distribution/registry/storage/driver/azure/parser.go

package azure

import (
	"errors"
	"fmt"

	"github.com/mitchellh/mapstructure"
)

const (
	defaultRealm                  = "core.windows.net"
	defaultCopyStatusPollMaxRetry = 5
	defaultCopyStatusPollDelay    = "100ms"
)

type Credentials struct {
	Type     string `mapstructure:"type"`
	ClientID string `mapstructure:"clientid"`
	TenantID string `mapstructure:"tenantid"`
	Secret   string `mapstructure:"secret"`
}

type Parameters struct {
	Container              string      `mapstructure:"container"`
	AccountName            string      `mapstructure:"accountname"`
	AccountKey             string      `mapstructure:"accountkey"`
	Credentials            Credentials `mapstructure:"credentials"`
	ConnectionString       string      `mapstructure:"connectionstring"`
	Realm                  string      `mapstructure:"realm"`
	RootDirectory          string      `mapstructure:"rootdirectory"`
	ServiceURL             string      `mapstructure:"serviceurl"`
	CopyStatusPollMaxRetry int         `mapstructure:"copy_status_poll_max_retry"`
	CopyStatusPollDelay    string      `mapstructure:"copy_status_poll_delay"`
}

func NewParameters(parameters map[string]interface{}) (*Parameters, error) {
	params := Parameters{
		Realm: defaultRealm,
	}
	if err := mapstructure.Decode(parameters, &params); err != nil {
		return nil, err
	}
	if params.AccountName == "" {
		return nil, errors.New("no accountname parameter provided")
	}
	if params.Container == "" {
		return nil, errors.New("no container parameter provider")
	}
	if params.ServiceURL == "" {
		params.ServiceURL = fmt.Sprintf("https://%s.blob.%s", params.AccountName, params.Realm)
	}
	if params.CopyStatusPollMaxRetry == 0 {
		params.CopyStatusPollMaxRetry = defaultCopyStatusPollMaxRetry
	}
	if params.CopyStatusPollDelay == "" {
		params.CopyStatusPollDelay = defaultCopyStatusPollDelay
	}
	return &params, nil
}