Implement account auto completion

Author: Parsa2820

examples/accounts.ledger

@@ -0,0 +1,6 @@
+account Assets:Cash
+account Assets:Bank
+account Expenses:Food
+account Income:Salary
+account Liabilities:CreditCard
+account Equity:OpeningBalances

package-lock.json

@@ -49,7 +49,20 @@
         "mac": "cmd+shift+t",
         "when": "editorTextFocus && resourceExtname == '.ledger'"
       }
-    ]
+    ],
+    "configuration": {
+      "title": "ledger-cli",
+      "properties": {
+        "ledger.accountFiles": {
+          "type": "array",
+          "items": {
+            "type": "string"
+          },
+          "default": [],
+          "description": "List of ledger files to include for account auto-completion."
+        }
+      }
+    }
   },
   "scripts": {
     "vscode:prepublish": "npm run compile",

package.json

@@ -0,0 +1,76 @@
+import * as vscode from 'vscode';
+
+export class LedgerAccountCompletion implements vscode.CompletionItemProvider {
+  accounts: string[] = [];
+
+  constructor() {
+    const defaultAccountsFile = vscode.Uri.joinPath(vscode.extensions.getExtension('parsa2820.ledger-cli')!.extensionUri, 'examples', 'accounts.ledger').fsPath;
+    const accountFiles = vscode.workspace.getConfiguration('ledger').get<string[]>('accountFiles', []);
+    if (accountFiles.length === 0) {
+      accountFiles.push(defaultAccountsFile);
+    }
+    accountFiles.forEach(file => {
+      this.loadAccountFile(file).then(accounts => {
+        this.accounts.push(...accounts);
+      });
+    });
+  }
+
+  provideCompletionItems(document: vscode.TextDocument, position: vscode.Position, token: vscode.CancellationToken, context: vscode.CompletionContext): vscode.ProviderResult<vscode.CompletionList<vscode.CompletionItem> | vscode.CompletionItem[]> {
+    return this.accounts.map(account => new vscode.CompletionItem(account, vscode.CompletionItemKind.Variable));
+  }
+
+  resolveCompletionItem?(item: vscode.CompletionItem, token: vscode.CancellationToken): vscode.ProviderResult<vscode.CompletionItem> {
+    throw new Error('Method not implemented.');
+  }
+
+  private async loadAccountFile(filePath: string): Promise<string[]> {
+    try {
+      const fileUri = vscode.Uri.file(filePath);
+      const fileData = await vscode.workspace.fs.readFile(fileUri);
+      const fileText = Buffer.from(fileData).toString('utf8');
+      return this.loadAccounts(fileText);
+    } catch (error) {
+      console.error(`Error reading account file ${filePath}:`, error);
+      return [];
+    }
+  }
+
+  private loadAccounts(fileText: string): string[] {
+    const accounts: string[] = [];
+    const lines = fileText.split(/\r?\n/);
+    for (const line of lines) {
+      if (this.isAccountDeclarationLine(line)) {
+        const account = this.extractAccountDeclarationLineEntry(line);
+        if (account) {
+          accounts.push(account);
+        }
+      }
+    }
+    return accounts;
+  }
+
+  private isAccountDeclarationLine(text: string): boolean {
+    return this.getAccountDeclarationPattern().test(text);
+  }
+
+  private extractAccountDeclarationLineEntry(text: string): string | null {
+    const pattern = this.getAccountDeclarationPattern();
+    const match = pattern.exec(text);
+    if (match && match.groups) {
+      return match.groups['account'].trim();
+    }
+    return null;
+  }
+
+  private getAccountDeclarationPattern(): RegExp {
+    const accountNamePattern = /(?<account>[\[\(]?[A-Za-z0-9:_\-]+)[\]\)]?/;
+    const commentPattern = /\s*(?<comment>[;#%\|\*].*)?/;
+    return new RegExp(
+      "^account\\s" +
+      accountNamePattern.source +
+      commentPattern.source +
+      "$"
+    );
+  }
+}

src/completion.ts

@@ -1,6 +1,7 @@
 import * as vscode from 'vscode';
 import { LedgerDocumentFormatter } from './formatter';
 import { registerCommands } from './commands';
+import { LedgerAccountCompletion } from './completion';
 
 export function activate(context: vscode.ExtensionContext) {
 
@@ -9,6 +10,9 @@ export function activate(context: vscode.ExtensionContext) {
 	const formatterRegistration = vscode.languages.registerDocumentFormattingEditProvider('ledger', new LedgerDocumentFormatter());
 	context.subscriptions.push(formatterRegistration);
 
+	const completionRegistration = vscode.languages.registerCompletionItemProvider('ledger', new LedgerAccountCompletion());
+	context.subscriptions.push(completionRegistration);
+
 	registerCommands(context);
 }