@liangjie559567/ultrapower 5.0.4

package/skills/project-session-manager/lib/session.sh

@@ -0,0 +1,169 @@
+#!/bin/bash
+# PSM Session Registry Management
+
+# Lock file for atomic registry operations
+PSM_LOCK_FILE="${PSM_DATA_DIR:-.psm}/.psm-lock"
+
+# Wrapper for atomic operations with file locking
+# Usage: psm_with_lock <command> [args...]
+psm_with_lock() {
+    local timeout="${PSM_LOCK_TIMEOUT:-5}"
+    (
+        flock -w "$timeout" 200 || {
+            echo "error|Failed to acquire lock after ${timeout}s" >&2
+            return 1
+        }
+        "$@"
+    ) 200>"$PSM_LOCK_FILE"
+}
+
+# Internal: Add session to registry (must be called via psm_with_lock)
+_psm_add_session_impl() {
+    local id="$1"
+    local type="$2"
+    local project="$3"
+    local ref="$4"
+    local branch="$5"
+    local base="$6"
+    local tmux_session="$7"
+    local worktree="$8"
+    local source_repo="$9"
+    local metadata="${10:-{}}"
+    local provider="${11:-github}"
+    local provider_ref="${12:-}"
+
+    local now=$(date -Iseconds)
+
+    local tmp=$(mktemp)
+    jq --arg id "$id" \
+       --arg type "$type" \
+       --arg project "$project" \
+       --arg ref "$ref" \
+       --arg branch "$branch" \
+       --arg base "$base" \
+       --arg tmux "$tmux_session" \
+       --arg worktree "$worktree" \
+       --arg source "$source_repo" \
+       --arg now "$now" \
+       --arg provider "$provider" \
+       --arg provider_ref "$provider_ref" \
+       --argjson meta "$metadata" \
+       '.sessions[$id] = {
+          "id": $id,
+          "type": $type,
+          "project": $project,
+          "ref": $ref,
+          "branch": $branch,
+          "base": $base,
+          "tmux": $tmux,
+          "worktree": $worktree,
+          "source_repo": $source,
+          "created_at": $now,
+          "last_accessed": $now,
+          "state": "active",
+          "provider": $provider,
+          "provider_ref": $provider_ref,
+          "metadata": $meta
+        } | .stats.total_created += 1' \
+       "$PSM_SESSIONS" > "$tmp" && mv "$tmp" "$PSM_SESSIONS"
+}
+
+# Add session to registry (with file locking)
+# Usage: psm_add_session <id> <type> <project> <ref> <branch> <base> <tmux> <worktree> <source_repo> <metadata_json> [provider] [provider_ref]
+psm_add_session() {
+    psm_with_lock _psm_add_session_impl "$@"
+}
+
+# Get session by ID
+# Usage: psm_get_session <id>
+psm_get_session() {
+    local id="$1"
+    jq -r --arg i "$id" '.sessions[$i] // empty' "$PSM_SESSIONS"
+}
+
+# Internal: Update session state (must be called via psm_with_lock)
+_psm_update_session_state_impl() {
+    local id="$1"
+    local state="$2"
+    local now=$(date -Iseconds)
+
+    local tmp=$(mktemp)
+    jq --arg id "$id" \
+       --arg state "$state" \
+       --arg now "$now" \
+       '.sessions[$id].state = $state | .sessions[$id].last_accessed = $now' \
+       "$PSM_SESSIONS" > "$tmp" && mv "$tmp" "$PSM_SESSIONS"
+}
+
+# Update session state (with file locking)
+# Usage: psm_update_session_state <id> <state>
+psm_update_session_state() {
+    psm_with_lock _psm_update_session_state_impl "$@"
+}
+
+# Internal: Remove session from registry (must be called via psm_with_lock)
+_psm_remove_session_impl() {
+    local id="$1"
+
+    local tmp=$(mktemp)
+    jq --arg id "$id" \
+       'del(.sessions[$id]) | .stats.total_cleaned += 1' \
+       "$PSM_SESSIONS" > "$tmp" && mv "$tmp" "$PSM_SESSIONS"
+}
+
+# Remove session from registry (with file locking)
+# Usage: psm_remove_session <id>
+psm_remove_session() {
+    psm_with_lock _psm_remove_session_impl "$@"
+}
+
+# List all sessions
+# Usage: psm_list_sessions [project]
+psm_list_sessions() {
+    local project="$1"
+
+    if [[ -n "$project" ]]; then
+        jq -r --arg p "$project" '.sessions | to_entries[] | select(.value.project == $p) | .value | "\(.id)|\(.type)|\(.state)|\(.worktree)"' "$PSM_SESSIONS"
+    else
+        jq -r '.sessions | to_entries[] | .value | "\(.id)|\(.type)|\(.state)|\(.worktree)"' "$PSM_SESSIONS"
+    fi
+}
+
+# Get sessions by state
+psm_get_sessions_by_state() {
+    local state="$1"
+    jq -r --arg s "$state" '.sessions | to_entries[] | select(.value.state == $s) | .value.id' "$PSM_SESSIONS"
+}
+
+# Get session count
+psm_session_count() {
+    jq -r '.sessions | length' "$PSM_SESSIONS"
+}
+
+# Write session metadata file in worktree
+# Usage: psm_write_session_metadata <worktree_path> <session_json>
+psm_write_session_metadata() {
+    local worktree_path="$1"
+    local session_json="$2"
+
+    echo "$session_json" > "${worktree_path}/.psm-session.json"
+}
+
+# Read session metadata from worktree
+psm_read_session_metadata() {
+    local worktree_path="$1"
+    local meta_file="${worktree_path}/.psm-session.json"
+
+    if [[ -f "$meta_file" ]]; then
+        cat "$meta_file"
+    fi
+}
+
+# Get all session IDs for cleanup check
+psm_get_review_sessions() {
+    jq -r '.sessions | to_entries[] | select(.value.type == "review") | "\(.value.id)|\(.value.metadata.pr_number // empty)|\(.value.project)"' "$PSM_SESSIONS"
+}
+
+psm_get_fix_sessions() {
+    jq -r '.sessions | to_entries[] | select(.value.type == "fix") | "\(.value.id)|\(.value.metadata.issue_number // empty)|\(.value.project)"' "$PSM_SESSIONS"
+}

