temp-526互联

package group

import (
   "Yearning-go/src/handler/common"
   "Yearning-go/src/lib"
   "Yearning-go/src/model"
   "encoding/json"
   "fmt"
   "github.com/cookieY/yee"
   "github.com/google/uuid"
   "net/http"
   "net/url"
   "strings"
)

func SuperGetRuseSource(c yee.Context) (err error) {
   var source []model.CoreDataSource
   var query []model.CoreDataSource
   model.DB().Select("source,source_id").Scopes(common.AccordingToGroupSourceIsQuery(0, 2)).Find(&source)
   model.DB().Select("source,source_id").Scopes(common.AccordingToGroupSourceIsQuery(1, 2)).Find(&query)
   return c.JSON(http.StatusOK, common.SuccessPayload(map[string]interface{}{"source": source, "query": query}))
}

func SuperGroup(c yee.Context) (err error) {
   u := new(common.PageList[[]model.CoreRoleGroup])
   if err = c.Bind(u); err != nil {
      return err
   }
   return c.JSON(http.StatusOK, u.Paging().Query(common.AccordingToOrderName(u.Expr.Text)).ToMessage())
}

func SuperGroupUpdate(c yee.Context) (err error) {
   user := new(lib.Token).JwtParse(c)
   if user.Username == "admin" {
      u := new(policy)
      if err = c.Bind(u); err != nil {
         c.Logger().Error(err.Error())
         return c.JSON(http.StatusOK, common.ERR_REQ_BIND)
      }
      g, err := json.Marshal(u.PermissionList)
      if err != nil {
         c.Logger().Error(err.Error())
         return c.JSON(http.StatusOK, common.ERR_COMMON_MESSAGE(err))
      }

      if u.ID == 0 {
         model.DB().Create(&model.CoreRoleGroup{
            Name:        u.Name,
            Permissions: g,
            GroupId:     uuid.New().String(),
         })
      } else {
         model.DB().Model(model.CoreRoleGroup{}).Scopes(common.AccordingToIDEqual(u.ID)).Updates(&model.CoreRoleGroup{Permissions: g, Name: u.Name})
      }
      return c.JSON(http.StatusOK, common.SuccessPayLoadToMessage(fmt.Sprintf(GROUP_CREATE_SUCCESS, u.Name)))
   }
   return c.JSON(http.StatusOK, common.ERR_REQ_FAKE)
}

func SuperClearUserRule(c yee.Context) (err error) {
   args := c.QueryParam("group_id")
   scape, _ := url.QueryUnescape(args)
   var j []model.CoreGrained
   model.DB().Scopes(common.AccordingToGroupNameIsLike(scape)).Find(&j)
   for _, i := range j {
      b, err := lib.ArrayRemove(i.Group, scape)
      if err != nil {
         return c.JSON(http.StatusOK, common.ERR_COMMON_MESSAGE(err))
      }
      go model.DB().Model(model.CoreGrained{}).Scopes(common.AccordingToUsernameEqual(i.Username)).Updates(&model.CoreGrained{Group: b})
   }
   model.DB().Model(model.CoreRoleGroup{}).Where("group_id = ?", scape).Delete(&model.CoreRoleGroup{})
   return c.JSON(http.StatusOK, common.SuccessPayLoadToMessage(fmt.Sprintf(GROUP_DELETE_SUCCESS, scape)))
}

func SuperUserRuleMarge(c yee.Context) (err error) {
   u := new(marge)
   if err = c.Bind(u); err != nil {
      return c.JSON(http.StatusOK, common.ERR_REQ_BIND)
   }
   m3 := lib.NewMultiUserRuleSet(strings.Split(u.Group, ","))
   return c.JSON(http.StatusOK, common.SuccessPayload(m3))
}