Files
ntrip-server/app/controller/home.js
2026-05-15 10:40:26 +08:00

513 lines
16 KiB
JavaScript

const { Controller } = require('egg');
const CryptoJS = require("crypto-js");
const KEY = CryptoJS.enc.Utf8.parse('BDQH');
const decryptRequest = (body) => {
return CryptoJS.AES.decrypt(body, KEY, {
mode: CryptoJS.mode.ECB,
padding: CryptoJS.pad.Pkcs7
}).toString(CryptoJS.enc.Utf8);
};
const checkUid = async (ctx) => {
const uid = ctx.request.body.uid;
const token = ctx.request.header.authorization;
const redis = ctx.app.redis;
let rdmcode;
if (!uid) {
const [buser, cuser, tuser] = await Promise.all([
redis.get('db0').get(token),
redis.get('db1').get(token),
redis.get('db5').get(token),
]);
rdmcode = buser || cuser || tuser;
} else {
let decryptUid = decryptRequest(uid).split('?')[0];
const redisType = {
"B": "db0",
"C": "db1",
"T": "db5",
};
const lastCode = decryptUid[decryptUid.length - 1];
rdmcode = await redis.get(redisType[lastCode]).get(token);
}
if (!rdmcode) {
return false;
}
return rdmcode;
};
const fileName = "home";
const setResponse = (ctx, code, msg, model, data = {}) => {
const modelType = {
"none": { code, msg },
"deconstruction": { code, msg, ...data },
"polymerization": { code, msg, data }
}
ctx.body = modelType[model];
}
class HomeController extends Controller {
// 保存挂载点
async saveMountpoint() {
const ctx = this.ctx;
try {
let rdmcode = await checkUid(ctx);
if (!rdmcode) {
setResponse(ctx, 401, 'token过期或错误', "none");
return;
}
const uid = JSON.parse(rdmcode).uid;
const body = ctx.request.body;
const result = await ctx.service.home.saveMountpoint(uid, body);
if (result == "error") {
setResponse(ctx, 500, '服务器错误', "none");
return;
}
if (result) {
setResponse(ctx, 200, '保存挂载点成功', "none");
} else {
setResponse(ctx, 404, '保存挂载点失败', "none");
}
} catch (error) {
const blockName = fileName + "--" + "saveMountpoint";
ctx.app.middleware.utils.log('../../../controllerLog', blockName + ">>>>" + error);
setResponse(ctx, 500, '服务器错误', "none");
}
}
// 修改挂载点密码
async editPassword() {
const ctx = this.ctx;
try {
let rdmcode = await checkUid(ctx);
if (!rdmcode) {
setResponse(ctx, 401, 'token过期或错误', "none");
return;
}
const uid = JSON.parse(rdmcode).uid;
const body = ctx.request.body;
const result = await ctx.service.home.editPassword(uid, body);
if (result == "error") {
setResponse(ctx, 500, '服务器错误', "none");
return;
}
if (result) {
setResponse(ctx, 200, '修改挂载点密码成功', "none");
} else {
setResponse(ctx, 404, '修改挂载点密码失败', "none");
}
} catch (error) {
const blockName = fileName + "--" + "editPassword";
ctx.app.middleware.utils.log('../../../controllerLog', blockName + ">>>>" + error);
setResponse(ctx, 500, '服务器错误', "none");
}
}
// 修改挂载点名称
async editMountpoint() {
const ctx = this.ctx;
try {
let rdmcode = await checkUid(ctx);
if (!rdmcode) {
setResponse(ctx, 401, 'token过期或错误', "none");
return;
}
const uid = JSON.parse(rdmcode).uid;
const body = ctx.request.body;
const result = await ctx.service.home.editMountpoint(uid, body);
if (result == "error") {
setResponse(ctx, 500, '服务器错误', "none");
return;
}
if (result) {
setResponse(ctx, 200, '修改挂载点名称成功', "none");
} else {
setResponse(ctx, 404, '修改挂载点名称失败', "none");
}
} catch (error) {
const blockName = fileName + "--" + "editMountpoint";
ctx.app.middleware.utils.log('../../../controllerLog', blockName + ">>>>" + error);
setResponse(ctx, 500, '服务器错误', "none");
}
}
// 删除挂载点
async delMountpoint() {
const ctx = this.ctx;
try {
// 验证用户token
let rdmcode = await checkUid(ctx);
if (!rdmcode) {
setResponse(ctx, 401, 'token过期或错误', "none");
return;
}
// 解析获取用户ID
const uid = JSON.parse(rdmcode).uid;
const body = ctx.request.body;
// 调用服务层删除挂载点
const result = await ctx.service.home.delMountpoint(uid, body);
if (result == "error") {
setResponse(ctx, 500, '服务器错误', "none");
return;
}
// 根据删除结果返回响应
if (result) {
setResponse(ctx, 200, '修改挂载点名称成功', "none");
} else {
setResponse(ctx, 404, '修改挂载点名称失败', "none");
}
} catch (error) {
// 记录异常日志
const blockName = fileName + "--" + "delMountpoint";
ctx.app.middleware.utils.log('../../../controllerLog', blockName + ">>>>" + error);
setResponse(ctx, 500, '服务器错误', "none");
}
}
// 获取挂载点
async getMountpoint() {
const ctx = this.ctx;
try {
let rdmcode = await checkUid(ctx);
if (!rdmcode) {
setResponse(ctx, 401, 'token过期或错误', "none");
return;
}
const uid = JSON.parse(rdmcode).uid;
const result = await ctx.service.home.getMountpoint(uid);
if (result == "error") {
setResponse(ctx, 500, '服务器错误', "none");
return;
}
if (result) {
const data = { mountpoint: result };
setResponse(ctx, 200, '获取挂载点成功', "deconstruction", data);
} else {
setResponse(ctx, 404, '获取挂载点失败', "none");
}
} catch (error) {
const blockName = fileName + "--" + "getMountpoint";
ctx.app.middleware.utils.log('../../../controllerLog', blockName + ">>>>" + error);
setResponse(ctx, 500, '服务器错误', "none");
}
}
// 搜索挂载点
async searchMountpoint() {
const ctx = this.ctx;
try {
let rdmcode = await checkUid(ctx);
if (!rdmcode) {
setResponse(ctx, 401, 'token过期或错误', "none");
return;
}
const uid = JSON.parse(rdmcode).uid;
const body = ctx.request.body;
const result = await ctx.service.home.searchMountpoint(uid, body);
if (result == "error") {
setResponse(ctx, 500, '服务器错误', "none");
return;
}
if (result) {
const data = { mountpoint: result.dataValues.mountpoint };
setResponse(ctx, 200, '搜索挂载点成功', "deconstruction", data);
} else {
setResponse(ctx, 203, '搜索挂载点失败', "none");
}
} catch (error) {
const blockName = fileName + "--" + "searchMountpoint";
ctx.app.middleware.utils.log('../../../controllerLog', blockName + ">>>>" + error);
setResponse(ctx, 500, '服务器错误', "none");
}
}
// 全局寻获挂载点
async findMountpoint() {
const ctx = this.ctx;
try {
const body = ctx.request.body;
const result = await ctx.service.home.findMountpoint(body);
if (result == "error") {
setResponse(ctx, 500, '服务器错误', "none");
return;
}
if (result) {
const data = {
mountpoint: result.mountpoint,
password: result.password,
status: result.status
};
setResponse(ctx, 200, '搜索挂载点成功', "deconstruction", data);
} else {
setResponse(ctx, 404, '搜索挂载点失败', "none");
}
} catch (error) {
const blockName = fileName + "--" + "findMountpoint";
ctx.app.middleware.utils.log('../../../controllerLog', blockName + ">>>>" + error);
setResponse(ctx, 500, '服务器错误', "none");
}
}
// 查看ntrip-clinet密码
async viewPassword() {
const ctx = this.ctx;
try {
let rdmcode = await checkUid(ctx);
if (!rdmcode) {
setResponse(ctx, 401, 'token过期或错误', "none");
return;
}
const uid = JSON.parse(rdmcode).uid;
const body = ctx.request.body;
const result = await ctx.service.home.viewPassword(uid, body);
if (result == "error") {
setResponse(ctx, 500, '服务器错误', "none");
return;
}
if (result) {
const data = { base: result.dataValues.base };
setResponse(ctx, 200, 'ntrip-clinet密码获取成功', "deconstruction", data);
} else {
setResponse(ctx, 404, 'ntrip-clinet密码获取失败', "none");
}
} catch (error) {
const blockName = fileName + "--" + "viewPassword";
ctx.app.middleware.utils.log('../../../controllerLog', blockName + ">>>>" + error);
setResponse(ctx, 500, '服务器错误', "none");
}
}
// 获取挂载点记录
async getRecord() {
const ctx = this.ctx;
try {
let rdmcode = await checkUid(ctx);
if (!rdmcode) {
setResponse(ctx, 401, 'token过期或错误', "none");
return;
}
const uid = JSON.parse(rdmcode).uid;
const body = ctx.request.body;
const result = await ctx.service.home.getRecord(uid, body);
if (result == "error") {
setResponse(ctx, 500, '服务器错误', "none");
return;
}
if (result) {
let recordMsg = [];
result.rows.forEach(element => {
recordMsg.push(element.dataValues);
});
const data = {
recordMsg: recordMsg,
count: result.count
};
setResponse(ctx, 200, '挂载点记录查询成功', "deconstruction", data);
} else {
setResponse(ctx, 404, '挂载点记录查询失败', "none");
}
} catch (error) {
const blockName = fileName + "--" + "getRecord";
ctx.app.middleware.utils.log('../../../controllerLog', blockName + ">>>>" + error);
setResponse(ctx, 500, '服务器错误', "none");
}
}
// 存储消息记录
async saveMsg() {
const ctx = this.ctx;
try {
let rdmcode = await checkUid(ctx);
if (!rdmcode) {
setResponse(ctx, 401, 'token过期或错误', "none");
return;
}
const uid = JSON.parse(rdmcode).uid;
const body = ctx.request.body;
const { mountpoint, original_msg } = body;
const form = {
mountpoint: mountpoint,
original_msg: original_msg,
created_at: new Date()
};
const result = await ctx.service.home.saveMsg(uid, form);
if (result == "error") {
setResponse(ctx, 500, '服务器错误', "none");
return;
}
if (result) {
setResponse(ctx, 200, '消息记录存储成功', "none");
} else {
setResponse(ctx, 404, '消息记录存储失败', "none");
}
} catch (error) {
const blockName = fileName + "--" + "saveMsg";
ctx.app.middleware.utils.log('../../../controllerLog', blockName + ">>>>" + error);
setResponse(ctx, 500, '服务器错误', "none");
}
}
// 设置ntrip-clinet密码
async setPassword() {
const ctx = this.ctx;
try {
let rdmcode = await checkUid(ctx);
if (!rdmcode) {
setResponse(ctx, 401, 'token过期或错误', "none");
return;
}
const uid = JSON.parse(rdmcode).uid;
const body = ctx.request.body;
const result = await ctx.service.home.setPassword(uid, body);
if (result == "error") {
setResponse(ctx, 500, '服务器错误', "none");
return;
}
if (result) {
setResponse(ctx, 200, 'ntrip-clinet密码设置成功', "none");
} else {
setResponse(ctx, 404, 'ntrip-clinet密码设置失败', "none");
}
} catch (error) {
const blockName = fileName + "--" + "setPassword";
ctx.app.middleware.utils.log('../../../controllerLog', blockName + ">>>>" + error);
setResponse(ctx, 500, '服务器错误', "none");
}
}
// 获取ntrip-clinet密码
async getPassword() {
const ctx = this.ctx;
try {
const body = ctx.request.body;
const result = await ctx.service.home.getPassword(body);
if (result == "error") {
setResponse(ctx, 500, '服务器错误', "none");
return;
}
if (result) {
const data = { password: result.dataValues.password };
setResponse(ctx, 200, 'ntrip-clinet密码获取成功', "deconstruction", data);
} else {
setResponse(ctx, 404, 'ntrip-clinet密码获取失败', "none");
}
} catch (error) {
const blockName = fileName + "--" + "getPassword";
ctx.app.middleware.utils.log('../../../controllerLog', blockName + ">>>>" + error);
setResponse(ctx, 500, '服务器错误', "none");
}
}
// 获取历史消息
async getHistoryMsg() {
const ctx = this.ctx;
try {
let rdmcode = await checkUid(ctx);
if (!rdmcode) {
setResponse(ctx, 401, 'token过期或错误', "none");
return;
}
const uid = JSON.parse(rdmcode).uid;
const body = ctx.request.body;
const result = await ctx.service.home.getHistoryMsg(uid, body);
if (result == "error") {
setResponse(ctx, 500, '服务器错误', "none");
return;
}
if (result) {
let historyMsg = [];
result.rows.forEach(element => {
historyMsg.push(element.dataValues);
});
const data = {
historyMsg: historyMsg,
count: result.count
};
setResponse(ctx, 200, '历史消息查找成功', "deconstruction", data);
} else {
setResponse(ctx, 404, '历史消息查找失败', "none");
}
} catch (error) {
const blockName = fileName + "--" + "getHistoryMsg";
ctx.app.middleware.utils.log('../../../controllerLog', blockName + ">>>>" + error);
setResponse(ctx, 500, '服务器错误', "none");
}
}
// 删除历史消息
async delHistoryMsg() {
const ctx = this.ctx;
try {
let rdmcode = await checkUid(ctx);
if (!rdmcode) {
setResponse(ctx, 401, 'token过期或错误', "none");
return;
}
const uid = JSON.parse(rdmcode).uid;
const body = ctx.request.body;
const model = decryptRequest(body.model);
const password = decryptRequest(body.password);
const requestMsg = body.userMsg;
const resetList = await ctx.service.home.delHistoryMsg(uid, model, password, requestMsg);
if (resetList == "error") {
setResponse(ctx, 500, '服务器错误', "none");
return;
}
if (resetList) {
setResponse(ctx, 200, '删除历史消息成功', "none");
} else {
setResponse(ctx, 404, '删除历史消息失败', "none");
}
} catch (error) {
const blockName = fileName + "--" + "delHistoryMsg";
ctx.app.middleware.utils.log('../../../controllerLog', blockName + ">>>>" + error);
setResponse(ctx, 500, '服务器错误', "none");
}
}
// 获取单位用户
async bsideManager() {
const ctx = this.ctx;
try {
let rdmcode = await checkUid(ctx);
if (!rdmcode) {
setResponse(ctx, 401, 'token过期或错误', "none");
return;
}
const uid = JSON.parse(rdmcode).uid;
let result;
if (uid[uid.length - 1] == 'B') {
const body = ctx.request.body;
result = await ctx.service.home.bsideManager(uid, body);
}
if (result == "error") {
setResponse(ctx, 500, '服务器错误', "none");
return;
}
if (result) {
let bsideUsers = [];
result.rows.forEach(element => {
bsideUsers.push(element.dataValues);
});
const data = {
bsideUsers: bsideUsers,
count: result.count
};
setResponse(ctx, 200, '获取单位用户成功', "deconstruction", data);
} else {
setResponse(ctx, 404, '获取单位用户失败', "none");
}
} catch (error) {
const blockName = fileName + "--" + "bsideManager";
ctx.app.middleware.utils.log('../../../controllerLog', blockName + ">>>>" + error);
setResponse(ctx, 500, '服务器错误', "none");
}
}
}
module.exports = HomeController;