Files
EasySoft-ZenTaoPMS/module/gitlab/model.php
2021-06-18 09:38:20 +08:00

451 lines
13 KiB
PHP

<?php
/**
* The model file of gitlab module of ZenTaoPMS.
*
* @copyright Copyright 2009-2015 青岛易软天创网络科技有限公司(QingDao Nature Easy Soft Network Technology Co,LTD, www.cnezsoft.com)
* @license ZPL (http://zpl.pub/page/zplv12.html)
* @author Chenqi <chenqi@cnezsoft.com>
* @package product
* @version $Id: $
* @link http://www.zentao.net
*/
class gitlabModel extends model
{
/**
* Get a gitlab by id.
*
* @param int $id
* @access public
* @return object
*/
public function getByID($id)
{
return $this->loadModel('pipeline')->getByID($id);
}
/**
* Get gitlab list.
*
* @param string $orderBy
* @param object $pager
* @access public
* @return array
*/
public function getList($orderBy = 'id_desc', $pager = null)
{
return $this->loadModel('pipeline')->getList('gitlab', $orderBy, $pager);
}
/**
* Get gitlab pairs
*
* @return array
*/
public function getPairs()
{
return $this->loadModel('pipeline')->getPairs('gitlab');
}
/**
* Create a gitlab.
*
* @access public
* @return bool
*/
public function create()
{
return $this->loadModel('pipeline')->create('gitlab');
}
/**
* Update a gitlab.
*
* @param int $id
* @access public
* @return bool
*/
public function update($id)
{
return $this->loadModel('pipeline')->update($id);
}
/**
* Get current user.
*
* @param string $host
* @param string $token
* @access public
* @return array
*/
public function apiGetCurrentUser($host, $token)
{
$api = rtrim($host, '/') . "/api/v4/user?private_token=$token";
return json_decode(commonModel::http($api));
}
/**
* Get gitlab user list.
*
* @param string $host
* @param string $token
* @access public
* @return array
*/
public function apiGetUsers($gitlab)
{
$api = rtrim($gitlab->url, '/') . '/api/v4/users?private_token=' . $gitlab->token;
$response = json_decode(commonModel::http($api));
if (!$response) return array();
$users = array();
foreach($response as $gitlabUser)
{
$user = new stdclass;
$user->id = $gitlabUser->id;
$user->realname = $gitlabUser->name;
$user->account = $gitlabUser->username;
$user->email = $gitlabUser->email;
$user->avatar = $gitlabUser->avatar_url;
$users[] = $user;
}
return $users;
}
/**
* Get matched gitlab users.
*
* @param array $gitlabUsers
* @param array $zentaoUsers
* @access public
* @return array
*/
public function getMatchedUsers($gitlabID, $gitlabUsers, $zentaoUsers)
{
$matches = new stdclass;
foreach($gitlabUsers as $gitlabUser)
{
foreach($zentaoUsers as $zentaoUser)
{
if($gitlabUser->account == $zentaoUser->account) $matches->accounts[$gitlabUser->account][] = $zentaoUser->account;
if($gitlabUser->realname == $zentaoUser->realname) $matches->names[$gitlabUser->realname][] = $zentaoUser->account;
if($gitlabUser->email == $zentaoUser->email) $matches->emails[$gitlabUser->email][] = $zentaoUser->account;
}
}
$bindedUsers = $this->dao->select('openID,account')->from(TABLE_OAUTH)
->where('providerType')->eq('gitlab')
->andWhere('providerID')->eq($gitlabID)
->fetchPairs();
$matchedUsers = array();
foreach($gitlabUsers as $gitlabUser)
{
if(isset($bindedUsers[$gitlabUser->id]))
{
$gitlabUser->zentaoAccount = $bindedUsers[$gitlabUser->id];
$matchedUsers[] = $gitlabUser;
continue;
}
$matchedZentaoUsers = array();
if(isset($matches->accounts[$gitlabUser->account])) $matchedZentaoUsers = array_merge($matchedZentaoUsers, $matches->accounts[$gitlabUser->account]);
if(isset($matches->emails[$gitlabUser->email])) $matchedZentaoUsers = array_merge($matchedZentaoUsers, $matches->emails[$gitlabUser->email]);
if(isset($matches->names[$gitlabUser->realname])) $matchedZentaoUsers = array_merge($matchedZentaoUsers, $matches->names[$gitlabUser->realname]);
$matchedZentaoUsers = array_unique($matchedZentaoUsers);
if(count($matchedZentaoUsers) == 1)
{
$gitlabUser->zentaoAccount = current($matchedZentaoUsers);
$matchedUsers[] = $gitlabUser;
}
}
return $matchedUsers;
}
/**
* Get projects of one gitlab.
*
* @param int $gitlabID
* @access public
* @return void
*/
public function apiGetProjects($gitlabID)
{
$gitlab = $this->getByID($gitlabID);
if(!$gitlab) return array();
$host = rtrim($gitlab->url, '/');
$host .= '/api/v4/projects';
$allResults = array();
for($page = 1; true; $page ++)
{
$results = json_decode(commonModel::http($host . "?private_token={$gitlab->token}&simple=true&membership=true&page={$page}&per_page=100"));
if(empty($results) or $page > 10) break;
$allResults = $allResults + $results;
}
return $allResults;
}
public function getProjectPairs($gitlabID)
{
$projects = $this->apiGetProjects($gitlabID);
$projectPairs = array();
foreach($projects as $project)
{
$projectPairs[$project->id] = $project->name_with_namespace;
}
return $projectPairs;
}
public function getProjectDisplayName($gitlabID, $projectID)
{
return array_key_exists($gitlabID, $projectID) ? $this->gitlab->getProjectPairs($gitlabID)[$projectID]: "";
}
/**
* Get gitlab api base url with access_token
*
* @param int $id
* @access public
* @return string gitlab api base url with access_token
*/
public function getApiRoot($id)
{
$gitlab = $this->getByID($id);
if(!$gitlab) return "";
$gitlab_url = rtrim($gitlab->url, '/').'/api/v4%s'."?private_token={$gitlab->token}";
return $gitlab_url;
}
/**
* Get hooks.
*
* @param int $gitlabID
* @param int $projectID
* @access public
* @return void
*/
public function apiGetHooks($gitlabID, $projectID)
{
$apiRoot = $this->getApiRoot($gitlabID);
$apiPath = "/projects/{$projectID}/hooks";
$url = sprintf($apiRoot, $apiPath);
$response = commonModel::http($url);
return $response;
}
/**
* Get specific hook.
*
* @param int $gitlabID
* @param int $projectID
* @param int $hookID
* @access public
* @return void
*/
public function apiGetHook($gitlabID, $projectID, $hookID)
{
$apiRoot = $this->getApiRoot($gitlabID);
$apiPath = "/projects/$projectID/hooks/$hookID)";
$url = sprintf($apiRoot, $apiPath);
$response = commonModel::http($url);
return;
}
/**
* Create hook.
*
* @param int $gitlabID
* @param int $projectID
* @param int $url
* @param int $token
* @access public
* @return void
*/
public function apiCreateHook($gitlabID, $projectID, $url, $token)
{
$apiRoot = $this->getApiRoot($gitlabID);
$postData = new stdclass;
$postData->enable_ssl_verification = "false";
$postData->issues_events = "true";
$postData->merge_requests_events = "true";
$postData->push_events = "true";
$postData->tag_push_events = "true";
$postData->url = $url;
$postData->token = $token;
$apiPath = "/projects/{$projectID}/hooks";
$url = sprintf($apiRoot, $apiPath);
$response = commonModel::http($url, $postData);
return $response;
}
/**
* Delete hook.
*
* @param int $gitlabID
* @param int $projectID
* @param int $hookID
* @access public
* @return void
*/
public function apiDeleteHook($gitlabID, $projectID, $hookID)
{
$apiRoot = $this->getApiRoot($gitlabID);
$apiPath = "/projects/{$projectID}/hooks/{$hookID}";
$url = sprintf($apiRoot, $apiPath);
$response = commonModel::http($url, null, array(CURLOPT_CUSTOMREQUEST => 'delete'));
return $response;
}
/**
* Update hook.
*
* @param int $gitlabID
* @param int $projectID
* @param int $hookID
* @access public
* @return void
*/
public function apiUpdateHook($gitlabID, $projectID, $hookID)
{
$apiRoot = $this->getApiRoot($gitlabID);
$postData = new stdclass;
$postData->enable_ssl_verification = "false";
$postData->issues_events = "true";
$postData->merge_requests_events = "true";
$postData->push_events = "true";
$postData->tag_push_events = "true";
$postData->url = $url;
$postData->token = $token;
$apiPath = "/projects/{$projectID}/hooks/{$hookID}";
$url = sprintf($apiRoot, $apiPath);
$response = commonModel::http($url, $postData, $options = array(CURLOPT_CUSTOMREQUEST => 'put'));
return $response;
}
/**
* Create Label for gitlab project.
*
* @param int $gitlabID
* @param int $projectID
* @param int $label
* @access public
* @return void
*/
public function apiCreateLabel($gitlabID, $projectID, $label)
{
$apiRoot = $this->getApiRoot($gitlabID);
if(empty($label->name) or empty($label->color)) return false;
$apiPath = "/projects/{$projectID}/labels/";
$url = sprintf($apiRoot, $apiPath);
$response = commonModel::http($url, $label);
return $response;
}
/**
* Get labels of project.
*
* @param int $gitlabID
* @param int $projectID
* @access public
* @return void
*/
public function apiGetLabels($gitlabID, $projectID)
{
$apiRoot = $this->getApiRoot($gitlabID);
$apiPath = "/projects/{$projectID}/labels/";
$url = sprintf($apiRoot, $apiPath);
$response = commonModel::http($url);
$labels = json_decode($response);
return $labels;
}
/**
* Check if predefined label exist in project.
*
* @param int $gitlabID
* @param int $projectID
* @access public
* @return void
*/
public function isLabelExists($gitlabID, $projectID)
{
$labels = $this->apiGetLabels($gitlabID, $projectID);
foreach($labels as $label)
{
if(strpos($label->name, $this->config->gitlab->taskLabel->name) == 0) return true;
if(strpos($label->name, $this->config->gitlab->bugLabel->name) == 0) return true;
}
return false;
}
/**
* Create predefined labels for project.
*
* @param int $gitlabID
* @param int $projectID
* @access public
* @return void
*/
public function initLabels($gitlabID, $projectID)
{
if($this->isLabelExists($gitlabID, $projectID)) return true;
$taskLabel = new stdclass();
$taskLabel->name = $this->config->gitlab->taskLabel->name;
$taskLabel->description = $this->config->gitlab->taskLabel->description;
$taskLabel->color = $this->config->gitlab->taskLabel->color;
$taskLabel->priority = $this->config->gitlab->taskLabel->priority;
$bugLabel = new stdclass();
$bugLabel->name = $this->config->gitlab->bugLabel->name;
$bugLabel->description = $this->config->gitlab->bugLabel->description;
$bugLabel->color = $this->config->gitlab->bugLabel->color;
$bugLabel->priority = $this->config->gitlab->bugLabel->priority;
$this->apiCreateLabel($gitlabID, $projectID, $taskLabel);
$this->apiCreateLabel($gitlabID, $projectID, $bugLabel);
return;
}
public function apiCreateIssue($gitlabID, $projectID, $issue)
{
$apiRoot = $this->getApiRoot($gitlabID);
$apiPath = "/projects/{$projectID}/issues/";
$url = sprintf($apiRoot, $apiPath);
$response = commonModel::http($url, $issue);
return $response;
}
public function pushTask($task, $gitlabID,$projectID)
{
$task->label = $this->config->gitlab->taskLabel->name;
}
public function pushBug($bug, $gitlabID,$projectID)
{
$bug->label = $this->config->gitlab->bugLabel->name;
}
}