go-mastodon/cmd/mstdn/cmd_stream.go

65 lines
1.4 KiB
Go
Raw Normal View History

2017-04-15 14:52:11 +02:00
package main
import (
"context"
2017-04-17 16:29:44 +02:00
"encoding/json"
2017-04-15 14:52:11 +02:00
"fmt"
"os"
"os/signal"
"github.com/fatih/color"
"github.com/mattn/go-mastodon"
"github.com/urfave/cli"
)
2017-04-17 16:43:11 +02:00
type SimpleJSON struct {
Username string `json:"username"`
Avatar string `json:"avatar"`
Content string `json:"content"`
}
2017-04-15 14:52:11 +02:00
func cmdStream(c *cli.Context) error {
2017-04-17 16:29:44 +02:00
asJSON := c.Bool("json")
2017-04-17 16:43:11 +02:00
asSimpleJSON := c.Bool("simplejson")
2017-04-15 14:52:11 +02:00
client := c.App.Metadata["client"].(*mastodon.Client)
ctx, cancel := context.WithCancel(context.Background())
2017-04-16 16:38:53 +02:00
defer cancel()
2017-04-15 14:52:11 +02:00
sc := make(chan os.Signal, 1)
signal.Notify(sc, os.Interrupt)
q, err := client.StreamingPublic(ctx)
if err != nil {
return err
}
go func() {
<-sc
cancel()
close(q)
}()
for e := range q {
2017-04-17 16:29:44 +02:00
if asJSON {
json.NewEncoder(c.App.Writer).Encode(e)
2017-04-17 16:43:11 +02:00
} else if asSimpleJSON {
if t, ok := e.(*mastodon.UpdateEvent); ok {
json.NewEncoder(c.App.Writer).Encode(&SimpleJSON{
Username: t.Status.Account.Username,
Avatar: t.Status.Account.AvatarStatic,
Content: textContent(t.Status.Content),
})
}
2017-04-17 16:29:44 +02:00
} else {
switch t := e.(type) {
case *mastodon.UpdateEvent:
color.Set(color.FgHiRed)
fmt.Fprintln(c.App.Writer, t.Status.Account.Username)
color.Set(color.Reset)
fmt.Fprintln(c.App.Writer, textContent(t.Status.Content))
case *mastodon.ErrorEvent:
color.Set(color.FgYellow)
fmt.Fprintln(c.App.Writer, t.Error())
color.Set(color.Reset)
}
2017-04-15 14:52:11 +02:00
}
}
return nil
}