Add bytes methods

Author: miguelmota

address.go

@@ -0,0 +1,49 @@
+package solsha3
+
+import (
+	"encoding/hex"
+	"reflect"
+
+	"github.com/ethereum/go-ethereum/common"
+)
+
+// Address address
+func Address(input interface{}) []byte {
+	switch v := input.(type) {
+	case common.Address:
+		return v.Bytes()[:]
+	case string:
+		v = removeHexPrefix(v)
+		if v == "" || v == "0" {
+			return []byte{0}
+		}
+
+		v = evenLengthHex(v)
+		decoded, err := hex.DecodeString(v)
+		if err != nil {
+			panic(err)
+		}
+
+		return decoded
+	case []byte:
+		return v
+	}
+
+	if isArray(input) {
+		return AddressArray(input)
+	}
+
+	return common.HexToAddress("").Bytes()[:]
+}
+
+// AddressArray address
+func AddressArray(input interface{}) []byte {
+	var values []byte
+	s := reflect.ValueOf(input)
+	for i := 0; i < s.Len(); i++ {
+		val := s.Index(i).Interface()
+		result := common.LeftPadBytes(Address(val), 32)
+		values = append(values, result...)
+	}
+	return values
+}

address_test.go

@@ -0,0 +1,62 @@
+package solsha3
+
+import (
+	"encoding/hex"
+	"fmt"
+	"testing"
+
+	"github.com/ethereum/go-ethereum/common"
+)
+
+func TestAddress(t *testing.T) {
+	t.Run("address", func(t *testing.T) {
+		for i, tt := range []struct {
+			in  []byte
+			out string
+		}{
+			{
+				SoliditySHA3(
+					[]string{"address"},
+					"0x0",
+				),
+				"bc36789e7a1e281436464229828f817d6612f7b477d66591ff96a9e064bcc98a",
+			},
+			{
+				SoliditySHA3(
+					[]string{"address"},
+					"0x0a",
+				),
+
+				"0ef9d8f8804d174666011a394cab7901679a8944d24249fd148a6a36071151f8",
+			},
+			{
+				SoliditySHA3(
+					[]string{"address"},
+					"0x12459c951127e0c374ff9105dda097662a027092",
+				),
+				"4b998b071d7bb74aee1ce2cdcc268cb0f6409b4a3387fc915617ec08415298ad",
+			},
+			{
+				SoliditySHA3(
+					[]string{"address"},
+					common.HexToAddress("0x12459c951127e0c374ff9105dda097662a027092"),
+				),
+				"4b998b071d7bb74aee1ce2cdcc268cb0f6409b4a3387fc915617ec08415298ad",
+			},
+			{
+				SoliditySHA3(
+					[]string{"string[]"},
+					[]string{"a", "b", "c"},
+				),
+				"4e03657aea45a94fc7d47ba826c8d667c0d1e6e33a64a036ec44f58fa12d6c45",
+			},
+		} {
+			t.Run(fmt.Sprintf("%v", i), func(t *testing.T) {
+				got := hex.EncodeToString(tt.in)
+				if got != tt.out {
+					t.Errorf("want %v, got %v", tt.out, got)
+				}
+			})
+		}
+	})
+}

bool.go

@@ -0,0 +1,36 @@
+package solsha3
+
+import (
+	"reflect"
+
+	"github.com/ethereum/go-ethereum/common"
+)
+
+// Bool bool
+func Bool(input interface{}) []byte {
+	switch v := input.(type) {
+	case bool:
+		if v {
+			return []byte{0x1}
+		}
+		return []byte{0x0}
+	}
+
+	if isArray(input) {
+		return BoolArray(input)
+	}
+
+	return []byte{0x0}
+}
+
+// BoolArray bool array
+func BoolArray(input interface{}) []byte {
+	var values []byte
+	s := reflect.ValueOf(input)
+	for i := 0; i < s.Len(); i++ {
+		val := s.Index(i).Interface()
+		result := common.LeftPadBytes(Bool(val), 32)
+		values = append(values, result...)
+	}
+	return values
+}

bool_test.go

@@ -0,0 +1,45 @@
+package solsha3
+
+import (
+	"encoding/hex"
+	"fmt"
+	"testing"
+)
+
+func TestBool(t *testing.T) {
+	t.Run("bool", func(t *testing.T) {
+		for i, tt := range []struct {
+			in  []byte
+			out string
+		}{
+			{
+				SoliditySHA3(
+					[]string{"bool"},
+					true,
+				),
+				"5fe7f977e71dba2ea1a68e21057beebb9be2ac30c6410aa38d4f3fbe41dcffd2",
+			},
+			{
+				SoliditySHA3(
+					[]string{"bool"},
+					false,
+				),
+				"bc36789e7a1e281436464229828f817d6612f7b477d66591ff96a9e064bcc98a",
+			},
+			{
+				SoliditySHA3(
+					[]string{"bool[]"},
+					[3]bool{true, false, true},
+				),
+				"5c6090c0461491a2941743bda5c3658bf1ea53bbd3edcde54e16205e18b45792",
+			},
+		} {
+			t.Run(fmt.Sprintf("%v", i), func(t *testing.T) {
+				got := hex.EncodeToString(tt.in)
+				if got != tt.out {
+					t.Errorf("want %v, got %v", tt.out, got)
+				}
+			})
+		}
+	})
+}