43 lines
966 B
Go
43 lines
966 B
Go
|
package yandeximport
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
|
||
|
|
"github.com/aws/aws-sdk-go/aws"
|
||
|
|
"github.com/aws/aws-sdk-go/aws/credentials"
|
||
|
|
"github.com/aws/aws-sdk-go/aws/session"
|
||
|
|
"github.com/aws/aws-sdk-go/service/s3"
|
||
|
|
)
|
||
|
|
|
||
|
|
const defaultS3Region = "ru-central1"
|
||
|
|
const defaultStorageEndpoint = "storage.yandexcloud.net"
|
||
|
|
|
||
|
|
func newYCStorageClient(storageEndpoint, accessKey, secretKey string) (*s3.S3, error) {
|
||
|
|
var creds *credentials.Credentials
|
||
|
|
|
||
|
|
if storageEndpoint == "" {
|
||
|
|
storageEndpoint = defaultStorageEndpoint
|
||
|
|
}
|
||
|
|
|
||
|
|
s3Config := &aws.Config{
|
||
|
|
Endpoint: aws.String(storageEndpoint),
|
||
|
|
Region: aws.String(defaultS3Region),
|
||
|
|
}
|
||
|
|
|
||
|
|
switch {
|
||
|
|
case accessKey != "" && secretKey != "":
|
||
|
|
creds = credentials.NewStaticCredentials(accessKey, secretKey, "")
|
||
|
|
default:
|
||
|
|
return nil, fmt.Errorf("either access or secret key not provided")
|
||
|
|
}
|
||
|
|
|
||
|
|
s3Config.Credentials = creds
|
||
|
|
newSession, err := session.NewSession(s3Config)
|
||
|
|
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
return s3.New(newSession), nil
|
||
|
|
}
|