Add browser extension and shell completions (Milestones 12-13)

M12: Browser Extension
- Chrome/Edge Manifest V3 extension
- Popup UI for saving pages
- Context menu integration (save page/selection/link)
- Background service worker
- Content script for extraction

M13: Shell Completions
- Bash, Zsh, Fish, PowerShell completions
- Dynamic node ID completion
- Dynamic tag completion
- Dynamic graph completion
- Auto-install command (--install)
This commit is contained in:
2026-02-03 11:35:41 +01:00
parent b1c62c5da9
commit f21426fc43
11 changed files with 1248 additions and 0 deletions

75
extension/README.md Normal file
View File

@@ -0,0 +1,75 @@
# Cortex Browser Extension
Save web content directly to your Cortex knowledge graph.
## Features
- One-click save from any webpage
- Right-click context menu integration
- Save selected text only
- Auto-extract page title and content
- Tag suggestions based on URL
## Installation
### Chrome / Edge
1. Open `chrome://extensions` (or `edge://extensions`)
2. Enable "Developer mode" in the top right
3. Click "Load unpacked"
4. Select this `extension` folder
### Firefox
1. Open `about:debugging`
2. Click "This Firefox"
3. Click "Load Temporary Add-on"
4. Select `manifest.json` from this folder
## Usage
### Popup
1. Click the Cortex icon in your browser toolbar
2. Edit the title and tags as needed
3. Choose whether to include page content
4. Click "Save"
### Context Menu
- **Save page**: Right-click anywhere on a page → "Save page to Cortex"
- **Save selection**: Select text, right-click → "Save selection to Cortex"
- **Save link**: Right-click a link → "Save link to Cortex"
## Requirements
The Cortex server must be running for the extension to work:
```bash
cortex serve
```
By default, the extension connects to `http://localhost:3100/api`.
## Development
The extension uses Manifest V3 and consists of:
- `manifest.json` - Extension configuration
- `popup/` - Popup UI (HTML, CSS, JS)
- `background/` - Service worker for context menus
- `content/` - Content script for page extraction
- `icons/` - Extension icons
## Icon Generation
To generate PNG icons from the SVG:
```bash
# Using ImageMagick
convert icons/icon.svg -resize 16x16 icons/icon-16.png
convert icons/icon.svg -resize 48x48 icons/icon-48.png
convert icons/icon.svg -resize 128x128 icons/icon-128.png
```
Or use an online SVG to PNG converter.

View File

@@ -0,0 +1,136 @@
// Background service worker for Cortex browser extension
const CORTEX_API = 'http://localhost:3100/api';
// Create context menu on install
chrome.runtime.onInstalled.addListener(() => {
// Save page context menu
chrome.contextMenus.create({
id: 'cortex-save-page',
title: 'Save page to Cortex',
contexts: ['page'],
});
// Save selection context menu
chrome.contextMenus.create({
id: 'cortex-save-selection',
title: 'Save selection to Cortex',
contexts: ['selection'],
});
// Save link context menu
chrome.contextMenus.create({
id: 'cortex-save-link',
title: 'Save link to Cortex',
contexts: ['link'],
});
});
// Handle context menu clicks
chrome.contextMenus.onClicked.addListener(async (info, tab) => {
try {
let content = '';
let title = tab.title;
if (info.menuItemId === 'cortex-save-selection') {
content = info.selectionText || '';
title = `Selection from: ${tab.title}`;
} else if (info.menuItemId === 'cortex-save-link') {
content = `Link: ${info.linkUrl}`;
title = info.linkUrl;
} else {
// Get full page content
const [{ result }] = await chrome.scripting.executeScript({
target: { tabId: tab.id },
func: extractPageContent,
});
content = result?.content || '';
title = result?.title || tab.title;
}
// Save to Cortex
await saveToCortex({
title,
content,
url: tab.url,
kind: 'memory',
tags: ['web-clip'],
});
// Show notification (if supported)
if (chrome.notifications) {
chrome.notifications.create({
type: 'basic',
iconUrl: 'icons/icon-48.png',
title: 'Saved to Cortex',
message: title.slice(0, 50),
});
}
} catch (err) {
console.error('Context menu action failed:', err);
}
});
// Function to extract page content (injected into page)
function extractPageContent() {
// Simple content extraction
// A full implementation would use Readability
const title = document.title;
// Try to get main content
const article = document.querySelector('article');
const main = document.querySelector('main');
const content = article?.textContent || main?.textContent || document.body.textContent || '';
// Clean up content
const cleanContent = content
.replace(/\s+/g, ' ')
.replace(/\n+/g, '\n')
.trim()
.slice(0, 50000); // Limit content size
return {
title,
content: cleanContent,
url: window.location.href,
};
}
// Save to Cortex API
async function saveToCortex(data) {
const response = await fetch(`${CORTEX_API}/nodes`, {
method: 'POST',
headers: {
'Content-Type': 'application/json',
},
body: JSON.stringify({
kind: data.kind || 'memory',
title: data.title,
content: data.content,
tags: data.tags || ['web-clip'],
metadata: {
source: {
type: 'url',
url: data.url,
savedAt: Date.now(),
},
},
}),
});
if (!response.ok) {
throw new Error(`HTTP ${response.status}`);
}
return response.json();
}
// Listen for messages from content script or popup
chrome.runtime.onMessage.addListener((request, sender, sendResponse) => {
if (request.action === 'save') {
saveToCortex(request.data)
.then(node => sendResponse({ success: true, nodeId: node.id }))
.catch(err => sendResponse({ success: false, error: err.message }));
return true; // Keep channel open for async response
}
});

