99 lines
3.6 KiB
PHP
99 lines
3.6 KiB
PHP
<?php
|
|
|
|
/**
|
|
* 邀请码与对应后台权限迁移（可重复执行）。
|
|
* 用法: php database/migrate_invitation_codes.php
|
|
*/
|
|
|
|
require __DIR__ . '/../vendor/autoload.php';
|
|
|
|
$app = new think\App();
|
|
$app->initialize();
|
|
|
|
use app\service\PermissionCatalog;
|
|
use think\facade\Db;
|
|
|
|
function invitationTableExists(string $table): bool
|
|
{
|
|
$db = Db::getConfig('database') ?: env('DB_NAME', 'ai_chat');
|
|
$rows = Db::query(
|
|
'SELECT TABLE_NAME FROM information_schema.TABLES WHERE TABLE_SCHEMA = ? AND TABLE_NAME = ? LIMIT 1',
|
|
[$db, $table]
|
|
);
|
|
return !empty($rows);
|
|
}
|
|
|
|
try {
|
|
if (!invitationTableExists('invitation_codes')) {
|
|
Db::execute("CREATE TABLE invitation_codes (
|
|
id BIGINT UNSIGNED AUTO_INCREMENT PRIMARY KEY,
|
|
code VARCHAR(32) NOT NULL UNIQUE,
|
|
department_id INT UNSIGNED NULL,
|
|
created_by INT UNSIGNED NULL,
|
|
used_by INT UNSIGNED NULL,
|
|
status ENUM('active','used','revoked') NOT NULL DEFAULT 'active',
|
|
expires_at TIMESTAMP NULL,
|
|
used_at TIMESTAMP NULL,
|
|
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
|
|
updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP,
|
|
INDEX idx_invitation_status (status),
|
|
INDEX idx_invitation_department (department_id),
|
|
INDEX idx_invitation_created (created_at)
|
|
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_unicode_ci");
|
|
echo "Created table invitation_codes\n";
|
|
}
|
|
|
|
if (invitationTableExists('sys_permissions')) {
|
|
$parent = Db::name('sys_permissions')->where('code', 'dir:org')->find();
|
|
if ($parent) {
|
|
$menu = Db::name('sys_permissions')->where('code', 'menu:invitations')->find();
|
|
if (!$menu) {
|
|
$menuId = Db::name('sys_permissions')->insertGetId([
|
|
'type' => 'menu',
|
|
'code' => 'menu:invitations',
|
|
'name' => '邀请码管理',
|
|
'parent_id' => (int) $parent['id'],
|
|
'path' => '/invitations',
|
|
'icon' => 'ticket',
|
|
'sort_order' => 25,
|
|
'is_system' => 1,
|
|
'created_at' => date('Y-m-d H:i:s'),
|
|
'updated_at' => date('Y-m-d H:i:s'),
|
|
]);
|
|
$menu = ['id' => $menuId];
|
|
echo "Added menu:invitations\n";
|
|
}
|
|
|
|
foreach ([
|
|
['btn:invitation:create', '生成邀请码'],
|
|
['btn:invitation:revoke', '作废邀请码'],
|
|
] as [$code, $name]) {
|
|
if (!Db::name('sys_permissions')->where('code', $code)->find()) {
|
|
Db::name('sys_permissions')->insert([
|
|
'type' => 'btn',
|
|
'code' => $code,
|
|
'name' => $name,
|
|
'parent_id' => (int) $menu['id'],
|
|
'sort_order' => 0,
|
|
'is_system' => 1,
|
|
'created_at' => date('Y-m-d H:i:s'),
|
|
'updated_at' => date('Y-m-d H:i:s'),
|
|
]);
|
|
echo "Added {$code}\n";
|
|
}
|
|
}
|
|
}
|
|
|
|
if (invitationTableExists('roles')) {
|
|
Db::name('roles')->where('slug', 'super_admin')->update([
|
|
'permissions' => json_encode(PermissionCatalog::fullPermissions(), JSON_UNESCAPED_UNICODE),
|
|
]);
|
|
}
|
|
}
|
|
|
|
echo "Invitation migration completed.\n";
|
|
} catch (\Throwable $e) {
|
|
echo 'Error: ' . $e->getMessage() . "\n";
|
|
exit(1);
|
|
}
|