go-mastodon/cmd/mstdn/main.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())
}