feat : csm pagination view

packages/vscode/src/extension.ts

@@ -80,10 +80,11 @@ export async function activate(context: vscode.ExtensionContext) {
       const fetchClusterNodes = async (
         baseBranchName = initialBaseBranchName,
         perPage?: number,
-        lastCommitId?: string
+        lastCommitId?: string,
+        command?: string
       ): Promise<ClusterNodesResult> => {
-        // Cache engine
-        if (!engine || engine.getBaseBranchName() !== baseBranchName) {
+        // Initialize engine if it doesn't exist or if the branch has changed
+        if (command === "refresh" || !engine || engine.getBaseBranchName() !== baseBranchName) {
           const workerPath = vscode.Uri.joinPath(context.extensionUri, "dist", "worker.js").fsPath;
           const gitLog = await fetchGitLogInParallel(gitPath, currentWorkspacePath, workerPath);
           const { owner, repo } = getRepo(gitConfig);

packages/vscode/src/webview-loader.ts

@@ -17,7 +17,6 @@ export default class WebviewLoader implements vscode.Disposable {
     const { fetchClusterNodes, fetchBranches, fetchCurrentBranch, fetchGithubInfo } = fetcher;
     const viewColumn = vscode.ViewColumn.One;
 
-    console.log("Initialize cache data");
     context.workspaceState.keys().forEach((key) => {
       context.workspaceState.update(key, undefined);
     });
@@ -35,50 +34,47 @@ export default class WebviewLoader implements vscode.Disposable {
       try {
         const { command, payload } = message;
 
-        if (command === "fetchAnalyzedData" || command === "refresh") {
-          try {
-            const requestPayload = payload ? JSON.parse(payload) : {};
-            const { baseBranchName, perPage, lastCommitId } = requestPayload;
-            const currentBranch = baseBranchName ?? (await fetchCurrentBranch());
-
-            if (perPage) {
-              console.log(`Paging request: perPage=${perPage}, lastCommitId=${lastCommitId}`);
-              const clusterData = await fetchClusterNodes(currentBranch, perPage, lastCommitId);
-              analyzedData = {
-                ...clusterData,
-                isLoadMore: !!lastCommitId,
-              };
-            } else {
-              const cacheKey = `${ANALYZE_DATA_KEY}_${currentBranch}`;
-              // "refresh": ignore the cache
-              // "fetchAnalyzedData": use the cache if exists
-              const storedAnalyzedData =
-                command === "fetchAnalyzedData" ? context.workspaceState.get<ClusterNode[]>(cacheKey) : undefined;
-
-              if (storedAnalyzedData) {
-                console.log("Cache data exists");
-                analyzedData = storedAnalyzedData;
-              } else {
-                console.log(command === "refresh" ? "Forced refresh: fetching new data" : "No cache Data");
-                analyzedData = await fetchClusterNodes(currentBranch);
-                context.workspaceState.update(cacheKey, analyzedData);
-              }
-              console.log("Current Stored data");
-              context.workspaceState.keys().forEach((key) => {
-                console.log(key);
-              });
-            }
-
-            const resMessage = {
-              command,
-              payload: analyzedData,
-            };
+        if (command === "refresh") {
+          const requestPayload = payload ? JSON.parse(payload) : {};
+          const { selectedBranch, perPage, lastCommitId } = requestPayload;
+          const currentBranch = selectedBranch ?? (await fetchCurrentBranch());
+
+          const clusterData = await fetchClusterNodes(currentBranch, perPage, lastCommitId, "refresh");
+          analyzedData = {
+            ...clusterData,
+            isLoadMore: !!lastCommitId,
+          };
+
+          await this.respondToMessage({
+            command,
+            payload: analyzedData,
+          });
+        }
+
+        if (command === "fetchAnalyzedData") {
+          const requestPayload = payload ? JSON.parse(payload) : {};
+          const { baseBranch, perPage, lastCommitId } = requestPayload;
+          const currentBranch = baseBranch ?? (await fetchCurrentBranch());
 
-            await this.respondToMessage(resMessage);
-          } catch (e) {
-            console.error("Error fetching analyzed data:", e);
-            throw e;
+          const cacheKey = `${ANALYZE_DATA_KEY}_${currentBranch}_${lastCommitId || "firstPage"}`;
+
+          const storedAnalyzedData = context.workspaceState.get<ClusterNodesResult>(cacheKey);
+
+          if (storedAnalyzedData) {
+            analyzedData = storedAnalyzedData;
+          } else {
+            const clusterData = await fetchClusterNodes(currentBranch, perPage, lastCommitId);
+            analyzedData = {
+              ...clusterData,
+              isLoadMore: !!lastCommitId,
+            };
+            context.workspaceState.update(cacheKey, analyzedData);
           }
+
+          await this.respondToMessage({
+            command,
+            payload: analyzedData,
+          });
         }
 
         if (command === "fetchBranchList") {
@@ -173,7 +169,7 @@ export default class WebviewLoader implements vscode.Disposable {
 
 type GithruFetcher<D = unknown, P extends unknown[] = []> = (...params: P) => Promise<D>;
 type GithruFetcherMap = {
-  fetchClusterNodes: GithruFetcher<ClusterNodesResult, [string?, number?, string?]>;
+  fetchClusterNodes: GithruFetcher<ClusterNodesResult, [string?, number?, string?, string?]>;
   fetchBranches: GithruFetcher<{ branchList: string[]; head: string | null }>;
   fetchCurrentBranch: GithruFetcher<string>;
   fetchGithubInfo: GithruFetcher<{ owner: string; repo: string }>;