Skip to content

Add connection string support #10

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
uncycler wants to merge 2 commits into
base: main
Choose a base branch
from
Open

Add connection string support #10

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
95a72fa
Add connection string support
uncycler Mar 16, 2025
a016632
Fix name
Mar 17, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions README.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -11,6 +11,7 @@ Furthermore it is possible to filter for a certain Storage Container Names or/an

* `--accountName` or `-n`: (mandatory) provide the name of the Storage Account
* `--accessKey` or `-k`: (mandatory) provide the Access Key of Storage Account
* `--connectionString` or `-x`: (optional) provide the connection string of Storage Account (instead of working with accountName and accessKey)
* `--container` or `-c`: (optional) filter for a specific container by it names. Substring match
* `--blob` or `-b`: (optional) filter for specific blobs by it names. Substring match
* `--metadata-filter` or `-m`: (optional) filter for metadata <key:value>. Shows only the blobs matching at least one given filter.
Expand Down
67 changes: 39 additions & 28 deletions src/main.go
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -24,13 +24,14 @@ type storageAccount struct {
}

type arguments struct {
AccountName string
AccessKey string
MSI string
ContainerName string
BlobName string
ShowContent bool
MetadataFilter []string
AccountName string
AccessKey string
ConnectionString string
MSI string
ContainerName string
BlobName string
ShowContent bool
MetadataFilter []string
}

var largs = arguments{}
Expand All @@ -55,12 +56,15 @@ var foundContainer []myContainer
func init() {
rootCmd.Flags().StringVarP(&largs.AccountName, "accountName", "n", "", "accountName of the Storage Account")
rootCmd.Flags().StringVarP(&largs.AccessKey, "accessKey", "k", "", "accessKey for the Storage Account")
rootCmd.Flags().StringVarP(&largs.ConnectionString, "connectionString", "x", "", "connectionString for the Storage Account")
rootCmd.Flags().StringVarP(&largs.MSI, "msi", "i", "", "user assigned managed Identity to Access the Storage Account")
rootCmd.Flags().StringVarP(&largs.ContainerName, "container", "c", "", "filter for container name with substring match")
rootCmd.Flags().StringVarP(&largs.BlobName, "blob", "b", "", "filter for blob name with substring match")
rootCmd.Flags().BoolVar(&largs.ShowContent, "show-content", false, "downloads and prints content of blobs in addition to other logs")
rootCmd.Flags().StringSliceVarP(&largs.MetadataFilter, "metadata-filter", "m", []string{}, "OR filter for blob metadata. Structure is <key>:<value>")
rootCmd.MarkFlagRequired("accountName")
if largs.ConnectionString != "" {
rootCmd.MarkFlagRequired("accountName")
}
rootCmd.SetVersionTemplate(getVersion())
}

Expand All @@ -74,31 +78,38 @@ func exec(args arguments) {
URL, _ := url.Parse(fmt.Sprintf(storageURLTemplate, args.AccountName))

// if AccessKey is provided use them
if len(args.AccessKey) > 0 {
keyCredentials, authErr := azblob.NewSharedKeyCredential(args.AccountName, args.AccessKey)
if authErr != nil {
log.Fatal("Error while Authentication with AccessKey", authErr)
if len(args.ConnectionString) > 0 {
client, clientError = azblob.NewClientFromConnectionString(args.ConnectionString, nil)
if clientError != nil {
log.Fatal("Error while Authentication with ConnectionString", clientError)
}
client, clientError = azblob.NewClientWithSharedKeyCredential(URL.String(), keyCredentials, nil)
} else {
var authErr error
var credentials *azidentity.ManagedIdentityCredential

// if user assigned managed identity is provided use them
if len(args.MSI) > 0 {
options := azidentity.ManagedIdentityCredentialOptions{}
options.ID = azidentity.ClientID(args.MSI)
credentials, authErr = azidentity.NewManagedIdentityCredential(&options)
if len(args.AccessKey) > 0 {
keyCredentials, authErr := azblob.NewSharedKeyCredential(args.AccountName, args.AccessKey)
if authErr != nil {
log.Fatal("Error while Authentication with AccessKey", authErr)
}
client, clientError = azblob.NewClientWithSharedKeyCredential(URL.String(), keyCredentials, nil)
} else {
// for system assigned managed identity we don't need to pass anything
credentials, authErr = azidentity.NewManagedIdentityCredential(nil)
}
var authErr error
var credentials *azidentity.ManagedIdentityCredential

// if user assigned managed identity is provided use them
if len(args.MSI) > 0 {
options := azidentity.ManagedIdentityCredentialOptions{}
options.ID = azidentity.ClientID(args.MSI)
credentials, authErr = azidentity.NewManagedIdentityCredential(&options)
} else {
// for system assigned managed identity we don't need to pass anything
credentials, authErr = azidentity.NewManagedIdentityCredential(nil)
}

if authErr != nil {
log.Fatal("Error while Authentication with DefaultCredentials", authErr)
}
if authErr != nil {
log.Fatal("Error while Authentication with DefaultCredentials", authErr)
}

client, clientError = azblob.NewClient(URL.String(), credentials, nil)
client, clientError = azblob.NewClient(URL.String(), credentials, nil)
}
}

if clientError != nil {
Expand Down