|
- <?php
-
- namespace app\admin\command;
-
- use think\addons\AddonException;
- use think\addons\Service;
- use think\Config;
- use think\console\Command;
- use think\console\Input;
- use think\console\input\Option;
- use think\console\Output;
- use think\Db;
- use think\Exception;
- use think\exception\PDOException;
-
- class Addon extends Command
- {
-
- protected function configure()
- {
- $this
- ->setName('addon')
- ->addOption('name', 'a', Option::VALUE_REQUIRED, 'addon name', null)
- ->addOption('action', 'c', Option::VALUE_REQUIRED, 'action(create/enable/disable/install/uninstall/refresh/upgrade/package/move)', 'create')
- ->addOption('force', 'f', Option::VALUE_OPTIONAL, 'force override', null)
- ->addOption('release', 'r', Option::VALUE_OPTIONAL, 'addon release version', null)
- ->addOption('uid', 'u', Option::VALUE_OPTIONAL, 'fastadmin uid', null)
- ->addOption('token', 't', Option::VALUE_OPTIONAL, 'fastadmin token', null)
- ->addOption('local', 'l', Option::VALUE_OPTIONAL, 'local package', null)
- ->setDescription('Addon manager');
- }
-
- protected function execute(Input $input, Output $output)
- {
- $name = $input->getOption('name') ?: '';
- $action = $input->getOption('action') ?: '';
- if (stripos($name, 'addons' . DS) !== false) {
- $name = explode(DS, $name)[1];
- }
- //强制覆盖
- $force = $input->getOption('force');
- //版本
- $release = $input->getOption('release') ?: '';
- //uid
- $uid = $input->getOption('uid') ?: '';
- //token
- $token = $input->getOption('token') ?: '';
-
- include dirname(__DIR__) . DS . 'common.php';
-
- if (!$name) {
- throw new Exception('Addon name could not be empty');
- }
- if (!$action || !in_array($action, ['create', 'disable', 'enable', 'install', 'uninstall', 'refresh', 'upgrade', 'package', 'move'])) {
- throw new Exception('Please input correct action name');
- }
-
- // 查询一次SQL,判断连接是否正常
- Db::execute("SELECT 1");
-
- $addonDir = ADDON_PATH . $name . DS;
- switch ($action) {
- case 'create':
- //非覆盖模式时如果存在则报错
- if (is_dir($addonDir) && !$force) {
- throw new Exception("addon already exists!\nIf you need to create again, use the parameter --force=true ");
- }
- //如果存在先移除
- if (is_dir($addonDir)) {
- rmdirs($addonDir);
- }
- mkdir($addonDir, 0755, true);
- mkdir($addonDir . DS . 'controller', 0755, true);
- $menuList = \app\common\library\Menu::export($name);
- $createMenu = $this->getCreateMenu($menuList);
- $prefix = Config::get('database.prefix');
- $createTableSql = '';
- try {
- $result = Db::query("SHOW CREATE TABLE `" . $prefix . $name . "`;");
- if (isset($result[0]) && isset($result[0]['Create Table'])) {
- $createTableSql = $result[0]['Create Table'];
- }
- } catch (PDOException $e) {
-
- }
-
- $data = [
- 'name' => $name,
- 'addon' => $name,
- 'addonClassName' => ucfirst($name),
- 'addonInstallMenu' => $createMenu ? "\$menu = " . var_export_short($createMenu, "\t") . ";\n\tMenu::create(\$menu);" : '',
- 'addonUninstallMenu' => $menuList ? 'Menu::delete("' . $name . '");' : '',
- 'addonEnableMenu' => $menuList ? 'Menu::enable("' . $name . '");' : '',
- 'addonDisableMenu' => $menuList ? 'Menu::disable("' . $name . '");' : '',
- ];
- $this->writeToFile("addon", $data, $addonDir . ucfirst($name) . '.php');
- $this->writeToFile("config", $data, $addonDir . 'config.php');
- $this->writeToFile("info", $data, $addonDir . 'info.ini');
- $this->writeToFile("controller", $data, $addonDir . 'controller' . DS . 'Index.php');
- if ($createTableSql) {
- $createTableSql = str_replace("`" . $prefix, '`__PREFIX__', $createTableSql);
- file_put_contents($addonDir . 'install.sql', $createTableSql);
- }
-
- $output->info("Create Successed!");
- break;
- case 'disable':
- case 'enable':
- try {
- //调用启用、禁用的方法
- Service::$action($name, 0);
- } catch (AddonException $e) {
- if ($e->getCode() != -3) {
- throw new Exception($e->getMessage());
- }
- if (!$force) {
- //如果有冲突文件则提醒
- $data = $e->getData();
- foreach ($data['conflictlist'] as $k => $v) {
- $output->warning($v);
- }
- $output->info("Are you sure you want to " . ($action == 'enable' ? 'override' : 'delete') . " all those files? Type 'yes' to continue: ");
- $line = fgets(defined('STDIN') ? STDIN : fopen('php://stdin', 'r'));
- if (trim($line) != 'yes') {
- throw new Exception("Operation is aborted!");
- }
- }
- //调用启用、禁用的方法
- Service::$action($name, 1);
- } catch (Exception $e) {
- throw new Exception($e->getMessage());
- }
- $output->info(ucfirst($action) . " Successed!");
- break;
- case 'install':
- //非覆盖模式时如果存在则报错
- if (is_dir($addonDir) && !$force) {
- throw new Exception("addon already exists!\nIf you need to install again, use the parameter --force=true ");
- }
- //如果存在先移除
- if (is_dir($addonDir)) {
- rmdirs($addonDir);
- }
- // 获取本地路径
- $local = $input->getOption('local');
- try {
- Service::install($name, 0, ['version' => $release], $local);
- } catch (AddonException $e) {
- if ($e->getCode() != -3) {
- throw new Exception($e->getMessage());
- }
- if (!$force) {
- //如果有冲突文件则提醒
- $data = $e->getData();
- foreach ($data['conflictlist'] as $k => $v) {
- $output->warning($v);
- }
- $output->info("Are you sure you want to override all those files? Type 'yes' to continue: ");
- $line = fgets(defined('STDIN') ? STDIN : fopen('php://stdin', 'r'));
- if (trim($line) != 'yes') {
- throw new Exception("Operation is aborted!");
- }
- }
- Service::install($name, 1, ['version' => $release, 'uid' => $uid, 'token' => $token], $local);
- } catch (Exception $e) {
- throw new Exception($e->getMessage());
- }
-
- $output->info("Install Successed!");
- break;
- case 'uninstall':
- //非覆盖模式时如果存在则报错
- if (!$force) {
- throw new Exception("If you need to uninstall addon, use the parameter --force=true ");
- }
- try {
- Service::uninstall($name, 0);
- } catch (AddonException $e) {
- if ($e->getCode() != -3) {
- throw new Exception($e->getMessage());
- }
- if (!$force) {
- //如果有冲突文件则提醒
- $data = $e->getData();
- foreach ($data['conflictlist'] as $k => $v) {
- $output->warning($v);
- }
- $output->info("Are you sure you want to delete all those files? Type 'yes' to continue: ");
- $line = fgets(defined('STDIN') ? STDIN : fopen('php://stdin', 'r'));
- if (trim($line) != 'yes') {
- throw new Exception("Operation is aborted!");
- }
- }
- Service::uninstall($name, 1);
- } catch (Exception $e) {
- throw new Exception($e->getMessage());
- }
-
- $output->info("Uninstall Successed!");
- break;
- case 'refresh':
- Service::refresh();
- $output->info("Refresh Successed!");
- break;
- case 'upgrade':
- Service::upgrade($name, ['version' => $release, 'uid' => $uid, 'token' => $token]);
- $output->info("Upgrade Successed!");
- break;
- case 'package':
- $infoFile = $addonDir . 'info.ini';
- if (!is_file($infoFile)) {
- throw new Exception(__('Addon info file was not found'));
- }
-
- $info = get_addon_info($name);
- if (!$info) {
- throw new Exception(__('Addon info file data incorrect'));
- }
- $infoname = isset($info['name']) ? $info['name'] : '';
- if (!$infoname || !preg_match("/^[a-z]+$/i", $infoname) || $infoname != $name) {
- throw new Exception(__('Addon info name incorrect'));
- }
-
- $infoversion = isset($info['version']) ? $info['version'] : '';
- if (!$infoversion || !preg_match("/^\d+\.\d+\.\d+$/i", $infoversion)) {
- throw new Exception(__('Addon info version incorrect'));
- }
-
- $addonTmpDir = RUNTIME_PATH . 'addons' . DS;
- if (!is_dir($addonTmpDir)) {
- @mkdir($addonTmpDir, 0755, true);
- }
- $addonFile = $addonTmpDir . $infoname . '-' . $infoversion . '.zip';
- if (!class_exists('ZipArchive')) {
- throw new Exception(__('ZinArchive not install'));
- }
- $zip = new \ZipArchive;
- $zip->open($addonFile, \ZipArchive::CREATE | \ZipArchive::OVERWRITE);
-
- $files = new \RecursiveIteratorIterator(
- new \RecursiveDirectoryIterator($addonDir), \RecursiveIteratorIterator::LEAVES_ONLY
- );
-
- foreach ($files as $name => $file) {
- if (!$file->isDir()) {
- $filePath = $file->getRealPath();
- $relativePath = str_replace(DS, '/', substr($filePath, strlen($addonDir)));
- if (!in_array($file->getFilename(), ['.git', '.DS_Store', 'Thumbs.db'])) {
- $zip->addFile($filePath, $relativePath);
- }
- }
- }
- $zip->close();
- $output->info("Package Successed!");
- break;
- case 'move':
- $movePath = [
- 'adminOnlySelfDir' => ['admin/behavior', 'admin/controller', 'admin/library', 'admin/model', 'admin/validate', 'admin/view'],
- 'adminAllSubDir' => ['admin/lang'],
- 'publicDir' => ['public/assets/addons', 'public/assets/js/backend']
- ];
- $paths = [];
- $appPath = str_replace('/', DS, APP_PATH);
- $rootPath = str_replace('/', DS, ROOT_PATH);
- foreach ($movePath as $k => $items) {
- switch ($k) {
- case 'adminOnlySelfDir':
- foreach ($items as $v) {
- $v = str_replace('/', DS, $v);
- $oldPath = $appPath . $v . DS . $name;
- $newPath = $rootPath . "addons" . DS . $name . DS . "application" . DS . $v . DS . $name;
- $paths[$oldPath] = $newPath;
- }
- break;
- case 'adminAllSubDir':
- foreach ($items as $v) {
- $v = str_replace('/', DS, $v);
- $vPath = $appPath . $v;
- $list = scandir($vPath);
- foreach ($list as $_v) {
- if (!in_array($_v, ['.', '..']) && is_dir($vPath . DS . $_v)) {
- $oldPath = $appPath . $v . DS . $_v . DS . $name;
- $newPath = $rootPath . "addons" . DS . $name . DS . "application" . DS . $v . DS . $_v . DS . $name;
- $paths[$oldPath] = $newPath;
- }
- }
- }
- break;
- case 'publicDir':
- foreach ($items as $v) {
- $v = str_replace('/', DS, $v);
- $oldPath = $rootPath . $v . DS . $name;
- $newPath = $rootPath . 'addons' . DS . $name . DS . $v . DS . $name;
- $paths[$oldPath] = $newPath;
- }
- break;
- }
- }
- foreach ($paths as $oldPath => $newPath) {
- if (is_dir($oldPath)) {
- if ($force) {
- if (is_dir($newPath)) {
- $list = scandir($newPath);
- foreach ($list as $_v) {
- if (!in_array($_v, ['.', '..'])) {
- $file = $newPath . DS . $_v;
- @chmod($file, 0777);
- @unlink($file);
- }
- }
- @rmdir($newPath);
- }
- }
- copydirs($oldPath, $newPath);
- }
- }
- break;
- default:
- break;
- }
- }
-
- /**
- * 获取创建菜单的数组
- * @param array $menu
- * @return array
- */
- protected function getCreateMenu($menu)
- {
- $result = [];
- foreach ($menu as $k => & $v) {
- $arr = [
- 'name' => $v['name'],
- 'title' => $v['title'],
- ];
- if ($v['icon'] != 'fa fa-circle-o') {
- $arr['icon'] = $v['icon'];
- }
- if ($v['ismenu']) {
- $arr['ismenu'] = $v['ismenu'];
- }
- if (isset($v['childlist']) && $v['childlist']) {
- $arr['sublist'] = $this->getCreateMenu($v['childlist']);
- }
- $result[] = $arr;
- }
- return $result;
- }
-
- /**
- * 写入到文件
- * @param string $name
- * @param array $data
- * @param string $pathname
- * @return mixed
- */
- protected function writeToFile($name, $data, $pathname)
- {
- $search = $replace = [];
- foreach ($data as $k => $v) {
- $search[] = "{%{$k}%}";
- $replace[] = $v;
- }
- $stub = file_get_contents($this->getStub($name));
- $content = str_replace($search, $replace, $stub);
-
- if (!is_dir(dirname($pathname))) {
- mkdir(strtolower(dirname($pathname)), 0755, true);
- }
- return file_put_contents($pathname, $content);
- }
-
- /**
- * 获取基础模板
- * @param string $name
- * @return string
- */
- protected function getStub($name)
- {
- return __DIR__ . '/Addon/stubs/' . $name . '.stub';
- }
-
- }
|