package main

import (
	"fmt"
	"log"
	"my-tools/counter"
	"my-tools/net"
)

const (
	mainMenu      = "请输入要执行的命令:\n1.清空DNS缓存&&重置winsock\n2.北京地铁票价计算器\n按回车退出\n====================================="
	busDaysPrompt = "请输入每月乘车天数(每天2次): "
	busFarePrompt = "请输入每次乘车票价: "
)

func main() {
	for {
		fmt.Println(mainMenu)
		input := getUserChoice()
		switch input {
		case 1:
			fmt.Println("================Start=====================")
			net.NewNetImpl().FlushDNSAndResetWinsock()
			fmt.Println("================End=====================")
		case 2:
			fmt.Println("北京地铁票价计算器")
			busDays, err := getUserInput(busDaysPrompt)
			if err != nil {
				log.Println(err)
				continue
			}
			busFare, err := getUserInput(busFarePrompt)
			if err != nil {
				log.Println(err)
				continue
			}
			fmt.Println("总票价为：", counter.NewBusImpl().CalculateBusFare(busDays, busFare))
		default:
			return
		}
		fmt.Println("按Enter继续")
		fmt.Scanln()
	}
}

func getUserChoice() int {
	var input int
	_, err := fmt.Scanln(&input)
	if err != nil {
		log.Println(err)
		return 0
	}
	return input
}

func getUserInput(prompt string) (int, error) {
	fmt.Print(prompt)
	var input int
	_, err := fmt.Scanln(&input)
	if err != nil {
		return 0, err
	}
	return input, nil
}