ai-mind-map 1.1.0

package/dist/utils/token-counter.js

@@ -0,0 +1,181 @@
+/**
+ * AI Mind Map — Fast Token Counter
+ *
+ * Provides a fast, dependency-free approximation of GPT-style token counts
+ * without loading the heavy tiktoken library.
+ *
+ * Algorithm:
+ *   1. Split input by whitespace and punctuation boundaries.
+ *   2. Count the resulting chunks.
+ *   3. Apply a calibrated multiplier (1.3) for sub-word tokenisation.
+ *   4. Add a correction for special / non-ASCII characters.
+ *
+ * Accuracy: within ~10% of the real cl100k_base token count.
+ * Performance: <1 ms for 10 K-character strings.
+ */
+/**
+ * Regex that splits text at token-like boundaries.
+ *
+ * This intentionally captures:
+ *  - Runs of word characters (a-z, A-Z, 0-9, _)
+ *  - Individual punctuation / symbol characters
+ *  - Individual non-ASCII characters (each counts as ~1 token)
+ */
+const TOKEN_SPLIT_RE = /[\w]+|[^\s\w]/g;
+/** Calibration multiplier.  Tuned against cl100k_base across many samples. */
+const BASE_MULTIPLIER = 1.3;
+/**
+ * Characters that typically expand into multiple BPE tokens.
+ * Each occurrence adds a small correction.
+ */
+const EXPENSIVE_CHAR_RE = /[{}()\[\]<>:;,."'`~!@#$%^&*+=|\\/?]/g;
+/** Extra token fraction added per expensive character. */
+const EXPENSIVE_CHAR_COST = 0.15;
+/**
+ * Estimate the token count of `text` using a fast heuristic.
+ *
+ * @param text - The input string.
+ * @returns Estimated token count (integer, ≥ 0).
+ */
+export function estimateTokens(text) {
+    if (!text) {
+        return 0;
+    }
+    // Fast-path for very short strings.
+    if (text.length <= 4) {
+        return 1;
+    }
+    const chunks = text.match(TOKEN_SPLIT_RE);
+    if (!chunks) {
+        return 0;
+    }
+    const baseCount = chunks.length;
+    // Count expensive characters for correction.
+    const expensiveMatches = text.match(EXPENSIVE_CHAR_RE);
+    const expensiveCount = expensiveMatches ? expensiveMatches.length : 0;
+    // Count non-ASCII characters (each often becomes 2-3 tokens in BPE).
+    let nonAsciiCount = 0;
+    for (let i = 0; i < text.length; i++) {
+        if (text.charCodeAt(i) > 127) {
+            nonAsciiCount++;
+        }
+    }
+    const estimate = baseCount * BASE_MULTIPLIER +
+        expensiveCount * EXPENSIVE_CHAR_COST +
+        nonAsciiCount * 0.5;
+    return Math.max(1, Math.round(estimate));
+}
+/**
+ * Estimate the token count for an array of strings (summed).
+ *
+ * @param texts - The input strings.
+ * @returns Total estimated token count.
+ */
+export function estimateTokensMany(texts) {
+    let total = 0;
+    for (const t of texts) {
+        total += estimateTokens(t);
+    }
+    return total;
+}
+// ── Smart Truncation ──────────────────────────────────────────
+/**
+ * Logical boundary characters / patterns for smart truncation,
+ * ordered from most desirable to least desirable.
+ */
+const BOUNDARY_PATTERNS = [
+    /\n\n/, // Paragraph / blank-line boundary
+    /\n}/, // End of block (function, class, object)
+    /\n\)/, // End of parenthesised block
+    /\n/, // Any newline
+    /\.\s/, // End of sentence
+    /;\s/, // End of statement
+    /,\s/, // After comma
+    /\s/, // Any whitespace (last resort)
+];
+/**
+ * Truncate `text` so that its estimated token count is ≤ `maxTokens`.
+ *
+ * The function tries to cut at a logical boundary (end of paragraph,
+ * end of function, end of sentence, etc.) rather than mid-word.
+ *
+ * @param text      - The input string.
+ * @param maxTokens - Maximum token budget.
+ * @param suffix    - Optional suffix appended to indicate truncation.
+ *                    Defaults to `"\n... [truncated]"`.
+ * @returns The (possibly truncated) string.
+ */
+export function truncateToTokenBudget(text, maxTokens, suffix = '\n... [truncated]') {
+    if (maxTokens <= 0) {
+        return suffix;
+    }
+    const currentTokens = estimateTokens(text);
+    if (currentTokens <= maxTokens) {
+        return text;
+    }
+    const suffixTokens = estimateTokens(suffix);
+    const targetTokens = maxTokens - suffixTokens;
+    if (targetTokens <= 0) {
+        return suffix;
+    }
+    // Estimate a rough character-to-token ratio for this text.
+    const ratio = text.length / currentTokens;
+    // Start with a character budget derived from the ratio, with a small buffer.
+    let charBudget = Math.floor(targetTokens * ratio * 0.95);
+    charBudget = Math.min(charBudget, text.length);
+    if (charBudget <= 0) {
+        return suffix;
+    }
+    // Try to find the best logical boundary before the char budget.
+    const candidate = text.slice(0, charBudget);
+    let bestCut = charBudget;
+    for (const pattern of BOUNDARY_PATTERNS) {
+        // Search backwards from the end of the candidate for the pattern.
+        const lastIndex = findLastMatch(candidate, pattern);
+        if (lastIndex !== -1 && lastIndex > charBudget * 0.5) {
+            // Only accept boundaries in the latter half to avoid over-truncation.
+            bestCut = lastIndex + findMatchLength(candidate, pattern, lastIndex);
+            break;
+        }
+    }
+    let truncated = text.slice(0, bestCut).trimEnd() + suffix;
+    // Safety check: verify we're within budget, shrink if needed.
+    let attempts = 0;
+    while (estimateTokens(truncated) > maxTokens && bestCut > 10 && attempts < 5) {
+        bestCut = Math.floor(bestCut * 0.85);
+        truncated = text.slice(0, bestCut).trimEnd() + suffix;
+        attempts++;
+    }
+    return truncated;
+}
+/**
+ * Find the last occurrence of `pattern` within `text`.
+ * Returns the start index, or -1 if not found.
+ */
+function findLastMatch(text, pattern) {
+    const global = new RegExp(pattern.source, 'g');
+    let lastIndex = -1;
+    let match;
+    while ((match = global.exec(text)) !== null) {
+        lastIndex = match.index;
+    }
+    return lastIndex;
+}
+/**
+ * Return the length of the match of `pattern` at `index` in `text`.
+ */
+function findMatchLength(text, pattern, index) {
+    const m = text.slice(index).match(pattern);
+    return m ? m[0].length : 0;
+}
+/**
+ * Check whether `text` fits within the given token budget.
+ *
+ * @param text      - The input string.
+ * @param maxTokens - Maximum allowed tokens.
+ * @returns `true` if estimated tokens ≤ maxTokens.
+ */
+export function fitsWithinBudget(text, maxTokens) {
+    return estimateTokens(text) <= maxTokens;
+}
+//# sourceMappingURL=token-counter.js.map

