package main

import (
	"fmt"
	"go-netbus/config"
	"go-netbus/core"
	"log"
	"os"
)

func init() {
	log.SetFlags(log.Ldate | log.Ltime | log.LstdFlags)

	fmt.Println("+----------------------------------------------------------------+")
	fmt.Println("| Welcome to use NetBus, code by winshu                          |")
	fmt.Println("| If you have any problem when using the tool,                   |")
	fmt.Println("| Please submit issue : https://gitee.com/winshu/go-netbus       |")
	fmt.Println("+----------------------------------------------------------------+")
	fmt.Println()
}

func printHelp() {
	fmt.Println(`A: "-server" load "config.ini" and start as server`)
	fmt.Println(`   "-client" load "config.ini" and start as client`)
	fmt.Println(`B: "-server <key> <port>" start as server, and listening at port x', e.g. -server 6666`)
	fmt.Println(`   "-client <key> <server:port> <local:port:mapping> [tunnel-count]" start as client,`)
	fmt.Println(`   "e.g. -client winshu 123.54.23.67:6666 127.0.0.1:3306:13306`)
	fmt.Println(`Generate trial key: `)
	fmt.Println(`   "-generate <key> [expired-time]" make a trial client key, e.g. -generate winshu 2019-12-31`)
	fmt.Println(`more details please read "README.md"`)
}

func main() {
	args := os.Args
	argc := len(os.Args)

	if argc < 2 {
		printHelp()
		os.Exit(0)
	}

	// 获取其余参数
	argsConfig := args[2:]

	switch args[1] {
	case "-server":
		// 外网服务
		serverConfig := config.InitServerConfig(argsConfig)
		core.Server(serverConfig)
	case "-client":
		// 内网服务
		clientConfig := config.InitClientConfig(argsConfig)
		core.Client(clientConfig)
	case "-generate":
		// 生成短期 key
		var seed, expired string
		if len(argsConfig) > 0 {
			seed = argsConfig[0]
		}
		if len(argsConfig) > 1 {
			expired = argsConfig[1]
		}
		if len(argsConfig) > 0 {
			trialKey, _ := config.NewKey(seed, expired)
			fmt.Println("You got a new key ->    ", trialKey)
		}
	case "-check":
		if len(argsConfig) == 2 {
			fmt.Println(config.CheckKey(argsConfig[0], argsConfig[1]))
		}
	case "-version":
		fmt.Println("Version", core.Version)
	default:
		printHelp()
	}
}