Files
EasySoft-ZenTaoPMS/module/gitlab/model.php
2021-06-21 11:16:37 +08:00

544 lines
16 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;
}
/**
* Create relationship between zentao product and gitlab project.
*
* @param int $gitlabID
* @param int $projectID
* @access public
* @return void
*/
public function createAssociat($products, $gitlabID, $gitlabProjectID)
{
$productIDs = $this->dao->select('id,program')->from(TABLE_PRODUCT)->fetchAll();
$projectID = array();
foreach($productIDs as $project) $projectID[$project->id] = $project->program;
$gitlabAssociat = new stdclass;
$gitlabAssociat->execution = 0;
$gitlabAssociat->AVersion = 0;
$gitlabAssociat->relation = 'interrated';
$gitlabAssociat->BVersion = 0;
$gitlabAssociat->extra = 0;
$gitlabAssociat->BID = $gitlabID;
foreach($products as $index => $prodcut)
{
$gitlabAssociat->BType = $this->getprojectpairs($gitlabID)[$gitlabProjectID];
$gitlabAssociat->Project = $projectID[$prodcut];
$gitlabAssociat->Product = $prodcut;
$this->dao->replace(TABLE_RELATION)->data($gitlabAssociat)->exec();
}
return true;
}
/**
* Create webhook for zentao.
*
* @param int $gitlabID
* @param int $projectID
* @access public
* @return void
*/
public function createWebhook($products, $gitlabID, $projectID)
{
$urls = $this->getWebhookUrls($gitlabID, $projectID);
foreach($products as $index => $product)
{
$url = sprintf($this->config->gitlab->zentaoApiWebhookUrl, commonModel::getSysURL(), $product, $gitlabID, $projectID);
if(! array_key_exists($url, array_flip($urls)))
{
$response = $this->apiCreateHook($gitlabID, $projectID, $url, $this->config->gitlab->zentaoApiWebhookToken);
}
}
return true;
}
/**
* 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 = json_decode(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 $response;
}
/**
* Get webhook urls
*
* @param int $gitlabID
* @param int $projectID
* @access public
* @return array $urls;
*/
public function getWebhookUrls($gitlabID, $projectID)
{
$urls = array();
$webhooks = $this->apiGetHooks($gitlabID, $projectID);
foreach($webhooks as $index => $webhook)
{
$urls[] = $webhook->url;
}
return $urls;
}
/**
* 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);
if(empty($labels)) return false;
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 = json_decode(commonModel::http($url, $issue));
return $response;
}
public function pushTask($gitlabID, $projectID, $task)
{
$task->label = $this->config->gitlab->taskLabel->name;
$response = $this->apiCreateIssue($gitlabID, $projectID, $task);
return $response;
}
public function pushBug($gitlabID, $projectID, $bug)
{
$bug->label = $this->config->gitlab->bugLabel->name;
$response = $this->apiCreateIssue($gitlabID, $projectID, $bug);
return $response;
}
public function parseWebhookBody($body)
{
$type = zget($body, 'object_kind', '');
if(!$type or !is_callable(array($this, "parse{$type}Webhook"))) return false;
return call_user_func_array(array($this, "parse{$type}Webhook"), array('body' => $body));
}
public function parseIssueWebhook($body)
{
$request = new stdclass;
$request->type = $body->object_kind;
$issue = $body->object_attributes;
$request->labels = $body->labels;
$request->project = $body->project->id;
}
}