gigaclaw 1.4.0

package/lib/chat/actions.js

@@ -0,0 +1,579 @@
+'use server';
+
+import { auth } from '../auth/index.js';
+import {
+  createChat as dbCreateChat,
+  getChatById,
+  getChatByWorkspaceId,
+  getMessagesByChatId,
+  deleteChat as dbDeleteChat,
+  deleteAllChatsByUser,
+  updateChatTitle,
+  toggleChatStarred,
+} from '../db/chats.js';
+import {
+  getNotifications as dbGetNotifications,
+  getUnreadCount as dbGetUnreadCount,
+  markAllRead as dbMarkAllRead,
+} from '../db/notifications.js';
+
+/**
+ * Get the authenticated user or throw.
+ */
+async function requireAuth() {
+  const session = await auth();
+  if (!session?.user?.id) {
+    throw new Error('Unauthorized');
+  }
+  return session.user;
+}
+
+/**
+ * Get all chats for the authenticated user (includes Telegram chats).
+ * @returns {Promise<object[]>}
+ */
+export async function getChats() {
+  const user = await requireAuth();
+  const { or, eq, desc } = await import('drizzle-orm');
+  const { getDb } = await import('../db/index.js');
+  const { chats, codeWorkspaces } = await import('../db/schema.js');
+  const db = getDb();
+  return db
+    .select({
+      id: chats.id,
+      userId: chats.userId,
+      title: chats.title,
+      starred: chats.starred,
+      codeWorkspaceId: chats.codeWorkspaceId,
+      containerName: codeWorkspaces.containerName,
+      createdAt: chats.createdAt,
+      updatedAt: chats.updatedAt,
+    })
+    .from(chats)
+    .leftJoin(codeWorkspaces, eq(chats.codeWorkspaceId, codeWorkspaces.id))
+    .where(or(eq(chats.userId, user.id), eq(chats.userId, 'telegram')))
+    .orderBy(desc(chats.updatedAt))
+    .all();
+}
+
+/**
+ * Get messages for a specific chat (with ownership check).
+ * @param {string} chatId
+ * @returns {Promise<object[]>}
+ */
+export async function getChatMessages(chatId) {
+  const user = await requireAuth();
+  const chat = getChatById(chatId);
+  if (!chat || (chat.userId !== user.id && chat.userId !== 'telegram')) {
+    return [];
+  }
+  return getMessagesByChatId(chatId);
+}
+
+/**
+ * Create a new chat.
+ * @param {string} [id] - Optional chat ID
+ * @param {string} [title='New Chat']
+ * @returns {Promise<object>}
+ */
+export async function createChat(id, title = 'New Chat') {
+  const user = await requireAuth();
+  return dbCreateChat(user.id, title, id);
+}
+
+/**
+ * Delete a chat (with ownership check).
+ * @param {string} chatId
+ * @returns {Promise<{success: boolean}>}
+ */
+export async function deleteChat(chatId) {
+  const user = await requireAuth();
+  const chat = getChatById(chatId);
+  if (!chat || chat.userId !== user.id) {
+    return { success: false };
+  }
+  dbDeleteChat(chatId);
+  return { success: true };
+}
+
+/**
+ * Get the title of a specific chat (with ownership check).
+ * @param {string} chatId
+ * @returns {Promise<string|null>}
+ */
+export async function getChatTitle(chatId) {
+  const user = await requireAuth();
+  const chat = getChatById(chatId);
+  if (!chat || (chat.userId !== user.id && chat.userId !== 'telegram')) {
+    return null;
+  }
+  return chat.title;
+}
+
+/**
+ * Get the title of the chat linked to a workspace (with ownership check).
+ * @param {string} workspaceId
+ * @returns {Promise<string|null>}
+ */
+export async function getChatTitleByWorkspace(workspaceId) {
+  const user = await requireAuth();
+  const chat = getChatByWorkspaceId(workspaceId);
+  if (!chat || (chat.userId !== user.id && chat.userId !== 'telegram')) {
+    return null;
+  }
+  return chat.title;
+}
+
+/**
+ * Rename a chat (with ownership check).
+ * @param {string} chatId
+ * @param {string} title
+ * @returns {Promise<{success: boolean}>}
+ */
+export async function renameChat(chatId, title) {
+  const user = await requireAuth();
+  const chat = getChatById(chatId);
+  if (!chat || chat.userId !== user.id) {
+    return { success: false };
+  }
+  updateChatTitle(chatId, title);
+  return { success: true };
+}
+
+/**
+ * Toggle a chat's starred status (with ownership check).
+ * @param {string} chatId
+ * @returns {Promise<{success: boolean, starred?: number}>}
+ */
+export async function starChat(chatId) {
+  const user = await requireAuth();
+  const chat = getChatById(chatId);
+  if (!chat || chat.userId !== user.id) {
+    return { success: false };
+  }
+  const starred = toggleChatStarred(chatId);
+  return { success: true, starred };
+}
+
+/**
+ * Delete all chats for the authenticated user.
+ * @returns {Promise<{success: boolean}>}
+ */
+export async function deleteAllChats() {
+  const user = await requireAuth();
+  deleteAllChatsByUser(user.id);
+  return { success: true };
+}
+
+/**
+ * Get all notifications, newest first.
+ * @returns {Promise<object[]>}
+ */
+export async function getNotifications() {
+  await requireAuth();
+  return dbGetNotifications();
+}
+
+/**
+ * Get count of unread notifications.
+ * @returns {Promise<number>}
+ */
+export async function getUnreadNotificationCount() {
+  await requireAuth();
+  return dbGetUnreadCount();
+}
+
+/**
+ * Mark all notifications as read.
+ * @returns {Promise<{success: boolean}>}
+ */
+export async function markNotificationsRead() {
+  await requireAuth();
+  dbMarkAllRead();
+  return { success: true };
+}
+
+// ─────────────────────────────────────────────────────────────────────────────
+// App info actions
+// ─────────────────────────────────────────────────────────────────────────────
+
+/**
+ * Get the installed package version and update status (auth-gated, never in client bundle).
+ * @returns {Promise<{ version: string, updateAvailable: string|null }>}
+ */
+export async function getAppVersion() {
+  await requireAuth();
+  const { getInstalledVersion } = await import('../cron.js');
+  const { getAvailableVersion, getReleaseNotes } = await import('../db/update-check.js');
+  const version = getInstalledVersion();
+  const available = getAvailableVersion();
+  const isNewer = available && available !== version;
+  return {
+    version,
+    updateAvailable: isNewer ? available : null,
+    changelog: isNewer ? getReleaseNotes() : null,
+  };
+}
+
+/**
+ * Trigger the upgrade-event-handler workflow via GitHub Actions.
+ * @returns {Promise<{ success: boolean }>}
+ */
+export async function triggerUpgrade() {
+  await requireAuth();
+  const { triggerWorkflowDispatch } = await import('../tools/github.js');
+  const { getAvailableVersion } = await import('../db/update-check.js');
+  const targetVersion = getAvailableVersion();
+  await triggerWorkflowDispatch('upgrade-event-handler.yml', 'main', {
+    target_version: targetVersion || '',
+  });
+  return { success: true };
+}
+
+// ─────────────────────────────────────────────────────────────────────────────
+// API Key actions
+// ─────────────────────────────────────────────────────────────────────────────
+
+/**
+ * Create (or replace) the API key.
+ * @returns {Promise<{ key: string, record: object } | { error: string }>}
+ */
+export async function createNewApiKey() {
+  const user = await requireAuth();
+  try {
+    const { createApiKeyRecord } = await import('../db/api-keys.js');
+    return createApiKeyRecord(user.id);
+  } catch (err) {
+    console.error('Failed to create API key:', err);
+    return { error: 'Failed to create API key' };
+  }
+}
+
+/**
+ * Get the current API key metadata (no hash).
+ * @returns {Promise<object|null>}
+ */
+export async function getApiKeys() {
+  await requireAuth();
+  try {
+    const { getApiKey } = await import('../db/api-keys.js');
+    return getApiKey();
+  } catch (err) {
+    console.error('Failed to get API key:', err);
+    return null;
+  }
+}
+
+/**
+ * Delete the API key.
+ * @returns {Promise<{ success: boolean } | { error: string }>}
+ */
+export async function deleteApiKey() {
+  await requireAuth();
+  try {
+    const mod = await import('../db/api-keys.js');
+    mod.deleteApiKey();
+    return { success: true };
+  } catch (err) {
+    console.error('Failed to delete API key:', err);
+    return { error: 'Failed to delete API key' };
+  }
+}
+
+// ─────────────────────────────────────────────────────────────────────────────
+// Code mode actions
+// ─────────────────────────────────────────────────────────────────────────────
+
+/**
+ * Get repositories accessible to the authenticated user.
+ * @returns {Promise<{full_name: string, default_branch: string}[]>}
+ */
+export async function getRepositories() {
+  await requireAuth();
+  try {
+    const { listRepositories } = await import('../tools/github.js');
+    return await listRepositories();
+  } catch (err) {
+    console.error('Failed to get repositories:', err);
+    return [];
+  }
+}
+
+/**
+ * Get branches for a repository.
+ * @param {string} repoFullName - e.g. "owner/repo"
+ * @returns {Promise<{name: string, isDefault: boolean}[]>}
+ */
+export async function getBranches(repoFullName) {
+  await requireAuth();
+  try {
+    const { listBranches } = await import('../tools/github.js');
+    return await listBranches(repoFullName);
+  } catch (err) {
+    console.error('Failed to get branches:', err);
+    return [];
+  }
+}
+
+/**
+ * Get chat metadata (codeWorkspaceId) for an existing chat.
+ * @param {string} chatId
+ * @returns {Promise<{codeWorkspaceId: string|null}|null>}
+ */
+export async function getChatMeta(chatId) {
+  const user = await requireAuth();
+  const chat = getChatById(chatId);
+  if (!chat || (chat.userId !== user.id && chat.userId !== 'telegram')) {
+    return null;
+  }
+  return {
+    title: chat.title || null,
+    starred: chat.starred || 0,
+    codeWorkspaceId: chat.codeWorkspaceId || null,
+  };
+}
+
+/**
+ * Get chat metadata by workspace ID (title, starred, chatId).
+ * @param {string} workspaceId
+ * @returns {Promise<{chatId: string, title: string|null, starred: number, codeWorkspaceId: string|null}|null>}
+ */
+export async function getChatMetaByWorkspace(workspaceId) {
+  const user = await requireAuth();
+  const chat = getChatByWorkspaceId(workspaceId);
+  if (!chat || (chat.userId !== user.id && chat.userId !== 'telegram')) {
+    return null;
+  }
+  return {
+    chatId: chat.id,
+    title: chat.title || null,
+    starred: chat.starred || 0,
+    codeWorkspaceId: chat.codeWorkspaceId || null,
+  };
+}
+
+/**
+ * Get workspace details by ID.
+ * @param {string} workspaceId
+ * @returns {Promise<{id: string, repo: string, branch: string, containerName: string|null}|null>}
+ */
+export async function getWorkspace(workspaceId) {
+  await requireAuth();
+  try {
+    const { getCodeWorkspaceById } = await import('../db/code-workspaces.js');
+    const ws = getCodeWorkspaceById(workspaceId);
+    if (!ws) return null;
+    return { id: ws.id, repo: ws.repo, branch: ws.branch, containerName: ws.containerName, codingAgent: ws.codingAgent };
+  } catch (err) {
+    console.error('Failed to get workspace:', err);
+    return null;
+  }
+}
+
+// ─────────────────────────────────────────────────────────────────────────────
+// Pull Request actions
+// ─────────────────────────────────────────────────────────────────────────────
+
+/**
+ * Get all open pull requests from GitHub.
+ * @returns {Promise<object[]>}
+ */
+export async function getPullRequests() {
+  await requireAuth();
+  try {
+    const { getOpenPullRequests } = await import('../tools/github.js');
+    return await getOpenPullRequests();
+  } catch (err) {
+    console.error('Failed to get pull requests:', err);
+    return [];
+  }
+}
+
+/**
+ * Get the count of open pull requests.
+ * @returns {Promise<number>}
+ */
+export async function getPullRequestCount() {
+  await requireAuth();
+  try {
+    const { getOpenPullRequests } = await import('../tools/github.js');
+    const prs = await getOpenPullRequests();
+    return prs.length;
+  } catch (err) {
+    console.error('Failed to get pull request count:', err);
+    return 0;
+  }
+}
+
+// ─────────────────────────────────────────────────────────────────────────────
+// Swarm actions
+// ─────────────────────────────────────────────────────────────────────────────
+
+/**
+ * Get swarm status (active + completed jobs with counts).
+ * @returns {Promise<object>}
+ */
+export async function getSwarmStatus(page = 1) {
+  await requireAuth();
+  try {
+    const { getSwarmStatus: fetchStatus } = await import('../tools/github.js');
+    return await fetchStatus(page);
+  } catch (err) {
+    console.error('Failed to get swarm status:', err);
+    return { error: 'Failed to get swarm status', runs: [], hasMore: false };
+  }
+}
+
+/**
+ * Get swarm config (crons + triggers).
+ * @returns {Promise<{ crons: object[], triggers: object[] }>}
+ */
+export async function getSwarmConfig() {
+  await requireAuth();
+  const { cronsFile, triggersFile } = await import('../paths.js');
+  const fs = await import('fs');
+  let crons = [];
+  let triggers = [];
+  try { crons = JSON.parse(fs.readFileSync(cronsFile, 'utf8')); } catch {}
+  try { triggers = JSON.parse(fs.readFileSync(triggersFile, 'utf8')); } catch {}
+  return { crons, triggers };
+}
+
+
+// ─────────────────────────────────────────────────────────────────────────────
+// Export actions
+// ─────────────────────────────────────────────────────────────────────────────
+
+/**
+ * Format a single chat's messages into the requested format string.
+ * @param {object} chat  - { id, title, createdAt, updatedAt }
+ * @param {object[]} msgs - [{ role, content, createdAt }]
+ * @param {'json'|'md'|'txt'} format
+ * @returns {string}
+ */
+function formatChatExport(chat, msgs, format) {
+  const title = chat.title || 'Untitled Chat';
+  const exportedAt = new Date().toISOString();
+
+  if (format === 'json') {
+    return JSON.stringify(
+      {
+        id: chat.id,
+        title,
+        exportedAt,
+        createdAt: new Date(chat.createdAt).toISOString(),
+        updatedAt: new Date(chat.updatedAt).toISOString(),
+        messages: msgs.map((m) => ({
+          id: m.id,
+          role: m.role,
+          content: m.content,
+          createdAt: new Date(m.createdAt).toISOString(),
+        })),
+      },
+      null,
+      2
+    );
+  }
+
+  if (format === 'md') {
+    const lines = [
+      `# ${title}`,
+      ``,
+      `**Exported:** ${exportedAt}`,
+      `**Created:** ${new Date(chat.createdAt).toISOString()}`,
+      ``,
+      `---`,
+      ``,
+    ];
+    for (const m of msgs) {
+      const ts = new Date(m.createdAt).toLocaleTimeString('en-US', { hour: '2-digit', minute: '2-digit' });
+      const role = m.role === 'user' ? '**User**' : '**Assistant**';
+      lines.push(`### ${role} · ${ts}`);
+      lines.push(``);
+      lines.push(m.content);
+      lines.push(``);
+      lines.push(`---`);
+      lines.push(``);
+    }
+    return lines.join('\n');
+  }
+
+  // Plain text (default)
+  const lines = [
+    `${title}`,
+    `Exported: ${exportedAt}`,
+    `Created:  ${new Date(chat.createdAt).toISOString()}`,
+    ``,
+    `${'─'.repeat(60)}`,
+    ``,
+  ];
+  for (const m of msgs) {
+    const ts = new Date(m.createdAt).toLocaleTimeString('en-US', { hour: '2-digit', minute: '2-digit' });
+    const speaker = m.role === 'user' ? 'User' : 'Assistant';
+    lines.push(`[${ts}] ${speaker}:`);
+    lines.push(m.content);
+    lines.push(``);
+  }
+  return lines.join('\n');
+}
+
+/**
+ * Export a single chat's messages.
+ * @param {string} chatId
+ * @param {'json'|'md'|'txt'} format
+ * @returns {Promise<{ filename: string, content: string, mimeType: string }>}
+ */
+export async function exportChat(chatId, format = 'md') {
+  const user = await requireAuth();
+  const chat = getChatById(chatId);
+  if (!chat || (chat.userId !== user.id && chat.userId !== 'telegram')) {
+    throw new Error('Chat not found or access denied');
+  }
+  const msgs = getMessagesByChatId(chatId);
+
+  const ext = format === 'json' ? 'json' : format === 'md' ? 'md' : 'txt';
+  const safeName = (chat.title || 'chat').replace(/[^a-z0-9_\-]/gi, '_').slice(0, 60);
+  const filename = `${safeName}_${chatId.slice(0, 8)}.${ext}`;
+  const mimeType =
+    format === 'json' ? 'application/json' : format === 'md' ? 'text/markdown' : 'text/plain';
+
+  return {
+    filename,
+    content: formatChatExport(chat, msgs, format),
+    mimeType,
+  };
+}
+
+/**
+ * Export all chats for the current user as an array of export objects.
+ * @param {'json'|'md'|'txt'} format
+ * @returns {Promise<Array<{ filename: string, content: string, mimeType: string }>>}
+ */
+export async function exportAllChats(format = 'md') {
+  const user = await requireAuth();
+  const { or, eq, desc } = await import('drizzle-orm');
+  const { getDb } = await import('../db/index.js');
+  const { chats } = await import('../db/schema.js');
+  const db = getDb();
+  const userChats = db
+    .select()
+    .from(chats)
+    .where(or(eq(chats.userId, user.id), eq(chats.userId, 'telegram')))
+    .orderBy(desc(chats.updatedAt))
+    .all();
+
+  const ext = format === 'json' ? 'json' : format === 'md' ? 'md' : 'txt';
+  const mimeType =
+    format === 'json' ? 'application/json' : format === 'md' ? 'text/markdown' : 'text/plain';
+
+  return userChats.map((chat) => {
+    const msgs = getMessagesByChatId(chat.id);
+    const safeName = (chat.title || 'chat').replace(/[^a-z0-9_\-]/gi, '_').slice(0, 60);
+    const filename = `${safeName}_${chat.id.slice(0, 8)}.${ext}`;
+    return {
+      filename,
+      content: formatChatExport(chat, msgs, format),
+      mimeType,
+    };
+  });
+}