package/skills/project-session-manager/lib/tmux.sh

@@ -0,0 +1,103 @@
+#!/bin/bash
+# PSM Tmux Session Management
+
+# Check if tmux is available
+psm_has_tmux() {
+    command -v tmux &> /dev/null
+}
+
+# Create a tmux session
+# Usage: psm_create_tmux_session <session_name> <working_dir>
+psm_create_tmux_session() {
+    local session_name="$1"
+    local working_dir="$2"
+
+    if ! psm_has_tmux; then
+        echo "error|tmux not found"
+        return 1
+    fi
+
+    # Check if session already exists
+    if tmux has-session -t "$session_name" 2>/dev/null; then
+        echo "exists|$session_name"
+        return 1
+    fi
+
+    # Create detached session
+    tmux new-session -d -s "$session_name" -c "$working_dir" 2>/dev/null || {
+        echo "error|Failed to create tmux session"
+        return 1
+    }
+
+    echo "created|$session_name"
+    return 0
+}
+
+# Launch Claude Code in tmux session
+# Usage: psm_launch_claude <session_name>
+psm_launch_claude() {
+    local session_name="$1"
+
+    if ! tmux has-session -t "$session_name" 2>/dev/null; then
+        echo "error|Session not found: $session_name"
+        return 1
+    fi
+
+    # Send claude command to the session
+    tmux send-keys -t "$session_name" "claude" Enter
+
+    echo "launched|$session_name"
+    return 0
+}
+
+# Kill a tmux session
+# Usage: psm_kill_tmux_session <session_name>
+psm_kill_tmux_session() {
+    local session_name="$1"
+
+    if ! tmux has-session -t "$session_name" 2>/dev/null; then
+        echo "not_found|$session_name"
+        return 0
+    fi
+
+    tmux kill-session -t "$session_name" 2>/dev/null || {
+        echo "error|Failed to kill session"
+        return 1
+    }
+
+    echo "killed|$session_name"
+    return 0
+}
+
+# List all PSM tmux sessions
+psm_list_tmux_sessions() {
+    if ! psm_has_tmux; then
+        return 0
+    fi
+
+    tmux list-sessions -F "#{session_name}|#{session_created}|#{session_attached}" 2>/dev/null | grep "^psm:" || true
+}
+
+# Check if a tmux session exists
+# Usage: psm_tmux_session_exists <session_name>
+psm_tmux_session_exists() {
+    local session_name="$1"
+    tmux has-session -t "$session_name" 2>/dev/null
+}
+
+# Get current tmux session name
+psm_current_tmux_session() {
+    if [[ -n "$TMUX" ]]; then
+        tmux display-message -p "#{session_name}" 2>/dev/null
+    fi
+}
+
+# Generate tmux session name
+# Usage: psm_tmux_session_name <alias> <type> <id>
+psm_tmux_session_name() {
+    local alias="$1"
+    local type="$2"
+    local id="$3"
+
+    echo "psm:${alias}:${type}-${id}"
+}

