dis-gravity 1.1.0 → 1.2.5

package/lib/onboarding.js

@@ -0,0 +1,512 @@
+'use strict';
+
+/**
+ * Dis-Gravity Shared Onboarding Engine
+ * Shared between CLI daemon (bin/dis-gravity.js) and VS Code Extension (dist/extension.js).
+ * Zero runtime dependencies (uses native Node 18+ global fetch).
+ */
+
+const fs = require('fs');
+const path = require('path');
+const os = require('os');
+const cp = require('child_process');
+
+const DISCORD_API_BASE = 'https://discord.com/api/v10';
+const GATEWAY_MESSAGE_CONTENT_LIMITED = 1 << 19; // 524288
+
+// Permissions needed: Manage Channels (16), View Channel (1024), Send Messages (2048),
+// Embed Links (16384), Attach Files (32768), Read Message History (65536),
+// Add Reactions (64), Send Messages in Threads (274877906944)
+const DISGRAVITY_PERMISSIONS = (
+  16n |
+  64n |
+  1024n |
+  2048n |
+  16384n |
+  32768n |
+  65536n |
+  274877906944n
+).toString(); // "274878024256"
+
+const CONFIG_DIR = path.join(os.homedir(), '.dis-gravity');
+const CONFIG_FILE = path.join(CONFIG_DIR, 'config.json');
+
+/**
+ * Validates whether string looks like a Discord Bot Token
+ */
+function looksLikeBotToken(token) {
+  if (!token || typeof token !== 'string') return false;
+  const trimmed = token.trim().replace(/^Bot\s+/i, '').trim();
+  // Format: [20+ chars].[5+ chars].[25+ chars]
+  return /^[A-Za-z0-9_-]{20,}\.[A-Za-z0-9_-]{5,}\.[A-Za-z0-9_-]{25,}$/.test(trimmed);
+}
+
+/**
+ * Builds OAuth2 invite link with exact required permissions and bot + applications.commands scopes
+ */
+function buildInviteUrl(clientId, permissions = DISGRAVITY_PERMISSIONS) {
+  if (!clientId) throw new Error('clientId is required to build invite URL');
+  return `https://discord.com/oauth2/authorize?client_id=${clientId}&scope=bot%20applications.commands&permissions=${permissions}`;
+}
+
+/**
+ * Low-level Discord REST API request handler with automatic rate-limit retry
+ */
+async function discordRequest(token, method, endpoint, body = null, retries = 3) {
+  if (!token) throw new Error('Discord token is required for API requests');
+  if (typeof fetch !== 'function') {
+    throw new Error('Native fetch is not available in this Node runtime. Node 18+ is required.');
+  }
+
+  const cleanToken = token.trim().replace(/^Bot\s+/i, '');
+  const url = endpoint.startsWith('http') ? endpoint : `${DISCORD_API_BASE}${endpoint.startsWith('/') ? '' : '/'}${endpoint}`;
+  const options = {
+    method,
+    headers: {
+      Authorization: `Bot ${cleanToken}`,
+      'Content-Type': 'application/json',
+      'User-Agent': 'DisGravity-Onboarding/1.1 (https://gitlab.com/rethinakumarsambath8/dis-gravity)'
+    }
+  };
+
+  if (body && ['POST', 'PUT', 'PATCH'].includes(method.toUpperCase())) {
+    options.body = typeof body === 'string' ? body : JSON.stringify(body);
+  }
+
+  const res = await fetch(url, options);
+
+  // Rate-limiting handler (HTTP 429)
+  if (res.status === 429 && retries > 0) {
+    let waitMs = 1500;
+    try {
+      const data = await res.json();
+      if (data.retry_after) waitMs = Math.ceil(data.retry_after * 1000) + 200;
+    } catch {
+      const retryHeader = res.headers.get('retry-after');
+      if (retryHeader) waitMs = Math.ceil(parseFloat(retryHeader) * 1000) + 200;
+    }
+    await new Promise((r) => setTimeout(r, waitMs));
+    return discordRequest(token, method, endpoint, body, retries - 1);
+  }
+
+  if (res.status === 204) {
+    return null;
+  }
+
+  let json;
+  const text = await res.text();
+  try {
+    json = JSON.parse(text);
+  } catch {
+    json = { raw: text };
+  }
+
+  if (!res.ok) {
+    const err = new Error(json.message || `Discord API error (${res.status}): ${text.substring(0, 200)}`);
+    err.status = res.status;
+    err.code = json.code;
+    err.data = json;
+    throw err;
+  }
+
+  return json;
+}
+
+/**
+ * Low-level Discord REST API multipart/form-data request handler for file attachments
+ * @param {string} token - Discord Bot Token
+ * @param {string} method - HTTP method ('POST' | 'PATCH')
+ * @param {string} endpoint - API endpoint
+ * @param {object} payloadJson - JSON payload object (content, embeds, components, etc.)
+ * @param {Array<{name: string, buffer: Buffer, contentType?: string}>} files - Files to attach
+ * @param {number} retries - Rate-limit retry count
+ */
+async function discordMultipartRequest(token, method, endpoint, payloadJson = {}, files = [], retries = 2) {
+  if (!token) throw new Error('Discord token is required for API requests');
+  if (typeof fetch !== 'function' || typeof FormData !== 'function' || typeof Blob !== 'function') {
+    throw new Error('Native fetch/FormData/Blob are required (Node 18+).');
+  }
+
+  const cleanToken = token.trim().replace(/^Bot\s+/i, '');
+  const url = endpoint.startsWith('http') ? endpoint : `${DISCORD_API_BASE}${endpoint.startsWith('/') ? '' : '/'}${endpoint}`;
+
+  const form = new FormData();
+  const attachmentsMeta = files.map((f, idx) => ({
+    id: idx,
+    filename: f.name || `file-${idx}.png`
+  }));
+
+  const bodyPayload = {
+    ...payloadJson,
+    ...(attachmentsMeta.length > 0 ? { attachments: attachmentsMeta } : {})
+  };
+  form.append('payload_json', JSON.stringify(bodyPayload));
+
+  files.forEach((f, idx) => {
+    const blob = new Blob([f.buffer], { type: f.contentType || 'image/png' });
+    form.append(`files[${idx}]`, blob, f.name || `file-${idx}.png`);
+  });
+
+  const res = await fetch(url, {
+    method,
+    headers: {
+      Authorization: `Bot ${cleanToken}`,
+      'User-Agent': 'DisGravity-Onboarding/1.2 (https://gitlab.com/rethinakumarsambath8/dis-gravity)'
+    },
+    body: form
+  });
+
+  if (res.status === 429 && retries > 0) {
+    let waitMs = 1500;
+    try {
+      const data = await res.json();
+      if (data.retry_after) waitMs = Math.ceil(data.retry_after * 1000) + 200;
+    } catch {}
+    await new Promise((r) => setTimeout(r, waitMs));
+    return discordMultipartRequest(token, method, endpoint, payloadJson, files, retries - 1);
+  }
+
+  if (res.status === 204) return null;
+
+  const text = await res.text();
+  let json;
+  try {
+    json = JSON.parse(text);
+  } catch {
+    json = { raw: text };
+  }
+
+  if (!res.ok) {
+    const err = new Error(json.message || `Discord API multipart error (${res.status}): ${text.substring(0, 200)}`);
+    err.status = res.status;
+    err.code = json.code;
+    err.data = json;
+    throw err;
+  }
+
+  return json;
+}
+
+/**
+ * Inspects the Discord application associated with the bot token.
+ * Returns appId, botId, ownerId, appName, teamMembers, flags
+ */
+async function inspectApplication(token) {
+  const [app, botUser] = await Promise.all([
+    discordRequest(token, 'GET', '/applications/@me'),
+    discordRequest(token, 'GET', '/users/@me')
+  ]);
+
+  const ownerId = app.owner ? app.owner.id : (app.team ? app.team.owner_user_id : null);
+  const teamMembers = app.team?.members ? app.team.members.map(m => ({ id: m.user.id, username: m.user.username })) : [];
+
+  return {
+    appId: app.id,
+    botId: botUser.id,
+    botTag: `${botUser.username}#${botUser.discriminator || '0'}`,
+    botUsername: botUser.username,
+    appName: app.name,
+    flags: app.flags || 0,
+    ownerId,
+    ownerUsername: app.owner?.username || 'Owner',
+    isTeam: Boolean(app.team),
+    teamMembers,
+    messageContentEnabled: Boolean((app.flags || 0) & GATEWAY_MESSAGE_CONTENT_LIMITED)
+  };
+}
+
+/**
+ * Ensures Message Content Intent is enabled.
+ * First tries PATCH /applications/@me with GATEWAY_MESSAGE_CONTENT_LIMITED flag.
+ * If API rejects or flag not set, returns fallback deep link instructions.
+ */
+async function ensureMessageContentIntent(token) {
+  const app = await discordRequest(token, 'GET', '/applications/@me');
+  const currentFlags = app.flags || 0;
+
+  if ((currentFlags & GATEWAY_MESSAGE_CONTENT_LIMITED) !== 0) {
+    return {
+      success: true,
+      alreadyEnabled: true,
+      flags: currentFlags
+    };
+  }
+
+  // Attempt automatic update via PATCH /applications/@me
+  const targetFlags = currentFlags | GATEWAY_MESSAGE_CONTENT_LIMITED;
+  try {
+    const patched = await discordRequest(token, 'PATCH', '/applications/@me', {
+      flags: targetFlags
+    });
+
+    if ((patched.flags & GATEWAY_MESSAGE_CONTENT_LIMITED) !== 0) {
+      return {
+        success: true,
+        alreadyEnabled: false,
+        autoEnabled: true,
+        flags: patched.flags
+      };
+    }
+  } catch (err) {
+    // Discord may reject PATCH for apps with certain flags or developer requirements
+  }
+
+  // Verification re-fetch
+  const recheck = await discordRequest(token, 'GET', '/applications/@me');
+  if ((recheck.flags & GATEWAY_MESSAGE_CONTENT_LIMITED) !== 0) {
+    return {
+      success: true,
+      alreadyEnabled: false,
+      autoEnabled: true,
+      flags: recheck.flags
+    };
+  }
+
+  return {
+    success: false,
+    autoEnabled: false,
+    portalUrl: `https://discord.com/developers/applications/${app.id}/bot`,
+    message: "Discord requires toggling 'Message Content Intent' manually in the Developer Portal for this application."
+  };
+}
+
+/**
+ * Fetches list of guilds the bot is currently in
+ */
+async function listUserGuilds(token) {
+  return discordRequest(token, 'GET', '/users/@me/guilds');
+}
+
+/**
+ * Polls for a new guild authorization or waits for user to authorize the bot
+ */
+async function waitForGuildJoin(token, options = {}) {
+  const timeoutMs = options.timeoutMs || 120000;
+  const pollIntervalMs = options.pollIntervalMs || 2500;
+  const initialGuildIds = new Set(options.initialGuildIds || []);
+  const startTime = Date.now();
+
+  while (Date.now() - startTime < timeoutMs) {
+    try {
+      const guilds = await listUserGuilds(token);
+      const newGuild = guilds.find(g => !initialGuildIds.has(g.id));
+      if (newGuild) {
+        return newGuild;
+      }
+      if (options.onPoll) {
+        options.onPoll({ elapsedMs: Date.now() - startTime, guildCount: guilds.length });
+      }
+    } catch {}
+
+    await new Promise(r => setTimeout(r, pollIntervalMs));
+  }
+
+  // Timeout reached, return whatever guilds we currently have
+  const finalGuilds = await listUserGuilds(token).catch(() => []);
+  if (finalGuilds.length === 1 && initialGuildIds.size === 0) {
+    return finalGuilds[0];
+  }
+  const err = new Error(`Timed out waiting for bot to join server (${Math.round(timeoutMs / 1000)}s).`);
+  err.guilds = finalGuilds;
+  throw err;
+}
+
+/**
+ * Builds permission overwrite objects for private channel (#dis-gravity)
+ * - @everyone: ViewChannel denied
+ * - bot: full permissions allowed
+ * - owner(s): full permissions allowed
+ */
+function privateChannelOverwrites(guildId, botId, ownerIds = []) {
+  const overwrites = [];
+
+  // Deny @everyone View Channel (1024)
+  overwrites.push({
+    id: guildId,
+    type: 0, // role
+    allow: '0',
+    deny: '1024'
+  });
+
+  // Allow Bot
+  if (botId) {
+    overwrites.push({
+      id: botId,
+      type: 1, // member
+      allow: DISGRAVITY_PERMISSIONS,
+      deny: '0'
+    });
+  }
+
+  // Allow Owners
+  const owners = Array.isArray(ownerIds) ? ownerIds : [ownerIds];
+  for (const oId of owners) {
+    if (oId && oId !== botId) {
+      overwrites.push({
+        id: oId,
+        type: 1, // member
+        allow: DISGRAVITY_PERMISSIONS,
+        deny: '0'
+      });
+    }
+  }
+
+  return overwrites;
+}
+
+/**
+ * Locates existing channel named #dis-gravity or creates a private one
+ */
+async function findOrCreateChannel(token, guildId, options = {}) {
+  const channelName = (options.channelName || 'dis-gravity').toLowerCase().replace(/\s+/g, '-');
+  const ownerIds = options.ownerIds || (options.ownerId ? [options.ownerId] : []);
+  const botId = options.botId;
+
+  // 1. Check existing channels
+  const channels = await discordRequest(token, 'GET', `/guilds/${guildId}/channels`);
+  const existing = channels.find(c => c.name === channelName && c.type === 0);
+  if (existing) {
+    return {
+      channel: existing,
+      created: false,
+      reused: true
+    };
+  }
+
+  // 2. Create private channel
+  const overwrites = privateChannelOverwrites(guildId, botId, ownerIds);
+  try {
+    const created = await discordRequest(token, 'POST', `/guilds/${guildId}/channels`, {
+      name: channelName,
+      type: 0, // GUILD_TEXT
+      topic: 'Dis-Gravity Remote AI IDE & Terminal Bridge',
+      permission_overwrites: overwrites
+    });
+
+    return {
+      channel: created,
+      created: true,
+      reused: false
+    };
+  } catch (err) {
+    // If bot lacks MANAGE_CHANNELS, fall back to first writable text channel
+    if (err.status === 403 || err.code === 50013) {
+      const fallback = channels.find(c => c.type === 0);
+      if (fallback) {
+        const warningErr = new Error(`Bot lacks 'Manage Channels' permission to create #${channelName}. You can use existing channel #${fallback.name} or grant the bot 'Manage Channels' in Server Settings.`);
+        warningErr.code = 'MISSING_MANAGE_CHANNELS';
+        warningErr.fallbackChannel = fallback;
+        warningErr.availableChannels = channels.filter(c => c.type === 0);
+        throw warningErr;
+      }
+    }
+    throw err;
+  }
+}
+
+/**
+ * Sends initial welcome and ping message to owner in the newly configured channel
+ */
+async function sendWelcomeMessage(token, channelId, options = {}) {
+  const ownerIds = options.ownerIds || (options.ownerId ? [options.ownerId] : []);
+  const hostName = options.hostName || os.hostname();
+  const provider = (options.provider || 'Antigravity / VS Code').toUpperCase();
+
+  const mentions = ownerIds.map(id => `<@${id}>`).join(' ');
+  const text = `🚀 **Dis-Gravity Bridge Activated!**\n\n` +
+    (mentions ? `Welcome ${mentions}! ` : '') +
+    `Your remote AI agent and terminal bridge is online.\n\n` +
+    `• **Host Machine:** \`${hostName}\` (${os.platform()} ${os.arch()})\n` +
+    `• **AI Engine:** 🟢 **${provider}**\n` +
+    `• **Channel:** <#${channelId}>\n\n` +
+    `> 💡 **Quick Test:** Send \`!info\` or \`!status\` here to view your system dashboard, or ask your AI coding agent a question!`;
+
+  return discordRequest(token, 'POST', `/channels/${channelId}/messages`, {
+    content: text
+  });
+}
+
+/**
+ * Loads configuration from ~/.dis-gravity/config.json
+ */
+function loadConfig() {
+  try {
+    if (fs.existsSync(CONFIG_FILE)) {
+      const content = fs.readFileSync(CONFIG_FILE, 'utf8');
+      return JSON.parse(content);
+    }
+  } catch {}
+  return {};
+}
+
+/**
+ * Saves configuration to ~/.dis-gravity/config.json with 0600 permissions
+ */
+function saveConfig(configData) {
+  try {
+    if (!fs.existsSync(CONFIG_DIR)) {
+      fs.mkdirSync(CONFIG_DIR, { recursive: true, mode: 0o700 });
+    }
+    const current = loadConfig();
+    const merged = { ...current, ...configData, updatedAt: new Date().toISOString() };
+    fs.writeFileSync(CONFIG_FILE, JSON.stringify(merged, null, 2), { encoding: 'utf8', mode: 0o600 });
+    try {
+      fs.chmodSync(CONFIG_FILE, 0o600);
+    } catch {}
+    return true;
+  } catch (err) {
+    throw new Error(`Failed to save config to ${CONFIG_FILE}: ${err.message}`);
+  }
+}
+
+/**
+ * Clears ~/.dis-gravity/config.json
+ */
+function clearConfig() {
+  try {
+    if (fs.existsSync(CONFIG_FILE)) {
+      fs.unlinkSync(CONFIG_FILE);
+      return true;
+    }
+  } catch {}
+  return false;
+}
+
+/**
+ * Opens a URL in the user's default browser across platforms
+ */
+function openBrowser(url) {
+  if (!url || typeof url !== 'string' || !/^https?:\/\/[^\s"<>`]+$/i.test(url)) return;
+  try {
+    if (process.platform === 'win32') {
+      cp.execFile('cmd.exe', ['/c', 'start', '', url], { windowsHide: true }, () => {});
+    } else if (process.platform === 'darwin') {
+      cp.execFile('open', [url], () => {});
+    } else {
+      cp.execFile('xdg-open', [url], () => {});
+    }
+  } catch {}
+}
+
+module.exports = {
+  DISCORD_API_BASE,
+  DISGRAVITY_PERMISSIONS,
+  GATEWAY_MESSAGE_CONTENT_LIMITED,
+  CONFIG_DIR,
+  CONFIG_FILE,
+  looksLikeBotToken,
+  buildInviteUrl,
+  discordRequest,
+  discordMultipartRequest,
+  inspectApplication,
+  ensureMessageContentIntent,
+  listUserGuilds,
+  waitForGuildJoin,
+  privateChannelOverwrites,
+  findOrCreateChannel,
+  sendWelcomeMessage,
+  loadConfig,
+  saveConfig,
+  clearConfig,
+  openBrowser
+};

package/manifest.json

@@ -0,0 +1,115 @@
+{
+  "manifest_version": "0.3",
+  "dxt_version": "0.3",
+  "name": "dis-gravity",
+  "display_name": "Dis-Gravity",
+  "version": "1.2.5",
+  "description": "Control Claude from Discord. Receive live notifications, approve agent tool calls in 38ms, render 2x Retina table cards, and execute remote commands directly from your phone.",
+  "author": {
+    "name": "Sambath Rethinakumar",
+    "url": "https://gitlab.com/rethinakumarsambath8"
+  },
+  "license": "MIT",
+  "homepage": "https://gitlab.com/rethinakumarsambath8/dis-gravity#readme",
+  "repository": {
+    "type": "git",
+    "url": "https://gitlab.com/rethinakumarsambath8/dis-gravity.git"
+  },
+  "icon": "media/mcp-icon.png",
+  "server": {
+    "type": "node",
+    "entry_point": "mcp/server.js",
+    "mcp_config": {
+      "command": "node",
+      "args": [
+        "${__dirname}/mcp/server.js"
+      ],
+      "env": {
+        "DISCORD_BOT_TOKEN": "${user_config.bot_token}",
+        "DISCORD_CHANNEL_ID": "${user_config.channel_id}",
+        "DISCORD_ALLOWED_USERS": "${user_config.allowed_users}"
+      }
+    }
+  },
+  "user_config": {
+    "bot_token": {
+      "type": "string",
+      "title": "Discord Bot Token",
+      "description": "Discord Bot Token from Discord Developer Portal (Only required field! Dis-Gravity auto-provisions intents, channels, and security)",
+      "default": "",
+      "sensitive": true,
+      "required": false
+    },
+    "channel_id": {
+      "type": "string",
+      "title": "Discord Channel ID (Optional)",
+      "description": "Discord Channel ID (Optional - Dis-Gravity automatically creates a private locked #dis-gravity channel if left blank)",
+      "default": "",
+      "sensitive": false,
+      "required": false
+    },
+    "allowed_users": {
+      "type": "string",
+      "title": "Allowed Discord User IDs",
+      "description": "Comma-separated Discord User IDs allowed to approve actions and run commands (optional)",
+      "default": "",
+      "sensitive": false,
+      "required": false
+    }
+  },
+  "tools": [
+    {
+      "name": "notify_discord",
+      "description": "Send a rich status notification, milestone update, or message to the user's linked Discord channel and mobile phone. Automatically renders Markdown tables into 2x Retina PNG cards."
+    },
+    {
+      "name": "ask_discord_approval",
+      "description": "Pause execution and request interactive 1-tap confirmation from the developer on their mobile Discord app. Halts until the developer taps [ Allow ] or [ Deny ]."
+    },
+    {
+      "name": "get_discord_bridge_status",
+      "description": "Retrieve current Dis-Gravity Discord connection status, active channel, and host system health."
+    },
+    {
+      "name": "run_workspace_command",
+      "description": "Execute a terminal shell command inside the workspace directory, with optional mobile Discord approval prompt for high-risk operations."
+    },
+    {
+      "name": "get_git_diff",
+      "description": "Retrieve current Git branch, short status summary, and unified diff stat for the active workspace."
+    },
+    {
+      "name": "render_table_card",
+      "description": "Convert structured tabular data (headers and rows) into a 2x Retina PNG card and deliver it to Discord."
+    },
+    {
+      "name": "read_workspace_file",
+      "description": "Safely inspect the content of a file within the workspace with path traversal protection."
+    },
+    {
+      "name": "list_workspace_tree",
+      "description": "View the directory tree structure of the workspace."
+    },
+    {
+      "name": "get_incoming_messages",
+      "description": "Retrieve recent messages and instructions sent from the developer in the linked Discord channel."
+    },
+    {
+      "name": "setup_discord_bridge",
+      "description": "Run automated frictionless onboarding wizard to configure Dis-Gravity. Validates bot token, auto-enables Message Content intent, opens OAuth invite link, auto-creates private #dis-gravity channel, auto-whitelists bot owner, and connects live bridge."
+    }
+  ],
+  "prompts": [
+    {
+      "name": "setup-discord-bridge",
+      "description": "Run the interactive 60-second setup wizard to connect Claude Desktop to Discord and your mobile phone.",
+      "arguments": ["bot_token"],
+      "text": "Run the Dis-Gravity Discord Bridge setup wizard using setup_discord_bridge with bot_token: ${arguments.bot_token}"
+    },
+    {
+      "name": "discord-bridge-status",
+      "description": "Check Dis-Gravity Discord bridge connection status, host health, and send a test notification to mobile.",
+      "text": "Check the Dis-Gravity Discord bridge status using get_discord_bridge_status and send a test notification using notify_discord."
+    }
+  ]
+}