Merge pull request #135 from digma-ai/workaround/go-links

workaround to make go links work

Author: doppleware

src/services/EditorHelper.ts

@@ -19,6 +19,7 @@ export interface EditorInfo {
 export interface InstrumentationInfo {
     instrumentationName?: string;
     spanName?: string;
+    fullName?: string
 }
 
 export class EditorHelper {

src/services/documentInfoProvider.ts

@@ -7,6 +7,7 @@ import { Token, TokenType } from './languages/tokens';
 import { Dictionary, Future } from './utils';
 import { EndpointInfo, SpanInfo, SymbolInfo, CodeObjectInfo } from './languages/extractors';
 import { InstrumentationInfo } from './EditorHelper';
+import { SymbolInformation } from 'vscode';
 
 export class DocumentInfoProvider implements vscode.Disposable
 {
@@ -47,11 +48,31 @@ export class DocumentInfoProvider implements vscode.Disposable
 
     public async searchForSpan(instrumentationInfo: InstrumentationInfo): Promise<SpanInfo|undefined>{
         const codeFileHint = instrumentationInfo.instrumentationName;
-        if (codeFileHint){
+        
+        if (codeFileHint ){
+            
+            if (vscode.window.activeTextEditor?.document.fileName.toLocaleLowerCase().endsWith(".go")){
+
+            //TODO: change to use document info we alrady scanned 
+                let regex = /(\(\*?.*\).*)/;
+                //workaround for GO
+                let match = instrumentationInfo.fullName?.match(regex)?.firstOrDefault();
+                if (match){
+                    match =match?.replace("(*","").replace(")","");
+                    let codeLocations:SymbolInformation[] =  await vscode.commands.executeCommand("vscode.executeWorkspaceSymbolProvider", match);
+                    if (codeLocations){
+                        codeLocations=codeLocations.filter(x=>x.kind===vscode.SymbolKind.Method && x.name===match);
+                        if (codeLocations.length===1){
+                            return new SpanInfo(instrumentationInfo.fullName!,instrumentationInfo.spanName!, codeLocations[0].location.range, codeLocations[0].location.uri);
+                        }
+                    }
+                }
+            }
+
 
             if (codeFileHint==='__main__'){
 
-                let doc = vscode.window.activeTextEditor?.document
+                let doc = vscode.window.activeTextEditor?.document;
                 if (doc){
                     const docInfo = await this.getDocumentInfo(doc);
                     if (docInfo){

src/services/languages/go/methodExtractor.ts

@@ -62,7 +62,7 @@ export class GoMethodExtractor implements IMethodExtractor{
             // "AuthController.Error" => "AuthController.Error"
             // "(*AuthController).Error" => "(*AuthController).Error"
             // "(AuthController).Error" => "AuthController.Error"
-            let name = s.name.replace(/\(([^*]+)\)\.(.+)/, "$1.$2")
+            let name = s.name.replace(/\(([^*]+)\)\.(.+)/, "$1.$2");
             return {
                 id: packageDefinitionName === "main" ? packagePath + '$_$' + `main.${name}` : packagePath + '$_$' + name,
                 name: name,

src/views/codeAnalytics/InsightListView/EndpointInsight.ts

@@ -179,7 +179,7 @@ export class SlowestSpansListViewItemsCreator implements IInsightListViewItemsCr
         var spansLocations = spans.map(span=> 
                                              { return {
                                                 slowspaninfo : span, 
-                                                spanSearchResult : this._documentInfoProvider.searchForSpan({ instrumentationName : span.spanInfo.instrumentationLibrary.split(".").join( " "), spanName :span.spanInfo.name })
+                                                spanSearchResult : this._documentInfoProvider.searchForSpan({ instrumentationName : span.spanInfo.instrumentationLibrary.split(".").join( " "), spanName :span.spanInfo.name, fullName:span.spanInfo.name })
                                                 };
                                              });