yuyue/Modules/Admin/Service/NodeService.class.php

<?php
namespace Admin\Service;

/**
 * NodeService
 */
class NodeService extends CommonService {
    /**
     * 节点类型
     * @var array
     */
    private $NODE_TYPE = array(
        1 => '应用（GROUP）',
        2 => '模块（MODULE）',
        3 => '操作（ACTION）'
    );

    /**
     * 得到带又层级的node数据
     * @return array
     */
    public function getNodes() {
        $category = new \Org\Util\Category($this->getModelName(),
                                           array('id', 'pid','title'));
       
        return $category->getList();
    }

	/**
     * 添加节点
     * @param  array $role 节点信息
     * @return array
     */
    public function addNode($node) {
        $Node = $this->getD();

        if (false === ($node = $Node->create($node))) {
            return $this->errorResultReturn($Node->getError());
        }

        if (false === $Node->add($node)) {
            return $this->errorResultReturn('系统错误！');
        }

        return $this->resultReturn(true);
    }
    
    /**
     * 得到应用节点
     * @param  array  $where 查询条件
     * @return array
     */
    public function getGroupNodes(array $where) {
        if (!isset($where) || !is_array($where)) {
            $where = array();
        }

        $map = array('level' => 1);
        return $this->getM()->where(array_merge($map, $where))->select();
    }

    /**
     * 得到模块节点
     * @param  array  $where 查询条件
     * @return array
     */
    public function getModuleNodes(array $where) {
        if (!isset($where) || !is_array($where)) {
            $where = array();
        }

        $map = array('level' => 2);
        return $this->getM()->where(array_merge($map, $where))->select();
    }

    /**
     * 得到操作节点
     * @param  array  $where 查询条件
     * @return array
     */
    public function getActionNodes(array $where) {
        if (!isset($where) || !is_array($where)) {
            $where = array();
        }

        $map = array('level' => 3);
        return $this->getM()->where(array_merge($map, $where))->select();
    }

    /**
     * 得到带有级别所有的节点
     * @return array
     */
    public function getLevelNodes() {
        $groups = $this->getGroupNodes();
        foreach ($groups as $i => $group) {
            $where['pid'] = $group['id'];
            $modules = $this->getModuleNodes($where);
            foreach ($modules as $j => $module) {
                $where['pid'] = $module['id'];
                $actions = $this->getActionNodes($where);
                $modules[$j]['actions'] = $actions;
            }

            $groups[$i]['modules'] = $modules;
        }

        return $groups;
    }

    /**
     * 得到节点的类型
     * @param  int    $type 节点的类型
     * @return string
     */
    public function getNodeType($type) {
        return $this->NODE_TYPE[$type];
    }

    /**
     * 添加模块管理节点
     * @param  string $name     节点名称
     * @param  string $ctrlName 控制器名称
     * @return boolean
     */
    public function addModuleNodes($name, $ctrlName) {
        $Node = $this->getM();

        // 得到顶级节点的id
        $pNode = $Node->field('id')->getByPid(0);
        if (is_null($pNode)) {
            return false;
        }

        $Node->startTrans();
        $node = array(
            'status' => 1,
            'created_at' => time(),
            'updated_at' => time()
        );

        // 模块节点
        $mNode = array(
            'pid' => $pNode['id'],
            'title' => "{$name}管理",
            'name' => $ctrlName,
            'level' => 2
        );
        $ms = $Node->add(array_merge($mNode, $node));

        // 模块id
        $pid = $this->getM()->getLastInsId();
        // 操作节点
        $node['pid'] = $pid;
        $node['level'] = 3;

        // index
        $index = array('title' => "{$name}管理", 'name' => "index");
        // add
        $add = array('title' => "添加{$name}", 'name' => "add");
        // create
        $create = array('title' => "创建{$name}", 'name' => "create");
        // edit
        $edit = array('title' => "编辑{$name}", 'name' => "edit");
        // update
        $update = array('title' => "更新{$name}", 'name' => "update");
        // delete
        $delete = array('title' => "删除{$name}", 'name' =>"delete");
        // medit
        $medit = array('title' => "批量操作{$name}", 'name' =>"medit");

        $nodes = array(
            array_merge($node, $index),
            array_merge($node, $add),
            array_merge($node, $create),
            array_merge($node, $edit),
            array_merge($node, $update),
            array_merge($node, $delete),
            array_merge($node, $medit),
        );

        $ns = $Node->addAll($nodes);
        if (false === $ms || false === $ns) {
            $Node->rollback();
            return false;
        }

        $Node->commit();
        return true;
    }

    /**
     * 删除模块管理节点
     * @param  string  $ctrlName 控制器名称
     * @return boolean
     */
    public function deleteModuleNodes($ctrlName) {
        $Node = $this->getM();
        $mNode = $Node->field('id')->getByName($ctrlName);

        if (is_null($mNode)) {
            return false;
        }

        $Node->delete($mNode['id']);
        $Node->where("pid={$mNode['id']}")->delete();

        return true;
    }

    /**
     * 设置节点状态
     * @param  int   $id     节点id
     * @param  int   $status 节点状态
     * @return mixed
     */
    public function setStatus($id, $status) {
        return $this->getM()
                    ->where("id={$id}")
                    ->save(array('status' => $status));
    }

    /**
     * 节点是否存在
     * @param  int     $id 节点id
     * @return boolean
     */
    public function existNode($id) {
        $node = $this->getM()->getById($id);
        return !empty($node);
    }

    protected function getModelName() {
        return 'Node';
    }
}