package mastodon import ( "bytes" "encoding/json" "fmt" "log" "net/http" "time" "github.com/urfave/cli/v2" ) // Mastodon struct holds data parsed via flags for the service type Mastodon struct { Title string Token string ServerURL string Message string } // Send parse values from *cli.context and return *cli.Command // and sets a status message for mastodon. func Send() *cli.Command { var mastodonOpts Mastodon return &cli.Command{ Name: "mastodon", Usage: "Set status message for mastodon", UsageText: "pingme mastodon --token '123' --url 'mastodon.social' --title 'PingMe' " + "--msg 'some message'", Description: `Mastodon uses application token to authorize and sets a status message`, Flags: []cli.Flag{ &cli.StringFlag{ Destination: &mastodonOpts.Token, Name: "token", Aliases: []string{"t"}, Required: true, Usage: "Application token for authorization.", EnvVars: []string{"MASTODON_TOKEN"}, }, &cli.StringFlag{ Destination: &mastodonOpts.Message, Name: "msg", Aliases: []string{"m"}, Usage: "Message content.", EnvVars: []string{"MASTODON_MESSAGE"}, }, &cli.StringFlag{ Destination: &mastodonOpts.Title, Name: "title", Usage: "Title of the message.", EnvVars: []string{"MASTODON_TITLE"}, }, &cli.StringFlag{ Destination: &mastodonOpts.ServerURL, Name: "url", Aliases: []string{"u"}, Value: "mastodon.social", Required: true, Usage: "URL of mastodon server i.e mastodon.social", EnvVars: []string{"MASTODON_SERVER"}, }, }, Action: func(ctx *cli.Context) error { endPointURL := "https://" + mastodonOpts.ServerURL + "/api/v1/statuses/" // Create a Bearer string by appending string access token bearer := "Bearer " + mastodonOpts.Token fullMessage := mastodonOpts.Title + "\n" + mastodonOpts.Message if err := sendMastodon(endPointURL, bearer, fullMessage); err != nil { return fmt.Errorf("failed to send message\n[ERROR] - %v", err) } return nil }, } } // sendMastodon function take the server url , authorization token // and message string to set the status. func sendMastodon(url string, token string, msg string) error { reqBody, err := json.Marshal(map[string]string{ "status": msg, }) if err != nil { return err } // Create a new request using http req, err := http.NewRequest("POST", url, bytes.NewBuffer(reqBody)) if err != nil { return err } // add authorization header to the request req.Header.Set("Authorization", token) req.Header.Set("Content-Type", "application/json; charset=UTF-8") // create a new http client and send request to server c := &http.Client{Timeout: 10 * time.Second} resp, err := c.Do(req) if err != nil { return err } defer resp.Body.Close() // decode response received from server var data map[string]interface{} err = json.NewDecoder(resp.Body).Decode(&data) if err != nil { return err } // check if server returned an error checkErr, ok := data["error"] if ok { return fmt.Errorf("%v", checkErr) } log.Printf("Success!!\nVisibility: %v\nURL: %v\n", data["visibility"], data["url"]) return nil }