@withbuddi/buddi 0.1.0-pre.45 → 0.1.0-pre.46

package/extension/background.js

@@ -910,6 +910,7 @@ function fieldFacts(read) {
 function fieldObservation(read) {
   return { field: fieldFacts(read) };
 }
+var DOWNLOAD_COMMANDS = /* @__PURE__ */ new Set(["navigate", "click", "press", "select", "fill"]);
 var BrowserCommands = class {
   #chrome;
   #sessions = /* @__PURE__ */ new Map();
@@ -936,6 +937,7 @@ var BrowserCommands = class {
   #ownerWaitMs;
   constructor(chrome2, options = {}) {
     this.#chrome = chrome2;
+    this.downloads = options.downloads;
     this.logins = new HeldLogins(options.onLogin ?? (() => void 0));
     chrome2.tabs.onRemoved?.addListener((tabId) => this.logins.clearTab(tabId));
     this.#onEvent = options.onEvent ?? (() => void 0);
@@ -959,8 +961,32 @@ var BrowserCommands = class {
     this.#uuid = options.uuid ?? (() => crypto.randomUUID());
     this.#contentFile = options.contentFile ?? "content.js";
   }
+  /** Downloads an agent's command starts, noticed while it runs (and a few seconds after). */
+  downloads;
+  /** The addresses of a session's tabs, for telling its downloads from the owner's. */
+  async #tabUrls(name) {
+    const session = this.#sessions.get(name);
+    if (!session) return [];
+    const urls = [];
+    for (const tabId of session.tabs.values()) {
+      const tab = await this.#chrome.tabs.get(tabId).catch(() => void 0);
+      if (tab?.url) urls.push(tab.url);
+    }
+    return urls;
+  }
   async run(command, cancel = new Cancellation()) {
     cancel.check();
+    if (this.downloads?.attached && !command.owner && DOWNLOAD_COMMANDS.has(command.name)) {
+      const action = this.downloads.begin(command.session, await this.#tabUrls(command.session));
+      try {
+        return await this.#dispatch(command, cancel);
+      } finally {
+        this.downloads.end(action, await this.#tabUrls(command.session).catch(() => []));
+      }
+    }
+    return this.#dispatch(command, cancel);
+  }
+  async #dispatch(command, cancel) {
     switch (command.name) {
       case "navigate":
         return this.#navigate(command, cancel);
@@ -1017,6 +1043,7 @@ var BrowserCommands = class {
     this.#held.clear();
     this.#heldLogins.clear();
     this.logins.clear();
+    this.downloads?.forget();
   }
   /* ---- take-over in place ---- */
   /**
@@ -1365,6 +1392,7 @@ var BrowserCommands = class {
     }
     this.#sessions.delete(command.session);
     this.#refs.delete(command.session);
+    this.downloads?.forget(command.session);
     return {};
   }
   /** The owner may have closed or adopted an agent tab; a session forgets those before it reports. */
@@ -1953,6 +1981,126 @@ function heldLoginFacts(raw) {
   return { never: sites, saved: kept };
 }
 
+// src/downloads.ts
+var DOWNLOAD_GRACE_MS = 15e3;
+var DOWNLOADS_PERMISSION = "downloads";
+function originOf(url) {
+  if (!url) return void 0;
+  try {
+    const parsed = new URL(url);
+    if (parsed.protocol === "blob:") return new URL(parsed.pathname).origin;
+    return parsed.protocol === "http:" || parsed.protocol === "https:" ? parsed.origin : void 0;
+  } catch {
+    return void 0;
+  }
+}
+function basename(file) {
+  return file.split(/[\\/]/).pop() || "download";
+}
+var AgentDownloads = class {
+  #actions = /* @__PURE__ */ new Map();
+  #next = 1;
+  #claimed = /* @__PURE__ */ new Map();
+  #api;
+  #send;
+  #now;
+  constructor(options) {
+    this.#send = options.send;
+    this.#now = options.now ?? (() => Date.now());
+  }
+  /** Chrome's downloads API, once the owner granted it. Listening starts here, once. */
+  attach(api) {
+    if (!api || this.#api) return;
+    this.#api = api;
+    api.onCreated.addListener((item) => this.created(item));
+    api.onChanged.addListener((delta) => {
+      void this.changed(delta).catch(() => void 0);
+    });
+  }
+  get attached() {
+    return this.#api !== void 0;
+  }
+  /** An agent command starts in this session, whose tabs are on these addresses. Returns the command's handle for `end`. */
+  begin(session, urls) {
+    this.#expire();
+    const id = this.#next++;
+    const origins = /* @__PURE__ */ new Set();
+    for (const url of urls) {
+      const origin = originOf(url);
+      if (origin) origins.add(origin);
+    }
+    this.#actions.set(id, { session, origins, running: true, until: 0, at: this.#now() });
+    return id;
+  }
+  /** That command ended; its tabs may have moved on to these. Its downloads still count for a few seconds, its own. */
+  end(id, urls = []) {
+    const action = this.#actions.get(id);
+    if (!action) return;
+    for (const url of urls) {
+      const origin = originOf(url);
+      if (origin) action.origins.add(origin);
+    }
+    action.running = false;
+    action.until = this.#now() + DOWNLOAD_GRACE_MS;
+  }
+  /** Commands whose few seconds are over count for nothing any more. */
+  #expire() {
+    const now = this.#now();
+    for (const [id, action] of this.#actions) if (!action.running && now > action.until) this.#actions.delete(id);
+  }
+  /** The session ended or the socket went: nothing more is its. */
+  forget(session) {
+    if (session === void 0) {
+      this.#actions.clear();
+      this.#claimed.clear();
+      return;
+    }
+    for (const [id, action] of this.#actions) if (action.session === session) this.#actions.delete(id);
+    for (const [id, owner] of this.#claimed) if (owner === session) this.#claimed.delete(id);
+  }
+  /** Which session's command this download belongs to, if any. */
+  #owner(item) {
+    if (item.byExtensionId) return void 0;
+    const origin = originOf(item.referrer) ?? originOf(item.finalUrl ?? item.url);
+    if (!origin) return void 0;
+    this.#expire();
+    let best;
+    for (const action of this.#actions.values()) {
+      if (!action.origins.has(origin)) continue;
+      if (!best || action.at > best.at) best = action;
+    }
+    return best?.session;
+  }
+  created(item) {
+    const session = this.#owner(item);
+    if (session) this.#claimed.set(item.id, session);
+  }
+  async changed(delta) {
+    const session = this.#claimed.get(delta.id);
+    const state = delta.state?.current;
+    if (!session || !state) return;
+    if (state === "interrupted") {
+      this.#claimed.delete(delta.id);
+      return;
+    }
+    if (state !== "complete") return;
+    this.#claimed.delete(delta.id);
+    const [item] = await (this.#api?.search({ id: delta.id }) ?? Promise.resolve([]));
+    if (!item?.filename) return;
+    const url = item.finalUrl || item.url || "";
+    const size = item.fileSize !== void 0 && item.fileSize >= 0 ? item.fileSize : item.totalBytes !== void 0 && item.totalBytes >= 0 ? item.totalBytes : void 0;
+    this.#send({
+      type: "download",
+      session,
+      path: item.filename,
+      filename: basename(item.filename),
+      url,
+      ...item.mime ? { mime: item.mime } : {},
+      ...size !== void 0 ? { size } : {}
+    });
+  }
+};
+
 // src/external.ts
 function isLoopbackOrigin(origin) {
   if (!origin) return false;
@@ -2028,7 +2176,15 @@ function flush(open) {
     }
   }
 }
-var commands = new BrowserCommands(chrome, { onFrame: send, onEvent: send, onLogin: send });
+var downloads = new AgentDownloads({ send });
+var attachDownloads = () => {
+  if (chrome.downloads) downloads.attach(chrome.downloads);
+};
+attachDownloads();
+chrome.permissions?.onAdded?.addListener((added) => {
+  if (added.permissions?.includes(DOWNLOADS_PERMISSION)) attachDownloads();
+});
+var commands = new BrowserCommands(chrome, { onFrame: send, onEvent: send, onLogin: send, downloads });
 var attempt = 0;
 var timer;
 var silence;

package/extension/manifest.json

@@ -1,7 +1,7 @@
 {
   "manifest_version": 3,
   "name": "buddi",
-  "version": "0.1.0.45",
+  "version": "0.1.0.46",
   "description": "Lets your buddi work in background tabs of this browser. The sites it opens are the ones you ask it for; it opens none on its own.",
   "minimum_chrome_version": "116",
   "key": "MIIBIjANBgkqhkiG9w0BAQEFAAOCAQ8AMIIBCgKCAQEAwR4JT/OlKKPcZQ33euO9mSv7RrSIWNVanldhT2Uzc9YfzF/X0lg5NfXLc+G/UGFjB8v+uMCbpMhRBZ2SVwDHQH3buv3VEUzuOIDTPHgv99T5lq5imZrDxmIE+EL/fWrsjhcXaIB4e4fn0tJqw6ivJUGmYpG5XhWn65nlLniHfFBKMpZ/y7bVnvfzAgMQUa13qTAVxPTpMRtNl/flQq4Q/vWYm28H7W7OzygDV+/4X2azUrB8yPk67QsipbdDLxiX2GMxjXuKK6iD5CQiytoxTsAqUzB886HbOZsxJfQpRIjrUgBmGjLyaKuz617ZGJW9zS23LmnlL6mkX3+ggca1AQIDAQAB",
@@ -13,6 +13,9 @@
     "storage",
     "alarms"
   ],
+  "optional_permissions": [
+    "downloads"
+  ],
   "host_permissions": [
     "http://*/*",
     "https://*/*"
@@ -42,5 +45,5 @@
     "48": "icons/48.png",
     "128": "icons/128.png"
   },
-  "version_name": "0.1.0-pre.45"
+  "version_name": "0.1.0-pre.46"
 }

package/extension/popup.html

@@ -46,7 +46,11 @@
         <p class="fact" id="paired-to"></p>
         <p class="quiet" id="working">Nothing open right now.</p>
       </div>
+      <div class="facts" id="downloads-box" hidden>
+        <p class="quiet" id="downloads-note">Let agents hand buddi the files they download in your Chrome, like a bank statement. Only downloads an agent's click starts are read; yours are never touched.</p>
+      </div>
       <div class="actions">
+        <button class="btn" id="allow-downloads" type="button" hidden>Allow downloads</button>
         <button class="btn" id="open-buddi" type="button">Open buddi</button>
       </div>
       <footer class="foot">

package/extension/popup.js

@@ -1,3 +1,6 @@
+// src/downloads.ts
+var DOWNLOADS_PERMISSION = "downloads";
+
 // src/popup.ts
 var SETTINGS_HASH = "#/settings/computer";
 var PILL = {
@@ -125,6 +128,19 @@ function wireCopy(button, read) {
     })();
   });
 }
+async function wireDownloads(doc, permissions) {
+  const box = byId(doc, "downloads-box");
+  const button = byId(doc, "allow-downloads");
+  if (!permissions || !box || !button) return;
+  const show = (granted) => {
+    box.hidden = granted;
+    button.hidden = granted;
+  };
+  show(await permissions.contains({ permissions: [DOWNLOADS_PERMISSION] }).catch(() => true));
+  button.addEventListener("click", () => {
+    void permissions.request({ permissions: [DOWNLOADS_PERMISSION] }).then(show, () => void 0);
+  });
+}
 async function countBuddiTabs(api) {
   try {
     const groups = new Set((await api.tabGroups.query({ title: "buddi" })).map((group) => group.id));
@@ -229,6 +245,7 @@ async function main() {
     });
   });
   byId(document, "open-buddi").addEventListener("click", () => open(""));
+  void wireDownloads(document, chrome.permissions);
   byId(document, "forget").addEventListener("click", async () => {
     const answer = await ask({ type: "buddi-forget" });
     if (answer?.state) {
@@ -253,5 +270,6 @@ export {
   triesApp,
   viewOf,
   wireCopy,
+  wireDownloads,
   workingLine
 };