feat(helpers): IsHexUnsafe

internal/commands/start.go

@@ -8,8 +8,7 @@ import (
 var StartCommand = Command(func(context Context, responder Responder, _ helpers.Formatter) error {
 	token := context.Arguments
 
-	if len(token) == 32 {
-		// crutch
+	if len(token) == 32 && helpers.IsHexUnsafe(token) {
 		client := context.GetClient()
 		client.User.RefreshToken = crypto.Encrypt(token)
 

internal/helpers/helpers.go

@@ -74,7 +74,26 @@ func HumanizeLesson(lesson string) string {
 		return "Русский язык (консультация)"
 	case "решение задач повышенной сложности":
 		return "Математика (консультация)"
+	case "основы духовно-нравственной культуры народов россии":
+		return "ОДНКНР"
+	case "изобразительное искусство":
+		return "ИЗО"
+	case "Мировая художественная культура":
+		return "МХК"
+	case "Основы религиозных культур и светской этики":
+		return "ОРКСЭ"
 	default:
 		return lesson
 	}
 }
+
+// IsHexUnsafe only checks if the string contains valid hex characters
+func IsHexUnsafe(hex string) bool {
+	for i := 0; i < len(hex); i++ {
+		if (hex[i] < '0' || hex[i] > '9') && (hex[i] < 'a' || hex[i] > 'f') && (hex[i] < 'A' || hex[i] > 'F') {
+			return false
+		}
+	}
+
+	return true
+}

internal/helpers/helpers_test.go

@@ -62,15 +62,11 @@ func TestGetEnvFile(t *testing.T) {
 		panic(err)
 	}
 
-	_, err = file.WriteString("test")
-
-	if err != nil {
+	if _, err = file.WriteString("test"); err != nil {
 		panic(err)
 	}
 
-	err = os.Setenv(UnknownVariable+"_FILE", file.Name())
-
-	if err != nil {
+	if err = os.Setenv(UnknownVariable+"_FILE", file.Name()); err != nil {
 		panic(err)
 	}
 
@@ -108,3 +104,21 @@ func TestHumanizeLesson(t *testing.T) {
 		}
 	}
 }
+
+func TestIsHexUnsafe(t *testing.T) {
+	tests := []struct {
+		input    string
+		expected bool
+	}{
+		{"d813c247bb06c2c8ac84ef4231658b2b", true},
+		{"d813c247bb06c2c8ac84ef4231658b2я", false},
+	}
+
+	for _, test := range tests {
+		result := IsHexUnsafe(test.input)
+
+		if result != test.expected {
+			t.Errorf("IsHexUnsafe(%s) = %t; want %t", test.input, result, test.expected)
+		}
+	}
+}