118 lines
2.7 KiB
Go
118 lines
2.7 KiB
Go
package remote
|
|
|
|
import (
|
|
"crypto/md5"
|
|
"fmt"
|
|
"os"
|
|
"strings"
|
|
|
|
artifactory "github.com/lusis/go-artifactory/src/artifactory.v401"
|
|
)
|
|
|
|
const ARTIF_TFSTATE_NAME = "terraform.tfstate"
|
|
|
|
func artifactoryFactory(conf map[string]string) (Client, error) {
|
|
userName, ok := conf["username"]
|
|
if !ok {
|
|
userName = os.Getenv("ARTIFACTORY_USERNAME")
|
|
if userName == "" {
|
|
return nil, fmt.Errorf(
|
|
"missing 'username' configuration or ARTIFACTORY_USERNAME environment variable")
|
|
}
|
|
}
|
|
password, ok := conf["password"]
|
|
if !ok {
|
|
password = os.Getenv("ARTIFACTORY_PASSWORD")
|
|
if password == "" {
|
|
return nil, fmt.Errorf(
|
|
"missing 'password' configuration or ARTIFACTORY_PASSWORD environment variable")
|
|
}
|
|
}
|
|
url, ok := conf["url"]
|
|
if !ok {
|
|
url = os.Getenv("ARTIFACTORY_URL")
|
|
if url == "" {
|
|
return nil, fmt.Errorf(
|
|
"missing 'url' configuration or ARTIFACTORY_URL environment variable")
|
|
}
|
|
}
|
|
repo, ok := conf["repo"]
|
|
if !ok {
|
|
return nil, fmt.Errorf(
|
|
"missing 'repo' configuration")
|
|
}
|
|
subpath, ok := conf["subpath"]
|
|
if !ok {
|
|
return nil, fmt.Errorf(
|
|
"missing 'subpath' configuration")
|
|
}
|
|
|
|
clientConf := &artifactory.ClientConfig{
|
|
BaseURL: url,
|
|
Username: userName,
|
|
Password: password,
|
|
}
|
|
nativeClient := artifactory.NewClient(clientConf)
|
|
|
|
return &ArtifactoryClient{
|
|
nativeClient: &nativeClient,
|
|
userName: userName,
|
|
password: password,
|
|
url: url,
|
|
repo: repo,
|
|
subpath: subpath,
|
|
}, nil
|
|
|
|
}
|
|
|
|
type ArtifactoryClient struct {
|
|
nativeClient *artifactory.ArtifactoryClient
|
|
userName string
|
|
password string
|
|
url string
|
|
repo string
|
|
subpath string
|
|
}
|
|
|
|
func (c *ArtifactoryClient) Get() (*Payload, error) {
|
|
p := fmt.Sprintf("%s/%s/%s", c.repo, c.subpath, ARTIF_TFSTATE_NAME)
|
|
output, err := c.nativeClient.Get(p, make(map[string]string))
|
|
if err != nil {
|
|
if strings.Contains(err.Error(), "404") {
|
|
return nil, nil
|
|
}
|
|
return nil, err
|
|
}
|
|
|
|
// TODO: migrate to using X-Checksum-Md5 header from artifactory
|
|
// needs to be exposed by go-artifactory first
|
|
|
|
hash := md5.Sum(output)
|
|
payload := &Payload{
|
|
Data: output,
|
|
MD5: hash[:md5.Size],
|
|
}
|
|
|
|
// If there was no data, then return nil
|
|
if len(payload.Data) == 0 {
|
|
return nil, nil
|
|
}
|
|
|
|
return payload, nil
|
|
}
|
|
|
|
func (c *ArtifactoryClient) Put(data []byte) error {
|
|
p := fmt.Sprintf("%s/%s/%s", c.repo, c.subpath, ARTIF_TFSTATE_NAME)
|
|
if _, err := c.nativeClient.Put(p, string(data), make(map[string]string)); err == nil {
|
|
return nil
|
|
} else {
|
|
return fmt.Errorf("Failed to upload state: %v", err)
|
|
}
|
|
}
|
|
|
|
func (c *ArtifactoryClient) Delete() error {
|
|
p := fmt.Sprintf("%s/%s/%s", c.repo, c.subpath, ARTIF_TFSTATE_NAME)
|
|
err := c.nativeClient.Delete(p)
|
|
return err
|
|
}
|