View File

@@ -0,0 +1,80 @@
// Content script for Cortex browser extension
// Injected into web pages to extract content
// Extract page content
function extractContent() {
const title = document.title;
// Get selected text
const selection = window.getSelection()?.toString() || '';
// Try to find main content using common selectors
let content = '';
const contentSelectors = [
'article',
'[role="main"]',
'main',
'.post-content',
'.article-content',
'.entry-content',
'.content',
'#content',
];
for (const selector of contentSelectors) {
const el = document.querySelector(selector);
if (el && el.textContent) {
content = el.textContent;
break;
}
}
// Fall back to body content
if (!content) {
content = document.body.textContent || '';
}
// Clean up content
content = content
.replace(/\s+/g, ' ')
.replace(/\n{3,}/g, '\n\n')
.trim()
.slice(0, 100000); // Limit to 100k chars
// Get meta description
const metaDesc = document.querySelector('meta[name="description"]')?.content || '';
// Get author
const author = document.querySelector('meta[name="author"]')?.content ||
document.querySelector('[rel="author"]')?.textContent || '';
// Get published date
const datePublished = document.querySelector('meta[property="article:published_time"]')?.content ||
document.querySelector('time[datetime]')?.getAttribute('datetime') || '';
return {
title,
content,
selection,
excerpt: metaDesc || content.slice(0, 200),
url: window.location.href,
author,
datePublished,
};
}
// Listen for messages from popup or background
chrome.runtime.onMessage.addListener((request, sender, sendResponse) => {
if (request.action === 'extract') {
try {
const data = extractContent();
sendResponse(data);
} catch (err) {
sendResponse({ error: err.message });
}
}
return true; // Keep channel open for async response
});
// Make extract function available globally for background script injection
window.__cortexExtract = extractContent;

23
extension/icons/icon.svg Normal file
View File

@@ -0,0 +1,23 @@
<svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 128 128">
<defs>
<linearGradient id="grad" x1="0%" y1="0%" x2="100%" y2="100%">
<stop offset="0%" style="stop-color:#4fc3f7;stop-opacity:1" />
<stop offset="100%" style="stop-color:#29b6f6;stop-opacity:1" />
</linearGradient>
</defs>
<circle cx="64" cy="64" r="56" fill="url(#grad)"/>
<g fill="#fff">
<!-- Brain/memory icon representation -->
<circle cx="64" cy="48" r="8"/>
<circle cx="44" cy="68" r="6"/>
<circle cx="84" cy="68" r="6"/>
<circle cx="54" cy="88" r="5"/>
<circle cx="74" cy="88" r="5"/>
<!-- Connections -->
<path d="M64 56 L44 62" stroke="#fff" stroke-width="2" fill="none"/>
<path d="M64 56 L84 62" stroke="#fff" stroke-width="2" fill="none"/>
<path d="M44 74 L54 83" stroke="#fff" stroke-width="2" fill="none"/>
<path d="M84 74 L74 83" stroke="#fff" stroke-width="2" fill="none"/>
<path d="M54 88 L74 88" stroke="#fff" stroke-width="2" fill="none"/>
</g>
</svg>

After

Width:  |  Height:  |  Size: 1012 B

36
extension/manifest.json Normal file
View File

@@ -0,0 +1,36 @@
{
"manifest_version": 3,
"name": "Cortex - Save to Memory",
"version": "1.0.0",
"description": "Save web content to your Cortex knowledge graph",
"permissions": [
"activeTab",
"contextMenus",
"storage"
],
"host_permissions": [
"http://localhost:3100/*"
],
"action": {
"default_popup": "popup/popup.html",
"default_icon": {
"16": "icons/icon-16.png",
"48": "icons/icon-48.png",
"128": "icons/icon-128.png"
}
},
"background": {
"service_worker": "background/background.js"
},
"content_scripts": [
{
"matches": ["<all_urls>"],
"js": ["content/content.js"]
}
],
"icons": {
"16": "icons/icon-16.png",
"48": "icons/icon-48.png",
"128": "icons/icon-128.png"
}
}

136
extension/popup/popup.css Normal file
View File

