@velvetmonkey/vault-core 2.0.145 → 2.0.147

package/dist/entities.js

@@ -248,8 +248,12 @@ const FRONTMATTER_TYPE_MAP = {
     country: 'locations', region: 'locations',
     concept: 'concepts', idea: 'concepts', topic: 'concepts',
 };
-function mapFrontmatterType(type) {
-    return FRONTMATTER_TYPE_MAP[type.toLowerCase()];
+function mapFrontmatterType(type, customCategories) {
+    const lower = type.toLowerCase();
+    // Custom categories take priority — frontmatter type becomes the category name
+    if (customCategories?.[lower])
+        return lower;
+    return FRONTMATTER_TYPE_MAP[lower];
 }
 /**
  * Categorize an entity based on its name and optional frontmatter type
@@ -296,10 +300,10 @@ const FOLDER_CATEGORY_MAP = {
     'finance': 'finance',
     'hobbies': 'hobbies',
 };
-function categorizeEntity(name, techKeywords, frontmatterType, notePath) {
-    // 0. Frontmatter type takes priority
+function categorizeEntity(name, techKeywords, frontmatterType, notePath, customCategories) {
+    // 0. Frontmatter type takes priority (custom categories checked first)
     if (frontmatterType) {
-        const mapped = mapFrontmatterType(frontmatterType);
+        const mapped = mapFrontmatterType(frontmatterType, customCategories);
         if (mapped)
             return mapped;
     }
@@ -442,6 +446,7 @@ async function scanDirectory(dirPath, basePath, excludeFolders) {
 export async function scanVaultEntities(vaultPath, options = {}) {
     const excludeFolders = options.excludeFolders ?? [];
     const techKeywords = options.techKeywords ?? DEFAULT_TECH_KEYWORDS;
+    const customCategories = options.customCategories;
     // Scan vault for all markdown files
     const allEntities = await scanDirectory(vaultPath, vaultPath, excludeFolders);
     // Filter out periodic notes, invalid entries, and stop-listed entities
@@ -494,7 +499,7 @@ export async function scanVaultEntities(vaultPath, options = {}) {
         },
     };
     for (const entity of uniqueEntities) {
-        const category = categorizeEntity(entity.name, techKeywords, entity.frontmatterType, entity.relativePath);
+        const category = categorizeEntity(entity.name, techKeywords, entity.frontmatterType, entity.relativePath, customCategories);
         // Store as EntityWithAliases object
         const entityObj = {
             name: entity.name,
@@ -502,41 +507,40 @@ export async function scanVaultEntities(vaultPath, options = {}) {
             aliases: entity.aliases,
             description: entity.description,
         };
+        // Initialize custom category array if needed
+        if (!index[category]) {
+            index[category] = [];
+        }
         index[category].push(entityObj);
     }
-    // Sort each category by name
+    // Sort each category by name (including custom categories)
     const sortByName = (a, b) => {
         const nameA = typeof a === 'string' ? a : a.name;
         const nameB = typeof b === 'string' ? b : b.name;
         return nameA.localeCompare(nameB);
     };
-    const allCategories = [
-        'technologies', 'acronyms', 'people', 'projects', 'organizations',
-        'locations', 'concepts', 'animals', 'media', 'events', 'documents',
-        'vehicles', 'health', 'finance', 'food', 'hobbies', 'other',
-    ];
-    for (const cat of allCategories) {
+    const allCategoryKeys = Object.keys(index).filter(k => k !== '_metadata');
+    for (const cat of allCategoryKeys) {
         index[cat].sort(sortByName);
     }
     // Update metadata
-    index._metadata.total_entities = allCategories.reduce((sum, cat) => sum + index[cat].length, 0);
+    index._metadata.total_entities = allCategoryKeys.reduce((sum, cat) => sum + index[cat].length, 0);
     return index;
 }
 /**
  * Get all entities as a flat array (for wikilink matching)
  * Handles both legacy string format and new EntityWithAliases format
  */
-/** All entity category keys (excludes _metadata) */
-const ALL_ENTITY_CATEGORIES = [
-    'technologies', 'acronyms', 'people', 'projects', 'organizations',
-    'locations', 'concepts', 'animals', 'media', 'events', 'documents',
-    'vehicles', 'health', 'finance', 'food', 'hobbies', 'periodical', 'other',
-];
+/** Get all category keys from an index (includes custom categories, excludes _metadata) */
+function getIndexCategories(index) {
+    return Object.keys(index).filter(k => k !== '_metadata');
+}
 export function getAllEntities(index) {
     const result = [];
-    for (const cat of ALL_ENTITY_CATEGORIES) {
-        if (index[cat])
-            result.push(...index[cat]);
+    for (const cat of getIndexCategories(index)) {
+        const entities = index[cat];
+        if (Array.isArray(entities))
+            result.push(...entities);
     }
     return result;
 }
@@ -552,8 +556,7 @@ export function getAllEntities(index) {
  */
 export function getAllEntitiesWithTypes(index) {
     const result = [];
-    const categories = ALL_ENTITY_CATEGORIES;
-    for (const category of categories) {
+    for (const category of getIndexCategories(index)) {
         const entities = index[category];
         // Skip undefined or empty categories
         if (!entities || !Array.isArray(entities)) {

package/dist/index.d.ts

@@ -4,7 +4,8 @@
  * Shared vault utilities for the Flywheel ecosystem.
  * Used by both Flywheel (read) and Flywheel Memory (write).
  */
-export type { EntityIndex, EntityCategory, EntityWithAliases, Entity, EntityWithType, ScanOptions, WikilinkOptions, WikilinkResult, ImplicitEntityConfig, ExtendedWikilinkOptions, ImplicitEntityMatch, ResolveAliasOptions, ProtectedZone, ProtectedZoneType, } from './types.js';
+export type { EntityIndex, EntityCategory, DefaultEntityCategory, EntityWithAliases, Entity, EntityWithType, ScanOptions, WikilinkOptions, WikilinkResult, ImplicitEntityConfig, ExtendedWikilinkOptions, ImplicitEntityMatch, ResolveAliasOptions, ProtectedZone, ProtectedZoneType, } from './types.js';
+export { DEFAULT_ENTITY_CATEGORIES, getIndexCategory, ensureIndexCategory, } from './types.js';
 export { COMMON_ENGLISH_WORDS } from './common-words.js';
 export { stem } from './stemmer.js';
 export { scanVaultEntities, getAllEntities, getAllEntitiesWithTypes, getEntityName, getEntityAliases, loadEntityCache, saveEntityCache, ENTITY_CACHE_VERSION, } from './entities.js';

package/dist/index.js

@@ -4,6 +4,8 @@
  * Shared vault utilities for the Flywheel ecosystem.
  * Used by both Flywheel (read) and Flywheel Memory (write).
  */
+// Type helpers and constants
+export { DEFAULT_ENTITY_CATEGORIES, getIndexCategory, ensureIndexCategory, } from './types.js';
 // Common English words (frequency list for alias filtering)
 export { COMMON_ENGLISH_WORDS } from './common-words.js';
 // Porter Stemmer (for morphological entity matching)

package/dist/queries.js

@@ -122,6 +122,10 @@ export function getEntityIndexFromDb(stateDb) {
             hubScore: entity.hubScore,
             description: entity.description,
         };
+        // Initialize custom category array if needed
+        if (!index[entity.category]) {
+            index[entity.category] = [];
+        }
         index[entity.category].push(entityObj);
     }
     return index;

package/dist/sqlite.js

@@ -193,13 +193,8 @@ export function openStateDb(vaultPath) {
         replaceAllEntities: db.transaction((index) => {
             // Clear existing entities
             stateDb.clearEntities.run();
-            // Insert all entities by category
-            const categories = [
-                'technologies', 'acronyms', 'people', 'projects',
-                'organizations', 'locations', 'concepts', 'animals',
-                'media', 'events', 'documents', 'vehicles', 'health',
-                'finance', 'food', 'hobbies', 'other',
-            ];
+            // Insert all entities by category (including custom categories)
+            const categories = Object.keys(index).filter(k => k !== '_metadata');
             let total = 0;
             for (const category of categories) {
                 const entities = index[category];

package/dist/types.d.ts

@@ -2,9 +2,15 @@
  * Types for vault-core shared utilities
  */
 /**
- * Categories for entity classification
+ * Built-in entity categories
  */
-export type EntityCategory = 'technologies' | 'acronyms' | 'people' | 'projects' | 'organizations' | 'locations' | 'concepts' | 'animals' | 'media' | 'events' | 'documents' | 'vehicles' | 'health' | 'finance' | 'food' | 'hobbies' | 'periodical' | 'other';
+export type DefaultEntityCategory = 'technologies' | 'acronyms' | 'people' | 'projects' | 'organizations' | 'locations' | 'concepts' | 'animals' | 'media' | 'events' | 'documents' | 'vehicles' | 'health' | 'finance' | 'food' | 'hobbies' | 'periodical' | 'other';
+/**
+ * Entity category — built-in defaults + user-defined custom categories.
+ * Custom categories are configured via FlywheelConfig.custom_categories
+ * and populated from frontmatter `type:` fields.
+ */
+export type EntityCategory = DefaultEntityCategory | (string & Record<never, never>);
 /**
  * Entity with optional aliases from frontmatter
  */
@@ -32,9 +38,11 @@ export interface EntityWithType {
     entity: EntityWithAliases;
     category: EntityCategory;
 }
+/** Built-in category keys (excludes _metadata) */
+export declare const DEFAULT_ENTITY_CATEGORIES: DefaultEntityCategory[];
 /**
- * Entity index structure matching wikilink-cache.py output
- * Now supports both string[] (legacy) and EntityWithAliases[] (v2)
+ * Entity index structure — built-in categories as typed properties,
+ * custom categories accessed dynamically via getIndexCategory().
  */
 export interface EntityIndex {
     technologies: Entity[];
@@ -55,6 +63,8 @@ export interface EntityIndex {
     hobbies: Entity[];
     periodical: Entity[];
     other: Entity[];
+    /** Dynamic custom categories */
+    [customCategory: string]: Entity[] | EntityIndex['_metadata'];
     _metadata: {
         total_entities: number;
         generated_at: string;
@@ -64,6 +74,10 @@ export interface EntityIndex {
         version?: number;
     };
 }
+/** Get entities for a category (handles both built-in and custom) */
+export declare function getIndexCategory(index: EntityIndex, category: string): Entity[];
+/** Ensure a category array exists on the index */
+export declare function ensureIndexCategory(index: EntityIndex, category: string): void;
 /**
  * A protected zone in content where wikilinks should not be applied
  */
@@ -88,6 +102,14 @@ export interface ScanOptions {
      * Tech keywords for categorization
      */
     techKeywords?: string[];
+    /**
+     * Custom entity categories from FlywheelConfig.
+     * Keys are frontmatter `type:` values; entities with matching types
+     * get that key as their category instead of the default classifier.
+     */
+    customCategories?: Record<string, {
+        type_boost?: number;
+    }>;
 }
 /**
  * Options for wikilink application

package/dist/types.js

@@ -1,5 +1,22 @@
 /**
  * Types for vault-core shared utilities
  */
-export {};
+/** Built-in category keys (excludes _metadata) */
+export const DEFAULT_ENTITY_CATEGORIES = [
+    'technologies', 'acronyms', 'people', 'projects', 'organizations',
+    'locations', 'concepts', 'animals', 'media', 'events', 'documents',
+    'vehicles', 'health', 'finance', 'food', 'hobbies', 'periodical', 'other',
+];
+/** Get entities for a category (handles both built-in and custom) */
+export function getIndexCategory(index, category) {
+    if (category === '_metadata')
+        return [];
+    return index[category] ?? [];
+}
+/** Ensure a category array exists on the index */
+export function ensureIndexCategory(index, category) {
+    if (category !== '_metadata' && !index[category]) {
+        index[category] = [];
+    }
+}
 //# sourceMappingURL=types.js.map

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@velvetmonkey/vault-core",
-  "version": "2.0.145",
+  "version": "2.0.147",
   "description": "Shared vault utilities for Flywheel ecosystem (entity scanning, wikilinks, protected zones)",
   "type": "module",
   "main": "dist/index.js",