Replace IntToBytes(value int) and ToBytes() with a single ToBytes() function

common/db.go

@@ -82,15 +82,8 @@ func StringToDataTyp(str string) (DataTyp, error) {
 	}
 }
 
-// Convert DataTyp to []byte
-func (d DataTyp) ToBytes() []byte {
-	bytes := make([]byte, 2)
-	binary.BigEndian.PutUint16(bytes, uint16(d))
-	return bytes
-}
-
-// Convert int to []byte
-func IntToBytes(value int) []byte {
+// ToBytes converts DataTyp or int to a byte slice
+func ToBytes[T ~uint16 | int](value T) []byte {
 	bytes := make([]byte, 2)
 	binary.BigEndian.PutUint16(bytes, uint16(value))
 	return bytes

common/transfer_statements.go

@@ -61,9 +61,9 @@ func GetTransferData(ctx context.Context, db storage.PrefixDb, publicKey string,
 	data := make(map[DataTyp]string)
 
 	for _, key := range keys {
-		value, err := db.Get(ctx, key.ToBytes())
+		value, err := db.Get(ctx, ToBytes(key))
 		if err != nil {
-			return "", fmt.Errorf("failed to get %s: %v", key.ToBytes(), err)
+			return "", fmt.Errorf("failed to get %s: %v", ToBytes(key), err)
 		}
 		data[key] = string(value)
 	}

common/vouchers.go

@@ -68,9 +68,9 @@ func GetVoucherData(ctx context.Context, db storage.PrefixDb, input string) (*da
 	data := make(map[DataTyp]string)
 
 	for _, key := range keys {
-		value, err := db.Get(ctx, key.ToBytes())
+		value, err := db.Get(ctx, ToBytes(key))
 		if err != nil {
-			return nil, fmt.Errorf("failed to get %s: %v", key.ToBytes(), err)
+			return nil, fmt.Errorf("failed to get %s: %v", ToBytes(key), err)
 		}
 		data[key] = string(value)
 	}

common/vouchers_test.go

@@ -85,7 +85,7 @@ func TestGetVoucherData(t *testing.T) {
 		t.Fatal(err)
 	}
 
-	prefix := IntToBytes(visedb.DATATYPE_USERDATA)
+	prefix := ToBytes(visedb.DATATYPE_USERDATA)
 	spdb := storage.NewSubPrefixDb(db, prefix)
 
 	// Test voucher data
@@ -98,7 +98,7 @@ func TestGetVoucherData(t *testing.T) {
 
 	// Put the data
 	for key, value := range mockData {
-		err = spdb.Put(ctx, []byte(key.ToBytes()), []byte(value))
+		err = spdb.Put(ctx, []byte(ToBytes(key)), []byte(value))
 		if err != nil {
 			t.Fatal(err)
 		}