148 lines
3.4 KiB
Go
148 lines
3.4 KiB
Go
package mastodon
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
"net/http"
|
|
"net/url"
|
|
"path"
|
|
"strings"
|
|
)
|
|
|
|
// Config is a setting for access mastodon APIs.
|
|
type Config struct {
|
|
Server string
|
|
ClientID string
|
|
ClientSecret string
|
|
AccessToken string
|
|
}
|
|
|
|
// Client is a API client for mastodon.
|
|
type Client struct {
|
|
http.Client
|
|
config *Config
|
|
}
|
|
|
|
func (c *Client) doAPI(ctx context.Context, method string, uri string, params url.Values, res interface{}) error {
|
|
u, err := url.Parse(c.config.Server)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
u.Path = path.Join(u.Path, uri)
|
|
|
|
req, err := http.NewRequest(method, u.String(), strings.NewReader(params.Encode()))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
req.WithContext(ctx)
|
|
req.Header.Set("Authorization", "Bearer "+c.config.AccessToken)
|
|
if params != nil {
|
|
req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
|
|
}
|
|
|
|
resp, err := c.Do(req)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer resp.Body.Close()
|
|
|
|
if resp.StatusCode != http.StatusOK {
|
|
return fmt.Errorf("bad request: %v", resp.Status)
|
|
} else if res == nil {
|
|
return nil
|
|
}
|
|
return json.NewDecoder(resp.Body).Decode(&res)
|
|
}
|
|
|
|
// NewClient return new mastodon API client.
|
|
func NewClient(config *Config) *Client {
|
|
return &Client{
|
|
Client: *http.DefaultClient,
|
|
config: config,
|
|
}
|
|
}
|
|
|
|
// Authenticate get access-token to the API.
|
|
func (c *Client) Authenticate(ctx context.Context, username, password string) error {
|
|
params := url.Values{}
|
|
params.Set("client_id", c.config.ClientID)
|
|
params.Set("client_secret", c.config.ClientSecret)
|
|
params.Set("grant_type", "password")
|
|
params.Set("username", username)
|
|
params.Set("password", password)
|
|
params.Set("scope", "read write follow")
|
|
|
|
u, err := url.Parse(c.config.Server)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
u.Path = path.Join(u.Path, "/oauth/token")
|
|
|
|
req, err := http.NewRequest(http.MethodPost, u.String(), strings.NewReader(params.Encode()))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
|
|
resp, err := c.Do(req)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer resp.Body.Close()
|
|
|
|
if resp.StatusCode != http.StatusOK {
|
|
return fmt.Errorf("bad authorization: %v", resp.Status)
|
|
}
|
|
|
|
res := struct {
|
|
AccessToken string `json:"access_token"`
|
|
}{}
|
|
err = json.NewDecoder(resp.Body).Decode(&res)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
c.config.AccessToken = res.AccessToken
|
|
return nil
|
|
}
|
|
|
|
// Toot is struct to post status.
|
|
type Toot struct {
|
|
Status string `json:"status"`
|
|
InReplyToID int64 `json:"in_reply_to_id"`
|
|
MediaIDs []int64 `json:"media_ids"`
|
|
Sensitive bool `json:"sensitive"`
|
|
SpoilerText string `json:"spoiler_text"`
|
|
Visibility string `json:"visibility"`
|
|
}
|
|
|
|
// Mention hold information for mention.
|
|
type Mention struct {
|
|
URL string `json:"url"`
|
|
Username string `json:"username"`
|
|
Acct string `json:"acct"`
|
|
ID int64 `json:"id"`
|
|
}
|
|
|
|
// Tag hold information for tag.
|
|
type Tag struct {
|
|
Name string `json:"name"`
|
|
URL string `json:"url"`
|
|
}
|
|
|
|
// Attachment hold information for attachment.
|
|
type Attachment struct {
|
|
ID int64 `json:"id"`
|
|
Type string `json:"type"`
|
|
URL string `json:"url"`
|
|
RemoteURL string `json:"remote_url"`
|
|
PreviewURL string `json:"preview_url"`
|
|
TextURL string `json:"text_url"`
|
|
}
|
|
|
|
// Results hold information for search result.
|
|
type Results struct {
|
|
Accounts []*Account `json:"accounts"`
|
|
Statuses []*Status `json:"statuses"`
|
|
Hashtags []string `json:"hashtags"`
|
|
}
|