//
// Copyright (C) 2024 veypi <i@veypi.com>
// 2024-09-20 16:10:16
// Distributed under terms of the MIT license.
//

package models

import (
	"github.com/veypi/OneAuth/cfg"
	"strings"
	"time"

	"github.com/google/uuid"
	"github.com/veypi/OneBD/rest/middlewares/crud"
	"github.com/veypi/utils/logv"
	"gorm.io/gorm"
)

type BaseModel struct {
	ID        string         `json:"id" gorm:"primaryKey;type:varchar(32)" methods:"get,patch,delete" parse:"path"`
	CreatedAt time.Time      `json:"created_at" methods:"*list" parse:"query"`
	UpdatedAt time.Time      `json:"updated_at" methods:"*list" parse:"query"`
	DeletedAt gorm.DeletedAt `gorm:"index" json:"-"`
}

func (m *BaseModel) BeforeCreate(tx *gorm.DB) error {
	if m.ID == "" {
		m.ID = strings.ReplaceAll(uuid.New().String(), "-", "")
	}
	return nil
}

var AllModels = crud.New()

func init() {
	AllModels.Append(User{}, AppUser{}, Resource{}, Access{}, Role{}, UserRole{}, Token{}, App{})
	AllModels.One2Many(App{}, Resource{})
	AllModels.One2Many(App{}, Role{})
	AllModels.One2Many(App{}, Access{})
	AllModels.Many2Many(AppUser{}, App{}, User{})
	AllModels.Many2Many(UserRole{}, App{}, User{}, Role{})
}

func Migrate() error {
	return crud.AutoMigrate(cfg.DB(), AllModels)
}

func Drop() error {
	return crud.AutoDrop(cfg.DB(), AllModels)
}

func InitDB() error {
	app := &App{}
	app.ID = cfg.Config.ID
	app.Name = "OA"
	app.Icon = "/favicon.ico"
	app.Key = cfg.Config.Key
	logv.AssertError(cfg.DB().Where("id = ?", app.ID).Attrs(app).FirstOrCreate(app).Error)
	initRole := map[string]map[string]uint{
		"user": {"admin": 5, "normal": 1},
		"app":  {"admin": 5, "normal": 2},
		"fs":   {"admin": 5, "normal": 5},
	}
	adminID := ""
	for r, roles := range initRole {
		resource := &Resource{
			AppID: app.ID,
			Name:  r,
		}
		logv.AssertError(cfg.DB().Where("app_id = ? AND name = ?", app.ID, r).FirstOrCreate(resource).Error)
		for rName, l := range roles {
			role := &Role{}
			logv.AssertError(cfg.DB().Where("app_id = ? AND name = ?", app.ID, rName).Attrs(&Role{
				BaseModel: BaseModel{
					ID: strings.ReplaceAll(uuid.New().String(), "-", ""),
				},
				AppID: app.ID,
				Name:  rName,
			}).FirstOrCreate(role).Error)
			logv.AssertError(cfg.DB().Where("app_id = ? AND role_id = ? AND name = ?", app.ID, role.ID, r).FirstOrCreate(&Access{
				AppID:      app.ID,
				RoleID:     &role.ID,
				ResourceID: &resource.ID,
				Name:       r,
				Level:      l,
			}).Error)
			if rName == "admin" {
				adminID = role.ID
			}
		}
	}
	if app.InitRoleID == nil {
		logv.AssertError(cfg.DB().Model(app).Update("init_role_id", adminID).Error)
	}
	return nil
}