68 lines
1.9 KiB
JavaScript
68 lines
1.9 KiB
JavaScript
|
let db = null;
|
||
|
|
exports.addCorpDefaultInfo = async (context, mongodb) => {
|
||
|
|
let { corpid } = context;
|
||
|
|
db = mongodb;
|
||
|
|
// 获取添加默认角色
|
||
|
|
await addDefaultRole(corpid);
|
||
|
|
// 获取添加默认模板
|
||
|
|
await addDefaultTemplate(corpid);
|
||
|
|
return "存储成功";
|
||
|
|
};
|
||
|
|
|
||
|
|
async function addDefaultRole(corpId) {
|
||
|
|
let res = await db.collection("default-administrator-role").find().toArray();
|
||
|
|
const roleList = res;
|
||
|
|
const promiseArr = roleList.map((item) => {
|
||
|
|
return defaultRoles(item.roleId, corpId);
|
||
|
|
});
|
||
|
|
await Promise.all(promiseArr);
|
||
|
|
}
|
||
|
|
|
||
|
|
async function defaultRoles(roleId, corpId) {
|
||
|
|
const count = await db
|
||
|
|
.collection("sys-role")
|
||
|
|
.countDocuments({ corpId, roleId });
|
||
|
|
if (count === 0) {
|
||
|
|
const res = await db
|
||
|
|
.collection("default-administrator-role")
|
||
|
|
.findOne({ roleId });
|
||
|
|
const { _id, ...rest } = res;
|
||
|
|
rest.corpId = corpId;
|
||
|
|
await db.collection("sys-role").insertOne(rest);
|
||
|
|
} else {
|
||
|
|
return {};
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
async function addDefaultTemplate(corpId) {
|
||
|
|
// 获取平台档案模版
|
||
|
|
const currentEnvApp = envConfig.getCurrentEnvApp();
|
||
|
|
const client = new MongoClient(currentEnvApp.databaseUri, {
|
||
|
|
useNewUrlParser: true,
|
||
|
|
useUnifiedTopology: true,
|
||
|
|
});
|
||
|
|
await client.connect();
|
||
|
|
const currentEnvDb = client.db(currentEnvApp.databaseName);
|
||
|
|
|
||
|
|
let res = await currentEnvDb
|
||
|
|
.collection("platform-template")
|
||
|
|
.find({ useType: "public" })
|
||
|
|
.toArray();
|
||
|
|
const templateList = res;
|
||
|
|
const promiseArr = templateList.map((item) => {
|
||
|
|
return defaultTemplate(item, corpId);
|
||
|
|
});
|
||
|
|
await Promise.all(promiseArr);
|
||
|
|
}
|
||
|
|
|
||
|
|
async function defaultTemplate(item, corpId) {
|
||
|
|
const count = await db
|
||
|
|
.collection("corp-template")
|
||
|
|
.countDocuments({ corpId, templateType: item.templateType });
|
||
|
|
if (count === 0) {
|
||
|
|
const { _id, ...rest } = item;
|
||
|
|
rest.corpId = corpId;
|
||
|
|
await db.collection("corp-template").insertOne(rest);
|
||
|
|
}
|
||
|
|
}
|