56 lines
1.2 KiB
JavaScript
56 lines
1.2 KiB
JavaScript
|
let db = null;
|
||
|
|
exports.main = async (event, mongodb) => {
|
||
|
|
db = mongodb;
|
||
|
|
switch (event.type) {
|
||
|
|
case "getSuiteToken":
|
||
|
|
return await getSuiteToken();
|
||
|
|
case "updateSuiteToken":
|
||
|
|
return await updateSuiteToken(event);
|
||
|
|
}
|
||
|
|
};
|
||
|
|
// 获取 SuiteToken
|
||
|
|
async function getSuiteToken() {
|
||
|
|
try {
|
||
|
|
const res = await db
|
||
|
|
.collection("weComToken")
|
||
|
|
.findOne({ type: "suiteToken" });
|
||
|
|
if (!res) {
|
||
|
|
return {
|
||
|
|
success: false,
|
||
|
|
message: "SuiteToken 不存在",
|
||
|
|
};
|
||
|
|
}
|
||
|
|
return {
|
||
|
|
success: true,
|
||
|
|
suiteToken: res.suite_token,
|
||
|
|
message: "获取成功",
|
||
|
|
};
|
||
|
|
} catch (error) {
|
||
|
|
return {
|
||
|
|
success: false,
|
||
|
|
message: `获取失败: ${error.message}`,
|
||
|
|
};
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// 更新 SuiteToken
|
||
|
|
async function updateSuiteToken({ suiteToken }) {
|
||
|
|
try {
|
||
|
|
const result = await db.collection("weComToken").updateOne(
|
||
|
|
{ type: "suiteToken" },
|
||
|
|
{ $set: { suite_token: suiteToken } },
|
||
|
|
{ upsert: true } // 如果不存在，则插入
|
||
|
|
);
|
||
|
|
return {
|
||
|
|
success: true,
|
||
|
|
message: "更新成功",
|
||
|
|
result,
|
||
|
|
};
|
||
|
|
} catch (error) {
|
||
|
|
return {
|
||
|
|
success: false,
|
||
|
|
message: `更新失败: ${error.message}`,
|
||
|
|
};
|
||
|
|
}
|
||
|
|
}
|