upda/terminal/app.go

package terminal

import (
	"encoding/json"
	"errors"
	"fmt"
	"git.myservermanager.com/varakh/upda/api"
	"git.myservermanager.com/varakh/upda/server"
	"git.myservermanager.com/varakh/upda/util"
	"github.com/go-resty/resty/v2"
	"github.com/urfave/cli/v2"
	"log"
	"os"
	"text/tabwriter"
)

const (
	name    = "upda-cli"
	desc    = "a commandline helper for upda"
	version = server.Version

	envServerUrl     = "UPDA_SERVER_URL"
	envAdminUser     = "UPDA_ADMIN_USER"
	envAdminPassword = "UPDA_ADMIN_PASSWORD"
	envWebhookId     = "UPDA_WEBHOOK_ID"
	envWebhookToken  = "UPDA_WEBHOOK_TOKEN"

	flagServerUrl      = "server-url"
	flagAdminUser      = "admin-user"
	flagAdminPass      = "admin-pass"
	flagWebhookId      = "webhook-id"
	flagWebhookToken   = "webhook-token"
	flagUpdatePageSize = "page-size"

	flagRaw = "raw"

	webhooksUrlPath = "/api/v1/webhooks"
	updatesUrlPath  = "/api/v1/updates"
)

func Start() {
	var raw bool
	var serverUrl string
	var adminUser string
	var adminPassword string
	var webhookId string
	var webhookToken string
	var updatePageSize int64

	rawFlag := &cli.BoolFlag{
		Name:        flagRaw,
		Usage:       "on success raw JSON data from response is returned",
		Aliases:     []string{"r"},
		Value:       false,
		Destination: &raw,
	}

	serverUrlFlag := &cli.StringFlag{
		Name:        flagServerUrl,
		Usage:       "the server url (FQDN without context path)",
		Required:    true,
		Aliases:     []string{"s"},
		EnvVars:     []string{envServerUrl},
		Destination: &serverUrl,
	}
	adminUserFlag := &cli.StringFlag{
		Name:        flagAdminUser,
		Usage:       "admin user",
		Required:    true,
		Aliases:     []string{"u"},
		EnvVars:     []string{envAdminUser},
		Destination: &adminUser,
	}
	adminPasswordFlag := &cli.StringFlag{
		Name:        flagAdminPass,
		Usage:       "admin password",
		Required:    true,
		Aliases:     []string{"p"},
		EnvVars:     []string{envAdminPassword},
		Destination: &adminPassword,
	}
	webhookIdFlag := &cli.StringFlag{
		Name:        flagWebhookId,
		Usage:       "webhook id",
		Required:    true,
		Aliases:     []string{"i"},
		EnvVars:     []string{envWebhookId},
		Destination: &webhookId,
	}
	webhookTokenFlag := &cli.StringFlag{
		Name:        flagWebhookToken,
		Usage:       "webhook token",
		Required:    true,
		Aliases:     []string{"t"},
		EnvVars:     []string{envWebhookToken},
		Destination: &webhookToken,
	}
	updatePageSizeFlag := &cli.Int64Flag{
		Name:        flagUpdatePageSize,
		Usage:       "update show page size",
		Value:       10000,
		Required:    false,
		Aliases:     []string{"ps"},
		Destination: &updatePageSize,
	}

	cli.VersionFlag = &cli.BoolFlag{
		Name:    "version",
		Aliases: []string{"v"},
		Usage:   "show version",
	}
	app := &cli.App{
		Name:                 name,
		Usage:                desc,
		Version:              version,
		EnableBashCompletion: true,
		Commands: []*cli.Command{
			{
				Name:    "webhook",
				Aliases: []string{"w"},
				Usage:   "Options for webhook",
				Subcommands: []*cli.Command{
					{
						Name:    "create",
						Usage:   "Creates a webhook",
						Aliases: []string{"c"},
						Flags: []cli.Flag{
							serverUrlFlag,
							adminUserFlag,
							adminPasswordFlag,
							rawFlag,
						},
						ArgsUsage: "<label> [<type (generic|diun, default: generic)>] [<ignore-host (true|false, default: false)>]",
						Action:    webhookCreate,
					},
					{
						Name:    "send",
						Usage:   "Sends data to a webhook",
						Aliases: []string{"s"},
						Flags: []cli.Flag{
							serverUrlFlag,
							webhookIdFlag,
							webhookTokenFlag,
						},
						ArgsUsage: "<json payload>",
						Action:    webhookSend,
					},
				},
			},
			{
				Name:    "update",
				Aliases: []string{"u"},
				Usage:   "Options for update",
				Subcommands: []*cli.Command{
					{
						Name:    "show",
						Usage:   "Shows updates",
						Aliases: []string{"s"},
						Flags: []cli.Flag{
							serverUrlFlag,
							adminUserFlag,
							adminPasswordFlag,
							updatePageSizeFlag,
							rawFlag,
						},
						Action: updateShow,
					},
				},
			},
		},
	}

	if err := app.Run(os.Args); err != nil {
		log.Fatal(err)
	}
}

