diff --git a/cost-calculator/.dockerignore b/cost-calculator/.dockerignore new file mode 100644 index 0000000..fb77c2e --- /dev/null +++ b/cost-calculator/.dockerignore @@ -0,0 +1,3 @@ +*.md +.git +.gitignore diff --git a/cost-calculator/.env.example b/cost-calculator/.env.example new file mode 100644 index 0000000..c7f968c --- /dev/null +++ b/cost-calculator/.env.example @@ -0,0 +1,6 @@ +DB_HOST=localhost +DB_PORT=3306 +DB_USER=something +DB_PASSWORD=something +DB_NAME=finleap +SERVER_PORT=8080 \ No newline at end of file diff --git a/cost-calculator/Dockerfile b/cost-calculator/Dockerfile new file mode 100644 index 0000000..168563b --- /dev/null +++ b/cost-calculator/Dockerfile @@ -0,0 +1,20 @@ +FROM golang:1.23-alpine AS builder + +WORKDIR /app + +COPY go.mod go.sum ./ +RUN go mod download + +COPY . . +RUN go mod tidy +RUN CGO_ENABLED=0 GOOS=linux go build -o cost-calculator ./cmd/main.go + +FROM alpine:latest + +WORKDIR /app + +COPY --from=builder /app/cost-calculator . + +EXPOSE 8080 + +CMD ["./cost-calculator"] diff --git a/cost-calculator/README.md b/cost-calculator/README.md new file mode 100644 index 0000000..a2835d5 --- /dev/null +++ b/cost-calculator/README.md @@ -0,0 +1,36 @@ +# Cost Calculator API + +Calculate EC2 instance costs by principalId from CloudTrail events. + +## Setup + +1. Copy `.env.example` to `.env` and configure your MySQL database settings +2. Install dependencies: `go mod tidy` +3. Run the application: `go run cmd/main.go` + +## API Endpoints + +- `GET /api/v1/costs/principals` - Get all principal costsls + +- `GET /api/v1/costs/principals/:id` - Get specific principal cost by ID +- `GET /api/v1/instances` - Get all EC2 instance costs + +## Example Response + +```json +[ + { + "principal_id": "AROA52BEGI3BHMK4NYVCU:ankitkarna@lftechnology.com", + "total_cost": 125.75, + "total_hours": 240.0, + "instances": 3 + } +] +``` + +## How it works + +The API joins `cloudtrail_events` and `ec2_costs` tables where: +- CloudTrail events contain `resourceIds` as JSON array with instance IDs +- EC2 costs contain actual cost and usage data per instance +- Costs are calculated per `principalId` who created/managed the resources \ No newline at end of file diff --git a/cost-calculator/cmd/main.go b/cost-calculator/cmd/main.go new file mode 100644 index 0000000..b1396bd --- /dev/null +++ b/cost-calculator/cmd/main.go @@ -0,0 +1,37 @@ +package main + +import ( + "cost-calculator/config" + "cost-calculator/database" + "cost-calculator/handlers" + "log" + + "github.com/gin-contrib/cors" + "github.com/gin-gonic/gin" +) + +func main() { + cfg := config.Load() + + database.Connect(cfg) + + r := gin.Default() + + r.Use(cors.New(cors.Config{ + AllowOrigins: []string{"http://localhost:5173", "http://127.0.0.1:5173", "http://3.108.187.54:5173"}, + AllowMethods: []string{"GET", "POST", "PUT", "DELETE", "OPTIONS"}, + AllowHeaders: []string{"Origin", "Content-Type", "Authorization"}, + AllowCredentials: true, + })) + + api := r.Group("/api/v1") + { + api.GET("/costs/principals", handlers.GetPrincipalCosts) + api.GET("/costs/principals/:id", handlers.GetPrincipalCostByID) + api.GET("/principals/resources", handlers.GetPrincipalResourceMapping) + api.GET("/instances", handlers.GetInstanceCosts) + } + + log.Printf("Server starting on port %s", cfg.ServerPort) + r.Run(":" + cfg.ServerPort) +} diff --git a/cost-calculator/config/config.go b/cost-calculator/config/config.go new file mode 100644 index 0000000..26cdcda --- /dev/null +++ b/cost-calculator/config/config.go @@ -0,0 +1,55 @@ +package config + +import ( + "bufio" + "os" + "strings" +) + +type Config struct { + DBHost string + DBPort string + DBUser string + DBPassword string + DBName string + ServerPort string +} + +func Load() *Config { + loadEnvFile(".env") + return &Config{ + DBHost: getEnv("DB_HOST", "localhost"), + DBPort: getEnv("DB_PORT", "3306"), + DBUser: getEnv("DB_USER", "root"), + DBPassword: getEnv("DB_PASSWORD", ""), + DBName: getEnv("DB_NAME", "cost_db"), + ServerPort: getEnv("SERVER_PORT", "8080"), + } +} + +func loadEnvFile(filename string) { + file, err := os.Open(filename) + if err != nil { + return + } + defer file.Close() + + scanner := bufio.NewScanner(file) + for scanner.Scan() { + line := strings.TrimSpace(scanner.Text()) + if line == "" || strings.HasPrefix(line, "#") { + continue + } + parts := strings.SplitN(line, "=", 2) + if len(parts) == 2 { + os.Setenv(parts[0], parts[1]) + } + } +} + +func getEnv(key, defaultValue string) string { + if value := os.Getenv(key); value != "" { + return value + } + return defaultValue +} diff --git a/cost-calculator/database/database.go b/cost-calculator/database/database.go new file mode 100644 index 0000000..a59096b --- /dev/null +++ b/cost-calculator/database/database.go @@ -0,0 +1,30 @@ +package database + +import ( + "cost-calculator/config" + "fmt" + "log" + + "gorm.io/driver/mysql" + "gorm.io/gorm" +) + +var DB *gorm.DB + +func Connect(cfg *config.Config) { + dsn := fmt.Sprintf("%s:%s@tcp(%s:%s)/%s?charset=utf8mb4&parseTime=True&loc=Local", + cfg.DBUser, cfg.DBPassword, cfg.DBHost, cfg.DBPort, cfg.DBName) + + var err error + DB, err = gorm.Open(mysql.Open(dsn), &gorm.Config{}) + if err != nil { + log.Fatal("Failed to connect to database:", err) + } + + // Tables already exist, no migration needed + log.Println("Database connected successfully") +} + +func GetDB() *gorm.DB { + return DB +} \ No newline at end of file diff --git a/cost-calculator/docker-compose.yml b/cost-calculator/docker-compose.yml new file mode 100644 index 0000000..ebfa116 --- /dev/null +++ b/cost-calculator/docker-compose.yml @@ -0,0 +1,37 @@ +version: '3.8' + +services: + app: + build: . + ports: + - "8080:8080" + environment: + - DB_HOST=${DB_HOST} + - DB_PORT=${DB_PORT} + - DB_USER=${DB_USER} + - DB_PASSWORD=${DB_PASSWORD} + - DB_NAME=${DB_NAME} + - SERVER_PORT=8080 + depends_on: + - db + networks: + - cost-calculator-network + + db: + image: mysql:8.0 + environment: + - MYSQL_ROOT_PASSWORD=${DB_PASSWORD} + - MYSQL_DATABASE=${DB_NAME} + ports: + - "3306:3306" + volumes: + - mysql-data:/var/lib/mysql + networks: + - cost-calculator-network + +networks: + cost-calculator-network: + driver: bridge + +volumes: + mysql-data: diff --git a/cost-calculator/go.mod b/cost-calculator/go.mod new file mode 100644 index 0000000..4ef3db3 --- /dev/null +++ b/cost-calculator/go.mod @@ -0,0 +1,39 @@ +module cost-calculator + +go 1.21 + +require ( + github.com/gin-gonic/gin v1.9.1 + gorm.io/driver/mysql v1.5.2 + gorm.io/gorm v1.25.5 +) + +require ( + github.com/bytedance/sonic v1.9.1 // indirect + github.com/chenzhuoyu/base64x v0.0.0-20221115062448-fe3a3abad311 // indirect + github.com/gabriel-vasile/mimetype v1.4.2 // indirect + github.com/gin-contrib/sse v0.1.0 // indirect + github.com/go-playground/locales v0.14.1 // indirect + github.com/go-playground/universal-translator v0.18.1 // indirect + github.com/go-playground/validator/v10 v10.14.0 // indirect + github.com/go-sql-driver/mysql v1.7.1 // indirect + github.com/goccy/go-json v0.10.2 // indirect + github.com/jinzhu/inflection v1.0.0 // indirect + github.com/jinzhu/now v1.1.5 // indirect + github.com/json-iterator/go v1.1.12 // indirect + github.com/klauspost/cpuid/v2 v2.2.4 // indirect + github.com/leodido/go-urn v1.2.4 // indirect + github.com/mattn/go-isatty v0.0.19 // indirect + github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd // indirect + github.com/modern-go/reflect2 v1.0.2 // indirect + github.com/pelletier/go-toml/v2 v2.0.8 // indirect + github.com/twitchyliquid64/golang-asm v0.15.1 // indirect + github.com/ugorji/go/codec v1.2.11 // indirect + golang.org/x/arch v0.3.0 // indirect + golang.org/x/crypto v0.9.0 // indirect + golang.org/x/net v0.10.0 // indirect + golang.org/x/sys v0.8.0 // indirect + golang.org/x/text v0.9.0 // indirect + google.golang.org/protobuf v1.30.0 // indirect + gopkg.in/yaml.v3 v3.0.1 // indirect +) diff --git a/cost-calculator/go.sum b/cost-calculator/go.sum new file mode 100644 index 0000000..f13461d --- /dev/null +++ b/cost-calculator/go.sum @@ -0,0 +1,98 @@ +github.com/bytedance/sonic v1.5.0/go.mod h1:ED5hyg4y6t3/9Ku1R6dU/4KyJ48DZ4jPhfY1O2AihPM= +github.com/bytedance/sonic v1.9.1 h1:6iJ6NqdoxCDr6mbY8h18oSO+cShGSMRGCEo7F2h0x8s= +github.com/bytedance/sonic v1.9.1/go.mod h1:i736AoUSYt75HyZLoJW9ERYxcy6eaN6h4BZXU064P/U= +github.com/chenzhuoyu/base64x v0.0.0-20211019084208-fb5309c8db06/go.mod h1:DH46F32mSOjUmXrMHnKwZdA8wcEefY7UVqBKYGjpdQY= +github.com/chenzhuoyu/base64x v0.0.0-20221115062448-fe3a3abad311 h1:qSGYFH7+jGhDF8vLC+iwCD4WpbV1EBDSzWkJODFLams= +github.com/chenzhuoyu/base64x v0.0.0-20221115062448-fe3a3abad311/go.mod h1:b583jCggY9gE99b6G5LEC39OIiVsWj+R97kbl5odCEk= +github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= +github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c= +github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= +github.com/gabriel-vasile/mimetype v1.4.2 h1:w5qFW6JKBz9Y393Y4q372O9A7cUSequkh1Q7OhCmWKU= +github.com/gabriel-vasile/mimetype v1.4.2/go.mod h1:zApsH/mKG4w07erKIaJPFiX0Tsq9BFQgN3qGY5GnNgA= +github.com/gin-contrib/sse v0.1.0 h1:Y/yl/+YNO8GZSjAhjMsSuLt29uWRFHdHYUb5lYOV9qE= +github.com/gin-contrib/sse v0.1.0/go.mod h1:RHrZQHXnP2xjPF+u1gW/2HnVO7nvIa9PG3Gm+fLHvGI= +github.com/gin-gonic/gin v1.9.1 h1:4idEAncQnU5cB7BeOkPtxjfCSye0AAm1R0RVIqJ+Jmg= +github.com/gin-gonic/gin v1.9.1/go.mod h1:hPrL7YrpYKXt5YId3A/Tnip5kqbEAP+KLuI3SUcPTeU= +github.com/go-playground/assert/v2 v2.2.0 h1:JvknZsQTYeFEAhQwI4qEt9cyV5ONwRHC+lYKSsYSR8s= +github.com/go-playground/assert/v2 v2.2.0/go.mod h1:VDjEfimB/XKnb+ZQfWdccd7VUvScMdVu0Titje2rxJ4= +github.com/go-playground/locales v0.14.1 h1:EWaQ/wswjilfKLTECiXz7Rh+3BjFhfDFKv/oXslEjJA= +github.com/go-playground/locales v0.14.1/go.mod h1:hxrqLVvrK65+Rwrd5Fc6F2O76J/NuW9t0sjnWqG1slY= +github.com/go-playground/universal-translator v0.18.1 h1:Bcnm0ZwsGyWbCzImXv+pAJnYK9S473LQFuzCbDbfSFY= +github.com/go-playground/universal-translator v0.18.1/go.mod h1:xekY+UJKNuX9WP91TpwSH2VMlDf28Uj24BCp08ZFTUY= +github.com/go-playground/validator/v10 v10.14.0 h1:vgvQWe3XCz3gIeFDm/HnTIbj6UGmg/+t63MyGU2n5js= +github.com/go-playground/validator/v10 v10.14.0/go.mod h1:9iXMNT7sEkjXb0I+enO7QXmzG6QCsPWY4zveKFVRSyU= +github.com/go-sql-driver/mysql v1.7.0/go.mod h1:OXbVy3sEdcQ2Doequ6Z5BW6fXNQTmx+9S1MCJN5yJMI= +github.com/go-sql-driver/mysql v1.7.1 h1:lUIinVbN1DY0xBg0eMOzmmtGoHwWBbvnWubQUrtU8EI= +github.com/go-sql-driver/mysql v1.7.1/go.mod h1:OXbVy3sEdcQ2Doequ6Z5BW6fXNQTmx+9S1MCJN5yJMI= +github.com/goccy/go-json v0.10.2 h1:CrxCmQqYDkv1z7lO7Wbh2HN93uovUHgrECaO5ZrCXAU= +github.com/goccy/go-json v0.10.2/go.mod h1:6MelG93GURQebXPDq3khkgXZkazVtN9CRI+MGFi0w8I= +github.com/golang/protobuf v1.5.0/go.mod h1:FsONVRAS9T7sI+LIUmWTfcYkHO4aIWwzhcaSAoJOfIk= +github.com/google/go-cmp v0.5.5 h1:Khx7svrCpmxxtHBq5j2mp/xVjsi8hQMfNLvJFAlrGgU= +github.com/google/go-cmp v0.5.5/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE= +github.com/google/gofuzz v1.0.0/go.mod h1:dBl0BpW6vV/+mYPU4Po3pmUjxk6FQPldtuIdl/M65Eg= +github.com/jinzhu/inflection v1.0.0 h1:K317FqzuhWc8YvSVlFMCCUb36O/S9MCKRDI7QkRKD/E= +github.com/jinzhu/inflection v1.0.0/go.mod h1:h+uFLlag+Qp1Va5pdKtLDYj+kHp5pxUVkryuEj+Srlc= +github.com/jinzhu/now v1.1.5 h1:/o9tlHleP7gOFmsnYNz3RGnqzefHA47wQpKrrdTIwXQ= +github.com/jinzhu/now v1.1.5/go.mod h1:d3SSVoowX0Lcu0IBviAWJpolVfI5UJVZZ7cO71lE/z8= +github.com/json-iterator/go v1.1.12 h1:PV8peI4a0ysnczrg+LtxykD8LfKY9ML6u2jnxaEnrnM= +github.com/json-iterator/go v1.1.12/go.mod h1:e30LSqwooZae/UwlEbR2852Gd8hjQvJoHmT4TnhNGBo= +github.com/klauspost/cpuid/v2 v2.0.9/go.mod h1:FInQzS24/EEf25PyTYn52gqo7WaD8xa0213Md/qVLRg= +github.com/klauspost/cpuid/v2 v2.2.4 h1:acbojRNwl3o09bUq+yDCtZFc1aiwaAAxtcn8YkZXnvk= +github.com/klauspost/cpuid/v2 v2.2.4/go.mod h1:RVVoqg1df56z8g3pUjL/3lE5UfnlrJX8tyFgg4nqhuY= +github.com/leodido/go-urn v1.2.4 h1:XlAE/cm/ms7TE/VMVoduSpNBoyc2dOxHs5MZSwAN63Q= +github.com/leodido/go-urn v1.2.4/go.mod h1:7ZrI8mTSeBSHl/UaRyKQW1qZeMgak41ANeCNaVckg+4= +github.com/mattn/go-isatty v0.0.19 h1:JITubQf0MOLdlGRuRq+jtsDlekdYPia9ZFsB8h/APPA= +github.com/mattn/go-isatty v0.0.19/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y= +github.com/modern-go/concurrent v0.0.0-20180228061459-e0a39a4cb421/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q= +github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd h1:TRLaZ9cD/w8PVh93nsPXa1VrQ6jlwL5oN8l14QlcNfg= +github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q= +github.com/modern-go/reflect2 v1.0.2 h1:xBagoLtFs94CBntxluKeaWgTMpvLxC4ur3nMaC9Gz0M= +github.com/modern-go/reflect2 v1.0.2/go.mod h1:yWuevngMOJpCy52FWWMvUC8ws7m/LJsjYzDa0/r8luk= +github.com/pelletier/go-toml/v2 v2.0.8 h1:0ctb6s9mE31h0/lhu+J6OPmVeDxJn+kYnJc2jZR9tGQ= +github.com/pelletier/go-toml/v2 v2.0.8/go.mod h1:vuYfssBdrU2XDZ9bYydBu6t+6a6PYNcZljzZR9VXg+4= +github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM= +github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4= +github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME= +github.com/stretchr/objx v0.4.0/go.mod h1:YvHI0jy2hoMjB+UWwv71VJQ9isScKT/TqJzVSSt89Yw= +github.com/stretchr/objx v0.5.0/go.mod h1:Yh+to48EsGEfYuaHDzXPcE3xhTkx73EhmCGUpEOglKo= +github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI= +github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg= +github.com/stretchr/testify v1.7.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg= +github.com/stretchr/testify v1.8.0/go.mod h1:yNjHg4UonilssWZ8iaSj1OCr/vHnekPRkoO+kdMU+MU= +github.com/stretchr/testify v1.8.1/go.mod h1:w2LPCIKwWwSfY2zedu0+kehJoqGctiVI29o6fzry7u4= +github.com/stretchr/testify v1.8.2/go.mod h1:w2LPCIKwWwSfY2zedu0+kehJoqGctiVI29o6fzry7u4= +github.com/stretchr/testify v1.8.3 h1:RP3t2pwF7cMEbC1dqtB6poj3niw/9gnV4Cjg5oW5gtY= +github.com/stretchr/testify v1.8.3/go.mod h1:sz/lmYIOXD/1dqDmKjjqLyZ2RngseejIcXlSw2iwfAo= +github.com/twitchyliquid64/golang-asm v0.15.1 h1:SU5vSMR7hnwNxj24w34ZyCi/FmDZTkS4MhqMhdFk5YI= +github.com/twitchyliquid64/golang-asm v0.15.1/go.mod h1:a1lVb/DtPvCB8fslRZhAngC2+aY1QWCk3Cedj/Gdt08= +github.com/ugorji/go/codec v1.2.11 h1:BMaWp1Bb6fHwEtbplGBGJ498wD+LKlNSl25MjdZY4dU= +github.com/ugorji/go/codec v1.2.11/go.mod h1:UNopzCgEMSXjBc6AOMqYvWC1ktqTAfzJZUZgYf6w6lg= +golang.org/x/arch v0.0.0-20210923205945-b76863e36670/go.mod h1:5om86z9Hs0C8fWVUuoMHwpExlXzs5Tkyp9hOrfG7pp8= +golang.org/x/arch v0.3.0 h1:02VY4/ZcO/gBOH6PUaoiptASxtXU10jazRCP865E97k= +golang.org/x/arch v0.3.0/go.mod h1:5om86z9Hs0C8fWVUuoMHwpExlXzs5Tkyp9hOrfG7pp8= +golang.org/x/crypto v0.9.0 h1:LF6fAI+IutBocDJ2OT0Q1g8plpYljMZ4+lty+dsqw3g= +golang.org/x/crypto v0.9.0/go.mod h1:yrmDGqONDYtNj3tH8X9dzUun2m2lzPa9ngI6/RUPGR0= +golang.org/x/net v0.10.0 h1:X2//UzNDwYmtCLn7To6G58Wr6f5ahEAQgKNzv9Y951M= +golang.org/x/net v0.10.0/go.mod h1:0qNGK6F8kojg2nk9dLZ2mShWaEBan6FAoqfSigmmuDg= +golang.org/x/sys v0.0.0-20220704084225-05e143d24a9e/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/sys v0.8.0 h1:EBmGv8NaZBZTWvrbjNoL6HVt+IVy3QDQpJs7VRIw3tU= +golang.org/x/sys v0.8.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/text v0.9.0 h1:2sjJmO8cDvYveuX97RDLsxlyUxLl+GHoLxBiRdHllBE= +golang.org/x/text v0.9.0/go.mod h1:e1OnstbJyHTd6l/uOt8jFFHp6TRDWZR/bV3emEE/zU8= +golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543 h1:E7g+9GITq07hpfrRu66IVDexMakfv52eLZ2CXBWiKr4= +golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0= +google.golang.org/protobuf v1.26.0-rc.1/go.mod h1:jlhhOSvTdKEhbULTjvd4ARK9grFBp09yW+WbY/TyQbw= +google.golang.org/protobuf v1.30.0 h1:kPPoIgf3TsEvrm0PFe15JQ+570QVxYzEvvHqChK+cng= +google.golang.org/protobuf v1.30.0/go.mod h1:HV8QOd/L58Z+nl8r43ehVNZIU/HEI6OcFqwMG9pJV4I= +gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405 h1:yhCVgyC4o1eVCa2tZl7eS0r+SDo693bJlVdllGtEeKM= +gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0= +gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM= +gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA= +gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM= +gorm.io/driver/mysql v1.5.2 h1:QC2HRskSE75wBuOxe0+iCkyJZ+RqpudsQtqkp+IMuXs= +gorm.io/driver/mysql v1.5.2/go.mod h1:pQLhh1Ut/WUAySdTHwBpBv6+JKcj+ua4ZFx1QQTBzb8= +gorm.io/gorm v1.25.2-0.20230530020048-26663ab9bf55/go.mod h1:L4uxeKpfBml98NYqVqwAdmV1a2nBtAec/cf3fpucW/k= +gorm.io/gorm v1.25.5 h1:zR9lOiiYf09VNh5Q1gphfyia1JpiClIWG9hQaxB/mls= +gorm.io/gorm v1.25.5/go.mod h1:hbnx/Oo0ChWMn1BIhpy1oYozzpM15i4YPuHDmfYtwg8= +rsc.io/pdf v0.1.1/go.mod h1:n8OzWcQ6Sp37PL01nO98y4iUCRdTGarVfzxY20ICaU4= diff --git a/cost-calculator/handlers/cost.go b/cost-calculator/handlers/cost.go new file mode 100644 index 0000000..50b93e2 --- /dev/null +++ b/cost-calculator/handlers/cost.go @@ -0,0 +1,125 @@ +package handlers + +import ( + "cost-calculator/database" + "cost-calculator/models" + "net/http" + + "github.com/gin-gonic/gin" +) + +func GetPrincipalCosts(c *gin.Context) { + var costs []struct { + PrincipalID string `json:"principal_id"` + ResourceIDs string `json:"resource_ids"` + } + + query := ` + SELECT DISTINCT + ce.principalId as principal_id, + ce.resourceIds as resource_ids + FROM cloudtrail_events ce + WHERE ce.principalId IS NOT NULL + AND ce.principalId != '' + AND ce.resourceIds IS NOT NULL + AND ce.resourceIds != '' + ORDER BY ce.principalId + ` + + if err := database.GetDB().Raw(query).Scan(&costs).Error; err != nil { + c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()}) + return + } + + c.JSON(http.StatusOK, costs) +} + +func GetPrincipalCostByID(c *gin.Context) { + principalID := c.Param("id") + + if principalID == "" { + c.JSON(http.StatusBadRequest, gin.H{"error": "Principal ID is required"}) + return + } + + var exists bool + if err := database.GetDB().Raw("SELECT EXISTS(SELECT 1 FROM cloudtrail_events WHERE principalId = ?)", principalID).Scan(&exists).Error; err != nil { + c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()}) + return + } + + if !exists { + c.JSON(http.StatusNotFound, gin.H{"error": "Principal ID not found"}) + return + } + + var cost struct { + PrincipalID string `json:"principal_id"` + TotalCost float64 `json:"total_cost"` + TotalHours float64 `json:"total_hours"` + InstanceIDs string `json:"instance_ids"` + CreatedAt string `json:"created_at"` + } + + query := ` + SELECT + ce.principalId as principal_id, + COALESCE(SUM(ec.cost_usd), 0) as total_cost, + COALESCE(SUM(ec.usage_hours), 0) as total_hours, + GROUP_CONCAT(DISTINCT ec.instance_id) as instance_ids, + MIN(ce.created_at) as created_at + FROM cloudtrail_events ce + LEFT JOIN ec2_costs ec ON JSON_CONTAINS(ce.resourceIds, CONCAT('"', ec.instance_id, '"')) + WHERE ce.principalId = ? + GROUP BY ce.principalId + ` + + if err := database.GetDB().Raw(query, principalID).Scan(&cost).Error; err != nil { + c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()}) + return + } + + if cost.PrincipalID == "" { + c.JSON(http.StatusNotFound, gin.H{"error": "Principal not found"}) + return + } + + c.JSON(http.StatusOK, cost) +} + +func GetPrincipalResourceMapping(c *gin.Context) { + var results []struct { + PrincipalID string `json:"principal_id"` + ResourceIDs string `json:"resource_ids"` + } + + query := ` + SELECT DISTINCT + ce.principalId as principal_id, + ce.resourceIds as resource_ids + FROM cloudtrail_events ce + WHERE ce.principalId IS NOT NULL + AND ce.principalId != '' + AND ce.resourceIds IS NOT NULL + AND ce.resourceIds != '' + ORDER BY ce.principalId + ` + + if err := database.GetDB().Raw(query).Scan(&results).Error; err != nil { + c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()}) + return + } + + c.JSON(http.StatusOK, results) +} + +func GetInstanceCosts(c *gin.Context) { + var costs []models.EC2Cost + + if err := database.GetDB().Find(&costs).Error; err != nil { + c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()}) + return + } + + c.JSON(http.StatusOK, costs) +} \ No newline at end of file diff --git a/cost-calculator/models/models.go b/cost-calculator/models/models.go new file mode 100644 index 0000000..11d5be6 --- /dev/null +++ b/cost-calculator/models/models.go @@ -0,0 +1,41 @@ +package models + +import "time" + +type CloudtrailEvent struct { + EventID string `json:"event_id" gorm:"column:eventID;primaryKey"` + AccountID string `json:"account_id" gorm:"column:accountId"` + EventTime time.Time `json:"event_time" gorm:"column:eventTime"` + PrincipalID string `json:"principal_id" gorm:"column:principalId"` + Username string `json:"username" gorm:"column:username"` + EventSource string `json:"event_source" gorm:"column:eventSource"` + EventName string `json:"event_name" gorm:"column:eventName"` + AwsRegion string `json:"aws_region" gorm:"column:awsRegion"` + ResourceIDs string `json:"resource_ids" gorm:"column:resourceIds"` + CreatedAt time.Time `json:"created_at" gorm:"column:created_at"` +} + +func (CloudtrailEvent) TableName() string { + return "cloudtrail_events" +} + +type EC2Cost struct { + Date string `json:"date" gorm:"column:date;primaryKey"` + InstanceID string `json:"instance_id" gorm:"column:instance_id;primaryKey"` + InstanceType string `json:"instance_type" gorm:"column:instance_type"` + CostUSD float64 `json:"cost_usd" gorm:"column:cost_usd"` + UsageHours float64 `json:"usage_hours" gorm:"column:usage_hours"` + CreatedAt time.Time `json:"created_at" gorm:"column:created_at"` + UpdatedAt time.Time `json:"updated_at" gorm:"column:updated_at"` +} + +func (EC2Cost) TableName() string { + return "ec2_costs" +} + +type AccountCost struct { + AccountID string `json:"account_id"` + TotalCost float64 `json:"total_cost"` + TotalHours float64 `json:"total_hours"` + Instances int `json:"instances"` +} \ No newline at end of file