force country to be 2 letters

lineageos-infra · Feb 6, 2024 · cba3f53 · cba3f53
1 parent fe17279
commit cba3f53
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/main.go b/main.go
@@ -135,6 +135,13 @@ func main() {
 				return
 			}
 
+			if len(stat.Country) != 2 {
+				w.WriteHeader(http.StatusBadRequest)
+				w.Header().Add("Content-Type", "text/plain")
+				w.Write([]byte("country must be a two letter iso code"))
+				return
+			}
+
 			if stat.Country != "Unknown" {
 				stat.Country = strings.ToUpper(stat.Country)
 			}