ApiSys/backend/Server/Web/Dao/EnvDao.class.php

<?php

/**
 * @name eolinker ams open source，eolinker开源版本
 * @link https://www.eolinker.com/
 * @package eolinker
 * @author www.eolinker.com 广州银云信息科技有限公司 ©2015-2018
 * eoLinker是目前全球领先、国内最大的在线API接口管理平台，提供自动生成API文档、API自动化测试、Mock测试、团队协作等功能，旨在解决由于前后端分离导致的开发效率低下问题。
 * 如在使用的过程中有任何问题，欢迎加入用户讨论群进行反馈，我们将会以最快的速度，最好的服务态度为您解决问题。
 *
 * eoLinker AMS开源版的开源协议遵循Apache License 2.0，如需获取最新的eolinker开源版以及相关资讯，请访问:https://www.eolinker.com/#/os/download
 *
 * 官方网站：https://www.eolinker.com/
 * 官方博客以及社区：http://blog.eolinker.com/
 * 使用教程以及帮助：http://help.eolinker.com/
 * 商务合作邮箱：market@eolinker.com
 * 用户讨论QQ群：284421832
 */
class EnvDao
{
    /**
     * 获取环境列表
     * @param $projectID int 项目ID
     * @return bool
     */
    public function getEnvList(&$projectID)
    {
        $db = getDatabase();

        $envList = $db->prepareExecuteAll("SELECT eo_api_env.envID,eo_api_env.envName FROM eo_api_env WHERE eo_api_env.projectID = ? ORDER BY eo_api_env.envID DESC;", array($projectID));

        if (is_array($envList)) {
            foreach ($envList as &$env) {
                $env['frontURIList'] = $db->prepareExecuteAll("SELECT eo_api_env_front_uri.applyProtocol,eo_api_env_front_uri.uri,eo_api_env_front_uri.uriID FROM eo_api_env_front_uri WHERE eo_api_env_front_uri.envID = ?;", array($env['envID']));
                $env['headerList'] = $db->prepareExecuteAll("SELECT eo_api_env_header.applyProtocol,eo_api_env_header.headerName,eo_api_env_header.headerValue,eo_api_env_header.headerID FROM eo_api_env_header WHERE eo_api_env_header.envID = ?;", array($env['envID']));
                $env['paramList'] = $db->prepareExecuteAll("SELECT eo_api_env_param.paramKey,eo_api_env_param.paramValue,eo_api_env_param.paramID FROM eo_api_env_param WHERE eo_api_env_param.envID = ?;", array($env['envID']));
                $env['additionalParamList'] = $db->prepareExecuteAll('SELECT eo_api_env_param_additional.paramKey,eo_api_env_param_additional.paramValue,eo_api_env_param_additional.paramID FROM eo_api_env_param_additional WHERE eo_api_env_param_additional.envID = ?;', array($env['envID']));
            }

        }

        if (empty($envList))
            return FALSE;
        else
            return $envList;
    }

    /**
     * 添加环境
     * @param int $projectID 项目ID
     * @param string $envName 环境名称
     * @param string $front_uri 前置URI
     * @param array $headers 请求头部
     * @param array $params 全局变量
     * @param int $apply_protocol 应用的请求类型,[-1]=>[所有请求类型]
     * @param array $additional_params 额外参数
     * @return bool|int
     */
    public function addEnv(&$projectID, &$envName, &$front_uri, &$headers, &$params, $apply_protocol, &$additional_params)
    {
        $db = getDatabase();
        try {
            $db->beginTransaction();
            //新建环境
            $db->prepareExecute("INSERT INTO eo_api_env (eo_api_env.envName,eo_api_env.projectID) VALUES (?,?);", array(
                $envName,
                $projectID
            ));
            $env_id = $db->getLastInsertID();
            if (empty($env_id))
                throw new \PDOException('addEnv Error');
            if ($front_uri) {
                //新建前置URI
                $db->prepareExecute("INSERT INTO eo_api_env_front_uri (eo_api_env_front_uri.envID,eo_api_env_front_uri.applyProtocol,eo_api_env_front_uri.uri) VALUES (?,?,?);", array(
                    $env_id,
                    $apply_protocol,
                    $front_uri
                ));
                if ($db->getAffectRow() < 1)
                    throw new \PDOException('addFrontURI Error');
            }
            if (!empty($headers)) {
                foreach ($headers as $k => $v) {
                    //新建请求头部
                    $db->prepareExecute("INSERT INTO eo_api_env_header (eo_api_env_header.envID,eo_api_env_header.applyProtocol,eo_api_env_header.headerName,eo_api_env_header.headerValue) VALUES (?,?,?,?);", array(
                        $env_id,
                        $apply_protocol,
                        $k,
                        $v
                    ));
                    if ($db->getAffectRow() < 1)
                        throw new \PDOException('addHeader Error');
                }
            }
            if (!empty($params)) {
                foreach ($params as $k => $v) {
                    //新建全局变量
                    $db->prepareExecute("INSERT INTO eo_api_env_param (eo_api_env_param.envID,eo_api_env_param.paramKey,eo_api_env_param.paramValue) VALUES (?,?,?);", array(
                        $env_id,
                        $k,
                        $v
                    ));
                    if ($db->getAffectRow() < 1)
                        throw new \PDOException('addParam Error');
                }
            }
            if (!empty($additional_params)) {
                foreach ($additional_params as $k => $v) {
                    //新建额外参数
                    $db->prepareExecute("INSERT INTO eo_api_env_param_additional(eo_api_env_param_additional.envID,eo_api_env_param_additional.paramKey,eo_api_env_param_additional.paramValue) VALUES (?,?,?);", array(
                        $env_id,
                        $k,
                        $v
                    ));
                    if ($db->getAffectRow() < 1)
                        throw new \PDOException('addAdditionalParam Error');
                }
            }
            $db->commit();
            $db->close();
            return $env_id;
        } catch (\PDOException $e) {
            $db->rollback();
            return FALSE;
        }
    }