package/dist/utils/token-counter.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"token-counter.js","sourceRoot":"","sources":["../../src/utils/token-counter.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH;;;;;;;GAOG;AACH,MAAM,cAAc,GAAG,gBAAgB,CAAC;AAExC,8EAA8E;AAC9E,MAAM,eAAe,GAAG,GAAG,CAAC;AAE5B;;;GAGG;AACH,MAAM,iBAAiB,GAAG,sCAAsC,CAAC;AAEjE,0DAA0D;AAC1D,MAAM,mBAAmB,GAAG,IAAI,CAAC;AAEjC;;;;;GAKG;AACH,MAAM,UAAU,cAAc,CAAC,IAAY;IACzC,IAAI,CAAC,IAAI,EAAE,CAAC;QACV,OAAO,CAAC,CAAC;IACX,CAAC;IAED,oCAAoC;IACpC,IAAI,IAAI,CAAC,MAAM,IAAI,CAAC,EAAE,CAAC;QACrB,OAAO,CAAC,CAAC;IACX,CAAC;IAED,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC;IAC1C,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,OAAO,CAAC,CAAC;IACX,CAAC;IAED,MAAM,SAAS,GAAG,MAAM,CAAC,MAAM,CAAC;IAEhC,6CAA6C;IAC7C,MAAM,gBAAgB,GAAG,IAAI,CAAC,KAAK,CAAC,iBAAiB,CAAC,CAAC;IACvD,MAAM,cAAc,GAAG,gBAAgB,CAAC,CAAC,CAAC,gBAAgB,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;IAEtE,qEAAqE;IACrE,IAAI,aAAa,GAAG,CAAC,CAAC;IACtB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QACrC,IAAI,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,GAAG,GAAG,EAAE,CAAC;YAC7B,aAAa,EAAE,CAAC;QAClB,CAAC;IACH,CAAC;IAED,MAAM,QAAQ,GACZ,SAAS,GAAG,eAAe;QAC3B,cAAc,GAAG,mBAAmB;QACpC,aAAa,GAAG,GAAG,CAAC;IAEtB,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC;AAC3C,CAAC;AAED;;;;;GAKG;AACH,MAAM,UAAU,kBAAkB,CAAC,KAAe;IAChD,IAAI,KAAK,GAAG,CAAC,CAAC;IACd,KAAK,MAAM,CAAC,IAAI,KAAK,EAAE,CAAC;QACtB,KAAK,IAAI,cAAc,CAAC,CAAC,CAAC,CAAC;IAC7B,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC;AAED,iEAAiE;AAEjE;;;GAGG;AACH,MAAM,iBAAiB,GAAa;IAClC,MAAM,EAAO,kCAAkC;IAC/C,KAAK,EAAQ,yCAAyC;IACtD,MAAM,EAAO,6BAA6B;IAC1C,IAAI,EAAS,cAAc;IAC3B,MAAM,EAAO,kBAAkB;IAC/B,KAAK,EAAQ,mBAAmB;IAChC,KAAK,EAAQ,cAAc;IAC3B,IAAI,EAAS,+BAA+B;CAC7C,CAAC;AAEF;;;;;;;;;;;GAWG;AACH,MAAM,UAAU,qBAAqB,CACnC,IAAY,EACZ,SAAiB,EACjB,SAAiB,mBAAmB;IAEpC,IAAI,SAAS,IAAI,CAAC,EAAE,CAAC;QACnB,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,MAAM,aAAa,GAAG,cAAc,CAAC,IAAI,CAAC,CAAC;IAC3C,IAAI,aAAa,IAAI,SAAS,EAAE,CAAC;QAC/B,OAAO,IAAI,CAAC;IACd,CAAC;IAED,MAAM,YAAY,GAAG,cAAc,CAAC,MAAM,CAAC,CAAC;IAC5C,MAAM,YAAY,GAAG,SAAS,GAAG,YAAY,CAAC;IAC9C,IAAI,YAAY,IAAI,CAAC,EAAE,CAAC;QACtB,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,2DAA2D;IAC3D,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,GAAG,aAAa,CAAC;IAE1C,6EAA6E;IAC7E,IAAI,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,GAAG,KAAK,GAAG,IAAI,CAAC,CAAC;IACzD,UAAU,GAAG,IAAI,CAAC,GAAG,CAAC,UAAU,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;IAE/C,IAAI,UAAU,IAAI,CAAC,EAAE,CAAC;QACpB,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,gEAAgE;IAChE,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,UAAU,CAAC,CAAC;IAC5C,IAAI,OAAO,GAAG,UAAU,CAAC;IAEzB,KAAK,MAAM,OAAO,IAAI,iBAAiB,EAAE,CAAC;QACxC,kEAAkE;QAClE,MAAM,SAAS,GAAG,aAAa,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;QACpD,IAAI,SAAS,KAAK,CAAC,CAAC,IAAI,SAAS,GAAG,UAAU,GAAG,GAAG,EAAE,CAAC;YACrD,sEAAsE;YACtE,OAAO,GAAG,SAAS,GAAG,eAAe,CAAC,SAAS,EAAE,OAAO,EAAE,SAAS,CAAC,CAAC;YACrE,MAAM;QACR,CAAC;IACH,CAAC;IAED,IAAI,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC,OAAO,EAAE,GAAG,MAAM,CAAC;IAE1D,8DAA8D;IAC9D,IAAI,QAAQ,GAAG,CAAC,CAAC;IACjB,OAAO,cAAc,CAAC,SAAS,CAAC,GAAG,SAAS,IAAI,OAAO,GAAG,EAAE,IAAI,QAAQ,GAAG,CAAC,EAAE,CAAC;QAC7E,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,GAAG,IAAI,CAAC,CAAC;QACrC,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC,OAAO,EAAE,GAAG,MAAM,CAAC;QACtD,QAAQ,EAAE,CAAC;IACb,CAAC;IAED,OAAO,SAAS,CAAC;AACnB,CAAC;AAED;;;GAGG;AACH,SAAS,aAAa,CAAC,IAAY,EAAE,OAAe;IAClD,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAC/C,IAAI,SAAS,GAAG,CAAC,CAAC,CAAC;IACnB,IAAI,KAA6B,CAAC;IAClC,OAAO,CAAC,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,KAAK,IAAI,EAAE,CAAC;QAC5C,SAAS,GAAG,KAAK,CAAC,KAAK,CAAC;IAC1B,CAAC;IACD,OAAO,SAAS,CAAC;AACnB,CAAC;AAED;;GAEG;AACH,SAAS,eAAe,CAAC,IAAY,EAAE,OAAe,EAAE,KAAa;IACnE,MAAM,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;IAC3C,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;AAC7B,CAAC;AAED;;;;;;GAMG;AACH,MAAM,UAAU,gBAAgB,CAAC,IAAY,EAAE,SAAiB;IAC9D,OAAO,cAAc,CAAC,IAAI,CAAC,IAAI,SAAS,CAAC;AAC3C,CAAC"}

