Support source code linkage given path prefix in launch command

example: `xprof --logdir=<logdir> --src_prefix=<prefixpath to repo>`
PiperOrigin-RevId: 844977541

Author: zzzaries

frontend/app/common/interfaces/capture_profile.ts

@@ -16,3 +16,9 @@ export interface CaptureProfileOptions {
   pythonTracerLevel: number;
   delay: number;
 }
+
+/** A serializable object with profiler configuration details. */
+export interface ProfilerConfig {
+  hideCaptureProfileButton: boolean;
+  srcPathPrefix: string;
+}

frontend/app/components/sidenav/sidenav.ts

@@ -7,7 +7,7 @@ import {NavigationEvent} from 'org_xprof/frontend/app/common/interfaces/navigati
 import {RunToolsMap} from 'org_xprof/frontend/app/common/interfaces/tool';
 import {CommunicationService} from 'org_xprof/frontend/app/services/communication_service/communication_service';
 import {DataServiceV2} from 'org_xprof/frontend/app/services/data_service_v2/data_service_v2';
-import {setCurrentRunAction, updateRunToolsMapAction} from 'org_xprof/frontend/app/store/actions';
+import {setCurrentRunAction, setProfilerConfigAction, updateRunToolsMapAction} from 'org_xprof/frontend/app/store/actions';
 import {getCurrentRun, getRunToolsMap} from 'org_xprof/frontend/app/store/selectors';
 import {firstValueFrom, Observable, ReplaySubject} from 'rxjs';
 import {takeUntil} from 'rxjs/operators';
@@ -171,6 +171,7 @@ export class SideNav implements OnInit, OnDestroy {
     const config = await firstValueFrom(
         this.dataService.getConfig().pipe(takeUntil(this.destroyed)));
     if (config) {
+      this.store.dispatch(setProfilerConfigAction({config}));
       this.hideCaptureProfileButton = config.hideCaptureProfileButton;
     }
   }

frontend/app/components/source_mapper/BUILD

