@quantiya/codevibe-antigravity-plugin 1.0.0 → 1.0.1

package/dist/server.js

@@ -2826,6 +2826,7 @@ var McpServer = class {
     this.wrapperPid = options.wrapperPid ?? null;
     this.cliLogPath = options.cliLogPath ?? null;
     this.appSyncClient = options.appSyncClient ?? new import_codevibe_core4.AppSyncClient();
+    this.workingDirectory = process.cwd();
     this.approvalDetector = options.approvalDetector ?? new ApprovalDetector(options.detectorOptions);
     this.paneObserver = options.paneObserver ?? new TmuxPaneObserver();
     this.transcriptTailer = options.transcriptTailer ?? new TranscriptTailer();
@@ -3296,6 +3297,89 @@ var McpServer = class {
     }
   }
   // ─── Mobile → desktop flow ──────────────────────────────────────────────
+  /**
+   * Download an attachment from S3 to the working directory and return a
+   * relative path suitable for agy's `@./path` syntax (inherited from
+   * gemini-cli — Antigravity CLI is downstream of Gemini CLI).
+   *
+   * Mirrors codevibe-gemini-plugin's downloadAttachment() including the
+   * AppSync nested-isEncrypted fallback (subscriptions return null on
+   * nested object fields even when DynamoDB has the value).
+   *
+   * Returns the relative path (e.g. `./.codevibe-attachments/attachment-xxx.png`)
+   * on success, or null on failure (logged + swallowed — one bad attachment
+   * shouldn't block the whole prompt).
+   */
+  async downloadAttachment(attachment, sessionId, sessionKey, eventIsEncrypted) {
+    try {
+      const shouldDecrypt = attachment.isEncrypted ?? eventIsEncrypted ?? false;
+      logger.info("Downloading attachment", {
+        id: attachment.id,
+        type: attachment.type,
+        filename: attachment.filename,
+        s3Key: attachment.s3Key,
+        attachmentIsEncrypted: attachment.isEncrypted,
+        eventIsEncrypted,
+        shouldDecrypt
+      });
+      const { downloadUrl } = await this.appSyncClient.getAttachmentDownloadUrl(attachment.s3Key);
+      const response = await fetch(downloadUrl);
+      if (!response.ok) {
+        throw new Error(`Failed to download attachment: ${response.status} ${response.statusText}`);
+      }
+      let buffer = Buffer.from(await response.arrayBuffer());
+      if (shouldDecrypt && sessionKey) {
+        try {
+          logger.info("Decrypting attachment", { id: attachment.id });
+          buffer = import_codevibe_core4.cryptoService.decryptData(buffer, sessionKey);
+          logger.info("Attachment decrypted successfully", {
+            id: attachment.id,
+            decryptedSize: buffer.length
+          });
+        } catch (decryptError) {
+          logger.error("Failed to decrypt attachment:", { id: attachment.id, error: String(decryptError) });
+          throw new Error("Failed to decrypt attachment");
+        }
+      } else if (shouldDecrypt && !sessionKey) {
+        logger.warn("Cannot decrypt attachment - no session key available", { id: attachment.id });
+      }
+      const attachmentsDir = path5.join(this.workingDirectory, ".codevibe-attachments");
+      if (!fs5.existsSync(attachmentsDir)) {
+        fs5.mkdirSync(attachmentsDir, { recursive: true });
+      }
+      let extension = "";
+      if (attachment.filename) {
+        const ext = path5.extname(attachment.filename);
+        if (ext) extension = ext;
+      }
+      if (!extension) {
+        const mimeToExt = {
+          "image/jpeg": ".jpg",
+          "image/png": ".png",
+          "image/gif": ".gif",
+          "image/webp": ".webp",
+          "image/heic": ".heic",
+          "application/pdf": ".pdf"
+        };
+        extension = mimeToExt[attachment.type] || ".bin";
+      }
+      const filename = `attachment-${attachment.id}${extension}`;
+      const absolutePath = path5.join(attachmentsDir, filename);
+      fs5.writeFileSync(absolutePath, buffer);
+      const relativePath = `./.codevibe-attachments/${filename}`;
+      logger.info("Attachment saved to working directory", {
+        id: attachment.id,
+        sessionId,
+        absolutePath,
+        relativePath,
+        size: buffer.length
+      });
+      return relativePath;
+    } catch (error) {
+      logger.error("Failed to download attachment:", { id: attachment.id, error: String(error) });
+      return null;
+    }
+  }
   async handleMobileEvent(evt) {
     if (!this.started) return;
     if (this.sessionDisabled) return;
@@ -3318,10 +3402,42 @@ var McpServer = class {
     evt = decrypted;
     await this.markDelivered(evt);
     if (evt.type === import_codevibe_core4.EventType.USER_PROMPT) {
-      this.mobileDeduper.track(session.sessionId, evt.content);
-      const result = await this.promptResponder.sendFreeForm(evt.content);
+      let promptContent = evt.content;
+      const attachments = evt.attachments ?? [];
+      if (attachments.length > 0) {
+        logger.info("Downloading attachments for USER_PROMPT", {
+          sessionId: session.sessionId,
+          eventId: evt.eventId,
+          count: attachments.length
+        });
+        const attachmentPaths = [];
+        for (const attachment of attachments) {
+          const relPath = await this.downloadAttachment(
+            attachment,
+            session.sessionId,
+            session.sessionKey ?? null,
+            evt.isEncrypted
+          );
+          if (relPath) attachmentPaths.push(relPath);
+        }
+        if (attachmentPaths.length > 0) {
+          const fileReferences = attachmentPaths.map((p) => `@${p}`).join(" ");
+          if (promptContent) {
+            promptContent = `${fileReferences} ${promptContent}`;
+          } else {
+            promptContent = `${fileReferences} Please analyze the attached file(s).`;
+          }
+          logger.info("Prompt augmented with attachment paths", {
+            sessionId: session.sessionId,
+            eventId: evt.eventId,
+            attachmentCount: attachmentPaths.length
+          });
+        }
+      }
+      this.mobileDeduper.track(session.sessionId, promptContent);
+      const result = await this.promptResponder.sendFreeForm(promptContent);
       if (!result.ok) {
-        this.mobileDeduper.forget(session.sessionId, evt.content);
+        this.mobileDeduper.forget(session.sessionId, promptContent);
         logger.warn("sendFreeForm failed", { reason: result.reason, details: result.details });
         return;
       }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@quantiya/codevibe-antigravity-plugin",
-  "version": "1.0.0",
+  "version": "1.0.1",
   "description": "Control Antigravity CLI from your iPhone and Android — real-time sync, approve file edits, send prompts by voice. Part of CodeVibe.",
   "main": "dist/server.js",
   "bin": {