import mongoose from 'mongoose'
import User from './model/user.js'

const connectDB = async () => {
    try {
        const conn = await mongoose.connect(process.env.MONGO_URI)
        console.log(`MongoDB Connected: ${conn.connection.host}`)
        // 检查是否存在管理员用户，如果不存在则创建一个默认管理员
        const adminUser = await User.findOne({ username: process.env.ADMIN_USERNAME })
        if (!adminUser) {
            await User.create({
                username: process.env.ADMIN_USERNAME,
                nickname: '管理员',
                password: process.env.ADMIN_PASSWORD,
                role: 'admin',
            })
        }
    } catch (error) {
        console.error('Error connecting to MongoDB:', error)
        process.exit(1)
    }
}

export default connectDB