+366 -1

backend/electron/datastore.ts

reviewed

··· 6 6 */ 7 7 8 8 import Database from 'better-sqlite3'; 9 9 - import type { TableName } from '../types/index.js'; 9 9 + import type { 10 10 + TableName, 11 11 + Address, 12 12 + Visit, 13 13 + Content, 14 14 + Tag, 15 15 + AddressTag, 16 16 + DatastoreStats, 17 17 + AddressFilter, 18 18 + VisitFilter, 19 19 + ContentFilter, 20 20 + AddressOptions, 21 21 + VisitOptions, 22 22 + ContentOptions, 23 23 + } from '../types/index.js'; 10 24 import { tableNames } from '../types/index.js'; 11 25 12 26 // SQL Schema ··· 376 390 console.error('main', 'TinyBase migration failed:', (error as Error).message); 377 391 } 378 392 } 393 393 + 394 394 + // ==================== Address Operations ==================== 395 395 + 396 396 + export function addAddress(uri: string, options: AddressOptions = {}): { id: string } { 397 397 + const normalizedUri = normalizeUrl(uri); 398 398 + const parsed = parseUrl(normalizedUri); 399 399 + const addressId = generateId('addr'); 400 400 + const timestamp = now(); 401 401 + 402 402 + getDb().prepare(` 403 403 + INSERT INTO addresses (id, uri, protocol, domain, path, title, mimeType, favicon, description, tags, metadata, createdAt, updatedAt, lastVisitAt, visitCount, starred, archived) 404 404 + VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?) 405 405 + `).run( 406 406 + addressId, 407 407 + normalizedUri, 408 408 + options.protocol || parsed.protocol, 409 409 + options.domain || parsed.domain, 410 410 + options.path || parsed.path, 411 411 + options.title || '', 412 412 + options.mimeType || 'text/html', 413 413 + options.favicon || '', 414 414 + options.description || '', 415 415 + options.tags || '', 416 416 + options.metadata || '{}', 417 417 + timestamp, 418 418 + timestamp, 419 419 + options.lastVisitAt || 0, 420 420 + options.visitCount || 0, 421 421 + options.starred || 0, 422 422 + options.archived || 0 423 423 + ); 424 424 + 425 425 + return { id: addressId }; 426 426 + } 427 427 + 428 428 + export function getAddress(id: string): Address | undefined { 429 429 + return getDb().prepare('SELECT * FROM addresses WHERE id = ?').get(id) as Address | undefined; 430 430 + } 431 431 + 432 432 + export function updateAddress(id: string, updates: Partial<Address>): Address | undefined { 433 433 + const existing = getAddress(id); 434 434 + if (!existing) return undefined; 435 435 + 436 436 + const updated = { ...existing, ...updates, updatedAt: now() }; 437 437 + const columns = Object.keys(updated).filter(k => k !== 'id'); 438 438 + const setClause = columns.map(col => `${col} = ?`).join(', '); 439 439 + const values = columns.map(col => updated[col as keyof Address]); 440 440 + 441 441 + getDb().prepare(`UPDATE addresses SET ${setClause} WHERE id = ?`).run(...values, id); 442 442 + return updated as Address; 443 443 + } 444 444 + 445 445 + export function queryAddresses(filter: AddressFilter = {}): Address[] { 446 446 + let sql = 'SELECT * FROM addresses WHERE 1=1'; 447 447 + const params: (string | number)[] = []; 448 448 + 449 449 + if (filter.domain) { 450 450 + sql += ' AND domain = ?'; 451 451 + params.push(filter.domain); 452 452 + } 453 453 + if (filter.protocol) { 454 454 + sql += ' AND protocol = ?'; 455 455 + params.push(filter.protocol); 456 456 + } 457 457 + if (filter.starred !== undefined) { 458 458 + sql += ' AND starred = ?'; 459 459 + params.push(filter.starred); 460 460 + } 461 461 + if (filter.tag) { 462 462 + sql += ' AND tags LIKE ?'; 463 463 + params.push(`%${filter.tag}%`); 464 464 + } 465 465 + 466 466 + const sortMap: Record<string, string> = { 467 467 + lastVisit: 'lastVisitAt DESC', 468 468 + visitCount: 'visitCount DESC', 469 469 + created: 'createdAt DESC' 470 470 + }; 471 471 + sql += ` ORDER BY ${sortMap[filter.sortBy || ''] || 'updatedAt DESC'}`; 472 472 + 473 473 + if (filter.limit) { 474 474 + sql += ' LIMIT ?'; 475 475 + params.push(filter.limit); 476 476 + } 477 477 + 478 478 + return getDb().prepare(sql).all(...params) as Address[]; 479 479 + } 480 480 + 481 481 + // ==================== Visit Operations ==================== 482 482 + 483 483 + export function addVisit(addressId: string, options: VisitOptions = {}): { id: string } { 484 484 + const visitId = generateId('visit'); 485 485 + const timestamp = now(); 486 486 + 487 487 + getDb().prepare(` 488 488 + INSERT INTO visits (id, addressId, timestamp, duration, source, sourceId, windowType, metadata, scrollDepth, interacted) 489 489 + VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?) 490 490 + `).run( 491 491 + visitId, 492 492 + addressId, 493 493 + options.timestamp || timestamp, 494 494 + options.duration || 0, 495 495 + options.source || 'direct', 496 496 + options.sourceId || '', 497 497 + options.windowType || 'main', 498 498 + options.metadata || '{}', 499 499 + options.scrollDepth || 0, 500 500 + options.interacted || 0 501 501 + ); 502 502 + 503 503 + // Update address visit stats 504 504 + getDb().prepare(` 505 505 + UPDATE addresses SET lastVisitAt = ?, visitCount = visitCount + 1, updatedAt = ? 506 506 + WHERE id = ? 507 507 + `).run(timestamp, timestamp, addressId); 508 508 + 509 509 + return { id: visitId }; 510 510 + } 511 511 + 512 512 + export function queryVisits(filter: VisitFilter = {}): Visit[] { 513 513 + let sql = 'SELECT * FROM visits WHERE 1=1'; 514 514 + const params: (string | number)[] = []; 515 515 + 516 516 + if (filter.addressId) { 517 517 + sql += ' AND addressId = ?'; 518 518 + params.push(filter.addressId); 519 519 + } 520 520 + if (filter.source) { 521 521 + sql += ' AND source = ?'; 522 522 + params.push(filter.source); 523 523 + } 524 524 + if (filter.since) { 525 525 + sql += ' AND timestamp >= ?'; 526 526 + params.push(filter.since); 527 527 + } 528 528 + 529 529 + sql += ' ORDER BY timestamp DESC'; 530 530 + 531 531 + if (filter.limit) { 532 532 + sql += ' LIMIT ?'; 533 533 + params.push(filter.limit); 534 534 + } 535 535 + 536 536 + return getDb().prepare(sql).all(...params) as Visit[]; 537 537 + } 538 538 + 539 539 + // ==================== Content Operations ==================== 540 540 + 541 541 + export function addContent(options: ContentOptions = {}): { id: string } { 542 542 + const contentId = generateId('content'); 543 543 + const timestamp = now(); 544 544 + 545 545 + getDb().prepare(` 546 546 + INSERT INTO content (id, title, content, mimeType, contentType, language, encoding, tags, addressRefs, parentId, metadata, createdAt, updatedAt, syncPath, synced, starred, archived) 547 547 + VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?) 548 548 + `).run( 549 549 + contentId, 550 550 + options.title || 'Untitled', 551 551 + options.content || '', 552 552 + options.mimeType || 'text/plain', 553 553 + options.contentType || 'plain', 554 554 + options.language || '', 555 555 + options.encoding || 'utf-8', 556 556 + options.tags || '', 557 557 + options.addressRefs || '', 558 558 + options.parentId || '', 559 559 + options.metadata || '{}', 560 560 + timestamp, 561 561 + timestamp, 562 562 + options.syncPath || '', 563 563 + options.synced || 0, 564 564 + options.starred || 0, 565 565 + options.archived || 0 566 566 + ); 567 567 + 568 568 + return { id: contentId }; 569 569 + } 570 570 + 571 571 + export function queryContent(filter: ContentFilter = {}): Content[] { 572 572 + let sql = 'SELECT * FROM content WHERE 1=1'; 573 573 + const params: (string | number)[] = []; 574 574 + 575 575 + if (filter.contentType) { 576 576 + sql += ' AND contentType = ?'; 577 577 + params.push(filter.contentType); 578 578 + } 579 579 + if (filter.mimeType) { 580 580 + sql += ' AND mimeType = ?'; 581 581 + params.push(filter.mimeType); 582 582 + } 583 583 + if (filter.synced !== undefined) { 584 584 + sql += ' AND synced = ?'; 585 585 + params.push(filter.synced); 586 586 + } 587 587 + if (filter.starred !== undefined) { 588 588 + sql += ' AND starred = ?'; 589 589 + params.push(filter.starred); 590 590 + } 591 591 + if (filter.tag) { 592 592 + sql += ' AND tags LIKE ?'; 593 593 + params.push(`%${filter.tag}%`); 594 594 + } 595 595 + 596 596 + const sortMap: Record<string, string> = { 597 597 + updated: 'updatedAt DESC', 598 598 + created: 'createdAt DESC' 599 599 + }; 600 600 + sql += ` ORDER BY ${sortMap[filter.sortBy || ''] || 'updatedAt DESC'}`; 601 601 + 602 602 + if (filter.limit) { 603 603 + sql += ' LIMIT ?'; 604 604 + params.push(filter.limit); 605 605 + } 606 606 + 607 607 + return getDb().prepare(sql).all(...params) as Content[]; 608 608 + } 609 609 + 610 610 + // ==================== Tag Operations ==================== 611 611 + 612 612 + export function getOrCreateTag(name: string): { tag: Tag; created: boolean } { 613 613 + const slug = name.toLowerCase().trim().replace(/\s+/g, '-'); 614 614 + const timestamp = now(); 615 615 + 616 616 + const existingTag = getDb().prepare('SELECT * FROM tags WHERE LOWER(name) = LOWER(?)').get(name) as Tag | undefined; 617 617 + if (existingTag) { 618 618 + return { tag: existingTag, created: false }; 619 619 + } 620 620 + 621 621 + const tagId = generateId('tag'); 622 622 + getDb().prepare(` 623 623 + INSERT INTO tags (id, name, slug, color, parentId, description, metadata, createdAt, updatedAt, frequency, lastUsedAt, frecencyScore) 624 624 + VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?) 625 625 + `).run(tagId, name.trim(), slug, '#999999', '', '', '{}', timestamp, timestamp, 0, 0, 0); 626 626 + 627 627 + const newTag = getDb().prepare('SELECT * FROM tags WHERE id = ?').get(tagId) as Tag; 628 628 + return { tag: newTag, created: true }; 629 629 + } 630 630 + 631 631 + export function tagAddress(addressId: string, tagId: string): { link: AddressTag; alreadyExists: boolean } { 632 632 + const timestamp = now(); 633 633 + 634 634 + const existingLink = getDb().prepare('SELECT * FROM address_tags WHERE addressId = ? AND tagId = ?').get(addressId, tagId) as AddressTag | undefined; 635 635 + if (existingLink) { 636 636 + return { link: existingLink, alreadyExists: true }; 637 637 + } 638 638 + 639 639 + const linkId = generateId('address_tag'); 640 640 + getDb().prepare('INSERT INTO address_tags (id, addressId, tagId, createdAt) VALUES (?, ?, ?, ?)').run(linkId, addressId, tagId, timestamp); 641 641 + 642 642 + // Update tag frequency and frecency 643 643 + const tag = getDb().prepare('SELECT * FROM tags WHERE id = ?').get(tagId) as Tag | undefined; 644 644 + if (tag) { 645 645 + const newFrequency = (tag.frequency || 0) + 1; 646 646 + const frecencyScore = calculateFrecency(newFrequency, timestamp); 647 647 + getDb().prepare('UPDATE tags SET frequency = ?, lastUsedAt = ?, frecencyScore = ?, updatedAt = ? WHERE id = ?') 648 648 + .run(newFrequency, timestamp, frecencyScore, timestamp, tagId); 649 649 + } 650 650 + 651 651 + const newLink = getDb().prepare('SELECT * FROM address_tags WHERE id = ?').get(linkId) as AddressTag; 652 652 + return { link: newLink, alreadyExists: false }; 653 653 + } 654 654 + 655 655 + export function untagAddress(addressId: string, tagId: string): boolean { 656 656 + const result = getDb().prepare('DELETE FROM address_tags WHERE addressId = ? AND tagId = ?').run(addressId, tagId); 657 657 + return result.changes > 0; 658 658 + } 659 659 + 660 660 + export function getTagsByFrecency(domain?: string): Tag[] { 661 661 + let tags = getDb().prepare('SELECT * FROM tags').all() as Tag[]; 662 662 + 663 663 + // Recalculate frecency scores 664 664 + tags = tags.map(tag => ({ 665 665 + ...tag, 666 666 + frecencyScore: calculateFrecency(tag.frequency || 0, tag.lastUsedAt || 0) 667 667 + })); 668 668 + 669 669 + // If domain provided, boost tags used on same-domain addresses 670 670 + if (domain) { 671 671 + const domainTagIds = new Set( 672 672 + (getDb().prepare(` 673 673 + SELECT DISTINCT at.tagId FROM address_tags at 674 674 + JOIN addresses a ON at.addressId = a.id 675 675 + WHERE a.domain = ? 676 676 + `).all(domain) as { tagId: string }[]).map(row => row.tagId) 677 677 + ); 678 678 + 679 679 + tags = tags.map(tag => ({ 680 680 + ...tag, 681 681 + frecencyScore: domainTagIds.has(tag.id) ? tag.frecencyScore * 2 : tag.frecencyScore 682 682 + })); 683 683 + } 684 684 + 685 685 + tags.sort((a, b) => b.frecencyScore - a.frecencyScore); 686 686 + return tags; 687 687 + } 688 688 + 689 689 + export function getAddressTags(addressId: string): Tag[] { 690 690 + return getDb().prepare(` 691 691 + SELECT t.* FROM tags t 692 692 + JOIN address_tags at ON t.id = at.tagId 693 693 + WHERE at.addressId = ? 694 694 + `).all(addressId) as Tag[]; 695 695 + } 696 696 + 697 697 + export function getAddressesByTag(tagId: string): Address[] { 698 698 + return getDb().prepare(` 699 699 + SELECT a.* FROM addresses a 700 700 + JOIN address_tags at ON a.id = at.addressId 701 701 + WHERE at.tagId = ? 702 702 + `).all(tagId) as Address[]; 703 703 + } 704 704 + 705 705 + export function getUntaggedAddresses(): Address[] { 706 706 + return getDb().prepare(` 707 707 + SELECT a.* FROM addresses a 708 708 + LEFT JOIN address_tags at ON a.id = at.addressId 709 709 + WHERE at.id IS NULL 710 710 + ORDER BY a.visitCount DESC 711 711 + `).all() as Address[]; 712 712 + } 713 713 + 714 714 + // ==================== Generic Table Operations ==================== 715 715 + 716 716 + export function getTable(tableName: TableName): Record<string, Record<string, unknown>> { 717 717 + const rows = getDb().prepare(`SELECT * FROM ${tableName}`).all() as Array<{ id: string } & Record<string, unknown>>; 718 718 + const table: Record<string, Record<string, unknown>> = {}; 719 719 + for (const row of rows) { 720 720 + table[row.id] = row; 721 721 + } 722 722 + return table; 723 723 + } 724 724 + 725 725 + export function setRow(tableName: TableName, rowId: string, rowData: Record<string, unknown>): void { 726 726 + const row: Record<string, unknown> = { id: rowId, ...rowData }; 727 727 + const columns = Object.keys(row); 728 728 + const placeholders = columns.map(() => '?').join(', '); 729 729 + const values = columns.map(col => row[col]); 730 730 + 731 731 + getDb().prepare(`INSERT OR REPLACE INTO ${tableName} (${columns.join(', ')}) VALUES (${placeholders})`).run(...values); 732 732 + } 733 733 + 734 734 + export function getStats(): DatastoreStats { 735 735 + const d = getDb(); 736 736 + return { 737 737 + totalAddresses: (d.prepare('SELECT COUNT(*) as count FROM addresses').get() as { count: number }).count, 738 738 + totalVisits: (d.prepare('SELECT COUNT(*) as count FROM visits').get() as { count: number }).count, 739 739 + avgVisitDuration: (d.prepare('SELECT AVG(duration) as avg FROM visits').get() as { avg: number | null }).avg || 0, 740 740 + totalContent: (d.prepare('SELECT COUNT(*) as count FROM content').get() as { count: number }).count, 741 741 + syncedContent: (d.prepare('SELECT COUNT(*) as count FROM content WHERE synced = 1').get() as { count: number }).count 742 742 + }; 743 743 + }

