[rush-sdk] Expose file change analysis on the ProjectChangeAnalyzer class

common/reviews/api/rush-lib.api.md

@@ -352,6 +352,14 @@ export interface ICloudBuildCacheProvider {
     updateCachedCredentialInteractiveAsync(terminal: ITerminal): Promise<void>;
 }
 
+// @beta (undocumented)
+export interface IClusterFileInfoOptions<TItem, TGroup> {
+    // (undocumented)
+    infoByPath: ReadonlyMap<string, TItem>;
+    // (undocumented)
+    lookup: LookupByPath<TGroup>;
+}
+
 // @beta (undocumented)
 export interface ICobuildCompletedState {
     cacheId: string;
@@ -517,14 +525,6 @@ export interface IGetChangedProjectsOptions {
     variant?: string;
 }
 
-// @beta (undocumented)
-export interface IGetChangesByProjectOptions {
-    // (undocumented)
-    changedFiles: Map<string, IFileDiffStatus>;
-    // (undocumented)
-    lookup: LookupByPath<RushConfigurationProject>;
-}
-
 // @beta
 export interface IGlobalCommand extends IRushCommand {
 }
@@ -1126,7 +1126,7 @@ export class ProjectChangeAnalyzer {
     _filterProjectDataAsync<T>(project: RushConfigurationProject, unfilteredProjectData: Map<string, T>, rootDir: string, terminal: ITerminal): Promise<Map<string, T>>;
     getChangedProjectsAsync(options: IGetChangedProjectsOptions): Promise<Set<RushConfigurationProject>>;
     // (undocumented)
-    protected getChangesByProject(options: IGetChangesByProjectOptions): Map<RushConfigurationProject, Map<string, IFileDiffStatus>>;
+    protected getChangesByProject(options: IClusterFileInfoOptions<IFileDiffStatus, RushConfigurationProject>): Map<RushConfigurationProject, Map<string, IFileDiffStatus>>;
     // @internal
     _tryGetProjectDependenciesAsync(project: RushConfigurationProject, terminal: ITerminal): Promise<Map<string, string> | undefined>;
     // @internal

libraries/rush-lib/src/index.ts

@@ -123,7 +123,7 @@ export {
 
 export {
   ProjectChangeAnalyzer,
-  type IGetChangesByProjectOptions,
+  type IClusterFileInfoOptions,
   type IGetChangedProjectsOptions,
   type IRawRepoState as _IRawRepoState
 } from './logic/ProjectChangeAnalyzer';

libraries/rush-lib/src/logic/ProjectChangeAnalyzer.ts

@@ -49,9 +49,31 @@ export interface IGetChangedProjectsOptions {
 /**
  * @beta
  */
-export interface IGetChangesByProjectOptions {
-  lookup: LookupByPath<RushConfigurationProject>;
-  changedFiles: Map<string, IFileDiffStatus>;
+export interface IClusterFileInfoOptions<TItem, TGroup> {
+  lookup: LookupByPath<TGroup>;
+  infoByPath: ReadonlyMap<string, TItem>;
+}
+
+function clusterFileInfo<TItem, TGroup>(
+  options: IClusterFileInfoOptions<TItem, TGroup>
+): Map<TGroup, Map<string, TItem>> {
+  const { lookup, infoByPath } = options;
+  const clusteredFileInfo: Map<TGroup, Map<string, TItem>> = new Map();
+
+  for (const [file, diffStatus] of infoByPath) {
+    const group: TGroup | undefined = lookup.findChildPath(file);
+    if (!group) {
+      continue;
+    }
+    let groupInfo: Map<string, TItem> | undefined = clusteredFileInfo.get(group);
+    if (!groupInfo) {
+      groupInfo = new Map();
+      clusteredFileInfo.set(group, groupInfo);
+    }
+    groupInfo.set(file, diffStatus);
+  }
+
+  return clusteredFileInfo;
 }
 
 interface IGitState {
@@ -244,7 +266,7 @@ export class ProjectChangeAnalyzer {
     const changesByProject: Map<
       RushConfigurationProject,
       Map<string, IFileDiffStatus>
-    > = this.getChangesByProject({ changedFiles, lookup });
+    > = this.getChangesByProject({ infoByPath: changedFiles, lookup });
 
     const changedProjects: Set<RushConfigurationProject> = new Set();
     if (enableFiltering) {
@@ -333,25 +355,9 @@ export class ProjectChangeAnalyzer {
   }
 
   protected getChangesByProject(
-    options: IGetChangesByProjectOptions
+    options: IClusterFileInfoOptions<IFileDiffStatus, RushConfigurationProject>
   ): Map<RushConfigurationProject, Map<string, IFileDiffStatus>> {
-    const { lookup, changedFiles } = options;
-    const changesByProject: Map<RushConfigurationProject, Map<string, IFileDiffStatus>> = new Map();
-
-    for (const [file, diffStatus] of changedFiles) {
-      const project: RushConfigurationProject | undefined = lookup.findChildPath(file);
-      if (!project) {
-        continue;
-      }
-      let projectChanges: Map<string, IFileDiffStatus> | undefined = changesByProject.get(project);
-      if (!projectChanges) {
-        projectChanges = new Map();
-        changesByProject.set(project, projectChanges);
-      }
-      projectChanges.set(file, diffStatus);
-    }
-
-    return changesByProject;
+    return clusterFileInfo(options);
   }
 
   private async _getDataAsync(terminal: ITerminal): Promise<IRawRepoState> {