@@ -0,0 +1,136 @@
* {
box-sizing: border-box;
margin: 0;
padding: 0;
}
body {
font-family: -apple-system, BlinkMacSystemFont, 'Segoe UI', Roboto, sans-serif;
font-size: 14px;
width: 320px;
padding: 16px;
background: #1a1a2e;
color: #eee;
}
.container {
display: flex;
flex-direction: column;
gap: 12px;
}
h1 {
font-size: 18px;
font-weight: 600;
color: #4fc3f7;
margin-bottom: 8px;
}
.status {
padding: 8px;
border-radius: 4px;
font-size: 12px;
display: none;
}
.status.success {
display: block;
background: #1b5e20;
color: #a5d6a7;
}
.status.error {
display: block;
background: #b71c1c;
color: #ef9a9a;
}
.field {
display: flex;
flex-direction: column;
gap: 4px;
}
.field.checkbox {
flex-direction: row;
align-items: center;
}
.field.checkbox label {
display: flex;
align-items: center;
gap: 8px;
cursor: pointer;
}
label {
font-size: 12px;
color: #aaa;
font-weight: 500;
}
input[type="text"],
select {
padding: 8px 12px;
border: 1px solid #333;
border-radius: 4px;
background: #0f0f23;
color: #eee;
font-size: 14px;
}
input[type="text"]:focus,
select:focus {
outline: none;
border-color: #4fc3f7;
}
select {
cursor: pointer;
}
input[type="checkbox"] {
width: 16px;
height: 16px;
cursor: pointer;
}
.actions {
display: flex;
gap: 8px;
margin-top: 8px;
}
button {
flex: 1;
padding: 10px 16px;
border: none;
border-radius: 4px;
font-size: 14px;
font-weight: 500;
cursor: pointer;
transition: background 0.2s;
}
button.primary {
background: #4fc3f7;
color: #000;
}
button.primary:hover {
background: #29b6f6;
}
button:not(.primary) {
background: #333;
color: #eee;
}
button:not(.primary):hover {
background: #444;
}
button:disabled {
opacity: 0.5;
cursor: not-allowed;
}

View File

@@ -0,0 +1,55 @@
<!DOCTYPE html>
<html>
<head>
<meta charset="UTF-8">
<link rel="stylesheet" href="popup.css">
</head>
<body>
<div class="container">
<h1>Save to Cortex</h1>
<div id="status" class="status"></div>
<div class="field">
<label for="title">Title</label>
<input type="text" id="title" placeholder="Page title">
</div>
<div class="field">
<label for="kind">Kind</label>
<select id="kind">
<option value="memory">Memory</option>
<option value="component">Component</option>
<option value="decision">Decision</option>
<option value="task">Task</option>
</select>
</div>
<div class="field">
<label for="tags">Tags</label>
<input type="text" id="tags" placeholder="comma, separated, tags">
</div>
<div class="field checkbox">
<label>
<input type="checkbox" id="includeContent" checked>
Include page content
</label>
</div>
<div class="field checkbox">
<label>
<input type="checkbox" id="selectionOnly">
Selection only
</label>
</div>
<div class="actions">
<button id="save" class="primary">Save</button>
<button id="cancel">Cancel</button>
</div>
</div>
<script src="popup.js"></script>
</body>
</html>

136
extension/popup/popup.js Normal file
View File

