This commit is contained in:
Vendored
+130
@@ -0,0 +1,130 @@
|
||||
"use strict";
|
||||
Object.defineProperty(exports, "__esModule", { value: true });
|
||||
const express_1 = require("express");
|
||||
const asyncHandler_1 = require("../utils/asyncHandler");
|
||||
const db_1 = require("../db");
|
||||
const schema_1 = require("../db/schema");
|
||||
const drizzle_orm_1 = require("drizzle-orm");
|
||||
const auth_1 = require("../utils/auth");
|
||||
const auth_2 = require("../utils/auth");
|
||||
const errorHandler_1 = require("../middleware/errorHandler");
|
||||
const validation_1 = require("../utils/validation");
|
||||
const router = (0, express_1.Router)();
|
||||
router.use(auth_2.authMiddleware);
|
||||
async function findUsernames(ids) {
|
||||
if (ids.length === 0)
|
||||
return new Map();
|
||||
const rows = await db_1.db.select().from(schema_1.users).where((0, drizzle_orm_1.or)(...ids.map((id) => (0, drizzle_orm_1.eq)(schema_1.users.id, id))));
|
||||
return new Map(rows.map((u) => [u.id, u.username]));
|
||||
}
|
||||
router.get('/', (0, asyncHandler_1.asyncHandler)(async (req, res) => {
|
||||
const userId = req.user.userId;
|
||||
const outgoingRows = await db_1.db
|
||||
.select()
|
||||
.from(schema_1.friendships)
|
||||
.where((0, drizzle_orm_1.and)((0, drizzle_orm_1.eq)(schema_1.friendships.userId, userId), (0, drizzle_orm_1.eq)(schema_1.friendships.status, 'pending')));
|
||||
const incomingRows = await db_1.db
|
||||
.select()
|
||||
.from(schema_1.friendships)
|
||||
.where((0, drizzle_orm_1.and)((0, drizzle_orm_1.eq)(schema_1.friendships.friendId, userId), (0, drizzle_orm_1.eq)(schema_1.friendships.status, 'pending')));
|
||||
const acceptedRows = await db_1.db
|
||||
.select()
|
||||
.from(schema_1.friendships)
|
||||
.where((0, drizzle_orm_1.and)((0, drizzle_orm_1.or)((0, drizzle_orm_1.eq)(schema_1.friendships.userId, userId), (0, drizzle_orm_1.eq)(schema_1.friendships.friendId, userId)), (0, drizzle_orm_1.eq)(schema_1.friendships.status, 'accepted')));
|
||||
const friendIds = acceptedRows.map((r) => (r.userId === userId ? r.friendId : r.userId));
|
||||
const outgoingIds = outgoingRows.map((r) => r.friendId);
|
||||
const incomingIds = incomingRows.map((r) => r.userId);
|
||||
const usernames = await findUsernames([...friendIds, ...outgoingIds, ...incomingIds]);
|
||||
const friends = acceptedRows.map((r) => {
|
||||
const friendId = r.userId === userId ? r.friendId : r.userId;
|
||||
return { id: friendId, username: usernames.get(friendId) ?? '' };
|
||||
});
|
||||
const outgoing = outgoingRows.map((r) => ({
|
||||
id: r.id,
|
||||
username: usernames.get(r.friendId) ?? '',
|
||||
requestId: r.id,
|
||||
status: 'pending',
|
||||
}));
|
||||
const incoming = incomingRows.map((r) => ({
|
||||
id: r.userId,
|
||||
username: usernames.get(r.userId) ?? '',
|
||||
requestId: r.id,
|
||||
status: 'pending',
|
||||
}));
|
||||
res.json({ friends, incoming, outgoing });
|
||||
}));
|
||||
router.post('/requests', (0, asyncHandler_1.asyncHandler)(async (req, res) => {
|
||||
const { username } = validation_1.friendRequestSchema.parse(req.body);
|
||||
const userId = req.user.userId;
|
||||
if (username.toLowerCase() === (await meUsername(userId))) {
|
||||
throw new errorHandler_1.AppError('SELF_REQUEST', 'You cannot add yourself', 400);
|
||||
}
|
||||
const target = await db_1.db.select().from(schema_1.users).where((0, drizzle_orm_1.eq)(schema_1.users.username, username)).limit(1);
|
||||
if (target.length === 0) {
|
||||
throw new errorHandler_1.AppError('USER_NOT_FOUND', 'No user with that username found', 404);
|
||||
}
|
||||
const existing = await db_1.db
|
||||
.select()
|
||||
.from(schema_1.friendships)
|
||||
.where((0, drizzle_orm_1.or)((0, drizzle_orm_1.and)((0, drizzle_orm_1.eq)(schema_1.friendships.userId, userId), (0, drizzle_orm_1.eq)(schema_1.friendships.friendId, target[0].id)), (0, drizzle_orm_1.and)((0, drizzle_orm_1.eq)(schema_1.friendships.userId, target[0].id), (0, drizzle_orm_1.eq)(schema_1.friendships.friendId, userId))))
|
||||
.limit(1);
|
||||
if (existing.length > 0) {
|
||||
throw new errorHandler_1.AppError('ALREADY_FRIENDS', existing[0].status === 'accepted' ? 'You are already friends' : 'Friend request already pending', 409);
|
||||
}
|
||||
const now = Date.now();
|
||||
await db_1.db.insert(schema_1.friendships).values({
|
||||
id: (0, auth_1.generateId)('friend'),
|
||||
userId,
|
||||
friendId: target[0].id,
|
||||
status: 'pending',
|
||||
createdAt: now,
|
||||
updatedAt: now,
|
||||
});
|
||||
res.status(201).json({ success: true });
|
||||
}));
|
||||
router.post('/requests/:id/accept', (0, asyncHandler_1.asyncHandler)(async (req, res) => {
|
||||
const userId = req.user.userId;
|
||||
const requestId = req.params.id;
|
||||
const row = await db_1.db.select().from(schema_1.friendships).where((0, drizzle_orm_1.eq)(schema_1.friendships.id, requestId)).limit(1);
|
||||
if (row.length === 0) {
|
||||
throw new errorHandler_1.AppError('NOT_FOUND', 'Request not found', 404);
|
||||
}
|
||||
if (row[0].friendId !== userId) {
|
||||
throw new errorHandler_1.AppError('FORBIDDEN', 'This request was not sent to you', 403);
|
||||
}
|
||||
if (row[0].status !== 'pending') {
|
||||
throw new errorHandler_1.AppError('INVALID_STATE', 'Request is no longer pending', 409);
|
||||
}
|
||||
await db_1.db
|
||||
.update(schema_1.friendships)
|
||||
.set({ status: 'accepted', updatedAt: Date.now() })
|
||||
.where((0, drizzle_orm_1.eq)(schema_1.friendships.id, requestId));
|
||||
res.json({ success: true });
|
||||
}));
|
||||
router.delete('/requests/:id', (0, asyncHandler_1.asyncHandler)(async (req, res) => {
|
||||
const userId = req.user.userId;
|
||||
const requestId = req.params.id;
|
||||
const row = await db_1.db.select().from(schema_1.friendships).where((0, drizzle_orm_1.eq)(schema_1.friendships.id, requestId)).limit(1);
|
||||
if (row.length === 0) {
|
||||
throw new errorHandler_1.AppError('NOT_FOUND', 'Request not found', 404);
|
||||
}
|
||||
if (row[0].userId !== userId && row[0].friendId !== userId) {
|
||||
throw new errorHandler_1.AppError('FORBIDDEN', 'Not allowed', 403);
|
||||
}
|
||||
await db_1.db.delete(schema_1.friendships).where((0, drizzle_orm_1.eq)(schema_1.friendships.id, requestId));
|
||||
res.json({ success: true });
|
||||
}));
|
||||
router.delete('/:friendId', (0, asyncHandler_1.asyncHandler)(async (req, res) => {
|
||||
const userId = req.user.userId;
|
||||
const friendId = req.params.friendId;
|
||||
await db_1.db
|
||||
.delete(schema_1.friendships)
|
||||
.where((0, drizzle_orm_1.or)((0, drizzle_orm_1.and)((0, drizzle_orm_1.eq)(schema_1.friendships.userId, userId), (0, drizzle_orm_1.eq)(schema_1.friendships.friendId, friendId)), (0, drizzle_orm_1.and)((0, drizzle_orm_1.eq)(schema_1.friendships.userId, friendId), (0, drizzle_orm_1.eq)(schema_1.friendships.friendId, userId))));
|
||||
res.json({ success: true });
|
||||
}));
|
||||
async function meUsername(userId) {
|
||||
const me = await db_1.db.select().from(schema_1.users).where((0, drizzle_orm_1.eq)(schema_1.users.id, userId)).limit(1);
|
||||
return me.length > 0 ? me[0].username.toLowerCase() : '';
|
||||
}
|
||||
exports.default = router;
|
||||
//# sourceMappingURL=friends.js.map
|
||||
Reference in New Issue
Block a user