64 lines
1.5 KiB
Go
64 lines
1.5 KiB
Go
package artifactory
|
|
|
|
import (
|
|
"crypto/md5"
|
|
"fmt"
|
|
"strings"
|
|
|
|
"github.com/hashicorp/terraform/internal/states/remote"
|
|
artifactory "github.com/lusis/go-artifactory/src/artifactory.v401"
|
|
)
|
|
|
|
const ARTIF_TFSTATE_NAME = "terraform.tfstate"
|
|
|
|
type ArtifactoryClient struct {
|
|
nativeClient *artifactory.ArtifactoryClient
|
|
userName string
|
|
password string
|
|
url string
|
|
repo string
|
|
subpath string
|
|
}
|
|
|
|
func (c *ArtifactoryClient) Get() (*remote.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 := &remote.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
|
|
}
|