Merge pull request #1 from trustwallet/vikmeup/asset-image

Add asset image

asset/image.go

@@ -0,0 +1,21 @@
+package asset
+
+import (
+	"fmt"
+
+	"github.com/trustwallet/go-primitives/coin"
+)
+
+func GetImageURL(endpoint, asset string) string {
+	coinId, tokenId, err := ParseID(asset)
+	if err != nil {
+		return ""
+	}
+	if c, ok := coin.Coins[coinId]; ok {
+		if len(tokenId) > 0 {
+			return fmt.Sprintf("%s/blockchains/%s/assets/%s/logo.png", endpoint, c.Handle, tokenId)
+		}
+		return fmt.Sprintf("%s/blockchains/%s/info/logo.png", endpoint, c.Handle)
+	}
+	return ""
+}

asset/image_test.go

@@ -0,0 +1,47 @@
+package asset
+
+import "testing"
+
+func TestGetImageURL(t *testing.T) {
+	type args struct {
+		endpoint string
+		asset    string
+	}
+	tests := []struct {
+		name string
+		args args
+		want string
+	}{
+		{
+			"Test coin",
+			args{
+				endpoint: "https://assets.com",
+				asset:    "c60",
+			},
+			"https://assets.com/blockchains/ethereum/info/logo.png",
+		},
+		{
+			"Test coin",
+			args{
+				endpoint: "https://assets.com",
+				asset:    "c60_t123",
+			},
+			"https://assets.com/blockchains/ethereum/assets/123/logo.png",
+		},
+		{
+			"Test invalid coin",
+			args{
+				endpoint: "https://assets.com",
+				asset:    "c123",
+			},
+			"",
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			if got := GetImageURL(tt.args.endpoint, tt.args.asset); got != tt.want {
+				t.Errorf("GetImageURL() = %v, want %v", got, tt.want)
+			}
+		})
+	}
+}