154 lines
4.0 KiB
JavaScript
154 lines
4.0 KiB
JavaScript
const express = require("express");
|
|
const router = express.Router();
|
|
const db = require("../db/mysql");
|
|
const authMiddleware = require("../middleware/authMiddleware");
|
|
|
|
|
|
// 获取用户信息
|
|
router.get(
|
|
"/userInfo",
|
|
authMiddleware,
|
|
(req, res) => {
|
|
res.json({
|
|
code: 200,
|
|
data: {
|
|
username: req.user.username
|
|
}
|
|
});
|
|
|
|
}
|
|
);
|
|
|
|
// 获取联系人列表
|
|
router.get(
|
|
"/contactsInfo",
|
|
authMiddleware,
|
|
(req, res) => {
|
|
const sql = `
|
|
SELECT
|
|
u.username
|
|
FROM chat_mysql.contacts c
|
|
JOIN chat_mysql.users u
|
|
ON c.contact_name = u.username
|
|
WHERE c.user_name = ?;
|
|
`;
|
|
db.query(sql, [req.user.username], (error, results) => {
|
|
if (error) {
|
|
console.error("数据库查询失败：", error);
|
|
|
|
return res.status(500).json({
|
|
code: 500,
|
|
message: "查询失败",
|
|
});
|
|
}
|
|
if (results.length > 0) {
|
|
res.json({
|
|
code: 200,
|
|
message: "查询成功",
|
|
data: {
|
|
contacts: results.map(contact => ({
|
|
contactName: contact.username
|
|
}))
|
|
},
|
|
});
|
|
} else {
|
|
res.json({
|
|
code: 401,
|
|
message: "未找到联系人",
|
|
data: {
|
|
contacts: []
|
|
}
|
|
});
|
|
}
|
|
});
|
|
|
|
}
|
|
);
|
|
|
|
// 获取所有聊天消息
|
|
router.get(
|
|
"/messagesInfo",
|
|
authMiddleware,
|
|
(req, res) => {
|
|
const currentUserName = req.user.username;
|
|
const sql = `
|
|
SELECT
|
|
id,
|
|
sender_name,
|
|
receiver_name,
|
|
content,
|
|
created_at
|
|
FROM chat_mysql.messages
|
|
WHERE sender_name = ? OR receiver_name = ?
|
|
ORDER BY created_at ASC;
|
|
`;
|
|
db.query(sql, [currentUserName, currentUserName], (error, results) => {
|
|
if (error) {
|
|
console.error("数据库查询失败：", error);
|
|
|
|
return res.status(500).json({
|
|
code: 500,
|
|
message: "查询失败",
|
|
});
|
|
}
|
|
if (results.length > 0) {
|
|
res.json({
|
|
code: 200,
|
|
message: "查询成功",
|
|
data: {
|
|
messages: results.map(message => ({
|
|
messageId: message.id,
|
|
senderName: message.sender_name,
|
|
receiverName: message.receiver_name,
|
|
content: message.content,
|
|
createdAt: message.created_at
|
|
}))
|
|
},
|
|
});
|
|
} else {
|
|
res.json({
|
|
code: 401,
|
|
message: "未找到联系人消息",
|
|
data: {
|
|
messages: []
|
|
}
|
|
});
|
|
}
|
|
});
|
|
|
|
}
|
|
);
|
|
|
|
// 写入聊天消息
|
|
router.post("/sendMessage", (req, res) => {
|
|
const message = req.body;
|
|
const sql = `
|
|
INSERT INTO messages
|
|
(sender_name, receiver_name, content)
|
|
VALUES (?, ?, ?)
|
|
`;
|
|
|
|
db.query(sql, [
|
|
message.senderName,
|
|
message.receiverName,
|
|
message.content
|
|
], (error, result) => {
|
|
if (error) {
|
|
console.error(error);
|
|
|
|
return res.status(500).json({
|
|
code: 500,
|
|
message: "发送消息失败",
|
|
});
|
|
}
|
|
|
|
res.json({
|
|
code: 200,
|
|
message: "发送消息成功",
|
|
data: null
|
|
});
|
|
});
|
|
});
|
|
|
|
|
|
module.exports = router; |