diff --git a/.gitignore b/.gitignore new file mode 100644 index 0000000..723ef36 --- /dev/null +++ b/.gitignore @@ -0,0 +1 @@ +.idea \ No newline at end of file diff --git a/api.go b/api.go new file mode 100644 index 0000000..2419e5c --- /dev/null +++ b/api.go @@ -0,0 +1,82 @@ +package main + +import ( + "errors" + "github.com/gin-gonic/gin" + "github.com/klass-lk/ginboot/types" + "net/http" + "strconv" + "strings" +) + +func GetAuthContext(c *gin.Context) (types.AuthContext, error) { + userId, exists := c.Get("user_id") + if !exists { + c.AbortWithStatus(http.StatusUnauthorized) + return types.AuthContext{}, errors.New("operation not permitted") + } + role, exists := c.Get("role") + if !exists { + c.AbortWithStatus(http.StatusUnauthorized) + return types.AuthContext{}, errors.New("operation not permitted") + } + return types.AuthContext{ + UserId: userId.(string), + Role: role.(string), + }, nil +} + +func BuildAuthRequestContext[T interface{}](c *gin.Context) (T, types.AuthContext, error) { + request, err := BuildRequest[T](c) + if err != nil { + return request, types.AuthContext{}, err + } + authContext, err := GetAuthContext(c) + if err != nil { + c.AbortWithStatus(http.StatusBadRequest) + return request, types.AuthContext{}, err + } + return request, authContext, nil +} + +func BuildPageRequest(c *gin.Context) types.PageRequest { + pageString := c.DefaultQuery("page", "1") + sizeString := c.DefaultQuery("size", "10") + sortString := c.DefaultQuery("sort", "_id,asc") + page, err := strconv.ParseInt(pageString, 10, 64) + if err != nil { + c.AbortWithStatus(http.StatusBadRequest) + } + size, err := strconv.ParseInt(sizeString, 10, 64) + if err != nil { + c.AbortWithStatus(http.StatusBadRequest) + } + sortSplit := strings.Split(sortString, ",") + var sort types.SortField + if len(sortSplit) > 1 { + direction := 1 + if sortSplit[1] == "desc" { + direction = -1 + } + sort = types.SortField{ + Field: sortSplit[0], + Direction: direction, + } + } else { + sort = types.SortField{ + Field: sortSplit[0], + Direction: 1, + } + } + + return types.PageRequest{Page: int(page), Size: int(size), Sort: sort} +} + +func BuildRequest[T interface{}](c *gin.Context) (T, error) { + var request T + if c.ShouldBindJSON(&request) != nil { + c.AbortWithStatus(http.StatusBadRequest) + return request, errors.New("bad request") + } + return request, nil +} diff --git a/errors/error_lib.go b/errors/error_lib.go new file mode 100644 index 0000000..f699d77 --- /dev/null +++ b/errors/error_lib.go @@ -0,0 +1,52 @@ +package errors + +import ( + "errors" + "fmt" + "github.com/gin-gonic/gin" + "net/http" +) + +type ApiError struct { + ErrorCode string `json:"error_code"` + Message string `json:"message"` +} + +func (e ApiError) New(messages ...string) ApiError { + args := make([]any, len(messages)) + for i, msg := range messages { + args[i] = msg + } + + message := fmt.Sprintf(e.Message, args...) + return ApiError{ + ErrorCode: e.ErrorCode, + Message: message, + } +} + +func (e ApiError) Error() string { + return fmt.Sprintf("%s: %s", e.ErrorCode, e.Message) +} + +type ErrorResponse struct { + ErrorCode string `json:"error_code"` + Message string `json:"message"` +} + +func SendError(c *gin.Context, err error) { + var customErr ApiError + if errors.As(err, &customErr) { + c.JSON(http.StatusBadRequest, gin.H{ + "error_code": customErr.ErrorCode, + "message": customErr.Message, + }) + return + } + // Handle other types of errors here + c.JSON(http.StatusInternalServerError, gin.H{ + "error_code": "Internal Server Error", + "message": "An unknown error occurred", + }) + return +} diff --git a/go.mod b/go.mod new file mode 100644 index 0000000..aff5791 --- /dev/null +++ b/go.mod @@ -0,0 +1,47 @@ +module github.com/klass-lk/ginboot + +go 1.22.0 + +require ( + github.com/dgrijalva/jwt-go v3.2.0+incompatible + github.com/gin-gonic/gin v1.10.0 + github.com/google/uuid v1.6.0 + go.mongodb.org/mongo-driver v1.17.1 +) + +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/golang/snappy v0.0.4 // indirect + github.com/json-iterator/go v1.1.12 // indirect + github.com/klauspost/compress v1.13.6 // 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/montanaflynn/stats v0.7.1 // 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 + github.com/xdg-go/pbkdf2 v1.0.0 // indirect + github.com/xdg-go/scram v1.1.2 // indirect + github.com/xdg-go/stringprep v1.0.4 // indirect + github.com/youmark/pkcs8 v0.0.0-20240726163527-a2c0da244d78 // indirect + golang.org/x/arch v0.8.0 // indirect + golang.org/x/crypto v0.26.0 // indirect + golang.org/x/net v0.25.0 // indirect + golang.org/x/sync v0.8.0 // indirect + golang.org/x/sys v0.23.0 // indirect + golang.org/x/text v0.17.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 0000000..ee19fde --- /dev/null +++ b/go.sum @@ -0,0 +1,133 @@ +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/dgrijalva/jwt-go v3.2.0+incompatible h1:7qlOGliEKZXTDg6OTjfoBKDXWrumCAMpl/TFQ4/5kLM= +github.com/dgrijalva/jwt-go v3.2.0+incompatible/go.mod h1:E3ru+11k8xSBh+hMPgOLZmtrrCbhqsmaPHjLKYnJCaQ= +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/golang/snappy v0.0.4 h1:yAGX7huGHXlcLOEtBnF4w7FQwA26wojNCwOYAEhLjQM= +github.com/golang/snappy v0.0.4/go.mod h1:/XxbfmMg8lxefKM7IXC3fBNl/7bRcc72aCRzEWrmP2Q= +github.com/google/go-cmp v0.6.0 h1:ofyhxvXcZhMsU5ulbFiLKl/XBFqE1GSq7atu8tAmTRI= +github.com/google/go-cmp v0.6.0/go.mod h1:17dUlkBOakJ0+DkrSSNjCkIjxS6bF9zb3elmeNGIjoY= +github.com/google/gofuzz v1.0.0/go.mod h1:dBl0BpW6vV/+mYPU4Po3pmUjxk6FQPldtuIdl/M65Eg= +github.com/google/uuid v1.6.0 h1:NIvaJDMOsjHA8n1jAhLSgzrAzy1Hgr+hNrb57e+94F0= +github.com/google/uuid v1.6.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo= +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/compress v1.13.6 h1:P76CopJELS0TiO2mebmnzgWaajssP/EszplttgQxcgc= +github.com/klauspost/compress v1.13.6/go.mod h1:/3/Vjq9QcHkK5uEr5lBEmyoZ1iFhe47etQ6QUkpK6sk= +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/montanaflynn/stats v0.7.1 h1:etflOAAHORrCC44V+aR6Ftzort912ZU+YLiSTuV8eaE= +github.com/montanaflynn/stats v0.7.1/go.mod h1:etXPPgVO6n31NxCd9KQUMvCM+ve0ruNzt6R8Bnaayow= +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= +github.com/xdg-go/pbkdf2 v1.0.0 h1:Su7DPu48wXMwC3bs7MCNG+z4FhcyEuz5dlvchbq0B0c= +github.com/xdg-go/pbkdf2 v1.0.0/go.mod h1:jrpuAogTd400dnrH08LKmI/xc1MbPOebTwRqcT5RDeI= +github.com/xdg-go/scram v1.1.2 h1:FHX5I5B4i4hKRVRBCFRxq1iQRej7WO3hhBuJf+UUySY= +github.com/xdg-go/scram v1.1.2/go.mod h1:RT/sEzTbU5y00aCK8UOx6R7YryM0iF1N2MOmC3kKLN4= +github.com/xdg-go/stringprep v1.0.4 h1:XLI/Ng3O1Atzq0oBs3TWm+5ZVgkq2aqdlvP9JtoZ6c8= +github.com/xdg-go/stringprep v1.0.4/go.mod h1:mPGuuIYwz7CmR2bT9j4GbQqutWS1zV24gijq1dTyGkM= +github.com/youmark/pkcs8 v0.0.0-20240726163527-a2c0da244d78 h1:ilQV1hzziu+LLM3zUTJ0trRztfwgjqKnBWNtSRkbmwM= +github.com/youmark/pkcs8 v0.0.0-20240726163527-a2c0da244d78/go.mod h1:aL8wCCfTfSfmXjznFBSZNN13rSJjlIOI1fUNAtF7rmI= +github.com/yuin/goldmark v1.4.13/go.mod h1:6yULJ656Px+3vBD8DxQVa3kxgyrAnzto9xy5taEt/CY= +go.mongodb.org/mongo-driver v1.17.1 h1:Wic5cJIwJgSpBhe3lx3+/RybR5PiYRMpVFgO7cOHyIM= +go.mongodb.org/mongo-driver v1.17.1/go.mod h1:wwWm/+BuOddhcq3n68LKRmgk2wXzmF6s0SFOa0GINL4= +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.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w= +golang.org/x/crypto v0.0.0-20210921155107-089bfa567519/go.mod h1:GvvjBRRGRdwPK5ydBHafDWAxML/pGHZbMvKqRZ5+Abc= +golang.org/x/crypto v0.26.0 h1:RrRspgV4mU+YwB4FYnuBoKsUapNIL5cohGAmSH3azsw= +golang.org/x/crypto v0.26.0/go.mod h1:GY7jblb9wI+FOo5y8/S2oY4zWP07AkOJ4+jxCqdqn54= +golang.org/x/mod v0.6.0-dev.0.20220419223038-86c51ed26bb4/go.mod h1:jJ57K6gSWd91VN4djpZkiMVwK6gcyfeH4XE8wZrZaV4= +golang.org/x/net v0.0.0-20190620200207-3b0461eec859/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s= +golang.org/x/net v0.0.0-20210226172049-e18ecbb05110/go.mod h1:m0MpNAwzfU5UDzcl9v0D8zg8gWTRqZa9RBIspLL5mdg= +golang.org/x/net v0.0.0-20220722155237-a158d28d115b/go.mod h1:XRhObCWvk6IyKnWLug+ECip1KBveYUHfp+8e9klMJ9c= +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/sync v0.0.0-20190423024810-112230192c58/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= +golang.org/x/sync v0.0.0-20220722155255-886fb9371eb4/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= +golang.org/x/sync v0.8.0 h1:3NFvSEYkUoMifnESzZl15y791HH1qU2xm6eCJU5ZPXQ= +golang.org/x/sync v0.8.0/go.mod h1:Czt+wKu1gCyEFDUtn0jG5QVvpJ6rzVqr5aXyt9drQfk= +golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY= +golang.org/x/sys v0.0.0-20201119102817-f84b799fce68/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.0.0-20210615035016-665e8c7367d1/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/sys v0.0.0-20220520151302-bc2c85ada10a/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/sys v0.0.0-20220722155257-8c9f86f7a55f/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +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.23.0 h1:YfKFowiIMvtgl1UERQoTPPToxltDeZfbj4H7dVUCwmM= +golang.org/x/sys v0.23.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA= +golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo= +golang.org/x/term v0.0.0-20210927222741-03fcf44c2211/go.mod h1:jbD1KX2456YbFQfuXm/mYQcufACuNUgVhRMnK/tPxf8= +golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ= +golang.org/x/text v0.3.3/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ= +golang.org/x/text v0.3.7/go.mod h1:u+2+/6zg+i71rQMx5EYifcz6MCKuco9NR6JIITiCfzQ= +golang.org/x/text v0.3.8/go.mod h1:E6s5w1FMmriuDzIBO73fBruAKo1PCIq6d2Q6DHfQ8WQ= +golang.org/x/text v0.17.0 h1:XtiM5bkSOt+ewxlOE/aE/AKEHibwj/6gvWMl9Rsh0Qc= +golang.org/x/text v0.17.0/go.mod h1:BuEKDfySbSR4drPmRPG/7iBdf8hvFMuRexcpahXilzY= +golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ= +golang.org/x/tools v0.0.0-20191119224855-298f0cb1881e/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo= +golang.org/x/tools v0.1.12/go.mod h1:hNGJHUnrk76NpqgfD5Aqm5Crs+Hm0VOH/i9J2+nxYbc= +golang.org/x/xerrors v0.0.0-20190717185122-a985d3407aa7/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/repositories/generic_repository.go b/repositories/generic_repository.go new file mode 100644 index 0000000..bb033d8 --- /dev/null +++ b/repositories/generic_repository.go @@ -0,0 +1,29 @@ +package repositories + +import ( + "github.com/klass-lk/ginboot/types" + "go.mongodb.org/mongo-driver/mongo" + "go.mongodb.org/mongo-driver/mongo/options" +) + +type GenericRepository[T types.Document] interface { + Query() *mongo.Collection + FindById(id interface{}) (T, error) + FindAllById(idList []string) ([]T, error) + Save(doc T) error + SaveOrUpdate(doc T) error + SaveAll(sms []T) error + Update(doc T) error + Delete(id string) error + FindOneBy(field string, value interface{}) (T, error) + FindOneByFilters(filters map[string]interface{}) (T, error) + FindBy(field string, value interface{}) ([]T, error) + FindByMultiple(filters map[string]interface{}) ([]T, error) + FindAll(opts ...*options.FindOptions) ([]T, error) + FindAllPaginated(pageRequest types.PageRequest) (types.PageResponse[T], error) + FindByPaginated(pageRequest types.PageRequest, filters map[string]interface{}) (types.PageResponse[T], error) + CountBy(field string, value interface{}) (int64, error) + CountByFilters(filters map[string]interface{}) (int64, error) + ExistsBy(field string, value interface{}) (bool, error) + ExistsByFilters(filters map[string]interface{}) (bool, error) +} diff --git a/repositories/mongo_repository.go b/repositories/mongo_repository.go new file mode 100644 index 0000000..5fc1281 --- /dev/null +++ b/repositories/mongo_repository.go @@ -0,0 +1,314 @@ +package repositories + +import ( + "context" + "fmt" + "github.com/klass-lk/ginboot/types" + "go.mongodb.org/mongo-driver/bson" + "go.mongodb.org/mongo-driver/mongo" + "go.mongodb.org/mongo-driver/mongo/options" + "math" + "time" +) + +type MongoRepository[T types.Document] struct { + collection *mongo.Collection +} + +func NewMongoRepository[T types.Document](db *mongo.Database) *MongoRepository[T] { + var doc T + return &MongoRepository[T]{ + collection: db.Collection(doc.GetCollectionName()), + } +} + +func (r *MongoRepository[T]) Query() *mongo.Collection { + return r.collection +} + +func (r *MongoRepository[T]) FindById(id interface{}) (T, error) { + ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second) + defer cancel() + + var result T + err := r.collection.FindOne(ctx, bson.M{"_id": id}).Decode(&result) + if err != nil { + return result, err + } + return result, nil +} + +func (r *MongoRepository[T]) FindAllById(idList []string) ([]T, error) { + ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second) + defer cancel() + filter := bson.M{"_id": bson.M{"$in": idList}} + cursor, err := r.collection.Find(ctx, filter) + if err != nil { + return nil, err + } + defer cursor.Close(ctx) + var result []T + for cursor.Next(ctx) { + var doc T + err := cursor.Decode(&doc) + if err != nil { + return nil, err + } + result = append(result, doc) + } + return result, nil +} + +func (r *MongoRepository[T]) Save(doc T) error { + ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second) + defer cancel() + + _, err := r.collection.InsertOne(ctx, doc) + return err +} + +func (r *MongoRepository[T]) SaveOrUpdate(doc T) error { + exists, err := r.ExistsBy("_id", doc.GetID()) + if err != nil { + return err + } + if exists { + return r.Update(doc) + } + return r.Save(doc) +} + +func (r *MongoRepository[T]) SaveAll(sms []T) error { + ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second) + defer cancel() + var documents []interface{} + for _, sm := range sms { + documents = append(documents, sm) + } + _, err := r.collection.InsertMany(ctx, documents) + if err != nil { + return err + } + return nil +} + +func (r *MongoRepository[T]) Update(doc T) error { + ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second) + defer cancel() + + filter := bson.M{"_id": doc.GetID()} + update := bson.M{"$set": doc} + _, err := r.collection.UpdateOne(ctx, filter, update) + return err +} + +func (r *MongoRepository[T]) Delete(id string) error { + ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second) + defer cancel() + + _, err := r.collection.DeleteOne(ctx, bson.M{"_id": id}) + return err +} + +func (r *MongoRepository[T]) FindOneBy(field string, value interface{}) (T, error) { + ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second) + defer cancel() + + var result T + err := r.collection.FindOne(ctx, bson.M{field: value}).Decode(&result) + if err != nil { + return result, err + } + return result, nil +} + +func (r *MongoRepository[T]) FindOneByFilters(filters map[string]interface{}) (T, error) { + ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second) + defer cancel() + + filter := bson.M{} + for field, value := range filters { + filter[field] = value + } + + var result T + err := r.collection.FindOne(ctx, filter).Decode(&result) + if err != nil { + return result, err + } + + return result, nil +} + +func (r *MongoRepository[T]) FindBy(field string, value interface{}) ([]T, error) { + ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second) + defer cancel() + + filter := bson.M{field: value} + cursor, err := r.collection.Find(ctx, filter) + if err != nil { + return nil, err + } + defer cursor.Close(ctx) + + var results []T + err = cursor.All(ctx, &results) + return results, err +} + +func (r *MongoRepository[T]) FindByMultiple(filters map[string]interface{}) ([]T, error) { + ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second) + defer cancel() + + filter := bson.M{} + for field, value := range filters { + filter[field] = value + } + + cursor, err := r.collection.Find(ctx, filter) + if err != nil { + return nil, err + } + defer cursor.Close(ctx) + + var results []T + err = cursor.All(ctx, &results) + return results, err +} + +func (r *MongoRepository[T]) FindAll(opts ...*options.FindOptions) ([]T, error) { + ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second) + defer cancel() + + cursor, err := r.collection.Find(ctx, bson.M{}, opts...) + if err != nil { + return nil, err + } + defer cursor.Close(ctx) + + var results []T + err = cursor.All(ctx, &results) + return results, err +} + +func (r *MongoRepository[T]) FindAllPaginated(pageRequest types.PageRequest) (types.PageResponse[T], error) { + ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second) + defer cancel() + skip := bson.D{{"$skip", (pageRequest.Page - 1) * pageRequest.Size}} + limit := bson.D{{"$limit", pageRequest.Size}} + sort := bson.D{{"$sort", bson.D{{pageRequest.Sort.Field, pageRequest.Sort.Direction}}}} + aggregationPipeline := mongo.Pipeline{ + skip, limit, sort, + } + + cursor, err := r.Query().Aggregate(ctx, aggregationPipeline) + if err != nil { + fmt.Println("Error running aggregation:", err) + return types.PageResponse[T]{}, err + } + defer cursor.Close(ctx) + + var results []T + if err = cursor.All(ctx, &results); err != nil { + fmt.Println("Error retrieving results:", err) + return types.PageResponse[T]{}, err + } + count, err := r.Query().CountDocuments(ctx, bson.M{}) + if err != nil { + return types.PageResponse[T]{}, err + } + totalPages := int(math.Ceil(float64(count) / float64(pageRequest.Size))) + pageResponse := types.PageResponse[T]{ + Contents: results, + NumberOfElements: pageRequest.Size, + Pageable: pageRequest, + TotalPages: totalPages, + TotalElements: int(count), + } + return pageResponse, nil +} + +func (r *MongoRepository[T]) FindByPaginated(pageRequest types.PageRequest, filters map[string]interface{}) (types.PageResponse[T], error) { + ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second) + defer cancel() + + filter := bson.M{} + for field, value := range filters { + filter[field] = value + } + + match := bson.D{{"$match", filter}} + skip := bson.D{{"$skip", (pageRequest.Page - 1) * pageRequest.Size}} + limit := bson.D{{"$limit", pageRequest.Size}} + sort := bson.D{{"$sort", bson.D{{pageRequest.Sort.Field, pageRequest.Sort.Direction}}}} + aggregationPipeline := mongo.Pipeline{ + match, sort, skip, limit, + } + + cursor, err := r.Query().Aggregate(ctx, aggregationPipeline) + if err != nil { + fmt.Println("Error running aggregation:", err) + return types.PageResponse[T]{}, err + } + defer cursor.Close(ctx) + + var results []T + if err = cursor.All(ctx, &results); err != nil { + fmt.Println("Error retrieving results:", err) + return types.PageResponse[T]{}, err + } + count, err := r.Query().CountDocuments(ctx, filter) + if err != nil { + return types.PageResponse[T]{}, err + } + totalPages := int(math.Ceil(float64(count) / float64(pageRequest.Size))) + pageResponse := types.PageResponse[T]{ + Contents: results, + NumberOfElements: pageRequest.Size, + Pageable: pageRequest, + TotalPages: totalPages, + TotalElements: int(count), + } + return pageResponse, nil +} + +func (r *MongoRepository[T]) CountBy(field string, value interface{}) (int64, error) { + ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second) + defer cancel() + count, err := r.collection.CountDocuments(ctx, bson.M{field: value}) + if err != nil { + return 0, err + } + return count, nil +} + +func (r *MongoRepository[T]) CountByFilters(filters map[string]interface{}) (int64, error) { + ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second) + defer cancel() + + filter := bson.M{} + for field, value := range filters { + filter[field] = value + } + count, err := r.collection.CountDocuments(ctx, filter) + if err != nil { + return 0, err + } + return count, nil +} + +func (r *MongoRepository[T]) ExistsBy(field string, value interface{}) (bool, error) { + count, err := r.CountBy(field, value) + if err != nil { + return false, err + } + return count > 0, nil +} + +func (r *MongoRepository[T]) ExistsByFilters(filters map[string]interface{}) (bool, error) { + count, err := r.CountByFilters(filters) + if err != nil { + return false, err + } + return count > 0, nil +} diff --git a/security/crypt.go b/security/crypt.go new file mode 100644 index 0000000..92a0c6d --- /dev/null +++ b/security/crypt.go @@ -0,0 +1,23 @@ +package security + +import "golang.org/x/crypto/bcrypt" + +type Crypt struct { +} + +func NewCrypt() *Crypt { + return &Crypt{} +} + +func (Crypt) GetPasswordHash(password string) (string, error) { + hash, err := bcrypt.GenerateFromPassword([]byte(password), 5) + if err != nil { + return "", err + } + return string(hash), nil +} + +func (Crypt) IsMatching(hash, password string) bool { + err := bcrypt.CompareHashAndPassword([]byte(hash), []byte(password)) + return err == nil +} diff --git a/security/jwt.go b/security/jwt.go new file mode 100644 index 0000000..debdf38 --- /dev/null +++ b/security/jwt.go @@ -0,0 +1,86 @@ +package security + +import ( + "errors" + "fmt" + "github.com/dgrijalva/jwt-go" + "github.com/google/uuid" + "os" + "time" +) + +type Claims struct { + Role string `json:"role"` + jwt.StandardClaims +} + +func GenerateTokens(userId string, role string) (string, string, error) { + var tokenSecret = os.Getenv("JWT_SECRET") + accessToken, err := generateJwtToken(userId, role, 24*time.Hour, tokenSecret) + if err != nil { + return "", "", err + } + + var refreshTokenSecret = os.Getenv("JWT_REFRESH_SECRET") + refreshToken, err := generateJwtToken(userId, role, 24*30*time.Hour, refreshTokenSecret) + if err != nil { + return "", "", err + } + return accessToken, refreshToken, nil +} + +func generateJwtToken(userId string, role string, duration time.Duration, secretKey string) (string, error) { + var jwtKeyBytes = []byte(secretKey) + expirationTime := time.Now().Add(duration) + claims := &Claims{ + Role: role, + StandardClaims: jwt.StandardClaims{ + ExpiresAt: expirationTime.Unix(), + Id: uuid.New().String(), + IssuedAt: time.Now().Unix(), + Issuer: "klass-lk", + Subject: userId, + }, + } + token := jwt.NewWithClaims(jwt.SigningMethodHS256, claims) + return token.SignedString(jwtKeyBytes) +} + +func ParseAccessToken(tokenString string) (*jwt.Token, error) { + var tokenSecret = os.Getenv("JWT_SECRET") + return parseJwtToken(tokenString, tokenSecret) +} +func ParseRefreshToken(tokenString string) (*jwt.Token, error) { + var tokenSecret = os.Getenv("JWT_REFRESH_SECRET") + return parseJwtToken(tokenString, tokenSecret) +} + +func parseJwtToken(tokenString string, secretKey string) (*jwt.Token, error) { + token, err := jwt.Parse(tokenString, func(token *jwt.Token) (interface{}, error) { + if _, ok := token.Method.(*jwt.SigningMethodHMAC); !ok { + return nil, fmt.Errorf("unexpected signing method: %v", token.Header["alg"]) + } + return []byte(secretKey), nil + }) + return token, err +} + +func ExtractClaims(token *jwt.Token) (jwt.MapClaims, error) { + claims, ok := token.Claims.(jwt.MapClaims) + if !ok { + return nil, errors.New("bearer token is invalid") + } + return claims, nil +} + +func IsExpired(claims jwt.MapClaims) bool { + return float64(time.Now().Unix()) > claims["exp"].(float64) +} + +func ExtractUserId(claims jwt.MapClaims) string { + return claims["sub"].(string) +} + +func ExtractRole(claims jwt.MapClaims) string { + return claims["role"].(string) +} diff --git a/security/password_encoder.go b/security/password_encoder.go new file mode 100644 index 0000000..86a6d9d --- /dev/null +++ b/security/password_encoder.go @@ -0,0 +1,6 @@ +package security + +type PasswordEncoder interface { + GetPasswordHash(password string) (string, error) + IsMatching(hash, password string) bool +} diff --git a/security/pbkdf2_encoder.go b/security/pbkdf2_encoder.go new file mode 100644 index 0000000..70d29fe --- /dev/null +++ b/security/pbkdf2_encoder.go @@ -0,0 +1,40 @@ +package security + +import ( + "crypto/sha512" + "crypto/subtle" + "encoding/base64" + "golang.org/x/crypto/pbkdf2" + "os" + "strconv" +) + +type PBKDF2Encoder struct { + Secret string + Iteration int + KeyLength int +} + +func NewPBKDF2Encoder() *PBKDF2Encoder { + secret := os.Getenv("PBKDF2_ENCODER_SECRET") + iteration, err := strconv.ParseInt(os.Getenv("PBKDF2_ENCODER_ITERATION"), 10, 64) + if err != nil { + panic("ENCODER_ITERATIONS_MISSING") + } + keyLength, err := strconv.ParseInt(os.Getenv("PBKDF2_ENCODER_KEY_LENGTH"), 10, 64) + if err != nil { + panic("ENCODER_ITERATIONS_MISSING") + } + return &PBKDF2Encoder{secret, int(iteration), int(keyLength)} +} + +func (P PBKDF2Encoder) GetPasswordHash(password string) (string, error) { + hash := pbkdf2.Key([]byte(password), []byte(P.Secret), P.Iteration, P.KeyLength, sha512.New) + encoded := base64.StdEncoding.EncodeToString(hash) + return encoded, nil +} + +func (P PBKDF2Encoder) IsMatching(hash, password string) bool { + encodedCharSeq, _ := P.GetPasswordHash(password) + return subtle.ConstantTimeCompare([]byte(encodedCharSeq), []byte(hash)) == 1 +} diff --git a/types/auth_context.go b/types/auth_context.go new file mode 100644 index 0000000..c193c66 --- /dev/null +++ b/types/auth_context.go @@ -0,0 +1,6 @@ +package types + +type AuthContext struct { + UserId string + Role string +} diff --git a/types/db_types.go b/types/db_types.go new file mode 100644 index 0000000..962ed76 --- /dev/null +++ b/types/db_types.go @@ -0,0 +1,20 @@ +package types + +type SortField struct { + Field string `json:"field"` + Direction int `json:"direction"` +} + +type PageRequest struct { + Page int `json:"page"` + Size int `json:"size"` + Sort SortField `json:"sort"` +} + +type PageResponse[T interface{}] struct { + Contents []T `json:"content"` + NumberOfElements int `json:"numberOfElements"` + Pageable PageRequest `json:"pageable"` + TotalPages int `json:"totalPages"` + TotalElements int `json:"totalElements"` +} diff --git a/types/mongo_document.go b/types/mongo_document.go new file mode 100644 index 0000000..c9edc50 --- /dev/null +++ b/types/mongo_document.go @@ -0,0 +1,6 @@ +package types + +type Document interface { + GetID() string + GetCollectionName() string +}