cloud66.go

package cloud66

import (
	"bytes"
	"context"
	"encoding/json"
	"errors"
	"fmt"
	"io"
	"log"
	"net/http"
	"net/http/httputil"
	"os"
	"path/filepath"
	"reflect"
	"runtime"
	"strconv"
	"strings"
	"time"

	"github.com/khash/oauth/oauth"
	"github.com/pborman/uuid"
)

const (
	baseURL = "https://app.cloud66.com"
)

type GenericResponse struct {
	Status  bool   `json:"ok"`
	Message string `json:"message"`
}

type ClientConfig struct {
	DefaultUserAgent string
	AgentPrefix      string
	BaseAPIURL       string
	ClientID         string
	ClientSecret     string
	RedirectURL      string
	Scope            string

	defaultAPIURL string
	authURL       string
	tokenURL      string
}

type Client struct {
	HTTP              *http.Client
	URL               string
	UserAgent         string
	Hostname          string
	AccountId         *int
	Debug             bool
	AdditionalHeaders http.Header
	Config            *ClientConfig
}

type Response struct {
	Response   json.RawMessage
	Count      int
	Pagination json.RawMessage
}

type Pagination struct {
	Previous int
	Next     int
	Current  int
}

type stackEnvironmentFilterFunction func(item interface{}, environment *string) bool

func (c *Client) Get(v interface{}, path string, queryStrings map[string]string, p *Pagination) error {
	return c.APIReq(v, "GET", path, nil, queryStrings, p)
}

func (c *Client) Patch(v interface{}, path string, body interface{}) error {
	return c.APIReq(v, "PATCH", path, body, nil, nil)
}

func (c *Client) Post(v interface{}, path string, body interface{}) error {
	return c.APIReq(v, "POST", path, body, nil, nil)
}

func (c *Client) Put(v interface{}, path string, body interface{}) error {
	return c.APIReq(v, "PUT", path, body, nil, nil)
}

func (c *Client) Delete(path string) error {
	return c.APIReq(nil, "DELETE", path, nil, nil, nil)
}

func (c *Client) NewRequest(method, path string, body interface{}, queryStrings map[string]string) (*http.Request, error) {
	var ctype string
	var rbody io.Reader

	switch t := body.(type) {
	case nil:
	case string:
		rbody = bytes.NewBufferString(t)
	case io.Reader:
		rbody = t
	default:
		v := reflect.ValueOf(body)
		if !v.IsValid() {
			break
		}
		if v.Type().Kind() == reflect.Ptr {
			v = reflect.Indirect(v)
			if !v.IsValid() {
				break
			}
		}

		j, err := json.Marshal(body)
		if err != nil {
			log.Fatal(err)
		}
		rbody = bytes.NewReader(j)
		ctype = "application/json"
	}
	apiURL := strings.TrimRight(c.URL, "/")
	if apiURL == "" {
		apiURL = c.Config.defaultAPIURL
	}

	var qs string
	if (queryStrings != nil) && (len(queryStrings) > 0) {
		for key, value := range queryStrings {
			if qs == "" {
				qs = "?"
			} else {
				qs = qs + "&"
			}
			qs = qs + key + "=" + value
		}
	}

	last_url := strings.TrimRight(apiURL+path, "/")
	if qs != "" {
		last_url = last_url + qs
	}

	req, err := http.NewRequest(method, last_url, rbody)
	if err != nil {
		return nil, err
	}

	req.Header.Set("Accept", "application/json")
	req.Header.Set("Request-Id", uuid.New())
	if os.Getenv("CXTOKEN") != "" {
		req.Header.Set("X-CxToken", os.Getenv("CXTOKEN"))
	}
	if c.AccountId != nil {
		req.Header.Set("X-Account", strconv.Itoa(*c.AccountId))
	}
	useragent := c.UserAgent
	if useragent == "" {
		useragent = c.Config.DefaultUserAgent
	}
	req.Header.Set("User-Agent", useragent)
	if ctype != "" {
		req.Header.Set("Content-Type", ctype)
	}
	for k, v := range c.AdditionalHeaders {
		req.Header[k] = v
	}
	return req, nil
}

func (c *Client) APIReq(v interface{}, meth, path string, body interface{}, queryStrings map[string]string, p *Pagination) error {
	req, err := c.NewRequest(meth, path, body, queryStrings)
	if err != nil {
		return err
	}
	return c.DoReq(req, v, p)
}

