package main

import (
	"fmt"
	"golang.org/x/crypto/bcrypt"
	"gorm.io/driver/postgres"
	"gorm.io/gorm"
	"gorm.io/gorm/schema"
	"log/slog"
	"platform/pkg/env"
	"platform/pkg/logs"
	"platform/pkg/u"
	client2 "platform/web/domains/client"
	proxy2 "platform/web/domains/proxy"
	m "platform/web/models"
	q "platform/web/queries"
)

func main() {

	env.Init()
	logs.Init()

	// 初始化数据库连接
	dsn := fmt.Sprintf(
		"host=%s user=%s password=%s dbname=%s port=%s sslmode=disable TimeZone=Asia/Shanghai",
		env.DbHost, env.DbUserName, env.DbPassword, env.DbName, env.DbPort,
	)
	db, err := gorm.Open(postgres.Open(dsn), &gorm.Config{
		NamingStrategy: schema.NamingStrategy{
			SingularTable: true,
		},
	})
	if err != nil {
		slog.Error("gorm 初始化数据库失败：", slog.Any("err", err))
		panic(err)
	}

	q.SetDefault(db)

	// 填充数据
	err = q.Q.Transaction(func(tx *q.Query) (err error) {

		// 代理
		err = q.Proxy.
			Select(q.Proxy.Version, q.Proxy.Name, q.Proxy.Host, q.Proxy.Type, q.Proxy.Secret).
			Create(&m.Proxy{
				Version: 1,
				Name:    "7a17e8b4-cdc3-4500-bf16-4a665991a7f6",
				Host:    "110.40.82.248",
				Type:    int32(proxy2.TypeSelfHosted),
				Secret:  u.P("api:123456"),
			}, &m.Proxy{
				Version: 1,
				Name:    "58e03f38-4cef-429c-8bb8-530142d0a745",
				Host:    "123.6.147.241",
				Type:    int32(proxy2.TypeThirdParty),
				Secret:  u.P("api:123456"),
			})
		if err != nil {
			return err
		}

		// 客户端
		testSecret, err := bcrypt.GenerateFromPassword([]byte("test"), bcrypt.DefaultCost)
		if err != nil {
			return err
		}

		tasksSecret, err := bcrypt.GenerateFromPassword([]byte("tasks"), bcrypt.DefaultCost)
		if err != nil {
			return err
		}

		proxySecret, err := bcrypt.GenerateFromPassword([]byte("proxy"), bcrypt.DefaultCost)
		if err != nil {
			return err
		}

		err = q.Client.
			Select(
				q.Client.ClientID,
				q.Client.ClientSecret,
				q.Client.GrantClient,
				q.Client.GrantRefresh,
				q.Client.GrantPassword,
				q.Client.Spec,
				q.Client.Name,
			).
			Create(&m.Client{
				ClientID:      "test",
				ClientSecret:  string(testSecret),
				GrantCode:     true,
				GrantClient:   true,
				GrantRefresh:  true,
				GrantPassword: true,
				Spec:          int32(client2.SpecTrusted),
				Name:          "默认客户端",
			}, &m.Client{
				ClientID:     "tasks",
				ClientSecret: string(tasksSecret),
				GrantClient:  true,
				Spec:         int32(client2.SpecTrusted),
				Name:         "异步任务处理服务",
			}, &m.Client{
				ClientID:     "proxy",
				ClientSecret: string(proxySecret),
				GrantClient:  true,
				Spec:         int32(client2.SpecTrusted),
				Name:         "代理转发服务",
			}, &m.Client{
				ClientID:    "edge",
				GrantClient: true,
				Spec:        int32(client2.SpecWeb),
				Name:        "代理边缘节点",
			})
		if err != nil {
			return err
		}

		return nil
	})
	if err != nil {
		panic(err)
	}

	slog.Info("✔ Data inserted successfully")
}