package db

import (
	"log"
	"strings"
	"time"

	"github.com/zu1k/nali/pkg/cdn"
	"github.com/zu1k/nali/pkg/download"
	"github.com/zu1k/nali/pkg/ip2region"
	"github.com/zu1k/nali/pkg/qqwry"
	"github.com/zu1k/nali/pkg/zxipv6wry"
)

func UpdateDB(dbNames ...string) {
	if len(dbNames) == 0 {
		dbNames = DbNameListForUpdate
	}

	done := make(map[string]struct{})
	for _, dbName := range dbNames {
		update, name := getUpdateFuncByName(dbName)
		if _, found := done[name]; !found {
			done[name] = struct{}{}
			if err := update(); err != nil {
				continue
			}
		}
	}
}

var DbNameListForUpdate = []string{
	"qqwry",
	"zxipv6wry",
	"ip2region",
	"cdn",
}

func getUpdateFuncByName(name string) (func() error, string) {
	name = strings.TrimSpace(name)
	if db := getDbByName(name); db != nil {
		// direct download if download-url not null
		if len(db.DownloadUrls) > 0 {
			return func() error {
				log.Printf("正在下载最新 %s 数据库...\n", db.Name)
				_, err := download.Download(db.File, db.DownloadUrls...)
				if err != nil {
					log.Printf("%s 数据库下载失败: %s\n", db.Name, db.File)
					log.Println("error:", err)
					return err
				} else {
					log.Printf("%s 数据库下载成功: %s\n", db.Name, db.File)
					return nil
				}
			}, string(db.Format)
		}

		// intenel download func
		switch db.Format {
		case FormatQQWry:
			return func() error {
				log.Println("正在下载最新 纯真 IPv4数据库...")
				_, err := qqwry.Download(getDbByName("qqwry").File)
				if err != nil {
					log.Println("数据库 QQWry 下载失败:", err)
				}
				return err
			}, FormatQQWry
		case FormatZXIPv6Wry:
			return func() error {
				log.Println("正在下载最新 ZX IPv6数据库...")
				_, err := zxipv6wry.Download(getDbByName("zxipv6wry").File)
				if err != nil {
					log.Println("数据库 ZXIPv6Wry 下载失败:", err)
				}
				return err
			}, FormatZXIPv6Wry
		case FormatIP2Region:
			return func() error {
				log.Println("正在下载最新 Ip2Region 数据库...")
				_, err := ip2region.Download(getDbByName("ip2region").File)
				if err != nil {
					log.Println("数据库 Ip2Region 下载失败:", err)
				}
				return err
			}, FormatZXIPv6Wry
		case FormatCDNYml:
			return func() error {
				log.Println("正在下载最新 CDN服务提供商数据库...")
				_, err := cdn.Download(getDbByName("cdn").File)
				if err != nil {
					log.Println("数据库 CDN 下载失败:", err)
				}
				return err
			}, FormatZXIPv6Wry
		default:
			return func() error {
				log.Println("暂不支持该类型数据库的自动更新")
				log.Println("可通过指定数据库的 download-urls 从特定链接下载数据库文件")
				return nil
			}, time.Now().String()
		}
	} else {
		return func() error {
			log.Fatalln("该名称的数据库未找到：", name)
			return nil
		}, time.Now().String()
	}
}