From cd8854d045857570b2769d6f17fd6e0b84961fce Mon Sep 17 00:00:00 2001
From: kha7iq <a.khaliq@outlook.my>
Date: Tue, 11 May 2021 23:03:56 +0800
Subject: [PATCH] feat(service): add support for new service mastodon

---
 service/mastodon/mastodon.go | 124 +++++++++++++++++++++++++++++++++++
 1 file changed, 124 insertions(+)
 create mode 100644 service/mastodon/mastodon.go

diff --git a/service/mastodon/mastodon.go b/service/mastodon/mastodon.go
new file mode 100644
index 0000000..abdd407
--- /dev/null
+++ b/service/mastodon/mastodon.go
@@ -0,0 +1,124 @@
+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
+}