Add literal functions

Author: aopoltorzhicky

tools/literal.go

@@ -31,9 +31,23 @@ func IsAddressLazy(address string) bool {
 	return len(address) == 36 && (strings.HasPrefix(address, "KT") || strings.HasPrefix(address, "tz"))
 }
 
+var (
+	addressRegex       = regexp.MustCompile("(tz|KT)[0-9A-Za-z]{34}")
+	operationHashRegex = regexp.MustCompile("(o)[0-9A-Za-z]{50}")
+	bigMapKeyHashRegex = regexp.MustCompile("(expr)[0-9A-Za-z]{50}")
+)
+
 // IsAddress -
 func IsAddress(str string) bool {
-	regexString := "(tz|KT)[0-9A-Za-z]{34}"
-	re := regexp.MustCompile(regexString)
-	return re.MatchString(str)
+	return addressRegex.MatchString(str)
+}
+
+// IsOperationHash -
+func IsOperationHash(str string) bool {
+	return operationHashRegex.MatchString(str)
+}
+
+// IsBigMapKeyHash -
+func IsBigMapKeyHash(str string) bool {
+	return bigMapKeyHashRegex.MatchString(str)
 }

tools/literal_test.go

@@ -1,6 +1,8 @@
 package tools
 
-import "testing"
+import (
+	"testing"
+)
 
 func TestIsContract(t *testing.T) {
 	tests := []struct {
@@ -20,6 +22,10 @@ func TestIsContract(t *testing.T) {
 			name:    "KT1Ap287P1NzsnToSJdA4aqSNjPomRaHBZSr",
 			address: "KT1Ap287P1NzsnToSJdA4aqSNjPomRaHBZSr",
 			want:    true,
+		}, {
+			name:    "expru2dKqDfZG8hu4wNGkiyunvq2hdSKuVYtcKta7BWP6Q18oNxKjS",
+			address: "expru2dKqDfZG8hu4wNGkiyunvq2hdSKuVYtcKta7BWP6Q18oNxKjS",
+			want:    false,
 		},
 	}
 	for _, tt := range tests {
@@ -30,3 +36,102 @@ func TestIsContract(t *testing.T) {
 		})
 	}
 }
+
+func TestIsBigMapKeyHash(t *testing.T) {
+	tests := []struct {
+		name string
+		str  string
+		want bool
+	}{
+		{
+			name: "KT1Ap287P1NzsnToSJdA4aqSNjPomRaHBZSr",
+			str:  "KT1Ap287P1NzsnToSJdA4aqSNjPomRaHBZSr",
+			want: false,
+		}, {
+			name: "exprtqoNj2hRg8PsPMaXLcy3dXjMM3B7nHKrRNqpfjbYpMbULbRj8k",
+			str:  "exprtqoNj2hRg8PsPMaXLcy3dXjMM3B7nHKrRNqpfjbYpMbULbRj8k",
+			want: true,
+		}, {
+			name: "expru2dKqDfZG8hu4wNGkiyunvq2hdSKuVYtcKta7BWP6Q18oNxKjS",
+			str:  "expru2dKqDfZG8hu4wNGkiyunvq2hdSKuVYtcKta7BWP6Q18oNxKjS",
+			want: true,
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			if got := IsBigMapKeyHash(tt.str); got != tt.want {
+				t.Errorf("IsBigMapKeyHash() = %v, want %v", got, tt.want)
+			}
+		})
+	}
+}
+
+func TestIsOperationHash(t *testing.T) {
+	tests := []struct {
+		name string
+		str  string
+		want bool
+	}{
+		{
+			name: "KT1Ap287P1NzsnToSJdA4aqSNjPomRaHBZSr",
+			str:  "KT1Ap287P1NzsnToSJdA4aqSNjPomRaHBZSr",
+			want: false,
+		}, {
+			name: "exprtqoNj2hRg8PsPMaXLcy3dXjMM3B7nHKrRNqpfjbYpMbULbRj8k",
+			str:  "exprtqoNj2hRg8PsPMaXLcy3dXjMM3B7nHKrRNqpfjbYpMbULbRj8k",
+			want: false,
+		}, {
+			name: "opDqhqYmqgmXTxcEcDXbJMWBThZkaQCovwV8BC3gwthEWYdPCWD",
+			str:  "opDqhqYmqgmXTxcEcDXbJMWBThZkaQCovwV8BC3gwthEWYdPCWD",
+			want: true,
+		}, {
+			name: "opRRiHEQacoet5rq7jgcd33K66bkj5qCdThxGnCQwyZtdFjZ8ph",
+			str:  "opRRiHEQacoet5rq7jgcd33K66bkj5qCdThxGnCQwyZtdFjZ8ph",
+			want: true,
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			if got := IsOperationHash(tt.str); got != tt.want {
+				t.Errorf("IsOperationHash() = %v, want %v", got, tt.want)
+			}
+		})
+	}
+}
+
+func TestIsAddress(t *testing.T) {
+	tests := []struct {
+		name string
+		str  string
+		want bool
+	}{
+		{
+			name: "KT1Ap287P1NzsnToSJdA4aqSNjPomRaHBZSr",
+			str:  "KT1Ap287P1NzsnToSJdA4aqSNjPomRaHBZSr",
+			want: true,
+		}, {
+			name: "exprtqoNj2hRg8PsPMaXLcy3dXjMM3B7nHKrRNqpfjbYpMbULbRj8k",
+			str:  "exprtqoNj2hRg8PsPMaXLcy3dXjMM3B7nHKrRNqpfjbYpMbULbRj8k",
+			want: false,
+		}, {
+			name: "opDqhqYmqgmXTxcEcDXbJMWBThZkaQCovwV8BC3gwthEWYdPCWD",
+			str:  "opDqhqYmqgmXTxcEcDXbJMWBThZkaQCovwV8BC3gwthEWYdPCWD",
+			want: false,
+		}, {
+			name: "opRRiHEQacoet5rq7jgcd33K66bkj5qCdThxGnCQwyZtdFjZ8ph",
+			str:  "opRRiHEQacoet5rq7jgcd33K66bkj5qCdThxGnCQwyZtdFjZ8ph",
+			want: false,
+		}, {
+			name: "tz1PUnJ3m435ZK4RTqhTEiSYF22YAUx5rEU1",
+			str:  "tz1PUnJ3m435ZK4RTqhTEiSYF22YAUx5rEU1",
+			want: true,
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			if got := IsAddress(tt.str); got != tt.want {
+				t.Errorf("IsAddress() = %v, want %v", got, tt.want)
+			}
+		})
+	}
+}