@fifine/aim-studio 0.0.1

package/dist/templates/aim/scripts-shell-archive/common/registry.sh

@@ -0,0 +1,247 @@
+#!/bin/bash
+# Registry utility functions for multi-agent pipeline
+#
+# Usage: source this file in other scripts
+#   source "$(dirname "$0")/common/registry.sh"
+#
+# Provides:
+#   registry_get_file        - Get registry file path
+#   registry_get_agent_by_id - Find agent by ID
+#   registry_get_agent_by_worktree - Find agent by worktree path
+#   registry_get_task_dir - Get task dir for a worktree
+#   registry_remove_by_id    - Remove agent by ID
+#   registry_remove_by_worktree - Remove agent by worktree path
+#   registry_add_agent       - Add agent to registry
+
+# Ensure dependencies are loaded
+if ! type get_repo_root &>/dev/null; then
+  echo "Error: paths.sh must be sourced before registry.sh" >&2
+  exit 1
+fi
+
+if ! type get_agents_dir &>/dev/null; then
+  echo "Error: developer.sh must be sourced before registry.sh" >&2
+  exit 1
+fi
+
+# =============================================================================
+# Registry File Access
+# =============================================================================
+
+# Get registry file path
+# Args: [repo_root]
+# Returns: path to registry.json
+registry_get_file() {
+  local repo_root="${1:-$(get_repo_root)}"
+  local agents_dir=$(get_agents_dir "$repo_root")
+  echo "${agents_dir}/registry.json"
+}
+
+# Ensure registry file exists with valid structure
+# Args: [repo_root]
+_ensure_registry() {
+  local repo_root="${1:-$(get_repo_root)}"
+  local registry_file=$(registry_get_file "$repo_root")
+  local agents_dir=$(dirname "$registry_file")
+
+  mkdir -p "$agents_dir"
+
+  if [[ ! -f "$registry_file" ]]; then
+    echo '{"agents":[]}' > "$registry_file"
+  fi
+}
+
+# =============================================================================
+# Agent Lookup
+# =============================================================================
+
+# Get agent by ID
+# Args: agent_id, [repo_root]
+# Returns: agent JSON object (compact), or empty if not found
+registry_get_agent_by_id() {
+  local agent_id="$1"
+  local repo_root="${2:-$(get_repo_root)}"
+  local registry_file=$(registry_get_file "$repo_root")
+
+  if [[ ! -f "$registry_file" ]]; then
+    return 1
+  fi
+
+  local agent=$(jq -c --arg id "$agent_id" '.agents[] | select(.id == $id)' "$registry_file" 2>/dev/null)
+
+  if [[ -n "$agent" ]] && [[ "$agent" != "null" ]]; then
+    echo "$agent"
+    return 0
+  fi
+
+  return 1
+}
+
+# Get agent by worktree path
+# Args: worktree_path, [repo_root]
+# Returns: agent JSON object (compact), or empty if not found
+registry_get_agent_by_worktree() {
+  local worktree_path="$1"
+  local repo_root="${2:-$(get_repo_root)}"
+  local registry_file=$(registry_get_file "$repo_root")
+
+  if [[ ! -f "$registry_file" ]]; then
+    return 1
+  fi
+
+  local agent=$(jq -c --arg path "$worktree_path" '.agents[] | select(.worktree_path == $path)' "$registry_file" 2>/dev/null)
+
+  if [[ -n "$agent" ]] && [[ "$agent" != "null" ]]; then
+    echo "$agent"
+    return 0
+  fi
+
+  return 1
+}
+
+# Search agent by ID or task_dir containing search term
+# Args: search_term, [repo_root]
+# Returns: first matching agent JSON object (compact), or empty if not found
+registry_search_agent() {
+  local search="$1"
+  local repo_root="${2:-$(get_repo_root)}"
+  local registry_file=$(registry_get_file "$repo_root")
+
+  if [[ ! -f "$registry_file" ]]; then
+    return 1
+  fi
+
+  local agent=$(jq -c --arg search "$search" \
+    '[.agents[] | select(.id == $search or (.task_dir | contains($search)))] | first' \
+    "$registry_file" 2>/dev/null)
+
+  if [[ -n "$agent" ]] && [[ "$agent" != "null" ]]; then
+    echo "$agent"
+    return 0
+  fi
+
+  return 1
+}
+
+# Get task directory for a worktree
+# Args: worktree_path, [repo_root]
+# Returns: task_dir value, or empty if not found
+registry_get_task_dir() {
+  local worktree_path="$1"
+  local repo_root="${2:-$(get_repo_root)}"
+  local registry_file=$(registry_get_file "$repo_root")
+
+  if [[ ! -f "$registry_file" ]]; then
+    return 1
+  fi
+
+  local task_dir=$(jq -r --arg path "$worktree_path" \
+    '.agents[] | select(.worktree_path == $path) | .task_dir' \
+    "$registry_file" 2>/dev/null)
+
+  if [[ -n "$task_dir" ]] && [[ "$task_dir" != "null" ]]; then
+    echo "$task_dir"
+    return 0
+  fi
+
+  return 1
+}
+
+# =============================================================================
+# Agent Modification
+# =============================================================================
+
+# Remove agent by ID
+# Args: agent_id, [repo_root]
+# Returns: 0 on success
+registry_remove_by_id() {
+  local agent_id="$1"
+  local repo_root="${2:-$(get_repo_root)}"
+  local registry_file=$(registry_get_file "$repo_root")
+
+  if [[ ! -f "$registry_file" ]]; then
+    return 0
+  fi
+
+  local updated=$(jq --arg id "$agent_id" \
+    '.agents = [.agents[] | select(.id != $id)]' \
+    "$registry_file")
+
+  echo "$updated" | jq '.' > "$registry_file"
+  return 0
+}
+
+# Remove agent by worktree path
+# Args: worktree_path, [repo_root]
+# Returns: 0 on success
+registry_remove_by_worktree() {
+  local worktree_path="$1"
+  local repo_root="${2:-$(get_repo_root)}"
+  local registry_file=$(registry_get_file "$repo_root")
+
+  if [[ ! -f "$registry_file" ]]; then
+    return 0
+  fi
+
+  local updated=$(jq --arg path "$worktree_path" \
+    '.agents = [.agents[] | select(.worktree_path != $path)]' \
+    "$registry_file")
+
+  echo "$updated" | jq '.' > "$registry_file"
+  return 0
+}
+
+# Add agent to registry (replaces if same ID exists)
+# Args: agent_id, worktree_path, pid, task_dir, [repo_root]
+# Returns: 0 on success
+registry_add_agent() {
+  local agent_id="$1"
+  local worktree_path="$2"
+  local pid="$3"
+  local task_dir="$4"
+  local repo_root="${5:-$(get_repo_root)}"
+
+  _ensure_registry "$repo_root"
+  local registry_file=$(registry_get_file "$repo_root")
+
+  local started_at=$(date -Iseconds)
+
+  # Remove existing agent with same ID
+  local registry=$(jq --arg id "$agent_id" \
+    '.agents = [.agents[] | select(.id != $id)]' \
+    "$registry_file")
+
+  # Create new agent record
+  local new_agent=$(jq -n \
+    --arg id "$agent_id" \
+    --arg worktree "$worktree_path" \
+    --arg pid "$pid" \
+    --arg started_at "$started_at" \
+    --arg task_dir "$task_dir" \
+    '{
+      id: $id,
+      worktree_path: $worktree,
+      pid: ($pid | tonumber),
+      started_at: $started_at,
+      task_dir: $task_dir
+    }')
+
+  # Add to registry
+  echo "$registry" | jq --argjson agent "$new_agent" '.agents += [$agent]' > "$registry_file"
+  return 0
+}
+
+# List all agents
+# Args: [repo_root]
+# Returns: JSON array of agents
+registry_list_agents() {
+  local repo_root="${1:-$(get_repo_root)}"
+  local registry_file=$(registry_get_file "$repo_root")
+
+  if [[ ! -f "$registry_file" ]]; then
+    echo '[]'
+    return 0
+  fi
+
+  jq '.agents' "$registry_file"
+}

