@kuandotdev/indicator 0.1.1 → 0.1.3

package/project/tools/import_agent_tools.sh

@@ -0,0 +1,503 @@
+#!/usr/bin/env bash
+# import_agent_tools.sh — detect/configure the target agent platform(s) and wire
+# the TradingView/market-analysis tools, skills, prompt files, package extension,
+# and MCP config needed by those agents.
+#
+# Usage:
+#   tools/import_agent_tools.sh                         # auto-detect; fallback all
+#   tools/import_agent_tools.sh --platform pi           # pi only
+#   tools/import_agent_tools.sh --platform claude-code  # Claude Code only
+#   tools/import_agent_tools.sh --platform claude-desktop
+#   tools/import_agent_tools.sh --platform pi,claude-code,codex
+#   tools/import_agent_tools.sh --platform all
+#   tools/import_agent_tools.sh --dry-run
+
+set -euo pipefail
+
+SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
+PROJECT_ROOT="$(cd "$SCRIPT_DIR/.." && pwd)"
+CONDA_ENV="${CONDA_ENV:-tv-indicator}"
+PLATFORM="auto"
+DRY_RUN=0
+GLOBAL_CLAUDE_SKILLS=0
+WRITE_PI_MCP=1
+WRITE_CLAUDE_DESKTOP_MCP=1
+WRITE_CLAUDE_CODE_MCP=1
+WRITE_CODEX_MCP=1
+WRITE_GEMINI_MCP=1
+WRITE_CURSOR_MCP=1
+
+usage() {
+  sed -n '1,18p' "$0" | sed 's/^# \{0,1\}//'
+  cat <<'EOF'
+
+Options:
+  --platform <names>        auto | all | pi | claude-desktop | claude-code | claude | codex | gemini | cursor
+                            Supports comma/space-separated lists.
+  --dry-run                 Print actions without writing
+  --global-claude-skills    Also symlink skills into ~/.claude/skills
+  --no-pi-mcp               Do not generate .pi/config.toml MCP entries
+  --no-claude-desktop-mcp   Do not update Claude Desktop MCP config
+  --no-claude-code-mcp      Do not write .mcp.json for Claude Code
+  --no-codex-mcp            Do not update ~/.codex/config.toml MCP config
+  --no-gemini-mcp           Do not update ~/.gemini/settings.json MCP config
+  --no-cursor-mcp           Do not write .cursor/mcp.json MCP config
+  -h, --help                Show help
+
+What it imports:
+  pi              -> .pi/APPEND_SYSTEM.md, .pi/skills, package extension, .pi/config.toml MCP
+  claude-desktop  -> Claude Desktop claude_desktop_config.json MCP servers
+  claude-code     -> CLAUDE.md, .claude/skills, .mcp.json MCP servers
+  codex           -> AGENTS.md, ~/.codex/config.toml MCP servers
+  gemini          -> GEMINI.md, ~/.gemini/settings.json MCP servers
+  cursor          -> .cursor/rules/*.mdc, .cursor/mcp.json MCP servers
+  all             -> all of the above
+EOF
+}
+
+while [[ $# -gt 0 ]]; do
+  case "$1" in
+    --platform)
+      PLATFORM="${2:-}"
+      [[ -n "$PLATFORM" ]] || { echo "--platform requires a value" >&2; exit 2; }
+      shift 2
+      ;;
+    --dry-run) DRY_RUN=1; shift ;;
+    --global-claude-skills) GLOBAL_CLAUDE_SKILLS=1; shift ;;
+    --no-pi-mcp) WRITE_PI_MCP=0; shift ;;
+    --no-claude-desktop-mcp) WRITE_CLAUDE_DESKTOP_MCP=0; shift ;;
+    --no-claude-code-mcp) WRITE_CLAUDE_CODE_MCP=0; shift ;;
+    --no-codex-mcp) WRITE_CODEX_MCP=0; shift ;;
+    --no-gemini-mcp) WRITE_GEMINI_MCP=0; shift ;;
+    --no-cursor-mcp) WRITE_CURSOR_MCP=0; shift ;;
+    -h|--help) usage; exit 0 ;;
+    *) echo "Unknown argument: $1" >&2; usage >&2; exit 2 ;;
+  esac
+done
+
+cd "$PROJECT_ROOT"
+
+log() { printf '%s\n' "$*"; }
+warn() { printf 'WARN: %s\n' "$*" >&2; }
+have_cmd() { command -v "$1" >/dev/null 2>&1; }
+
+mcp_command_path() {
+  local exe="$1"
+  local resolved=""
+  if have_cmd conda; then
+    resolved="$(conda run -n "$CONDA_ENV" which "$exe" 2>/dev/null || true)"
+  fi
+  if [[ -z "$resolved" ]]; then
+    resolved="$(command -v "$exe" 2>/dev/null || true)"
+  fi
+  if [[ -z "$resolved" ]]; then
+    resolved="$exe"
+  fi
+  printf '%s\n' "$resolved"
+}
+
+mcp_servers_json() {
+  local data_cmd browser_cmd agent_cmd
+  data_cmd="$(mcp_command_path training-view-mcp-data)"
+  browser_cmd="$(mcp_command_path training-view-mcp-browser)"
+  agent_cmd="$(mcp_command_path training-view-mcp-agent)"
+
+  python3 - "$data_cmd" "$browser_cmd" "$agent_cmd" <<'PY'
+import json, sys
+print(json.dumps({
+    "training-view-data": {"command": sys.argv[1]},
+    "training-view-browser": {"command": sys.argv[2], "env": {"TV_BROWSER_HEADLESS": "true"}},
+    "training-view-agent": {"command": sys.argv[3]},
+}, indent=2))
+PY
+}
+
+write_mcp_json_config() {
+  local path="$1"
+  local label="$2"
+  local servers_json
+  servers_json="$(mcp_servers_json)"
+
+  if [[ "$DRY_RUN" == "1" ]]; then
+    log "[dry-run] merge TradingView MCP servers (training-view-data, -browser, -agent) into $label: $path"
+    printf '%s\n' "$servers_json"
+    return
+  fi
+
+  mkdir -p "$(dirname "$path")"
+  python3 - "$path" "$servers_json" "$label" <<'PY'
+import json
+import sys
+from pathlib import Path
+
+path = Path(sys.argv[1]).expanduser()
+servers = json.loads(sys.argv[2])
+label = sys.argv[3]
+
+if path.exists() and path.read_text(encoding="utf-8").strip():
+    try:
+        data = json.loads(path.read_text(encoding="utf-8"))
+    except json.JSONDecodeError:
+        backup = path.with_suffix(path.suffix + ".bak")
+        backup.write_text(path.read_text(encoding="utf-8"), encoding="utf-8")
+        print(f"Existing {label} config was invalid JSON; backed up to {backup}")
+        data = {}
+else:
+    data = {}
+
+mcp = data.get("mcpServers")
+if not isinstance(mcp, dict):
+    mcp = {}
+
+# Evict legacy tv-* names so re-importing after the rename doesn't
+# leave stale duplicates alongside the new training-view-* entries.
+for legacy in ("tv-data", "tv-browser", "tv-agent"):
+    mcp.pop(legacy, None)
+
+for name, cfg in servers.items():
+    mcp[name] = cfg
+
+data["mcpServers"] = mcp
+path.write_text(json.dumps(data, indent=2) + "\n", encoding="utf-8")
+print(f"Wrote {label} MCP config: {path}")
+PY
+}
+
+claude_desktop_config_path() {
+  case "$(uname -s 2>/dev/null || echo unknown)" in
+    Darwin) printf '%s\n' "$HOME/Library/Application Support/Claude/claude_desktop_config.json" ;;
+    Linux) printf '%s\n' "$HOME/.config/Claude/claude_desktop_config.json" ;;
+    MINGW*|MSYS*|CYGWIN*) printf '%s\n' "${APPDATA:-$HOME/AppData/Roaming}/Claude/claude_desktop_config.json" ;;
+    *) printf '%s\n' "$HOME/.config/Claude/claude_desktop_config.json" ;;
+  esac
+}
+
+write_toml_mcp_block_config() {
+  local path="$1"
+  local label="$2"
+  local prefix="$3"
+  local data_cmd browser_cmd agent_cmd
+  data_cmd="$(mcp_command_path training-view-mcp-data)"
+  browser_cmd="$(mcp_command_path training-view-mcp-browser)"
+  agent_cmd="$(mcp_command_path training-view-mcp-agent)"
+
+  local block
+  block="# BEGIN tv-indicator agent tools (managed by tools/import_agent_tools.sh)
+[$prefix.training-view-data]
+command = \"$data_cmd\"
+
+[$prefix.training-view-browser]
+command = \"$browser_cmd\"
+env = { TV_BROWSER_HEADLESS = \"true\" }
+
+[$prefix.training-view-agent]
+command = \"$agent_cmd\"
+# END tv-indicator agent tools
+"
+
+  if [[ "$DRY_RUN" == "1" ]]; then
+    log "[dry-run] write/update managed MCP block in $label: $path"
+    printf '%s\n' "$block"
+    return
+  fi
+
+  mkdir -p "$(dirname "$path")"
+  python3 - "$path" "$block" "$prefix" "$label" <<'PY'
+import sys
+from pathlib import Path
+
+path = Path(sys.argv[1]).expanduser()
+block = sys.argv[2]
+prefix = sys.argv[3]
+label = sys.argv[4]
+start = "# BEGIN tv-indicator agent tools"
+end = "# END tv-indicator agent tools"
+text = path.read_text(encoding="utf-8") if path.exists() else ""
+
+if start not in text:
+    for section in (
+        f"[{prefix}.training-view-data]",
+        f"[{prefix}.training-view-browser]",
+        f"[{prefix}.training-view-agent]",
+    ):
+        if section in text:
+            print(f"Skipping {label} MCP write because {section} already exists outside managed block")
+            sys.exit(0)
+
+if start in text and end in text:
+    before = text.split(start, 1)[0].rstrip()
+    after = text.split(end, 1)[1].lstrip()
+    new_text = (before + "\n\n" if before else "") + block + ("\n" + after if after else "")
+else:
+    new_text = (text.rstrip() + "\n\n" if text.strip() else "") + block
+
+path.write_text(new_text, encoding="utf-8")
+print(f"Wrote {label} MCP config: {path}")
+PY
+}
+
+json_update_pi_settings() {
+  local settings="$PROJECT_ROOT/.pi/settings.json"
+  local package_source="../packages/tradingview-stock-data-skill"
+
+  if [[ "$DRY_RUN" == "1" ]]; then
+    log "[dry-run] ensure .pi/settings.json loads package extension only: $package_source (skills disabled to avoid duplicate skill conflicts)"
+    return
+  fi
+
+  mkdir -p "$PROJECT_ROOT/.pi"
+  python3 - "$settings" "$package_source" <<'PY'
+import json
+import sys
+from pathlib import Path
+
+settings_path = Path(sys.argv[1])
+source = sys.argv[2]
+if settings_path.exists() and settings_path.read_text(encoding="utf-8").strip():
+    try:
+        data = json.loads(settings_path.read_text(encoding="utf-8"))
+    except json.JSONDecodeError:
+        backup = settings_path.with_suffix(settings_path.suffix + ".bak")
+        backup.write_text(settings_path.read_text(encoding="utf-8"), encoding="utf-8")
+        print(f"Existing settings JSON was invalid; backed up to {backup}")
+        data = {}
+else:
+    data = {}
+
+packages = data.get("packages", [])
+if not isinstance(packages, list):
+    packages = []
+
+filtered = []
+for item in packages:
+    if item == source:
+        continue
+    if isinstance(item, dict) and item.get("source") == source:
+        continue
+    filtered.append(item)
+
+filtered.append({"source": source, "skills": []})
+data["packages"] = filtered
+settings_path.write_text(json.dumps(data, indent=2) + "\n", encoding="utf-8")
+PY
+}
+
+write_pi_import_sentinel() {
+  local sentinel="$PROJECT_ROOT/.pi/.tv-indicator-auto-imported"
+  if [[ "$DRY_RUN" == "1" ]]; then
+    log "[dry-run] write Pi auto-import sentinel: .pi/.tv-indicator-auto-imported"
+    return
+  fi
+  mkdir -p "$PROJECT_ROOT/.pi"
+  {
+    echo "Managed by tools/import_agent_tools.sh"
+    echo "platform=pi"
+    echo "updated_at=$(date -u +%Y-%m-%dT%H:%M:%SZ)"
+  } > "$sentinel"
+}
+
+write_pi_mcp_config() {
+  [[ "$WRITE_PI_MCP" == "1" ]] || return 0
+
+  local config="$PROJECT_ROOT/.pi/config.toml"
+  local data_cmd browser_cmd agent_cmd
+  data_cmd="$(mcp_command_path training-view-mcp-data)"
+  browser_cmd="$(mcp_command_path training-view-mcp-browser)"
+  agent_cmd="$(mcp_command_path training-view-mcp-agent)"
+
+  local block
+  block="# BEGIN tv-indicator agent tools (managed by tools/import_agent_tools.sh)
+[mcp_servers.training-view-data]
+command = \"$data_cmd\"
+
+[mcp_servers.training-view-browser]
+command = \"$browser_cmd\"
+env = { TV_BROWSER_HEADLESS = \"true\" }
+
+[mcp_servers.training-view-agent]
+command = \"$agent_cmd\"
+# END tv-indicator agent tools
+"
+
+  if [[ "$DRY_RUN" == "1" ]]; then
+    log "[dry-run] write/update managed MCP block in .pi/config.toml"
+    printf '%s\n' "$block"
+    return
+  fi
+
+  mkdir -p "$PROJECT_ROOT/.pi"
+  python3 - "$config" "$block" <<'PY'
+import sys
+from pathlib import Path
+
+path = Path(sys.argv[1])
+block = sys.argv[2]
+start = "# BEGIN tv-indicator agent tools"
+end = "# END tv-indicator agent tools"
+text = path.read_text(encoding="utf-8") if path.exists() else ""
+
+if start not in text:
+    for section in (
+        "[mcp_servers.training-view-data]",
+        "[mcp_servers.training-view-browser]",
+        "[mcp_servers.training-view-agent]",
+    ):
+        if section in text:
+            print(f"Skipping .pi/config.toml MCP write because {section} already exists outside managed block")
+            sys.exit(0)
+
+if start in text and end in text:
+    before = text.split(start, 1)[0].rstrip()
+    after = text.split(end, 1)[1].lstrip()
+    new_text = (before + "\n\n" if before else "") + block + ("\n" + after if after else "")
+else:
+    new_text = (text.rstrip() + "\n\n" if text.strip() else "") + block
+
+path.write_text(new_text, encoding="utf-8")
+print(f"Wrote managed MCP config: {path}")
+PY
+}
+
+install_project_skills() {
+  log "Refreshing project skills, context files, docs, and active startup prompt..."
+  if [[ "$DRY_RUN" == "1" ]]; then
+    log "[dry-run] ./tools/install_skills.sh"
+  else
+    ./tools/install_skills.sh
+  fi
+}
+
+install_global_claude_skills_if_requested() {
+  [[ "$GLOBAL_CLAUDE_SKILLS" == "1" ]] || return 0
+  log "Installing user-global Claude skills..."
+  if [[ "$DRY_RUN" == "1" ]]; then
+    log "[dry-run] TV_INSTALL_GLOBAL_CLAUDE=1 ./tools/install_skills.sh"
+  else
+    TV_INSTALL_GLOBAL_CLAUDE=1 ./tools/install_skills.sh
+  fi
+}
+
+detect_active_platform() {
+  if [[ -n "${PI_CODING_AGENT_SESSION_DIR:-}${PI_SESSION_ID:-}" ]]; then echo "pi"; return; fi
+  if [[ -n "${CLAUDE_SESSION_ID:-}${CLAUDECODE:-}" ]]; then echo "claude-code"; return; fi
+  if [[ -n "${CODEX_SANDBOX:-}${CODEX_SESSION_ID:-}" ]]; then echo "codex"; return; fi
+  if [[ -n "${GEMINI_CLI:-}${GEMINI_SESSION_ID:-}" ]]; then echo "gemini"; return; fi
+  echo "all"
+}
+
+add_unique_platform() {
+  local current="$1"
+  local item="$2"
+  case " $current " in
+    *" $item "*) printf '%s\n' "$current" ;;
+    *) printf '%s\n' "${current:+$current }$item" ;;
+  esac
+}
+
+normalize_platforms() {
+  local raw="$1"
+  [[ "$raw" == "auto" ]] && raw="$(detect_active_platform)"
+  raw="${raw//,/ }"
+
+  local selected=""
+  local token mapped
+  for token in $raw; do
+    case "$token" in
+      all)
+        for mapped in pi claude-desktop claude-code codex gemini cursor; do
+          selected="$(add_unique_platform "$selected" "$mapped")"
+        done
+        ;;
+      pi) mapped="pi"; selected="$(add_unique_platform "$selected" "$mapped")" ;;
+      claude|claude-code|claude_code) mapped="claude-code"; selected="$(add_unique_platform "$selected" "$mapped")" ;;
+      claude-desktop|claude_desktop|desktop) mapped="claude-desktop"; selected="$(add_unique_platform "$selected" "$mapped")" ;;
+      codex|opencode|aider) mapped="codex"; selected="$(add_unique_platform "$selected" "$mapped")" ;;
+      gemini) mapped="gemini"; selected="$(add_unique_platform "$selected" "$mapped")" ;;
+      cursor) mapped="cursor"; selected="$(add_unique_platform "$selected" "$mapped")" ;;
+      "") ;;
+      *) echo "Unsupported platform: $token" >&2; exit 2 ;;
+    esac
+  done
+  printf '%s\n' "$selected"
+}
+
+print_detection_summary() {
+  log "Agent platform import"
+  log "  project:  $PROJECT_ROOT"
+  log "  request:  $PLATFORM"
+  log "  selected: $(normalize_platforms "$PLATFORM")"
+  log "  detected commands: pi=$(have_cmd pi && echo yes || echo no), claude=$(have_cmd claude && echo yes || echo no), codex=$(have_cmd codex && echo yes || echo no), gemini=$(have_cmd gemini && echo yes || echo no)"
+  log ""
+}
+
+configure_pi() {
+  log "[pi] Importing Pi package extension and MCP tools..."
+  json_update_pi_settings
+  write_pi_mcp_config
+  write_pi_import_sentinel
+  log "[pi] Uses .pi/APPEND_SYSTEM.md, .pi/skills, .pi/settings.json package extension, and .pi/config.toml MCP."
+}
+
+configure_claude_desktop() {
+  log "[claude-desktop] Importing Claude Desktop MCP servers..."
+  if [[ "$WRITE_CLAUDE_DESKTOP_MCP" == "1" ]]; then
+    write_mcp_json_config "$(claude_desktop_config_path)" "Claude Desktop"
+  fi
+  log "[claude-desktop] Uses Claude Desktop mcpServers: training-view-data, training-view-browser, training-view-agent."
+}
+
+configure_claude_code() {
+  log "[claude-code] Importing Claude Code project context, skills, and MCP servers..."
+  install_global_claude_skills_if_requested
+  if [[ "$WRITE_CLAUDE_CODE_MCP" == "1" ]]; then
+    write_mcp_json_config "$PROJECT_ROOT/.mcp.json" "Claude Code project"
+  fi
+  log "[claude-code] Uses CLAUDE.md, .claude/skills/, and .mcp.json."
+}
+
+configure_codex() {
+  log "[codex] Importing Codex/OpenCode/Aider project instructions and MCP servers..."
+  if [[ "$WRITE_CODEX_MCP" == "1" ]]; then
+    write_toml_mcp_block_config "$HOME/.codex/config.toml" "Codex" "mcpServers"
+  fi
+  log "[codex] Uses AGENTS.md and ~/.codex/config.toml MCP servers."
+}
+
+configure_gemini() {
+  log "[gemini] Importing Gemini project instructions and MCP servers..."
+  if [[ "$WRITE_GEMINI_MCP" == "1" ]]; then
+    write_mcp_json_config "$HOME/.gemini/settings.json" "Gemini CLI"
+  fi
+  log "[gemini] Uses GEMINI.md and ~/.gemini/settings.json MCP servers."
+}
+
+configure_cursor() {
+  log "[cursor] Importing Cursor rules and MCP servers..."
+  if [[ "$WRITE_CURSOR_MCP" == "1" ]]; then
+    write_mcp_json_config "$PROJECT_ROOT/.cursor/mcp.json" "Cursor project"
+  fi
+  log "[cursor] Uses .cursor/rules/*.mdc and .cursor/mcp.json."
+}
+
+main() {
+  print_detection_summary
+  install_project_skills
+
+  local selected
+  selected="$(normalize_platforms "$PLATFORM")"
+  for p in $selected; do
+    case "$p" in
+      pi) configure_pi ;;
+      claude-desktop) configure_claude_desktop ;;
+      claude-code) configure_claude_code ;;
+      codex) configure_codex ;;
+      gemini) configure_gemini ;;
+      cursor) configure_cursor ;;
+    esac
+  done
+
+  log ""
+  log "Done. Restart or reload the selected agent(s) to pick up changes."
+}
+
+main