package seed

import (
	"context"
	"fmt"
	"math/rand"
	"time"

	"quyun/v2/app/commands"
	"quyun/v2/database"
	"quyun/v2/database/fields"
	"quyun/v2/database/models"
	"quyun/v2/pkg/consts"
	"quyun/v2/providers/postgres"

	"github.com/google/uuid"
	"github.com/spf13/cast"
	"github.com/spf13/cobra"
	"go.ipao.vip/atom"
	"go.ipao.vip/atom/container"
	"go.ipao.vip/gen/types"
	"go.uber.org/dig"
	"gorm.io/gorm"
)

func defaultProviders() container.Providers {
	return commands.Default(container.Providers{
		postgres.DefaultProvider(),
		database.DefaultProvider(),
	}...)
}

func Command() atom.Option {
	return atom.Command(
		atom.Name("seed"),
		atom.Short("seed initial data"),
		atom.RunE(Serve),
		atom.Providers(defaultProviders()),
	)
}

type Service struct {
	dig.In

	DB *gorm.DB
}

func Serve(cmd *cobra.Command, args []string) error {
	return container.Container.Invoke(func(ctx context.Context, svc Service) error {
		models.SetDefault(svc.DB)
		fmt.Println("Seeding data...")

		// 1. Users
		// Creator
		creator := &models.User{
			Username: "creator",
			Phone:    "13800000001",
			Nickname: "梅派传人小林",
			Avatar:   "https://api.dicebear.com/7.x/avataaars/svg?seed=Master1",
			Balance:  10000,
			Status:   consts.UserStatusVerified,
			Roles:    types.Array[consts.Role]{consts.RoleCreator},
		}
		if err := models.UserQuery.WithContext(ctx).Create(creator); err != nil {
			fmt.Printf("Create creator failed (maybe exists): %v\n", err)
			creator, _ = models.UserQuery.WithContext(ctx).Where(models.UserQuery.Phone.Eq("13800000001")).First()
		}

		// Buyer
		buyer := &models.User{
			Username: "test",
			Phone:    "13800138000",
			Nickname: "戏迷小张",
			Avatar:   "https://api.dicebear.com/7.x/avataaars/svg?seed=Zhang",
			Balance:  5000,
			Status:   consts.UserStatusVerified,
			Roles:    types.Array[consts.Role]{consts.RoleUser},
		}
		if err := models.UserQuery.WithContext(ctx).Create(buyer); err != nil {
			fmt.Printf("Create buyer failed: %v\n", err)
			buyer, _ = models.UserQuery.WithContext(ctx).Where(models.UserQuery.Phone.Eq("13800138000")).First()
		}

		// Superadmin
		superAdmin := &models.User{
			Username: "superadmin",
			Phone:    "13800009999",
			Nickname: "平台管理员",
			Avatar:   "https://api.dicebear.com/7.x/avataaars/svg?seed=Admin",
			Balance:  0,
			Status:   consts.UserStatusVerified,
			Roles:    types.Array[consts.Role]{consts.RoleSuperAdmin},
		}
		if err := models.UserQuery.WithContext(ctx).Create(superAdmin); err != nil {
			fmt.Printf("Create superadmin failed: %v\n", err)
			superAdmin, _ = models.UserQuery.WithContext(ctx).Where(models.UserQuery.Username.Eq("superadmin")).First()
		}

		// 2. Tenant
		tenant := &models.Tenant{
			UserID: creator.ID,
			Name:   "梅派艺术工作室",
			Code:   "meipai_" + cast.ToString(rand.Intn(1000)),
			UUID:   types.UUID(uuid.New()),
			Status: consts.TenantStatusVerified,
		}
		if err := models.TenantQuery.WithContext(ctx).Create(tenant); err != nil {
			fmt.Printf("Create tenant failed: %v\n", err)
			tenant, _ = models.TenantQuery.WithContext(ctx).Where(models.TenantQuery.UserID.Eq(creator.ID)).First()
		}

		// Tenant membership (buyer joins as member)
		member := &models.TenantUser{
			TenantID: tenant.ID,
			UserID:   buyer.ID,
			Role:     types.Array[consts.TenantUserRole]{consts.TenantUserRoleMember},
			Status:   consts.UserStatusVerified,
		}
		if err := models.TenantUserQuery.WithContext(ctx).Create(member); err != nil {
			fmt.Printf("Create tenant member failed: %v\n", err)
		}

		// 3. Contents
		titles := []string{
			"《锁麟囊》春秋亭 (程砚秋)", "昆曲《牡丹亭》游园惊梦", "越剧《红楼梦》葬花",
			"京剧《霸王别姬》全本实录", "京剧打击乐基础教程", "豫剧唱腔发音技巧",
			"黄梅戏《女驸马》选段", "评剧《花为媒》报花名", "秦腔《三滴血》",
			"河北梆子《大登殿》",
		}
		covers := []string{
			"https://images.unsplash.com/photo-1514306191717-452ec28c7f31?ixlib=rb-1.2.1&auto=format&fit=crop&w=400&q=60",
			"https://images.unsplash.com/photo-1557683316-973673baf926?ixlib=rb-1.2.1&auto=format&fit=crop&w=400&q=60",
			"https://images.unsplash.com/photo-1469571486292-0ba58a3f068b?ixlib=rb-1.2.1&auto=format&fit=crop&w=400&q=60",
		}

		var seededContents []*models.Content
		for i, title := range titles {
			price := int64((i % 3) * 1000) // 0, 10.00, 20.00
			if i == 3 {
				price = 990
			} // 9.90

			c := &models.Content{
				TenantID:    tenant.ID,
				UserID:      creator.ID,
				Title:       title,
				Description: fmt.Sprintf("这是关于 %s 的详细介绍...", title),
				Genre:       "京剧",
				Status:      consts.ContentStatusPublished,
				Visibility:  consts.ContentVisibilityPublic,
				Views:       int32(rand.Intn(10000)),
				Likes:       int32(rand.Intn(1000)),
			}
			if err := models.ContentQuery.WithContext(ctx).Create(c); err == nil {
				seededContents = append(seededContents, c)
			}
			// Price
			models.ContentPriceQuery.WithContext(ctx).Create(&models.ContentPrice{
				TenantID:    tenant.ID,
				UserID:      creator.ID,
				ContentID:   c.ID,
				PriceAmount: price,
				Currency:    "CNY",
			})

			// Asset (Cover)
			ma := &models.MediaAsset{
				TenantID:  tenant.ID,
				UserID:    creator.ID,
				Type:      consts.MediaAssetTypeImage,
				Status:    consts.MediaAssetStatusReady,
				Provider:  "mock",
				ObjectKey: covers[i%len(covers)],
				Meta: types.NewJSONType(fields.MediaAssetMeta{
					Size: 1024,
				}),
			}
			models.MediaAssetQuery.WithContext(ctx).Create(ma)

			models.ContentAssetQuery.WithContext(ctx).Create(&models.ContentAsset{
				TenantID:  tenant.ID,
				UserID:    creator.ID,
				ContentID: c.ID,
				AssetID:   ma.ID,
				Role:      consts.ContentAssetRoleCover,
			})
		}

		// 4. Coupons
		cp1 := &models.Coupon{
			TenantID:       tenant.ID,
			Title:          "新人立减券",
			Type:           consts.CouponTypeFixAmount,
			Value:          500, // 5.00
			MinOrderAmount: 1000,
			TotalQuantity:  100,
			StartAt:        time.Now().Add(-24 * time.Hour),
			EndAt:          time.Now().Add(30 * 24 * time.Hour),
		}
		if err := models.CouponQuery.WithContext(ctx).Create(cp1); err != nil {
			fmt.Printf("Create coupon failed: %v\n", err)
		}

		// Give to buyer
		models.UserCouponQuery.WithContext(ctx).Create(&models.UserCoupon{
			UserID:   buyer.ID,
			CouponID: cp1.ID,
			Status:   consts.UserCouponStatusUnused,
		})

		// 5. Orders & library access (first content)
		if len(seededContents) > 0 {
			content := seededContents[0]
			order := &models.Order{
				TenantID:       tenant.ID,
				UserID:         buyer.ID,
				Type:           consts.OrderTypeContentPurchase,
				Status:         consts.OrderStatusPaid,
				Currency:       consts.CurrencyCNY,
				AmountOriginal: 990,
				AmountDiscount: 0,
				AmountPaid:     990,
				IdempotencyKey: uuid.NewString(),
				PaidAt:         time.Now().Add(-2 * time.Hour),
			}
			if err := models.OrderQuery.WithContext(ctx).Create(order); err != nil {
				fmt.Printf("Create order failed: %v\n", err)
			} else {
				models.OrderItemQuery.WithContext(ctx).Create(&models.OrderItem{
					TenantID:      tenant.ID,
					UserID:        buyer.ID,
					OrderID:       order.ID,
					ContentID:     content.ID,
					ContentUserID: creator.ID,
					AmountPaid:    order.AmountPaid,
				})
				models.ContentAccessQuery.WithContext(ctx).Create(&models.ContentAccess{
					TenantID:  tenant.ID,
					UserID:    buyer.ID,
					ContentID: content.ID,
					OrderID:   order.ID,
					Status:    consts.ContentAccessStatusActive,
				})
			}
		}

		// 6. Creator join request & invite
		models.TenantJoinRequestQuery.WithContext(ctx).Create(&models.TenantJoinRequest{
			TenantID: tenant.ID,
			UserID:   buyer.ID,
			Status:   "pending",
			Reason:   "申请加入租户用于创作",
		})
		models.TenantInviteQuery.WithContext(ctx).Create(&models.TenantInvite{
			TenantID:  tenant.ID,
			UserID:    creator.ID,
			Code:      "invite" + cast.ToString(rand.Intn(100000)),
			Status:    "active",
			MaxUses:   5,
			UsedCount: 0,
			Remark:    "staging seed invite",
			ExpiresAt: time.Now().Add(7 * 24 * time.Hour),
		})

		// 7. Notifications & templates
		models.NotificationQuery.WithContext(ctx).Create(&models.Notification{
			UserID:   buyer.ID,
			TenantID: tenant.ID,
			Type:     string(consts.NotificationTypeSystem),
			Title:    "欢迎注册",
			Content:  "欢迎来到曲韵平台！",
			IsRead:   false,
		})
		models.NotificationTemplateQuery.WithContext(ctx).Create(&models.NotificationTemplate{
			TenantID: 0,
			Name:     "订单支付通知",
			Type:     consts.NotificationTypeOrder,
			Title:    "订单支付成功",
			Content:  "您的订单已支付成功。",
			IsActive: true,
		})
		models.NotificationTemplateQuery.WithContext(ctx).Create(&models.NotificationTemplate{
			TenantID: 0,
			Name:     "内容审核通知",
			Type:     consts.NotificationTypeAudit,
			Title:    "内容审核通过",
			Content:  "您提交的内容已通过审核。",
			IsActive: true,
		})

		// 8. System config
		models.SystemConfigQuery.WithContext(ctx).Create(&models.SystemConfig{
			ConfigKey:   "site_name",
			Value:       types.JSON([]byte(`{"value":"曲韵平台"}`)),
			Description: "站点名称",
		})

		// 9. Audit log
		models.AuditLogQuery.WithContext(ctx).Create(&models.AuditLog{
			TenantID:   tenant.ID,
			OperatorID: superAdmin.ID,
			Action:     "seed",
			TargetID:   fmt.Sprintf("tenant:%d", tenant.ID),
			Detail:     "staging seed data",
		})

		// 10. Content report
		if len(seededContents) > 0 {
			models.ContentReportQuery.WithContext(ctx).Create(&models.ContentReport{
				TenantID:   tenant.ID,
				ContentID:  seededContents[0].ID,
				ReporterID: buyer.ID,
				Reason:     "spam",
				Detail:     "疑似广告内容",
				Status:     "pending",
			})
		}

		// 11. Comments
		if len(seededContents) > 0 {
			models.CommentQuery.WithContext(ctx).Create(&models.Comment{
				TenantID:  tenant.ID,
				UserID:    buyer.ID,
				ContentID: seededContents[0].ID,
				Content:   "好喜欢这段演出！",
				Likes:     1,
			})
		}

		fmt.Println("Seed done.")
		return nil
	})
}