package main

import (
	"errors"
	"log"
	"time"

	"github.com/gin-gonic/gin"
	"github.com/glebarez/sqlite"
	"gorm.io/gorm"
	"gorm.io/gorm/logger"

	"github.com/wangjia/pay/config"
	"github.com/wangjia/pay/internal/channel"
	"github.com/wangjia/pay/internal/model"
	"github.com/wangjia/pay/internal/router"
)

func main() {
	config.Load()

	db := initDB()
	autoMigrate(db)
	seed(db)
	seedWechat(db)

	reg := channel.NewRegistry(db)

	gin.SetMode(config.C.Server.Mode)
	r := gin.New()
	r.Use(gin.Logger(), gin.Recovery())

	orderSvc := router.Setup(r, db, reg)

	if config.C.QuerySync.Enabled {
		orderSvc.StartQuerySync(
			time.Duration(config.C.QuerySync.IntervalSec)*time.Second,
			time.Duration(config.C.QuerySync.MaxAgeMin)*time.Minute,
		)
		log.Printf("查单兜底已启动：每 %ds 一次", config.C.QuerySync.IntervalSec)
	}

	// 业务回调重试兜底（webhook 丢失/业务方短暂不可用时补推）。
	orderSvc.StartBizNotifyRetry(60 * time.Second)

	addr := ":" + config.C.Server.Port
	log.Printf("支付服务启动 %s (mode=%s, base_url=%s)", addr, config.C.Server.Mode, config.C.Server.BaseURL)
	if err := r.Run(addr); err != nil {
		log.Fatalf("启动失败: %v", err)
	}
}

func initDB() *gorm.DB {
	logLevel := logger.Silent
	if config.C.Server.Mode == "debug" {
		logLevel = logger.Info
	}
	gormCfg := &gorm.Config{Logger: logger.Default.LogMode(logLevel), TranslateError: true}

	switch config.C.Database.Driver {
	case "sqlite", "":
		db, err := gorm.Open(sqlite.Open(config.C.Database.DSN), gormCfg)
		if err != nil {
			log.Fatalf("连接 sqlite 失败: %v", err)
		}
		return db
	case "mysql":
		log.Fatal("mysql 驱动尚未启用：go get gorm.io/driver/mysql 后在 initDB 接入")
		return nil
	default:
		log.Fatalf("未知 database.driver: %s", config.C.Database.Driver)
		return nil
	}
}

func autoMigrate(db *gorm.DB) {
	if err := db.AutoMigrate(
		&model.Merchant{},
		&model.Product{},
		&model.Order{},
		&model.NotifyLog{},
		&model.BizNotifyLog{},
	); err != nil {
		log.Fatalf("自动迁移失败: %v", err)
	}
	log.Println("AutoMigrate 完成")
}

// seed 据 config 的 alipay_sandbox upsert 一个支付宝商户，并为其补两个测试套餐。
func seed(db *gorm.DB) {
	sb := config.C.AlipaySandbox
	if !sb.Enabled {
		log.Println("[seed] alipay_sandbox.enabled=false，跳过沙箱商户初始化")
		return
	}
	var m model.Merchant
	err := db.Where("code = ?", sb.MerchantCode).First(&m).Error
	if errors.Is(err, gorm.ErrRecordNotFound) {
		m = model.Merchant{
			Code:    sb.MerchantCode,
			Name:    sb.MerchantName,
			Channel: "alipay",
		}
	} else if err != nil {
		log.Fatalf("[seed] 查询商户失败: %v", err)
	}
	m.Production = sb.Production
	m.Enabled = true
	m.AppID = sb.AppID
	m.AppPrivateKey = sb.AppPrivateKey
	m.AlipayPublicKey = sb.AlipayPublicKey
	if err := db.Save(&m).Error; err != nil {
		log.Fatalf("[seed] 保存商户失败: %v", err)
	}
	log.Printf("[seed] 支付宝沙箱商户就绪: code=%s app_id=%s", m.Code, m.AppID)

	var cnt int64
	db.Model(&model.Product{}).Where("merchant_id = ?", m.ID).Count(&cnt)
	if cnt == 0 {
		samples := []model.Product{
			{MerchantID: m.ID, Name: "测试套餐 A", Description: "沙箱联调用", Price: "0.01", Active: true, Sort: 1},
			{MerchantID: m.ID, Name: "测试套餐 B", Description: "沙箱联调用", Price: "0.02", Active: true, Sort: 2},
		}
		if err := db.Create(&samples).Error; err != nil {
			log.Fatalf("[seed] 创建测试套餐失败: %v", err)
		}
		log.Printf("[seed] 已为商户 %s 创建 %d 个测试套餐", m.Code, len(samples))
	}

	seedPlans(db, m.ID)
}