func (c *Client) DoReq(req *http.Request, v interface{}, p *Pagination) error {

	if c.Debug {
		dump, err := httputil.DumpRequestOut(req, true)
		if err != nil {
			log.Println(err)
		} else {
			os.Stderr.Write(dump)
			os.Stderr.Write([]byte{'\n', '\n'})
		}
	}

	var check_pagination bool
	if (req.Method == "GET") && (p != nil) {
		check_pagination = true
	} else {
		check_pagination = false
	}

	httpClient := c.HTTP
	if httpClient == nil {
		httpClient = http.DefaultClient
	}

	res, err := httpClient.Do(req)
	if err != nil {
		return err
	}
	defer res.Body.Close()
	if c.Debug {
		dump, err := httputil.DumpResponse(res, true)
		if err != nil {
			log.Println(err)
		} else {
			os.Stderr.Write(dump)
			os.Stderr.Write([]byte{'\n'})
		}
	}
	if err = checkResp(res); err != nil {
		return err
	}

	// open the wrapper
	var r Response
	err = json.NewDecoder(res.Body).Decode(&r)
	if err != nil {
		return err
	}

	buffer := bytes.NewBuffer(r.Response)

	switch t := v.(type) {
	case nil:
	case io.Writer:
		_, err = io.Copy(t, buffer)
	default:
		err = json.NewDecoder(buffer).Decode(v)
	}

	if (err == nil) && check_pagination {
		pagination := bytes.NewBuffer(r.Pagination)
		err = json.NewDecoder(pagination).Decode(p)
	}

	return err
}

type Error struct {
	error
	Id string
}

type errorResp struct {
	Error       string `json:"error"`
	Description string `json:"error_description"`
	Details     string `json:"details"`
}

func checkResp(res *http.Response) error {
	if res.StatusCode/100 != 2 { // 200, 201, 202, etc
		var e errorResp
		err := json.NewDecoder(res.Body).Decode(&e)
		if err != nil {
			return errors.New("Unexpected error: " + res.Status)
		}
		if e.Details != "" {
			return Error{error: errors.New(e.Details), Id: e.Error}
		} else {
			return Error{error: errors.New(e.Description), Id: e.Error}
		}

	}
	if msg := res.Header.Get("X-Cloud66-Warning"); msg != "" {
		fmt.Fprintln(os.Stderr, strings.TrimSpace(msg))
	}
	return nil
}

func (c *Client) GetAuthorizeURL() string {
	config := &oauth.Config{
		ClientId:     c.Config.ClientID,
		ClientSecret: c.Config.ClientSecret,
		RedirectURL:  c.Config.RedirectURL,
		Scope:        c.Config.Scope,
		AuthURL:      c.Config.authURL,
		TokenURL:     c.Config.tokenURL,
	}

	return config.AuthCodeURL("")
}

func (c *Client) Authorize(tokenDir, tokenFile, token string) {
	err := os.MkdirAll(tokenDir, 0777)
	if err != nil {
		fmt.Printf("Failed to create directory for the token at %s\n", tokenDir)
	}
	cachefile := filepath.Join(tokenDir, tokenFile)

	config := &oauth.Config{
		ClientId:     c.Config.ClientID,
		ClientSecret: c.Config.ClientSecret,
		RedirectURL:  c.Config.RedirectURL,
		Scope:        c.Config.Scope,
		AuthURL:      c.Config.authURL,
		TokenURL:     c.Config.tokenURL,
		TokenCache:   oauth.CacheFile(cachefile),
	}
	transport := &oauth.Transport{Config: config}
	_, err = config.TokenCache.Token()

	// do we already have access?
	if err != nil {
		_, err := transport.Exchange(token)
		if err != nil {
			log.Fatal("Exchange:", err)
		}

		log.Printf("token is cached in %v\n", config.TokenCache)
		os.Exit(1)
	}
}

func GetClient(tokenFile, tokenDir, version string, config *ClientConfig) Client {
	c := Client{
		Config: config,
	}

	cachefile := filepath.Join(tokenDir, tokenFile)
	config.DefaultUserAgent = config.AgentPrefix + "/" + version + " (" + runtime.GOOS + "; " + runtime.GOARCH + ")"

	oauthConfig := &oauth.Config{
		ClientId:     config.ClientID,
		ClientSecret: config.ClientSecret,
		RedirectURL:  config.RedirectURL,
		Scope:        config.Scope,
		AuthURL:      config.authURL,
		TokenURL:     config.tokenURL,
		TokenCache:   oauth.CacheFile(cachefile),
	}

	hostname, err := os.Hostname()
	if err != nil {
		log.Printf("unable to get the hostname\n", err.Error())
	}
	transport := &oauth.Transport{Config: oauthConfig}
	token, _ := oauthConfig.TokenCache.Token()
	transport.Token = token
	c.HTTP = transport.Client()
	c.Hostname = hostname

	return c
}

func NewClientConfig(baseAPIURL string) *ClientConfig {
	return &ClientConfig{
		defaultAPIURL: baseAPIURL + "/api/3",
		authURL:       baseAPIURL + "/oauth/authorize",
		tokenURL:      baseAPIURL + "/oauth/token",
	}
}

func FetchTokenFromCallback(timeout time.Duration) (string, error) {
	var token string

	m := http.NewServeMux()
	srv := &http.Server{Addr: "127.0.0.1:34543", Handler: m}
	codeCh := make(chan string)

	m.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
		fmt.Fprintf(w, "Authorized. You can close this window now!")
		codeCh <- r.URL.Query().Get("code")
	})

	go func() {
		if err := srv.ListenAndServe(); err != http.ErrServerClosed {
			log.Fatalf("ListenAndServe(): %s", err)
		}
	}()

	select {
	case code := <-codeCh:
		srv.Shutdown(context.Background())
		token = code
	}

	return token, nil
}