105 lines
2.3 KiB
Go
105 lines
2.3 KiB
Go
package vanity
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
"net/http"
|
|
"strings"
|
|
)
|
|
|
|
type Info struct {
|
|
Version string `json:"version"`
|
|
NumPackages int `json:"num_packages"`
|
|
Packages []Package `json:"packages"`
|
|
}
|
|
|
|
type Package struct {
|
|
Name string `json:"name"`
|
|
Description string `json:"description"`
|
|
Branch string `json:"branch"`
|
|
WebURL string `json:"web_url"`
|
|
CloneHTTP string `json:"clone_http"`
|
|
CloneSSH string `json:"clone_ssh"`
|
|
}
|
|
|
|
func (p Package) Module(domain string) string {
|
|
return fmt.Sprintf("%s/%s", strings.TrimSuffix(domain, "/"), strings.ToLower(p.Name))
|
|
}
|
|
|
|
// Info gets Info from a vanity server
|
|
func (c *Client) Info(ctx context.Context) (Info, error) {
|
|
var info Info
|
|
resp, err := c.crud(ctx, Package{}, http.MethodOptions)
|
|
if err != nil {
|
|
return info, err
|
|
}
|
|
defer resp.Body.Close()
|
|
|
|
if resp.StatusCode != http.StatusOK {
|
|
return info, fmt.Errorf("could not get info: %s", resp.Status)
|
|
}
|
|
|
|
if err := json.NewDecoder(resp.Body).Decode(&info); err != nil {
|
|
return info, err
|
|
}
|
|
|
|
return info, nil
|
|
}
|
|
|
|
// Add adds a new Package to a vanity server
|
|
func (c *Client) Add(ctx context.Context, pkg Package) error {
|
|
resp, err := c.crud(ctx, pkg, http.MethodPost)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if resp.StatusCode != http.StatusCreated {
|
|
return fmt.Errorf("could not add package: %s", resp.Status)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// Update updates a Package on a vanity server
|
|
func (c *Client) Update(ctx context.Context, pkg Package) error {
|
|
resp, err := c.crud(ctx, pkg, http.MethodPatch)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if resp.StatusCode != http.StatusOK {
|
|
return fmt.Errorf("could not update package: %s", resp.Status)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// Remove removes a Package from a vanity server
|
|
func (c *Client) Remove(ctx context.Context, pkg Package) error {
|
|
resp, err := c.crud(ctx, pkg, http.MethodDelete)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if resp.StatusCode != http.StatusOK {
|
|
return fmt.Errorf("could not remove package: %s", resp.Status)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (c *Client) crud(ctx context.Context, pkg Package, method string) (*http.Response, error) {
|
|
payload, err := json.Marshal(pkg)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
req, err := c.newRequest(ctx, method, c.server, bytes.NewReader(payload))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return c.http.Do(req)
|
|
}
|