Files
heros/build-templates/wechatgame/cloud_functions/cocos_cloud/modules/auth.js
2025-08-19 08:42:19 +08:00

284 lines
7.3 KiB
JavaScript

// 登录认证模块
const {
getNewUserInitData,
mergeUserDataWithDefaults,
checkDataVersionCompatibility
} = require('../user_init_data');
const user_db_name = "cocos_users";
/**
* 用户登录处理
* @param {Object} db 数据库实例
* @param {Object} wxContext 微信上下文
* @returns {Object} 登录结果
*/
async function login(db, wxContext) {
try {
let user = await getOrCreaterUser(db, wxContext.OPENID);
if (!user) {
return {
code: -4,
msg: "Failed to get or create user"
};
}
return {
code: 200,
data: {
user_id: user._id,
openid: user._openid,
regist_time: user.regist_time,
data: user.data,
fight_heros: user.fight_heros,
heros: user.heros,
items: user.items,
tals: user.tals,
equips: user.equips,
data_version: user.data_version,
last_save_time: user.last_save_time || null
},
msg: "Login successful"
};
} catch (error) {
console.error("Login error:", error);
return {
code: -5,
msg: `Login error: ${error.message}`
};
}
}
/**
* 获取或创建用户
* @param {Object} db 数据库实例
* @param {string} openid 用户openid
* @returns {Object} 用户数据
*/
async function getOrCreaterUser(db, openid) {
try {
let res = await db.collection(user_db_name).where({ _openid: openid }).get();
let userData = null;
if (res == null || res.data == null || res.data.length <= 0) {
// 创建新用户时使用初始化数据配置
let initData = getNewUserInitData();
userData = {
_openid: openid,
regist_time: Date.now(),
data: initData.data,
fight_heros: initData.fight_heros,
heros: initData.heros,
items: initData.items,
tals: initData.tals,
equips: initData.equips,
data_version: initData.data_version,
init_time: initData.init_time
};
let addResult = await db.collection(user_db_name).add({
data: userData
});
userData._id = addResult._id;
console.log(`New user created: ${openid}, version: ${initData.data_version}`);
} else {
userData = res.data[0];
// 检查数据版本兼容性
const versionCheck = checkDataVersionCompatibility(userData.data_version);
console.log(`User ${openid} data version check:`, versionCheck);
if (versionCheck.needsUpgrade) {
// 使用新的数据管理系统合并和升级数据
const upgradedData = mergeUserDataWithDefaults({
data: userData.data,
fight_heros: userData.fight_heros,
heros: userData.heros,
items: userData.items,
tals: userData.tals,
equips: userData.equips,
data_version: userData.data_version
});
// 更新用户数据
userData.data = upgradedData.data;
userData.fight_heros = upgradedData.fight_heros;
userData.heros = upgradedData.heros;
userData.items = upgradedData.items;
userData.tals = upgradedData.tals;
userData.equips = upgradedData.equips;
userData.data_version = upgradedData.data_version;
console.log(`User ${openid} data upgraded to version: ${upgradedData.data_version}`);
}
}
return userData;
} catch (err) {
console.error(`Get or create user err`, err);
return null;
}
}
/**
* 获取用户基本信息
* @param {Object} db 数据库实例
* @param {string} openid 用户openid
* @returns {Object} 用户基本信息
*/
async function getUserInfo(db, openid) {
try {
let user = await getOrCreaterUser(db, openid);
if (!user) {
return {
code: -4,
msg: "User not found"
};
}
return {
code: 200,
data: {
user_id: user._id,
openid: user._openid,
regist_time: user.regist_time,
init_time: user.init_time,
data_version: user.data_version,
last_save_time: user.last_save_time
},
msg: "User info retrieved successfully"
};
} catch (error) {
console.error("Get user info error:", error);
return {
code: -5,
msg: `Get user info error: ${error.message}`
};
}
}
/**
* 检查用户数据版本
* @param {Object} db 数据库实例
* @param {string} openid 用户openid
* @returns {Object} 版本检查结果
*/
async function checkVersion(db, openid) {
try {
let user = await getOrCreaterUser(db, openid);
if (!user) {
return {
code: -4,
msg: "User not found"
};
}
const versionCheck = checkDataVersionCompatibility(user.data_version);
return {
code: 200,
data: {
user_version: user.data_version || "unknown",
current_version: require('../user_init_data').DATA_VERSION,
compatibility: versionCheck,
init_time: user.init_time,
regist_time: user.regist_time,
last_save_time: user.last_save_time
},
msg: "Version information retrieved successfully"
};
} catch (error) {
console.error("Check version error:", error);
return {
code: -5,
msg: `Check version error: ${error.message}`
};
}
}
/**
* 强制升级用户数据
* @param {Object} db 数据库实例
* @param {string} openid 用户openid
* @returns {Object} 升级结果
*/
async function upgradeUserData(db, openid) {
try {
const _ = db.command;
let user = await getOrCreaterUser(db, openid);
if (!user) {
return {
code: -4,
msg: "User not found"
};
}
// 强制升级用户数据
const upgradedData = mergeUserDataWithDefaults({
data: user.data,
fight_heros: user.fight_heros,
heros: user.heros,
items: user.items,
tals: user.tals,
equips: user.equips,
data_version: user.data_version
});
let upgradeDataRes = await db.collection(user_db_name).doc(user._id).update({
data: {
data: _.set(upgradedData.data),
fight_heros: _.set(upgradedData.fight_heros),
heros: _.set(upgradedData.heros),
items: _.set(upgradedData.items),
tals: _.set(upgradedData.tals),
equips: _.set(upgradedData.equips),
data_version: _.set(upgradedData.data_version),
last_save_time: _.set(Date.now()),
upgrade_time: _.set(Date.now())
}
});
if (upgradeDataRes?.stats?.updated >= 1) {
return {
code: 200,
data: {
old_version: user.data_version || "unknown",
new_version: upgradedData.data_version,
upgrade_time: Date.now(),
data: upgradedData.data,
fight_heros: upgradedData.fight_heros,
heros: upgradedData.heros,
items: upgradedData.items,
tals: upgradedData.tals,
equips: upgradedData.equips
},
msg: "Data upgrade completed successfully"
};
} else {
return {
code: -1,
msg: `Upgrade fail, ${JSON.stringify(upgradeDataRes)}`
};
}
} catch (error) {
console.error("Upgrade user data error:", error);
return {
code: -5,
msg: `Upgrade error: ${error.message}`
};
}
}
module.exports = {
login,
getOrCreaterUser,
getUserInfo,
checkVersion,
upgradeUserData
};