100 lines
2.5 KiB
JavaScript
100 lines
2.5 KiB
JavaScript
#!/usr/bin/env node
|
||
|
||
import { config } from 'dotenv'
|
||
import path from 'path'
|
||
import { fileURLToPath } from 'url'
|
||
import { client, db, notificationSettings, systemSettings, users } from '../app/drizzle/db.ts'
|
||
import bcrypt from 'bcryptjs'
|
||
import { eq } from 'drizzle-orm'
|
||
|
||
const currentDir = path.dirname(fileURLToPath(import.meta.url))
|
||
config({ path: path.resolve(process.cwd(), '.env') })
|
||
|
||
if (!process.env.DATABASE_URL) {
|
||
console.error('❌ DATABASE_URL 环境变量未设置')
|
||
process.exit(1)
|
||
}
|
||
|
||
async function main() {
|
||
try {
|
||
// 检查是否已有超级管理员
|
||
const existingSuperAdmin = await db
|
||
.select()
|
||
.from(users)
|
||
.where(eq(users.role, 'SUPER_ADMIN'))
|
||
.limit(1)
|
||
|
||
if (existingSuperAdmin.length > 0) {
|
||
console.log('✅ 管理员已存在')
|
||
return existingSuperAdmin[0]
|
||
}
|
||
|
||
// 没有超级管理员但 username=admin 已被其他角色占用时跳过创建，避免产生重复 username
|
||
const existingAdminName = await db
|
||
.select()
|
||
.from(users)
|
||
.where(eq(users.username, 'admin'))
|
||
.limit(1)
|
||
|
||
if (existingAdminName.length > 0) {
|
||
console.warn(
|
||
`⚠️ username=admin 已被用户#${existingAdminName[0].id}（role=${existingAdminName[0].role}）占用，跳过创建管理员`
|
||
)
|
||
return existingAdminName[0]
|
||
}
|
||
|
||
// 创建管理员
|
||
const hashedPassword = await bcrypt.hash('admin123', 10)
|
||
const [admin] = await db
|
||
.insert(users)
|
||
.values({
|
||
username: 'admin',
|
||
name: '超级管理员',
|
||
password: hashedPassword,
|
||
role: 'SUPER_ADMIN',
|
||
forcePasswordChange: false
|
||
})
|
||
.returning()
|
||
|
||
// 创建通知设置
|
||
await db
|
||
.insert(notificationSettings)
|
||
.values({
|
||
userId: admin.id,
|
||
enabled: true,
|
||
songRequestEnabled: true,
|
||
songVotedEnabled: true,
|
||
songPlayedEnabled: true,
|
||
refreshInterval: 60,
|
||
songVotedThreshold: 1
|
||
})
|
||
.onConflictDoNothing()
|
||
|
||
// 创建系统设置
|
||
await db
|
||
.insert(systemSettings)
|
||
.values({
|
||
id: 1,
|
||
enablePlayTimeSelection: false
|
||
})
|
||
.onConflictDoNothing()
|
||
|
||
console.log('✅ 管理员创建成功 (admin/admin123)')
|
||
return admin
|
||
} catch (error) {
|
||
console.error('❌ 创建管理员失败:', error)
|
||
throw error
|
||
}
|
||
}
|
||
|
||
main()
|
||
.then(async () => {
|
||
await client.end()
|
||
process.exit(0)
|
||
})
|
||
.catch(async (error) => {
|
||
console.error('❌ 初始化失败:', error)
|
||
await client.end()
|
||
process.exit(1)
|
||
})
|