fix: empty country code on user creation (#36)

2 files changed, 5 insertions, 1 deletions

diff --git a/backend/controllers/loginController.go b/backend/controllers/loginController.go
index 84ef22f..3e59211 100644
--- a/backend/controllers/loginController.go
+++ b/backend/controllers/loginController.go
@@ -50,6 +50,10 @@ func Login(c *gin.Context) {
                                 c.JSON(http.StatusBadRequest, models.ErrorResponse(err.Error()))
                                 return
                         }
+                        // Empty country code check
+                        if user.LocCountryCode == "" {
+                                user.LocCountryCode = "XX"
+                        }
                         // Insert new user to database
                         database.DB.Exec(`INSERT INTO users (steam_id, username, avatar_link, country_code)
                         VALUES ($1, $2, $3, $4)`, steamID, user.PersonaName, user.AvatarFull, user.LocCountryCode)
diff --git a/backend/database/init.sql b/backend/database/init.sql
index 1b35c02..7e225f0 100644
--- a/backend/database/init.sql
+++ b/backend/database/init.sql
@@ -12,7 +12,7 @@ CREATE TABLE users (
   steam_id TEXT,
   username TEXT NOT NULL,
   avatar_link TEXT NOT NULL,
-  country_code CHAR(2) NOT NULL DEFAULT 'XX',
+  country_code CHAR(2) NOT NULL,
   created_at TIMESTAMP NOT NULL DEFAULT now(),
   updated_at TIMESTAMP NOT NULL DEFAULT now(),
   PRIMARY KEY (steam_id)