@@ -25,6 +25,7 @@ xprof_ng_module(
         "@org_xprof//frontend/app/common/angular:angular_material_select",
         "@org_xprof//frontend/app/common/angular:angular_material_tooltip",
         "@org_xprof//frontend/app/common/constants",
+        "@org_xprof//frontend/app/common/interfaces",
         "@org_xprof//frontend/app/components/stack_trace_snippet",
         "@org_xprof//frontend/app/components/stack_trace_snippet:message",
         "@org_xprof//frontend/app/services/data_service_v2:data_service_v2_interface",

frontend/app/components/source_mapper/source_mapper.ng.html

@@ -9,6 +9,7 @@
       </mat-select>
     </mat-form-field>
     <stack-trace-snippet
+      [srcPathPrefix]="srcPathPrefix"
       [sourceFileAndLineNumber]="adaptedSourceFileAndLineNumber"
       [stackTrace]="adaptedStackTrace"
       [sourceContextWindow]="sourceContextWindow"

frontend/app/components/source_mapper/source_mapper.ts

@@ -2,9 +2,10 @@ import {Component, inject, Input, OnChanges, OnDestroy, SimpleChanges} from '@an
 import {Store} from '@ngrx/store';
 import {GRAPH_TYPE_DEFAULT, GRAPH_TYPE_ORIGINAL_HLO} from 'org_xprof/frontend/app/common/constants/constants';
 import {FileExtensionType} from 'org_xprof/frontend/app/common/constants/enums';
+import {ProfilerConfig} from 'org_xprof/frontend/app/common/interfaces/capture_profile';
 import {DATA_SERVICE_INTERFACE_TOKEN, DataServiceV2Interface} from 'org_xprof/frontend/app/services/data_service_v2/data_service_v2_interface';
 import {Address} from 'org_xprof/frontend/app/services/source_code_service/source_code_service_interface';
-import {getTagsState} from 'org_xprof/frontend/app/store/selectors';
+import {getProfilerConfig, getTagsState} from 'org_xprof/frontend/app/store/selectors';
 import {ReplaySubject} from 'rxjs';
 import {takeUntil} from 'rxjs/operators';
 
@@ -63,6 +64,8 @@ export class SourceMapper implements OnDestroy, OnChanges {
   sourceFileNames: string[] = [];
   selectedSourceFileName = '';
   tags: string[] = [];
+  // The prefix of the source file path if specified by users.
+  srcPathPrefix = '';
 
   constructor(private readonly store: Store<{}>) {
     this.store.select(getTagsState)
@@ -71,6 +74,12 @@ export class SourceMapper implements OnDestroy, OnChanges {
           if (!tags || tags.length === 0) return;
           this.tags = tags;
         });
+    this.store.select(getProfilerConfig)
+        .pipe(takeUntil(this.destroyed))
+        .subscribe((config: ProfilerConfig) => {
+          if (!config) return;
+          this.srcPathPrefix = config.srcPathPrefix;
+        });
   }
 
   ngOnChanges(changes: SimpleChanges) {

frontend/app/components/stack_trace_snippet/stack_frame_snippet.ts

@@ -20,6 +20,7 @@ export class StackFrameSnippet implements OnChanges, OnDestroy {
   @Input() sourceCodeSnippetAddress: Address|undefined = undefined;
   @Input() topOfStack: boolean|undefined = undefined;
   @Input() usingSourceFileAndLineNumber: boolean|undefined = undefined;
+  @Input() srcPathPrefix = '';
   private readonly route: ActivatedRoute = inject(ActivatedRoute);
   private readonly sourceCodeService: SourceCodeServiceInterface =
       inject(SOURCE_CODE_SERVICE_INTERFACE_TOKEN);
@@ -30,12 +31,15 @@ export class StackFrameSnippet implements OnChanges, OnDestroy {
   codeSearchLink: string|undefined = undefined;
   codeSearchLinkTooltip: string|undefined = undefined;
   lineNumberToMetricMap: Map<number, Metric>|undefined = undefined;
+  isCodeFetchEnabled = false;
 
   constructor() {
     this.route.params.pipe(takeUntil(this.destroy$)).subscribe((params) => {
-      this.sessionId = (params || {})['sessionId'];
+      this.sessionId = (params || {})['sessionId'] || (params || {})['run'] ||
+          this.sessionId;
       this.reload();
     });
+    this.isCodeFetchEnabled = this.sourceCodeService.isCodeFetchEnabled();
   }
 
   ngOnDestroy(): void {
@@ -60,7 +64,8 @@ export class StackFrameSnippet implements OnChanges, OnDestroy {
   }
 
   get loaded() {
-    return this.frame !== undefined || this.failure !== undefined;
+    return this.frame !== undefined || this.failure !== undefined ||
+        !this.isCodeFetchEnabled;
   }
 
   get isAtTopOfStack(): boolean {
@@ -111,7 +116,7 @@ export class StackFrameSnippet implements OnChanges, OnDestroy {
     this.sourceCodeService
         .codeSearchLink(
             this.sessionId, this.sourceCodeSnippetAddress.fileName,
-            this.sourceCodeSnippetAddress.lineNumber)
+            this.sourceCodeSnippetAddress.lineNumber, this.srcPathPrefix)
         .pipe(takeUntil(this.destroy$))
         .subscribe({
           next: (link) => {

frontend/app/components/stack_trace_snippet/stack_trace_snippet.ng.html

@@ -5,7 +5,8 @@
       *ngFor="let addr of sourceCodeSnippetAddresses; let isFirst = first; trackBy: trackByIndex"
       [sourceCodeSnippetAddress]="addr"
       [topOfStack]="isFirst"
-      [usingSourceFileAndLineNumber]="usingSourceFileAndLineNumber">
+      [usingSourceFileAndLineNumber]="usingSourceFileAndLineNumber"
+      [srcPathPrefix]="srcPathPrefix">
     </stack-frame-snippet>
   </ng-container>
 

frontend/app/components/stack_trace_snippet/stack_trace_snippet.ts

@@ -31,6 +31,14 @@ export class StackTraceSnippet implements OnChanges {
    * The number of lines to show around the stack frame.
    */
   @Input() sourceContextWindow = 40;
+  /**
+   * The prefix of the source file path.
+   *
+   * This is used to construct the src link given different repositories that
+   * serves the source code.
+   */
+  @Input() srcPathPrefix = '';
+
   sourceCodeSnippetAddresses: readonly Address[] = [];
 
   ngOnChanges(changes: SimpleChanges) {

frontend/app/services/data_service_v2/data_service_v2.ts

@@ -4,13 +4,13 @@ import {Injectable} from '@angular/core';
 import {Store} from '@ngrx/store';
 import {API_PREFIX, CAPTURE_PROFILE_API, CONFIG_API, DATA_API, GRAPH_TYPE_DEFAULT, GRAPHVIZ_PAN_ZOOM_CONTROL, HLO_MODULE_LIST_API, HOSTS_API, LOCAL_URL, PLUGIN_NAME, RUN_TOOLS_API, RUNS_API, USE_SAVED_RESULT} from 'org_xprof/frontend/app/common/constants/constants';
 import {FileExtensionType} from 'org_xprof/frontend/app/common/constants/enums';
-import {CaptureProfileOptions, CaptureProfileResponse} from 'org_xprof/frontend/app/common/interfaces/capture_profile';
+import {CaptureProfileOptions, CaptureProfileResponse, ProfilerConfig, } from 'org_xprof/frontend/app/common/interfaces/capture_profile';
 import {DataTable} from 'org_xprof/frontend/app/common/interfaces/data_table';
 import {HostMetadata} from 'org_xprof/frontend/app/common/interfaces/hosts';
 import {type SmartSuggestionReport} from 'org_xprof/frontend/app/common/interfaces/smart_suggestion.jsonpb_decls';
 import * as utils from 'org_xprof/frontend/app/common/utils/utils';
 import {OpProfileData, OpProfileSummary} from 'org_xprof/frontend/app/components/op_profile/op_profile_data';
-import {DataServiceV2Interface, ProfilerConfig} from 'org_xprof/frontend/app/services/data_service_v2/data_service_v2_interface';
+import {DataServiceV2Interface} from 'org_xprof/frontend/app/services/data_service_v2/data_service_v2_interface';
 import {setErrorMessageStateAction} from 'org_xprof/frontend/app/store/actions';
 import {Observable, of} from 'rxjs';
 import {catchError} from 'rxjs/operators';
@@ -307,14 +307,17 @@ export class DataServiceV2 implements DataServiceV2Interface {
     // Host is not specified for hlo text view now, as we assume metadata the
     // same across all hosts.
     const host = '';
-    const params = this.getHttpParams('', '')
-                       .set('run', sessionId)
-                       .set('tag', tool)
-                       .set('host', host)
-                       .set('graph_type', graphType)
-                       .set('module_name', moduleName)
-                       .set('type', type)
-                       .set('show_metadata', String(showMetadata));
+    let params = this.getHttpParams('', '')
+                     .set('run', sessionId)
+                     .set('tag', tool)
+                     .set('host', host)
+                     .set('graph_type', graphType)
+                     .set('module_name', moduleName)
+                     .set('type', type)
+                     .set('show_metadata', String(showMetadata));
+    if (programId) {
+      params = params.set('program_id', String(programId));
+    }
     return this.get(this.pathPrefix + DATA_API, {
       'params': params,
       'responseType': responseType,

frontend/app/services/data_service_v2/data_service_v2_interface.ts

@@ -4,17 +4,13 @@
  */
 
 import {InjectionToken} from '@angular/core';
+import {ProfilerConfig} from 'org_xprof/frontend/app/common/interfaces/capture_profile';
 import {DataTable} from 'org_xprof/frontend/app/common/interfaces/data_table';
 import {GraphTypeObject} from 'org_xprof/frontend/app/common/interfaces/graph_viewer';
 import {HostMetadata} from 'org_xprof/frontend/app/common/interfaces/hosts';
+import {type SmartSuggestionReport} from 'org_xprof/frontend/app/common/interfaces/smart_suggestion.jsonpb_decls';
 import {OpProfileData, OpProfileSummary} from 'org_xprof/frontend/app/components/op_profile/op_profile_data';
 import {Observable} from 'rxjs';
-import {type SmartSuggestionReport} from 'org_xprof/frontend/app/common/interfaces/smart_suggestion.jsonpb_decls';
-
-/** A serializable object with profiler configuration details. */
-export interface ProfilerConfig {
-  hideCaptureProfileButton: boolean;
-}
 
 /** The data service class that calls API and return response. */
 export interface DataServiceV2Interface {