4 files changed, 290 insertions, 0 deletions

diff --git a/backend/controllers/userController.go b/backend/controllers/userController.go
index b23a303..ab29e32 100644
--- a/backend/controllers/userController.go
+++ b/backend/controllers/userController.go
@@ -67,3 +67,33 @@ func FetchUser(c *gin.Context) {
         })
         return
 }
+
+func UpdateCountryCode(c *gin.Context) {
+        // Check if user exists
+        user, exists := c.Get("user")
+        if !exists {
+                c.JSON(http.StatusUnauthorized, models.ErrorResponse("User not logged in."))
+                return
+        }
+        code := c.Query("country_code")
+        if code == "" {
+                c.JSON(http.StatusNotFound, models.ErrorResponse("Enter a valid country code."))
+                return
+        }
+        var validCode string
+        err := database.DB.QueryRow(`SELECT country_code FROM countries WHERE country_code = $1;`, code).Scan(&validCode)
+        if err != nil {
+                c.JSON(http.StatusNotFound, models.ErrorResponse(err.Error()))
+                return
+        }
+        // Valid code, update profile
+        _, err = database.DB.Exec(`UPDATE users SET country_code = $1 WHERE steam_id = $2`, validCode, user.(models.User).SteamID)
+        if err != nil {
+                c.JSON(http.StatusNotFound, models.ErrorResponse(err.Error()))
+                return
+        }
+        c.JSON(http.StatusOK, models.Response{
+                Success: true,
+                Message: "Successfully updated country code.",
+        })
+}
diff --git a/backend/database/countries.sql b/backend/database/countries.sql
new file mode 100644
index 0000000..0272f6f
--- /dev/null
+++ b/backend/database/countries.sql
@@ -0,0 +1,252 @@
+INSERT INTO countries (country_code, country_name) VALUES
+('AF', 'AFGHANISTAN'),
+('AL', 'ALBANIA'),
+('DZ', 'ALGERIA'),
+('AS', 'AMERICAN SAMOA'),
+('AD', 'ANDORRA'),
+('AO', 'ANGOLA'),
+('AI', 'ANGUILLA'),
+('AQ', 'ANTARCTICA'),
+('AG', 'ANTIGUA AND BARBUDA'),
+('AR', 'ARGENTINA'),
+('AM', 'ARMENIA'),
+('AW', 'ARUBA'),
+('AU', 'AUSTRALIA'),
+('AT', 'AUSTRIA'),
+('AZ', 'AZERBAIJAN'),
+('BS', 'BAHAMAS'),
+('BH', 'BAHRAIN'),
+('BD', 'BANGLADESH'),
+('BB', 'BARBADOS'),
+('BY', 'BELARUS'),
+('BE', 'BELGIUM'),
+('BZ', 'BELIZE'),
+('BJ', 'BENIN'),
+('BM', 'BERMUDA'),
+('BT', 'BHUTAN'),
+('BO', 'BOLIVIA'),
+('BA', 'BOSNIA AND HERZEGOVINA'),
+('BW', 'BOTSWANA'),
+('BV', 'BOUVET ISLAND'),
+('BR', 'BRAZIL'),
+('IO', 'BRITISH INDIAN OCEAN TERRITORY'),
+('BN', 'BRUNEI DARUSSALAM'),
+('BG', 'BULGARIA'),
+('BF', 'BURKINA FASO'),
+('BI', 'BURUNDI'),
+('KH', 'CAMBODIA'),
+('CM', 'CAMEROON'),
+('CA', 'CANADA'),
+('CV', 'CAPE VERDE'),
+('KY', 'CAYMAN ISLANDS'),
+('CF', 'CENTRAL AFRICAN REPUBLIC'),
+('TD', 'CHAD'),
+('CL', 'CHILE'),
+('CN', 'CHINA'),
+('CX', 'CHRISTMAS ISLAND'),
+('CC', 'COCOS (KEELING) ISLANDS'),
+('CO', 'COLOMBIA'),
+('KM', 'COMOROS'),
+('CG', 'CONGO'),
+('CD', 'CONGO, THE DEMOCRATIC REPUBLIC OF THE'),
+('CK', 'COOK ISLANDS'),
+('CR', 'COSTA RICA'),
+('CI', 'COTE D''IVOIRE'),
+('HR', 'CROATIA'),
+('CU', 'CUBA'),
+('CY', 'CYPRUS'),
+('CZ', 'CZECHIA'),
+('DK', 'DENMARK'),
+('DJ', 'DJIBOUTI'),
+('DM', 'DOMINICA'),
+('DO', 'DOMINICAN REPUBLIC'),
+('EC', 'ECUADOR'),
+('EG', 'EGYPT'),
+('SV', 'EL SALVADOR'),
+('GQ', 'EQUATORIAL GUINEA'),
+('ER', 'ERITREA'),
+('EE', 'ESTONIA'),
+('ET', 'ETHIOPIA'),
+('FK', 'FALKLAND ISLANDS (MALVINAS)'),
+('FO', 'FAROE ISLANDS'),
+('FJ', 'FIJI'),
+('FI', 'FINLAND'),
+('FR', 'FRANCE'),
+('GF', 'FRENCH GUIANA'),
+('PF', 'FRENCH POLYNESIA'),
+('TF', 'FRENCH SOUTHERN TERRITORIES'),
+('GA', 'GABON'),
+('GM', 'GAMBIA'),
+('GE', 'GEORGIA'),
+('DE', 'GERMANY'),
+('GH', 'GHANA'),
+('GI', 'GIBRALTAR'),
+('GR', 'GREECE'),
+('GL', 'GREENLAND'),
+('GD', 'GRENADA'),
+('GP', 'GUADELOUPE'),
+('GU', 'GUAM'),
+('GT', 'GUATEMALA'),
+('GN', 'GUINEA'),
+('GW', 'GUINEA-BISSAU'),
+('GY', 'GUYANA'),
+('HT', 'HAITI'),
+('HM', 'HEARD ISLAND AND MCDONALD ISLANDS'),
+('VA', 'VATICAN CITY STATE'),
+('HN', 'HONDURAS'),
+('HK', 'HONG KONG'),
+('HU', 'HUNGARY'),
+('IS', 'ICELAND'),
+('IN', 'INDIA'),
+('ID', 'INDONESIA'),
+('IR', 'IRAN'),
+('IQ', 'IRAQ'),
+('IE', 'IRELAND'),
+('IL', 'ISRAEL'),
+('IT', 'ITALY'),
+('JM', 'JAMAICA'),
+('JP', 'JAPAN'),
+('JO', 'JORDAN'),
+('KZ', 'KAZAKHSTAN'),
+('KE', 'KENYA'),
+('KI', 'KIRIBATI'),
+('KP', 'NORTH KOREA'),
+('KR', 'SOUTH KOREA'),
+('KW', 'KUWAIT'),
+('KG', 'KYRGYZSTAN'),
+('LA', 'LAO'),
+('LV', 'LATVIA'),
+('LB', 'LEBANON'),
+('LS', 'LESOTHO'),
+('LR', 'LIBERIA'),
+('LY', 'LIBYAN ARAB JAMAHIRIYA'),
+('LI', 'LIECHTENSTEIN'),
+('LT', 'LITHUANIA'),
+('LU', 'LUXEMBOURG'),
+('MO', 'MACAO'),
+('MK', 'NORTH MACEDONIA'),
+('MG', 'MADAGASCAR'),
+('MW', 'MALAWI'),
+('MY', 'MALAYSIA'),
+('MV', 'MALDIVES'),
+('ML', 'MALI'),
+('MT', 'MALTA'),
+('MH', 'MARSHALL ISLANDS'),
+('MQ', 'MARTINIQUE'),
+('MR', 'MAURITANIA'),
+('MU', 'MAURITIUS'),
+('YT', 'MAYOTTE'),
+('MX', 'MEXICO'),
+('FM', 'MICRONESIA'),
+('MD', 'MOLDOVA'),
+('MC', 'MONACO'),
+('MN', 'MONGOLIA'),
+('MS', 'MONTSERRAT'),
+('MA', 'MOROCCO'),
+('MZ', 'MOZAMBIQUE'),
+('MM', 'MYANMAR'),
+('NA', 'NAMIBIA'),
+('NR', 'NAURU'),
+('NP', 'NEPAL'),
+('NL', 'NETHERLANDS'),
+('AN', 'NETHERLANDS ANTILLES'),
+('NC', 'NEW CALEDONIA'),
+('NZ', 'NEW ZEALAND'),
+('NI', 'NICARAGUA'),
+('NE', 'NIGER'),
+('NG', 'NIGERIA'),
+('NU', 'NIUE'),
+('NF', 'NORFOLK ISLAND'),
+('MP', 'NORTHERN MARIANA ISLANDS'),
+('NO', 'NORWAY'),
+('OM', 'OMAN'),
+('PK', 'PAKISTAN'),
+('PW', 'PALAU'),
+('PS', 'PALESTINE'),
+('PA', 'PANAMA'),
+('PG', 'PAPUA NEW GUINEA'),
+('PY', 'PARAGUAY'),
+('PE', 'PERU'),
+('PH', 'PHILIPPINES'),
+('PN', 'PITCAIRN'),
+('PL', 'POLAND'),
+('PT', 'PORTUGAL'),
+('PR', 'PUERTO RICO'),
+('QA', 'QATAR'),
+('RE', 'REUNION'),
+('RO', 'ROMANIA'),
+('RU', 'RUSSIAN FEDERATION'),
+('RW', 'RWANDA'),
+('SH', 'SAINT HELENA'),
+('KN', 'SAINT KITTS AND NEVIS'),
+('LC', 'SAINT LUCIA'),
+('PM', 'SAINT PIERRE AND MIQUELON'),
+('VC', 'SAINT VINCENT AND THE GRENADINES'),
+('WS', 'SAMOA'),
+('SM', 'SAN MARINO'),
+('ST', 'SAO TOME AND PRINCIPE'),
+('SA', 'SAUDI ARABIA'),
+('SN', 'SENEGAL'),
+('RS', 'SERBIA'),
+('SC', 'SEYCHELLES'),
+('SL', 'SIERRA LEONE'),
+('SG', 'SINGAPORE'),
+('SK', 'SLOVAKIA'),
+('SI', 'SLOVENIA'),
+('SB', 'SOLOMON ISLANDS'),
+('SO', 'SOMALIA'),
+('ZA', 'SOUTH AFRICA'),
+('GS', 'SOUTH GEORGIA AND THE SOUTH SANDWICH ISLANDS'),
+('ES', 'SPAIN'),
+('LK', 'SRI LANKA'),
+('SD', 'SUDAN'),
+('SR', 'SURINAME'),
+('SJ', 'SVALBARD AND JAN MAYEN'),
+('SZ', 'SWAZILAND'),
+('SE', 'SWEDEN'),
+('CH', 'SWITZERLAND'),
+('SY', 'SYRIAN ARAB REPUBLIC'),
+('TW', 'TAIWAN'),
+('TJ', 'TAJIKISTAN'),
+('TZ', 'TANZANIA'),
+('TH', 'THAILAND'),
+('TL', 'TIMOR-LESTE'),
+('TG', 'TOGO'),
+('TK', 'TOKELAU'),
+('TO', 'TONGA'),
+('TT', 'TRINIDAD AND TOBAGO'),
+('TN', 'TUNISIA'),
+('TR', 'TURKEY'),
+('TM', 'TURKMENISTAN'),
+('TC', 'TURKS AND CAICOS ISLANDS'),
+('TV', 'TUVALU'),
+('UG', 'UGANDA'),
+('UA', 'UKRAINE'),
+('AE', 'UNITED ARAB EMIRATES'),
+('GB', 'UNITED KINGDOM'),
+('US', 'UNITED STATES'),
+('UM', 'UNITED STATES MINOR OUTLYING ISLANDS'),
+('UY', 'URUGUAY'),
+('UZ', 'UZBEKISTAN'),
+('VU', 'VANUATU'),
+('VE', 'VENEZUELA'),
+('VN', 'VIET NAM'),
+('VG', 'VIRGIN ISLANDS, BRITISH'),
+('VI', 'VIRGIN ISLANDS, U.S.'),
+('WF', 'WALLIS AND FUTUNA'),
+('EH', 'WESTERN SAHARA'),
+('YE', 'YEMEN'),
+('ZM', 'ZAMBIA'),
+('ZW', 'ZIMBABWE'),
+('ME', 'MONTENEGRO'),
+('XK', 'KOSOVO'),
+('AX', 'ALAND ISLANDS'),
+('BQ', 'BONAIRE'),
+('CW', 'CURACAO'),
+('GG', 'GUERNSEY'),
+('IM', 'ISLE OF MAN'),
+('JE', 'JERSEY'),
+('BL', 'SAINT BARTHELEMY'),
+('MF', 'SAINT MARTIN'),
+('SX', 'SINT MAARTEN'),
+('SS', 'SOUTH SUDAN');
\ No newline at end of file
diff --git a/backend/database/init.sql b/backend/database/init.sql
index 12a2487..345cee3 100644
--- a/backend/database/init.sql
+++ b/backend/database/init.sql
@@ -4,6 +4,7 @@ DROP TABLE IF EXISTS records_sp;
 DROP TABLE IF EXISTS records_mp;
 DROP TABLE IF EXISTS maps;
 DROP TABLE IF EXISTS users;
