From 4e735cc8b03d87f850286461474d8f110580e444 Mon Sep 17 00:00:00 2001
From: Mononaut <github@monospace.live>
Date: Mon, 20 Jan 2025 07:30:27 +0000
Subject: [PATCH] fix stratum tree rendering with different branch lengths

---
 .../stratum-list/stratum-list.component.ts    | 28 ++++++++++++++-----
 1 file changed, 21 insertions(+), 7 deletions(-)

diff --git a/frontend/src/app/components/stratum/stratum-list/stratum-list.component.ts b/frontend/src/app/components/stratum/stratum-list/stratum-list.component.ts
index 0af9f0976..6f252babe 100644
--- a/frontend/src/app/components/stratum/stratum-list/stratum-list.component.ts
+++ b/frontend/src/app/components/stratum/stratum-list/stratum-list.component.ts
@@ -8,8 +8,10 @@ import { SinglePoolStats } from '../../../interfaces/node-api.interface';
 
 type MerkleCellType = ' ' | '┬' | '├' | '└' | '│' | '─' | 'leaf';
 
+
 interface TaggedStratumJob extends StratumJob {
   tag: string;
+  merkleBranchIds: string[];
 }
 
 interface MerkleCell {
@@ -46,6 +48,18 @@ function parseTag(scriptSig: string): string {
   return (ascii.match(/\/.*\//)?.[0] || ascii).trim();
 }
 
+function getMerkleBranchIds(merkleBranches: string[], numBranches: number): string[] {
+  let lastHash = '';
+  const ids: string[] = [];
+  for (let i = 0; i < numBranches; i++) {
+    if (merkleBranches[i]) {
+      lastHash = merkleBranches[i];
+    }
+    ids.push(`${i}-${lastHash}`);
+  }
+  return ids;
+}
+
 @Component({
   selector: 'app-stratum-list',
   templateUrl: './stratum-list.component.html',
@@ -81,16 +95,15 @@ export class StratumList implements OnInit, OnDestroy {
   }
 
   processJobs(rawJobs: Record<string, StratumJob>): PoolRow[] {
+    const numBranches = Math.max(...Object.values(rawJobs).map(job => job.merkleBranches.length));
     const jobs: Record<string, TaggedStratumJob> = {};
     for (const [id, job] of Object.entries(rawJobs)) {
-      jobs[id] = { ...job, tag: parseTag(job.scriptsig) };
+      jobs[id] = { ...job, tag: parseTag(job.scriptsig), merkleBranchIds: getMerkleBranchIds(job.merkleBranches, numBranches) };
     }
     if (Object.keys(jobs).length === 0) {
       return [];
     }
 
-    const numBranches = Math.max(...Object.values(jobs).map(job => job.merkleBranches.length));
-
     let trees: MerkleTree[] = Object.keys(jobs).map(job => ({
       job,
       size: 1,
@@ -100,12 +113,13 @@ export class StratumList implements OnInit, OnDestroy {
     for (let col = numBranches - 1; col >= 0; col--) {
       const groups: Record<string, MerkleTree[]> = {};
       for (const tree of trees) {
-        const hash = jobs[tree.job].merkleBranches[col];
-        if (!groups[hash]) {
-          groups[hash] = [];
+        const branchId = jobs[tree.job].merkleBranchIds[col];
+        if (!groups[branchId]) {
+          groups[branchId] = [];
         }
-        groups[hash].push(tree);
+        groups[branchId].push(tree);
       }
+
       trees = Object.values(groups).map(group => ({
         hash: jobs[group[0].job].merkleBranches[col],
         job: group[0].job,