Miao-Yunzai/plugins/genshin/model/db/UserDB.js

101 lines
1.9 KiB
JavaScript
Raw Blame History

This file contains ambiguous Unicode characters

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

import BaseModel from './BaseModel.js'
import lodash from 'lodash'
import { UserGameDB } from './index.js'
import MysUtil from '../mys/MysUtil.js'
import MysUserDB from './MysUserDB.js'
const { Types } = BaseModel
const COLUMNS = {
// 用户IDqq为数字
id: {
type: Types.STRING,
autoIncrement: false,
primaryKey: true
},
type: {
type: Types.STRING,
defaultValue: 'qq',
notNull: true
},
// 昵称
name: Types.STRING,
// 头像
face: Types.STRING,
ltuids: Types.STRING,
games: {
type: Types.STRING,
get () {
let data = this.getDataValue('games')
let ret = {}
try {
data = JSON.parse(data) || {}
} catch (e) {
data = {}
}
MysUtil.eachGame((game) => {
let ds = data[game] || {}
ret[game] = {
uid: ds.uid || '',
data: ds.data || {}
}
})
return ret
},
set (data) {
this.setDataValue('games', JSON.stringify(data))
}
},
data: Types.STRING
}
class UserDB extends BaseModel {
static async find (id, type = 'qq') {
// user_id
id = type === 'qq' ? '' + id : type + id
// DB查询
let user = await UserDB.findByPk(id)
if (!user) {
user = await UserDB.build({
id,
type
})
}
return user
}
async saveDB (user) {
let db = this
let ltuids = []
lodash.forEach(user.mysUsers, (mys) => {
if (mys.ck && mys.ltuid) {
ltuids.push(mys.ltuid)
}
})
db.ltuids = ltuids.join(',')
let games = {}
lodash.forEach(user._games, (gameDs, game) => {
games[game] = {
uid: gameDs.uid,
data: {}
}
lodash.forEach(gameDs.data, (ds, uid) => {
games[game].data[uid] = {
uid: ds.uid,
type: ds.type
}
})
})
db.games = games
await this.save()
}
}
BaseModel.initDB(UserDB, COLUMNS)
await UserDB.sync()
export default UserDB