feat: support custom initializationOptions

Author: chrismwendt

src/connection.ts

@@ -71,8 +71,9 @@ export const webSocketTransport = ({
                 map(({ params }) => params)
             ),
         unsubscribe: () => {
-            attempt(() => socket.close())
-            attempt(() => connection.dispose())
+            console.log(new Error('unsub'))
+            console.log('attempt', attempt(() => socket.close()))
+            console.log('attempt2', attempt(() => connection.dispose()))
         },
     }
 }

src/features/definition.ts

@@ -15,7 +15,7 @@ export const definitionFeature: Feature<typeof DefinitionRequest.type, 'definiti
                         DefinitionRequest.type,
                         convertProviderParams({ textDocument, position }, { clientToServerURI })
                     )
-                    rewriteUris(result, serverToClientURI)
+                    rewriteUris(result, uri => serverToClientURI(uri, scopeRootUri))
                     return convertLocations(sourcegraph, result as Location | Location[] | null)
                 },
             }

src/features/feature.ts

@@ -14,7 +14,7 @@ export interface Feature<R extends RequestType<any, any, any, any>, C extends ke
         sourcegraph: typeof import('sourcegraph')
         scopeRootUri: URL | null
         clientToServerURI: (uri: URL) => URL
-        serverToClientURI: (uri: URL) => URL
+        serverToClientURI: (uri: URL, scopeRootUri: URL | null) => URL
         registerOptions: RegistrationOptions<R>
     }): Unsubscribable
 }

src/features/hover.ts

@@ -15,7 +15,7 @@ export const hoverFeature: Feature<typeof HoverRequest.type, 'hoverProvider'> =
                         HoverRequest.type,
                         convertProviderParams({ textDocument, position }, { clientToServerURI })
                     )
-                    rewriteUris(result, serverToClientURI)
+                    rewriteUris(result, uri => serverToClientURI(uri, scopeRootUri))
                     return convertHover(sourcegraph, result)
                 },
             }

src/features/references.ts

@@ -15,7 +15,7 @@ export const referencesFeature: Feature<typeof ReferencesRequest.type, 'referenc
                         ...convertProviderParams({ textDocument, position }, { clientToServerURI }),
                         context,
                     })
-                    rewriteUris(result, serverToClientURI)
+                    rewriteUris(result, uri => serverToClientURI(uri, scopeRootUri))
                     return convertLocations(sourcegraph, result)
                 },
             }

src/index.ts

@@ -1,7 +1,7 @@
 import { differenceBy, identity } from 'lodash'
 import * as path from 'path'
 import { from, noop, Subscription, Unsubscribable } from 'rxjs'
