mirror of https://github.com/kha7iq/pingme
commit
f0c2c37837
@ -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
|
||||||
|
}
|
Loading…
Reference in new issue