package lbryinc import ( "bytes" "encoding/json" "errors" "fmt" "io/ioutil" "net/http" "net/url" "time" log "github.com/sirupsen/logrus" ) const ( defaultServerAddress = "https://api.lbry.com" timeout = 5 * time.Second headerForwardedFor = "X-Forwarded-For" userObjectPath = "user" userMeMethod = "me" userHasVerifiedEmailMethod = "has_verified_email" ) // Client stores data about internal-apis call it is about to make. type Client struct { AuthToken string Logger *log.Logger serverAddress string extraHeaders map[string]string } // ClientOpts allow to provide extra parameters to NewClient: // - ServerAddress // - RemoteIP — to forward the IP of a frontend client making the request type ClientOpts struct { ServerAddress string RemoteIP string } // APIResponse reflects internal-apis JSON response format. type APIResponse struct { Success bool `json:"success"` Error *string `json:"error"` Data *ResponseData `json:"data"` } // ResponseData is a map containing parsed json response. type ResponseData map[string]interface{} func makeMethodPath(obj, method string) string { return fmt.Sprintf("/%s/%s", obj, method) } // NewClient returns a client instance for internal-apis. It requires authToken to be provided // for authentication. func NewClient(authToken string, opts *ClientOpts) Client { c := Client{ serverAddress: defaultServerAddress, extraHeaders: make(map[string]string), AuthToken: authToken, Logger: log.StandardLogger(), } if opts != nil { if opts.ServerAddress != "" { c.serverAddress = opts.ServerAddress } if opts.RemoteIP != "" { c.extraHeaders[headerForwardedFor] = opts.RemoteIP } } return c } func (c Client) getEndpointURL(object, method string) string { return fmt.Sprintf("%s%s", c.serverAddress, makeMethodPath(object, method)) } func (c Client) prepareParams(params map[string]interface{}) (string, error) { form := url.Values{} form.Add("auth_token", c.AuthToken) for k, v := range params { if k == "auth_token" { return "", errors.New("extra auth_token supplied in request params") } form.Add(k, fmt.Sprintf("%v", v)) } return form.Encode(), nil } func (c Client) doCall(url string, payload string) ([]byte, error) { var body []byte c.Logger.Debugf("sending payload: %s", payload) req, err := http.NewRequest(http.MethodPost, url, bytes.NewBuffer([]byte(payload))) if err != nil { return body, err } req.Header.Add("Accept", "application/json") req.Header.Add("Content-Type", "application/x-www-form-urlencoded") for k, v := range c.extraHeaders { req.Header.Set(k, v) } client := &http.Client{Timeout: timeout} r, err := client.Do(req) if err != nil { return body, err } defer r.Body.Close() return ioutil.ReadAll(r.Body) } // Call calls a remote internal-apis server, returning a response, // wrapped into standardized API Response struct. func (c Client) Call(object, method string, params map[string]interface{}) (ResponseData, error) { var rd ResponseData payload, err := c.prepareParams(params) if err != nil { return rd, err } body, err := c.doCall(c.getEndpointURL(object, method), payload) if err != nil { return rd, err } var ar APIResponse err = json.Unmarshal(body, &ar) if err != nil { return rd, err } if !ar.Success { return rd, errors.New(*ar.Error) } return *ar.Data, err } // UserMe returns user details for the user associated with the current auth_token func (c Client) UserMe() (ResponseData, error) { return c.Call(userObjectPath, userMeMethod, map[string]interface{}{}) } func (c Client) UserHasVerifiedEmail() (ResponseData, error) { return c.Call(userObjectPath, userHasVerifiedEmailMethod, map[string]interface{}{}) }