@@ -0,0 +1,136 @@
// Popup script for Cortex browser extension
const CORTEX_API = 'http://localhost:3100/api';
// DOM elements
const titleInput = document.getElementById('title');
const kindSelect = document.getElementById('kind');
const tagsInput = document.getElementById('tags');
const includeContent = document.getElementById('includeContent');
const selectionOnly = document.getElementById('selectionOnly');
const saveButton = document.getElementById('save');
const cancelButton = document.getElementById('cancel');
const statusDiv = document.getElementById('status');
let pageData = null;
// Initialize popup
async function init() {
try {
// Get current tab
const [tab] = await chrome.tabs.query({ active: true, currentWindow: true });
// Request content from content script
const response = await chrome.tabs.sendMessage(tab.id, { action: 'extract' });
pageData = response;
// Pre-fill form
titleInput.value = response.title || tab.title || '';
// Suggest tags based on URL
const url = new URL(tab.url);
const suggestedTags = ['web-clip'];
if (url.hostname.includes('github')) suggestedTags.push('github');
if (url.hostname.includes('docs')) suggestedTags.push('documentation');
if (url.hostname.includes('stackoverflow')) suggestedTags.push('stackoverflow');
tagsInput.value = suggestedTags.join(', ');
// Enable selection only if there's selected text
if (response.selection) {
selectionOnly.disabled = false;
} else {
selectionOnly.disabled = true;
}
} catch (err) {
showStatus('Could not extract page content', 'error');
console.error('Init error:', err);
}
}
// Save to Cortex
async function save() {
saveButton.disabled = true;
saveButton.textContent = 'Saving...';
try {
const content = selectionOnly.checked
? pageData?.selection
: (includeContent.checked ? pageData?.content : '');
const tags = tagsInput.value
.split(',')
.map(t => t.trim())
.filter(t => t.length > 0);
const [tab] = await chrome.tabs.query({ active: true, currentWindow: true });
const response = await fetch(`${CORTEX_API}/nodes`, {
method: 'POST',
headers: {
'Content-Type': 'application/json',
},
body: JSON.stringify({
kind: kindSelect.value,
title: titleInput.value,
content: content,
tags: tags,
metadata: {
source: {
type: 'url',
url: tab.url,
savedAt: Date.now(),
},
},
}),
});
if (!response.ok) {
throw new Error(`HTTP ${response.status}`);
}
const node = await response.json();
showStatus(`Saved! ID: ${node.id.slice(0, 8)}`, 'success');
// Close popup after brief delay
setTimeout(() => window.close(), 1500);
} catch (err) {
showStatus(`Failed to save: ${err.message}`, 'error');
console.error('Save error:', err);
// Check if Cortex server is running
if (err.message.includes('fetch')) {
showStatus('Is Cortex server running? (cortex serve)', 'error');
}
} finally {
saveButton.disabled = false;
saveButton.textContent = 'Save';
}
}
// Show status message
function showStatus(message, type) {
statusDiv.textContent = message;
statusDiv.className = `status ${type}`;
}
// Event listeners
saveButton.addEventListener('click', save);
cancelButton.addEventListener('click', () => window.close());
// Toggle selection only checkbox when include content is unchecked
includeContent.addEventListener('change', () => {
if (!includeContent.checked) {
selectionOnly.checked = false;
}
});
selectionOnly.addEventListener('change', () => {
if (selectionOnly.checked) {
includeContent.checked = true;
}
});
// Initialize on load
init();

View File

@@ -0,0 +1,169 @@
import { Command } from 'commander';
import chalk from 'chalk';
import * as fs from 'fs';
import * as path from 'path';
import * as os from 'os';
import { generateCompletions, Shell } from '../completions';
import { listNodes } from '../../core/store';
import { listGraphs } from '../../core/graphs';
import { getDb } from '../../core/db';
export const completionsCommand = new Command('completions')
.description('Generate shell completions')
.argument('[shell]', 'Shell type: bash, zsh, fish, powershell')
.option('--install', 'Install completions to appropriate location')
.action(async (shell: string | undefined, opts) => {
const validShells: Shell[] = ['bash', 'zsh', 'fish', 'powershell'];
if (!shell) {
console.log(chalk.cyan('Available shell completions:\n'));
console.log(' cortex completions bash');
console.log(' cortex completions zsh');
console.log(' cortex completions fish');
console.log(' cortex completions powershell');
console.log(chalk.dim('\nAdd --install to auto-install for your shell'));
return;
}
const shellType = shell.toLowerCase() as Shell;
if (!validShells.includes(shellType)) {
console.error(chalk.red(`Invalid shell: ${shell}`));
console.log(chalk.dim(`Valid shells: ${validShells.join(', ')}`));
process.exit(1);
}
const completionScript = generateCompletions(shellType);
if (opts.install) {
installCompletions(shellType, completionScript);
} else {
console.log(completionScript);
}
});
function installCompletions(shell: Shell, script: string): void {
const home = os.homedir();
switch (shell) {
case 'bash': {
const bashrc = path.join(home, '.bashrc');
const completionMarker = '# Cortex CLI completions';
// Check if already installed
if (fs.existsSync(bashrc)) {
const content = fs.readFileSync(bashrc, 'utf-8');
if (content.includes(completionMarker)) {
console.log(chalk.yellow('Completions already installed in ~/.bashrc'));
return;
}
}
fs.appendFileSync(bashrc, `\n${completionMarker}\n${script}\n`);
console.log(chalk.green('✓ Installed bash completions to ~/.bashrc'));
console.log(chalk.dim(' Run: source ~/.bashrc'));
break;
}
case 'zsh': {
const zshCompletions = path.join(home, '.zsh', 'completions');
const targetFile = path.join(zshCompletions, '_cortex');
fs.mkdirSync(zshCompletions, { recursive: true });
fs.writeFileSync(targetFile, script);
console.log(chalk.green(`✓ Installed zsh completions to ${targetFile}`));
console.log(chalk.dim(' Add to ~/.zshrc: fpath=(~/.zsh/completions $fpath)'));
console.log(chalk.dim(' Then run: autoload -Uz compinit && compinit'));
break;
}
case 'fish': {
const fishCompletions = path.join(home, '.config', 'fish', 'completions');
const targetFile = path.join(fishCompletions, 'cortex.fish');
fs.mkdirSync(fishCompletions, { recursive: true });
fs.writeFileSync(targetFile, script);
console.log(chalk.green(`✓ Installed fish completions to ${targetFile}`));
console.log(chalk.dim(' Completions will be loaded automatically'));
break;
}
case 'powershell': {
// PowerShell profile varies by platform
const profilePaths = [
path.join(home, 'Documents', 'WindowsPowerShell', 'Microsoft.PowerShell_profile.ps1'),
path.join(home, 'Documents', 'PowerShell', 'Microsoft.PowerShell_profile.ps1'),
path.join(home, '.config', 'powershell', 'Microsoft.PowerShell_profile.ps1'),
];
let profilePath = profilePaths.find(p => fs.existsSync(path.dirname(p)));
if (!profilePath) {
profilePath = profilePaths[0];
fs.mkdirSync(path.dirname(profilePath), { recursive: true });
}
const marker = '# Cortex CLI completions';
if (fs.existsSync(profilePath)) {
const content = fs.readFileSync(profilePath, 'utf-8');
if (content.includes(marker)) {
console.log(chalk.yellow(`Completions already installed in ${profilePath}`));
return;
}
}
fs.appendFileSync(profilePath, `\n${marker}\n${script}\n`);
console.log(chalk.green(`✓ Installed PowerShell completions to ${profilePath}`));
console.log(chalk.dim(' Restart PowerShell to load completions'));
break;
}
}
}
// Hidden helper commands for dynamic completions
export const getNodesCommand = new Command('--get-nodes')
.argument('[prefix]', 'Node ID prefix to filter')
.description('Helper for shell completions')
.action(async (prefix?: string) => {
try {
const nodes = listNodes({ limit: 30, includeStale: false });
const filtered = prefix
? nodes.filter(n => n.id.startsWith(prefix) || n.title.toLowerCase().includes(prefix.toLowerCase()))
: nodes;
// Output tab-separated: id<TAB>title for completion with description
for (const node of filtered.slice(0, 20)) {
console.log(`${node.id.slice(0, 8)}\t${node.title.slice(0, 40)}`);
}
} catch {
// Silently fail for completion scripts
}
});
export const getTagsCommand = new Command('--get-tags')
.description('Helper for shell completions')
.action(async () => {
try {
const db = getDb();
const tags = db.prepare('SELECT DISTINCT tag FROM node_tags ORDER BY tag').all() as { tag: string }[];
for (const { tag } of tags.slice(0, 50)) {
console.log(tag);
}
} catch {
// Silently fail
}
});
export const getGraphsCommand = new Command('--get-graphs')
.description('Helper for shell completions')
.action(async () => {
try {
const graphs = listGraphs();
for (const graph of graphs) {
console.log(graph.name);
}
} catch {
// Silently fail
}
});