package/dist/templates/aim/scripts-shell-archive/common/task-queue.sh

@@ -0,0 +1,142 @@
+#!/bin/bash
+# Task queue utility functions
+#
+# Usage: source this file in other scripts
+#   source "$(dirname "$0")/common/task-queue.sh"
+#
+# Provides:
+#   list_pending_tasks     - List tasks with pending status
+#   get_task_stats         - Get P0/P1/P2/P3 counts
+
+# Ensure paths.sh is loaded
+if ! type get_repo_root &>/dev/null; then
+  echo "Error: paths.sh must be sourced before task-queue.sh" >&2
+  exit 1
+fi
+
+# =============================================================================
+# Public Functions
+# =============================================================================
+
+# List tasks by status
+# Args: [filter_status]
+# Output: formatted list to stdout
+list_tasks_by_status() {
+  local filter_status="${1:-}"
+  local repo_root="${2:-$(get_repo_root)}"
+
+  local tasks_dir=$(get_tasks_dir "$repo_root")
+
+  if [[ ! -d "$tasks_dir" ]]; then
+    return 0
+  fi
+
+  for d in "$tasks_dir"/*/; do
+    if [[ -d "$d" ]] && [[ "$(basename "$d")" != "archive" ]]; then
+      local task_json="$d/$FILE_TASK_JSON"
+      if [[ -f "$task_json" ]]; then
+        local id=$(jq -r '.id' "$task_json")
+        local title=$(jq -r '.title // .name' "$task_json")
+        local priority=$(jq -r '.priority // "P2"' "$task_json")
+        local status=$(jq -r '.status // "planning"' "$task_json")
+        local assignee=$(jq -r '.assignee // "-"' "$task_json")
+
+        # Apply filter
+        if [[ -n "$filter_status" ]] && [[ "$status" != "$filter_status" ]]; then
+          continue
+        fi
+
+        echo "$priority|$id|$title|$status|$assignee"
+      fi
+    fi
+  done
+}
+
+# List pending tasks
+list_pending_tasks() {
+  list_tasks_by_status "planning" "$@"
+}
+
+# List tasks assigned to a specific developer
+# Args: developer_name, [filter_status], [repo_root]
+# Output: formatted list to stdout
+list_tasks_by_assignee() {
+  local assignee="$1"
+  local filter_status="${2:-}"
+  local repo_root="${3:-$(get_repo_root)}"
+
+  local tasks_dir=$(get_tasks_dir "$repo_root")
+
+  if [[ ! -d "$tasks_dir" ]]; then
+    return 0
+  fi
+
+  for d in "$tasks_dir"/*/; do
+    if [[ -d "$d" ]] && [[ "$(basename "$d")" != "archive" ]]; then
+      local task_json="$d/$FILE_TASK_JSON"
+      if [[ -f "$task_json" ]]; then
+        local id=$(jq -r '.id' "$task_json")
+        local title=$(jq -r '.title // .name' "$task_json")
+        local priority=$(jq -r '.priority // "P2"' "$task_json")
+        local status=$(jq -r '.status // "planning"' "$task_json")
+        local task_assignee=$(jq -r '.assignee // "-"' "$task_json")
+
+        # Apply assignee filter
+        if [[ "$task_assignee" != "$assignee" ]]; then
+          continue
+        fi
+
+        # Apply status filter
+        if [[ -n "$filter_status" ]] && [[ "$status" != "$filter_status" ]]; then
+          continue
+        fi
+
+        echo "$priority|$id|$title|$status|$task_assignee"
+      fi
+    fi
+  done
+}
+
+# List my tasks (current developer)
+# Args: [filter_status], [repo_root]
+list_my_tasks() {
+  local filter_status="${1:-}"
+  local repo_root="${2:-$(get_repo_root)}"
+  local developer=$(get_developer "$repo_root")
+
+  if [[ -z "$developer" ]]; then
+    echo "Error: Developer not set" >&2
+    return 1
+  fi
+
+  list_tasks_by_assignee "$developer" "$filter_status" "$repo_root"
+}
+
+# Get task statistics
+# Output: "P0:N P1:N P2:N P3:N Total:N"
+get_task_stats() {
+  local repo_root="${1:-$(get_repo_root)}"
+  local tasks_dir=$(get_tasks_dir "$repo_root")
+
+  local p0=0 p1=0 p2=0 p3=0 total=0
+
+  if [[ -d "$tasks_dir" ]]; then
+    for d in "$tasks_dir"/*/; do
+      if [[ -d "$d" ]] && [[ "$(basename "$d")" != "archive" ]]; then
+        local task_json="$d/$FILE_TASK_JSON"
+        if [[ -f "$task_json" ]]; then
+          local priority=$(jq -r '.priority // "P2"' "$task_json" 2>/dev/null)
+          case "$priority" in
+            P0) ((p0++)) ;;
+            P1) ((p1++)) ;;
+            P2) ((p2++)) ;;
+            P3) ((p3++)) ;;
+          esac
+          ((total++))
+        fi
+      fi
+    done
+  fi
+
+  echo "P0:$p0 P1:$p1 P2:$p2 P3:$p3 Total:$total"
+}

package/dist/templates/aim/scripts-shell-archive/common/task-utils.sh

@@ -0,0 +1,151 @@
+#!/bin/bash
+# Task utility functions
+#
+# Usage: source this file in other scripts
+#   source "$(dirname "$0")/common/task-utils.sh"
+#
+# Provides:
+#   is_safe_task_path      - Validate task path is safe to operate on
+#   find_task_by_name      - Find task directory by name
+#   archive_task_dir       - Archive task to monthly directory
+
+# Ensure dependencies are loaded
+if ! type get_repo_root &>/dev/null; then
+  echo "Error: paths.sh must be sourced before task-utils.sh" >&2
+  exit 1
+fi
+
+# =============================================================================
+# Path Safety
+# =============================================================================
+
+# Check if a relative task path is safe to operate on
+# Args: task_path (relative), repo_root
+# Returns: 0 if safe, 1 if dangerous
+# Outputs: error message to stderr if unsafe
+is_safe_task_path() {
+  local task_path="$1"
+  local repo_root="${2:-$(get_repo_root)}"
+
+  # Check empty or null
+  if [[ -z "$task_path" ]] || [[ "$task_path" = "null" ]]; then
+    echo "Error: empty or null task path" >&2
+    return 1
+  fi
+
+  # Reject absolute paths
+  if [[ "$task_path" = /* ]]; then
+    echo "Error: absolute path not allowed: $task_path" >&2
+    return 1
+  fi
+
+  # Reject ".", "..", paths starting with "./" or "../", or containing ".."
+  if [[ "$task_path" = "." ]] || [[ "$task_path" = ".." ]] || \
+     [[ "$task_path" = "./" ]] || [[ "$task_path" == ./* ]] || \
+     [[ "$task_path" == *".."* ]]; then
+    echo "Error: path traversal not allowed: $task_path" >&2
+    return 1
+  fi
+
+  # Final check: ensure resolved path is not the repo root
+  local abs_path="${repo_root}/${task_path}"
+  if [[ -e "$abs_path" ]]; then
+    local resolved=$(realpath "$abs_path" 2>/dev/null)
+    local root_resolved=$(realpath "$repo_root" 2>/dev/null)
+    if [[ "$resolved" = "$root_resolved" ]]; then
+      echo "Error: path resolves to repo root: $task_path" >&2
+      return 1
+    fi
+  fi
+
+  return 0
+}
+
+# =============================================================================
+# Task Lookup
+# =============================================================================
+
+# Find task directory by name (exact or suffix match)
+# Args: task_name, tasks_dir
+# Returns: absolute path to task directory, or empty if not found
+find_task_by_name() {
+  local task_name="$1"
+  local tasks_dir="$2"
+
+  if [[ -z "$task_name" ]] || [[ -z "$tasks_dir" ]]; then
+    return 1
+  fi
+
+  # Try exact match first
+  local task_dir=$(find "$tasks_dir" -maxdepth 1 -type d -name "${task_name}" 2>/dev/null | head -1)
+
+  # Try suffix match (e.g., "my-task" matches "01-21-my-task")
+  if [[ -z "$task_dir" ]]; then
+    task_dir=$(find "$tasks_dir" -maxdepth 1 -type d -name "*-${task_name}" 2>/dev/null | head -1)
+  fi
+
+  if [[ -n "$task_dir" ]] && [[ -d "$task_dir" ]]; then
+    echo "$task_dir"
+    return 0
+  fi
+
+  return 1
+}
+
+# =============================================================================
+# Archive Operations
+# =============================================================================
+
+# Archive a task directory to archive/{YYYY-MM}/
+# Args: task_dir_abs, [repo_root]
+# Returns: 0 on success, 1 on error
+# Outputs: archive destination path
+archive_task_dir() {
+  local task_dir_abs="$1"
+  local repo_root="${2:-$(get_repo_root)}"
+
+  if [[ ! -d "$task_dir_abs" ]]; then
+    echo "Error: task directory not found: $task_dir_abs" >&2
+    return 1
+  fi
+
+  # Get tasks directory (parent of the task)
+  local tasks_dir=$(dirname "$task_dir_abs")
+  local archive_dir="$tasks_dir/archive"
+  local year_month=$(date +%Y-%m)
+  local month_dir="$archive_dir/$year_month"
+
+  # Create archive directory
+  mkdir -p "$month_dir"
+
+  # Move task to archive
+  local task_name=$(basename "$task_dir_abs")
+  mv "$task_dir_abs" "$month_dir/"
+
+  # Output the destination
+  echo "$month_dir/$task_name"
+  return 0
+}
+
+# Complete archive workflow: archive directory
+# Args: task_dir_abs, [repo_root]
+# Returns: 0 on success
+# Outputs: lines with status info
+archive_task_complete() {
+  local task_dir_abs="$1"
+  local repo_root="${2:-$(get_repo_root)}"
+
+  if [[ ! -d "$task_dir_abs" ]]; then
+    echo "Error: task directory not found: $task_dir_abs" >&2
+    return 1
+  fi
+
+  # Archive the directory
+  local archive_dest
+  if archive_dest=$(archive_task_dir "$task_dir_abs" "$repo_root"); then
+    echo "archived_to:$archive_dest"
+    return 0
+  fi
+
+  return 1
+}

package/dist/templates/aim/scripts-shell-archive/common/worktree.sh

@@ -0,0 +1,128 @@
+#!/bin/bash
+# Worktree utilities for Multi-Agent Pipeline
+#
+# Usage: source this file in multi-agent scripts
+#   source "$(dirname "$0")/../common/worktree.sh"
+#
+# Provides:
+#   get_worktree_config         - Get worktree.yaml path
+#   get_worktree_base_dir       - Get worktree storage directory
+#   get_worktree_copy_files     - Get files to copy list
+#   get_worktree_post_create_hooks - Get post-create hooks
+#   get_agents_dir              - Get agents registry directory
+#
+# Requires: paths.sh (for get_repo_root)
+
+# Ensure paths.sh is loaded
+if ! type get_repo_root &>/dev/null; then
+  SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
+  source "$SCRIPT_DIR/paths.sh"
+fi
+
+# =============================================================================
+# Worktree Configuration
+# =============================================================================
+
+# Worktree config file relative path (relative to repo root)
+WORKTREE_CONFIG_PATH="$DIR_WORKFLOW/worktree.yaml"
+
+# Get worktree.yaml config file path
+# Args: $1 - repo_root (optional)
+# Returns: absolute path to config file
+get_worktree_config() {
+  local repo_root="${1:-$(get_repo_root)}"
+  echo "$repo_root/$WORKTREE_CONFIG_PATH"
+}
+
+# Read simple value from worktree.yaml
+# Args: $1 - key, $2 - config_file (optional)
+# Returns: value
+_yaml_get_value() {
+  local key="$1"
+  local config="${2:-$(get_worktree_config)}"
+  grep "^${key}:" "$config" 2>/dev/null | sed "s/^${key}:[[:space:]]*//" | tr -d '"' | tr -d "'"
+}
+
+# Read list from worktree.yaml
+# Args: $1 - section, $2 - config_file (optional)
+# Returns: list items (one per line)
+_yaml_get_list() {
+  local section="$1"
+  local config="${2:-$(get_worktree_config)}"
+  local in_section=0
+
+  while IFS= read -r line; do
+    if [[ "$line" =~ ^${section}: ]]; then
+      in_section=1
+      continue
+    fi
+
+    if [ $in_section -eq 1 ]; then
+      # Exit when encountering new top-level key
+      if [[ "$line" =~ ^[a-z_]+: ]] && [[ ! "$line" =~ ^[[:space:]] ]]; then
+        break
+      fi
+      # Read list item
+      if [[ "$line" =~ ^[[:space:]]*-[[:space:]](.+)$ ]]; then
+        echo "${BASH_REMATCH[1]}" | tr -d '"' | tr -d "'"
+      fi
+    fi
+  done < "$config"
+}
+
+# Get worktree base directory
+# Args: $1 - repo_root (optional)
+# Returns: absolute path to worktree base directory
+get_worktree_base_dir() {
+  local repo_root="${1:-$(get_repo_root)}"
+  local config=$(get_worktree_config "$repo_root")
+  local worktree_dir=$(_yaml_get_value "worktree_dir" "$config")
+
+  # Default value
+  if [ -z "$worktree_dir" ]; then
+    worktree_dir="../worktrees"
+  fi
+
+  # Handle relative path
+  if [[ "$worktree_dir" == ../* ]] || [[ "$worktree_dir" == ./* ]]; then
+    # Relative to repo_root
+    echo "$repo_root/$worktree_dir"
+  else
+    # Absolute path
+    echo "$worktree_dir"
+  fi
+}
+
+# Get files to copy list
+# Args: $1 - repo_root (optional)
+# Returns: file list (one per line)
+get_worktree_copy_files() {
+  local repo_root="${1:-$(get_repo_root)}"
+  local config=$(get_worktree_config "$repo_root")
+  _yaml_get_list "copy" "$config"
+}
+
+# Get post_create hooks
+# Args: $1 - repo_root (optional)
+# Returns: command list (one per line)
+get_worktree_post_create_hooks() {
+  local repo_root="${1:-$(get_repo_root)}"
+  local config=$(get_worktree_config "$repo_root")
+  _yaml_get_list "post_create" "$config"
+}
+
+# =============================================================================
+# Agents Registry
+# =============================================================================
+
+# Get agents directory for current developer
+# Args: $1 - repo_root (optional)
+# Returns: absolute path to agents directory
+get_agents_dir() {
+  local repo_root="${1:-$(get_repo_root)}"
+  local workspace_dir=$(get_workspace_dir "$repo_root")
+
+  if [[ -n "$workspace_dir" ]]; then
+    echo "$workspace_dir/.agents"
+  fi
+}