35 lines
633 B
Go
35 lines
633 B
Go
package digitalocean
|
|
|
|
import (
|
|
"log"
|
|
"os"
|
|
|
|
"github.com/pearkes/digitalocean"
|
|
)
|
|
|
|
type Config struct {
|
|
Token string `mapstructure:"token"`
|
|
}
|
|
|
|
// Client() returns a new client for accessing digital
|
|
// ocean.
|
|
//
|
|
func (c *Config) Client() (*digitalocean.Client, error) {
|
|
|
|
// If we have env vars set (like in the acc) tests,
|
|
// we need to override the values passed in here.
|
|
if v := os.Getenv("DIGITALOCEAN_TOKEN"); v != "" {
|
|
c.Token = v
|
|
}
|
|
|
|
client, err := digitalocean.NewClient(c.Token)
|
|
|
|
log.Printf("[INFO] DigitalOcean Client configured for URL: %s", client.URL)
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return client, nil
|
|
}
|