2023-05-16 10:47:08 +08:00

391 lines
14 KiB
PHP

<?php
/**
* The control file of webhook module of ZenTaoPMS.
*
* @copyright Copyright 2009-2017 禅道软件(青岛)有限公司(ZenTao Software (Qingdao) Co., Ltd. www.cnezsoft.com)
* @license ZPL(http://zpl.pub/page/zplv12.html) or AGPL(https://www.gnu.org/licenses/agpl-3.0.en.html)
* @author Gang Liu <liugang@cnezsoft.com>
* @package webhook
* @version $Id$
* @link http://www.zentao.net
*/
class webhook extends control
{
/**
* Construct
*
* @param string $moduleName
* @param string $methodName
* @access public
* @return void
*/
public function __construct($moduleName = '', $methodName = '')
{
parent::__construct($moduleName, $methodName);
$this->loadModel('message');
}
/**
* Browse webhooks.
*
* @param string $orderBy
* @param int $recTotal
* @param int $recPerPage
* @param int $pageID
* @access public
* @return void
*/
public function browse($orderBy = 'id_desc', $recTotal = 0, $recPerPage = 20, $pageID = 1)
{
$this->app->loadClass('pager', $static = true);
$pager = new pager($recTotal, $recPerPage, $pageID);
/* Unset selectedDepts cookie. */
setcookie('selectedDepts', '', 0, $this->config->webRoot, '', $this->config->cookieSecure, true);
$this->view->title = $this->lang->webhook->api . $this->lang->colon . $this->lang->webhook->list;
$this->view->webhooks = $this->webhook->getList($orderBy, $pager);
$this->view->position[] = html::a(inlink('browse'), $this->lang->webhook->api);
$this->view->position[] = $this->lang->webhook->common;
$this->view->orderBy = $orderBy;
$this->view->pager = $pager;
$this->display();
}
/**
* Create a webhook.
*
* @access public
* @return void
*/
public function create()
{
if($_POST)
{
$webhookID = $this->webhook->create();
if(dao::isError()) return $this->send(array('result' => 'fail', 'message' => dao::getError()));
if($this->viewType == 'json') return $this->send(array('result' => 'success', 'message' => $this->lang->saveSuccess, 'id' => $webhookID));
return $this->send(array('result' => 'success', 'message' => $this->lang->saveSuccess, 'locate' => inlink('browse')));
}
unset($this->lang->webhook->typeList['']);
$this->app->loadLang('action');
$this->view->title = $this->lang->webhook->api . $this->lang->colon . $this->lang->webhook->create;
$this->view->products = $this->loadModel('product')->getPairs();
$this->view->executions = $this->loadModel('execution')->getPairs();
$this->view->position[] = html::a(inlink('browse'), $this->lang->webhook->api);
$this->view->position[] = html::a(inlink('browse'), $this->lang->webhook->common);
$this->view->position[] = $this->lang->webhook->create;
$this->display();
}
/**
* Edit a webhook.
*
* @param int $id
* @access public
* @return void
*/
public function edit($id)
{
if($_POST)
{
$this->webhook->update($id);
if(dao::isError()) return $this->send(array('result' => 'fail', 'message' => dao::getError()));
return $this->send(array('result' => 'success', 'message' => $this->lang->saveSuccess, 'locate' => inlink('browse')));
}
$webhook = $this->webhook->getByID($id);
$this->app->loadLang('action');
$this->view->title = $this->lang->webhook->edit . $this->lang->colon . $webhook->name;
$this->view->position[] = html::a(inlink('browse'), $this->lang->webhook->api);
$this->view->position[] = html::a(inlink('browse'), $this->lang->webhook->common);
$this->view->position[] = $this->lang->webhook->edit;
$this->view->products = $this->loadModel('product')->getPairs();
$this->view->executions = $this->loadModel('execution')->getPairs();
$this->view->webhook = $webhook;
$this->display();
}
/**
* Delete a webhook.
*
* @param int $id
* @access public
* @return void
*/
public function delete($id)
{
$this->webhook->delete(TABLE_WEBHOOK, $id);
if(dao::isError()) return $this->send(array('result' => 'fail', 'message' => dao::getError()));
$this->send(array('result' => 'success'));
}
/**
* Browse logs of a webhook.
*
* @param int $id
* @param string $orderBy
* @param int $recTotal
* @param int $recPerPage
* @param int $pageID
* @access public
* @return void
*/
public function log($id, $orderBy = 'id_desc', $recTotal = 0, $recPerPage = 20, $pageID = 1)
{
/* Save session. */
$uri = $this->app->getURI(true);
$this->session->set('productList', $uri, 'product');
$this->session->set('productPlanList', $uri, 'product');
$this->session->set('releaseList', $uri, 'product');
$this->session->set('storyList', $uri, 'product');
$this->session->set('executionList', $uri, 'execution');
$this->session->set('taskList', $uri, 'execution');
$this->session->set('buildList', $uri, 'execution');
$this->session->set('bugList', $uri, 'qa');
$this->session->set('caseList', $uri, 'qa');
$this->session->set('testtaskList', $uri, 'qa');
$this->session->set('todoList', $uri, 'my');
$this->session->set('docList', $uri, 'doc');
$this->app->loadClass('pager', $static = true);
$pager = new pager($recTotal, $recPerPage, $pageID);
$webhook = $this->webhook->getByID($id);
$this->view->title = $this->lang->webhook->log . $this->lang->colon . $webhook->name;
$this->view->logs = $this->webhook->getLogList($id, $orderBy, $pager);
$this->view->position[] = html::a(inlink('browse'), $this->lang->webhook->api);
$this->view->position[] = html::a(inlink('browse'), $this->lang->webhook->common);
$this->view->position[] = $this->lang->webhook->log;
$this->view->webhook = $webhook;
$this->view->orderBy = $orderBy;
$this->view->pager = $pager;
$this->display();
}
/**
* Bind dingtalk userid.
*
* @param int $id
* @param int $recTotal
* @param int $recPerPage
* @param int $pageID
* @access public
* @return void
*/
public function bind($id, $recTotal = 0, $recPerPage = 15, $pageID = 1)
{
if($_POST)
{
$this->webhook->bind($id);
if(dao::isError()) return print(js::error(dao::getError()));
return print(js::reload('parent'));
}
$webhook = $this->webhook->getById($id);
if($webhook->type != 'dinguser' && $webhook->type != 'wechatuser' && $webhook->type != 'feishuuser')
{
echo js::alert($this->lang->webhook->note->bind);
return print(js::locate($this->createLink('webhook', 'browse')));
}
$webhook->secret = json_decode($webhook->secret);
/* Get selected depts. */
if($this->get->selectedDepts)
{
setcookie('selectedDepts', $this->get->selectedDepts, 0, $this->config->webRoot, '', $this->config->cookieSecure, true);
$_COOKIE['selectedDepts'] = $this->get->selectedDepts;
}
$selectedDepts = $this->cookie->selectedDepts ? $this->cookie->selectedDepts : '';
if($webhook->type == 'dinguser')
{
$this->app->loadClass('dingapi', true);
$dingapi = new dingapi($webhook->secret->appKey, $webhook->secret->appSecret, $webhook->secret->agentId);
$response = $dingapi->getUsers($selectedDepts);
}
elseif($webhook->type == 'wechatuser')
{
$this->app->loadClass('wechatapi', true);
$wechatApi = new wechatapi($webhook->secret->appKey, $webhook->secret->appSecret, $webhook->secret->agentId);
$response = $wechatApi->getAllUsers();
}
elseif($webhook->type == 'feishuuser')
{
$this->app->loadClass('feishuapi', true);
$feishuApi = new feishuapi($webhook->secret->appId, $webhook->secret->appSecret);
$response = $feishuApi->getAllUsers($selectedDepts);
}
if($response['result'] == 'fail')
{
if($response['message'] == 'nodept')
{
echo js::error($this->lang->webhook->error->noDept);
return print(js::locate($this->createLink('webhook', 'chooseDept', "id=$id")));
}
echo js::error($response['message']);
return print(js::locate($this->createLink('webhook', 'browse')));
}
$oauthUsers = $response['data'];
$bindedPairs = $this->webhook->getBoundUsers($id);
$useridPairs = array('' => '');
foreach($oauthUsers as $name => $userid) $useridPairs[$userid] = $name;
$this->app->loadClass('pager', $static = true);
$pager = new pager($recTotal, $recPerPage, $pageID);
$users = $this->loadModel('user')->getByQuery('inside', $query = '', $pager);
$unbindUsers = array();
$bindedUsers = array();
foreach($users as $user)
{
if(isset($bindedPairs[$user->account])) $bindedUsers[$user->account] = $user;
if(!isset($bindedPairs[$user->account])) $unbindUsers[$user->account] = $user;
}
$users = $unbindUsers + $bindedUsers;
$this->view->title = $this->lang->webhook->bind;
$this->view->position[] = html::a($this->createLink('webhook', 'browse'), $this->lang->webhook->common);
$this->view->position[] = $this->lang->webhook->bind;
$this->view->webhook = $webhook;
$this->view->oauthUsers = $oauthUsers;
$this->view->useridPairs = $useridPairs;
$this->view->users = $users;
$this->view->pager = $pager;
$this->view->bindedUsers = $bindedPairs;
$this->view->selectedDepts = $selectedDepts;
$this->display();
}
/**
* choose dept.
*
* @param int $id
* @access public
* @return void
*/
public function chooseDept($id)
{
$webhook = $this->webhook->getById($id);
if($webhook->type != 'dinguser' && $webhook->type != 'wechatuser' && $webhook->type != 'feishuuser')
{
echo js::alert($this->lang->webhook->note->bind);
return print(js::locate($this->createLink('webhook', 'browse')));
}
$webhook->secret = json_decode($webhook->secret);
if($webhook->type == 'dinguser')
{
$this->app->loadClass('dingapi', true);
$dingapi = new dingapi($webhook->secret->appKey, $webhook->secret->appSecret, $webhook->secret->agentId);
$response = $dingapi->getDeptTree();
}
if($webhook->type == 'feishuuser') $response = array('result' => 'success', 'data' => array());
if($response['result'] == 'fail')
{
echo js::error($response['message']);
return print(js::locate($this->createLink('webhook', 'browse')));
}
if($response['result'] == 'selected')
{
$locateLink = $this->createLink('webhook', 'bind', "id={$id}");
$locateLink .= strpos($locateLink, '?') !== false ? '&' : '?';
$locateLink .= 'selectedDepts=' . join(',', $response['data']);
return print(js::locate($locateLink));
}
$this->view->title = $this->lang->webhook->chooseDept;
$this->view->position[] = $this->lang->webhook->chooseDept;
$this->view->webhookType = $webhook->type;
$this->view->deptTree = $response['data'];
$this->view->webhookID = $id;
$this->display();
}
public function ajaxGetFeishuDeptList($webhookID)
{
$webhook = $this->webhook->getById($webhookID);
$webhook->secret = json_decode($webhook->secret);
if($_POST)
{
$this->app->loadClass('feishuapi', true);
$feishuApi = new feishuapi($webhook->secret->appId, $webhook->secret->appSecret);
$departmentID = $_POST['departmentID'] ? $_POST['departmentID'] : '';
$depts = $feishuApi->getChildDeptTree($departmentID);
echo json_encode($depts, true);
}
else
{
$this->app->loadClass('feishuapi', true);
$feishuApi = new feishuapi($webhook->secret->appId, $webhook->secret->appSecret);
$depts = $feishuApi->getDeptTree();
echo json_encode($depts, true);
}
}
/**
* Send data by async.
*
* @access public
* @return void
*/
public function asyncSend()
{
$webhooks = $this->webhook->getList($orderBy = 'id_desc', $pager = null, $decode = false);
if(empty($webhooks))
{
echo "NO WEBHOOK EXIST.\n";
return false;
}
$dataList = $this->webhook->getDataList();
if(empty($dataList))
{
echo "OK\n";
return true;
}
$this->webhook->setSentStatus(array_keys($dataList), 'senting');
$now = helper::now();
$diff = 0;
foreach($dataList as $data)
{
$webhook = zget($webhooks, $data->objectID, '');
if($webhook)
{
/* if connect time is out then ignore it.*/
if($diff < 29)
{
$time = time();
$result = $this->webhook->fetchHook($webhook, $data->data, $data->action);
$diff = time() - $time;
}
$this->webhook->saveLog($webhook, $data->action, $data->data, $result);
}
$this->webhook->setSentStatus($data->id, 'sended', $now);
}
$this->dao->delete()->from(TABLE_NOTIFY)->where('status')->eq('sended')->exec();
echo "OK\n";
return true;
}
}