package model

import (
	"perms-system-server/internal/model/dept"
	"perms-system-server/internal/model/perm"
	"perms-system-server/internal/model/product"
	"perms-system-server/internal/model/productmember"
	"perms-system-server/internal/model/role"
	"perms-system-server/internal/model/roleperm"
	"perms-system-server/internal/model/user"
	"perms-system-server/internal/model/userperm"
	"perms-system-server/internal/model/userrole"

	"github.com/zeromicro/go-zero/core/stores/cache"
	"github.com/zeromicro/go-zero/core/stores/sqlx"
)

type Models struct {
	SysProductModel       product.SysProductModel
	SysPermModel          perm.SysPermModel
	SysRoleModel          role.SysRoleModel
	SysRolePermModel      roleperm.SysRolePermModel
	SysDeptModel          dept.SysDeptModel
	SysUserModel          user.SysUserModel
	SysUserRoleModel      userrole.SysUserRoleModel
	SysUserPermModel      userperm.SysUserPermModel
	SysProductMemberModel productmember.SysProductMemberModel
}

func NewModels(conn sqlx.SqlConn, cacheConf cache.CacheConf, cachePrefix string) *Models {
	return &Models{
		SysProductModel:       product.NewSysProductModel(conn, cacheConf, cachePrefix),
		SysPermModel:          perm.NewSysPermModel(conn, cacheConf, cachePrefix),
		SysRoleModel:          role.NewSysRoleModel(conn, cacheConf, cachePrefix),
		SysRolePermModel:      roleperm.NewSysRolePermModel(conn, cacheConf, cachePrefix),
		SysDeptModel:          dept.NewSysDeptModel(conn, cacheConf, cachePrefix),
		SysUserModel:          user.NewSysUserModel(conn, cacheConf, cachePrefix),
		SysUserRoleModel:      userrole.NewSysUserRoleModel(conn, cacheConf, cachePrefix),
		SysUserPermModel:      userperm.NewSysUserPermModel(conn, cacheConf, cachePrefix),
		SysProductMemberModel: productmember.NewSysProductMemberModel(conn, cacheConf, cachePrefix),
	}
}