start portainer stack by name

pull/1/head
Cian Hatton 3 years ago
parent 1f2526c31b
commit c8741d96eb

@ -12,29 +12,96 @@ const (
applicationJson = "application/json" applicationJson = "application/json"
) )
// api docs
// https://app.swaggerhub.com/apis/portainer/portainer-ce/2.9.3
type PortainerClient struct { type PortainerClient struct {
authToken string authToken string
baseUrl string baseUrl string
} }
func NewPortainerClient(baseUrl string) *PortainerClient { type Credentials struct {
return &PortainerClient{ Username string `json:"username"`
baseUrl: baseUrl, Password string `json:"password"`
BaseUrl string `json:"baseUrl"`
}
func NewPortainerClient(creds Credentials) (*PortainerClient, error) {
c := &PortainerClient{
baseUrl: creds.BaseUrl,
} }
return c, c.Login(creds.Username, creds.Password)
} }
func (c *PortainerClient) IsLoggedIn() bool { func (c *PortainerClient) IsLoggedIn() bool {
return c.authToken != "" return c.authToken != ""
} }
func (c *PortainerClient) Login(username, password string) error {
payload := map[string]string{"Username": username, "Password": password}
body, err := c.post("api/auth", payload)
if err != nil {
return err
}
type JwtToken struct {
Token string `json:"jwt"`
}
token := JwtToken{}
if err := json.Unmarshal(body, &token); err != nil {
return err
}
c.authToken = token.Token
return nil
}
func (c *PortainerClient) GetAllStacks() ([]Stack, error) {
b, err := c.get("stacks")
if err != nil {
return nil, err
}
var stacks []Stack
if err := json.Unmarshal(b, &stacks); err != nil {
return nil, err
}
return stacks, nil
}
func (c *PortainerClient) GetStackByName(name string) (*Stack, error) {
stacks, err := c.GetAllStacks()
if err != nil {
return nil, err
}
for _, s := range stacks {
if s.Name == name {
return &s, nil
}
}
return nil, nil
}
func (c *PortainerClient) StartStack(stackId int) error {
url := fmt.Sprintf("api/stacks/%d/start", stackId)
b, err := c.post(url, nil)
fmt.Println(string(b))
return err
}
func (c *PortainerClient) post(path string, payload interface{}) ([]byte, error) { func (c *PortainerClient) post(path string, payload interface{}) ([]byte, error) {
jsonBytes, err := json.Marshal(payload) jsonBytes, err := json.Marshal(payload)
if err != nil { if err != nil {
return nil, err return nil, err
} }
url := fmt.Sprintf("%s/%s", c.baseUrl, path) url := fmt.Sprintf("%s/%s", c.baseUrl, path)
resp, err := http.Post(url, applicationJson, bytes.NewBuffer(jsonBytes)) // Create a Bearer string by appending string access token
var bearer = "Bearer " + c.authToken
// Create a new request using http
req, err := http.NewRequest("POST", url, bytes.NewBuffer(jsonBytes))
// add authorization header to the req
req.Header.Add("Authorization", bearer)
// Send req using http Client
client := &http.Client{}
resp, err := client.Do(req)
if err != nil { if err != nil {
return nil, err return nil, err
} }
@ -46,19 +113,28 @@ func (c *PortainerClient) post(path string, payload interface{}) ([]byte, error)
return body, nil return body, nil
} }
func (c *PortainerClient) Login(username, password string) error { func (c *PortainerClient) get(path string) ([]byte, error) {
payload := map[string]string{"Username": username, "Password": password} url := fmt.Sprintf("%s/api/%s", c.baseUrl, path)
body, err := c.post("api/auth", payload)
// Create a Bearer string by appending string access token
var bearer = "Bearer " + c.authToken
// Create a new request using http
req, err := http.NewRequest("GET", url, nil)
// add authorization header to the req
req.Header.Add("Authorization", bearer)
// Send req using http Client
client := &http.Client{}
resp, err := client.Do(req)
if err != nil { if err != nil {
return err return nil, err
}
type JwtToken struct {
Token string `json:"jwt"`
} }
token := JwtToken{} defer resp.Body.Close()
if err := json.Unmarshal(body, &token); err != nil {
return err body, err := ioutil.ReadAll(resp.Body)
if err != nil {
return nil, err
} }
c.authToken = token.Token return body, nil
return nil
} }

