push.go

package command

import (
	"fmt"
	"net/url"
	"os"

	"github.com/gotify/cli/v2/config"
	"github.com/gotify/cli/v2/utils"
	"github.com/gotify/go-api-client/v2/auth"
	"github.com/gotify/go-api-client/v2/client/message"
	"github.com/gotify/go-api-client/v2/gotify"
	"github.com/gotify/go-api-client/v2/models"
	"gopkg.in/urfave/cli.v1"
)

func Push() cli.Command {
	return cli.Command{
		Name:        "push",
		Aliases:     []string{"p"},
		Usage:       "Pushes a message",
		ArgsUsage:   "<message-text>",
		Description: "the message can also provided in stdin f.ex:\n   echo my text | gotify push",
		Flags: []cli.Flag{
			cli.IntFlag{Name: "priority,p", Usage: "Set the priority"},
			cli.StringFlag{Name: "title,t", Usage: "Set the title (empty for app name)"},
			cli.StringFlag{Name: "token", Usage: "Override the app token", EnvVar: "GOTIFY_TOKEN"},
			cli.StringFlag{Name: "url", Usage: "Override the Gotify URL"},
			cli.BoolFlag{Name: "quiet,q", Usage: "Do not output anything (on success)"},
			cli.StringFlag{Name: "contentType", Usage: "The content type of the message. See https://gotify.net/docs/msgextras#client-display"},
			cli.StringFlag{Name: "clickUrl", Usage: "An URL to open upon clicking the notification. See https://gotify.net/docs/msgextras#client-notification"},
			cli.BoolFlag{Name: "disable-unescape-backslash", Usage: "Disable evaluating \\n and \\t (if set, \\n and \\t will be seen as a string)"},
			cli.BoolFlag{Name: "no-split", Usage: "Do not split the message on null character when reading from stdin"},
		},
		Action: doPush,
	}
}

func doPush(ctx *cli.Context) {
	conf, confErr := config.ReadConfig(config.GetLocations())

	msgText := make(chan string)
	null := '\x00'
	if ctx.Bool("no-split") {
		go readMessage(ctx.Args(), os.Stdin, msgText, nil)
	} else {
		go readMessage(ctx.Args(), os.Stdin, msgText, &null)
	}

	priority := ctx.Int("priority")
	title := ctx.String("title")
	token := ctx.String("token")
	quiet := ctx.Bool("quiet")
	contentType := ctx.String("contentType")
	clickUrl := ctx.String("clickUrl")

	if token == "" {
		if confErr != nil {
			utils.Exit1With("token is not configured, run 'gotify init'")
			return
		}
		token = conf.Token
	}

	stringURL := ctx.String("url")
	if stringURL == "" {
		if confErr != nil {
			utils.Exit1With("url is not configured, run 'gotify init'")
			return
		}
		stringURL = conf.URL
	}

	if !ctx.IsSet("priority") && conf != nil {
		priority = conf.DefaultPriority
	}

	parsedURL, err := url.Parse(stringURL)
	if err != nil {
		utils.Exit1With("invalid url", stringURL)
		return
	}

	var sent bool
	for msgText := range msgText {
		if !ctx.Bool("disable-unescape-backslash") {
			msgText = utils.Evaluate(msgText)
		}

		msg := models.MessageExternal{
			Message:  msgText,
			Title:    title,
			Priority: priority,
		}

		msg.Extras = map[string]interface{}{}

		if contentType != "" {
			msg.Extras["client::display"] = map[string]interface{}{
				"contentType": contentType,
			}
		}

		if clickUrl != "" {
			msg.Extras["client::notification"] = map[string]interface{}{
				"click": map[string]string{
					"url": clickUrl,
				},
			}
		}

		pushMessage(parsedURL, token, msg, quiet)

		sent = true
	}
	if !sent {
		utils.Exit1With("no message sent! a message must be set, either as argument or via stdin")
	}
}

func pushMessage(parsedURL *url.URL, token string, msg models.MessageExternal, quiet bool) {

	client := gotify.NewClient(parsedURL, utils.CreateHTTPClient())

	params := message.NewCreateMessageParams()
	params.Body = &msg
	_, err := client.Message.CreateMessage(params, auth.TokenAuth(token))
	if err == nil {
		if !quiet {
			fmt.Println("message created")
		}
	} else {
		utils.Exit1With(err)
	}
}