2026-04-21T18-34-26Z__iter_03

import * as vscode from "vscode";
import * as aws from "aws-sdk";

export function activate(context: vscode.ExtensionContext) {
  const disposable = vscode.commands.registerCommand(
    "qrcode-extension.openQRCodePanel",
    () => {
      // 1) Create the webview as before
      const panel = vscode.window.createWebviewPanel(
        "qrCodePanel",
        "QR Code + Data Viewer",
        vscode.ViewColumn.One,
        { enableScripts: true }
      );

      // 2) Generate a session ID
      const sessionId = Date.now().toString();

      // 3) Notify server about this session
      fetch("ht…

import * as vscode from "vscode";
import * as aws from "aws-sdk";

export function activate(context: vscode.ExtensionContext) {
  const disposable = vscode.commands.registerCommand(
    "qrcode-extension.openQRCodePanel",
    () => {
      // 1) Create the webview as before
      const panel = vscode.window.createWebviewPanel(
        "qrCodePanel",
        "QR Code + Data Viewer",
        vscode.ViewColumn.One,
        { enableScripts: true }
      );

      // 2) Generate a session ID
      const sessionId = Date.now().toString();

      // 3) Notify server about this session
      fetch("http://localhost:3000/init-session", {
        method: "POST",
        headers: { "Content-Type": "application/json" },
        body: JSON.stringify({ sessionId }),
      }).catch((err) => {
        console.error("Failed to init session with server:", err);
      });

      // 4) Provide minimal HTML that shows the QR code and polls for scan
      panel.webview.html = getWebviewContent(sessionId);

      // 5) Listen for messages FROM the webview
      panel.webview.onDidReceiveMessage(async (message) => {
        if (message.command === "scanned") {
          const { s3Url } = message;
          try {
            // Fetch the file from S3 (public or private)
            const fileContent = await fetchFromPublicS3Url(s3Url);

            // Extract the base name (no extension) from the S3 URL
            const baseName = pickBaseFilenameNoExtension(s3Url);
            // e.g. "22_generate_parenthesis" from "22_generate_parenthesis.py"

            // Create an untitled document with no extension
            // e.g. "untitled:22_generate_parenthesis"
            const untitledUri = vscode.Uri.parse(
              `untitled:${baseName}`
            );

            // Open the doc
            const doc = await vscode.workspace.openTextDocument(
              untitledUri
            );
            const editor = await vscode.window.showTextDocument(
              doc
            );
            // Paste the content
            await editor.edit((editBuilder) => {
              editBuilder.insert(
                new vscode.Position(0, 0),
                fileContent
              );
            });

            // Optionally close the panel
            panel.dispose();
          } catch (err) {
            console.error(
              "[Extension] Error fetching data from S3: ",
              err
            );
            vscode.window.showErrorMessage(
              "Failed to load file from S3."
            );
          }
        }
      });
    }
  );

  context.subscriptions.push(disposable);
}

// Fetch from public S3 URL
async function fetchFromPublicS3Url(url: string): Promise<string> {
  const response = await fetch(url);
  if (!response.ok) {
    throw new Error(`Fetch to ${url} returned status ${response.status}`);
  }
  return response.text();
}

/**
 * Extract the final filename from the S3 URL, then remove the extension.
 * e.g. "https://bucket.s3.amazonaws.com/foo/bar/22_generate_parenthesis.py"
 * -> "22_generate_parenthesis.py" -> "22_generate_parenthesis"
 */
function pickBaseFilenameNoExtension(url: string): string {
  try {
    const lastPart = url.split("/").pop() ?? "untitledFile";
    // Remove dot extension if present
    // This regex removes the last .xyz portion, if any
    const baseName = lastPart.replace(/\.[^/\.]*$/, "");
    return baseName || "untitledFile";
  } catch {
    return "untitledFile";
  }
}

// Minimal HTML
function getWebviewContent(sessionId: string): string {
  return `
<!DOCTYPE html>
<html>
<head>
  <meta charset="UTF-8" />
  <title>QR Code Demo</title>
  <script src="https://cdn.jsdelivr.net/npm/qrcode@1.5.1/build/qrcode.min.js"></script>
</head>
<body>
  <h2>My QR Code (Session: ${sessionId})</h2>
  <canvas id="qrcode"></canvas>
  <div id="status"></div>
  <script>
    const vscode = acquireVsCodeApi();
    const sessionId = "${sessionId}";

    QRCode.toCanvas(document.getElementById('qrcode'), sessionId, (err) => {
      if (err) console.error(err);
      console.log("QR code rendered!");
    });

    async function checkForScan() {
      try {
        const res = await fetch("http://localhost:3000/check-scan", {
          method: "POST",
          headers: { "Content-Type": "application/json" },
          body: JSON.stringify({ sessionId }),
        });
        const data = await res.json();
        if (data.scanned) {
          document.getElementById("status").innerText = "Scanned! Opening file...";
          vscode.postMessage({
            command: "scanned",
            s3Url: data.s3Url,
          });
        } else {
          document.getElementById("status").innerText = "Not scanned yet. Polling again in 5s...";
        }
      } catch (err) {
        console.error(err);
        document.getElementById("status").innerText = "Error polling server. Retrying in 5s...";
        setTimeout(checkForScan, 5000);
      }
    }

    checkForScan();
  </script>
</body>
</html>`;
}