zentao/module/zanode/zen.php

<?php
/**
 * The zen file of zanode module of ZenTaoPMS.
 *
 * @copyright   Copyright 2009-2023 禅道软件（青岛）有限公司(ZenTao Software (Qingdao) Co., Ltd. www.zentao.net)
 * @license     ZPL(https://zpl.pub/page/zplv12.html) or AGPL(https://www.gnu.org/licenses/agpl-3.0.en.html)
 * @author      Qiyu Xie <xieqiyu@easycorp.ltd>
 * @package     admin
 * @link        https://www.zentao.net
 */
class zanodeZen extends zanode
{
    /**
     * 操作执行节点。
     * Handle node.
     *
     * @param  int    $nodeID
     * @param  string $type boot|destroy|suspend|reboot|resume
     * @access protected
     * @return void
     */
    protected function handleNode($nodeID, $type)
    {
        $node = $this->zanode->getNodeByID($nodeID);

        if(in_array($node->status, array('restoring', 'creating_img', 'creating_snap')))
        {
            return $this->sendError(sprintf($this->lang->zanode->busy, $this->lang->zanode->statusList[$node->status]), true);
        }

        $url    = 'http://' . $node->ip . ':' . $node->hzap . '/api/v1/kvm/' . $node->name . '/' . $type;
        $param  = array('vmUniqueName' => $node->name);
        $result = commonModel::http($url, $param, array(), array("Authorization:$node->tokenSN"), 'data', 'POST', 10);
        $result = json_decode($result, true);

        if(empty($result)) return $this->sendError($this->lang->zanode->notFoundAgent, true);

        if($result['code'] != 'success') return $this->sendError(zget($this->lang->zanode->apiError, $result['code'], $result['msg']), true);

        if($type != 'reboot')
        {
            $status = $type == 'suspend' ? 'suspend' : 'running';
            if($type == 'destroy') $status = 'shutoff';

            $this->dao->update(TABLE_ZAHOST)->set('status')->eq($status)->where('id')->eq($nodeID)->exec();
        }
        $this->loadModel('action')->create('zanode', $nodeID, ucfirst($type));
        return $this->sendSuccess(array('message' => $this->lang->zanode->actionSuccess, 'load' => true));
    }

    /**
     * 处理创建 zanode 请求数据。
     * Processing request data for creating zanode.
     *
     * @access protected
     * @return object
     */
    protected function prepareCreateExtras()
    {
        if($this->post->hostType == 'physics') $this->config->zanode->create->requiredFields = $this->config->zanode->create->physicsRequiredFields;

        if($this->loadModel('zahost')->hiddenHost() || $this->post->hostType == 'physics')
        {
            $this->config->zanode->form->create['memory']['required']   = false;
            $this->config->zanode->form->create['diskSize']['required'] = false;
        }


        $data = form::data()
            ->setDefault('type', 'node')
            ->setDefault('createdDate', helper::now())
            ->setDefault('createdBy', $this->app->user->account)
            ->setDefault('status', 'running')
            ->setIF($this->post->hostType != 'physics', 'hostType', '')
            ->setIF($this->post->hostType == 'physics', 'parent', 0)
            ->setIF($this->post->hostType == 'physics', 'osName', $this->post->osNamePhysics)
            ->setIF($this->post->hostType == 'physics', 'secret', md5($this->post->name . time()))
            ->setIF($this->post->hostType == 'physics', 'status', 'offline')
            ->get();

        $checkResult = $this->zanode->checkFields4Create($data);
        if(!$checkResult) return $this->send(array('result' => 'fail', 'message' => dao::getError()));

        if($data->hostType != 'physics')
        {
            $result = $this->zanode->linkAgentService($data);
            if(!$result) return $this->send(array('result' => 'fail', 'message' => dao::getError()));

            $data->mac = $result->data->mac;
            $data->vnc = (int)$result->data->vnc;
        }

        return $data;
    }

    /**
     * 处理创建快照请求数据。
     * Processing request data for creating snapshot.
     *
     * @param  int nodeID
     * @access protected
     * @return object
     */
    protected function prepareCreateSnapshotExtras($node)
    {
        $data = form::data()->get();
        if(is_numeric($data->name)) return $this->sendError(array('name' => sprintf($this->lang->error->code, $this->lang->zanode->name)));

        $snapshot = new stdClass();
        $snapshot->host        = $node->id;
        $snapshot->name        = $data->name;
        $snapshot->desc        = $data->desc;
        $snapshot->status      = 'creating';
        $snapshot->osName      = $node->osName;
        $snapshot->memory      = 0;
        $snapshot->disk        = 0;
        $snapshot->fileSize    = 0;
        $snapshot->from        = 'snapshot';
        $snapshot->createdBy   = $this->app->user->account;
        $snapshot->createdDate = helper::now();

        return $snapshot;
    }

    /**
     * 获取导出镜像状态。
     * Get task status by zagent api.
     *
     * @param  object     $node
     * @param  int        $taskID
     * @param  string     $type
     * @param  string     $status
     * @access protected
     * @return false|array
     */
    protected function getTaskStatus($node, $taskID = 0, $type = '', $status = '')
    {
        $agnetUrl = 'http://' . $node->ip . ':' . $node->hzap . '/api/v1/task/getStatus';
        $result   = json_decode(commonModel::http($agnetUrl, array(), array(CURLOPT_CUSTOMREQUEST => 'POST'), array("Authorization:$node->tokenSN"), 'json', 'POST', 10));
        if(empty($result) || $result->code != 'success') return false;

        $data = $result->data;
        if(empty($data)) return array();

        if($status && !$taskID && isset($data->$status)) return $data->$status;
        if(!$taskID) return $data;

        foreach($data as $status => $tasks)
        {
            if(empty($tasks)) continue;

            foreach($tasks as $task)
            {
                if(!empty($tasks['inprogress']) && $task->task != $tasks['inprogress'][0]->task && $task->status == 'created') $task->status = 'pending';
                if($type == $task->type && $taskID == $task->task) return $task;
            }
        }
        return $result;
    }

    /**
     * 获取宿主机中zagent、nginx、websockify、novnc的运行及安装状态.
     * Get service status from host.
     *
     * @param  object $node
     * @access protected
     * @return array
     */
    protected function getServiceStatus($node)
    {
        $result = json_decode(commonModel::http("http://{$node->ip}:{$node->zap}/api/v1/service/check", json_encode(array('services' => 'all')), array(), array("Authorization:$node->tokenSN"), 'data', 'POST', 10));

        if(empty($result->data->ztfStatus) || $result->code != 'success') return $this->lang->zanode->init->serviceStatus;

        return array('ZenAgent' => 'ready', 'ZTF' => $result->data->ztfStatus);
    }

    /**
     * 执行节点安装应用（支持ztf,zendata）.
     * Install service to node by name.
     *
     * @param  object $node
     * @param  string $name
     * @access protected
     * @return array
     */
    protected function installService($node, $name)
    {
        $param = array(
            'name'   => strtolower($name),
            'secret' => $node->secret,
            'server' => getWebRoot(true),
        );
        $result = json_decode(commonModel::http("http://{$node->ip}:{$node->zap}/api/v1/service/setup", json_encode($param), array(), array("Authorization:$node->tokenSN")));

        if(empty($result->data) || $result->code != 'success') return $this->lang->zanode->init->serviceStatus;

        return array('ZenAgent' => 'ready', 'ZTF' => $result->data->ztfStatus,);
    }
}