diff --git a/go.mod b/go.mod new file mode 100644 index 000000000..f9ad23531 --- /dev/null +++ b/go.mod @@ -0,0 +1,34 @@ +module challenge2016 + +go 1.22.4 + +require github.com/gin-gonic/gin v1.10.0 + +require ( + github.com/bytedance/sonic v1.11.6 // indirect + github.com/bytedance/sonic/loader v0.1.1 // indirect + github.com/cloudwego/base64x v0.1.4 // indirect + github.com/cloudwego/iasm v0.2.0 // indirect + github.com/gabriel-vasile/mimetype v1.4.3 // 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.20.0 // indirect + github.com/goccy/go-json v0.10.2 // indirect + github.com/json-iterator/go v1.1.12 // indirect + github.com/klauspost/cpuid/v2 v2.2.7 // indirect + github.com/leodido/go-urn v1.4.0 // indirect + github.com/mattn/go-isatty v0.0.20 // 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.2.2 // indirect + github.com/twitchyliquid64/golang-asm v0.15.1 // indirect + github.com/ugorji/go/codec v1.2.12 // indirect + golang.org/x/arch v0.8.0 // indirect + golang.org/x/crypto v0.23.0 // indirect + golang.org/x/net v0.25.0 // indirect + golang.org/x/sys v0.20.0 // indirect + golang.org/x/text v0.15.0 // indirect + google.golang.org/protobuf v1.34.1 // indirect + gopkg.in/yaml.v3 v3.0.1 // indirect +) diff --git a/go.sum b/go.sum new file mode 100644 index 000000000..7f08abb20 --- /dev/null +++ b/go.sum @@ -0,0 +1,89 @@ +github.com/bytedance/sonic v1.11.6 h1:oUp34TzMlL+OY1OUWxHqsdkgC/Zfc85zGqw9siXjrc0= +github.com/bytedance/sonic v1.11.6/go.mod h1:LysEHSvpvDySVdC2f87zGWf6CIKJcAvqab1ZaiQtds4= +github.com/bytedance/sonic/loader v0.1.1 h1:c+e5Pt1k/cy5wMveRDyk2X4B9hF4g7an8N3zCYjJFNM= +github.com/bytedance/sonic/loader v0.1.1/go.mod h1:ncP89zfokxS5LZrJxl5z0UJcsk4M4yY2JpfqGeCtNLU= +github.com/cloudwego/base64x v0.1.4 h1:jwCgWpFanWmN8xoIUHa2rtzmkd5J2plF/dnLS6Xd/0Y= +github.com/cloudwego/base64x v0.1.4/go.mod h1:0zlkT4Wn5C6NdauXdJRhSKRlJvmclQ1hhJgA0rcu/8w= +github.com/cloudwego/iasm v0.2.0 h1:1KNIy1I1H9hNNFEEH3DVnI4UujN+1zjpuk6gwHLTssg= +github.com/cloudwego/iasm v0.2.0/go.mod h1:8rXZaNYT2n95jn+zTI1sDr+IgcD2GVs0nlbbQPiEFhY= +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.3 h1:in2uUcidCuFcDKtdcBxlR0rJ1+fsokWf+uqxgUFjbI0= +github.com/gabriel-vasile/mimetype v1.4.3/go.mod h1:d8uq/6HKRL6CGdk+aubisF/M5GcPfT7nKyLpA0lbSSk= +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.10.0 h1:nTuyha1TYqgedzytsKYqna+DfLos46nTv2ygFy86HFU= +github.com/gin-gonic/gin v1.10.0/go.mod h1:4PMNQiOhvDRa013RKVbsiNwoyezlm2rm0uX/T7kzp5Y= +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.20.0 h1:K9ISHbSaI0lyB2eWMPJo+kOS/FBExVwjEviJTixqxL8= +github.com/go-playground/validator/v10 v10.20.0/go.mod h1:dbuPbCMFw/DrkbEynArYaCwl3amGuJotoKCe95atGMM= +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/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/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.7 h1:ZWSB3igEs+d0qvnxR/ZBzXVmxkgt8DdzP6m9pfuVLDM= +github.com/klauspost/cpuid/v2 v2.2.7/go.mod h1:Lcz8mBdAVJIBVzewtcLocK12l3Y+JytZYpaMropDUws= +github.com/knz/go-libedit v1.10.1/go.mod h1:MZTVkCWyz0oBc7JOWP3wNAzd002ZbM/5hgShxwh4x8M= +github.com/leodido/go-urn v1.4.0 h1:WT9HwE9SGECu3lg4d/dIA+jxlljEa1/ffXKmRjqdmIQ= +github.com/leodido/go-urn v1.4.0/go.mod h1:bvxc+MVxLKB4z00jd1z+Dvzr47oO32F/QSNjSBOlFxI= +github.com/mattn/go-isatty v0.0.20 h1:xfD0iDuEKnDkl03q4limB+vH+GxLEtL/jb4xVJSWWEY= +github.com/mattn/go-isatty v0.0.20/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.2.2 h1:aYUidT7k73Pcl9nb2gScu7NSrKCSHIDE89b3+6Wq+LM= +github.com/pelletier/go-toml/v2 v2.2.2/go.mod h1:1t835xjRzz80PqgE6HHgN2JOsmgYu/h4qDAS4n929Rs= +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/objx v0.5.2/go.mod h1:FRsXN1f5AsAjCGJKqEizvkpNtU+EGNCLh3NxZ/8L+MA= +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.4/go.mod h1:sz/lmYIOXD/1dqDmKjjqLyZ2RngseejIcXlSw2iwfAo= +github.com/stretchr/testify v1.9.0 h1:HtqpIVDClZ4nwg75+f6Lvsy/wHu+3BoSGCbBAcpTsTg= +github.com/stretchr/testify v1.9.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY= +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.12 h1:9LC83zGrHhuUA9l16C9AHXAqEV/2wBQ4nkvumAE65EE= +github.com/ugorji/go/codec v1.2.12/go.mod h1:UNopzCgEMSXjBc6AOMqYvWC1ktqTAfzJZUZgYf6w6lg= +golang.org/x/arch v0.0.0-20210923205945-b76863e36670/go.mod h1:5om86z9Hs0C8fWVUuoMHwpExlXzs5Tkyp9hOrfG7pp8= +golang.org/x/arch v0.8.0 h1:3wRIsP3pM4yUptoR96otTUOXI367OS0+c9eeRi9doIc= +golang.org/x/arch v0.8.0/go.mod h1:FEVrYAQjsQXMVJ1nsMoVVXPZg6p2JE2mx8psSWTDQys= +golang.org/x/crypto v0.23.0 h1:dIJU/v2J8Mdglj/8rJ6UUOM3Zc9zLZxVZwwxMooUSAI= +golang.org/x/crypto v0.23.0/go.mod h1:CKFgDieR+mRhux2Lsu27y0fO304Db0wZe70UKqHu0v8= +golang.org/x/net v0.25.0 h1:d/OCCoBEUq33pjydKrGQhw7IlUPI2Oylr+8qLx49kac= +golang.org/x/net v0.25.0/go.mod h1:JkAGAh7GEvH74S6FOH42FLoXpXbE/aqXSrIQjXgsiwM= +golang.org/x/sys v0.5.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/sys v0.20.0 h1:Od9JTbYCk261bKm4M/mw7AklTlFYIa0bIp9BgSm1S8Y= +golang.org/x/sys v0.20.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA= +golang.org/x/text v0.15.0 h1:h1V/4gjBv8v9cjcR6+AR5+/cIYK5N/WAgiv4xlsEtAk= +golang.org/x/text v0.15.0/go.mod h1:18ZOQIKpY8NJVqYksKHtTdi31H5itFRjB5/qKTNYzSU= +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.34.1 h1:9ddQBjfCyZPOHPUiPxpYESBLc+T8P3E+Vo4IbKZgFWg= +google.golang.org/protobuf v1.34.1/go.mod h1:c6P6GXX6sHbq/GpV6MGZEdwhWPcYBgnhAHhKbcUYpos= +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= +nullprogram.com/x/optparse v1.0.0/go.mod h1:KdyPE+Igbe0jQUrVfMqDMeJQIJZEuyV7pjYmp6pbG50= +rsc.io/pdf v0.1.1/go.mod h1:n8OzWcQ6Sp37PL01nO98y4iUCRdTGarVfzxY20ICaU4= diff --git a/handlers/permission_handler.go b/handlers/permission_handler.go new file mode 100644 index 000000000..2c4217bf8 --- /dev/null +++ b/handlers/permission_handler.go @@ -0,0 +1,97 @@ +package handlers + +import ( + "challenge2016/models" + "challenge2016/services" + "challenge2016/utils" + "net/http" + "strings" + + "github.com/gin-gonic/gin" +) + +var distributors = make(map[string]*models.Distributor) + +func InitializePermissions(csvPath string) error { + regions, err := utils.ReadCitiesCSV(csvPath) + if err != nil { + return err + } + services.SetRegions(regions) + return nil +} + +type CheckRequest struct { + Distributor string `json:"distributor" binding:"required"` + Region string `json:"region" binding:"required"` +} + +type AssignRequest struct { + ParentDistributor string `json:"parent_distributor"` + NewDistributor string `json:"new_distributor" binding:"required"` + Permissions []models.PermissionRule `json:"permissions" binding:"required"` +} + +func CheckPermission(c *gin.Context) { + var req CheckRequest + if err := c.ShouldBindJSON(&req); err != nil { + c.JSON(http.StatusBadRequest, gin.H{"error": "Invalid request format"}) + return + } + + dist, exists := distributors[strings.ToUpper(req.Distributor)] + if !exists { + c.JSON(http.StatusNotFound, gin.H{"error": "Distributor not found"}) + return + } + + allowed := services.CheckPermission(dist, req.Region) + c.JSON(http.StatusOK, gin.H{"allowed": allowed}) +} + +func AssignPermission(c *gin.Context) { + var req AssignRequest + if err := c.ShouldBindJSON(&req); err != nil { + c.JSON(http.StatusBadRequest, gin.H{"error": "Invalid request format"}) + return + } + + + newDistName := strings.ToUpper(req.NewDistributor) + parentDistName := strings.ToUpper(req.ParentDistributor) + + if parentDistName == "" { + + for _, perm := range req.Permissions { + if _, valid := services.NormalizeRegion(perm.Region); !valid { + c.JSON(http.StatusBadRequest, gin.H{"error": "Invalid region in permissions: " + perm.Region}) + return + } + } + distributors[newDistName] = &models.Distributor{ + Name: newDistName, + Permissions: req.Permissions, + Parent: nil, + } + c.JSON(http.StatusCreated, gin.H{"message": "Distributor created"}) + return + } + + parent, exists := distributors[parentDistName] + if !exists { + c.JSON(http.StatusNotFound, gin.H{"error": "Parent distributor not found"}) + return + } + + if !services.CanAssignPermission(parent, req.Permissions) { + c.JSON(http.StatusForbidden, gin.H{"error": "Cannot assign permissions outside parent's scope"}) + return + } + + distributors[newDistName] = &models.Distributor{ + Name: newDistName, + Permissions: req.Permissions, + Parent: parent, + } + c.JSON(http.StatusCreated, gin.H{"message": "Sub-distributor created"}) +} \ No newline at end of file diff --git a/main.go b/main.go new file mode 100644 index 000000000..a96593e0b --- /dev/null +++ b/main.go @@ -0,0 +1,28 @@ +package main + +import ( + "challenge2016/handlers" + "github.com/gin-gonic/gin" +) + +func main() { + + // Initialize Gin router + router := gin.Default() + + // Load CSV data + err := handlers.InitializePermissions("cities.csv") + if err != nil { + panic("Failed to load cities data: " + err.Error()) + } + + // API endpoints + v1 := router.Group("/api/v1") + { + v1.POST("/permissions/check", handlers.CheckPermission) + v1.POST("/permissions/assign", handlers.AssignPermission) + } + + // Start the server + router.Run(":8080") +} \ No newline at end of file diff --git a/models/distribution.go b/models/distribution.go new file mode 100644 index 000000000..88bd3942c --- /dev/null +++ b/models/distribution.go @@ -0,0 +1,21 @@ +package models + +type Region struct { + CityCode string + ProvinceCode string + CountryCode string + CityName string + ProvinceName string + CountryName string +} + +type PermissionRule struct { + Action string // "INCLUDE" or "EXCLUDE" + Region string // Format: "CITY-PROVINCE-COUNTRY" or "PROVINCE-COUNTRY" or "COUNTRY" +} + +type Distributor struct { + Name string + Permissions []PermissionRule + Parent *Distributor // Nil for top-level distributor +} diff --git a/services/persmission_service.go b/services/persmission_service.go new file mode 100644 index 000000000..91f30e52d --- /dev/null +++ b/services/persmission_service.go @@ -0,0 +1,136 @@ +package services + +import ( + "challenge2016/models" + "strings" +) + +var regions []models.Region + +func SetRegions(r []models.Region) { + regions = r +} + +//Converts input to standardized uppercase names as per cities.csv file +func NormalizeRegion(region string) (string, bool) { + parts := strings.Split(strings.ToUpper(region), "-") + if len(parts) < 1 || len(parts) > 3 { + return "", false + } + + inputCountry := parts[len(parts)-1] + var inputProvince, inputCity string + if len(parts) >= 2 { + inputProvince = parts[len(parts)-2] + } + if len(parts) == 3 { + inputCity = parts[0] + } + + for _, r := range regions { + if matchesRegion(r, inputCity, inputProvince, inputCountry) { + if inputCity != "" { + return strings.ToUpper(r.CityName + "-" + r.ProvinceName + "-" + r.CountryName), true + } else if inputProvince != "" { + return strings.ToUpper(r.ProvinceName + "-" + r.CountryName), true + } + return strings.ToUpper(r.CountryName), true + } + } + return "", false +} + +func matchesRegion(r models.Region, city, province, country string) bool { + return (strings.ToUpper(r.CountryCode) == country || strings.ToUpper(r.CountryName) == country) && + (province == "" || strings.ToUpper(r.ProvinceCode) == province || strings.ToUpper(r.ProvinceName) == province) && + (city == "" || strings.ToUpper(r.CityCode) == city || strings.ToUpper(r.CityName) == city) +} + +func CheckPermission(distributor *models.Distributor, targetRegion string) bool { + normalizedRegion, valid := NormalizeRegion(targetRegion) + if !valid { + return false + } + + parts := strings.Split(normalizedRegion, "-") + country := parts[len(parts)-1] + var province, city string + if len(parts) >= 2 { + province = parts[len(parts)-2] + } + if len(parts) == 3 { + city = parts[0] + } + + current := distributor + for current != nil { + if !checkSingleDistributorPermission(current, city, province, country) { + return false + } + current = current.Parent + } + return true +} + +func checkSingleDistributorPermission(dist *models.Distributor, city, province, country string) bool { + hasInclude := false + isExcluded := false + + for _, rule := range dist.Permissions { + normalizedRule, valid := NormalizeRegion(rule.Region) + if !valid { + continue + } + + parts := strings.Split(normalizedRule, "-") + ruleCountry := parts[len(parts)-1] + var ruleProvince, ruleCity string + if len(parts) >= 2 { + ruleProvince = parts[len(parts)-2] + } + if len(parts) == 3 { + ruleCity = parts[0] + } + + if strings.ToUpper(rule.Action) == "INCLUDE" { + if matchesRegionExact(ruleCity, ruleProvince, ruleCountry, city, province, country) { + hasInclude = true + } + } else if strings.ToUpper(rule.Action) == "EXCLUDE" { + if matchesRegionExact(ruleCity, ruleProvince, ruleCountry, city, province, country) { + isExcluded = true + } + } + } + + return hasInclude && !isExcluded +} + +func matchesRegionExact(ruleCity, ruleProvince, ruleCountry, targetCity, targetProvince, targetCountry string) bool { + if ruleCountry != targetCountry { + return false + } + if ruleProvince != "" && ruleProvince != targetProvince { + return false + } + if ruleCity != "" && ruleCity != targetCity { + return false + } + return true +} + +func CanAssignPermission(parent *models.Distributor, childPermissions []models.PermissionRule) bool { + for _, childRule := range childPermissions { + normalizedRule, valid := NormalizeRegion(childRule.Region) + if !valid { + return false + } + + if strings.ToUpper(childRule.Action) == "INCLUDE" { + if !CheckPermission(parent, normalizedRule) { + return false + } + } + } + return true +} \ No newline at end of file diff --git a/test_commands.sh b/test_commands.sh new file mode 100644 index 000000000..2035269af --- /dev/null +++ b/test_commands.sh @@ -0,0 +1,103 @@ +#!/bin/bash + +# Test Commands for challenge2016 API + + +echo "1. Assign DISTRIBUTOR1 (Top-Level Distributor)" +curl -X POST http://localhost:8080/api/v1/permissions/assign \ +-H "Content-Type: application/json" \ +-d '{ + "new_distributor": "DISTRIBUTOR1", + "permissions": [ + {"action": "INCLUDE", "region": "IN"}, + {"action": "INCLUDE", "region": "United States"}, + {"action": "EXCLUDE", "region": "KARNATAKA-INDIA"}, + {"action": "EXCLUDE", "region": "CHENNAI-TAMIL NADU-INDIA"} + ] +}' +# Expected Output: {"message":"Distributor created"} + +echo -e "\n2. Assign DISTRIBUTOR2 (Sub-Distributor under DISTRIBUTOR1)" +curl -X POST http://localhost:8080/api/v1/permissions/assign \ +-H "Content-Type: application/json" \ +-d '{ + "parent_distributor": "DISTRIBUTOR1", + "new_distributor": "DISTRIBUTOR2", + "permissions": [ + {"action": "INCLUDE", "region": "INDIA"}, + {"action": "EXCLUDE", "region": "TAMIL NADU-INDIA"} + ] +}' +# Expected Output: {"message":"Sub-distributor created"} + +echo -e "\n3. Assign DISTRIBUTOR3 (Sub-Distributor under DISTRIBUTOR2)" +curl -X POST http://localhost:8080/api/v1/permissions/assign \ +-H "Content-Type: application/json" \ +-d '{ + "parent_distributor": "DISTRIBUTOR2", + "new_distributor": "DISTRIBUTOR3", + "permissions": [ + {"action": "INCLUDE", "region": "HYDERABAD-TELANGANA-INDIA"} + ] +}' +# Expected Output: {"message":"Sub-distributor created"} + +echo -e "\n4. Check Permission: Allowed Region (DISTRIBUTOR1)" +curl -X POST http://localhost:8080/api/v1/permissions/check \ +-H "Content-Type: application/json" \ +-d '{"distributor": "DISTRIBUTOR1", "region": "CHICAGO-ILLINOIS-UNITED STATES"}' +# Expected Output: {"allowed":true} + +echo -e "\n5. Check Permission: Excluded City (DISTRIBUTOR1)" +curl -X POST http://localhost:8080/api/v1/permissions/check \ +-H "Content-Type: application/json" \ +-d '{"distributor": "DISTRIBUTOR1", "region": "CHENNAI-TAMIL NADU-INDIA"}' +# Expected Output: {"allowed":false} + +echo -e "\n6. Check Permission: Excluded State (DISTRIBUTOR1)" +curl -X POST http://localhost:8080/api/v1/permissions/check \ +-H "Content-Type: application/json" \ +-d '{"distributor": "DISTRIBUTOR1", "region": "BANGALORE-KARNATAKA-INDIA"}' +# Expected Output: {"allowed":false} + +echo -e "\n7. Check Permission: Sub-Distributor Allowed (DISTRIBUTOR3)" +curl -X POST http://localhost:8080/api/v1/permissions/check \ +-H "Content-Type: application/json" \ +-d '{"distributor": "DISTRIBUTOR3", "region": "HYDERABAD-TELANGANA-INDIA"}' +# Expected Output: {"allowed":true} + +echo -e "\n8. Check Permission: Sub-Distributor Excluded (DISTRIBUTOR2)" +curl -X POST http://localhost:8080/api/v1/permissions/check \ +-H "Content-Type: application/json" \ +-d '{"distributor": "DISTRIBUTOR2", "region": "MADURAI-TAMIL NADU-INDIA"}' +# Expected Output: {"allowed":false} + +echo -e "\n9. Check Permission: Invalid Region (DISTRIBUTOR1)" +curl -X POST http://localhost:8080/api/v1/permissions/check \ +-H "Content-Type: application/json" \ +-d '{"distributor": "DISTRIBUTOR1", "region": "FAKE-CITY-FAKE-FAKE"}' +# Expected Output: {"allowed":false} + +echo -e "\n10. Invalid Sub-Distributor Assignment (DISTRIBUTOR2)" +curl -X POST http://localhost:8080/api/v1/permissions/assign \ +-H "Content-Type: application/json" \ +-d '{ + "parent_distributor": "DISTRIBUTOR2", + "new_distributor": "DISTRIBUTOR4", + "permissions": [ + {"action": "INCLUDE", "region": "CHINA"} + ] +}' +# Expected Output: {"error":"Cannot assign permissions outside parent's scope"} + +echo -e "\n11. Case Insensitivity Test (DISTRIBUTOR1)" +curl -X POST http://localhost:8080/api/v1/permissions/check \ +-H "Content-Type: application/json" \ +-d '{"distributor": "distributor1", "region": "chicago-illinois-united states"}' +# Expected Output: {"allowed":true} + +echo -e "\n12. Code Usage Test (DISTRIBUTOR1)" +curl -X POST http://localhost:8080/api/v1/permissions/check \ +-H "Content-Type: application/json" \ +-d '{"distributor": "DISTRIBUTOR1", "region": "CHICA-IL-US"}' +# Expected Output: {"allowed":true} \ No newline at end of file diff --git a/utils/csv_reader.go b/utils/csv_reader.go new file mode 100644 index 000000000..76640a4e1 --- /dev/null +++ b/utils/csv_reader.go @@ -0,0 +1,42 @@ +package utils + +import ( + "challenge2016/models" + "encoding/csv" + "os" + "strings" +) + +func ReadCitiesCSV(filename string) ([]models.Region, error) { + file, err := os.Open(filename) + if err != nil { + return nil, err + } + defer file.Close() + + reader := csv.NewReader(file) + records, err := reader.ReadAll() + if err != nil { + return nil, err + } + + var regions []models.Region + + // Skip header row + for _, record := range records[1:] { + if len(record) != 6 { + continue // Skip malformed rows + } + region := models.Region{ + CityCode: strings.TrimSpace(record[0]), + ProvinceCode: strings.TrimSpace(record[1]), + CountryCode: strings.TrimSpace(record[2]), + CityName: strings.TrimSpace(record[3]), + ProvinceName: strings.TrimSpace(record[4]), + CountryName: strings.TrimSpace(record[5]), + } + regions = append(regions, region) + } + + return regions, nil +} \ No newline at end of file