package security

import (
	"fmt"
	"gitee.com/kristas/booting-go/framework/common/util/lang"
	"gitee.com/kristas/booting-go/framework/core/bean"
	"gitee.com/kristas/booting-go/framework/logging"
	"github.com/casbin/casbin/v2"
	"github.com/casbin/casbin/v2/model"
	"github.com/casbin/gorm-adapter/v3"
	"gorm.io/gorm"
)

/*
https://casbin.org/docs/zh-CN/overview
*/

const CasbinDatasource = "casbin_datasource"

type Casbin struct {
	bean.Component
	e     *casbin.Enforcer
	DB    *gorm.DB          `db:"casbin_datasource"`
	Model map[string]string `value:"casbin.model"`
	Log   logging.Logger    `bean:"log"`
}

func (r *Casbin) AfterInit() {
	e, err := InitDatabase(r.Model, r.DB)
	if err != nil {
		r.Log.Error(err)
	}
	r.e = e
}

func (r *Casbin) Repository() string {
	return ""
}

func (r Casbin) GetEnforcer() *casbin.Enforcer {
	return r.e
}

func InitDatabase(m map[string]string, db *gorm.DB) (*casbin.Enforcer, error) {
	adapter, err := gormadapter.NewAdapterByDB(db)
	if err != nil {
		return nil, err
	}
	newModelFromString, err := model.NewModelFromString(parseModel(m))
	if err != nil {
		panic(err)
	}
	e, err := casbin.NewEnforcer(newModelFromString, adapter)
	if err != nil {
		return nil, err
	}
	err = e.LoadPolicy()
	e.AddPolicy("Administrator", "/*", "(GET)|(POST)|(PUT)|(PATCH)|(OPTIONS)")
	e.AddGroupingPolicy("admin", "Administrator")
	if err != nil {
		return nil, err
	}
	return e, nil
}

func parseModel(m map[string]string) string {
	s := lang.NewEmptyString()
	for k, v := range m {
		s = s.Concat(fmt.Sprintf("[%s]\n%s\n", k, v))
	}
	return s.String()
}