cmd

var ListenFlag = &cli.StringFlag{
	Name:    "listen",
	Value:   ":8080",
	Usage:   "监听地址，格式为 ip:port，例如 :8080",
	Sources: cli.EnvVars("LISTEN"),
}

var LogFlag = &cli.StringFlag{
	Name:    "log",
	Value:   "/var/log/bepusdt/",
	Usage:   "日志文件保存路径",
	Sources: cli.EnvVars("LOG"),
}

var PostgresDSNFlag = &cli.StringFlag{
	Name:    "postgres",
	Value:   "",
	Usage:   "PostgreSQL 数据库连接字符串 (DSN)，例如: postgres://user:password@localhost:5432/bepusdt?sslmode=disable&connect_timeout=3",
	Sources: cli.EnvVars("POSTGRESQL_DSN"),
}

var Reset = &cli.Command{
	Name:  "reset",
	Usage: "忘记密码时，此命令可重置账号密码登录入口",
	Flags: []cli.Flag{SQLiteFlag, PostgresDSNFlag},
	Before: func(ctx context.Context, c *cli.Command) (context.Context, error) {
		sqlite := c.String("sqlite")
		postgres := c.String("postgres")
		if err := model.Init(sqlite, postgres); err != nil {
			return ctx, fmt.Errorf("数据库初始化失败 %w", err)
		}

		return ctx, task.Init()
	},
	After: func(ctx context.Context, c *cli.Command) error {
		model.Close()

		return nil
	},
	Action: func(ctx context.Context, cmd *cli.Command) error {
		hash := utils.Md5String(time.Now().String())

		username := hash[8:16]
		password := hash[0:8]
		entrance := fmt.Sprintf("/%s", hash[10:20])
		encrypt, _ := bcrypt.GenerateFromPassword([]byte(password), bcrypt.DefaultCost)

		model.SetK(model.AdminSecure, entrance)
		model.SetK(model.AdminUsername, username)
		model.SetK(model.AdminPassword, string(encrypt))

		fmt.Println("重置成功，对应信息如下：")
		fmt.Printf("管理员账号：%s\n管理员密码：%s\n后台管理入口：%s\n", username, password, entrance)
		fmt.Println("请妥善保存以上信息！")
		fmt.Println("-------------------------------")

		return nil
	},
}

var SQLiteFlag = &cli.StringFlag{
	Name:    "sqlite",
	Value:   "/var/lib/bepusdt/sqlite.db",
	Usage:   "SQLite 数据库文件路径",
	Sources: cli.EnvVars("SQLITE"),
}

var Start = &cli.Command{
	Name:  "start",
	Usage: "启动收款网关",
	Flags: []cli.Flag{SQLiteFlag, PostgresDSNFlag, LogFlag, ListenFlag},
	Before: func(ctx context.Context, c *cli.Command) (context.Context, error) {
		postgres := c.String("postgres")
		sqlite := c.String("sqlite")
		if err := model.Init(sqlite, postgres); err != nil {
			return ctx, fmt.Errorf("数据库初始化失败 %w", err)
		}

		if err := log.Init(c.String("log")); err != nil {
			return ctx, fmt.Errorf("日志初始化失败 %w", err)
		}

		return ctx, task.Init()
	},
	After: func(ctx context.Context, c *cli.Command) error {
		log.Close()
		model.Close()

		return nil
	},
	Action: start,
}

var Version = &cli.Command{
	Name:  "version",
	Usage: "显示版本信息",
	Action: func(ctx context.Context, cmd *cli.Command) error {
		fmt.Println("BEpusdt 版本：" + app.Version)

		return nil
	},
}