func webhookCreate(cCtx *cli.Context) error {
	if err := failIfFlagsNotPresent(cCtx, []string{flagServerUrl, flagAdminUser, flagAdminPass}); err != nil {
		return cli.Exit(err, 1)
	}
	if !cCtx.Args().Present() {
		return cli.Exit(errors.New("args required - try 'webhook create help'"), 1)
	}
	// validate label
	label := cCtx.Args().First()
	if label == "" || len(label) > 255 {
		return cli.Exit(errors.New("label cannot be blank or only be 255 characters long"), 1)
	}

	// validate type
	t := cCtx.Args().Get(1)
	validTypes := []string{api.WebhookTypeGeneric.Value(), api.WebhookTypeDiun.Value()}
	if t == "" {
		t = api.WebhookTypeGeneric.Value()
	}

	if !util.FindInSlice(validTypes, t) {
		return cli.Exit(errors.New(fmt.Sprintf("type must be one of %v", validTypes)), 1)
	}

	ignoreHost := cCtx.Args().Get(2) == "true"

	// fully constructed payload
	payload := api.CreateWebhookRequest{
		Label:      label,
		Type:       t,
		IgnoreHost: ignoreHost,
	}

	var successRes api.WebhookSingleResponse
	var errorRes api.ErrorResponse
	url := cCtx.String(flagServerUrl) + webhooksUrlPath
	client := resty.New()
	client.SetDisableWarn(true)
	res, err := client.R().
		SetBasicAuth(cCtx.String(flagAdminUser), cCtx.String(flagAdminPass)).
		SetHeader("Content-Type", "application/json").
		SetBody(&payload).
		SetResult(&successRes).
		SetError(&errorRes).
		Post(url)

	if err != nil {
		return cli.Exit(fmt.Errorf("error during webhook creation: %w", err), 1)
	}
	if !res.IsSuccess() {
		return cli.Exit(fmt.Sprintf("error during webhook creation: (%d) %+v", res.StatusCode(), errorRes), 1)
	}

	if cCtx.Bool(flagRaw) {
		fmt.Println(string(res.Body()))
		return nil
	}

	fmt.Printf("ID\t%v\n", successRes.Data.ID)
	fmt.Printf("URL\t%v\n", fmt.Sprintf("%s/%s", url, successRes.Data.ID))
	fmt.Printf("Token\t%v\n", successRes.Data.Token)
	fmt.Printf("Type\t%v\n", successRes.Data.Type)

	return nil
}

func webhookSend(cCtx *cli.Context) error {
	if err := failIfFlagsNotPresent(cCtx, []string{flagServerUrl, flagWebhookId, flagWebhookToken}); err != nil {
		return cli.Exit(err, 1)
	}

	if !cCtx.Args().Present() || cCtx.Args().Len() < 1 {
		return cli.Exit(errors.New("args required - try 'webhook send help'"), 1)
	}
	// validate payload is valid json
	payloadArg := cCtx.Args().First()
	if payloadArg == "" {
		return cli.Exit(errors.New("payload cannot be blank"), 1)
	}
	if !json.Valid([]byte(payloadArg)) {
		return cli.Exit(errors.New("payload is not valid JSON"), 1)
	}

	var errorRes api.ErrorResponse
	url := cCtx.String(flagServerUrl) + webhooksUrlPath + "/" + cCtx.String(flagWebhookId)
	client := resty.New()
	client.SetDisableWarn(true)
	res, err := client.R().
		SetHeader("Content-Type", "application/json").
		SetHeader(server.HeaderWebhookToken, cCtx.String(flagWebhookToken)).
		SetBody(payloadArg).
		SetError(&errorRes).
		Post(url)

	if err != nil {
		return cli.Exit(fmt.Errorf("error during webhook invocation: %w", err), 1)
	}
	if !res.IsSuccess() {
		return cli.Exit(fmt.Sprintf("error during webhook invocation: (%d) %+v", res.StatusCode(), errorRes), 1)
	}

	return nil
}

func updateShow(cCtx *cli.Context) error {
	if err := failIfFlagsNotPresent(cCtx, []string{flagServerUrl, flagAdminUser, flagAdminPass}); err != nil {
		return cli.Exit(err, 1)
	}

	var successRes api.UpdateDataPageResponse
	var errorRes api.ErrorResponse
	url := cCtx.String(flagServerUrl) + updatesUrlPath + fmt.Sprintf("?pageSize=%d", cCtx.Int64(flagUpdatePageSize))
	client := resty.New()
	client.SetDisableWarn(true)
	res, err := client.R().
		SetBasicAuth(cCtx.String(flagAdminUser), cCtx.String(flagAdminPass)).
		SetHeader("Content-Type", "application/json").
		SetResult(&successRes).
		SetError(&errorRes).
		Get(url)

	if err != nil {
		return cli.Exit(fmt.Errorf("error during showing updates: %w", err), 1)
	}
	if !res.IsSuccess() {
		return cli.Exit(fmt.Sprintf("error during showing updates: (%d) %+v", res.StatusCode(), errorRes), 1)
	}

	if cCtx.Bool(flagRaw) {
		fmt.Println(string(res.Body()))
		return nil
	}

	w := tabwriter.NewWriter(os.Stdout, 10, 1, 1, ' ', tabwriter.Debug)

	for _, u := range successRes.Data.Content {
		if _, err = fmt.Fprintf(w, "%v\t %v\t %v\t %v\t %v\n", u.Application, u.Host, u.Provider, u.Version, u.State); err != nil {
			return cli.Exit(fmt.Sprintf("error during showing updates: %+v", errorRes), 1)
		}
	}
	if err = w.Flush(); err != nil {
		return cli.Exit(fmt.Sprintf("error during showing updates: %+v", errorRes), 1)
	}

	return nil
}

func failIfFlagsNotPresent(cCtx *cli.Context, flagKeys []string) error {
	if flagKeys == nil {
		return errors.New("flagKeys cannot be null")
	}

	for _, key := range flagKeys {
		if cCtx.String(key) == "" {
			return errors.New(fmt.Sprintf("'%v' is required but blank", key))
		}
	}

	return nil
}