chore: Initial commit of Opencord base state

2026-02-18 02:49:21 +01:00
commit 4fd17084a5
124 changed files with 17955 additions and 0 deletions
@@ -0,0 +1,409 @@
+import { eq, inArray } from 'drizzle-orm';
+import { getDb, schema } from '../db/index.js';
+import { generateSnowflake } from '../utils/snowflake.js';
+import { connectionManager } from './handler.js';
+import { isMember, getChannelServerId, isDmMember } from '../utils/permissions.js';
+import type { User, MessageWithUser, Attachment, DmMessageWithUser } from '@opencord/shared';
+
+function sanitizeUser(row: typeof schema.users.$inferSelect): User {
+  return {
+    id: row.id,
+    username: row.username,
+    displayName: row.displayName,
+    avatar: row.avatar,
+    status: (row.status ?? 'offline') as User['status'],
+    customStatus: row.customStatus,
+    createdAt: row.createdAt,
+  };
+}
+
+function getMessageWithUser(messageId: string): MessageWithUser | null {
+  const db = getDb();
+  const message = db.select().from(schema.messages).where(eq(schema.messages.id, messageId)).get();
+  if (!message) return null;
+
+  const user = db.select().from(schema.users).where(eq(schema.users.id, message.userId)).get();
+  if (!user) return null;
+
+  const attachmentRows = db.select()
+    .from(schema.attachments)
+    .where(eq(schema.attachments.messageId, messageId))
+    .all();
+
+  const attachments: Attachment[] = attachmentRows.map(a => ({
+    id: a.id,
+    messageId: a.messageId ?? messageId,
+    filename: a.filename,
+    originalName: a.originalName,
+    mimetype: a.mimetype,
+    size: a.size,
+    createdAt: a.createdAt,
+  }));
+
+  return {
+    id: message.id,
+    channelId: message.channelId,
+    userId: message.userId,
+    content: message.content,
+    editedAt: message.editedAt,
+    createdAt: message.createdAt,
+    user: sanitizeUser(user),
+    attachments,
+  };
+}
+
+// Typing timeout tracking
+const typingTimeouts: Map<string, NodeJS.Timeout> = new Map();
+
+export function handleClientEvent(
+  event: Record<string, unknown>,
+  userId: string,
+  username: string,
+): void {
+  const type = event.type as string;
+
+  switch (type) {
+    case 'message_create':
+      handleMessageCreate(event, userId);
+      break;
+    case 'message_edit':
+      handleMessageEdit(event, userId);
+      break;
+    case 'message_delete':
+      handleMessageDelete(event, userId);
+      break;
+    case 'typing_start':
+      handleTypingStart(event, userId, username);
+      break;
+    case 'presence_update':
+      handlePresenceUpdate(event, userId);
+      break;
+    case 'voice_join':
+      handleVoiceJoin(event, userId);
+      break;
+    case 'voice_leave':
+      handleVoiceLeave(userId);
+      break;
+    case 'dm_message_create':
+      handleDmMessageCreate(event, userId);
+      break;
+    default:
+      connectionManager.sendToUser(userId, {
+        type: 'error',
+        message: `Unknown event type: ${type}`,
+      });
+  }
+}
+
+function handleMessageCreate(event: Record<string, unknown>, userId: string): void {
+  const channelId = event.channelId as string;
+  const content = event.content as string;
+
+  if (!channelId || typeof channelId !== 'string') {
+    connectionManager.sendToUser(userId, { type: 'error', message: 'channelId is required' });
+    return;
+  }
+
+  if (!content || typeof content !== 'string' || content.trim().length === 0) {
+    connectionManager.sendToUser(userId, { type: 'error', message: 'content is required' });
+    return;
+  }
+
+  const serverId = getChannelServerId(channelId);
+  if (!serverId) {
+    connectionManager.sendToUser(userId, { type: 'error', message: 'Channel not found' });
+    return;
+  }
+
+  if (!isMember(serverId, userId)) {
+    connectionManager.sendToUser(userId, { type: 'error', message: 'Not a member of this server' });
+    return;
+  }
+
+  const db = getDb();
+  const messageId = generateSnowflake();
+  const now = Date.now();
+
+  db.insert(schema.messages).values({
+    id: messageId,
+    channelId,
+    userId,
+    content: content.trim(),
+    createdAt: now,
+  }).run();
+
+  const messageWithUser = getMessageWithUser(messageId);
+  if (messageWithUser) {
+    // Broadcast to all server members (including sender)
+    connectionManager.sendToServer(serverId, {
+      type: 'message_created',
+      message: messageWithUser,
+    });
+  }
+}
+
+function handleMessageEdit(event: Record<string, unknown>, userId: string): void {
+  const messageId = event.messageId as string;
+  const content = event.content as string;
+
+  if (!messageId || typeof messageId !== 'string') {
+    connectionManager.sendToUser(userId, { type: 'error', message: 'messageId is required' });
+    return;
+  }
+
+  if (!content || typeof content !== 'string' || content.trim().length === 0) {
+    connectionManager.sendToUser(userId, { type: 'error', message: 'content is required' });
+    return;
+  }
+
+  const db = getDb();
+  const message = db.select().from(schema.messages).where(eq(schema.messages.id, messageId)).get();
+  if (!message) {
+    connectionManager.sendToUser(userId, { type: 'error', message: 'Message not found' });
+    return;
+  }
+
+  if (message.userId !== userId) {
+    connectionManager.sendToUser(userId, { type: 'error', message: 'You can only edit your own messages' });
+    return;
+  }
+
+  const now = Date.now();
+  db.update(schema.messages)
+    .set({ content: content.trim(), editedAt: now })
+    .where(eq(schema.messages.id, messageId))
+    .run();
+
+  const serverId = getChannelServerId(message.channelId);
+  if (!serverId) return;
+
+  const updatedMessage = getMessageWithUser(messageId);
+  if (updatedMessage) {
+    connectionManager.sendToServer(serverId, {
+      type: 'message_updated',
+      message: updatedMessage,
+    });
+  }
+}
+
+function handleMessageDelete(event: Record<string, unknown>, userId: string): void {
+  const messageId = event.messageId as string;
+
+  if (!messageId || typeof messageId !== 'string') {
+    connectionManager.sendToUser(userId, { type: 'error', message: 'messageId is required' });
+    return;
+  }
+
+  const db = getDb();
+  const message = db.select().from(schema.messages).where(eq(schema.messages.id, messageId)).get();
+  if (!message) {
+    connectionManager.sendToUser(userId, { type: 'error', message: 'Message not found' });
+    return;
+  }
+
+  const serverId = getChannelServerId(message.channelId);
+  if (!serverId) return;
+
+  // Allow author or admin to delete
+  const isAuthor = message.userId === userId;
+  const memberRow = db.select()
+    .from(schema.serverMembers)
+    .where(eq(schema.serverMembers.serverId, serverId))
+    .all()
+    .find(m => m.userId === userId);
+
+  const isAdminRole = memberRow?.role === 'admin' || memberRow?.role === 'owner';
+
+  if (!isAuthor && !isAdminRole) {
+    connectionManager.sendToUser(userId, { type: 'error', message: 'You cannot delete this message' });
+    return;
+  }
+
+  // Delete attachments then message
+  db.delete(schema.attachments).where(eq(schema.attachments.messageId, messageId)).run();
+  db.delete(schema.messages).where(eq(schema.messages.id, messageId)).run();
+
+  connectionManager.sendToServer(serverId, {
+    type: 'message_deleted',
+    messageId,
+    channelId: message.channelId,
+  });
+}
+
+function handleTypingStart(event: Record<string, unknown>, userId: string, username: string): void {
+  const channelId = event.channelId as string;
+
+  if (!channelId || typeof channelId !== 'string') return;
+
+  const serverId = getChannelServerId(channelId);
+  if (!serverId) return;
+
+  if (!isMember(serverId, userId)) return;
+
+  // Clear previous typing timeout for this user+channel
+  const key = `${userId}:${channelId}`;
+  const existing = typingTimeouts.get(key);
+  if (existing) {
+    clearTimeout(existing);
+  }
+
+  // Broadcast typing event (exclude sender)
+  connectionManager.sendToServer(serverId, {
+    type: 'typing',
+    channelId,
+    userId,
+    username,
+  }, userId);
+
+  // Auto-expire typing after 5 seconds
+  const timeout = setTimeout(() => {
+    typingTimeouts.delete(key);
+  }, 5000);
+  typingTimeouts.set(key, timeout);
+}
+
+function handlePresenceUpdate(event: Record<string, unknown>, userId: string): void {
+  const status = event.status as string;
+
+  if (!status || !['online', 'idle', 'dnd'].includes(status)) {
+    connectionManager.sendToUser(userId, { type: 'error', message: 'Status must be "online", "idle", or "dnd"' });
+    return;
+  }
+
+  const db = getDb();
+  db.update(schema.users).set({ status }).where(eq(schema.users.id, userId)).run();
+
+  // Broadcast to all servers user is in
+  const userServers = connectionManager.getUserServers(userId);
+  for (const serverId of userServers) {
+    connectionManager.sendToServer(serverId, {
+      type: 'presence_update',
+      userId,
+      status,
+    }, userId);
+  }
+
+  // Also send to self (other tabs)
+  connectionManager.sendToUser(userId, {
+    type: 'presence_update',
+    userId,
+    status,
+  });
+}
+
+function handleVoiceJoin(event: Record<string, unknown>, userId: string): void {
+  const channelId = event.channelId as string;
+
+  if (!channelId || typeof channelId !== 'string') {
+    connectionManager.sendToUser(userId, { type: 'error', message: 'channelId is required' });
+    return;
+  }
+
+  const serverId = getChannelServerId(channelId);
+  if (!serverId) {
+    connectionManager.sendToUser(userId, { type: 'error', message: 'Channel not found' });
+    return;
+  }
+
+  if (!isMember(serverId, userId)) {
+    connectionManager.sendToUser(userId, { type: 'error', message: 'Not a member of this server' });
+    return;
+  }
+
+  // Leave any current voice channel
+  const previousChannel = connectionManager.leaveAllVoice(userId);
+  if (previousChannel) {
+    const prevServerId = getChannelServerId(previousChannel);
+    if (prevServerId) {
+      connectionManager.sendToServer(prevServerId, {
+        type: 'voice_state_update',
+        channelId: previousChannel,
+        userId,
+        action: 'leave',
+      });
+    }
+  }
+
+  // Join new voice channel
+  connectionManager.joinVoice(channelId, userId);
+
+  // Broadcast to server
+  connectionManager.sendToServer(serverId, {
+    type: 'voice_state_update',
+    channelId,
+    userId,
+    action: 'join',
+  });
+}
+
+function handleVoiceLeave(userId: string): void {
+  const channelId = connectionManager.leaveAllVoice(userId);
+  if (channelId) {
+    const serverId = getChannelServerId(channelId);
+    if (serverId) {
+      connectionManager.sendToServer(serverId, {
+        type: 'voice_state_update',
+        channelId,
+        userId,
+        action: 'leave',
+      });
+    }
+  }
+}
+
+function handleDmMessageCreate(event: Record<string, unknown>, userId: string): void {
+  const dmChannelId = event.dmChannelId as string;
+  const content = event.content as string;
+
+  if (!dmChannelId || typeof dmChannelId !== 'string') {
+    connectionManager.sendToUser(userId, { type: 'error', message: 'dmChannelId is required' });
+    return;
+  }
+
+  if (!content || typeof content !== 'string' || content.trim().length === 0) {
+    connectionManager.sendToUser(userId, { type: 'error', message: 'content is required' });
+    return;
+  }
+
+  if (!isDmMember(dmChannelId, userId)) {
+    connectionManager.sendToUser(userId, { type: 'error', message: 'Not a member of this DM channel' });
+    return;
+  }
+
+  const db = getDb();
+  const messageId = generateSnowflake();
+  const now = Date.now();
+
+  db.insert(schema.dmMessages).values({
+    id: messageId,
+    dmChannelId,
+    userId,
+    content: content.trim(),
+    createdAt: now,
+  }).run();
+
+  const user = db.select().from(schema.users).where(eq(schema.users.id, userId)).get();
+  if (!user) return;
+
+  const dmMessage: DmMessageWithUser = {
+    id: messageId,
+    dmChannelId,
+    userId,
+    content: content.trim(),
+    createdAt: now,
+    user: sanitizeUser(user),
+  };
+
+  // Send to all DM members
+  const dmMembers = db.select()
+    .from(schema.dmMembers)
+    .where(eq(schema.dmMembers.dmChannelId, dmChannelId))
+    .all();
+
+  for (const member of dmMembers) {
+    connectionManager.sendToUser(member.userId, {
+      type: 'dm_message_created',
+      message: dmMessage,
+    });
+  }
+}