shoud-cli 1.0.10 → 1.0.11

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "shoud-cli",
-  "version": "1.0.10",
+  "version": "1.0.11",
   "description": "SHOUD Terminal Agent: Give your computer a job.",
   "main": "bin/shoud.js",
   "bin": {

package/src/tools/index.js

@@ -112,4 +112,60 @@ function executeTool(toolName, input) {
   }
 }
 
+/**
+ * Read an image/video file and return a data URL for Gemini.
+ */
+function readImage(filePath) {
+  const safePath = sanitizePath(filePath);
+  try {
+    const stats = fs.statSync(safePath);
+    if (!stats.isFile()) throw new Error('Path is not a file.');
+    const buffer = fs.readFileSync(safePath);
+    const base64 = buffer.toString('base64');
+    const mimeType = getMimeType(safePath); // you need to define this helper
+    return `data:${mimeType};base64,${base64}`;
+  } catch (err) {
+    throw new Error(`Failed to read image: ${err.message}`);
+  }
+}
+
+// Helper: determine MIME type from file extension
+function getMimeType(filePath) {
+  const ext = path.extname(filePath).toLowerCase();
+  const map = {
+    '.jpg': 'image/jpeg',
+    '.jpeg': 'image/jpeg',
+    '.png': 'image/png',
+    '.gif': 'image/gif',
+    '.webp': 'image/webp',
+    '.mp4': 'video/mp4',
+    '.mov': 'video/quicktime',
+    '.avi': 'video/x-msvideo',
+    '.webm': 'video/webm',
+    '.mkv': 'video/x-matroska',
+    '.pdf': 'application/pdf',
+  };
+  return map[ext] || 'application/octet-stream';
+}
+
+// In executeTool, add:
+function executeTool(toolName, input) {
+  try {
+    switch (toolName) {
+      case 'execute_shell':
+        return executeShell(input.command);
+      case 'read_file':
+        return readFile(input.path);
+      case 'write_file':
+        return writeFile(input.path, input.content);
+      case 'read_image':   // new
+        return readImage(input.path);
+      default:
+        throw new Error(`Tool "${toolName}" is not supported.`);
+    }
+  } catch (error) {
+    return `Tool execution failed: ${error.message}`;
+  }
+}
+
 module.exports = { executeTool };