Allow VS Code to recognise files with "sourcekit-lsp" scheme and pass it to SourceKitLSP inorder to provide Semantic Functionality

Author: lokesh-tr

src/WorkspaceContext.ts

@@ -522,7 +522,10 @@ export class WorkspaceContext implements vscode.Disposable {
     async focusUri(uri?: vscode.Uri) {
         this.currentDocument = uri ?? null;
         this.updateContextKeysForFile();
-        if (this.currentDocument?.scheme === "file") {
+        if (
+            this.currentDocument?.scheme === "file" ||
+            this.currentDocument?.scheme === "sourcekit-lsp"
+        ) {
             await this.focusPackageUri(this.currentDocument);
         }
     }

src/sourcekit-lsp/LanguageClientManager.ts

@@ -43,6 +43,7 @@ export class LanguageClientManager {
 
     // document selector used by language client
     static appleLangDocumentSelector = [
+        { scheme: "sourcekit-lsp", language: "swift" },
         { scheme: "file", language: "swift" },
         { scheme: "untitled", language: "swift" },
         { scheme: "file", language: "objective-c" },

src/sourcekit-lsp/getReferenceDocument.ts

@@ -22,7 +22,7 @@ export function activateGetReferenceDocument(client: langclient.LanguageClient):
         {
             provideTextDocumentContent: async (uri, token) => {
                 const params: GetReferenceDocumentParams = {
-                    uri: uri.toString(true),
+                    uri: uri.toString(),
                 };
 
                 const result = await client.sendRequest(GetReferenceDocumentRequest, params, token);

src/sourcekit-lsp/peekDocuments.ts

@@ -32,10 +32,7 @@ export function activatePeekDocuments(client: langclient.LanguageClient): vscode
 
             await vscode.commands.executeCommand(
                 "editor.action.peekLocations",
-                vscode.Uri.from({
-                    scheme: "file",
-                    path: new URL(params.uri).pathname,
-                }),
+                vscode.Uri.parse(new URL(params.uri).href, true),
                 new vscode.Position(params.position.line, params.position.character),
                 locations,
                 "peek"