-import { map, scan, startWith } from 'rxjs/operators'
+import { map, scan, startWith, buffer, debounceTime } from 'rxjs/operators'
 import { DocumentSelector, ProgressReporter, WorkspaceRoot } from 'sourcegraph'
 import * as uuid from 'uuid'
 import {
@@ -71,11 +71,12 @@ export interface RegisterOptions {
     sourcegraph: SourcegraphAPI
     supportsWorkspaceFolders?: boolean
     clientToServerURI?: (uri: URL) => URL
-    serverToClientURI?: (uri: URL) => URL
+    serverToClientURI?: (uri: URL, scopeRootUri: URL | null) => URL
     afterInitialize?: (initializeResult: InitializeResult) => Promise<void> | void
     logger?: Logger
     transport: () => Promise<LSPConnection> | LSPConnection
     documentSelector: DocumentSelector
+    additionalInitializationOptions?: (url: URL) => any
 }
 export async function register({
     sourcegraph,
@@ -87,6 +88,7 @@ export async function register({
     afterInitialize = noop,
     transport: createConnection,
     documentSelector,
+    additionalInitializationOptions,
 }: RegisterOptions): Promise<LSPClient> {
     const subscriptions = new Subscription()
     // tslint:disable-next-line:no-object-literal-type-assertion
@@ -153,7 +155,10 @@ export async function register({
         }
     }
 
-    async function connect(clientRootUri: URL | null, initParams: InitializeParams): Promise<LSPConnection> {
+    async function connect(
+        clientRootUri: URL | null,
+        initParams: InitializeParams & { originalRootUri?: string | null }
+    ): Promise<LSPConnection> {
         const subscriptions = new Subscription()
         const decorationType = sourcegraph.app.createDecorationType()
         const connection = await createConnection()
@@ -188,7 +193,7 @@ export async function register({
         subscriptions.add(
             connection.observeNotification(PublishDiagnosticsNotification.type).subscribe(params => {
                 const uri = new URL(params.uri)
-                const sourcegraphTextDocumentUri = serverToClientURI(uri)
+                const sourcegraphTextDocumentUri = serverToClientURI(uri, clientRootUri)
                 diagnosticsByUri.set(sourcegraphTextDocumentUri.href, params.diagnostics)
                 for (const appWindow of sourcegraph.app.windows) {
                     for (const viewComponent of appWindow.visibleViewComponents) {
@@ -256,6 +261,7 @@ export async function register({
                     }
                 })
         )
+        connection.closeEvent.subscribe(() => console.log('closed'))
         await initializeConnection(connection, clientRootUri, initParams)
         return connection
     }
@@ -265,7 +271,13 @@ export async function register({
         clientRootUri: URL | null,
         initParams: InitializeParams
     ): Promise<void> {
-        const initializeResult = await connection.sendRequest(InitializeRequest.type, initParams)
+        // total hack
+        if (initParams.initializationOptions && clientRootUri) {
+            initParams.initializationOptions = initParams.initializationOptions(clientRootUri)
+        } else {
+            console.log('Garr, did not work', initParams.initializationOptions, clientRootUri)
+        }
+        const initializeResult = await connection.sendRequest(InitializeRequest.type, { ...initParams, rootPath: '/' })
         // Tell language server about all currently open text documents under this root
         syncTextDocuments(connection)
 
@@ -292,6 +304,8 @@ export async function register({
                 rootUri: null,
                 capabilities: clientCapabilities,
                 workspaceFolders: sourcegraph.workspace.roots.map(toLSPWorkspaceFolder({ clientToServerURI })),
+                originalRootUri: null,
+                initializationOptions: additionalInitializationOptions,
             }
         )
         subscriptions.add(connection)
@@ -359,15 +373,18 @@ export async function register({
                 {
                     processId: null,
                     rootUri: serverRootUri.href,
+                    originalRootUri: workspaceFolder.href,
                     capabilities: clientCapabilities,
                     workspaceFolders: null,
+                    initializationOptions: additionalInitializationOptions,
                 }
             )
             subscriptions.add(connection)
             try {
                 return await fn(connection)
             } finally {
                 connectionsByRootUri.delete(workspaceFolder.href)
+                console.log('unsub')
                 connection.unsubscribe()
             }
         }
@@ -380,9 +397,11 @@ export async function register({
                             new URL(root.uri.toString()),
                             {
                                 processId: null,
+                                originalRootUri: root.uri.toString(),
                                 rootUri: serverRootUri.href,
                                 capabilities: clientCapabilities,
                                 workspaceFolders: null,
+                                initializationOptions: additionalInitializationOptions,
                             }
                         )
                         subscriptions.add(connection)
@@ -399,6 +418,7 @@ export async function register({
         subscriptions.add(
             from(sourcegraph.workspace.rootChanges)
                 .pipe(
+                    buffer(from(sourcegraph.workspace.rootChanges).pipe(debounceTime(1000))),
                     startWith(null),
                     map(() => [...sourcegraph.workspace.roots]),
                     scan((before, after) => {
@@ -408,6 +428,7 @@ export async function register({
 
                         // Close connections for removed workspaces
                         const removed = differenceBy(before, after, root => root.uri.toString())
+                        console.log(before, after, added, removed)
                         // tslint:disable-next-line no-floating-promises
                         Promise.all(
                             removed.map(async root => {