View File

@@ -0,0 +1,397 @@
/**
* Shell completion generators for Cortex CLI
*/
export type Shell = 'bash' | 'zsh' | 'fish' | 'powershell';
const COMMANDS = [
'add', 'query', 'show', 'list', 'update', 'remove', 'link', 'graph',
'children', 'decay', 'serve', 'history', 'diff', 'restore', 'capture',
'context', 'config', 'index', 'journal', 'ingest', 'clip', 'export',
'viz', 'import', 'backup', 'restore-backup', 'list-backups', 'graphs',
'use', 'init', 'smart-search', 'ss', 'what', 'now', 'tui', 'ui',
];
const KINDS = ['memory', 'component', 'task', 'decision'];
const STATUSES = ['active', 'todo', 'in_progress', 'done', 'deprecated'];
const EDGE_TYPES = ['depends_on', 'contains', 'implements', 'blocked_by', 'subtask_of', 'relates_to', 'supersedes', 'about'];
export function generateCompletions(shell: Shell): string {
switch (shell) {
case 'bash':
return generateBashCompletions();
case 'zsh':
return generateZshCompletions();
case 'fish':
return generateFishCompletions();
case 'powershell':
return generatePowerShellCompletions();
}
}
function generateBashCompletions(): string {
return `# Cortex CLI completions for Bash
# Add to ~/.bashrc or /etc/bash_completion.d/cortex
_cortex_completions() {
local cur prev words cword
_init_completion -n : || return
local commands="${COMMANDS.join(' ')}"
local kinds="${KINDS.join(' ')}"
local statuses="${STATUSES.join(' ')}"
local edge_types="${EDGE_TYPES.join(' ')}"
case "\${prev}" in
cortex)
COMPREPLY=($(compgen -W "\${commands}" -- "\${cur}"))
return 0
;;
add)
COMPREPLY=($(compgen -W "\${kinds}" -- "\${cur}"))
return 0
;;
--kind|-k)
COMPREPLY=($(compgen -W "\${kinds}" -- "\${cur}"))
return 0
;;
--status|-s)
COMPREPLY=($(compgen -W "\${statuses}" -- "\${cur}"))
return 0
;;
--type)
COMPREPLY=($(compgen -W "\${edge_types}" -- "\${cur}"))
return 0
;;
--tags|-t)
local tags=$(cortex --get-tags 2>/dev/null)
COMPREPLY=($(compgen -W "\${tags}" -- "\${cur}"))
return 0
;;
--graph)
local graphs=$(cortex --get-graphs 2>/dev/null)
COMPREPLY=($(compgen -W "\${graphs}" -- "\${cur}"))
return 0
;;
show|update|remove|children|history|diff|restore)
local nodes=$(cortex --get-nodes "\${cur}" 2>/dev/null)
COMPREPLY=($(compgen -W "\${nodes}" -- "\${cur}"))
return 0
;;
link)
local nodes=$(cortex --get-nodes "\${cur}" 2>/dev/null)
COMPREPLY=($(compgen -W "\${nodes}" -- "\${cur}"))
return 0
;;
use)
local graphs=$(cortex --get-graphs 2>/dev/null)
COMPREPLY=($(compgen -W "\${graphs}" -- "\${cur}"))
return 0
;;
import)
COMPREPLY=($(compgen -W "obsidian markdown" -- "\${cur}"))
return 0
;;
export|--format|-f)
COMPREPLY=($(compgen -W "html svg mermaid markdown jsonld" -- "\${cur}"))
return 0
;;
esac
if [[ "\${cur}" == -* ]]; then
local opts="--help --version --kind --status --tags --limit --format --output --graph --all-graphs"
COMPREPLY=($(compgen -W "\${opts}" -- "\${cur}"))
return 0
fi
}
complete -F _cortex_completions cortex
`;
}
function generateZshCompletions(): string {
return `#compdef cortex
# Cortex CLI completions for Zsh
# Add to ~/.zsh/completions/_cortex
_cortex() {
local -a commands
commands=(
'add:Add a node to the knowledge graph'
'query:Search the knowledge graph'
'show:Show a node and its connections'
'list:List nodes'
'update:Update a node'
'remove:Remove a node'
'link:Create a link between nodes'
'graph:Display graph structure'
'children:List child nodes'
'decay:Mark old nodes as stale'
'serve:Start web server'
'history:View node version history'
'diff:Compare node versions'
'restore:Restore node to previous version'
'capture:Capture conversation as memory'
'context:Show context for session'
'config:Configure capture settings'
'index:Index project codebase'
'journal:Daily journal'
'ingest:Ingest content from URL'
'clip:Quick clip a URL'
'export:Export graph as HTML/SVG/Mermaid'
'viz:Export interactive visualization'
'import:Import from Obsidian/Markdown'
'backup:Create database backup'
'restore-backup:Restore from backup'
'list-backups:List backup files'
'graphs:Manage knowledge graphs'
'use:Switch active graph'
'init:Initialize project with graph'
'smart-search:Context-aware search'
'ss:Alias for smart-search'
'what:What should I know right now?'
'now:Show current context'
'tui:Launch interactive TUI'
'ui:Alias for tui'
)
local -a kinds=(memory component task decision)
local -a statuses=(active todo in_progress done deprecated)
local -a edge_types=(depends_on contains implements blocked_by subtask_of relates_to supersedes about)
_arguments -C \\
'1:command:->command' \\
'*::arg:->args'
case "$state" in
command)
_describe -t commands 'cortex commands' commands
;;
args)
case "$words[1]" in
add)
_arguments \\
'1:kind:(memory component task decision)' \\
'--title[Node title]:title:' \\
'--content[Node content]:content:' \\
'--tags[Tags]:tags:->tags' \\
'--status[Status]:status:(${STATUSES.join(' ')})'
;;
show|update|remove|children|history)
_arguments '1:node:->nodes'
;;
link)
_arguments \\
'1:from:->nodes' \\
'2:to:->nodes' \\
'--type[Edge type]:type:(${EDGE_TYPES.join(' ')})'
;;
list|query)
_arguments \\
'--kind[Filter by kind]:kind:(${KINDS.join(' ')})' \\
'--status[Filter by status]:status:(${STATUSES.join(' ')})' \\
'--tags[Filter by tags]:tags:->tags' \\
'--limit[Max results]:limit:' \\
'--graph[Query specific graph]:graph:->graphs' \\
'--all-graphs[Search all graphs]'
;;
use)
_arguments '1:graph:->graphs'
;;
import)
_arguments \\
'1:source:(obsidian markdown)' \\
'2:path:_files -/'
;;
export)
_arguments \\
'--format[Export format]:format:(html svg mermaid markdown jsonld)' \\
'--output[Output file]:file:_files' \\
'--kind[Filter by kind]:kind:(${KINDS.join(' ')})'
;;
esac
;;
esac
case "$state" in
nodes)
local -a nodes
nodes=(\${(f)"$(cortex --get-nodes 2>/dev/null)"})
_describe -t nodes 'nodes' nodes
;;
tags)
local -a tags
tags=(\${(f)"$(cortex --get-tags 2>/dev/null)"})
_describe -t tags 'tags' tags
;;
graphs)
local -a graphs
graphs=(\${(f)"$(cortex --get-graphs 2>/dev/null)"})
_describe -t graphs 'graphs' graphs
;;
esac
}
_cortex
`;
}
function generateFishCompletions(): string {
return `# Cortex CLI completions for Fish
# Add to ~/.config/fish/completions/cortex.fish
# Disable file completions for cortex
complete -c cortex -f
# Main commands
complete -c cortex -n __fish_use_subcommand -a add -d 'Add a node'
complete -c cortex -n __fish_use_subcommand -a query -d 'Search the graph'
complete -c cortex -n __fish_use_subcommand -a show -d 'Show a node'
complete -c cortex -n __fish_use_subcommand -a list -d 'List nodes'
complete -c cortex -n __fish_use_subcommand -a update -d 'Update a node'
complete -c cortex -n __fish_use_subcommand -a remove -d 'Remove a node'
complete -c cortex -n __fish_use_subcommand -a link -d 'Link nodes'
complete -c cortex -n __fish_use_subcommand -a graph -d 'Display graph'
complete -c cortex -n __fish_use_subcommand -a children -d 'List children'
complete -c cortex -n __fish_use_subcommand -a decay -d 'Mark stale nodes'
complete -c cortex -n __fish_use_subcommand -a serve -d 'Start server'
complete -c cortex -n __fish_use_subcommand -a history -d 'View history'
complete -c cortex -n __fish_use_subcommand -a diff -d 'Compare versions'
complete -c cortex -n __fish_use_subcommand -a restore -d 'Restore version'
complete -c cortex -n __fish_use_subcommand -a capture -d 'Capture memory'
complete -c cortex -n __fish_use_subcommand -a context -d 'Show context'
complete -c cortex -n __fish_use_subcommand -a config -d 'Configure'
complete -c cortex -n __fish_use_subcommand -a index -d 'Index project'
complete -c cortex -n __fish_use_subcommand -a journal -d 'Daily journal'
complete -c cortex -n __fish_use_subcommand -a ingest -d 'Ingest URL'
complete -c cortex -n __fish_use_subcommand -a clip -d 'Clip URL'
complete -c cortex -n __fish_use_subcommand -a export -d 'Export graph'
complete -c cortex -n __fish_use_subcommand -a viz -d 'Visualize'
complete -c cortex -n __fish_use_subcommand -a import -d 'Import data'
complete -c cortex -n __fish_use_subcommand -a backup -d 'Backup database'
complete -c cortex -n __fish_use_subcommand -a restore-backup -d 'Restore backup'
complete -c cortex -n __fish_use_subcommand -a list-backups -d 'List backups'
complete -c cortex -n __fish_use_subcommand -a graphs -d 'Manage graphs'
complete -c cortex -n __fish_use_subcommand -a use -d 'Switch graph'
complete -c cortex -n __fish_use_subcommand -a init -d 'Init project'
complete -c cortex -n __fish_use_subcommand -a smart-search -d 'Smart search'
complete -c cortex -n __fish_use_subcommand -a ss -d 'Smart search alias'
complete -c cortex -n __fish_use_subcommand -a what -d 'What should I know?'
complete -c cortex -n __fish_use_subcommand -a now -d 'Current context'
complete -c cortex -n __fish_use_subcommand -a tui -d 'Interactive TUI'
complete -c cortex -n __fish_use_subcommand -a ui -d 'TUI alias'
# Kind completions
complete -c cortex -n '__fish_seen_subcommand_from add' -a 'memory component task decision'
# Node ID completions
complete -c cortex -n '__fish_seen_subcommand_from show update remove children history' -a '(cortex --get-nodes 2>/dev/null)'
# Graph completions
complete -c cortex -n '__fish_seen_subcommand_from use' -a '(cortex --get-graphs 2>/dev/null)'
# Import source completions
complete -c cortex -n '__fish_seen_subcommand_from import' -a 'obsidian markdown'
# Export format completions
complete -c cortex -n '__fish_seen_subcommand_from export' -l format -a 'html svg mermaid markdown jsonld'
# Common options
complete -c cortex -l kind -s k -d 'Filter by kind' -a 'memory component task decision'
complete -c cortex -l status -s s -d 'Filter by status' -a 'active todo in_progress done deprecated'
complete -c cortex -l tags -s t -d 'Filter by tags' -a '(cortex --get-tags 2>/dev/null)'
complete -c cortex -l limit -s l -d 'Max results'
complete -c cortex -l graph -d 'Specific graph' -a '(cortex --get-graphs 2>/dev/null)'
complete -c cortex -l all-graphs -d 'Search all graphs'
complete -c cortex -l help -s h -d 'Show help'
`;
}
function generatePowerShellCompletions(): string {
return `# Cortex CLI completions for PowerShell
# Add to your $PROFILE
Register-ArgumentCompleter -Native -CommandName cortex -ScriptBlock {
param($wordToComplete, $commandAst, $cursorPosition)
$commands = @('add', 'query', 'show', 'list', 'update', 'remove', 'link', 'graph',
'children', 'decay', 'serve', 'history', 'diff', 'restore', 'capture',
'context', 'config', 'index', 'journal', 'ingest', 'clip', 'export',
'viz', 'import', 'backup', 'restore-backup', 'list-backups', 'graphs',
'use', 'init', 'smart-search', 'ss', 'what', 'now', 'tui', 'ui')
$kinds = @('memory', 'component', 'task', 'decision')
$statuses = @('active', 'todo', 'in_progress', 'done', 'deprecated')
$edgeTypes = @('depends_on', 'contains', 'implements', 'blocked_by', 'subtask_of', 'relates_to', 'supersedes', 'about')
$elements = $commandAst.CommandElements
$command = if ($elements.Count -gt 1) { $elements[1].Value } else { $null }
# Command completion
if ($elements.Count -le 2) {
$commands | Where-Object { $_ -like "$wordToComplete*" } | ForEach-Object {
[System.Management.Automation.CompletionResult]::new($_, $_, 'ParameterValue', $_)
}
return
}
# Sub-command argument completion
switch ($command) {
'add' {
if ($elements.Count -eq 3) {
$kinds | Where-Object { $_ -like "$wordToComplete*" } | ForEach-Object {
[System.Management.Automation.CompletionResult]::new($_, $_, 'ParameterValue', $_)
}
}
}
{ $_ -in @('show', 'update', 'remove', 'children', 'history') } {
$nodes = cortex --get-nodes $wordToComplete 2>$null
if ($nodes) {
$nodes -split "\\n" | Where-Object { $_ -like "$wordToComplete*" } | ForEach-Object {
[System.Management.Automation.CompletionResult]::new($_, $_, 'ParameterValue', $_)
}
}
}
'use' {
$graphs = cortex --get-graphs 2>$null
if ($graphs) {
$graphs -split "\\n" | Where-Object { $_ -like "$wordToComplete*" } | ForEach-Object {
[System.Management.Automation.CompletionResult]::new($_, $_, 'ParameterValue', $_)
}
}
}
'import' {
if ($elements.Count -eq 3) {
@('obsidian', 'markdown') | Where-Object { $_ -like "$wordToComplete*" } | ForEach-Object {
[System.Management.Automation.CompletionResult]::new($_, $_, 'ParameterValue', $_)
}
}
}
}
# Option completion
$prevElement = if ($elements.Count -gt 2) { $elements[-2].Value } else { $null }
switch ($prevElement) {
{ $_ -in @('--kind', '-k') } {
$kinds | Where-Object { $_ -like "$wordToComplete*" } | ForEach-Object {
[System.Management.Automation.CompletionResult]::new($_, $_, 'ParameterValue', $_)
}
}
{ $_ -in @('--status', '-s') } {
$statuses | Where-Object { $_ -like "$wordToComplete*" } | ForEach-Object {
[System.Management.Automation.CompletionResult]::new($_, $_, 'ParameterValue', $_)
}
}
'--type' {
$edgeTypes | Where-Object { $_ -like "$wordToComplete*" } | ForEach-Object {
[System.Management.Automation.CompletionResult]::new($_, $_, 'ParameterValue', $_)
}
}
{ $_ -in @('--format', '-f') } {
@('html', 'svg', 'mermaid', 'markdown', 'jsonld') | Where-Object { $_ -like "$wordToComplete*" } | ForEach-Object {
[System.Management.Automation.CompletionResult]::new($_, $_, 'ParameterValue', $_)
}
}
}
}
`;
}

View File

@@ -25,6 +25,7 @@ import { backupCommand, restoreDbCommand, listBackupsCommand } from './commands/
import { graphsCommand, useCommand, initCommand } from './commands/graphs';
import { smartSearchCommand, ssCommand, whatCommand, contextAwareCommand } from './commands/smart';
import { tuiCommand, uiCommand } from './commands/tui';
import { completionsCommand, getNodesCommand, getTagsCommand, getGraphsCommand } from './commands/completions';
import { closeDb } from '../core/db';
import { migrateOldDatabase } from '../core/db';
@@ -75,6 +76,10 @@ program.addCommand(whatCommand);
program.addCommand(contextAwareCommand);
program.addCommand(tuiCommand);
program.addCommand(uiCommand);
program.addCommand(completionsCommand);
program.addCommand(getNodesCommand);
program.addCommand(getTagsCommand);
program.addCommand(getGraphsCommand);
// Check for old database migration
migrateOldDatabase();