// seedPlans 幂等 upsert 业务套餐（带 biz_code，供 jiu 等业务系统按 biz_code 查 product）。
// 价格权威在此表；权益（时长/档位/设备/功能）由业务方按 biz_code 映射。
func seedPlans(db *gorm.DB, merchantID uint64) {
	// Name 会成为支付宝收银台的 subject（一行约 12-14 汉字可见）：
	// 带品牌前缀、短句式，避免截断（2026-07-04 用户反馈收银台信息太少/被截断）。
	plans := []model.Product{
		{Name: "岩美酒库·联调测试", Description: "1 分钱端到端联调用（标准权益）", Price: "0.01", BizCode: "test_liandiao", Sort: 0},
		{Name: "岩美酒库·首月特惠30天", Description: "标准权益 · 30 天 · 每门店限一次（限次由 jiu 侧校验）", Price: "1.00", BizCode: "promo_first_month", Sort: 1},
		{Name: "岩美酒库·标准版月付", Description: "单店/单仓库 · 2 客户端 · 千张图片分享", Price: "299.00", BizCode: "monthly_standard", Sort: 2},
		{Name: "岩美酒库·高级版月付", Description: "单店/多仓库 · 5 客户端 · 万张照片 · 免费 AI 周/月度商业分析", Price: "599.00", BizCode: "monthly_pro", Sort: 3},
		{Name: "岩美酒库·标准版年付", Description: "单店/单仓库 · 2 客户端 · 千张图片分享", Price: "2999.00", BizCode: "annual_standard", Sort: 4},
		{Name: "岩美酒库·高级版年付", Description: "单店/多仓库 · 5 客户端 · 万张照片 · 免费 AI 周/月度商业分析", Price: "5999.00", BizCode: "annual_pro", Sort: 5},
	}
	for _, p := range plans {
		var existing model.Product
		err := db.Where("biz_code = ?", p.BizCode).First(&existing).Error
		if errors.Is(err, gorm.ErrRecordNotFound) {
			p.MerchantID = merchantID
			p.Active = true
			if err := db.Create(&p).Error; err != nil {
				log.Fatalf("[seed] 创建套餐 %s 失败: %v", p.BizCode, err)
			}
			continue
		} else if err != nil {
			log.Fatalf("[seed] 查询套餐 %s 失败: %v", p.BizCode, err)
		}
		// 已存在：更新价格/文案/归属（价格权威可从此改）。
		db.Model(&existing).Updates(map[string]any{
			"merchant_id": merchantID, "name": p.Name, "description": p.Description,
			"price": p.Price, "active": true, "sort": p.Sort,
		})
	}
	log.Printf("[seed] 业务套餐就绪：%d 个（含联调 0.01）", len(plans))
}

// seedWechat 据 config 的 wechat upsert 一个微信商户（V3 Native），并为其补两个测试套餐。
// 微信无沙箱：wechat.enabled=true 即视为正式（Production=true），只能用真实商户号做 1 分钱实测。
func seedWechat(db *gorm.DB) {
	wc := config.C.Wechat
	if !wc.Enabled {
		log.Println("[seed] wechat.enabled=false，跳过微信商户初始化")
		return
	}
	var m model.Merchant
	err := db.Where("code = ?", wc.MerchantCode).First(&m).Error
	if errors.Is(err, gorm.ErrRecordNotFound) {
		m = model.Merchant{
			Code:    wc.MerchantCode,
			Name:    wc.MerchantName,
			Channel: "wechat",
		}
	} else if err != nil {
		log.Fatalf("[seed] 查询微信商户失败: %v", err)
	}
	m.Production = true // 微信无沙箱，一律正式网关
	m.Enabled = true
	m.MchID = wc.MchID
	m.WxAppID = wc.AppID
	m.CertSerial = wc.CertSerial
	m.WxPrivateKey = wc.PrivateKey
	m.APIv3Key = wc.APIv3Key
	if err := db.Save(&m).Error; err != nil {
		log.Fatalf("[seed] 保存微信商户失败: %v", err)
	}
	log.Printf("[seed] 微信商户就绪: code=%s mch_id=%s", m.Code, m.MchID)

	var cnt int64
	db.Model(&model.Product{}).Where("merchant_id = ?", m.ID).Count(&cnt)
	if cnt == 0 {
		samples := []model.Product{
			{MerchantID: m.ID, Name: "微信测试套餐 A", Description: "微信联调用", Price: "0.01", Active: true, Sort: 1},
			{MerchantID: m.ID, Name: "微信测试套餐 B", Description: "微信联调用", Price: "0.02", Active: true, Sort: 2},
		}
		if err := db.Create(&samples).Error; err != nil {
			log.Fatalf("[seed] 创建微信测试套餐失败: %v", err)
		}
		log.Printf("[seed] 已为微信商户 %s 创建 %d 个测试套餐", m.Code, len(samples))
	}
}