72 lines
2.0 KiB
Go
72 lines
2.0 KiB
Go
|
package triton
|
||
|
|
||
|
import (
|
||
|
"encoding/json"
|
||
|
"fmt"
|
||
|
"net/http"
|
||
|
|
||
|
"github.com/hashicorp/errwrap"
|
||
|
)
|
||
|
|
||
|
type ConfigClient struct {
|
||
|
*Client
|
||
|
}
|
||
|
|
||
|
// Config returns a c used for accessing functions pertaining
|
||
|
// to Config functionality in the Triton API.
|
||
|
func (c *Client) Config() *ConfigClient {
|
||
|
return &ConfigClient{c}
|
||
|
}
|
||
|
|
||
|
// Config represents configuration for your account.
|
||
|
type Config struct {
|
||
|
// DefaultNetwork is the network that docker containers are provisioned on.
|
||
|
DefaultNetwork string `json:"default_network"`
|
||
|
}
|
||
|
|
||
|
type GetConfigInput struct{}
|
||
|
|
||
|
// GetConfig outputs configuration for your account.
|
||
|
func (client *ConfigClient) GetConfig(input *GetConfigInput) (*Config, error) {
|
||
|
respReader, err := client.executeRequest(http.MethodGet, fmt.Sprintf("/%s/config", client.accountName), nil)
|
||
|
if respReader != nil {
|
||
|
defer respReader.Close()
|
||
|
}
|
||
|
if err != nil {
|
||
|
return nil, errwrap.Wrapf("Error executing GetConfig request: {{err}}", err)
|
||
|
}
|
||
|
|
||
|
var result *Config
|
||
|
decoder := json.NewDecoder(respReader)
|
||
|
if err = decoder.Decode(&result); err != nil {
|
||
|
return nil, errwrap.Wrapf("Error decoding GetConfig response: {{err}}", err)
|
||
|
}
|
||
|
|
||
|
return result, nil
|
||
|
}
|
||
|
|
||
|
type UpdateConfigInput struct {
|
||
|
// DefaultNetwork is the network that docker containers are provisioned on.
|
||
|
DefaultNetwork string `json:"default_network"`
|
||
|
}
|
||
|
|
||
|
// UpdateConfig updates configuration values for your account.
|
||
|
// TODO(jen20) Work out a safe way to test this (after networks c implemented)
|
||
|
func (client *ConfigClient) UpdateConfig(input *UpdateConfigInput) (*Config, error) {
|
||
|
respReader, err := client.executeRequest(http.MethodPut, fmt.Sprintf("/%s/config", client.accountName), input)
|
||
|
if respReader != nil {
|
||
|
defer respReader.Close()
|
||
|
}
|
||
|
if err != nil {
|
||
|
return nil, errwrap.Wrapf("Error executing UpdateConfig request: {{err}}", err)
|
||
|
}
|
||
|
|
||
|
var result *Config
|
||
|
decoder := json.NewDecoder(respReader)
|
||
|
if err = decoder.Decode(&result); err != nil {
|
||
|
return nil, errwrap.Wrapf("Error decoding UpdateConfig response: {{err}}", err)
|
||
|
}
|
||
|
|
||
|
return result, nil
|
||
|
}
|