@phenixstar/talon 1.0.2 → 1.1.0

package/talon

@@ -20,13 +20,45 @@ else
 fi
 COMPOSE_FILE="$COMPOSE_BASE"
 
-# Load .env if present
+# Talon home directory
+TALON_HOME="${TALON_HOME:-$HOME/.talon}"
+
+# Load .env — check CWD first, then ~/.talon/
 if [ -f .env ]; then
   set -a
   source .env
   set +a
+elif [ -f "$TALON_HOME/.env" ]; then
+  set -a
+  source "$TALON_HOME/.env"
+  set +a
 fi
 
+# Resolve host paths for Docker volume mounts.
+# CWD-local takes priority over ~/.talon/ (project mode vs global mode).
+resolve_talon_paths() {
+  if [ -d "./configs" ]; then TALON_CONFIGS_DIR="$(pwd)/configs"
+  elif [ -d "$TALON_HOME/configs" ]; then TALON_CONFIGS_DIR="$TALON_HOME/configs"
+  else TALON_CONFIGS_DIR="$(pwd)/configs"; fi
+
+  if [ -d "./prompts" ]; then TALON_PROMPTS_DIR="$(pwd)/prompts"
+  else TALON_PROMPTS_DIR="$(cd "$(dirname "$0")" && pwd)/prompts"; fi
+
+  if [ -d "./audit-logs" ]; then TALON_AUDIT_DIR="$(pwd)/audit-logs"
+  elif [ -d "$TALON_HOME/audit-logs" ]; then TALON_AUDIT_DIR="$TALON_HOME/audit-logs"
+  else TALON_AUDIT_DIR="$TALON_HOME/audit-logs"; fi
+
+  if [ -d "./credentials" ]; then TALON_CREDENTIALS_DIR="$(pwd)/credentials"
+  elif [ -d "$TALON_HOME/credentials" ]; then TALON_CREDENTIALS_DIR="$TALON_HOME/credentials"
+  else TALON_CREDENTIALS_DIR="$TALON_HOME/credentials"; fi
+
+  if [ -d "./repos" ]; then TALON_REPOS_DIR="$(pwd)/repos"
+  elif [ -d "$TALON_HOME/repos" ]; then TALON_REPOS_DIR="$TALON_HOME/repos"
+  else TALON_REPOS_DIR="$(pwd)/repos"; fi
+
+  export TALON_CONFIGS_DIR TALON_PROMPTS_DIR TALON_AUDIT_DIR TALON_CREDENTIALS_DIR TALON_REPOS_DIR
+}
+
 show_help() {
   cat << 'EOF'
 
@@ -45,6 +77,8 @@ Usage:
   ./talon start URL=<url> REPO=<name>   Start a pentest workflow
   ./talon benchmark TARGET=<name>       Run benchmark and compute F1 metrics
   ./talon evolve GENERATIONS=<n>        Run N evolution generations on gene pool
+  ./talon tui <workflow-id>              Live TUI dashboard for a running workflow
+  ./talon tui --list                    List workspaces in TUI
   ./talon workspaces                    List all workspaces
   ./talon logs ID=<workflow-id>         Tail logs for a specific workflow
   ./talon stop                          Stop all containers
@@ -145,9 +179,15 @@ cmd_start() {
     exit 1
   fi
 
-  # Check for API key (Bedrock and router modes can bypass this)
+  # Resolve all host paths for Docker mounts
+  resolve_talon_paths
+
+  # Check for API key (Bedrock, Vertex, router, and custom base URL modes can bypass this)
   if [ -z "$ANTHROPIC_API_KEY" ] && [ -z "$CLAUDE_CODE_OAUTH_TOKEN" ]; then
-    if [ "$CLAUDE_CODE_USE_BEDROCK" = "1" ]; then
+    if [ -n "$ANTHROPIC_BASE_URL" ] && [ -n "$ANTHROPIC_AUTH_TOKEN" ]; then
+      # Custom base URL mode — use auth token as API key for SDK initialization
+      echo "Using custom base URL: $ANTHROPIC_BASE_URL"
+    elif [ "$CLAUDE_CODE_USE_BEDROCK" = "1" ]; then
       # Bedrock mode — validate required AWS credentials
       MISSING=""
       [ -z "$AWS_REGION" ] && MISSING="$MISSING AWS_REGION"
@@ -197,16 +237,19 @@ cmd_start() {
 
   # Determine container path for REPO
   # - If REPO is already a container path (/benchmarks/*, /repos/*), use as-is
-  # - Otherwise, treat as a folder name under ./repos/ (mounted at /repos in container)
+  # - Otherwise, check resolved repos dir (CWD/repos/ then ~/.talon/repos/)
   case "$REPO" in
     /benchmarks/*|/repos/*)
       CONTAINER_REPO="$REPO"
       ;;
     *)
-      if [ ! -d "./repos/$REPO" ]; then
-        echo "ERROR: Repository not found at ./repos/$REPO"
+      if [ ! -d "$TALON_REPOS_DIR/$REPO" ]; then
+        echo "ERROR: Repository not found"
+        echo "  Checked: ./repos/$REPO"
+        echo "  Checked: $TALON_HOME/repos/$REPO"
         echo ""
-        echo "Place your target repository under the ./repos/ directory"
+        echo "Add a repo:  talon repo add <git-url>"
+        echo "Or clone:    git clone <url> $TALON_REPOS_DIR/$REPO"
         exit 1
       fi
       CONTAINER_REPO="/repos/$REPO"
@@ -234,16 +277,21 @@ cmd_start() {
     export ANTHROPIC_AUTH_TOKEN="$TALON_ROUTER_KEY"
   fi
 
-  # Ensure audit-logs directory exists with write permissions for container user (UID 1001)
-  mkdir -p ./audit-logs ./credentials
-  chmod 777 ./audit-logs  # 777 required: container UID 1001 differs from host UID
+  # Ensure directories exist with write permissions for container user (UID 1001)
+  mkdir -p "$TALON_AUDIT_DIR" "$TALON_CREDENTIALS_DIR"
+  chmod 777 "$TALON_AUDIT_DIR"  # 777 required: container UID 1001 differs from host UID
 
   # Ensure repo deliverables directory is writable by container user (UID 1001)
-  if [ -d "./repos/$REPO" ]; then
-    mkdir -p "./repos/$REPO/deliverables"
-    chmod 777 "./repos/$REPO/deliverables"  # 777 required: container UID 1001 differs from host UID
+  if [ -d "$TALON_REPOS_DIR/$REPO" ]; then
+    mkdir -p "$TALON_REPOS_DIR/$REPO/deliverables"
+    chmod 777 "$TALON_REPOS_DIR/$REPO/deliverables"  # 777 required: container UID 1001 differs from host UID
   fi
 
+  # Log resolved paths
+  echo "Repos:      $TALON_REPOS_DIR"
+  echo "Configs:    $TALON_CONFIGS_DIR"
+  echo "Audit logs: $TALON_AUDIT_DIR"
+
   # Start router BEFORE main containers so worker can reach it on boot
   if [ "$ROUTER" = "true" ]; then
     if [ -z "$OPENAI_API_KEY" ] && [ -z "$OPENROUTER_API_KEY" ]; then
@@ -409,6 +457,30 @@ cmd_doctor() {
   node dist/cli/doctor.js "$@"
 }
 
+cmd_repo() {
+  if ! command -v node &>/dev/null; then
+    echo "Node.js is required. Install: https://nodejs.org/en/download"
+    exit 1
+  fi
+  if [ ! -f "dist/cli/repo-commands.js" ]; then
+    echo "Building Talon..."
+    npm install --silent && npm run build
+  fi
+  node dist/cli/repo-commands.js "$@"
+}
+
+cmd_tui() {
+  if ! command -v node &>/dev/null; then
+    echo "Node.js is required for TUI. Install: https://nodejs.org/en/download"
+    exit 1
+  fi
+  if [ ! -f "dist/tui/tui-entry.js" ]; then
+    echo "Building Talon..."
+    npm install --silent && npm run build
+  fi
+  node dist/tui/tui-entry.js "$@"
+}
+
 cmd_stop() {
   parse_args "$@"
 
@@ -429,6 +501,10 @@ case "${1:-help}" in
     shift
     cmd_doctor "$@"
     ;;
+  repo)
+    shift
+    cmd_repo "$@"
+    ;;
   start)
     shift
     cmd_start "$@"
@@ -445,6 +521,10 @@ case "${1:-help}" in
     shift
     cmd_evolve "$@"
     ;;
+  tui)
+    shift
+    cmd_tui "$@"
+    ;;
   workspaces)
     shift
     cmd_workspaces