Files
2026-06-05 17:38:10 +08:00

203 lines
7.1 KiB
JavaScript

'use strict';
const Controller = require('../core/base_controller');
const fs = require('fs');
const fileName = "BSide";
class BSideController extends Controller {
// 注册B端用户
async addBSide() {
const { ctx } = this;
try {
let rdmcode = await this.checkUid(ctx);
if (!rdmcode) {
this.setResponse(ctx, 401, 'token过期或错误', "none");
return;
}
const requestMsg = ctx.request.body.requestMsg;
requestMsg.forEach(element => {
const salt = this.generateRandomString();
element.uid = this.generateNumberString('B');
element.salt = this.encryptRequest(salt);
element.status = 0;
element.role = this.decryptRequest(element.role);
element.account_name = this.decryptRequest(element.account_name);
element.password = this.shaForm('123456', salt);
element.organization = element.organization == "" ? null : this.decryptRequest(element.organization);
element.user_name = element.user_name == "" ? "" : this.decryptRequest(element.user_name);
element.phone = element.phone == "" ? "" : this.decryptRequest(element.phone);
element.msg = element.msg == "" ? "" : this.decryptRequest(element.msg);
element.created_at = new Date();
element.updated_at = new Date();
element.login_at = new Date();
element.login_status = 0;
});
const result = await ctx.service.bSide.addBSide(requestMsg);
if (result == "error") {
this.setResponse(ctx, 500, '服务器错误', "none");
return;
}
if (result) {
this.setResponse(ctx, 200, '创建B端新用户成功', "none");
} else {
this.setResponse(ctx, 404, '创建B端新用户失败', "none");
}
} catch (error) {
const blockName = fileName + "--" + "addBSide";
ctx.app.middleware.log('../../controllerLog', blockName + ">>>>" + error);
this.setResponse(ctx, 500, '服务器错误', "none");
}
}
// 查看B端用户
async showBSide() {
const { ctx } = this;
try {
let rdmcode = await this.checkUid(ctx);
if (!rdmcode) {
this.setResponse(ctx, 401, 'token过期或错误', "none");
return;
}
const body = ctx.request.body;
const result = await ctx.service.bSide.showBSide(body);
if (result == "error") {
this.setResponse(ctx, 500, '服务器错误', "none");
return;
}
if (result) {
let BSideMsg = [];
result.rows.forEach(element => {
BSideMsg.push(element.dataValues);
});
const data = {
BSideMsg: BSideMsg,
count: result.count
};
this.setResponse(ctx, 200, '搜索B端用户成功', "deconstruction", data);
} else {
this.setResponse(ctx, 404, '搜索B端用户失败', "none");
}
} catch (error) {
const blockName = fileName + "--" + "showBSide";
ctx.app.middleware.log('../../controllerLog', blockName + ">>>>" + error);
this.setResponse(ctx, 500, '服务器错误', "none");
}
}
// 编辑B端用户
async editBSide() {
const { ctx } = this;
try {
let rdmcode = await this.checkUid(ctx);
if (!rdmcode) {
this.setResponse(ctx, 401, 'token过期或错误', "none");
return;
}
const uid = JSON.parse(rdmcode).uid;
const body = ctx.request.body;
const { model, updateMsg, requestMsg } = body
const modelType = this.decryptRequest(model);
const result = await ctx.service.bSide.editBSide(uid, modelType, updateMsg, requestMsg);
if (result == "error") {
this.setResponse(ctx, 500, '服务器错误', "none");
return;
}
if (result) {
this.setResponse(ctx, 200, '编辑B端用户信息成功', "none");
} else {
this.setResponse(ctx, 404, '编辑B端用户信息失败', "none");
}
} catch (error) {
const blockName = fileName + "--" + "editBSide";
ctx.app.middleware.log('../../controllerLog', blockName + ">>>>" + error);
this.setResponse(ctx, 500, '服务器错误', "none");
}
}
// 删除B端用户
async deleteBSide() {
const { ctx } = this;
try {
let rdmcode = await this.checkUid(ctx);
if (!rdmcode) {
this.setResponse(ctx, 401, 'token过期或错误', "none");
return;
}
const uid = JSON.parse(rdmcode).uid;
const body = ctx.request.body;
const { model, password, requestMsg } = body
const modelType = this.decryptRequest(model);
const deCodePassword = this.decryptRequest(password);
const result = await ctx.service.bSide.deleteBSide(uid, modelType, deCodePassword, requestMsg);
if (result == "error") {
this.setResponse(ctx, 500, '服务器错误', "none");
return;
}
if (result) {
this.setResponse(ctx, 200, '删除B端用户成功', "none");
} else {
this.setResponse(ctx, 404, '删除B端用户失败', "none");
}
} catch (error) {
const blockName = fileName + "--" + "deleteBSide";
ctx.app.middleware.log('../../controllerLog', blockName + ">>>>" + error);
this.setResponse(ctx, 500, '服务器错误', "none");
}
}
// 管理系统导出B端用户
async downloadBside() {
const { ctx } = this;
try {
let rdmcode = await this.checkUid(ctx);
if (!rdmcode) {
this.setResponse(ctx, 401, 'token过期或错误', "none");
return;
}
const uid = JSON.parse(rdmcode).uid;
const { form, searchMsg, dataHeaders } = ctx.request.body;
const result = await ctx.service.bSide.downloadBside(uid, searchMsg);
if (result == "error") {
this.setResponse(ctx, 500, '服务器错误', "none");
return;
}
if (result.code) {
const file = this.downloadData(form, result.data, dataHeaders);
if (file) {
const { filePath, completeName, buf } = file;
try {
fs.writeFileSync(filePath, Buffer.from(buf));
const data = {
// 本机服务器地址
// 'http://127.0.0.1:8012'
// 'http://192.168.31.28:8012'
// 公司服务器地址
// 'http://123.57.205.46:8012'
// 'http://manager.bdrd.hwasmart.com//accountManager'
fileUrl: 'http://127.0.0.1:8012' + `/public/${completeName}?time=${new Date().getTime()}`,
fileName: completeName,
}
this.setResponse(ctx, 200, '文件已成功写入', "deconstruction", data);
setTimeout(() => {
fs.unlinkSync(filePath);
}, 600000);
} catch (error) {
this.setResponse(ctx, 500, '写入文件时发生错误', "none");
}
} else {
this.setResponse(ctx, 404, '操作失败', "none");
}
} else {
this.setResponse(ctx, 404, '操作失败', "none");
}
} catch (error) {
const blockName = fileName + "--" + "downloadBside";
ctx.app.middleware.log('../../controllerLog', blockName + ">>>>" + error);
this.setResponse(ctx, 500, '服务器错误', "none");
}
}
}
module.exports = BSideController;