    /**
     * 删除环境
     * @param $projectID int 项目ID
     * @param $env_id int 环境ID
     * @return bool
     */
    public function deleteEnv(&$projectID, &$env_id)
    {
        $db = getDatabase();
        $db->beginTransaction();
        $result = $db->prepareExecute('SELECT * FROM eo_api_env_front_uri WHERE eo_api_env_front_uri.envID = ?;', array(
            $env_id
        ));
        if (!empty($result)) {
            //删除旧的前置URI
            $db->prepareExecute("DELETE FROM eo_api_env_front_uri WHERE eo_api_env_front_uri.envID = ?;", array(
                $env_id
            ));
            if ($db->getAffectRow() < 1) {
                $db->rollback();
                return FALSE;
            }
        }
        $result = $db->prepareExecute('SELECT * FROM eo_api_env_header WHERE eo_api_env_header.envID = ?;', array(
            $env_id
        ));
        if (!empty($result)) {
            //删除旧的请求头部
            $db->prepareExecute("DELETE FROM eo_api_env_header WHERE eo_api_env_header.envID = ?;", array(
                $env_id
            ));
            if ($db->getAffectRow() < 1) {
                $db->rollback();
                return FALSE;
            }
        }
        $result = $db->prepareExecute('SELECT * FROM eo_api_env_param WHERE eo_api_env_param.envID = ?;', array(
            $env_id
        ));
        if (!empty($result)) {
            //删除旧的全局变量
            $db->prepareExecute("DELETE FROM eo_api_env_param WHERE eo_api_env_param.envID = ?;", array(
                $env_id
            ));
            if ($db->getAffectRow() < 1) {
                $db->rollback();
                return FALSE;
            }
        }
        $result = $db->prepareExecute('SELECT * FROM eo_api_env_param_additional WHERE eo_api_env_param_additional.envID = ?;', array(
            $env_id
        ));
        if (!empty($result)) {
            //删除额外参数
            $db->prepareExecuteAll('DELETE FROM eo_api_env_param_additional WHERE eo_api_env_param_additional.envID = ?;', array(
                $env_id
            ));
            if ($db->getAffectRow() < 1) {
                $db->rollback();
                return FALSE;
            }
        }
        //删除环境
        $db->prepareExecute("DELETE FROM eo_api_env WHERE eo_api_env.envID = ? AND eo_api_env.projectID = ?;", array(
            $env_id,
            $projectID
        ));
        if ($db->getAffectRow() > 0) {
            $db->commit();
            return TRUE;
        } else {
            $db->rollback();
            return FALSE;
        }

    }

