77 lines
1.5 KiB
Go
77 lines
1.5 KiB
Go
|
package gateeditor
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"encoding/json"
|
||
|
|
"fmt"
|
||
|
|
"net/http"
|
||
|
|
)
|
||
|
|
|
||
|
|
func NewClient(url string) *Client {
|
||
|
|
return &Client{
|
||
|
|
url: url,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
type Client struct {
|
||
|
|
url string
|
||
|
|
}
|
||
|
|
|
||
|
|
type Route struct {
|
||
|
|
Host []string `json:"host"`
|
||
|
|
Backend string `json:"backend"`
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c *Client) listRoutes() ([]Route, error) {
|
||
|
|
body := bytes.NewBufferString(`{"action": "ListRoutes"}`)
|
||
|
|
resp, err := http.Post(c.url, "application/json", body)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
defer resp.Body.Close()
|
||
|
|
|
||
|
|
var response struct {
|
||
|
|
Routes []Route `json:"routes"`
|
||
|
|
Error string `json:"error"`
|
||
|
|
}
|
||
|
|
if err := json.NewDecoder(resp.Body).Decode(&response); err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
if response.Error != "" {
|
||
|
|
return nil, fmt.Errorf("error listing routes: %s", response.Error)
|
||
|
|
}
|
||
|
|
return response.Routes, nil
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c *Client) updateRoutes(routes []Route) error {
|
||
|
|
payload := struct {
|
||
|
|
Action string `json:"action"`
|
||
|
|
Routes []Route `json:"routes"`
|
||
|
|
}{
|
||
|
|
Action: "UpdateRoutes",
|
||
|
|
Routes: routes,
|
||
|
|
}
|
||
|
|
payloadBytes, err := json.Marshal(payload)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
resp, err := http.Post(c.url, "application/json", bytes.NewBuffer(payloadBytes))
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
defer resp.Body.Close()
|
||
|
|
|
||
|
|
var response struct {
|
||
|
|
Message string `json:"message"`
|
||
|
|
Error string `json:"error"`
|
||
|
|
}
|
||
|
|
if err := json.NewDecoder(resp.Body).Decode(&response); err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
if response.Error != "" {
|
||
|
|
return fmt.Errorf("error updating routes: %s", response.Error)
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
}
|