mirror of https://github.com/kha7iq/pingme
puneeth8994
4 years ago
committed by
puneeth.s
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