591dcb4285
- Add KUGETSU_BASE_BRANCH env var (default: origin/main) - Update check_task_completion() to use configurable base branch - Update create_worktree() to use configurable base branch Closes #165
115 lines
3.4 KiB
Bash
Executable File
115 lines
3.4 KiB
Bash
Executable File
#!/bin/bash
|
|
set -euo pipefail
|
|
|
|
KUGETSU_DIR="${KUGETSU_DIR:-$HOME/.kugetsu}"
|
|
SESSIONS_DIR="$KUGETSU_DIR/sessions"
|
|
WORKTREES_DIR="$KUGETSU_DIR/worktrees"
|
|
REPOS_CONFIG="$KUGETSU_DIR/repos.json"
|
|
INDEX_FILE="$KUGETSU_DIR/index.json"
|
|
NOTIFICATIONS_FILE="$KUGETSU_DIR/notifications.json"
|
|
LOGS_DIR="$KUGETSU_DIR/logs"
|
|
ENV_DIR="${ENV_DIR:-$KUGETSU_DIR/env}"
|
|
VERBOSITY_DIR="$KUGETSU_DIR/verbosity"
|
|
|
|
MAX_CONCURRENT_AGENTS="${MAX_CONCURRENT_AGENTS:-3}"
|
|
KUGETSU_VERBOSITY="${KUGETSU_VERBOSITY:-default}"
|
|
CONTEXT_DIR="${CONTEXT_DIR:-$KUGETSU_DIR/context}"
|
|
ENABLE_CONTEXT_DUMP="${ENABLE_CONTEXT_DUMP:-true}"
|
|
WORKTREE_CHECK_PR_STATUS="${WORKTREE_CHECK_PR_STATUS:-true}"
|
|
|
|
QUEUE_DIR="${QUEUE_DIR:-$KUGETSU_DIR/queue}"
|
|
QUEUE_ITEMS_DIR="${QUEUE_ITEMS_DIR:-$QUEUE_DIR/items}"
|
|
QUEUE_DAEMON_PID_FILE="${QUEUE_DAEMON_PID_FILE:-$QUEUE_DIR/daemon.pid}"
|
|
QUEUE_DAEMON_LOCK_FILE="${QUEUE_DAEMON_LOCK_FILE:-$QUEUE_DIR/daemon.lock}"
|
|
QUEUE_DAEMON_LOG_FILE="${QUEUE_DAEMON_LOG_FILE:-$QUEUE_DIR/daemon.log}"
|
|
QUEUE_DAEMON_INTERVAL_MINUTES="${QUEUE_DAEMON_INTERVAL_MINUTES:-5}"
|
|
QUEUE_CLEANUP_AGE_DAYS="${QUEUE_CLEANUP_AGE_DAYS:-7}"
|
|
TASK_TIMEOUT_HOURS="${TASK_TIMEOUT_HOURS:-1}"
|
|
|
|
NETWORK_RETRY_ATTEMPTS="${NETWORK_RETRY_ATTEMPTS:-3}"
|
|
NETWORK_RETRY_DELAY_SECONDS="${NETWORK_RETRY_DELAY_SECONDS:-5}"
|
|
KUGETSU_BASE_BRANCH="${KUGETSU_BASE_BRANCH:-origin/main}"
|
|
|
|
# Load user config overrides (~/.kugetsu/config)
|
|
if [ -f "$KUGETSU_DIR/config" ]; then
|
|
source "$KUGETSU_DIR/config"
|
|
fi
|
|
|
|
mask_sensitive_vars() {
|
|
local line="${1:-}"
|
|
for var in GITEA_TOKEN GITHUB_TOKEN GITLAB_TOKEN API_KEY PASSWORD TOKEN SECRET; do
|
|
if [[ "$line" =~ $var ]]; then
|
|
line=$(echo "$line" | sed -E "s/=.*/=***MASKED***/")
|
|
fi
|
|
done
|
|
echo "$line"
|
|
}
|
|
|
|
strip_ansi_codes() {
|
|
local line="${1:-}"
|
|
echo "$line" | sed 's/\x1b\[[0-9;]*m//g' | sed 's/\x1b\[[0-9;]*[a-zA-Z]//g'
|
|
}
|
|
|
|
load_agent_env() {
|
|
local agent_type="${1:-base}"
|
|
local env_file="$ENV_DIR/${agent_type}.env"
|
|
|
|
if [ -f "$env_file" ]; then
|
|
set -a
|
|
source "$env_file"
|
|
set +a
|
|
elif [ -f "$ENV_DIR/default.env" ]; then
|
|
set -a
|
|
source "$ENV_DIR/default.env"
|
|
set +a
|
|
elif [ -f "$ENV_DIR/pm-agent.env" ]; then
|
|
set -a
|
|
source "$ENV_DIR/pm-agent.env"
|
|
set +a
|
|
fi
|
|
}
|
|
|
|
set_debug_mode() {
|
|
local filtered_args=()
|
|
local debug_mode=false
|
|
|
|
for arg in "$@"; do
|
|
case "$arg" in
|
|
--debug)
|
|
debug_mode=true
|
|
;;
|
|
*)
|
|
filtered_args+=("$arg")
|
|
;;
|
|
esac
|
|
done
|
|
|
|
if [ "$debug_mode" = true ]; then
|
|
export KUGETSU_VERBOSITY="debug"
|
|
echo "[DEBUG] Debug mode enabled" >&2
|
|
fi
|
|
|
|
echo "${filtered_args[@]}"
|
|
}
|
|
|
|
retry_with_backoff() {
|
|
local max_attempts="${1:-$NETWORK_RETRY_ATTEMPTS}"
|
|
local delay_seconds="${2:-$NETWORK_RETRY_DELAY_SECONDS}"
|
|
local command="$3"
|
|
local remaining_attempts=$max_attempts
|
|
|
|
while [ $remaining_attempts -gt 0 ]; do
|
|
if eval "$command"; then
|
|
return 0
|
|
fi
|
|
remaining_attempts=$((remaining_attempts - 1))
|
|
if [ $remaining_attempts -gt 0 ]; then
|
|
log "warn" "retry_with_backoff" "Command failed, $remaining_attempts retries remaining. Waiting ${delay_seconds}s..."
|
|
sleep "$delay_seconds"
|
|
delay_seconds=$((delay_seconds * 2))
|
|
fi
|
|
done
|
|
log "error" "retry_with_backoff" "Command failed after $max_attempts attempts"
|
|
return 1
|
|
}
|