package/install.ps1

@@ -0,0 +1,321 @@
+#Requires -Version 5.1
+<#
+.SYNOPSIS
+    AI Mind Map — One-line installer for Windows.
+
+.DESCRIPTION
+    Installs the AI Mind Map MCP server by cloning the repository,
+    building from source, and optionally configuring detected AI agents.
+
+.PARAMETER InstallDir
+    Custom installation directory. Default: $env:USERPROFILE\.ai-mind-map
+
+.PARAMETER SkipConfig
+    Skip automatic AI agent configuration.
+
+.PARAMETER Update
+    Force update mode — pulls latest changes and rebuilds.
+
+.EXAMPLE
+    # Default install
+    .\install.ps1
+
+    # Custom directory
+    .\install.ps1 -InstallDir "D:\tools\ai-mind-map"
+
+    # Update existing installation
+    .\install.ps1 -Update
+
+.LINK
+    https://github.com/shdra06/ai-mind-map
+#>
+[CmdletBinding()]
+param(
+    [string]$InstallDir = "",
+    [switch]$SkipConfig,
+    [switch]$Update
+)
+
+Set-StrictMode -Version Latest
+$ErrorActionPreference = "Stop"
+
+# ── Constants ────────────────────────────────────────────────────────────────
+$REPO_URL        = "https://github.com/shdra06/ai-mind-map.git"
+$MIN_NODE_MAJOR  = 18
+$ENTRY_POINT     = "dist\index.js"
+
+# ── Banner ───────────────────────────────────────────────────────────────────
+function Show-Banner {
+    $banner = @"
+
+     █████╗ ██╗    ███╗   ███╗██╗███╗   ██╗██████╗     ███╗   ███╗ █████╗ ██████╗
+    ██╔══██╗██║    ████╗ ████║██║████╗  ██║██╔══██╗    ████╗ ████║██╔══██╗██╔══██╗
+    ███████║██║    ██╔████╔██║██║██╔██╗ ██║██║  ██║    ██╔████╔██║███████║██████╔╝
+    ██╔══██║██║    ██║╚██╔╝██║██║██║╚██╗██║██║  ██║    ██║╚██╔╝██║██╔══██║██╔═══╝
+    ██║  ██║██║    ██║ ╚═╝ ██║██║██║ ╚████║██████╔╝    ██║ ╚═╝ ██║██║  ██║██║
+    ╚═╝  ╚═╝╚═╝    ╚═╝     ╚═╝╚═╝╚═╝  ╚═══╝╚═════╝     ╚═╝     ╚═╝╚═╝  ╚═╝╚═╝
+
+    MCP Server — Reduce AI token usage by 80-99%%
+    https://github.com/shdra06/ai-mind-map
+
+"@
+    Write-Host $banner -ForegroundColor Cyan
+}
+
+# ── Helpers ──────────────────────────────────────────────────────────────────
+function Write-Step   { param([string]$Msg) Write-Host "  [*] $Msg" -ForegroundColor Blue }
+function Write-Ok     { param([string]$Msg) Write-Host "  [✓] $Msg" -ForegroundColor Green }
+function Write-Warn   { param([string]$Msg) Write-Host "  [!] $Msg" -ForegroundColor Yellow }
+function Write-Err    { param([string]$Msg) Write-Host "  [✗] $Msg" -ForegroundColor Red }
+
+function Test-CommandExists {
+    param([string]$Command)
+    $null -ne (Get-Command $Command -ErrorAction SilentlyContinue)
+}
+
+function Invoke-StepCommand {
+    param([string]$Label, [string]$Exe, [string[]]$Arguments, [string]$WorkDir)
+    Write-Step $Label
+    $psi = @{
+        FilePath     = $Exe
+        ArgumentList = $Arguments
+        WorkingDirectory = $WorkDir
+        NoNewWindow  = $true
+        Wait         = $true
+        PassThru     = $true
+    }
+    $proc = Start-Process @psi
+    if ($proc.ExitCode -ne 0) {
+        Write-Err "$Label failed with exit code $($proc.ExitCode)."
+        exit 1
+    }
+}
+
+# ── Pre-flight Checks ───────────────────────────────────────────────────────
+function Assert-Prerequisites {
+    Write-Host "`n  Checking prerequisites..." -ForegroundColor White
+
+    # Node.js
+    if (-not (Test-CommandExists "node")) {
+        Write-Err "Node.js is not installed."
+        Write-Err "Download it from: https://nodejs.org (v$MIN_NODE_MAJOR or later)"
+        exit 1
+    }
+    $nodeVersionRaw = (node --version 2>&1).ToString().Trim()
+    $nodeMatch = [regex]::Match($nodeVersionRaw, 'v(\d+)\.')
+    if (-not $nodeMatch.Success) {
+        Write-Err "Could not determine Node.js version from: $nodeVersionRaw"
+        exit 1
+    }
+    $nodeMajor = [int]$nodeMatch.Groups[1].Value
+    if ($nodeMajor -lt $MIN_NODE_MAJOR) {
+        Write-Err "Node.js v$MIN_NODE_MAJOR+ is required. Found: $nodeVersionRaw"
+        Write-Err "Download the latest LTS from: https://nodejs.org"
+        exit 1
+    }
+    Write-Ok "Node.js $nodeVersionRaw"
+
+    # npm (comes with Node, but verify)
+    if (-not (Test-CommandExists "npm")) {
+        Write-Err "npm is not installed. It should come with Node.js — try reinstalling Node."
+        exit 1
+    }
+    $npmVersion = (npm --version 2>&1).ToString().Trim()
+    Write-Ok "npm v$npmVersion"
+
+    # Git
+    if (-not (Test-CommandExists "git")) {
+        Write-Err "Git is not installed."
+        Write-Err "Download it from: https://git-scm.com/download/win"
+        exit 1
+    }
+    $gitVersion = (git --version 2>&1).ToString().Trim()
+    Write-Ok "$gitVersion"
+}
+
+# ── Installation ─────────────────────────────────────────────────────────────
+function Install-AiMindMap {
+    $targetDir = if ($InstallDir -ne "") { $InstallDir } else { Join-Path $env:USERPROFILE ".ai-mind-map" }
+    $isExisting = Test-Path (Join-Path $targetDir ".git")
+
+    # Decide: fresh install or update
+    if ($isExisting -and -not $Update) {
+        Write-Warn "AI Mind Map is already installed at: $targetDir"
+        $choice = Read-Host "  Update to latest version? (Y/n)"
+        if ($choice -match '^[Nn]') {
+            Write-Host "`n  Installation cancelled." -ForegroundColor Gray
+            exit 0
+        }
+        $Update = $true
+    }
+
+    if ($Update -and $isExisting) {
+        # ── Update path ──────────────────────────────────────────────────
+        Write-Host "`n  Updating AI Mind Map..." -ForegroundColor White
+        Write-Step "Pulling latest changes..."
+        $gitPull = Start-Process -FilePath "git" -ArgumentList "pull","--ff-only" `
+            -WorkingDirectory $targetDir -NoNewWindow -Wait -PassThru
+        if ($gitPull.ExitCode -ne 0) {
+            Write-Warn "Fast-forward pull failed. Trying git pull --rebase..."
+            $gitRebase = Start-Process -FilePath "git" -ArgumentList "pull","--rebase" `
+                -WorkingDirectory $targetDir -NoNewWindow -Wait -PassThru
+            if ($gitRebase.ExitCode -ne 0) {
+                Write-Err "Git pull failed. You may have local changes."
+                Write-Err "Resolve manually in: $targetDir"
+                exit 1
+            }
+        }
+        Write-Ok "Repository updated"
+    }
+    else {
+        # ── Fresh install ────────────────────────────────────────────────
+        Write-Host "`n  Installing AI Mind Map..." -ForegroundColor White
+        Write-Step "Cloning repository to: $targetDir"
+        if (Test-Path $targetDir) {
+            Write-Warn "Directory exists but is not a git repo. Removing..."
+            Remove-Item -Recurse -Force $targetDir
+        }
+        $gitClone = Start-Process -FilePath "git" -ArgumentList "clone",$REPO_URL,$targetDir `
+            -NoNewWindow -Wait -PassThru
+        if ($gitClone.ExitCode -ne 0) {
+            Write-Err "Failed to clone repository."
+            Write-Err "Check your internet connection and try again."
+            exit 1
+        }
+        Write-Ok "Repository cloned"
+    }
+
+    # ── Install dependencies ─────────────────────────────────────────────
+    Invoke-StepCommand -Label "Installing dependencies (this may take a minute)..." `
+        -Exe "npm" -Arguments @("install","--legacy-peer-deps") -WorkDir $targetDir
+    Write-Ok "Dependencies installed"
+
+    # ── Build ────────────────────────────────────────────────────────────
+    Invoke-StepCommand -Label "Building TypeScript..." `
+        -Exe "npx" -Arguments @("tsc") -WorkDir $targetDir
+    Write-Ok "Build complete"
+
+    # ── Verify build ─────────────────────────────────────────────────────
+    $entryFile = Join-Path $targetDir $ENTRY_POINT
+    if (-not (Test-Path $entryFile)) {
+        Write-Err "Build verification failed: $ENTRY_POINT not found."
+        Write-Err "Please report this issue: https://github.com/shdra06/ai-mind-map/issues"
+        exit 1
+    }
+    Write-Ok "Build verified ($ENTRY_POINT exists)"
+
+    # ── PATH management ──────────────────────────────────────────────────
+    Add-ToUserPath -Dir $targetDir
+
+    # ── Agent configuration ──────────────────────────────────────────────
+    if (-not $SkipConfig) {
+        Configure-Agents -Dir $targetDir
+    }
+
+    # ── Done ─────────────────────────────────────────────────────────────
+    Show-Success -Dir $targetDir
+}
+
+# ── PATH ─────────────────────────────────────────────────────────────────────
+function Add-ToUserPath {
+    param([string]$Dir)
+
+    $currentPath = [Environment]::GetEnvironmentVariable("PATH", "User")
+    if ($currentPath -and $currentPath.Split(';') -contains $Dir) {
+        Write-Ok "Install directory already in PATH"
+        return
+    }
+
+    try {
+        $newPath = if ($currentPath) { "$currentPath;$Dir" } else { $Dir }
+        [Environment]::SetEnvironmentVariable("PATH", $newPath, "User")
+        # Also update current session
+        $env:PATH = "$env:PATH;$Dir"
+        Write-Ok "Added to user PATH: $Dir"
+        Write-Warn "Restart your terminal for PATH changes to take effect."
+    }
+    catch {
+        Write-Warn "Could not update PATH automatically."
+        Write-Warn "Manually add this directory to your PATH: $Dir"
+    }
+}
+
+# ── Agent Auto-Configuration ────────────────────────────────────────────────
+function Configure-Agents {
+    param([string]$Dir)
+
+    Write-Host "`n  Detecting AI agents..." -ForegroundColor White
+
+    $agents = @()
+
+    # Claude Desktop / Claude Code
+    $claudeConfig = Join-Path $env:APPDATA "Claude\claude_desktop_config.json"
+    if (Test-Path $claudeConfig) { $agents += "Claude Desktop" }
+    $claudeCodeDir = Join-Path $env:USERPROFILE ".claude"
+    if (Test-Path $claudeCodeDir) { $agents += "Claude Code" }
+
+    # Cursor
+    $cursorDir = Join-Path $env:USERPROFILE ".cursor"
+    if (Test-Path $cursorDir) { $agents += "Cursor" }
+
+    # VS Code
+    $vscodeDir = Join-Path $env:APPDATA "Code\User"
+    if (Test-Path $vscodeDir) { $agents += "VS Code" }
+
+    # Windsurf / Codeium
+    $windsurfDir = Join-Path $env:APPDATA "Windsurf"
+    if (Test-Path $windsurfDir) { $agents += "Windsurf" }
+
+    if ($agents.Count -eq 0) {
+        Write-Warn "No AI agents detected. See README for manual setup."
+        return
+    }
+
+    Write-Ok "Detected: $($agents -join ', ')"
+    Write-Host ""
+    Write-Host "  To configure an agent, add this to its MCP config:" -ForegroundColor White
+    $escapedDir = ($Dir -replace '\\', '\\')
+    Write-Host @"
+
+    {
+      "mcpServers": {
+        "ai-mind-map": {
+          "command": "node",
+          "args": [
+            "$escapedDir\\dist\\index.js",
+            "--project-root",
+            "<YOUR_PROJECT_PATH>"
+          ]
+        }
+      }
+    }
+
+"@ -ForegroundColor Gray
+}
+
+# ── Success ──────────────────────────────────────────────────────────────────
+function Show-Success {
+    param([string]$Dir)
+
+    $divider = "─" * 60
+    Write-Host ""
+    Write-Host "  $divider" -ForegroundColor Green
+    Write-Host "  ✅  AI Mind Map installed successfully!" -ForegroundColor Green
+    Write-Host "  $divider" -ForegroundColor Green
+    Write-Host ""
+    Write-Host "  Location:  $Dir" -ForegroundColor White
+    Write-Host ""
+    Write-Host "  Next steps:" -ForegroundColor White
+    Write-Host "    1. Configure your AI agent (see above or README)" -ForegroundColor Gray
+    Write-Host "    2. Test with:  node `"$Dir\$ENTRY_POINT`" --project-root ." -ForegroundColor Gray
+    Write-Host "    3. Star the repo: https://github.com/shdra06/ai-mind-map" -ForegroundColor Gray
+    Write-Host ""
+    Write-Host "  To update later:  .\install.ps1 -Update" -ForegroundColor Yellow
+    Write-Host "  Full docs:        https://github.com/shdra06/ai-mind-map#readme" -ForegroundColor Yellow
+    Write-Host ""
+}
+
+# ── Main ─────────────────────────────────────────────────────────────────────
+Show-Banner
+Assert-Prerequisites
+Install-AiMindMap