修改后端接口

This commit is contained in:
2025-09-25 17:43:54 +08:00
parent 5b6b7e0a96
commit 76b5393182
31 changed files with 2155 additions and 468 deletions

View File

@@ -1,51 +1,63 @@
const jwt = require('jsonwebtoken')
const jwt = require('jsonwebtoken');
const User = require('../models/User');
const AdminStaff = require('../models/AdminStaff');
const bcrypt = require('bcryptjs');
// JWT配置
const JWT_SECRET = process.env.JWT_SECRET || 'your-secret-key-change-in-production'
// 临时用户数据(实际项目中应该从数据库获取)
const users = [
{
id: 1,
username: 'admin',
password: '123456',
name: '系统管理员',
role: 'admin',
email: 'admin@example.com'
}
]
const JWT_SECRET = process.env.JWT_SECRET || 'your-secret-key-change-in-production';
exports.login = async (req, res) => {
try {
const { username, password } = req.body
const { username, password } = req.body;
// 查找用户
const user = users.find(u => u.username === username && u.password === password)
// 从数据库查找用户
const user = await User.findOne({
where: {
username,
status: 'active'
}
});
if (user) {
const token = jwt.sign({
id: user.id,
username: user.username,
role: user.role
}, JWT_SECRET, { expiresIn: '2h' })
return res.json({
code: 200,
message: '登录成功',
data: { token }
})
if (!user) {
return res.status(401).json({
code: 401,
message: '用户名或密码错误'
});
}
res.status(401).json({
code: 401,
message: '用户名或密码错误'
})
// 验证密码
const isPasswordValid = await bcrypt.compare(password, user.password);
if (!isPasswordValid) {
return res.status(401).json({
code: 401,
message: '用户名或密码错误'
});
}
// 更新最后登录时间
await user.update({
last_login: new Date()
});
const token = jwt.sign({
id: user.id,
username: user.username,
role: user.role
}, JWT_SECRET, { expiresIn: '2h' });
return res.json({
code: 200,
message: '登录成功',
data: { token }
});
} catch (err) {
console.error('登录错误:', err);
res.status(500).json({
code: 500,
message: '服务器错误',
error: err.message
})
});
}
}
@@ -53,51 +65,104 @@ exports.login = async (req, res) => {
exports.getUserInfo = async (req, res) => {
try {
// 从token中解析用户信息
const token = req.headers.authorization?.replace('Bearer ', '')
const token = req.headers.authorization?.replace('Bearer ', '');
if (!token) {
return res.status(401).json({
code: 401,
message: '未提供认证令牌'
})
});
}
try {
const decoded = jwt.verify(token, JWT_SECRET)
const user = users.find(u => u.id === decoded.id)
const decoded = jwt.verify(token, JWT_SECRET);
if (user) {
const userInfo = {
id: user.id,
username: user.username,
name: user.name,
role: user.role,
avatar: '',
email: user.email,
permissions: ['dashboard', 'users', 'settings']
}
return res.json({
code: 200,
message: '获取用户信息成功',
data: userInfo
})
} else {
// 从数据库获取用户信息
const user = await User.findByPk(decoded.id, {
include: [
{
model: AdminStaff,
as: 'staffInfo'
}
]
});
if (!user || user.status !== 'active') {
return res.status(401).json({
code: 401,
message: '用户不存在'
})
message: '用户不存在或已禁用'
});
}
// 获取员工信息
let staffInfo = null;
try {
staffInfo = await AdminStaff.findOne({
where: {
phone: user.username
}
});
} catch (error) {
console.warn('获取员工信息失败:', error);
}
// 根据角色设置权限
const permissions = getPermissionsByRole(user.role);
const userInfo = {
id: user.id,
username: user.username,
name: staffInfo?.name || user.username,
role: user.role,
avatar: '',
email: '',
phone: staffInfo?.phone || user.username,
department: staffInfo?.department_id ? {
id: staffInfo.department_id,
name: ''
} : null,
position: staffInfo?.position_id ? {
id: staffInfo.position_id,
name: ''
} : null,
permissions
};
return res.json({
code: 200,
message: '获取用户信息成功',
data: userInfo
});
} catch (jwtError) {
console.error('JWT验证错误:', jwtError);
return res.status(401).json({
code: 401,
message: '认证令牌无效'
})
});
}
} catch (err) {
console.error('获取用户信息错误:', err);
res.status(500).json({
code: 500,
message: '服务器错误',
error: err.message
})
});
}
}
};
// 根据角色获取权限
exports.getPermissionsByRole = (role) => {
const basePermissions = ['dashboard'];
switch (role) {
case 'admin':
return [...basePermissions, 'users', 'settings', 'supervision', 'approval', 'personnel', 'warehouse', 'epidemic', 'service', 'visualization'];
case 'manager':
return [...basePermissions, 'supervision', 'approval', 'personnel', 'warehouse', 'epidemic', 'service'];
case 'inspector':
return [...basePermissions, 'supervision', 'epidemic'];
case 'clerk':
return [...basePermissions, 'approval', 'personnel', 'warehouse', 'service'];
default:
return basePermissions;
}
};

View File

