feat: add catalog name completion in package.json (#75)

* feat: support catalog completion

* fix: remove incorrect range in catalog completion

* [autofix.ci] apply automated fixes

* fix: update completion item kind to text

* update

---------

Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com>
Co-authored-by: Vida Xie <vida_2020@163.com>

Author: 3 people

playground/package.json

@@ -14,7 +14,7 @@
     "ofetch": "^2.0.0-alpha.1",
     "array-includes": "latest",
     "axios": "",
-    "is-number": "",
+    "is-number": "catalog:",
     "lodash": "catalog:"
   }
 }

playground/pnpm-workspace.yaml

@@ -1,7 +1,9 @@
 catalog:
   lodash: 2.4.2
+  is-number: ^4.0.0
 catalogs:
   prod:
     '@deno/doc': jsr:^0.189.1
+    is-number: 5.0.0
   dev:
     is-number: ~7.0.0

src/core/workspace.ts

@@ -67,6 +67,10 @@ class WorkspaceContext {
     this.#invalidatedPaths.add(path)
   }
 
+  async getCatalogs(): Promise<CatalogsInfo | undefined> {
+    return this.#catalogs?.promise
+  }
+
   #createResolvedDependencyInfo(dependency: DependencyInfo, catalogs?: CatalogsInfo): ResolvedDependencyInfo {
     const resolution = resolveDependencySpec(dependency.rawName, dependency.rawSpec, catalogs)
 
@@ -110,7 +114,7 @@ class WorkspaceContext {
 
     const [info, catalogs] = await Promise.all([
       getDocumentText(uri).then((text) => extractor.getPackageManifestInfo(text)),
-      this.#catalogs!.promise,
+      this.#catalogs?.promise,
     ])
 
     if (!info)

src/providers/completion-item/catalog.ts

@@ -0,0 +1,33 @@
+import type { CompletionItemProvider, Position, TextDocument } from 'vscode'
+import { getResolvedDependencyByOffset, getWorkspaceContext } from '#core/workspace'
+import { CompletionItem, CompletionItemKind } from 'vscode'
+
+export class CatalogCompletionItemProvider implements CompletionItemProvider {
+  static triggers = [':']
+
+  async provideCompletionItems(document: TextDocument, position: Position) {
+    const offset = document.offsetAt(position)
+    const info = await getResolvedDependencyByOffset(document.uri, offset)
+    if (!info?.rawSpec.startsWith('catalog:'))
+      return
+
+    const ctx = await getWorkspaceContext(document.uri)
+    if (!ctx)
+      return
+
+    const catalogs = await ctx.getCatalogs()
+
+    if (!catalogs)
+      return
+
+    return Object.entries(catalogs).flatMap(([name, catalog]) => {
+      const version = catalog[info.resolvedName]
+      if (!version)
+        return []
+
+      const item = new CompletionItem(name, CompletionItemKind.Value)
+      item.detail = version
+      return [item]
+    })
+  }
+}

src/providers/completion-item/index.ts

@@ -1,7 +1,8 @@
-import { SUPPORTED_DOCUMENT_PATTERN } from '#constants'
+import { PACKAGE_JSON_BASENAME, SUPPORTED_DOCUMENT_PATTERN } from '#constants'
 import { config } from '#state'
 import { watchEffect } from 'reactive-vscode'
 import { languages } from 'vscode'
+import { CatalogCompletionItemProvider } from './catalog'
 import { VersionCompletionItemProvider } from './version'
 
 export function useCompletionItem() {
@@ -17,4 +18,14 @@ export function useCompletionItem() {
 
     onCleanup(() => disposable.dispose())
   })
+
+  watchEffect((onCleanup) => {
+    const disposable = languages.registerCompletionItemProvider(
+      { pattern: `**/${PACKAGE_JSON_BASENAME}` },
+      new CatalogCompletionItemProvider(),
+      ...CatalogCompletionItemProvider.triggers,
+    )
+
+    onCleanup(() => disposable.dispose())
+  })
 }