opencode-see-image 0.5.0 → 0.5.2

package/index.ts

@@ -38,20 +38,63 @@ function opencodeDbPath(): string {
   return path.join(dataDir, "opencode.db")
 }
 
-function resolveFromDb(filename: string): ResolvedImage | null {
+function resolveFromDb(
+  filename: string,
+  sessionID?: string,
+): ResolvedImage | null {
   const dbPath = opencodeDbPath()
   if (!fs.existsSync(dbPath)) return null
 
   try {
     const db = new Database(dbPath, { readonly: true })
-    const rows = db
-      .query(
-        `SELECT data FROM part
-         WHERE json_extract(data, '$.type') = 'file'
-           AND json_extract(data, '$.filename') = ?
-         ORDER BY time_created DESC LIMIT 1`,
-      )
-      .all(filename) as Array<{ data: string }>
+    let rows: Array<{ data: string }>
+
+    if (!filename || filename === "clipboard") {
+      // No filename: get the most recent file part, scoped to the current
+      // session if known so we don't grab an image pasted into a different
+      // conversation.
+      if (sessionID) {
+        rows = db
+          .query(
+            `SELECT data FROM part
+             WHERE session_id = ?
+               AND json_extract(data, '$.type') = 'file'
+               AND json_extract(data, '$.url') LIKE 'data:%'
+             ORDER BY time_created DESC LIMIT 1`,
+          )
+          .all(sessionID) as Array<{ data: string }>
+      } else {
+        rows = db
+          .query(
+            `SELECT data FROM part
+             WHERE json_extract(data, '$.type') = 'file'
+               AND json_extract(data, '$.url') LIKE 'data:%'
+             ORDER BY time_created DESC LIMIT 1`,
+          )
+          .all() as Array<{ data: string }>
+      }
+    } else {
+      if (sessionID) {
+        rows = db
+          .query(
+            `SELECT data FROM part
+             WHERE session_id = ?
+               AND json_extract(data, '$.type') = 'file'
+               AND json_extract(data, '$.filename') = ?
+             ORDER BY time_created DESC LIMIT 1`,
+          )
+          .all(sessionID, filename) as Array<{ data: string }>
+      } else {
+        rows = db
+          .query(
+            `SELECT data FROM part
+             WHERE json_extract(data, '$.type') = 'file'
+               AND json_extract(data, '$.filename') = ?
+             ORDER BY time_created DESC LIMIT 1`,
+          )
+          .all(filename) as Array<{ data: string }>
+      }
+    }
 
     db.close()
 
@@ -129,20 +172,19 @@ function resolveFromFilesystem(
   }
 }
 
-function resolveImage(name: string, cwd: string): ResolvedImage {
-  if (name === "clipboard") {
-    const fromDb = resolveFromDb("")
-    if (fromDb) return fromDb
-  }
-
-  const fromDb = resolveFromDb(name)
+function resolveImage(name: string, cwd: string, sessionID?: string): ResolvedImage {
+  // DB first: handles clipboard pastes, dragged files, screenshots.
+  // For "clipboard" or empty name, gets the most recent file part.
+  // Scoped to the current session if known.
+  const fromDb = resolveFromDb(name, sessionID)
   if (fromDb) return fromDb
 
+  // Filesystem fallback for files not yet in the DB.
   const fromFs = resolveFromFilesystem(name, cwd)
   if (fromFs) return fromFs
 
   throw new Error(
-    `see_image: could not find "${name}". Searched opencode DB and filesystem (cwd, ~/Desktop, ~/Downloads, temp). Pass an absolute filePath instead.`,
+    `see_image: could not find "${name}". Searched opencode DB and filesystem. Pass an absolute filePath instead.`,
   )
 }
 
@@ -405,7 +447,7 @@ const SeeImagePlugin: Plugin = async (ctx) => {
         ),
     },
     async execute(args, context) {
-      const resolved = resolveImage(args.filePath, context.directory)
+      const resolved = resolveImage(args.filePath, context.directory, context.sessionID)
 
       const prompt =
         args.question && args.question.trim().length > 0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "opencode-see-image",
-  "version": "0.5.0",
+  "version": "0.5.2",
   "description": "Give non-vision opencode models the ability to see images/screenshots by routing them to a vision-capable model (MiniMax M3 via opencode-go by default).",
   "type": "module",
   "main": "index.ts",