mirror of
https://github.com/yusufipk/OpenFrame.git
synced 2026-09-11 09:36:08 +00:00
- Introduce `withCacheControl` utility function for API responses. - Apply `private, no-store` to authentication and data modification (POST, PATCH, DELETE) routes. - Apply `private, no-cache` to sensitive data retrieval (GET) routes. - Enhance security by preventing caching of private user data. - Ensure fresh data is always fetched for authenticated API responses.
142 lines
4.8 KiB
TypeScript
142 lines
4.8 KiB
TypeScript
import { NextRequest } from 'next/server';
|
|
import { db } from '@/lib/db';
|
|
import { auth } from '@/lib/auth';
|
|
import { WorkspaceMemberRole } from '@prisma/client';
|
|
import { rateLimit } from '@/lib/rate-limit';
|
|
import { apiErrors, successResponse, withCacheControl } from '@/lib/api-response';
|
|
|
|
type RouteParams = { params: Promise<{ workspaceId: string }> };
|
|
|
|
// GET /api/workspaces/[workspaceId]/members - List members
|
|
export async function GET(request: NextRequest, { params }: RouteParams) {
|
|
try {
|
|
const session = await auth();
|
|
const { workspaceId } = await params;
|
|
|
|
if (!session?.user?.id) {
|
|
return apiErrors.unauthorized();
|
|
}
|
|
|
|
const workspace = await db.workspace.findUnique({
|
|
where: { id: workspaceId },
|
|
include: {
|
|
members: { where: { userId: session.user.id } },
|
|
},
|
|
});
|
|
|
|
if (!workspace) {
|
|
return apiErrors.notFound('Workspace');
|
|
}
|
|
|
|
const isOwner = workspace.ownerId === session.user.id;
|
|
const isMember = workspace.members.length > 0;
|
|
|
|
if (!isOwner && !isMember) {
|
|
return apiErrors.forbidden('Access denied');
|
|
}
|
|
|
|
const members = await db.workspaceMember.findMany({
|
|
where: { workspaceId },
|
|
include: {
|
|
user: { select: { id: true, name: true, image: true } },
|
|
},
|
|
orderBy: { createdAt: 'asc' },
|
|
});
|
|
|
|
// Include the owner as well
|
|
const owner = await db.user.findUnique({
|
|
where: { id: workspace.ownerId },
|
|
select: { id: true, name: true, image: true },
|
|
});
|
|
|
|
const response = successResponse({ members, owner });
|
|
return withCacheControl(response, 'private, max-age=60, stale-while-revalidate=120');
|
|
} catch (error) {
|
|
console.error('Error fetching workspace members:', error);
|
|
return apiErrors.internalError('Failed to fetch members');
|
|
}
|
|
}
|
|
|
|
// POST /api/workspaces/[workspaceId]/members - Invite a member
|
|
export async function POST(request: NextRequest, { params }: RouteParams) {
|
|
try {
|
|
const limited = await rateLimit(request, 'invite-member');
|
|
if (limited) return limited;
|
|
|
|
const session = await auth();
|
|
const { workspaceId } = await params;
|
|
|
|
if (!session?.user?.id) {
|
|
return apiErrors.unauthorized();
|
|
}
|
|
|
|
// Check if user is owner or admin
|
|
const workspace = await db.workspace.findUnique({
|
|
where: { id: workspaceId },
|
|
include: { members: { where: { userId: session.user.id } } },
|
|
});
|
|
|
|
if (!workspace) {
|
|
return apiErrors.notFound('Workspace');
|
|
}
|
|
|
|
const isOwner = workspace.ownerId === session.user.id;
|
|
const isAdmin = workspace.members[0]?.role === WorkspaceMemberRole.ADMIN;
|
|
|
|
if (!isOwner && !isAdmin) {
|
|
return apiErrors.forbidden('Only workspace owners and admins can invite members');
|
|
}
|
|
|
|
const body = await request.json();
|
|
const { email, role } = body;
|
|
|
|
if (!email || typeof email !== 'string') {
|
|
return apiErrors.badRequest('Email is required');
|
|
}
|
|
|
|
// Validate role
|
|
const validRoles = ['ADMIN', 'COMMENTATOR'];
|
|
const memberRole = validRoles.includes(role) ? role : 'COMMENTATOR';
|
|
|
|
// Find user by email
|
|
const userToInvite = await db.user.findUnique({
|
|
where: { email: email.toLowerCase().trim() },
|
|
});
|
|
|
|
if (!userToInvite) {
|
|
const response = successResponse({ message: 'If the user exists, an invitation has been sent.' });
|
|
return withCacheControl(response, 'private, no-store');
|
|
}
|
|
|
|
if (userToInvite.id === workspace.ownerId) {
|
|
return apiErrors.badRequest('Cannot invite the workspace owner as a member');
|
|
}
|
|
|
|
// Check if already a member
|
|
const existingMember = await db.workspaceMember.findUnique({
|
|
where: { workspaceId_userId: { workspaceId, userId: userToInvite.id } },
|
|
});
|
|
|
|
if (existingMember) {
|
|
return apiErrors.conflict('User is already a member of this workspace');
|
|
}
|
|
|
|
const member = await db.workspaceMember.create({
|
|
data: {
|
|
workspaceId,
|
|
userId: userToInvite.id,
|
|
role: memberRole as WorkspaceMemberRole,
|
|
},
|
|
include: {
|
|
user: { select: { id: true, name: true, image: true } },
|
|
},
|
|
});
|
|
|
|
const response = successResponse(member, 201);
|
|
return withCacheControl(response, 'private, no-store');
|
|
} catch (error) {
|
|
console.error('Error inviting workspace member:', error);
|
|
return apiErrors.internalError('Failed to invite member');
|
|
}
|
|
}
|