    /**
     * 修改环境
     * @param $env_id int 环境ID
     * @param $envName string 环境名称
     * @param $front_uri string 前置uri
     * @param $headers array 请求头部
     * @param $params array 全局变量
     * @param $apply_protocol int 请求协议
     * @param $additional_params array 额外参数
     * @return bool
     */
    public function editEnv(&$env_id, &$envName, &$front_uri, &$headers, &$params, $apply_protocol, &$additional_params)
    {
        $db = getDatabase();
        try {
            $db->beginTransaction();
            $db->prepareExecute("UPDATE eo_api_env SET eo_api_env.envName = ? WHERE eo_api_env.envID = ?;", array(
                $envName,
                $env_id
            ));
            //删除旧的前置URI
            $db->prepareExecute("DELETE FROM eo_api_env_front_uri WHERE eo_api_env_front_uri.envID = ?;", array(
                $env_id
            ));

            //删除旧的请求头部
            $db->prepareExecute("DELETE FROM eo_api_env_header WHERE eo_api_env_header.envID = ?;", array(
                $env_id
            ));

            //删除旧的全局变量
            $db->prepareExecute("DELETE FROM eo_api_env_param WHERE eo_api_env_param.envID = ?;", array(
                $env_id
            ));

            //删除旧的额外参数
            $db->prepareExecuteAll('DELETE FROM eo_api_env_param_additional WHERE eo_api_env_param_additional.envID = ?;', array(
                $env_id
            ));

            if ($front_uri) {
                // 新建前置URI
                $db->prepareExecute("INSERT INTO eo_api_env_front_uri (eo_api_env_front_uri.envID,eo_api_env_front_uri.applyProtocol,eo_api_env_front_uri.uri) VALUES (?,?,?);", array(
                    $env_id,
                    $apply_protocol,
                    $front_uri
                ));
                if ($db->getAffectRow() < 1)
                    throw new \PDOException('addFrontURI Error');
            }
            if (!empty($headers)) {
                foreach ($headers as $k => $v) {
                    // 新建请求头部
                    $db->prepareExecute("INSERT INTO eo_api_env_header (eo_api_env_header.envID,eo_api_env_header.applyProtocol,eo_api_env_header.headerName,eo_api_env_header.headerValue) VALUES (?,?,?,?);", array(
                        $env_id,
                        $apply_protocol,
                        $k,
                        $v
                    ));
                    if ($db->getAffectRow() < 1)
                        throw new \PDOException('addHeader Error');
                }
            }
            if (!empty($params)) {
                foreach ($params as $k => $v) {
                    // 新建全局变量
                    $db->prepareExecute("INSERT INTO eo_api_env_param (eo_api_env_param.envID,eo_api_env_param.paramKey,eo_api_env_param.paramValue) VALUES (?,?,?);", array(
                        $env_id,
                        $k,
                        $v
                    ));
                    if ($db->getAffectRow() < 1)
                        throw new \PDOException('addParam Error');
                }
            }
            if (!empty($additional_params)) {
                foreach ($additional_params as $k => $v) {
                    //新建额外参数
                    $db->prepareExecute("INSERT INTO eo_api_env_param_additional(eo_api_env_param_additional.envID,eo_api_env_param_additional.paramKey,eo_api_env_param_additional.paramValue) VALUES (?,?,?);", array(
                        $env_id,
                        $k,
                        $v
                    ));
                    if ($db->getAffectRow() < 1)
                        throw new \PDOException('addAdditionalParam Error');
                }
            }
            $db->commit();
            return TRUE;
        } catch (\PDOException $e) {
            $db->rollback();
            return FALSE;
        }
    }

    /**
     * 获取环境信息
     * @param int $env_id 环境ID
     * @return bool|array
     */
    public function getEnvInfoFromDB(&$env_id)
    {
        $db = getDatabase();
        $env = $db->prepareExecute("SELECT eo_api_env.envID,eo_api_env.envName FROM eo_api_env WHERE eo_api_env.envID = ?;", array($env_id));
        $env['frontURIList'] = $db->prepareExecute("SELECT eo_api_env_front_uri.applyProtocol,eo_api_env_front_uri.uri FROM eo_api_env_front_uri WHERE eo_api_env_front_uri.envID = ?;", array($env_id));
        $env['headerList'] = $db->prepareExecuteAll("SELECT eo_api_env_header.applyProtocol,eo_api_env_header.headerName,eo_api_env_header.headerValue FROM eo_api_env_header WHERE eo_api_env_header.envID = ?;", array($env_id));
        $env['paramList'] = $db->prepareExecuteAll("SELECT eo_api_env_param.paramKey,eo_api_env_param.paramValue FROM eo_api_env_param WHERE eo_api_env_param.envID = ?;", array($env_id));
        $env['additionalParamList'] = $db->prepareExecuteAll('SELECT eo_api_env_param_additional.paramKey,eo_api_env_param_additional.paramValue FROM eo_api_env_param_additional WHERE eo_api_env_param_additional.envID = ?;', array($env_id));
        if ($env)
            return $env;
        else
            return FALSE;
    }

    /**
     * 获取环境名称
     * @param $envID int 环境ID
     * @return bool
     */
    public function getEnvName(&$envID)
    {
        $db = getDatabase();
        $result = $db->prepareExecute("SELECT eo_api_env.envName FROM eo_api_env WHERE eo_api_env.envID = ?;", array($envID));

        if (empty($result))
            return FALSE;
        else
            return $result['envName'];
    }

    /**
     * 检查项目环境权限
     * @param $envID int 环境ID
     * @param $userID int 用户ID
     * @return bool
     */
    public function checkEnvPermission(&$envID, &$userID)
    {
        $db = getDatabase();
        $result = $db->prepareExecute('SELECT eo_conn_project.projectID FROM eo_api_env LEFT JOIN eo_conn_project ON eo_api_env.projectID = eo_conn_project.projectID WHERE eo_api_env.envID = ? AND eo_conn_project.userID = ?;', array(
            $envID,
            $userID
        ));
        if (empty($result)) {
            return FALSE;
        } else {
            return $result['projectID'];
        }
    }
}