@@ -0,0 +1,248 @@
const Farmer = require('../models/Farmer');
const { Op } = require('sequelize');
// 获取养殖户列表
const getFarmers = async (req, res) => {
try {
const { page = 1, pageSize = 10, search = '', farmType = '', animalType = '', status = '' } = req.query;
const whereCondition = {};
// 搜索条件
if (search) {
whereCondition[Op.or] = [
{ nickname: { [Op.like]: `%${search}%` } },
{ real_name: { [Op.like]: `%${search}%` } },
{ farm_name: { [Op.like]: `%${search}%` } },
{ account: { [Op.like]: `%${search}%` } }
];
}
// 养殖场类型筛选
if (farmType) {
whereCondition.farm_type = farmType;
}
// 养殖种类筛选
if (animalType) {
whereCondition.animal_type = animalType;
}
// 状态筛选
if (status) {
whereCondition.status = status;
}
const offset = (page - 1) * pageSize;
const { count, rows } = await Farmer.findAndCountAll({
where: whereCondition,
offset,
limit: parseInt(pageSize),
order: [['register_time', 'DESC']]
});
// 格式化数据以便前端使用
const formattedData = rows.map(farmer => ({
id: farmer.id,
key: farmer.id.toString(),
account: farmer.account,
nickname: farmer.nickname,
real_name: farmer.real_name,
farm_name: farmer.farm_name,
farm_type: farmer.farm_type,
animal_type: farmer.animal_type,
animal_count: farmer.animal_count,
address: farmer.address,
register_time: farmer.register_time.toLocaleString('zh-CN'),
registrar: farmer.registrar,
status: farmer.status === 'active' ? '正常' : farmer.status === 'inactive' ? '暂停' : '关闭',
status_value: farmer.status
}));
res.json({
data: formattedData,
total: count,
page: parseInt(page),
pageSize: parseInt(pageSize)
});
} catch (error) {
console.error('获取养殖户列表失败:', error);
res.status(500).json({ message: error.message });
}
};
// 新增养殖户
const createFarmer = async (req, res) => {
try {
const {
account,
nickname,
real_name,
farm_name,
farm_type,
animal_type,
animal_count,
address,
status = 'active'
} = req.body;
// 检查账号是否已存在
const existingFarmer = await Farmer.findOne({ where: { account } });
if (existingFarmer) {
return res.status(400).json({ message: '该账号已存在' });
}
const farmer = await Farmer.create({
account,
nickname,
real_name,
farm_name,
farm_type,
animal_type,
animal_count,
address,
registrar: req.user?.username || 'system',
status
});
res.status(201).json({
success: true,
message: '养殖户创建成功',
data: farmer
});
} catch (error) {
console.error('创建养殖户失败:', error);
res.status(500).json({ message: error.message });
}
};
// 编辑养殖户
const updateFarmer = async (req, res) => {
try {
const { id } = req.params;
const {
nickname,
real_name,
farm_name,
farm_type,
animal_type,
animal_count,
address,
status
} = req.body;
const farmer = await Farmer.findByPk(id);
if (!farmer) {
return res.status(404).json({ message: '养殖户不存在' });
}
await farmer.update({
nickname,
real_name,
farm_name,
farm_type,
animal_type,
animal_count,
address,
status
});
res.json({
success: true,
message: '养殖户信息更新成功',
data: farmer
});
} catch (error) {
console.error('更新养殖户信息失败:', error);
res.status(500).json({ message: error.message });
}
};
// 删除养殖户
const deleteFarmer = async (req, res) => {
try {
const { id } = req.params;
const farmer = await Farmer.findByPk(id);
if (!farmer) {
return res.status(404).json({ message: '养殖户不存在' });
}
await farmer.destroy();
res.json({
success: true,
message: '养殖户删除成功'
});
} catch (error) {
console.error('删除养殖户失败:', error);
res.status(500).json({ message: error.message });
}
};
// 重置养殖户密码(如果需要)
const resetFarmerPassword = async (req, res) => {
try {
const { id } = req.params;
const farmer = await Farmer.findByPk(id);
if (!farmer) {
return res.status(404).json({ message: '养殖户不存在' });
}
// 在实际应用中,这里应该生成一个临时密码并发送给用户
// 为了演示,我们只返回成功信息
res.json({
success: true,
message: '密码重置成功,新密码已发送到用户手机'
});
} catch (error) {
console.error('重置养殖户密码失败:', error);
res.status(500).json({ message: error.message });
}
};
// 获取养殖类型列表
const getFarmTypes = async (req, res) => {
try {
const farmTypes = [
{ value: '规模', label: '规模' },
{ value: '散养', label: '散养' },
{ value: '其他', label: '其他' }
];
res.json(farmTypes);
} catch (error) {
console.error('获取养殖类型列表失败:', error);
res.status(500).json({ message: error.message });
}
};
// 获取养殖种类列表
const getAnimalTypes = async (req, res) => {
try {
const animalTypes = [
{ value: '牛', label: '牛' },
{ value: '羊', label: '羊' },
{ value: '猪', label: '猪' },
{ value: '鸡', label: '鸡' },
{ value: '其他', label: '其他' }
];
res.json(animalTypes);
} catch (error) {
console.error('获取养殖种类列表失败:', error);
res.status(500).json({ message: error.message });
}
};
module.exports = {
getFarmers,
createFarmer,
updateFarmer,
deleteFarmer,
resetFarmerPassword,
getFarmTypes,
getAnimalTypes
};