safety-agent-cli 0.1.0 → 0.1.1

package/dist/commands/scan.d.ts

@@ -0,0 +1 @@
+export declare function scanCommand(args: string[]): Promise<void>;

package/dist/commands/scan.js

@@ -0,0 +1,100 @@
+import { createClient } from "safety-agent";
+function showHelp() {
+    console.log("Usage: superagent scan --repo <url> [options]");
+    console.log("");
+    console.log("Scan a repository for AI agent-targeted attacks like repo poisoning and prompt injection");
+    console.log("");
+    console.log("Options:");
+    console.log("  --help              Show this help message");
+    console.log("  --repo <url>        Repository URL to scan (required)");
+    console.log("  --branch <ref>      Branch, tag, or commit to checkout");
+    console.log("  --model <id>        Model to use (default: anthropic/claude-sonnet-4-5)");
+    console.log("");
+    console.log("Examples:");
+    console.log("  superagent scan --repo https://github.com/user/repo");
+    console.log("  superagent scan --repo https://github.com/user/repo --branch main");
+    console.log("  superagent scan --repo https://github.com/user/repo --model openai/gpt-4o");
+}
+export async function scanCommand(args) {
+    // Check for --help flag
+    if (args.includes("--help") || args.includes("-h")) {
+        showHelp();
+        process.exit(0);
+    }
+    // Parse --repo flag
+    let repo;
+    const repoFlagIndex = args.indexOf("--repo");
+    if (repoFlagIndex !== -1) {
+        repo = args[repoFlagIndex + 1];
+        if (!repo || repo.startsWith("--")) {
+            console.error("❌ ERROR: --repo flag requires a URL");
+            process.exit(1);
+        }
+    }
+    if (!repo) {
+        console.error("❌ ERROR: --repo flag is required");
+        console.error("");
+        showHelp();
+        process.exit(1);
+    }
+    // Parse --branch flag
+    let branch;
+    const branchFlagIndex = args.indexOf("--branch");
+    if (branchFlagIndex !== -1) {
+        branch = args[branchFlagIndex + 1];
+        if (!branch || branch.startsWith("--")) {
+            console.error("❌ ERROR: --branch flag requires a value");
+            process.exit(1);
+        }
+    }
+    // Parse --model flag
+    let model;
+    const modelFlagIndex = args.indexOf("--model");
+    if (modelFlagIndex !== -1) {
+        model = args[modelFlagIndex + 1];
+        if (!model || model.startsWith("--")) {
+            console.error("❌ ERROR: --model flag requires a value");
+            process.exit(1);
+        }
+    }
+    // Ensure API keys are available
+    if (!process.env.SUPERAGENT_API_KEY) {
+        console.error("❌ ERROR: SUPERAGENT_API_KEY environment variable not set");
+        process.exit(2);
+    }
+    if (!process.env.DAYTONA_API_KEY) {
+        console.error("❌ ERROR: DAYTONA_API_KEY environment variable not set");
+        console.error("The scan command requires a Daytona API key for sandbox execution.");
+        process.exit(2);
+    }
+    // Create client instance
+    const client = createClient({
+        apiKey: process.env.SUPERAGENT_API_KEY,
+    });
+    console.error(`🔍 Scanning repository: ${repo}`);
+    if (branch) {
+        console.error(`   Branch: ${branch}`);
+    }
+    if (model) {
+        console.error(`   Model: ${model}`);
+    }
+    console.error("");
+    try {
+        const result = await client.scan({
+            repo,
+            branch,
+            model: model,
+        });
+        // Output result as JSON
+        const output = {
+            result: result.result,
+            usage: result.usage,
+        };
+        console.log(JSON.stringify(output, null, 2));
+        process.exit(0);
+    }
+    catch (error) {
+        console.error(`❌ Scan failed: ${error.message}`);
+        process.exit(2);
+    }
+}

package/dist/index.js

@@ -1,6 +1,7 @@
 #!/usr/bin/env node
 import { guardCommand } from './commands/guard.js';
 import { redactCommand } from './commands/redact.js';
+import { scanCommand } from './commands/scan.js';
 function showHelp() {
     console.log('Usage: superagent <command> [options]');
     console.log('');
@@ -9,6 +10,7 @@ function showHelp() {
     console.log('Commands:');
     console.log('  guard   Analyze prompts for security threats');
     console.log('  redact  Remove sensitive data from text');
+    console.log('  scan    Scan repositories for AI agent-targeted attacks');
     console.log('');
     console.log('Options:');
     console.log('  --help  Show help for a command');
@@ -16,8 +18,10 @@ function showHelp() {
     console.log('Examples:');
     console.log('  superagent guard --help');
     console.log('  superagent redact --help');
+    console.log('  superagent scan --help');
     console.log('  superagent guard "rm -rf /"');
     console.log('  superagent redact "My email is john@example.com"');
+    console.log('  superagent scan --repo https://github.com/user/repo');
 }
 const args = process.argv.slice(2);
 const command = args[0];
@@ -32,6 +36,9 @@ switch (command) {
     case 'redact':
         await redactCommand(args.slice(1));
         break;
+    case 'scan':
+        await scanCommand(args.slice(1));
+        break;
     default:
         console.error(`Unknown command: ${command}`);
         console.error('');

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "safety-agent-cli",
-  "version": "0.1.0",
+  "version": "0.1.1",
   "description": "CLI for Superagent - validate prompts and tool calls for security",
   "type": "module",
   "main": "./dist/index.js",