feat: implemented provideReferences

vuejs · Jun 9, 2022 · 6b05d17 · 6b05d17
1 parent 86079ad
commit 6b05d17
Show file tree

Hide file tree

Showing 3 changed files with 29 additions and 6 deletions.
diff --git a/src/monaco/Monaco.vue b/src/monaco/Monaco.vue
@@ -31,7 +31,7 @@ const editor = shallowRef<monaco.editor.IStandaloneCodeEditor | undefined>(undef
 
 const currentModel = shallowRef<monaco.editor.ITextModel>(
   getOrCreateModel(
-    monaco.Uri.parse('playground:///demo.vue'),
+    monaco.Uri.parse('file:///demo.vue'),
     'vue',
     props.value ?? ''
   )

diff --git a/src/monaco/code2monaco.ts b/src/monaco/code2monaco.ts
@@ -141,11 +141,19 @@ export function asMarkdownString(markdownString: vscode.Hover['contents']): mona
     }
 }
 
-export function asLocation(definition: vscode.LocationLink): monaco.languages.Location {
-    return {
-        uri: asUri(definition.targetUri),
-        range: asRange(definition.targetRange),
-    };
+export function asLocation(definition: vscode.LocationLink | vscode.Location): monaco.languages.Location {
+    if ('targetUri' in definition) {
+        return {
+            uri: asUri(definition.targetUri),
+            range: asRange(definition.targetRange),
+        };
+    }
+    else {
+        return {
+            uri: asUri(definition.uri),
+            range: asRange(definition.range),
+        };
+    }
 }
 
 export function asUri(uri: vscode.URI): monaco.Uri {

diff --git a/src/monaco/ls.ts b/src/monaco/ls.ts
@@ -195,6 +195,21 @@ export async function setupLs(modelsMap: Ref<Map<string, monaco.editor.ITextMode
     const completionItems = new WeakMap<monaco.languages.CompletionItem, vscode.CompletionItem>();
 
     disposables.value.push(
+        // TODO: registerTokensProviderFactory
+        // TODO: setTokensProvider
+        // TODO: setMonarchTokensProvider
+        monaco.languages.registerReferenceProvider(lang, {
+            provideReferences: async (model, position) => {
+                const codeResult = await ls.findReferences(
+                    model.uri.toString(),
+                    monaco2code.asPosition(position),
+                );
+                // TODO: can't show if only one result from libs
+                if (codeResult) {
+                    return codeResult.map(code2monaco.asLocation);
+                }
+            },
+        }),
         monaco.languages.registerCompletionItemProvider(lang, {
             // https://github.com/johnsoncodehk/volar/blob/2f786182250d27e99cc3714fbfc7d209616e2289/packages/vue-language-server/src/registers/registerlanguageFeatures.ts#L57
             triggerCharacters: '!@#$%^&*()_+-=`~{}|[]\:";\'<>?,./ '.split(''),