@ -0,0 +1,34 @@
package client
type Stack struct {
ID int `json:"Id"`
Name string `json:"Name"`
Type int `json:"Type"`
EndpointID int `json:"EndpointId"`
SwarmID string `json:"SwarmId"`
EntryPoint string `json:"EntryPoint"`
Env []interface{} `json:"Env"`
ResourceControl struct {
ID int `json:"Id"`
ResourceID string `json:"ResourceId"`
SubResourceIds []interface{} `json:"SubResourceIds"`
Type int `json:"Type"`
UserAccesses []interface{} `json:"UserAccesses"`
TeamAccesses []interface{} `json:"TeamAccesses"`
Public bool `json:"Public"`
AdministratorsOnly bool `json:"AdministratorsOnly"`
System bool `json:"System"`
} `json:"ResourceControl"`
Status int `json:"Status"`
ProjectPath string `json:"ProjectPath"`
CreationDate int `json:"CreationDate"`
CreatedBy string `json:"CreatedBy"`
UpdateDate int `json:"UpdateDate"`
UpdatedBy string `json:"UpdatedBy"`
AdditionalFiles interface{} `json:"AdditionalFiles"`
AutoUpdate interface{} `json:"AutoUpdate"`
GitConfig interface{} `json:"GitConfig"`
FromAppTemplate bool `json:"FromAppTemplate"`
Namespace string `json:"Namespace"`
IsComposeFormat bool `json:"IsComposeFormat"`
}

@ -7,21 +7,12 @@ import (
"log" "log"
"os" "os"
"os/user" "os/user"
"strings"
"github.com/chatton/portainer/client" "github.com/chatton/portainer/client"
) )
const ( func loadCreds() client.Credentials {
applicationJson = "application/json"
baseUrl = "http://qnap:9000"
)
type PortainerCredentials struct {
Username string `json:"username"`
Password string `json:"password"`
}
func loadCreds() PortainerCredentials {
usr, _ := user.Current() usr, _ := user.Current()
credPath := fmt.Sprintf("%s/.homelab/portainer-creds.json", usr.HomeDir) credPath := fmt.Sprintf("%s/.homelab/portainer-creds.json", usr.HomeDir)
@ -34,7 +25,7 @@ func loadCreds() PortainerCredentials {
log.Fatal(err) log.Fatal(err)
} }
creds := PortainerCredentials{} creds := client.Credentials{}
if err := json.Unmarshal(fileBytes, &creds); err != nil { if err := json.Unmarshal(fileBytes, &creds); err != nil {
log.Fatal(err) log.Fatal(err)
} }
@ -42,10 +33,25 @@ func loadCreds() PortainerCredentials {
} }
func main() { func main() {
c := client.NewPortainerClient(baseUrl) args := os.Args
if len(args) != 2 {
fmt.Println("must specify name of stack to start!")
os.Exit(1)
}
stackName := args[1]
creds := loadCreds() creds := loadCreds()
err := c.Login(creds.Username, creds.Password) c, err := client.NewPortainerClient(creds)
if err != nil { if err != nil {
log.Fatalln(err) log.Fatal(err)
}
s, err := c.GetStackByName(stackName)
if err != nil {
log.Fatal(err)
}
err = c.StartStack(s.ID)
if err != nil && strings.Contains(err.Error(), "is already running") {
log.Fatal(err)
} }
} }

Loading…
Cancel
Save