package/skills/project-session-manager/lib/worktree.sh

@@ -0,0 +1,204 @@
+#!/bin/bash
+# PSM Worktree Management
+
+# Validate worktree path is under PSM worktree root before deletion
+# Returns 0 if valid, 1 if invalid
+# Usage: validate_worktree_path <path>
+validate_worktree_path() {
+    local path="$1"
+    local worktree_root
+    worktree_root=$(psm_get_worktree_root 2>/dev/null) || return 1
+
+    # Path must exist and be a directory
+    if [[ ! -d "$path" ]]; then
+        return 1
+    fi
+
+    # Resolve to absolute paths for comparison
+    local abs_path abs_root
+    abs_path=$(cd "$path" 2>/dev/null && pwd) || return 1
+    abs_root=$(cd "$worktree_root" 2>/dev/null && pwd) || return 1
+
+    # Check path is under root and doesn't contain ..
+    if [[ "$abs_path" != "$abs_root"/* ]] || [[ "$path" == *".."* ]]; then
+        echo "error|Invalid worktree path: not under PSM root" >&2
+        return 1
+    fi
+    return 0
+}
+
+# Create a worktree for PR review
+# Usage: psm_create_pr_worktree <local_repo> <alias> <pr_number> <pr_branch>
+psm_create_pr_worktree() {
+    local local_repo="$1"
+    local alias="$2"
+    local pr_number="$3"
+    local pr_branch="$4"
+
+    local worktree_root=$(psm_get_worktree_root)
+    local worktree_path="${worktree_root}/${alias}/pr-${pr_number}"
+
+    # Check if worktree already exists
+    if [[ -d "$worktree_path" ]]; then
+        echo "exists|$worktree_path"
+        return 1
+    fi
+
+    # Ensure parent directory exists
+    mkdir -p "${worktree_root}/${alias}"
+
+    # Fetch the PR branch
+    cd "$local_repo" || return 1
+    git fetch origin "pull/${pr_number}/head:psm-pr-${pr_number}-review" 2>/dev/null || {
+        echo "error|Failed to fetch PR #${pr_number}"
+        return 1
+    }
+
+    # Create worktree
+    git worktree add "$worktree_path" "psm-pr-${pr_number}-review" 2>/dev/null || {
+        echo "error|Failed to create worktree"
+        return 1
+    }
+
+    echo "created|$worktree_path"
+    return 0
+}
+
+# Create a worktree for issue fix
+# Usage: psm_create_issue_worktree <local_repo> <alias> <issue_number> <slug> <base_branch>
+psm_create_issue_worktree() {
+    local local_repo="$1"
+    local alias="$2"
+    local issue_number="$3"
+    local slug="$4"
+    local base_branch="${5:-main}"
+
+    local worktree_root=$(psm_get_worktree_root)
+    local worktree_path="${worktree_root}/${alias}/issue-${issue_number}"
+    local branch_name="fix/${issue_number}-${slug}"
+
+    # Check if worktree already exists
+    if [[ -d "$worktree_path" ]]; then
+        echo "exists|$worktree_path|$branch_name"
+        return 1
+    fi
+
+    mkdir -p "${worktree_root}/${alias}"
+
+    cd "$local_repo" || return 1
+
+    # Fetch latest from origin
+    git fetch origin "$base_branch" 2>/dev/null || {
+        echo "error|Failed to fetch $base_branch"
+        return 1
+    }
+
+    # Create and checkout new branch
+    git branch "$branch_name" "origin/$base_branch" 2>/dev/null || {
+        # Branch might already exist
+        true
+    }
+
+    # Create worktree
+    git worktree add "$worktree_path" "$branch_name" 2>/dev/null || {
+        echo "error|Failed to create worktree"
+        return 1
+    }
+
+    echo "created|$worktree_path|$branch_name"
+    return 0
+}
+
+# Create a worktree for feature development
+# Usage: psm_create_feature_worktree <local_repo> <alias> <feature_name> <base_branch>
+psm_create_feature_worktree() {
+    local local_repo="$1"
+    local alias="$2"
+    local feature_name="$3"
+    local base_branch="${4:-main}"
+
+    local worktree_root=$(psm_get_worktree_root)
+    local safe_name=$(psm_sanitize "$feature_name")
+    local worktree_path="${worktree_root}/${alias}/feat-${safe_name}"
+    local branch_name="feature/${safe_name}"
+
+    # Check if worktree already exists
+    if [[ -d "$worktree_path" ]]; then
+        echo "exists|$worktree_path|$branch_name"
+        return 1
+    fi
+
+    mkdir -p "${worktree_root}/${alias}"
+
+    cd "$local_repo" || return 1
+
+    # Fetch latest
+    git fetch origin "$base_branch" 2>/dev/null || {
+        echo "error|Failed to fetch $base_branch"
+        return 1
+    }
+
+    # Create branch
+    git branch "$branch_name" "origin/$base_branch" 2>/dev/null || true
+
+    # Create worktree
+    git worktree add "$worktree_path" "$branch_name" 2>/dev/null || {
+        echo "error|Failed to create worktree"
+        return 1
+    }
+
+    echo "created|$worktree_path|$branch_name"
+    return 0
+}
+
+# Remove a worktree
+# Usage: psm_remove_worktree <local_repo> <worktree_path>
+psm_remove_worktree() {
+    local local_repo="$1"
+    local worktree_path="$2"
+
+    if [[ ! -d "$worktree_path" ]]; then
+        echo "not_found|$worktree_path"
+        return 1
+    fi
+
+    # Check for uncommitted changes
+    if [[ -d "$worktree_path/.git" ]] || [[ -f "$worktree_path/.git" ]]; then
+        cd "$worktree_path" || return 1
+        if [[ -n $(git status --porcelain 2>/dev/null) ]]; then
+            echo "dirty|$worktree_path"
+            return 1
+        fi
+    fi
+
+    cd "$local_repo" || return 1
+
+    # Validate path is under PSM worktree root before any deletion
+    if validate_worktree_path "$worktree_path"; then
+        git worktree remove "$worktree_path" --force 2>/dev/null || {
+            # Force remove the directory if git worktree remove fails
+            rm -rf "$worktree_path"
+        }
+    else
+        echo "error|Refusing to delete path outside worktree root: $worktree_path" >&2
+        return 1
+    fi
+
+    echo "removed|$worktree_path"
+    return 0
+}
+
+# List all PSM worktrees
+psm_list_worktrees() {
+    local worktree_root=$(psm_get_worktree_root)
+
+    if [[ ! -d "$worktree_root" ]]; then
+        return 0
+    fi
+
+    find "$worktree_root" -mindepth 2 -maxdepth 2 -type d 2>/dev/null | while read -r dir; do
+        local alias=$(basename "$(dirname "$dir")")
+        local name=$(basename "$dir")
+        echo "${alias}:${name}|${dir}"
+    done
+}