swirl/biz/role.go

package biz

import (
	"context"

	"github.com/cuigh/auxo/net/web"
	"github.com/cuigh/swirl/dao"
)

type RoleBiz interface {
	Search(name string) ([]*dao.Role, error)
	Find(id string) (role *dao.Role, err error)
	Create(role *dao.Role, user web.User) (err error)
	Delete(id, name string, user web.User) (err error)
	Update(r *dao.Role, user web.User) (err error)
	GetPerms(ids []string) ([]string, error)
}

func NewRole(d dao.Interface, eb EventBiz) RoleBiz {
	return &roleBiz{d: d, eb: eb}
}

type roleBiz struct {
	d  dao.Interface
	eb EventBiz
}

func (b *roleBiz) Search(name string) (roles []*dao.Role, err error) {
	return b.d.RoleSearch(context.TODO(), name)
}

func (b *roleBiz) Create(role *dao.Role, user web.User) (err error) {
	r := &dao.Role{
		ID:          createId(),
		Name:        role.Name,
		Description: role.Description,
		Perms:       role.Perms,
		CreatedAt:   now(),
		CreatedBy:   newOperator(user),
	}
	r.UpdatedAt = r.CreatedAt
	r.UpdatedBy = r.CreatedBy
	err = b.d.RoleCreate(context.TODO(), r)
	if err == nil {
		b.eb.CreateRole(EventActionCreate, r.ID, role.Name, user)
	}
	return
}

func (b *roleBiz) Delete(id, name string, user web.User) (err error) {
	err = b.d.RoleDelete(context.TODO(), id)
	if err == nil {
		b.eb.CreateRole(EventActionDelete, id, name, user)
	}
	return
}

func (b *roleBiz) Find(id string) (role *dao.Role, err error) {
	return b.d.RoleGet(context.TODO(), id)
}

func (b *roleBiz) Update(role *dao.Role, user web.User) (err error) {
	r := &dao.Role{
		ID:          role.ID,
		Name:        role.Name,
		Description: role.Description,
		Perms:       role.Perms,
		UpdatedAt:   now(),
		UpdatedBy:   newOperator(user),
	}
	err = b.d.RoleUpdate(context.TODO(), r)
	if err == nil {
		b.eb.CreateRole(EventActionUpdate, role.ID, role.Name, user)
	}
	return
}

func (b *roleBiz) GetPerms(ids []string) ([]string, error) {
	m := make(map[string]struct{})

	for _, id := range ids {
		r, err := b.d.RoleGet(context.TODO(), id)
		if err != nil {
			return nil, err
		}

		for _, p := range r.Perms {
			m[p] = struct{}{}
		}
	}

	perms := make([]string, 0, len(m))
	for p := range m {
		perms = append(perms, p)
	}
	return perms, nil
}