package main

import (
	"errors"
	"fmt"
	"maps"
	"os"
	"slices"
	"strings"

	"google.golang.org/grpc"
)

type command func(grpc.ClientConnInterface) error

var allCommands = map[string]command{
	"getAllAssets": getAllAssets,
	"transact":     transact,
	"listen":       listen,
}

func main() {
	commands := os.Args[1:]
	if len(commands) == 0 {
		printUsage()
		panic(errors.New("missing command"))
	}

	for _, name := range commands {
		if _, exists := allCommands[name]; !exists {
			printUsage()
			panic(fmt.Errorf("unknown command: %s", name))
		}
		fmt.Println("command:", name)
	}

	client := newGrpcConnection()
	defer client.Close()

	for _, name := range commands {
		command := allCommands[name]

		if err := command(client); err != nil {
			if errors.Is(err, errExpected) {
				fmt.Println(err)
				return
			}

			panic(err)
		}
	}
}

func printUsage() {
	fmt.Println("Arguments: <command1> [<command2> ...]")
	fmt.Println("Available commands:", availableCommands())
}

func availableCommands() string {
	commandNames := slices.Collect(maps.Keys(allCommands))
	return strings.Join(commandNames, ", ")
}