[#54 Computed Fields]

README.md

@@ -115,18 +115,49 @@ const blogs = await mddb.getFiles({
 });
 ```
 
-# Configuring `markdowndb.config.js`
+## Computed Fields
+
+This feature helps you define functions that compute additional fields you want to include.
+
+### Step 1: Define the Computed Field Function
+
+Next, define a function that computes the additional field you want to include. In this example, we have a function named `addTitle` that extracts the title from the first heading in the AST (Abstract Syntax Tree) of a Markdown file.
+
+```javascript
+const addTitle = (fileInfo, ast) => {
+  // Find the first header node in the AST
+  const headerNode = ast.children.find((node) => node.type === "heading");
+
+  // Extract the text content from the header node
+  const title = headerNode
+    ? headerNode.children.map((child) => child.value).join("")
+    : "";
+
+  // Add the title property to the fileInfo
+  fileInfo.title = title;
+};
+```
+
+### Step 2: Indexing the Folder with Computed Fields
+
+Now, use the `client.indexFolder` method to scan and index the folder containing your Markdown files. Pass the `addTitle` function in the `computedFields` option array to include the computed title in the database.
+
+```javascript
+client.indexFolder(folderPath: "PATH_TO_FOLDER", customConfig: { computedFields: [addTitle] });
+```
+
+## Configuring `markdowndb.config.js`
 
 - Implement computed fields to dynamically calculate values based on specified logic or dependencies.
 - Specify the patterns for including or excluding files in MarkdownDB.
 
-## Example Configuration
+### Example Configuration
 
 Here's an example `markdowndb.config.js` with custom configurations:
 
 ```javascript
 export default {
-  customFields: [
+  computedFields: [
     (fileInfo, ast) => {
       // Your custom logic here
     },

src/lib/CustomConfig.ts

@@ -0,0 +1,6 @@
+import { FileInfo } from "./process.js";
+import { Root } from "remark-parse/lib/index.js";
+
+export interface CustomConfig {
+  computedFields?: ((fileInfo: FileInfo, ast: Root) => any)[];
+}

src/lib/indexFolder.ts

@@ -1,22 +1,26 @@
+import { CustomConfig } from "./CustomConfig.js";
 import { FileInfo, processFile } from "./process.js";
 import { recursiveWalkDir } from "./recursiveWalkDir.js";
 
 export function indexFolder(
   folderPath: string,
   pathToUrlResolver: (filePath: string) => string,
+  config: CustomConfig,
   ignorePatterns?: RegExp[]
 ) {
   const filePathsToIndex = recursiveWalkDir(folderPath);
   const filteredFilePathsToIndex = filePathsToIndex.filter((filePath) =>
     shouldIncludeFile(filePath, ignorePatterns)
   );
   const files: FileInfo[] = [];
+  const computedFields = config.computedFields || [];
   for (const filePath of filteredFilePathsToIndex) {
     const fileObject = processFile(
       folderPath,
       filePath,
       pathToUrlResolver,
-      filePathsToIndex
+      filePathsToIndex,
+      computedFields
     );
     files.push(fileObject);
   }

src/lib/markdowndb.ts

@@ -12,6 +12,7 @@
   getUniqueValues,
 } from "./databaseUtils.js";
 import fs from "fs";
+import { CustomConfig } from "./CustomConfig.js";
 
 const defaultFilePathToUrl = (filePath: string) => {
   let url = filePath
@@ -22,7 +23,7 @@
   return encodeURI(url);
 };
 
 const resolveLinkToUrlPath = (link: string, sourceFilePath?: string) => {
   if (!sourceFilePath) {
     return link;
   }
@@ -72,16 +73,19 @@
     // TODO support glob patterns
     ignorePatterns = [],
     pathToUrlResolver = defaultFilePathToUrl,
+    customConfig = {},
   }: {
     folderPath: string;
     ignorePatterns?: RegExp[];
     pathToUrlResolver?: (filePath: string) => string;
+    customConfig: CustomConfig;
   }) {
     await resetDatabaseTables(this.db);
 
     const fileObjects = indexFolder(
       folderPath,
       pathToUrlResolver,
+      customConfig,
       ignorePatterns
     );
     const filesToInsert = fileObjects.map(mapFileToInsert);
@@ -248,7 +252,7 @@
   }
 }
 
 function writeJsonToFile(filePath: string, jsonData: any[]) {
   try {
     const directory = path.dirname(filePath);
     if (!fs.existsSync(directory)) {
@@ -257,7 +261,7 @@
 
     const jsonString = JSON.stringify(jsonData, null, 2);
     fs.writeFileSync(filePath, jsonString);
   } catch (error: any) {
     console.error(`Error writing data to ${filePath}: ${error}`);
   }
 }

src/lib/parseFile.ts

@@ -28,6 +28,7 @@
   metadata.tasks = tasks;
 
   return {
+    ast,
     metadata,
     links,
   };
@@ -64,7 +65,7 @@
 
   const nodes = selectAll("*", ast);
   for (let index = 0; index < nodes.length; index++) {
     const node: any = nodes[index];
     if (node.value) {
       const textTags = node.value.match(/(?:^|\s)(#(\w+|\/|-|_)+)/g);
       if (textTags) {

src/lib/process.ts

@@ -4,6 +4,7 @@ import path from "path";
 
 import { File } from "./schema.js";
 import { WikiLink, parseFile } from "./parseFile.js";
+import { Root } from "remark-parse/lib/index.js";
 
 export interface FileInfo extends File {
   tags: string[];
@@ -16,7 +17,8 @@ export function processFile(
   rootFolder: string,
   filePath: string,
   pathToUrlResolver: (filePath: string) => string,
-  filePathsToIndex: string[]
+  filePathsToIndex: string[],
+  computedFields: ((fileInfo: FileInfo, ast: Root) => any)[]
 ) {
   // Remove rootFolder from filePath
   const relativePath = path.relative(rootFolder, filePath);
@@ -52,7 +54,7 @@ export function processFile(
     flag: "r",
   });
 
-  const { metadata, links } = parseFile(source, {
+  const { ast, metadata, links } = parseFile(source, {
     from: relativePath,
     permalinks: filePathsToIndex,
   });
@@ -66,6 +68,10 @@ export function processFile(
 
   const tags = metadata?.tags || [];
   fileInfo.tags = tags;
+  for (let index = 0; index < computedFields.length; index++) {
+    const customFieldFunction = computedFields[index];
+    customFieldFunction(fileInfo, ast);
+  }
 
   return fileInfo;
 }