+34 -1

backend/electron/index.ts

reviewed

··· 4 4 * Exports database functions and types for the Electron main process. 5 5 */ 6 6 7 7 - // Database functions 7 7 + // Database lifecycle and helpers 8 8 export { 9 9 initDatabase, 10 10 closeDatabase, ··· 17 17 calculateFrecency, 18 18 } from './datastore.js'; 19 19 20 20 + // Datastore operations 21 21 + export { 22 22 + // Addresses 23 23 + addAddress, 24 24 + getAddress, 25 25 + updateAddress, 26 26 + queryAddresses, 27 27 + // Visits 28 28 + addVisit, 29 29 + queryVisits, 30 30 + // Content 31 31 + addContent, 32 32 + queryContent, 33 33 + // Tags 34 34 + getOrCreateTag, 35 35 + tagAddress, 36 36 + untagAddress, 37 37 + getTagsByFrecency, 38 38 + getAddressTags, 39 39 + getAddressesByTag, 40 40 + getUntaggedAddresses, 41 41 + // Generic 42 42 + getTable, 43 43 + setRow, 44 44 + getStats, 45 45 + } from './datastore.js'; 46 46 + 20 47 // Re-export shared data types 21 48 export type { 22 49 Address, ··· 28 55 ExtensionSetting, 29 56 DatastoreStats, 30 57 TableName, 58 58 + AddressFilter, 59 59 + VisitFilter, 60 60 + ContentFilter, 61 61 + AddressOptions, 62 62 + VisitOptions, 63 63 + ContentOptions, 31 64 } from '../types/index.js'; 32 65 33 66 export { tableNames } from '../types/index.js';

