[Tables] connection string parser

sdk/tables/aztable/connection_string.go

@@ -0,0 +1,101 @@
+// Copyright (c) Microsoft Corporation. All rights reserved.
+// Licensed under the MIT License.
+
+package aztable
+
+import (
+	"errors"
+	"fmt"
+	"strings"
+
+	"github.com/Azure/azure-sdk-for-go/sdk/azcore"
+)
+
+var ErrConnectionString = errors.New("connection string is either blank or malformed")
+
+func NewTableClientFromConnectionString(tableName string, connectionString string, options *TableClientOptions) (*TableClient, error) {
+	endpoint, credential, err := parseConnectionString(connectionString)
+	if err != nil {
+		return nil, err
+	}
+	return NewTableClient(tableName, endpoint, *credential, options)
+}
+
+func NewTableServiceClientFrommConnectionString(connectionString string, options *TableClientOptions) (*TableServiceClient, error) {
+	endpoint, credential, err := parseConnectionString(connectionString)
+	if err != nil {
+		return nil, err
+	}
+	return NewTableServiceClient(endpoint, *credential, options)
+}
+
+func convertConnStrToMap(connStr string) (map[string]string, error) {
+	ret := make(map[string]string)
+	connStr = strings.TrimRight(connStr, ";")
+
+	splitString := strings.Split(connStr, ";")
+	if len(splitString) == 0 {
+		return ret, ErrConnectionString
+	}
+	for _, stringPart := range splitString {
+		parts := strings.Split(stringPart, "=")
+		if len(parts) != 2 {
+			return ret, ErrConnectionString
+		}
+		ret[parts[0]] = parts[1]
+	}
+	return ret, nil
+}
+
+func parseConnectionString(connStr string) (string, *azcore.Credential, error) {
+	var serviceURL string
+	var cred azcore.Credential
+
+	connStrMap, err := convertConnStrToMap(connStr)
+	if err != nil {
+		return "", nil, err
+	}
+
+	accountName, ok := connStrMap["AccountName"]
+	if !ok {
+		return "", nil, ErrConnectionString
+	}
+	accountKey, ok := connStrMap["AccountKey"]
+	if !ok {
+		return "", nil, ErrConnectionString
+	}
+
+	if accountName == "" || accountKey == "" {
+		// Try sharedaccesssignature
+		sharedAccessSignature, ok := connStrMap["sharedaccesssignature"]
+		if !ok {
+			return serviceURL, nil, ErrConnectionString
+		}
+		return sharedAccessSignature, nil, errors.New("there is not support for SharedAccessSignature yet")
+		// cred = azcore.SharedAccessSignature(sharedAccessSignature)
+		// TODO: fix this when shared access signatures are added.
+	}
+	defaultProtocol, ok := connStrMap["DefaultEndpointsProtocol"]
+	if !ok {
+		defaultProtocol = "https"
+	}
+
+	endpointSuffix, ok := connStrMap["EndpointSuffix"]
+	if !ok {
+		endpointSuffix = "core.windows.net"
+	}
+
+	tableEndpoint, ok := connStrMap["TableEndpoint"]
+	if ok {
+		cred, err = NewSharedKeyCredential(accountName, accountKey)
+		return tableEndpoint, &cred, err
+	}
+	serviceURL = fmt.Sprintf("%v://%v.table.%v", defaultProtocol, accountName, endpointSuffix)
+
+	cred, err = NewSharedKeyCredential(accountName, accountKey)
+	if err != nil {
+		return "", nil, err
+	}
+
+	return serviceURL, &cred, nil
+}

sdk/tables/aztable/connection_string_test.go

@@ -0,0 +1,70 @@
+// Copyright (c) Microsoft Corporation. All rights reserved.
+// Licensed under the MIT License.
+
+package aztable
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/require"
+)
+
+func TestConnectionStringParser(t *testing.T) {
+	require := require.New(t)
+
+	connStr := "DefaultEndpointsProtocol=https;AccountName=dummyaccount;AccountKey=secretkeykey;EndpointSuffix=core.windows.net"
+	serviceURL, cred, err := parseConnectionString(connStr)
+	require.NoError(err)
+	require.Equal(serviceURL, "https://dummyaccount.table.core.windows.net")
+	require.NotNil(cred)
+}
+
+func TestConnectionStringParserHTTP(t *testing.T) {
+	require := require.New(t)
+
+	connStr := "DefaultEndpointsProtocol=http;AccountName=dummyaccount;AccountKey=secretkeykey;EndpointSuffix=core.windows.net"
+	serviceURL, cred, err := parseConnectionString(connStr)
+	require.NoError(err)
+	require.Equal(serviceURL, "http://dummyaccount.table.core.windows.net")
+	require.NotNil(cred)
+}
+
+func TestConnectionStringParserBasic(t *testing.T) {
+	require := require.New(t)
+
+	connStr := "AccountName=dummyaccount;AccountKey=secretkeykey"
+	serviceURL, cred, err := parseConnectionString(connStr)
+	require.NoError(err)
+	require.Equal(serviceURL, "https://dummyaccount.table.core.windows.net")
+	require.NotNil(cred)
+}
+
+func TestConnectionStringParserCustomDomain(t *testing.T) {
+	require := require.New(t)
+
+	connStr := "AccountName=dummyaccount;AccountKey=secretkeykey;TableEndpoint=www.mydomain.com;"
+	serviceURL, cred, err := parseConnectionString(connStr)
+	require.NoError(err)
+	require.Equal(serviceURL, "www.mydomain.com")
+	require.NotNil(cred)
+}
+
+func TestConnectionStringParserInvalid(t *testing.T) {
+	badConnectionStrings := []string{
+		"",
+		"foobar",
+		"foo;bar;baz",
+		"foo=;bar=;",
+		"=",
+		";",
+		"=;==",
+		"foobar=baz=foo"
+	}
+	require := require.New(t)
+
+	for _, badConnStr := range badConnectionStrings {
+		_, _, err := parseConnectionString(badConnStr)
+		require.Error(err)
+		require.Contains(err.Error(), ErrConnectionString.Error())
+	}
+}