package scripts

import (
	"fmt"
	"mygo/dbhelper"
	"mygo/libs"
	"mygo/model"
	"mygo/response"
	"mygo/say"
	"mygo/validator"

	"github.com/labstack/echo/v4"
	"gorm.io/gorm"
)

//获取角色授权信息
func GetRoutesByRoleId(c echo.Context) error {
	fmt.Println("获取角色授权信息 ")
	id := c.QueryParam("id")
	role_id := c.QueryParam("role_id")
	data := response.RoleRoutesResponse{}
	if id == "" || role_id == "" {
		return say.Success(c, 10001, data)
	}

	db := dbhelper.GetMySqlDB()
	//获取角色对应路由列表
	ids, _ := model.GetRouteIdsByRoleId(db, role_id)

	routes, err := model.GetRoleRoutes(db, ids)

	if err != nil {
		return say.Success(c, 10000, data)
	}

	data.Routes = routes

	return say.Success(c, 0, data)
}

//角色授权
func RoleBindRoute(c echo.Context) error {
	id := c.QueryParam("id")
	data := response.NoneResponse{}
	body := &validator.RoleBindRouteBody{}
	err := c.Bind(body)
	if err != nil {
		return say.Success(c, 10000, data)
	}
	if id == "" || len(body.Routes) <= 0 || body.RoleID == "" {
		return say.Success(c, 10001, data)
	}

	db := dbhelper.GetMySqlDB()

	role_id := body.RoleID
	role, err := model.GetRoleById(db, role_id)
	if err != nil || role.ID == "" {
		return say.Success(c, 10006, data)
	}

	//启动事务
	txerr := db.Transaction(func(tx *gorm.DB) error {
		// 在事务中执行一些 db 操作（从这里开始，您应该使用 'tx' 而不是 'db'）

		//批量删除存在的授权
		rr := model.RoleRoute{}
		if err := tx.Debug().Where("role_id = ? and deleted_at is NULL", role_id).Delete(&rr).Error; err != nil {
			// 返回任何错误都会回滚事务
			return err
		}

		for _, route_id := range body.Routes {

			r, err := model.GetRoleRoute(tx, role_id, route_id)
			if err != nil {
				return err
			}

			if r.ID != "" {
				continue
			}
			id := libs.GenId()
			route := model.RoleRoute{
				ID:      id,
				RouteID: route_id,
				RoleID:  role_id,
			}

			if err := tx.Create(&route).Error; err != nil {
				// 返回任何错误都会回滚事务
				return err
			}
		}

		// 返回 nil 提交事务
		return nil
	})

	if txerr != nil {
		return say.Success(c, 10000, data)
	}

	return say.Success(c, 0, data)
}