keys/admin/main.go

package main

import (
	"context"
	"errors"
	"fmt"
	"io"
	"log"
	"net/http"
	"os"

	"github.com/spf13/viper"
	"github.com/urfave/cli/v3"
)

var (
	server string
	token  string
)

func load() error {
	viper.SetConfigName("config")
	viper.SetConfigType("toml")
	viper.AddConfigPath("$HOME/.local/share/keys")
	err := viper.ReadInConfig()
	if err != nil {
		return err
	}

	server = viper.GetString("server")
	if server == "" {
		return errors.New("'server' property is required")
	}

	token = viper.GetString("token")
	if token == "" {
		return errors.New("'token' property is required")
	}

	return nil
}

func main() {
	cmd := &cli.Command{
		Name:  "keys-admin",
		Usage: "keys project server administration tool",
		Commands: []*cli.Command{
			{
				Name:  "version",
				Usage: "print version and exit",
				Action: func(context.Context, *cli.Command) error {
					fmt.Println("version 1.0")
					return nil
				},
			},
			{
				Name:  "clients",
				Usage: "fetch all clients",
				Flags: []cli.Flag{
					&cli.StringFlag{
						Name:  "id",
						Usage: "optional id of a client to fetch",
					},
				},
				Action: clients,
			},
			{
				Name:  "logs",
				Usage: "fetch all/specific logs",
				Flags: []cli.Flag{
					&cli.StringFlag{
						Name:  "id",
						Usage: "optional id of a client to log",
					},
					&cli.StringFlag{
						Name:  "date",
						Usage: "optional date of the log, id required, in ISO8601 date format",
					},
					&cli.StringFlag{
						Name:  "skip",
						Usage: "optional skip bytes in log file",
					},
					&cli.StringFlag{
						Name:  "take",
						Usage: "optional take bytes from log file",
					},
				},
				Action: logs,
			},
		},
	}

	err := cmd.Run(context.Background(), os.Args)
	if err != nil {
		log.Fatal(err)
	}
}

func clients(ctx context.Context, cmd *cli.Command) error {
	err := load()
	if err != nil {
		return err
	}

	var url string
	id := cmd.String("id")
	if id != "" {
		url = fmt.Sprintf("%s/admin/clients/%s", server, id)
	} else {
		url = fmt.Sprintf("%s/admin/clients", server)
	}

	req, err := http.NewRequest(http.MethodGet, url, nil)
	if err != nil {
		return err
	}

	req.Header.Set("Authorization", token)

	res, err := http.DefaultClient.Do(req)
	if err != nil {
		return err
	}

	body, err := io.ReadAll(res.Body)
	if err != nil {
		return err
	}

	fmt.Printf("%s", body)
	return nil
}

func logs(ctx context.Context, cmd *cli.Command) error {
	err := load()
	if err != nil {
		return err
	}

	var url string
	id := cmd.String("id")
	if id != "" {
		params := map[string]string{
			"date": cmd.String("date"),
			"skip": cmd.String("skip"),
			"take": cmd.String("take"),
		}
		paramsStr := ""
		for k, v := range params {
			if v == "" {
				continue
			}

			if paramsStr == "" {
				paramsStr = fmt.Sprintf("?%s=%s", k, v)
			} else {
				paramsStr += fmt.Sprintf("&%s=%s", k, v)
			}
		}

		url = fmt.Sprintf("%s/admin/%s/logs%s", server, id, paramsStr)
	} else {
		url = fmt.Sprintf("%s/admin/logs", server)
	}

	req, err := http.NewRequest(http.MethodGet, url, nil)
	if err != nil {
		return err
	}

	req.Header.Set("Authorization", token)

	res, err := http.DefaultClient.Do(req)
	if err != nil {
		return err
	}

	body, err := io.ReadAll(res.Body)
	if err != nil {
		return err
	}

	fmt.Printf("%s", body)
	return nil
}