variable name check

Assembler/config/generalConfig.go

@@ -4,3 +4,5 @@ const AmntBitsCode = 8
 const AmntBitsParam = 8
 
 var Testing bool = false
+
+const VariableNameRegex = "^[a-zA-Z][a-zA-Z0-9_]*$"

Assembler/core/commandAssembler.go

@@ -76,7 +76,7 @@ func getSecondWord(commandName string, words []string, acceptStringParam bool) (
 
 	strParam := words[1]
 
-	if utils.IsVariableName(strParam) {
+	if acceptStringParam && utils.IsValidVarName(strParam) {
 		param := data.NewStringParam(strParam)
 		return &param, nil
 	}

Assembler/utils/validations.go

@@ -1,10 +1,17 @@
 package utils
 
 import (
+	"assembler/config"
 	"math"
+	"regexp"
 )
 
 func IsOverflow(num uint, availableBits int) bool {
 	largestNumber := math.Pow(2, float64(availableBits))
 	return num >= uint(math.Floor(largestNumber))
 }
+
+func IsValidVarName(str string) bool {
+	matched, err := regexp.MatchString(config.VariableNameRegex, str)
+	return matched && err == nil
+}

Assembler/utils/validations_test.go

@@ -26,3 +26,38 @@ func TestOverflow(t *testing.T) {
 		}
 	}
 }
+
+func TestIsValidVarName(t *testing.T) {
+	var tests = []struct {
+		param    string
+		expected bool
+	}{
+		{"", false},
+		{"a", true},
+		{"var", true},
+		{"Var", true},
+		{"vaR", true},
+		{"VAR", true},
+		{"var_able", true},
+		{"var_Able", true},
+		{"var_ABLE", true},
+		{"va0r_4ABLE1", true},
+		{"va.", false},
+		{"va.r", false},
+		{"va-r", false},
+		{"va*r", false},
+		{"va^r", false},
+		{"va&r", false},
+		{"&var", false},
+		{"var&", false},
+		{"jmp", true},
+	}
+
+	for _, test := range tests {
+		got := IsValidVarName(test.param)
+
+		if test.expected != got {
+			t.Errorf("For var name '%s': Expected: %t, Got: %t", test.param, test.expected, got)
+		}
+	}
+}