From f5738e39f5702b6992b5378718e7c462964570fb Mon Sep 17 00:00:00 2001 From: Nabil Ould Hamou Date: Tue, 16 Jan 2024 00:36:49 +0100 Subject: [PATCH] Initial commit --- api/api.go | 13 ++++ controllers/authController.go | 127 ++++++++++++++++++++++++++++++++++ controllers/fileController.go | 36 ++++++++++ controllers/userController.go | 50 +++++++++++++ go.mod | 40 +++++++++++ go.sum | 99 ++++++++++++++++++++++++++ initializers/database.go | 19 +++++ initializers/loadEnv.go | 14 ++++ main.go | 40 +++++++++++ middleware/requireAuth.go | 35 ++++++++++ migrate/migration.go | 26 +++++++ models/fileModel.go | 10 +++ models/postModel.go | 20 ++++++ models/userModel.go | 23 ++++++ tokens/tokenParser.go | 45 ++++++++++++ 15 files changed, 597 insertions(+) create mode 100644 api/api.go create mode 100644 controllers/authController.go create mode 100644 controllers/fileController.go create mode 100644 controllers/userController.go create mode 100644 go.mod create mode 100644 go.sum create mode 100644 initializers/database.go create mode 100644 initializers/loadEnv.go create mode 100644 main.go create mode 100644 middleware/requireAuth.go create mode 100644 migrate/migration.go create mode 100644 models/fileModel.go create mode 100644 models/postModel.go create mode 100644 models/userModel.go create mode 100644 tokens/tokenParser.go diff --git a/api/api.go b/api/api.go new file mode 100644 index 0000000..6aaa75f --- /dev/null +++ b/api/api.go @@ -0,0 +1,13 @@ +package api + +import "github.com/gin-gonic/gin" + +var Router *gin.Engine +var Files *gin.RouterGroup +var Api *gin.RouterGroup + +func CreateRouter() { + Router = gin.Default() + Files = Router.Group("/files") + Api = Router.Group("/api") +} diff --git a/controllers/authController.go b/controllers/authController.go new file mode 100644 index 0000000..4c39476 --- /dev/null +++ b/controllers/authController.go @@ -0,0 +1,127 @@ +package controllers + +import ( + "azote-backend/initializers" + "azote-backend/models" + "errors" + "github.com/gin-gonic/gin" + "github.com/golang-jwt/jwt" + "github.com/google/uuid" + "golang.org/x/crypto/bcrypt" + "gorm.io/gorm" + "net/http" + "os" + "time" +) + +func Signup(c *gin.Context) { + var body struct { + Username string + Email string + Password string + } + + if err := c.Bind(&body); err != nil { + c.JSON(http.StatusBadRequest, gin.H{"error": "Invalid request format"}) + return + } + + if body.Username == "" || body.Email == "" || body.Password == "" { + c.JSON(http.StatusBadRequest, gin.H{"error": "Missing required fields"}) + return + } + + hashedPassword, err := bcrypt.GenerateFromPassword([]byte(body.Password), bcrypt.DefaultCost) + if err != nil { + c.Status(http.StatusInternalServerError) + } + + user := models.User{ + Username: body.Username, + Email: body.Email, + Password: string(hashedPassword), + } + result := initializers.DB.Create(&user) + + if result.Error != nil { + c.JSON(http.StatusBadRequest, gin.H{ + "error": "An account already exists with that email.", + }) + return + } + + tokenString, err := createToken(user.ID) + if err != nil { + c.JSON(http.StatusInternalServerError, gin.H{ + "error": err, + }) + return + } + + c.JSON(http.StatusAccepted, gin.H{ + "user": user, + "token": tokenString, + }) +} + +func Login(c *gin.Context) { + var body struct { + Email string + Password string + } + + if err := c.Bind(&body); err != nil { + c.JSON(http.StatusBadRequest, gin.H{"error": "Invalid request format"}) + return + } + + if body.Email == "" || body.Password == "" { + c.JSON(http.StatusBadRequest, gin.H{"error": "Missing required fields"}) + return + } + + var user models.User + result := initializers.DB.First(&user, "email = ?", body.Email) + + if result.Error != nil { + if errors.Is(result.Error, gorm.ErrRecordNotFound) { + c.JSON(http.StatusBadRequest, gin.H{"error": "Email/Password is invalid."}) + } else { + c.JSON(http.StatusInternalServerError, gin.H{"error": "Database error"}) + } + return + } + + err := bcrypt.CompareHashAndPassword([]byte(user.Password), []byte(body.Password)) + if err != nil { + c.JSON(http.StatusBadRequest, gin.H{"error": "Email/Password is invalid."}) + return + } + + tokenString, err := createToken(user.ID) + if err != nil { + c.JSON(http.StatusInternalServerError, gin.H{ + "error": err, + }) + return + } + + c.JSON(http.StatusAccepted, gin.H{ + "user": user, + "token": tokenString, + }) +} + +func createToken(userId uuid.UUID) (string, error) { + token := jwt.NewWithClaims(jwt.SigningMethodHS256, jwt.MapClaims{ + "bearer": userId, + "expiresAt": time.Now().Add(time.Hour * 24 * 30).Unix(), + }) + + tokenString, err := token.SignedString([]byte(os.Getenv("JWT_SECRET"))) + if err != nil { + return "", errors.New("could not create token") + } + + return tokenString, nil +} diff --git a/controllers/fileController.go b/controllers/fileController.go new file mode 100644 index 0000000..7e98faf --- /dev/null +++ b/controllers/fileController.go @@ -0,0 +1,36 @@ +package controllers + +import ( + "azote-backend/initializers" + "azote-backend/models" + "github.com/gin-gonic/gin" + "github.com/google/uuid" + "net/http" + "strings" +) + +func UploadFile(c *gin.Context) { + file, _ := c.FormFile("file") + splitName := strings.Split(file.Filename, ".") + path := "assets/images/" + uuid.New().String() + "." + splitName[len(splitName)-1] + if err := c.SaveUploadedFile(file, path); err != nil { + c.JSON(http.StatusInternalServerError, gin.H{"error": "Unknown error"}) + return + } + + f := models.File{ + Location: path, + } + + result := initializers.DB.Create(&f) + if result.Error != nil { + c.JSON(http.StatusInternalServerError, gin.H{ + "error": "An internal server error occurred", + }) + return + } + + c.JSON(http.StatusAccepted, gin.H{ + "recipeId": "feur", + }) +} diff --git a/controllers/userController.go b/controllers/userController.go new file mode 100644 index 0000000..0acab89 --- /dev/null +++ b/controllers/userController.go @@ -0,0 +1,50 @@ +package controllers + +import ( + "azote-backend/initializers" + "azote-backend/models" + "errors" + "github.com/gin-gonic/gin" + "github.com/google/uuid" + "gorm.io/gorm" + "net/http" +) + +// GetUsers Returns all users +func GetUsers(c *gin.Context) { + var users []models.User + + result := initializers.DB.Find(&users) + if result.Error != nil { + c.Status(http.StatusBadRequest) + return + } + + c.JSON(http.StatusAccepted, gin.H{ + "users": users, + }) +} + +func GetUserById(c *gin.Context) { + userId := c.Param("id") + + uniqueId, err := uuid.Parse(userId) + if err != nil { + c.JSON(http.StatusBadRequest, gin.H{"error": "Invalid id format"}) + return + } + + var user models.User + result := initializers.DB.First(&user, "id = ?", uniqueId) + + if result.Error != nil { + if errors.Is(result.Error, gorm.ErrRecordNotFound) { + c.JSON(http.StatusNotFound, gin.H{"error": "User not found"}) + } else { + c.JSON(http.StatusInternalServerError, gin.H{"error": "Database error"}) + } + return + } + + c.JSON(http.StatusOK, user) +} diff --git a/go.mod b/go.mod new file mode 100644 index 0000000..c256c31 --- /dev/null +++ b/go.mod @@ -0,0 +1,40 @@ +module azote-backend + +go 1.20 + +require ( + github.com/bytedance/sonic v1.10.2 // indirect + github.com/chenzhuoyu/base64x v0.0.0-20230717121745-296ad89f973d // indirect + github.com/chenzhuoyu/iasm v0.9.1 // indirect + github.com/gabriel-vasile/mimetype v1.4.3 // indirect + github.com/gin-contrib/sse v0.1.0 // indirect + github.com/gin-gonic/gin v1.9.1 // 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.17.0 // indirect + github.com/go-sql-driver/mysql v1.7.1 // indirect + github.com/goccy/go-json v0.10.2 // indirect + github.com/golang-jwt/jwt v3.2.2+incompatible // indirect + github.com/google/uuid v1.5.0 // indirect + github.com/jinzhu/inflection v1.0.0 // indirect + github.com/jinzhu/now v1.1.5 // indirect + github.com/joho/godotenv v1.5.1 // indirect + github.com/json-iterator/go v1.1.12 // indirect + github.com/klauspost/cpuid/v2 v2.2.6 // indirect + github.com/leodido/go-urn v1.2.4 // 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.1.1 // indirect + github.com/twitchyliquid64/golang-asm v0.15.1 // indirect + github.com/ugorji/go/codec v1.2.12 // indirect + golang.org/x/arch v0.7.0 // indirect + golang.org/x/crypto v0.18.0 // indirect + golang.org/x/net v0.20.0 // indirect + golang.org/x/sys v0.16.0 // indirect + golang.org/x/text v0.14.0 // indirect + google.golang.org/protobuf v1.32.0 // indirect + gopkg.in/yaml.v3 v3.0.1 // indirect + gorm.io/driver/mysql v1.5.2 // indirect + gorm.io/gorm v1.25.5 // indirect +) diff --git a/go.sum b/go.sum new file mode 100644 index 0000000..129c3c8 --- /dev/null +++ b/go.sum @@ -0,0 +1,99 @@ +github.com/bytedance/sonic v1.5.0/go.mod h1:ED5hyg4y6t3/9Ku1R6dU/4KyJ48DZ4jPhfY1O2AihPM= +github.com/bytedance/sonic v1.10.0-rc/go.mod h1:ElCzW+ufi8qKqNW0FY314xriJhyJhuoJ3gFZdAHF7NM= +github.com/bytedance/sonic v1.10.2 h1:GQebETVBxYB7JGWJtLBi07OVzWwt+8dWA00gEVW2ZFE= +github.com/bytedance/sonic v1.10.2/go.mod h1:iZcSUejdk5aukTND/Eu/ivjQuEL0Cu9/rf50Hi0u/g4= +github.com/chenzhuoyu/base64x v0.0.0-20211019084208-fb5309c8db06/go.mod h1:DH46F32mSOjUmXrMHnKwZdA8wcEefY7UVqBKYGjpdQY= +github.com/chenzhuoyu/base64x v0.0.0-20221115062448-fe3a3abad311/go.mod h1:b583jCggY9gE99b6G5LEC39OIiVsWj+R97kbl5odCEk= +github.com/chenzhuoyu/base64x v0.0.0-20230717121745-296ad89f973d h1:77cEq6EriyTZ0g/qfRdp61a3Uu/AWrgIq2s0ClJV1g0= +github.com/chenzhuoyu/base64x v0.0.0-20230717121745-296ad89f973d/go.mod h1:8EPpVsBuRksnlj1mLy4AWzRNQYxauNi62uWcE3to6eA= +github.com/chenzhuoyu/iasm v0.9.0/go.mod h1:Xjy2NpN3h7aUqeqM+woSuuvxmIe6+DDsiNLIrkAmYog= +github.com/chenzhuoyu/iasm v0.9.1 h1:tUHQJXo3NhBqw6s33wkGn9SP3bvrWLdlVIJ3hQBL7P0= +github.com/chenzhuoyu/iasm v0.9.1/go.mod h1:Xjy2NpN3h7aUqeqM+woSuuvxmIe6+DDsiNLIrkAmYog= +github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= +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.9.1 h1:4idEAncQnU5cB7BeOkPtxjfCSye0AAm1R0RVIqJ+Jmg= +github.com/gin-gonic/gin v1.9.1/go.mod h1:hPrL7YrpYKXt5YId3A/Tnip5kqbEAP+KLuI3SUcPTeU= +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.17.0 h1:SmVVlfAOtlZncTxRuinDPomC2DkXJ4E5T9gDA0AIH74= +github.com/go-playground/validator/v10 v10.17.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-jwt/jwt v3.2.2+incompatible h1:IfV12K8xAKAnZqdXVzCZ+TOjboZ2keLg81eXfW3O+oY= +github.com/golang-jwt/jwt v3.2.2+incompatible/go.mod h1:8pz2t5EyA70fFQQSrl6XZXzqecmYZeUEB8OUGHkxJ+I= +github.com/google/gofuzz v1.0.0/go.mod h1:dBl0BpW6vV/+mYPU4Po3pmUjxk6FQPldtuIdl/M65Eg= +github.com/google/uuid v1.5.0 h1:1p67kYwdtXjb0gL0BPiP1Av9wiZPo5A8z2cWkTZ+eyU= +github.com/google/uuid v1.5.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo= +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/joho/godotenv v1.5.1 h1:7eLL/+HRGLY0ldzfGMeQkb7vMd0as4CfYvUVzLqw0N0= +github.com/joho/godotenv v1.5.1/go.mod h1:f4LDr5Voq0i2e/R5DDNOoa2zzDfwtkZa6DnEwAbqwq4= +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.6 h1:ndNyv040zDGIDh8thGkXYjnFtiN02M1PVVF+JE/48xc= +github.com/klauspost/cpuid/v2 v2.2.6/go.mod h1:Lcz8mBdAVJIBVzewtcLocK12l3Y+JytZYpaMropDUws= +github.com/knz/go-libedit v1.10.1/go.mod h1:MZTVkCWyz0oBc7JOWP3wNAzd002ZbM/5hgShxwh4x8M= +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.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.1.1 h1:LWAJwfNvjQZCFIDKWYQaM62NcYeYViCmWIwmOStowAI= +github.com/pelletier/go-toml/v2 v2.1.1/go.mod h1:tJU2Z3ZkXwnxa4DPO899bsyIoywizdUvyaeZurnPPDc= +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.4/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.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.7.0 h1:pskyeJh/3AmoQ8CPE95vxHLqp1G1GfGNXTmcl9NEKTc= +golang.org/x/arch v0.7.0/go.mod h1:FEVrYAQjsQXMVJ1nsMoVVXPZg6p2JE2mx8psSWTDQys= +golang.org/x/crypto v0.18.0 h1:PGVlW0xEltQnzFZ55hkuX5+KLyrMYhHld1YHO4AKcdc= +golang.org/x/crypto v0.18.0/go.mod h1:R0j02AL6hcrfOiy9T4ZYp/rcWeMxM3L6QYxlOuEG1mg= +golang.org/x/net v0.20.0 h1:aCL9BSgETF1k+blQaYUBx9hJ9LOGP3gAVemcZlf1Kpo= +golang.org/x/net v0.20.0/go.mod h1:z8BVo6PvndSri0LbOE3hAn0apkU+1YvI6E70E9jsnvY= +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.16.0 h1:xWw16ngr6ZMtmxDyKyIgsE93KNKz5HKmMa3b8ALHidU= +golang.org/x/sys v0.16.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA= +golang.org/x/text v0.14.0 h1:ScX5w1eTa3QqT8oi6+ziP7dTV1S2+ALU0bI+0zXKWiQ= +golang.org/x/text v0.14.0/go.mod h1:18ZOQIKpY8NJVqYksKHtTdi31H5itFRjB5/qKTNYzSU= +google.golang.org/protobuf v1.32.0 h1:pPC6BG5ex8PDFnkbrGU3EixyhKcQ2aDuBS36lqK/C7I= +google.golang.org/protobuf v1.32.0/go.mod h1:c6P6GXX6sHbq/GpV6MGZEdwhWPcYBgnhAHhKbcUYpos= +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= +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/initializers/database.go b/initializers/database.go new file mode 100644 index 0000000..4d74469 --- /dev/null +++ b/initializers/database.go @@ -0,0 +1,19 @@ +package initializers + +import ( + "gorm.io/driver/mysql" + "gorm.io/gorm" + "log" + "os" +) + +var DB *gorm.DB + +func ConnectToDB() { + dsn := os.Getenv("DSN") + var err error + DB, err = gorm.Open(mysql.Open(dsn), &gorm.Config{}) + if err != nil { + log.Fatal("Could not connect to the database:\n" + err.Error()) + } +} diff --git a/initializers/loadEnv.go b/initializers/loadEnv.go new file mode 100644 index 0000000..72df89b --- /dev/null +++ b/initializers/loadEnv.go @@ -0,0 +1,14 @@ +package initializers + +import ( + "github.com/joho/godotenv" + "log" +) + +func LoadEnv() { + err := godotenv.Load() + + if err != nil { + log.Fatal("Error loading .env file!") + } +} diff --git a/main.go b/main.go new file mode 100644 index 0000000..1aedc0e --- /dev/null +++ b/main.go @@ -0,0 +1,40 @@ +package main + +import ( + "azote-backend/api" + "azote-backend/controllers" + "azote-backend/initializers" + "github.com/gin-gonic/gin" + "log" + "os" +) + +func init() { + initializers.LoadEnv() + initializers.ConnectToDB() +} + +func main() { + gin.SetMode(os.Getenv("GIN_MODE")) + + api.CreateRouter() + + // Auth + api.Api.POST("/signup", controllers.Signup) + api.Api.POST("/login", controllers.Login) + + // Users + api.Api.GET("/users", controllers.GetUsers) + api.Api.GET("/users/:id", controllers.GetUserById) + + // Files + api.Router.Static("assets", "./assets/images") + + // Posts + + // Starting + err := api.Router.Run() + if err != nil { + log.Fatal("Router could not be created!\n" + err.Error()) + } +} diff --git a/middleware/requireAuth.go b/middleware/requireAuth.go new file mode 100644 index 0000000..8174673 --- /dev/null +++ b/middleware/requireAuth.go @@ -0,0 +1,35 @@ +package middleware + +import ( + "azote-backend/initializers" + "azote-backend/models" + "azote-backend/tokens" + "github.com/gin-gonic/gin" + "github.com/google/uuid" + "net/http" + "time" +) + +func RequireAuth(c *gin.Context) { + + session, err := token.ParseToken(c) + if err != nil { + c.AbortWithStatus(http.StatusUnauthorized) + return + } + + if time.Now().Unix() > session.ExpiresAt.Unix() { + c.AbortWithStatus(http.StatusUnauthorized) + return + } + + var user models.User + initializers.DB.First(&user, "id = ?", session.Bearer) + if user.ID == uuid.Nil { + c.AbortWithStatus(http.StatusUnauthorized) + return + } + + c.Set("user", user) + c.Next() +} diff --git a/migrate/migration.go b/migrate/migration.go new file mode 100644 index 0000000..da9385f --- /dev/null +++ b/migrate/migration.go @@ -0,0 +1,26 @@ +package main + +import ( + "azote-backend/initializers" + "azote-backend/models" + "log" +) + +func init() { + initializers.LoadEnv() + initializers.ConnectToDB() +} + +func main() { + log.Println("Migrating models to db...") + + err := initializers.DB.AutoMigrate( + &models.User{}, + &models.File{}, + &models.Post{}) + if err != nil { + log.Fatalf("Automatic migration has failed : %v", err) + } + + log.Println("Migration successful.") +} diff --git a/models/fileModel.go b/models/fileModel.go new file mode 100644 index 0000000..beebce7 --- /dev/null +++ b/models/fileModel.go @@ -0,0 +1,10 @@ +package models + +import "github.com/google/uuid" + +type File struct { + ID uint `gorm:"primaryKey;autoIncrement;"` + FileName string + PostID *uuid.UUID + UserID *uuid.UUID +} diff --git a/models/postModel.go b/models/postModel.go new file mode 100644 index 0000000..12af4b0 --- /dev/null +++ b/models/postModel.go @@ -0,0 +1,20 @@ +package models + +import ( + "github.com/google/uuid" + "gorm.io/gorm" + "time" +) + +type Post struct { + ID uuid.UUID `gorm:"type:char(36);primary_key;"` + CreatedAt time.Time + Author uuid.UUID + Text string + Files []File `gorm:"foreignKey:PostID;constraint:OnUpdate:CASCADE,OnDelete:SET NULL;"` +} + +func (post *Post) BeforeCreate(tx *gorm.DB) (err error) { + post.ID = uuid.New() + return +} diff --git a/models/userModel.go b/models/userModel.go new file mode 100644 index 0000000..affc965 --- /dev/null +++ b/models/userModel.go @@ -0,0 +1,23 @@ +package models + +import ( + "github.com/google/uuid" + "gorm.io/gorm" + "time" +) + +type User struct { + ID uuid.UUID `gorm:"type:char(36);primary_key;"` + CreatedAt time.Time + UpdatedAt time.Time + Avatar File `gorm:"foreignKey:UserID;"` + Username string `gorm:"unique"` + Email string `gorm:"unique"` + Password string `json:"-"` + Posts []Post `gorm:"foreignKey:Author;constraint:OnUpdate:CASCADE,OnDelete:SET NULL;"` +} + +func (user *User) BeforeCreate(tx *gorm.DB) (err error) { + user.ID = uuid.New() + return +} diff --git a/tokens/tokenParser.go b/tokens/tokenParser.go new file mode 100644 index 0000000..2b664f1 --- /dev/null +++ b/tokens/tokenParser.go @@ -0,0 +1,45 @@ +package token + +import ( + "azote-backend/initializers" + "azote-backend/models" + "errors" + "fmt" + "github.com/gin-gonic/gin" + "github.com/golang-jwt/jwt" + "github.com/google/uuid" + "math" + "os" + "time" +) + +type UserSession struct { + Bearer uuid.UUID + ExpiresAt time.Time +} + +func ParseToken(c *gin.Context) (UserSession, error) { + tokenString, err := c.Cookie("Authorization") + if len(tokenString) == 0 || err != nil { + return UserSession{}, errors.New("cookie not found") + } + + 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(os.Getenv("JWT_SECRET")), nil + }) + + if claims, ok := token.Claims.(jwt.MapClaims); ok && token.Valid { + var user models.User + initializers.DB.First(&user, "id = ?", claims["bearer"]) + + sec, dec := math.Modf(claims["expiresAt"].(float64)) + + return UserSession{Bearer: user.ID, ExpiresAt: time.Unix(int64(sec), int64(dec*(1e9)))}, nil + + } else { + return UserSession{}, errors.New("token is not valid") + } +}