package main

import (
	"context"
	"os"
	"os/signal"
	"os/user"
	"syscall"
	"time"

	"github.com/google/uuid"
	"github.com/spf13/cobra"
	"google.golang.org/grpc"
)

var (
	cliCmd = &cobra.Command{
		Use:   "",
		Short: "IMCTL is the CLI app to interact with an INetMock server",
		PersistentPreRunE: func(cmd *cobra.Command, args []string) error {
			return initGRPCConnection()
		},
	}

	inetMockSocketPath string
	outputFormat       string
	grpcTimeout        time.Duration
	appCtx             context.Context
	appCancel          context.CancelFunc
	conn               *grpc.ClientConn
)

func init() {
	cliCmd.PersistentFlags().StringVar(&inetMockSocketPath, "socket-path", "unix:///var/run/inetmock.sock", "Path to the INetMock socket file")
	cliCmd.PersistentFlags().StringVarP(&outputFormat, "format", "f", "table", "Output format to use. Possible values: table, json, yaml")
	cliCmd.PersistentFlags().DurationVar(&grpcTimeout, "grpc-timeout", 5*time.Second, "Timeout to connect to the gRPC API")

	cliCmd.AddCommand(endpointsCmd, handlerCmd, healthCmd, auditCmd)
	endpointsCmd.AddCommand(getEndpoints)
	handlerCmd.AddCommand(getHandlersCmd)
	healthCmd.AddCommand(generalHealthCmd, containerHealthCmd)

	currentUser := ""
	if usr, err := user.Current(); err == nil {
		currentUser = usr.Username
	} else {
		currentUser = uuid.New().String()
	}

	watchEventsCmd.PersistentFlags().StringVar(
		&listenerName,
		"listener-name",
		currentUser,
		"set listener name - defaults to the current username, if the user cannot be determined a random UUID will be used",
	)
	auditCmd.AddCommand(watchEventsCmd, addFileCmd, removeFileCmd)
}

func initGRPCConnection() (err error) {
	appCtx, appCancel = context.WithCancel(context.Background())

	signals := make(chan os.Signal, 1)
	signal.Notify(signals, syscall.SIGINT, syscall.SIGTERM, syscall.SIGQUIT)

	go func() {
		<-signals
		appCancel()
	}()

	dialCtx, cancel := context.WithTimeout(appCtx, grpcTimeout)
	conn, err = grpc.DialContext(dialCtx, inetMockSocketPath, grpc.WithInsecure())
	cancel()

	return
}