aboutsummaryrefslogtreecommitdiff
path: root/mastodon
diff options
context:
space:
mode:
authorr <r@freesoftwareextremist.com>2020-01-31 18:23:18 +0000
committerr <r@freesoftwareextremist.com>2020-01-31 18:23:18 +0000
commite73df8de9ab5a00aa377d5cd7cfe335e06774ea0 (patch)
tree12655bfc159a40ecf85f6dc7e1d78cc64086d965 /mastodon
parentcd9306294d18d0b357fae3c1894b4f3f2d14c9c0 (diff)
downloadbloat-e73df8de9ab5a00aa377d5cd7cfe335e06774ea0.tar.gz
bloat-e73df8de9ab5a00aa377d5cd7cfe335e06774ea0.zip
Remove websocket api
Diffstat (limited to 'mastodon')
-rw-r--r--mastodon/go.mod6
-rw-r--r--mastodon/go.sum4
-rw-r--r--mastodon/streaming_ws.go195
3 files changed, 1 insertions, 204 deletions
diff --git a/mastodon/go.mod b/mastodon/go.mod
index 505f461..cc414ea 100644
--- a/mastodon/go.mod
+++ b/mastodon/go.mod
@@ -2,8 +2,4 @@ module mastodon
go 1.13
-require (
- github.com/gorilla/mux v1.7.3 // indirect
- github.com/gorilla/websocket v1.4.1
- github.com/tomnomnom/linkheader v0.0.0-20180905144013-02ca5825eb80
-)
+require github.com/tomnomnom/linkheader v0.0.0-20180905144013-02ca5825eb80
diff --git a/mastodon/go.sum b/mastodon/go.sum
index 6440913..d67f56e 100644
--- a/mastodon/go.sum
+++ b/mastodon/go.sum
@@ -1,6 +1,2 @@
-github.com/gorilla/mux v1.7.3 h1:gnP5JzjVOuiZD07fKKToCAOjS0yOpj/qPETTXCCS6hw=
-github.com/gorilla/mux v1.7.3/go.mod h1:1lud6UwP+6orDFRuTfBEV8e9/aOM/c4fVVCaMa2zaAs=
-github.com/gorilla/websocket v1.4.1 h1:q7AeDBpnBk8AogcD4DSag/Ukw/KV+YhzLj2bP5HvKCM=
-github.com/gorilla/websocket v1.4.1/go.mod h1:YR8l580nyteQvAITg2hZ9XVh4b55+EU/adAjf1fMHhE=
github.com/tomnomnom/linkheader v0.0.0-20180905144013-02ca5825eb80 h1:nrZ3ySNYwJbSpD6ce9duiP+QkD3JuLCcWkdaehUS/3Y=
github.com/tomnomnom/linkheader v0.0.0-20180905144013-02ca5825eb80/go.mod h1:iFyPdL66DjUD96XmzVL3ZntbzcflLnznH0fr99w5VqE=
diff --git a/mastodon/streaming_ws.go b/mastodon/streaming_ws.go
deleted file mode 100644
index 838f65b..0000000
--- a/mastodon/streaming_ws.go
+++ /dev/null
@@ -1,195 +0,0 @@
-package mastodon
-
-import (
- "context"
- "encoding/json"
- "fmt"
- "net/url"
- "path"
- "strings"
-
- "github.com/gorilla/websocket"
-)
-
-// WSClient is a WebSocket client.
-type WSClient struct {
- websocket.Dialer
- client *Client
-}
-
-// NewWSClient return WebSocket client.
-func (c *Client) NewWSClient() *WSClient { return &WSClient{client: c} }
-
-// Stream is a struct of data that flows in streaming.
-type Stream struct {
- Event string `json:"event"`
- Payload interface{} `json:"payload"`
-}
-
-// StreamingWSUser return channel to read events on home using WebSocket.
-func (c *WSClient) StreamingWSUser(ctx context.Context) (chan Event, error) {
- return c.streamingWS(ctx, "user", "")
-}
-
-// StreamingWSPublic return channel to read events on public using WebSocket.
-func (c *WSClient) StreamingWSPublic(ctx context.Context, isLocal bool) (chan Event, error) {
- s := "public"
- if isLocal {
- s += ":local"
- }
-
- return c.streamingWS(ctx, s, "")
-}
-
-// StreamingWSHashtag return channel to read events on tagged timeline using WebSocket.
-func (c *WSClient) StreamingWSHashtag(ctx context.Context, tag string, isLocal bool) (chan Event, error) {
- s := "hashtag"
- if isLocal {
- s += ":local"
- }
-
- return c.streamingWS(ctx, s, tag)
-}
-
-// StreamingWSList return channel to read events on a list using WebSocket.
-func (c *WSClient) StreamingWSList(ctx context.Context, id string) (chan Event, error) {
- return c.streamingWS(ctx, "list", string(id))
-}
-
-func (c *WSClient) streamingWS(ctx context.Context, stream, tag string) (chan Event, error) {
- params := url.Values{}
- params.Set("access_token", c.client.config.AccessToken)
- params.Set("stream", stream)
- if tag != "" {
- params.Set("tag", tag)
- }
-
- u, err := changeWebSocketScheme(c.client.config.Server)
- if err != nil {
- return nil, err
- }
- u.Path = path.Join(u.Path, "/api/v1/streaming")
- u.RawQuery = params.Encode()
-
- q := make(chan Event)
- go func() {
- defer close(q)
- for {
- err := c.handleWS(ctx, u.String(), q)
- if err != nil {
- return
- }
- }
- }()
-
- return q, nil
-}
-
-func (c *WSClient) handleWS(ctx context.Context, rawurl string, q chan Event) error {
- conn, err := c.dialRedirect(rawurl)
- if err != nil {
- q <- &ErrorEvent{err: err}
-
- // End.
- return err
- }
-
- // Close the WebSocket when the context is canceled.
- go func() {
- <-ctx.Done()
- conn.Close()
- }()
-
- for {
- select {
- case <-ctx.Done():
- q <- &ErrorEvent{err: ctx.Err()}
-
- // End.
- return ctx.Err()
- default:
- }
-
- var s Stream
- err := conn.ReadJSON(&s)
- if err != nil {
- q <- &ErrorEvent{err: err}
-
- // Reconnect.
- break
- }
-
- err = nil
- switch s.Event {
- case "update":
- var status Status
- err = json.Unmarshal([]byte(s.Payload.(string)), &status)
- if err == nil {
- q <- &UpdateEvent{Status: &status}
- }
- case "notification":
- var notification Notification
- err = json.Unmarshal([]byte(s.Payload.(string)), &notification)
- if err == nil {
- q <- &NotificationEvent{Notification: &notification}
- }
- case "delete":
- if f, ok := s.Payload.(float64); ok {
- q <- &DeleteEvent{ID: fmt.Sprint(int64(f))}
- } else {
- q <- &DeleteEvent{ID: strings.TrimSpace(s.Payload.(string))}
- }
- }
- if err != nil {
- q <- &ErrorEvent{err}
- }
- }
-
- return nil
-}
-
-func (c *WSClient) dialRedirect(rawurl string) (conn *websocket.Conn, err error) {
- for {
- conn, rawurl, err = c.dial(rawurl)
- if err != nil {
- return nil, err
- } else if conn != nil {
- return conn, nil
- }
- }
-}
-
-func (c *WSClient) dial(rawurl string) (*websocket.Conn, string, error) {
- conn, resp, err := c.Dial(rawurl, nil)
- if err != nil && err != websocket.ErrBadHandshake {
- return nil, "", err
- }
- defer resp.Body.Close()
-
- if loc := resp.Header.Get("Location"); loc != "" {
- u, err := changeWebSocketScheme(loc)
- if err != nil {
- return nil, "", err
- }
-
- return nil, u.String(), nil
- }
-
- return conn, "", err
-}
-
-func changeWebSocketScheme(rawurl string) (*url.URL, error) {
- u, err := url.Parse(rawurl)
- if err != nil {
- return nil, err
- }
-
- switch u.Scheme {
- case "http":
- u.Scheme = "ws"
- case "https":
- u.Scheme = "wss"
- }
-
- return u, nil
-}