nia-mcp-server 1.0.5__py3-none-any.whl → 1.0.7__py3-none-any.whl

nia_mcp_server/assets/rules/vscode_rules.md

@@ -0,0 +1,312 @@
+# NIA Integration for Visual Studio Code
+
+## VSCode-Specific Configuration
+
+### 1. Workspace Settings
+Configure `.vscode/settings.json` for NIA integration:
+
+```json
+{
+  "nia.defaultRepositories": [
+    "owner/main-project",
+    "owner/dependency-lib"
+  ],
+  "nia.searchHistory": true,
+  "nia.autoIndex": {
+    "enabled": true,
+    "patterns": ["github.com/*/*"]
+  },
+  "nia.apiEndpoint": "https://api.trynia.ai",
+  "nia.cacheTimeout": 3600
+}
+```
+
+### 2. Tasks Configuration
+Set up `.vscode/tasks.json` for common NIA operations:
+
+```json
+{
+  "version": "2.0.0",
+  "tasks": [
+    {
+      "label": "NIA: Index Current Repository",
+      "type": "shell",
+      "command": "echo 'index_repository ${workspaceFolder}'",
+      "problemMatcher": []
+    },
+    {
+      "label": "NIA: List Indexed Repositories",
+      "type": "shell",
+      "command": "echo 'list_repositories'",
+      "problemMatcher": []
+    },
+    {
+      "label": "NIA: Search Codebase",
+      "type": "shell",
+      "command": "echo 'search_codebase \"${input:searchQuery}\"'",
+      "problemMatcher": []
+    }
+  ],
+  "inputs": [
+    {
+      "id": "searchQuery",
+      "type": "promptString",
+      "description": "Enter your search query"
+    }
+  ]
+}
+```
+
+### 3. Code Snippets
+Add to `.vscode/nia.code-snippets`:
+
+```json
+{
+  "Nia Index Repository": {
+    "prefix": "nia-index",
+    "body": [
+      "index_repository ${1:https://github.com/owner/repo}"
+    ],
+    "description": "Index a repository with Nia"
+  },
+  "Nia Search Code": {
+    "prefix": "nia-search",
+    "body": [
+      "search_codebase \"${1:How does authentication work?}\""
+    ],
+    "description": "Search indexed codebases"
+  },
+  "Nia Web Search": {
+    "prefix": "nia-web",
+    "body": [
+      "nia_web_search \"${1:find RAG implementation libraries}\""
+    ],
+    "description": "Search the web with Nia"
+  },
+  "Nia Deep Research": {
+    "prefix": "nia-research",
+    "body": [
+      "nia_deep_research_agent \"${1:compare X vs Y for use case}\""
+    ],
+    "description": "Perform deep research analysis"
+  }
+}
+```
+
+### 4. Launch Configuration
+Configure `.vscode/launch.json` for debugging with NIA context:
+
+```json
+{
+  "version": "0.2.0",
+  "configurations": [
+    {
+      "name": "Debug with NIA Context",
+      "type": "node",
+      "request": "launch",
+      "program": "${workspaceFolder}/index.js",
+      "env": {
+        "NIA_API_KEY": "${env:NIA_API_KEY}",
+        "NIA_INDEXED_REPOS": "${workspaceFolder}/.nia/indexed_repos.json"
+      }
+    }
+  ]
+}
+```
+
+### 5. Extensions Recommendations
+Add to `.vscode/extensions.json`:
+
+```json
+{
+  "recommendations": [
+    "github.copilot",
+    "continue.continue",
+    "codeium.codeium"
+  ]
+}
+```
+
+## VSCode Command Palette Integration
+
+### Quick Commands
+Access via `Cmd/Ctrl + Shift + P`:
+
+1. **Nia: Index This Repository**
+   - Automatically indexes the current workspace
+   - Shows progress in status bar
+   - Notifies when complete
+
+2. **Nia: Search Symbol**
+   - Search for functions, classes, or variables
+   - Across all indexed repositories
+   - Jump to definition support
+
+3. **Nia: Explain Code**
+   - Select code and search for explanations
+   - Find similar implementations
+   - Understand patterns
+
+4. **Nia: Find Examples**
+   - Search for usage examples
+   - Filter by language or framework
+   - Copy-paste ready snippets
+
+## VSCode-Specific Workflows
+
+### 1. Project Onboarding
+When opening a new project:
+```
+1. Check if repo is indexed: list_repositories
+2. If not, index it: index_repository [current repo]
+3. Index related dependencies
+4. Search for architecture overview
+5. Create workspace documentation
+```
+
+### 2. Code Review Assistance
+During code reviews:
+```
+1. Search for similar patterns in indexed repos
+2. Find best practices for the specific feature
+3. Check for common pitfalls
+4. Suggest improvements based on examples
+```
+
+### 3. Debugging Workflow
+When debugging issues:
+```
+1. Search for error messages across repos
+2. Find similar bug fixes
+3. Understand the problematic code context
+4. Search for test cases
+```
+
+### 4. Learning & Documentation
+For learning new codebases:
+```
+1. Index the main repository
+2. Search for entry points (main, index, app)
+3. Understand the architecture
+4. Find example usage patterns
+```
+
+## Terminal Integration
+
+### Integrated Terminal Commands
+Use in VSCode's integrated terminal:
+
+```bash
+# Quick index
+alias nia-index='echo "index_repository $(git remote get-url origin)"'
+
+# Search current project
+alias nia-search='echo "search_codebase"'
+
+# List all indexed
+alias nia-list='echo "list_repositories"'
+```
+
+### PowerShell Functions
+For Windows users in `.vscode/powershell_profile.ps1`:
+
+```powershell
+function Nia-Index {
+    param($repo)
+    Write-Host "index_repository $repo"
+}
+
+function Nia-Search {
+    param($query)
+    Write-Host "search_codebase `"$query`""
+}
+```
+
+## Output Formatting for VSCode
+
+### 1. File References
+Format as clickable links:
+```
+📁 authentication/middleware.ts:45
+📁 lib/auth/session.ts:23-67
+📁 pages/api/auth/[...nextauth].ts:12
+```
+
+### 2. Code Blocks
+Use proper syntax highlighting:
+```typescript
+// From: lib/auth/session.ts
+export async function getSession(req: Request) {
+  // Implementation details
+}
+```
+
+### 3. Problem Markers
+For issues found during search:
+```
+⚠️ Warning: Deprecated pattern found
+   File: utils/oldAuth.js:34
+   Suggestion: Use new auth method
+
+❌ Error: Security vulnerability
+   File: api/user.js:56
+   Fix: Validate input parameters
+```
+
+## Keyboard Shortcuts
+Suggested keybindings for `.vscode/keybindings.json`:
+
+```json
+[
+  {
+    "key": "ctrl+shift+n i",
+    "command": "workbench.action.terminal.sendSequence",
+    "args": { "text": "index_repository " }
+  },
+  {
+    "key": "ctrl+shift+n s",
+    "command": "workbench.action.terminal.sendSequence",
+    "args": { "text": "search_codebase \"" }
+  },
+  {
+    "key": "ctrl+shift+n l",
+    "command": "workbench.action.terminal.sendSequence",
+    "args": { "text": "list_repositories\n" }
+  }
+]
+```
+
+## Performance Optimization
+
+### 1. Workspace-Specific Cache
+Store frequently searched queries:
+```
+.vscode/.nia-cache/
+├── search-results.json
+├── indexed-files.json
+└── common-queries.json
+```
+
+### 2. Search Filters
+Use repository context for faster searches:
+```
+search_codebase "authentication" repositories=["current-project"]
+```
+
+### 3. Batch Operations
+Index related repositories together:
+```
+index_repository repo1
+index_repository repo2
+index_repository repo3
+```
+
+### 4. Documentation Indexing
+When indexing documentation:
+- If user provides just a URL without patterns, use `["/*"]` to index everything
+- Or ask: "Should I index the entire documentation or specific sections?"
+- Common patterns: `["/docs/*", "/api/*", "/guide/*", "/reference/*"]`
+- Example: `index_documentation "https://docs.example.com" ["/*"]`
+- Can index multiple doc sites for comprehensive searches
+
+Remember: Make Nia feel like a native part of the VSCode experience!

nia_mcp_server/assets/rules/windsurf_rules.md

@@ -0,0 +1,92 @@
+# Nia Integration for Windsurf Cascade
+
+## Quick Reference
+- **List indexed repos**: `list_repositories`
+- **Index new repo**: `index_repository repo_url`
+- **Index docs**: `index_documentation url` (use `["/*"]` for entire site)
+- **Search code**: `search_codebase "your question"`
+- **Search docs**: `search_documentation "your question"`
+- **Web search**: `nia_web_search "find X library"`
+- **Deep research**: `nia_deep_research_agent "compare X vs Y"`
+
+## Windsurf Cascade Guidelines
+
+### 1. Flow-Based Development
+When using Cascade's flow feature:
+- Start flows by checking indexed repositories with `list_repositories`
+- Auto-suggest indexing when users mention GitHub URLs
+- Use Nia search results to inform multi-step flows
+- Reference specific files found via search in subsequent steps
+
+### 2. Memory Integration
+Create memories for frequently used repositories:
+- "Remember: Project X is indexed at owner/repo"
+- "Remember: Use Nia to search for implementation patterns"
+- Memories will automatically recall indexed repos context
+
+### 3. Code Understanding Workflow
+For codebase exploration:
+```
+1. Check if indexed: list_repositories
+2. If not indexed: index_repository [url]
+3. Search naturally: search_codebase "How does X work?"
+4. Use results in flow steps
+```
+
+### 4. Search Optimization for Cascade
+
+#### Natural Language Queries
+- Write complete questions: "How is authentication implemented in this Next.js app?"
+- Include context: "Find React hooks that handle user state"
+- Be specific: "Show me where API routes are defined"
+
+#### Multi-Step Flows
+1. Research: `nia_deep_research_agent "compare state management libraries"`
+2. Find examples: `nia_web_search "Redux toolkit examples"`
+3. Index chosen repo: `index_repository https://github.com/...`
+4. Explore patterns: `search_codebase "action creators pattern"`
+
+### 5. Cascade Commands
+
+When users ask about code:
+- Always check if the repository is indexed first
+- Suggest indexing if working with a new codebase
+- Use search results to build comprehensive responses
+- Create flows that combine multiple searches
+
+### 6. Best Practices
+
+#### For New Projects
+1. Index the main repository immediately
+2. Search for architecture overview: "How is this project structured?"
+3. Create memories of key findings
+4. Build flows based on discovered patterns
+
+#### For Debugging
+1. Search for error messages across indexed repos
+2. Find similar implementations that work
+3. Compare working vs broken code
+4. Create fix flow based on findings
+
+#### For Feature Development
+1. Search for similar features in indexed repos
+2. Understand existing patterns
+3. Find best practices examples
+4. Generate code following discovered patterns
+
+### 7. Documentation Indexing
+
+When indexing documentation:
+- If user provides just a URL without patterns, use `["/*"]` to index everything
+- Or ask: "Should I index the entire documentation or specific sections?"
+- Common patterns: `["/docs/*", "/api/*", "/guide/*", "/reference/*"]`
+- Example: `index_documentation "https://docs.example.com" ["/*"]`
+
+### 8. Integration Tips
+
+- Nia complements Cascade's abilities by providing deep code search
+- Use Nia for finding specific implementations
+- Let Cascade handle the creative synthesis
+- Combine both for powerful development workflows
+
+Remember: Nia provides the knowledge, Cascade provides the flow!

