package main

import (
	"errors"
	"os"

	"github.com/rs/zerolog/log"
	"github.com/spf13/cobra"

	"github.com/httprunner/httprunner/v5/pkg/uixt"
)

var rootCmd = &cobra.Command{
	Use:     "wcl",
	Short:   "Monitor FIFA World Cup Live",
	Version: "2022.12.03.0018",
	RunE: func(cmd *cobra.Command, args []string) error {
		var device uixt.IDevice
		var bundleID string
		if iosApp != "" {
			log.Info().Str("bundleID", iosApp).Msg("init ios device")
			device = initIOSDevice(uuid)
			bundleID = iosApp
		} else if androidApp != "" {
			log.Info().Str("bundleID", androidApp).Msg("init android device")
			device = initAndroidDevice(uuid)
			bundleID = androidApp
		} else {
			return errors.New("android or ios app bundldID is required")
		}

		wc := NewWorldCupLive(device, matchName, bundleID, duration, interval)

		if auto {
			wc.EnterLive(bundleID)
		}

		wc.Start()
		return nil
	},
}

var (
	uuid       string
	iosApp     string
	androidApp string
	auto       bool
	duration   int
	interval   int
	logLevel   string
	matchName  string
	perf       []string
)

func main() {
	rootCmd.PersistentFlags().StringVarP(&logLevel, "log-level", "l", "INFO", "set log level")
	rootCmd.PersistentFlags().StringVarP(&uuid, "uuid", "u", "", "specify device serial or udid")
	rootCmd.PersistentFlags().StringVar(&iosApp, "ios", "", "run ios app")
	rootCmd.PersistentFlags().StringVar(&androidApp, "android", "", "run android app")
	rootCmd.PersistentFlags().BoolVar(&auto, "auto", false, "auto enter live")
	rootCmd.PersistentFlags().IntVarP(&duration, "duration", "d", 30, "set duration in seconds")
	rootCmd.PersistentFlags().IntVarP(&interval, "interval", "i", 15, "set interval in seconds")
	rootCmd.PersistentFlags().StringVarP(&matchName, "match-name", "n", "", "specify match name")
	rootCmd.PersistentFlags().StringSliceVarP(&perf, "perf", "p", nil,
		"specify performance monitor, e.g. sys_cpu,sys_mem,sys_net,sys_disk,fps,network,gpu")

	err := rootCmd.Execute()
	if err != nil {
		os.Exit(1)
	}
}