sessiontoken.go

package signalfx

import (
	"bytes"
	"context"
	"encoding/json"
	"fmt"
	"io"
	"io/ioutil"
	"net/http"

	"github.com/signalfx/signalfx-go/sessiontoken"
)

// SessionTokenAPIURL is the base URL for interacting with org tokens.
const SessionTokenAPIURL = "/v2/session"

// CreateOrgToken creates a org token.
func (c *Client) CreateSessionToken(ctx context.Context, tokenRequest *sessiontoken.CreateTokenRequest) (*sessiontoken.Token, error) {
	payload, err := json.Marshal(tokenRequest)
	if err != nil {
		return nil, err
	}

	// we need to explicitly pass an empty token (which means it wont get set in the header)
	// the API accepts either no token or a valid token, but not an empty token.
	resp, err := c.doRequestWithToken(ctx, "POST", SessionTokenAPIURL, nil, bytes.NewReader(payload), "")
	if resp != nil {
		defer resp.Body.Close()
	}
	if err != nil {
		return nil, err
	}

	if resp.StatusCode != http.StatusOK {
		message, _ := ioutil.ReadAll(resp.Body)
		return nil, fmt.Errorf("Bad status %d: %s", resp.StatusCode, message)
	}

	sessionToken := &sessiontoken.Token{}

	err = json.NewDecoder(resp.Body).Decode(sessionToken)
	_, _ = io.Copy(ioutil.Discard, resp.Body)

	return sessionToken, err
}

// DeleteOrgToken deletes a token.
func (c *Client) DeleteSessionToken(ctx context.Context, token string) error {
	resp, err := c.doRequestWithToken(ctx, "DELETE", SessionTokenAPIURL, nil, nil, token)
	if resp != nil {
		defer resp.Body.Close()
	}
	if err != nil {
		return err
	}

	if resp.StatusCode != http.StatusNoContent {
		message, _ := ioutil.ReadAll(resp.Body)
		return fmt.Errorf("Unexpected status code: %d: %s", resp.StatusCode, message)
	}
	_, _ = io.Copy(ioutil.Discard, resp.Body)

	return nil
}