nia_mcp_server/profiles.py

@@ -0,0 +1,263 @@
+"""
+Nia Profile Configurations
+Defines IDE/Editor profile configurations for rule transformation
+"""
+from typing import Dict, Any, Optional, List
+
+# Profile configurations define how rules are transformed and where they're placed
+PROFILE_CONFIGS: Dict[str, Dict[str, Any]] = {
+    "cursor": {
+        "name": "Cursor",
+        "target_dir": ".cursor/rules",
+        "file_extension": ".mdc",
+        "file_map": {
+            "cursor_rules.md": "nia.mdc"
+        },
+        "mcp_config": True,
+        "format": "mdc",
+        "features": ["mcp", "composer", "inline_edits"],
+        "global_replacements": {
+            "# Nia": "# Nia for Cursor",
+            "{{IDE}}": "Cursor"
+        }
+    },
+    
+    "vscode": {
+        "name": "Visual Studio Code",
+        "target_dir": ".vscode",
+        "file_extension": ".md",
+        "file_map": {
+            "nia_rules.md": "nia-guide.md",
+            "vscode_rules.md": "nia-vscode-integration.md"
+        },
+        "mcp_config": False,
+        "format": "markdown",
+        "features": ["tasks", "snippets", "terminal_integration"],
+        "global_replacements": {
+            "# Nia": "# Nia for VSCode",
+            "{{IDE}}": "VSCode"
+        },
+        "additional_files": {
+            "tasks.json": "vscode_tasks_template",
+            "nia.code-snippets": "vscode_snippets_template"
+        }
+    },
+    
+    "claude": {
+        "name": "Claude Desktop",
+        "target_dir": ".claude",
+        "file_extension": ".md",
+        "file_map": {
+            "nia_rules.md": "nia_rules.md",
+            "claude_rules.md": "nia_claude_integration.md"
+        },
+        "mcp_config": False,
+        "format": "markdown",
+        "features": ["conversational", "context_aware", "multi_step"],
+        "global_replacements": {
+            "# Nia": "# Nia for Claude Desktop",
+            "{{IDE}}": "Claude"
+        }
+    },
+    
+    "windsurf": {
+        "name": "Windsurf",
+        "target_dir": ".windsurfrules",
+        "file_extension": ".md",
+        "file_map": {
+            "nia_rules.md": "nia_rules.md",
+            "windsurf_rules.md": "windsurf_nia_guide.md"
+        },
+        "mcp_config": True,
+        "format": "markdown",
+        "features": ["cascade", "memories", "flows"],
+        "global_replacements": {
+            "# Nia": "# Nia for Windsurf Cascade",
+            "{{IDE}}": "Windsurf"
+        }
+    },
+    
+    "cline": {
+        "name": "Cline",
+        "target_dir": ".cline",
+        "file_extension": ".md",
+        "file_map": {
+            "nia_rules.md": "nia_rules.md"
+        },
+        "mcp_config": True,
+        "format": "markdown",
+        "features": ["autonomous", "task_planning"],
+        "global_replacements": {
+            "# Nia": "# Nia for Cline",
+            "{{IDE}}": "Cline"
+        }
+    },
+    
+    "codex": {
+        "name": "OpenAI Codex",
+        "target_dir": ".codex",
+        "file_extension": ".md",
+        "file_map": {
+            "nia_rules.md": "nia_codex_guide.md"
+        },
+        "mcp_config": False,
+        "format": "markdown",
+        "features": ["completion", "generation"],
+        "global_replacements": {
+            "# Nia": "# Nia for Codex",
+            "{{IDE}}": "Codex"
+        }
+    },
+    
+    "zed": {
+        "name": "Zed",
+        "target_dir": ".zed",
+        "file_extension": ".md",
+        "file_map": {
+            "nia_rules.md": "nia_assistant.md"
+        },
+        "mcp_config": False,
+        "format": "markdown",
+        "features": ["assistant", "collaboration"],
+        "global_replacements": {
+            "# Nia": "# Nia for Zed",
+            "{{IDE}}": "Zed"
+        }
+    },
+    
+    "jetbrains": {
+        "name": "JetBrains IDEs",
+        "target_dir": ".idea/nia",
+        "file_extension": ".md",
+        "file_map": {
+            "nia_rules.md": "nia_guide.md"
+        },
+        "mcp_config": False,
+        "format": "markdown",
+        "features": ["ai_assistant", "code_completion"],
+        "global_replacements": {
+            "# Nia": "# Nia for JetBrains",
+            "{{IDE}}": "JetBrains IDE"
+        }
+    },
+    
+    "neovim": {
+        "name": "Neovim",
+        "target_dir": ".config/nvim/nia",
+        "file_extension": ".md",
+        "file_map": {
+            "nia_rules.md": "nia_guide.md"
+        },
+        "mcp_config": False,
+        "format": "markdown",
+        "features": ["copilot", "cmp"],
+        "global_replacements": {
+            "# Nia": "# Nia for Neovim",
+            "{{IDE}}": "Neovim"
+        }
+    },
+    
+    "sublime": {
+        "name": "Sublime Text",
+        "target_dir": ".sublime",
+        "file_extension": ".md",
+        "file_map": {
+            "nia_rules.md": "nia_guide.md"
+        },
+        "mcp_config": False,
+        "format": "markdown",
+        "features": ["copilot"],
+        "global_replacements": {
+            "# Nia": "# Nia for Sublime Text",
+            "{{IDE}}": "Sublime Text"
+        }
+    }
+}
+
+# Additional template configurations for specific file types
+TEMPLATE_CONFIGS = {
+    "vscode_tasks_template": {
+        "content": """{
+  "version": "2.0.0",
+  "tasks": [
+    {
+      "label": "Nia: Index Repository",
+      "type": "shell",
+      "command": "echo 'Run: index_repository ${input:repoUrl}'",
+      "problemMatcher": []
+    },
+    {
+      "label": "Nia: Search Codebase",
+      "type": "shell",
+      "command": "echo 'Run: search_codebase \\"${input:searchQuery}\\"'",
+      "problemMatcher": []
+    },
+    {
+      "label": "Nia: List Repositories",
+      "type": "shell",
+      "command": "echo 'Run: list_repositories'",
+      "problemMatcher": []
+    }
+  ],
+  "inputs": [
+    {
+      "id": "repoUrl",
+      "type": "promptString",
+      "description": "GitHub repository URL"
+    },
+    {
+      "id": "searchQuery",
+      "type": "promptString",
+      "description": "Search query"
+    }
+  ]
+}"""
+    },
+    
+    "vscode_snippets_template": {
+        "content": """{
+  "Nia Index": {
+    "prefix": "nia-index",
+    "body": ["index_repository ${1:repo_url}"],
+    "description": "Index a repository with Nia"
+  },
+  "Nia Search": {
+    "prefix": "nia-search",
+    "body": ["search_codebase \\"${1:query}\\""],
+    "description": "Search indexed repositories"
+  },
+  "Nia Web Search": {
+    "prefix": "nia-web",
+    "body": ["nia_web_search \\"${1:query}\\""],
+    "description": "Search the web with Nia"
+  },
+  "Nia Research": {
+    "prefix": "nia-research",
+    "body": ["nia_deep_research_agent \\"${1:query}\\""],
+    "description": "Perform deep research with Nia"
+  }
+}"""
+    }
+}
+
+
+def get_profile_config(profile: str) -> Optional[Dict[str, Any]]:
+    """Get configuration for a specific profile"""
+    return PROFILE_CONFIGS.get(profile)
+
+
+def get_supported_profiles() -> List[str]:
+    """Get list of all supported profiles"""
+    return list(PROFILE_CONFIGS.keys())
+
+
+def get_profile_features(profile: str) -> List[str]:
+    """Get features supported by a profile"""
+    config = get_profile_config(profile)
+    return config.get("features", []) if config else []
+
+
+def is_mcp_enabled(profile: str) -> bool:
+    """Check if a profile supports MCP"""
+    config = get_profile_config(profile)
+    return config.get("mcp_config", False) if config else False