+72

backend/types/index.ts

reviewed

··· 114 114 syncedContent: number; 115 115 } 116 116 117 117 + // ==================== Filter Types ==================== 118 118 + 119 119 + export interface AddressFilter { 120 120 + domain?: string; 121 121 + protocol?: string; 122 122 + starred?: number; 123 123 + tag?: string; 124 124 + sortBy?: 'lastVisit' | 'visitCount' | 'created'; 125 125 + limit?: number; 126 126 + } 127 127 + 128 128 + export interface VisitFilter { 129 129 + addressId?: string; 130 130 + source?: string; 131 131 + since?: number; 132 132 + limit?: number; 133 133 + } 134 134 + 135 135 + export interface ContentFilter { 136 136 + contentType?: string; 137 137 + mimeType?: string; 138 138 + synced?: number; 139 139 + starred?: number; 140 140 + tag?: string; 141 141 + sortBy?: 'updated' | 'created'; 142 142 + limit?: number; 143 143 + } 144 144 + 145 145 + export interface AddressOptions { 146 146 + protocol?: string; 147 147 + domain?: string; 148 148 + path?: string; 149 149 + title?: string; 150 150 + mimeType?: string; 151 151 + favicon?: string; 152 152 + description?: string; 153 153 + tags?: string; 154 154 + metadata?: string; 155 155 + lastVisitAt?: number; 156 156 + visitCount?: number; 157 157 + starred?: number; 158 158 + archived?: number; 159 159 + } 160 160 + 161 161 + export interface VisitOptions { 162 162 + timestamp?: number; 163 163 + duration?: number; 164 164 + source?: string; 165 165 + sourceId?: string; 166 166 + windowType?: string; 167 167 + metadata?: string; 168 168 + scrollDepth?: number; 169 169 + interacted?: number; 170 170 + } 171 171 + 172 172 + export interface ContentOptions { 173 173 + title?: string; 174 174 + content?: string; 175 175 + mimeType?: string; 176 176 + contentType?: string; 177 177 + language?: string; 178 178 + encoding?: string; 179 179 + tags?: string; 180 180 + addressRefs?: string; 181 181 + parentId?: string; 182 182 + metadata?: string; 183 183 + syncPath?: string; 184 184 + synced?: number; 185 185 + starred?: number; 186 186 + archived?: number; 187 187 + } 188 188 + 117 189 // ==================== Table Names ==================== 118 190 119 191 export type TableName =

