Merge pull request #43 from Flajt/fix-HandleNewKey

Fix handle new key
Flajt · Oct 20, 2023 · f1ea46e · f1ea46e
2 parents 4a00139 + e8c9892
commit f1ea46e
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/get_new_key/getNewKey.go → get_new_key/get_new_key.go b/get_new_key/getNewKey.go → get_new_key/get_new_key.go
@@ -32,7 +32,14 @@ func HandleGetNewKey(w http.ResponseWriter, r *http.Request) {
 		if success {
 			log.Info().Msg("Valid App Check token,adding api key to response")
 			w.WriteHeader(http.StatusOK)
-			w.Write([]byte(apiKeys[1]))
+			if len(apiKeys) == 1 {
+				w.Write([]byte(apiKeys[0]))
+			} else if len(apiKeys) == 2 {
+				w.Write([]byte(apiKeys[1]))
+			} else {
+				log.Debug().Msgf("%d api keys found", len(apiKeys))
+				panic("Invalid number of api keys found!")
+			}
 			return
 		} else {
 			log.Error().Msg("Invalid App Check token")