package ftp

import (
	"log"
	"net"
	"os/user"
	"path/filepath"
	"strings"

	"github.com/spf13/cobra"
	"goftp.io/server/v2"
)

var ports int
var authuser, authpwd string
var path, ip, publicIp string
var username, groupname, passiveports string
var cert, key string
var forceTls bool

type Auth struct {
}

func (a Auth) CheckPasswd(ctx *server.Context, user string, name string) (bool, error) {
	if authuser == "" && authpwd == "" {
		return true, nil
	}
	if authuser == user && authpwd == name {
		return true, nil
	}
	return false, nil
}

// ftpCmd represents the ftp command
var Cmd = &cobra.Command{
	Use:   "ftp",
	Short: `FTP文件服务器`,
	Long:  `FTP文件服务器`,
	Run: func(cmd *cobra.Command, args []string) {
		if username == "" || groupname == "" {
			u, err := user.Current()
			if err == nil {
				username = u.Username
				groupname = u.Username
			} else {
				username = "root"
				groupname = "root"
			}
		}
		if publicIp == "" {
			if tmp, err := net.Dial("udp", "139.199.163.65:443"); err == nil {
				publicIp = strings.Split(tmp.LocalAddr().String(), ":")[0]
				tmp.Close()
			}
		}
		path, _ = filepath.Abs(path)
		driver, err := NewDriver(path)
		if err != nil {
			log.Fatal("Error starting server:", err)
		}
		opts := &server.Options{
			Auth:           Auth{},
			Driver:         driver,
			Perm:           server.NewSimplePerm(username, groupname),
			Name:           "B612 FTP Server",
			Hostname:       ip,
			PublicIP:       publicIp,
			PassivePorts:   passiveports,
			Port:           ports,
			WelcomeMessage: "B612 FTP Server",
			Logger:         nil,
			RateLimit:      0,
			TLS:            key != "" && cert != "",
			ForceTLS:       forceTls,
			CertFile:       cert,
			KeyFile:        key,
		}
		log.Printf("Starting ftp server on %v:%v", opts.Hostname, opts.Port)
		log.Printf("AuthUser %v, Password %v", authuser, authpwd)
		log.Printf("Public IP %v", publicIp)
		log.Printf("Path %v", path)
		log.Printf("Passive Ports %v", passiveports)
		log.Printf("TLS %v", opts.TLS)
		log.Printf("Username %v, Groupname %v", username, groupname)
		server, err := server.NewServer(opts)
		if err != nil {
			log.Fatal("Error starting server:", err)
		}
		err = server.ListenAndServe()
		if err != nil {
			log.Fatal("Error starting server:", err)
		}
	},
}

func init() {
	Cmd.Flags().StringVarP(&publicIp, "public-ip", "I", "", "公网IP")
	Cmd.Flags().StringVarP(&username, "username", "U", "", "FTP用户名")
	Cmd.Flags().StringVarP(&groupname, "groupname", "G", "", "FTP组名")
	Cmd.Flags().StringVarP(&passiveports, "passiveports", "P", "50000-60000", "被动模式端口范围，格式：50000-60000")
	Cmd.Flags().IntVarP(&ports, "port", "p", 21, "监听端口")
	Cmd.Flags().StringVarP(&ip, "ip", "i", "0.0.0.0", "监听地址")
	Cmd.Flags().StringVarP(&authuser, "auth-user", "u", "", "用户名，默认为任意")
	Cmd.Flags().StringVarP(&authpwd, "auth-passwd", "k", "", "密码，默认为任意")
	Cmd.Flags().StringVarP(&path, "folder", "f", "./", "本地文件地址")
	Cmd.Flags().BoolVarP(&forceTls, "force-tls", "t", false, "强制使用TLS加密传输")
	Cmd.Flags().StringVarP(&cert, "cert", "C", "", "TLS证书文件")
	Cmd.Flags().StringVarP(&key, "key", "K", "", "TLS密钥文件")
}