370 lines
7.1 KiB
Go
370 lines
7.1 KiB
Go
package main
|
|
|
|
import (
|
|
"bufio"
|
|
"bytes"
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
"io"
|
|
"io/ioutil"
|
|
"net/url"
|
|
"os"
|
|
"path/filepath"
|
|
"runtime"
|
|
"strings"
|
|
|
|
"github.com/fatih/color"
|
|
"github.com/mattn/go-mastodon"
|
|
"github.com/mattn/go-tty"
|
|
"github.com/urfave/cli"
|
|
"golang.org/x/net/html"
|
|
)
|
|
|
|
func readFile(filename string) ([]byte, error) {
|
|
if filename == "-" {
|
|
return ioutil.ReadAll(os.Stdin)
|
|
}
|
|
return ioutil.ReadFile(filename)
|
|
}
|
|
|
|
func textContent(s string) string {
|
|
doc, err := html.Parse(strings.NewReader(s))
|
|
if err != nil {
|
|
return s
|
|
}
|
|
var buf bytes.Buffer
|
|
|
|
var extractText func(node *html.Node, w *bytes.Buffer)
|
|
extractText = func(node *html.Node, w *bytes.Buffer) {
|
|
if node.Type == html.TextNode {
|
|
data := strings.Trim(node.Data, "\r\n")
|
|
if data != "" {
|
|
w.WriteString(data)
|
|
}
|
|
}
|
|
for c := node.FirstChild; c != nil; c = c.NextSibling {
|
|
extractText(c, w)
|
|
}
|
|
if node.Type == html.ElementNode {
|
|
name := strings.ToLower(node.Data)
|
|
if name == "br" {
|
|
w.WriteString("\n")
|
|
}
|
|
}
|
|
}
|
|
extractText(doc, &buf)
|
|
return buf.String()
|
|
}
|
|
|
|
var (
|
|
readUsername = func() (string, error) {
|
|
b, _, err := bufio.NewReader(os.Stdin).ReadLine()
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
return string(b), nil
|
|
}
|
|
readPassword func() (string, error)
|
|
)
|
|
|
|
func prompt() (string, string, error) {
|
|
fmt.Print("E-Mail: ")
|
|
email, err := readUsername()
|
|
if err != nil {
|
|
return "", "", err
|
|
}
|
|
|
|
fmt.Print("Password: ")
|
|
var password string
|
|
if readPassword == nil {
|
|
var t *tty.TTY
|
|
t, err = tty.Open()
|
|
if err != nil {
|
|
return "", "", err
|
|
}
|
|
defer t.Close()
|
|
password, err = t.ReadPassword()
|
|
} else {
|
|
password, err = readPassword()
|
|
}
|
|
if err != nil {
|
|
return "", "", err
|
|
}
|
|
return email, password, nil
|
|
}
|
|
|
|
func configFile(c *cli.Context) (string, error) {
|
|
dir := os.Getenv("HOME")
|
|
if runtime.GOOS == "windows" {
|
|
dir = os.Getenv("APPDATA")
|
|
if dir == "" {
|
|
dir = filepath.Join(os.Getenv("USERPROFILE"), "Application Data", "mstdn")
|
|
}
|
|
dir = filepath.Join(dir, "mstdn")
|
|
} else {
|
|
dir = filepath.Join(dir, ".config", "mstdn")
|
|
}
|
|
if err := os.MkdirAll(dir, 0700); err != nil {
|
|
return "", err
|
|
}
|
|
var file string
|
|
profile := c.String("profile")
|
|
if profile != "" {
|
|
file = filepath.Join(dir, "settings-"+profile+".json")
|
|
} else {
|
|
file = filepath.Join(dir, "settings.json")
|
|
}
|
|
return file, nil
|
|
}
|
|
|
|
func getConfig(c *cli.Context) (string, *mastodon.Config, error) {
|
|
file, err := configFile(c)
|
|
if err != nil {
|
|
return "", nil, err
|
|
}
|
|
b, err := ioutil.ReadFile(file)
|
|
if err != nil && !os.IsNotExist(err) {
|
|
return "", nil, err
|
|
}
|
|
config := &mastodon.Config{
|
|
Server: "https://mstdn.jp",
|
|
ClientID: "1e463436008428a60ed14ff1f7bc0b4d923e14fc4a6827fa99560b0c0222612f",
|
|
ClientSecret: "72b63de5bc11111a5aa1a7b690672d78ad6a207ce32e16ea26115048ec5d234d",
|
|
}
|
|
if err == nil {
|
|
err = json.Unmarshal(b, &config)
|
|
if err != nil {
|
|
return "", nil, fmt.Errorf("could not unmarshal %v: %v", file, err)
|
|
}
|
|
}
|
|
return file, config, nil
|
|
}
|
|
|
|
func authenticate(client *mastodon.Client, config *mastodon.Config, file string) error {
|
|
email, password, err := prompt()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
err = client.Authenticate(context.Background(), email, password)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
b, err := json.MarshalIndent(config, "", " ")
|
|
if err != nil {
|
|
return fmt.Errorf("failed to store file: %v", err)
|
|
}
|
|
err = ioutil.WriteFile(file, b, 0700)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to store file: %v", err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func argstr(c *cli.Context) string {
|
|
a := []string{}
|
|
for i := 0; i < c.NArg(); i++ {
|
|
a = append(a, c.Args().Get(i))
|
|
}
|
|
return strings.Join(a, " ")
|
|
}
|
|
|
|
func fatalIf(err error) {
|
|
if err == nil {
|
|
return
|
|
}
|
|
fmt.Fprintf(os.Stderr, "%s: %v\n", os.Args[0], err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
func makeApp() *cli.App {
|
|
app := cli.NewApp()
|
|
app.Name = "mstdn"
|
|
app.Usage = "mastodon client"
|
|
app.Version = "0.0.1"
|
|
app.Flags = []cli.Flag{
|
|
cli.StringFlag{
|
|
Name: "profile",
|
|
Usage: "profile name",
|
|
Value: "",
|
|
},
|
|
}
|
|
app.Commands = []cli.Command{
|
|
{
|
|
Name: "toot",
|
|
Usage: "post toot",
|
|
Flags: []cli.Flag{
|
|
cli.StringFlag{
|
|
Name: "ff",
|
|
Usage: "post utf-8 string from a file(\"-\" means STDIN)",
|
|
Value: "",
|
|
},
|
|
cli.IntFlag{
|
|
Name: "i",
|
|
Usage: "in-reply-to",
|
|
Value: 0,
|
|
},
|
|
},
|
|
Action: cmdToot,
|
|
},
|
|
{
|
|
Name: "stream",
|
|
Usage: "stream statuses",
|
|
Flags: []cli.Flag{
|
|
cli.StringFlag{
|
|
Name: "type",
|
|
Usage: "stream type (public,public/local,user:NAME,hashtag:TAG)",
|
|
},
|
|
cli.BoolFlag{
|
|
Name: "json",
|
|
Usage: "output JSON",
|
|
},
|
|
cli.BoolFlag{
|
|
Name: "simplejson",
|
|
Usage: "output simple JSON",
|
|
},
|
|
cli.StringFlag{
|
|
Name: "template",
|
|
Usage: "output with tamplate format",
|
|
},
|
|
},
|
|
Action: cmdStream,
|
|
},
|
|
{
|
|
Name: "timeline",
|
|
Usage: "show timeline",
|
|
Action: cmdTimeline,
|
|
},
|
|
{
|
|
Name: "notification",
|
|
Usage: "show notification",
|
|
Action: cmdNotification,
|
|
},
|
|
{
|
|
Name: "instance",
|
|
Usage: "show instance information",
|
|
Action: cmdInstance,
|
|
},
|
|
{
|
|
Name: "account",
|
|
Usage: "show account information",
|
|
Action: cmdAccount,
|
|
},
|
|
{
|
|
Name: "search",
|
|
Usage: "search content",
|
|
Action: cmdSearch,
|
|
},
|
|
{
|
|
Name: "follow",
|
|
Usage: "follow account",
|
|
Action: cmdFollow,
|
|
},
|
|
{
|
|
Name: "followers",
|
|
Usage: "show followers",
|
|
Action: cmdFollowers,
|
|
},
|
|
{
|
|
Name: "upload",
|
|
Usage: "upload file",
|
|
Action: cmdUpload,
|
|
},
|
|
{
|
|
Name: "delete",
|
|
Usage: "delete status",
|
|
Action: cmdDelete,
|
|
},
|
|
{
|
|
Name: "init",
|
|
Usage: "initialize profile",
|
|
Action: func(c *cli.Context) error { return nil },
|
|
},
|
|
}
|
|
app.Setup()
|
|
return app
|
|
}
|
|
|
|
type screen struct {
|
|
host string
|
|
}
|
|
|
|
func newScreen(config *mastodon.Config) *screen {
|
|
var host string
|
|
u, err := url.Parse(config.Server)
|
|
if err == nil {
|
|
host = u.Host
|
|
}
|
|
return &screen{host}
|
|
}
|
|
|
|
func (s *screen) acct(a string) string {
|
|
if !strings.Contains(a, "@") {
|
|
a += "@" + s.host
|
|
}
|
|
return a
|
|
}
|
|
|
|
func (s *screen) displayError(w io.Writer, e error) {
|
|
color.Set(color.FgYellow)
|
|
fmt.Fprintln(w, e.Error())
|
|
color.Set(color.Reset)
|
|
}
|
|
|
|
func (s *screen) displayStatus(w io.Writer, t *mastodon.Status) {
|
|
if t == nil {
|
|
return
|
|
}
|
|
if t.Reblog != nil {
|
|
color.Set(color.FgHiRed)
|
|
fmt.Fprint(w, s.acct(t.Account.Acct))
|
|
color.Set(color.Reset)
|
|
fmt.Fprint(w, " reblogged ")
|
|
color.Set(color.FgHiBlue)
|
|
fmt.Fprintln(w, s.acct(t.Reblog.Account.Acct))
|
|
fmt.Fprintln(w, textContent(t.Reblog.Content))
|
|
color.Set(color.Reset)
|
|
} else {
|
|
color.Set(color.FgHiRed)
|
|
fmt.Fprintln(w, s.acct(t.Account.Acct))
|
|
color.Set(color.Reset)
|
|
fmt.Fprintln(w, textContent(t.Content))
|
|
}
|
|
}
|
|
|
|
func run() int {
|
|
app := makeApp()
|
|
|
|
app.Before = func(c *cli.Context) error {
|
|
if c.Args().Get(0) == "init" {
|
|
file, err := configFile(c)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
os.Remove(file)
|
|
}
|
|
|
|
file, config, err := getConfig(c)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
client := mastodon.NewClient(config)
|
|
app.Metadata = map[string]interface{}{
|
|
"client": client,
|
|
"config": config,
|
|
}
|
|
if config.AccessToken == "" {
|
|
return authenticate(client, config, file)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
fatalIf(app.Run(os.Args))
|
|
return 0
|
|
}
|
|
|
|
func main() {
|
|
os.Exit(run())
|
|
}
|