+57 -356

index.js

reviewed

··· 22 22 initDatabase, 23 23 closeDatabase, 24 24 getDb, 25 25 - generateId, 26 26 - now, 27 27 - parseUrl, 28 28 - normalizeUrl, 29 25 isValidTable, 30 30 - calculateFrecency, 26 26 + // Datastore operations 27 27 + addAddress, 28 28 + getAddress, 29 29 + updateAddress, 30 30 + queryAddresses, 31 31 + addVisit, 32 32 + queryVisits, 33 33 + addContent, 34 34 + queryContent, 35 35 + getOrCreateTag, 36 36 + tagAddress, 37 37 + untagAddress, 38 38 + getTagsByFrecency, 39 39 + getAddressTags, 40 40 + getAddressesByTag, 41 41 + getUntaggedAddresses, 42 42 + getTable, 43 43 + setRow, 44 44 + getStats, 31 45 } from './dist/backend/electron/index.js'; 32 46 import unhandled from 'electron-unhandled'; 33 47 ··· 102 116 103 117 console.log('PROFILE', PROFILE, app.isPackaged ? '(packaged)' : '(source)'); 104 118 119 119 + // Test profiles skip certain behaviors (devtools, dialogs, etc.) 120 120 + const isTestProfile = PROFILE.startsWith('test'); 121 121 + 105 122 // Profile dirs are subdir of userData dir 106 123 // ..................................... ↓ we set this per profile 107 124 // ··· 155 172 }; 156 173 157 174 // ***** System / OS / Theme ***** 175 175 + 176 176 + // Use system theme by default 177 177 + nativeTheme.themeSource = 'system'; 158 178 159 179 // system dark mode handling 160 180 ipcMain.handle('dark-mode:toggle', () => { ··· 897 917 const win = new BrowserWindow(winPrefs); 898 918 win.loadURL(webCoreAddress); 899 919 900 900 - // Setup devtools for the background window (always open in debug mode) 901 901 - if (DEBUG) { 920 920 + // Setup devtools for the background window (debug mode, but not in tests) 921 921 + if (DEBUG && !isTestProfile) { 902 922 win.webContents.openDevTools({ mode: 'detach', activate: false }); 903 923 } 904 924 ··· 1535 1555 ipcMain.handle('datastore-add-address', async (ev, data) => { 1536 1556 try { 1537 1557 const { uri, options = {} } = data; 1538 1538 - const normalizedUri = normalizeUrl(uri); 1539 1539 - const parsed = parseUrl(normalizedUri); 1540 1540 - const addressId = generateId('addr'); 1541 1541 - const timestamp = now(); 1542 1542 - const db = getDb(); 1543 1543 - 1544 1544 - const stmt = db.prepare(` 1545 1545 - INSERT INTO addresses (id, uri, protocol, domain, path, title, mimeType, favicon, description, tags, metadata, createdAt, updatedAt, lastVisitAt, visitCount, starred, archived) 1546 1546 - VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?) 1547 1547 - `); 1548 1548 - 1549 1549 - stmt.run( 1550 1550 - addressId, 1551 1551 - normalizedUri, 1552 1552 - options.protocol || parsed.protocol, 1553 1553 - options.domain || parsed.domain, 1554 1554 - options.path || parsed.path, 1555 1555 - options.title || '', 1556 1556 - options.mimeType || 'text/html', 1557 1557 - options.favicon || '', 1558 1558 - options.description || '', 1559 1559 - options.tags || '', 1560 1560 - options.metadata || '{}', 1561 1561 - timestamp, 1562 1562 - timestamp, 1563 1563 - options.lastVisitAt || 0, 1564 1564 - options.visitCount || 0, 1565 1565 - options.starred || 0, 1566 1566 - options.archived || 0 1567 1567 - ); 1568 1568 - 1569 1569 - return { success: true, id: addressId }; 1558 1558 + const result = addAddress(uri, options); 1559 1559 + return { success: true, id: result.id }; 1570 1560 } catch (error) { 1571 1561 console.error('datastore-add-address error:', error); 1572 1562 return { success: false, error: error.message }; ··· 1576 1566 ipcMain.handle('datastore-get-address', async (ev, data) => { 1577 1567 try { 1578 1568 const { id } = data; 1579 1579 - const db = getDb(); 1580 1580 - const row = db.prepare('SELECT * FROM addresses WHERE id = ?').get(id); 1569 1569 + const row = getAddress(id); 1581 1570 return { success: true, data: row || {} }; 1582 1571 } catch (error) { 1583 1572 console.error('datastore-get-address error:', error); ··· 1588 1577 ipcMain.handle('datastore-update-address', async (ev, data) => { 1589 1578 try { 1590 1579 const { id, updates } = data; 1591 1591 - const db = getDb(); 1592 1592 - const existing = db.prepare('SELECT * FROM addresses WHERE id = ?').get(id); 1593 1593 - if (!existing) { 1580 1580 + const updated = updateAddress(id, updates); 1581 1581 + if (!updated) { 1594 1582 return { success: false, error: 'Address not found' }; 1595 1583 } 1596 1596 - 1597 1597 - const updated = { ...existing, ...updates, updatedAt: now() }; 1598 1598 - const columns = Object.keys(updated).filter(k => k !== 'id'); 1599 1599 - const setClause = columns.map(col => `${col} = ?`).join(', '); 1600 1600 - const values = columns.map(col => updated[col]); 1601 1601 - 1602 1602 - db.prepare(`UPDATE addresses SET ${setClause} WHERE id = ?`).run(...values, id); 1603 1603 - return { success: true, data: { id, ...updated } }; 1584 1584 + return { success: true, data: updated }; 1604 1585 } catch (error) { 1605 1586 console.error('datastore-update-address error:', error); 1606 1587 return { success: false, error: error.message }; ··· 1610 1591 ipcMain.handle('datastore-query-addresses', async (ev, data) => { 1611 1592 try { 1612 1593 const { filter = {} } = data; 1613 1613 - const db = getDb(); 1614 1614 - 1615 1615 - let sql = 'SELECT * FROM addresses WHERE 1=1'; 1616 1616 - const params = []; 1617 1617 - 1618 1618 - if (filter.domain) { 1619 1619 - sql += ' AND domain = ?'; 1620 1620 - params.push(filter.domain); 1621 1621 - } 1622 1622 - if (filter.protocol) { 1623 1623 - sql += ' AND protocol = ?'; 1624 1624 - params.push(filter.protocol); 1625 1625 - } 1626 1626 - if (filter.starred !== undefined) { 1627 1627 - sql += ' AND starred = ?'; 1628 1628 - params.push(filter.starred); 1629 1629 - } 1630 1630 - if (filter.tag) { 1631 1631 - sql += ' AND tags LIKE ?'; 1632 1632 - params.push(`%${filter.tag}%`); 1633 1633 - } 1634 1634 - 1635 1635 - // Sort 1636 1636 - const sortMap = { 1637 1637 - lastVisit: 'lastVisitAt DESC', 1638 1638 - visitCount: 'visitCount DESC', 1639 1639 - created: 'createdAt DESC' 1640 1640 - }; 1641 1641 - sql += ` ORDER BY ${sortMap[filter.sortBy] || 'updatedAt DESC'}`; 1642 1642 - 1643 1643 - // Limit 1644 1644 - if (filter.limit) { 1645 1645 - sql += ' LIMIT ?'; 1646 1646 - params.push(filter.limit); 1647 1647 - } 1648 1648 - 1649 1649 - const results = db.prepare(sql).all(...params); 1594 1594 + const results = queryAddresses(filter); 1650 1595 return { success: true, data: results }; 1651 1596 } catch (error) { 1652 1597 console.error('datastore-query-addresses error:', error); ··· 1657 1602 ipcMain.handle('datastore-add-visit', async (ev, data) => { 1658 1603 try { 1659 1604 const { addressId, options = {} } = data; 1660 1660 - const visitId = generateId('visit'); 1661 1661 - const timestamp = now(); 1662 1662 - const db = getDb(); 1663 1663 - 1664 1664 - db.prepare(` 1665 1665 - INSERT INTO visits (id, addressId, timestamp, duration, source, sourceId, windowType, metadata, scrollDepth, interacted) 1666 1666 - VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?) 1667 1667 - `).run( 1668 1668 - visitId, 1669 1669 - addressId, 1670 1670 - options.timestamp || timestamp, 1671 1671 - options.duration || 0, 1672 1672 - options.source || 'direct', 1673 1673 - options.sourceId || '', 1674 1674 - options.windowType || 'main', 1675 1675 - options.metadata || '{}', 1676 1676 - options.scrollDepth || 0, 1677 1677 - options.interacted || 0 1678 1678 - ); 1679 1679 - 1680 1680 - // Update address visit stats 1681 1681 - db.prepare(` 1682 1682 - UPDATE addresses SET lastVisitAt = ?, visitCount = visitCount + 1, updatedAt = ? 1683 1683 - WHERE id = ? 1684 1684 - `).run(timestamp, timestamp, addressId); 1685 1685 - 1686 1686 - return { success: true, id: visitId }; 1605 1605 + const result = addVisit(addressId, options); 1606 1606 + return { success: true, id: result.id }; 1687 1607 } catch (error) { 1688 1608 console.error('datastore-add-visit error:', error); 1689 1609 return { success: false, error: error.message }; ··· 1693 1613 ipcMain.handle('datastore-query-visits', async (ev, data) => { 1694 1614 try { 1695 1615 const { filter = {} } = data; 1696 1696 - const db = getDb(); 1697 1697 - 1698 1698 - let sql = 'SELECT * FROM visits WHERE 1=1'; 1699 1699 - const params = []; 1700 1700 - 1701 1701 - if (filter.addressId) { 1702 1702 - sql += ' AND addressId = ?'; 1703 1703 - params.push(filter.addressId); 1704 1704 - } 1705 1705 - if (filter.source) { 1706 1706 - sql += ' AND source = ?'; 1707 1707 - params.push(filter.source); 1708 1708 - } 1709 1709 - if (filter.since) { 1710 1710 - const since = typeof filter.since === 'number' ? filter.since : now() - filter.since; 1711 1711 - sql += ' AND timestamp >= ?'; 1712 1712 - params.push(since); 1713 1713 - } 1714 1714 - 1715 1715 - sql += ' ORDER BY timestamp DESC'; 1716 1716 - 1717 1717 - if (filter.limit) { 1718 1718 - sql += ' LIMIT ?'; 1719 1719 - params.push(filter.limit); 1720 1720 - } 1721 1721 - 1722 1722 - const results = db.prepare(sql).all(...params); 1616 1616 + const results = queryVisits(filter); 1723 1617 return { success: true, data: results }; 1724 1618 } catch (error) { 1725 1619 console.error('datastore-query-visits error:', error); ··· 1730 1624 ipcMain.handle('datastore-add-content', async (ev, data) => { 1731 1625 try { 1732 1626 const { options = {} } = data; 1733 1733 - const contentId = generateId('content'); 1734 1734 - const timestamp = now(); 1735 1735 - const db = getDb(); 1736 1736 - 1737 1737 - db.prepare(` 1738 1738 - INSERT INTO content (id, title, content, mimeType, contentType, language, encoding, tags, addressRefs, parentId, metadata, createdAt, updatedAt, syncPath, synced, starred, archived) 1739 1739 - VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?) 1740 1740 - `).run( 1741 1741 - contentId, 1742 1742 - options.title || 'Untitled', 1743 1743 - options.content || '', 1744 1744 - options.mimeType || 'text/plain', 1745 1745 - options.contentType || 'plain', 1746 1746 - options.language || '', 1747 1747 - options.encoding || 'utf-8', 1748 1748 - options.tags || '', 1749 1749 - options.addressRefs || '', 1750 1750 - options.parentId || '', 1751 1751 - options.metadata || '{}', 1752 1752 - timestamp, 1753 1753 - timestamp, 1754 1754 - options.syncPath || '', 1755 1755 - options.synced || 0, 1756 1756 - options.starred || 0, 1757 1757 - options.archived || 0 1758 1758 - ); 1759 1759 - return { success: true, id: contentId }; 1627 1627 + const result = addContent(options); 1628 1628 + return { success: true, id: result.id }; 1760 1629 } catch (error) { 1761 1630 console.error('datastore-add-content error:', error); 1762 1631 return { success: false, error: error.message }; ··· 1766 1635 ipcMain.handle('datastore-query-content', async (ev, data) => { 1767 1636 try { 1768 1637 const { filter = {} } = data; 1769 1769 - const db = getDb(); 1770 1770 - 1771 1771 - let sql = 'SELECT * FROM content WHERE 1=1'; 1772 1772 - const params = []; 1773 1773 - 1774 1774 - if (filter.contentType) { 1775 1775 - sql += ' AND contentType = ?'; 1776 1776 - params.push(filter.contentType); 1777 1777 - } 1778 1778 - if (filter.mimeType) { 1779 1779 - sql += ' AND mimeType = ?'; 1780 1780 - params.push(filter.mimeType); 1781 1781 - } 1782 1782 - if (filter.synced !== undefined) { 1783 1783 - sql += ' AND synced = ?'; 1784 1784 - params.push(filter.synced); 1785 1785 - } 1786 1786 - if (filter.starred !== undefined) { 1787 1787 - sql += ' AND starred = ?'; 1788 1788 - params.push(filter.starred); 1789 1789 - } 1790 1790 - if (filter.tag) { 1791 1791 - sql += ' AND tags LIKE ?'; 1792 1792 - params.push(`%${filter.tag}%`); 1793 1793 - } 1794 1794 - 1795 1795 - // Sort 1796 1796 - const sortMap = { 1797 1797 - updated: 'updatedAt DESC', 1798 1798 - created: 'createdAt DESC' 1799 1799 - }; 1800 1800 - sql += ` ORDER BY ${sortMap[filter.sortBy] || 'updatedAt DESC'}`; 1801 1801 - 1802 1802 - if (filter.limit) { 1803 1803 - sql += ' LIMIT ?'; 1804 1804 - params.push(filter.limit); 1805 1805 - } 1806 1806 - 1807 1807 - const results = db.prepare(sql).all(...params); 1638 1638 + const results = queryContent(filter); 1808 1639 return { success: true, data: results }; 1809 1640 } catch (error) { 1810 1641 console.error('datastore-query-content error:', error); ··· 1815 1646 ipcMain.handle('datastore-get-table', async (ev, data) => { 1816 1647 try { 1817 1648 const { tableName } = data; 1818 1818 - const db = getDb(); 1819 1819 - // Validate table name against known tables 1820 1820 - const validTables = ['addresses', 'visits', 'content', 'tags', 'address_tags', 'blobs', 'scripts_data', 'feeds', 'extensions', 'extension_settings']; 1821 1821 - if (!validTables.includes(tableName)) { 1649 1649 + if (!isValidTable(tableName)) { 1822 1650 return { success: false, error: `Invalid table name: ${tableName}` }; 1823 1651 } 1824 1824 - const rows = db.prepare(`SELECT * FROM ${tableName}`).all(); 1825 1825 - // Convert to object keyed by id for compatibility 1826 1826 - const table = {}; 1827 1827 - for (const row of rows) { 1828 1828 - table[row.id] = row; 1829 1829 - } 1652 1652 + const table = getTable(tableName); 1830 1653 return { success: true, data: table }; 1831 1654 } catch (error) { 1832 1655 console.error('datastore-get-table error:', error); ··· 1837 1660 ipcMain.handle('datastore-set-row', async (ev, data) => { 1838 1661 try { 1839 1662 const { tableName, rowId, rowData } = data; 1840 1840 - const db = getDb(); 1841 1841 - // Validate table name against known tables 1842 1842 - const validTables = ['addresses', 'visits', 'content', 'tags', 'address_tags', 'blobs', 'scripts_data', 'feeds', 'extensions', 'extension_settings']; 1843 1843 - if (!validTables.includes(tableName)) { 1663 1663 + if (!isValidTable(tableName)) { 1844 1664 return { success: false, error: `Invalid table name: ${tableName}` }; 1845 1665 } 1846 1846 - const row = { id: rowId, ...rowData }; 1847 1847 - const columns = Object.keys(row); 1848 1848 - const placeholders = columns.map(() => '?').join(', '); 1849 1849 - const values = columns.map(col => row[col]); 1850 1850 - 1851 1851 - db.prepare(`INSERT OR REPLACE INTO ${tableName} (${columns.join(', ')}) VALUES (${placeholders})`).run(...values); 1666 1666 + setRow(tableName, rowId, rowData); 1852 1667 return { success: true }; 1853 1668 } catch (error) { 1854 1669 console.error('datastore-set-row error:', error); ··· 1858 1673 1859 1674 ipcMain.handle('datastore-get-stats', async () => { 1860 1675 try { 1861 1861 - const db = getDb(); 1862 1862 - const stats = { 1863 1863 - totalAddresses: db.prepare('SELECT COUNT(*) as count FROM addresses').get().count, 1864 1864 - totalVisits: db.prepare('SELECT COUNT(*) as count FROM visits').get().count, 1865 1865 - avgVisitDuration: db.prepare('SELECT AVG(duration) as avg FROM visits').get().avg || 0, 1866 1866 - totalContent: db.prepare('SELECT COUNT(*) as count FROM content').get().count, 1867 1867 - syncedContent: db.prepare('SELECT COUNT(*) as count FROM content WHERE synced = 1').get().count 1868 1868 - }; 1676 1676 + const stats = getStats(); 1869 1677 return { success: true, data: stats }; 1870 1678 } catch (error) { 1871 1679 console.error('datastore-get-stats error:', error); ··· 1875 1683 1876 1684 // ***** Tag IPC Handlers ***** 1877 1685 1878 1878 - // Get or create a tag by name 1879 1686 ipcMain.handle('datastore-get-or-create-tag', async (ev, data) => { 1880 1687 try { 1881 1688 const { name } = data; 1882 1882 - console.log('datastore-get-or-create-tag:', name); 1883 1883 - const slug = name.toLowerCase().trim().replace(/\s+/g, '-'); 1884 1884 - const timestamp = now(); 1885 1885 - const db = getDb(); 1886 1886 - 1887 1887 - // Look for existing tag by name (case-insensitive) 1888 1888 - const existingTag = db.prepare('SELECT * FROM tags WHERE LOWER(name) = LOWER(?)').get(name); 1889 1889 - 1890 1890 - if (existingTag) { 1891 1891 - console.log(' -> found existing tag:', existingTag.id); 1892 1892 - return { success: true, data: existingTag, created: false }; 1893 1893 - } 1894 1894 - 1895 1895 - // Create new tag 1896 1896 - const tagId = generateId('tag'); 1897 1897 - db.prepare(` 1898 1898 - INSERT INTO tags (id, name, slug, color, parentId, description, metadata, createdAt, updatedAt, frequency, lastUsedAt, frecencyScore) 1899 1899 - VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?) 1900 1900 - `).run(tagId, name.trim(), slug, '#999999', '', '', '{}', timestamp, timestamp, 0, 0, 0); 1901 1901 - 1902 1902 - const newTag = db.prepare('SELECT * FROM tags WHERE id = ?').get(tagId); 1903 1903 - console.log(' -> created new tag:', tagId); 1904 1904 - return { success: true, data: newTag, created: true }; 1689 1689 + const result = getOrCreateTag(name); 1690 1690 + return { success: true, data: result.tag, created: result.created }; 1905 1691 } catch (error) { 1906 1692 console.error('datastore-get-or-create-tag error:', error); 1907 1693 return { success: false, error: error.message }; 1908 1694 } 1909 1695 }); 1910 1696 1911 1911 - // Tag an address (create the link and update frecency) 1912 1697 ipcMain.handle('datastore-tag-address', async (ev, data) => { 1913 1698 try { 1914 1699 const { addressId, tagId } = data; 1915 1915 - console.log('datastore-tag-address:', { addressId, tagId }); 1916 1916 - const timestamp = now(); 1917 1917 - const db = getDb(); 1918 1918 - 1919 1919 - // Check if link already exists 1920 1920 - const existingLink = db.prepare('SELECT * FROM address_tags WHERE addressId = ? AND tagId = ?').get(addressId, tagId); 1921 1921 - if (existingLink) { 1922 1922 - return { success: true, data: existingLink, alreadyExists: true }; 1923 1923 - } 1924 1924 - 1925 1925 - // Create the link 1926 1926 - const linkId = generateId('address_tag'); 1927 1927 - db.prepare('INSERT INTO address_tags (id, addressId, tagId, createdAt) VALUES (?, ?, ?, ?)').run(linkId, addressId, tagId, timestamp); 1928 1928 - 1929 1929 - // Update tag frequency and frecency 1930 1930 - const tag = db.prepare('SELECT * FROM tags WHERE id = ?').get(tagId); 1931 1931 - if (tag) { 1932 1932 - const newFrequency = (tag.frequency || 0) + 1; 1933 1933 - const frecencyScore = calculateFrecency(newFrequency, timestamp); 1934 1934 - db.prepare('UPDATE tags SET frequency = ?, lastUsedAt = ?, frecencyScore = ?, updatedAt = ? WHERE id = ?') 1935 1935 - .run(newFrequency, timestamp, frecencyScore, timestamp, tagId); 1936 1936 - } 1937 1937 - 1938 1938 - const newLink = db.prepare('SELECT * FROM address_tags WHERE id = ?').get(linkId); 1939 1939 - return { success: true, data: newLink }; 1700 1700 + const result = tagAddress(addressId, tagId); 1701 1701 + return { success: true, data: result.link, alreadyExists: result.alreadyExists }; 1940 1702 } catch (error) { 1941 1703 console.error('datastore-tag-address error:', error); 1942 1704 return { success: false, error: error.message }; 1943 1705 } 1944 1706 }); 1945 1707 1946 1946 - // Untag an address (remove the link) 1947 1708 ipcMain.handle('datastore-untag-address', async (ev, data) => { 1948 1709 try { 1949 1710 const { addressId, tagId } = data; 1950 1950 - const db = getDb(); 1951 1951 - 1952 1952 - const result = db.prepare('DELETE FROM address_tags WHERE addressId = ? AND tagId = ?').run(addressId, tagId); 1953 1953 - return { success: true, removed: result.changes > 0 }; 1711 1711 + const removed = untagAddress(addressId, tagId); 1712 1712 + return { success: true, removed }; 1954 1713 } catch (error) { 1955 1714 console.error('datastore-untag-address error:', error); 1956 1715 return { success: false, error: error.message }; 1957 1716 } 1958 1717 }); 1959 1718 1960 1960 - // Get all tags sorted by frecency 1961 1719 ipcMain.handle('datastore-get-tags-by-frecency', async (ev, data = {}) => { 1962 1720 try { 1963 1721 const { domain } = data || {}; 1964 1964 - const db = getDb(); 1965 1965 - let tags = db.prepare('SELECT * FROM tags').all(); 1966 1966 - console.log('datastore-get-tags-by-frecency: tags table has', tags.length, 'tags'); 1967 1967 - 1968 1968 - // Recalculate frecency scores (they decay over time) 1969 1969 - tags = tags.map(tag => ({ 1970 1970 - ...tag, 1971 1971 - frecencyScore: calculateFrecency(tag.frequency || 0, tag.lastUsedAt || 0) 1972 1972 - })); 1973 1973 - 1974 1974 - // If domain provided, boost tags used on same-domain addresses 1975 1975 - if (domain) { 1976 1976 - // Find tag IDs used on addresses with matching domain using JOIN 1977 1977 - const domainTagIds = new Set( 1978 1978 - db.prepare(` 1979 1979 - SELECT DISTINCT at.tagId FROM address_tags at 1980 1980 - JOIN addresses a ON at.addressId = a.id 1981 1981 - WHERE a.domain = ? 1982 1982 - `).all(domain).map(row => row.tagId) 1983 1983 - ); 1984 1984 - 1985 1985 - // Apply 2x boost 1986 1986 - tags = tags.map(tag => ({ 1987 1987 - ...tag, 1988 1988 - frecencyScore: domainTagIds.has(tag.id) ? tag.frecencyScore * 2 : tag.frecencyScore 1989 1989 - })); 1990 1990 - } 1991 1991 - 1992 1992 - // Sort by frecency descending 1993 1993 - tags.sort((a, b) => b.frecencyScore - a.frecencyScore); 1994 1994 - 1722 1722 + const tags = getTagsByFrecency(domain); 1995 1723 return { success: true, data: tags }; 1996 1724 } catch (error) { 1997 1725 console.error('datastore-get-tags-by-frecency error:', error); ··· 1999 1727 } 2000 1728 }); 2001 1729 2002 2002 - // Get tags for a specific address 2003 1730 ipcMain.handle('datastore-get-address-tags', async (ev, data) => { 2004 1731 try { 2005 1732 const { addressId } = data; 2006 2006 - const db = getDb(); 2007 2007 - 2008 2008 - // Use JOIN to get tags directly 2009 2009 - const tags = db.prepare(` 2010 2010 - SELECT t.* FROM tags t 2011 2011 - JOIN address_tags at ON t.id = at.tagId 2012 2012 - WHERE at.addressId = ? 2013 2013 - `).all(addressId); 2014 2014 - 1733 1733 + const tags = getAddressTags(addressId); 2015 1734 return { success: true, data: tags }; 2016 1735 } catch (error) { 2017 1736 console.error('datastore-get-address-tags error:', error); ··· 2019 1738 } 2020 1739 }); 2021 1740 2022 2022 - // Get addresses with a specific tag 2023 1741 ipcMain.handle('datastore-get-addresses-by-tag', async (ev, data) => { 2024 1742 try { 2025 1743 const { tagId } = data; 2026 2026 - const db = getDb(); 2027 2027 - 2028 2028 - // Use JOIN to get addresses directly 2029 2029 - const addresses = db.prepare(` 2030 2030 - SELECT a.* FROM addresses a 2031 2031 - JOIN address_tags at ON a.id = at.addressId 2032 2032 - WHERE at.tagId = ? 2033 2033 - `).all(tagId); 2034 2034 - 1744 1744 + const addresses = getAddressesByTag(tagId); 2035 1745 return { success: true, data: addresses }; 2036 1746 } catch (error) { 2037 1747 console.error('datastore-get-addresses-by-tag error:', error); ··· 2039 1749 } 2040 1750 }); 2041 1751 2042 2042 - // Get addresses that have no tags 2043 1752 ipcMain.handle('datastore-get-untagged-addresses', async (ev, data) => { 2044 1753 try { 2045 2045 - const db = getDb(); 2046 2046 - // Use LEFT JOIN + NULL check to find untagged addresses 2047 2047 - const addresses = db.prepare(` 2048 2048 - SELECT a.* FROM addresses a 2049 2049 - LEFT JOIN address_tags at ON a.id = at.addressId 2050 2050 - WHERE at.id IS NULL 2051 2051 - ORDER BY a.visitCount DESC 2052 2052 - `).all(); 2053 2053 - 1754 1754 + const addresses = getUntaggedAddresses(); 2054 1755 return { success: true, data: addresses }; 2055 1756 } catch (error) { 2056 1757 console.error('datastore-get-untagged-addresses error:', error); ··· 2784 2485 2785 2486 console.log('winDevtoolsConfig:', bw.id, 'openDevTools:', params.openDevTools, 'address:', params.address); 2786 2487 2787 2787 - // Check if devTools should be opened 2788 2788 - if (params.openDevTools === true) { 2488 2488 + // Check if devTools should be opened (never in test profiles) 2489 2489 + if (params.openDevTools === true && !isTestProfile) { 2789 2490 const isDetached = params.detachedDevTools === true; 2790 2491 // Determine if detached mode should be used 2791 2492 // activate: false prevents devtools from stealing focus (only works with detach/undocked)