+DROP TABLE IF EXISTS countries;
 
 DROP TABLE IF EXISTS demos;
 
@@ -69,4 +70,10 @@ CREATE TABLE titles (
   title_name TEXT NOT NULL,
   PRIMARY KEY (user_id),
   FOREIGN KEY (user_id) REFERENCES users(steam_id)
+);
+
+CREATE TABLE countries (
+  country_code CHAR(2),
+  country_name TEXT NOT NULL,
+  PRIMARY KEY (country_code)
 );
\ No newline at end of file
diff --git a/backend/routes/routes.go b/backend/routes/routes.go
index e9de891..43dfd65 100644
--- a/backend/routes/routes.go
+++ b/backend/routes/routes.go
@@ -16,6 +16,7 @@ func InitRoutes(router *gin.Engine) {
                 v1.GET("/", middleware.CheckAuth, controllers.Home)
                 v1.GET("/login", controllers.Login)
                 v1.GET("/profile", middleware.CheckAuth, controllers.Profile)
+                v1.PUT("/profile", middleware.CheckAuth, controllers.UpdateCountryCode)
                 v1.GET("/user/:id", middleware.CheckAuth, controllers.FetchUser)
                 v1.GET("/demo", controllers.DownloadDemoWithID)
                 v1.POST("/maps/:id/record", middleware.CheckAuth, controllers.CreateRecordWithDemo)