@h-rig/core 0.0.6-alpha.0

package/dist/src/taskGraphLayout.js

@@ -0,0 +1,374 @@
+// @bun
+// packages/core/src/rigSelectors.ts
+function selectRunsByTask(snapshot, taskId) {
+  if (!taskId)
+    return [];
+  return snapshot?.runs.filter((run) => run.taskId === taskId) ?? [];
+}
+function selectPendingApprovals(snapshot) {
+  return snapshot?.approvals.filter((approval) => approval.status === "pending") ?? [];
+}
+function selectUserInputsForRun(snapshot, runId) {
+  if (!runId)
+    return [];
+  return (snapshot?.userInputs ?? []).filter((request) => request.runId === runId);
+}
+
+// packages/core/src/taskGraphCodes.ts
+var TASK_CODE_RE = /^\[([A-Z0-9]+(?:-[A-Z0-9]+)*)\]\s*/;
+function extractTaskCode(title) {
+  const match = title.match(TASK_CODE_RE);
+  return match?.[1] ?? null;
+}
+function extractTaskGroupKey(title) {
+  const code = extractTaskCode(title);
+  if (!code)
+    return null;
+  const parts = code.split("-");
+  const suffix = parts.at(-1) ?? "";
+  if (/^\d+$/.test(suffix)) {
+    return parts.slice(0, -1).join("-");
+  }
+  return parts[0] ?? code;
+}
+function stripTaskCode(label) {
+  return label.replace(TASK_CODE_RE, "");
+}
+
+// packages/core/src/taskGraph.ts
+function isObjectRecord(value) {
+  return typeof value === "object" && value !== null && !Array.isArray(value);
+}
+function readTaskMetadataStringList(task, key) {
+  const metadata = isObjectRecord(task.metadata) ? task.metadata : null;
+  const value = metadata?.[key];
+  return Array.isArray(value) ? value.filter((entry) => typeof entry === "string" && entry.length > 0) : [];
+}
+function readTaskSourceIssueId(task) {
+  const metadata = isObjectRecord(task.metadata) ? task.metadata : null;
+  return typeof metadata?.sourceIssueId === "string" && metadata.sourceIssueId.length > 0 ? metadata.sourceIssueId : null;
+}
+function resolveTaskReference(ref, tasksById, taskIdByExternalRef, taskIdBySourceIssueId) {
+  if (tasksById.has(ref))
+    return ref;
+  return taskIdBySourceIssueId.get(ref) ?? taskIdByExternalRef.get(ref) ?? null;
+}
+function buildTaskReferenceIndex(tasks) {
+  return {
+    tasksById: new Map(tasks.map((task) => [task.id, task])),
+    taskIdByExternalRef: new Map(tasks.flatMap((task) => task.externalId ? [[task.externalId, task.id]] : [])),
+    taskIdBySourceIssueId: new Map(tasks.flatMap((task) => {
+      const sourceIssueId = readTaskSourceIssueId(task);
+      return sourceIssueId ? [[sourceIssueId, task.id]] : [];
+    }))
+  };
+}
+
+// packages/core/src/taskGraphLayout.ts
+var CARD_WIDTH = 200;
+var CARD_HEIGHT = 110;
+var CELL_V_PAD = 12;
+var CELL_H_PAD = 12;
+var ROW_GAP = 28;
+var COL_GAP = 40;
+var LANE_LABEL_W = 120;
+var STAGE_HDR_H = 32;
+var PALETTE = [
+  { bg: "#3a2d12", border: "#8d6b19", edge: "#d6a11d" },
+  { bg: "#102642", border: "#245fbf", edge: "#66a2ff" },
+  { bg: "#2c173f", border: "#7b39d4", edge: "#a76df5" },
+  { bg: "#112d1c", border: "#2d8f4e", edge: "#62d882" },
+  { bg: "#3a2314", border: "#c86d1c", edge: "#e69654" },
+  { bg: "#31152b", border: "#bf3d88", edge: "#f07ebb" },
+  { bg: "#132c35", border: "#1783a6", edge: "#53c4e5" },
+  { bg: "#26310f", border: "#6d9a19", edge: "#a7da42" }
+];
+function isObjectRecord2(value) {
+  return typeof value === "object" && value !== null && !Array.isArray(value);
+}
+function readIssueType(task) {
+  const metadata = isObjectRecord2(task.metadata) ? task.metadata : null;
+  return typeof metadata?.issueType === "string" ? metadata.issueType : null;
+}
+function isGraphTask(task) {
+  return readIssueType(task) !== "epic";
+}
+function findEpicAncestor(task, resolve, tasksById) {
+  const visited = new Set;
+  let current = task;
+  let epic = null;
+  while (!visited.has(current.id)) {
+    visited.add(current.id);
+    const parentRef = readTaskMetadataStringList(current, "parentChildDeps")[0];
+    if (!parentRef)
+      break;
+    const parentId = resolve(parentRef);
+    if (!parentId)
+      break;
+    const parent = tasksById.get(parentId);
+    if (!parent)
+      break;
+    if (readIssueType(parent) === "epic") {
+      epic = parent;
+    }
+    current = parent;
+  }
+  return epic;
+}
+function getRowKey(task, resolve, tasksById) {
+  const epic = findEpicAncestor(task, resolve, tasksById);
+  if (epic) {
+    return `group:${epic.id}`;
+  }
+  const codeGroup = extractTaskGroupKey(task.title);
+  if (codeGroup) {
+    return `code:${codeGroup}`;
+  }
+  return `type:${readIssueType(task) ?? "task"}`;
+}
+function getRowLabel(task, rowKey, resolve, tasksById) {
+  if (rowKey.startsWith("group:")) {
+    const groupId = rowKey.slice("group:".length);
+    return tasksById.get(groupId)?.title ?? groupId;
+  }
+  if (rowKey.startsWith("code:")) {
+    return rowKey.slice("code:".length);
+  }
+  const code = extractTaskCode(task.title);
+  if (code)
+    return code;
+  const issueType = readIssueType(task);
+  if (issueType === "task")
+    return "Tasks";
+  if (issueType)
+    return `${issueType[0]?.toUpperCase() ?? ""}${issueType.slice(1)}`;
+  const parentRef = readTaskMetadataStringList(task, "parentChildDeps")[0];
+  if (parentRef) {
+    const parentId = resolve(parentRef);
+    if (parentId) {
+      return tasksById.get(parentId)?.title ?? "Tasks";
+    }
+  }
+  return "Tasks";
+}
+function computeDepths(ids, edges) {
+  const blockers = new Map;
+  for (const edge of edges) {
+    if (!ids.has(edge.source) || !ids.has(edge.target))
+      continue;
+    const current = blockers.get(edge.target) ?? [];
+    current.push(edge.source);
+    blockers.set(edge.target, current);
+  }
+  const memo = new Map;
+  const visit = (id, stack) => {
+    const cached = memo.get(id);
+    if (cached !== undefined)
+      return cached;
+    if (stack.has(id))
+      return 0;
+    stack.add(id);
+    const deps = blockers.get(id);
+    if (!deps || deps.length === 0) {
+      memo.set(id, 0);
+      return 0;
+    }
+    let maxDepth = 0;
+    for (const dep of deps) {
+      maxDepth = Math.max(maxDepth, visit(dep, stack) + 1);
+    }
+    stack.delete(id);
+    memo.set(id, maxDepth);
+    return maxDepth;
+  };
+  for (const id of ids) {
+    visit(id, new Set);
+  }
+  return memo;
+}
+function buildTaskGraphLayout(snapshot, tasks, options) {
+  const showParentChild = options?.showParentChild ?? false;
+  const graphTasks = tasks.filter(isGraphTask);
+  if (graphTasks.length === 0) {
+    return {
+      lanes: [],
+      stages: [],
+      nodes: [],
+      edges: [],
+      totalWidth: 0,
+      totalHeight: 0,
+      taskCount: 0
+    };
+  }
+  const { tasksById, taskIdByExternalRef, taskIdBySourceIssueId } = buildTaskReferenceIndex(tasks);
+  const resolve = (ref) => resolveTaskReference(ref, tasksById, taskIdByExternalRef, taskIdBySourceIssueId);
+  const rows = new Map;
+  const rowLabelByKey = new Map;
+  for (const task of graphTasks) {
+    const rowKey = getRowKey(task, resolve, tasksById);
+    const current = rows.get(rowKey) ?? [];
+    current.push(task);
+    rows.set(rowKey, current);
+    if (!rowLabelByKey.has(rowKey)) {
+      rowLabelByKey.set(rowKey, getRowLabel(task, rowKey, resolve, tasksById));
+    }
+  }
+  const orderedRows = [...rows.entries()].sort((left, right) => {
+    if (left[1].length !== right[1].length)
+      return right[1].length - left[1].length;
+    return (rowLabelByKey.get(left[0]) ?? left[0]).localeCompare(rowLabelByKey.get(right[0]) ?? right[0]);
+  });
+  const rowIndexByKey = new Map(orderedRows.map(([key], index) => [key, index]));
+  const rowIndexByTaskId = new Map;
+  for (const task of graphTasks) {
+    rowIndexByTaskId.set(task.id, rowIndexByKey.get(getRowKey(task, resolve, tasksById)) ?? 0);
+  }
+  const blockingEdgesRaw = [];
+  const depsIn = new Map;
+  const depsOut = new Map;
+  const edges = [];
+  for (const task of graphTasks) {
+    for (const ref of readTaskMetadataStringList(task, "dependencies")) {
+      const sourceId = resolve(ref);
+      if (!sourceId)
+        continue;
+      blockingEdgesRaw.push({ source: sourceId, target: task.id });
+      depsOut.set(sourceId, (depsOut.get(sourceId) ?? 0) + 1);
+      depsIn.set(task.id, (depsIn.get(task.id) ?? 0) + 1);
+      const color = PALETTE[(rowIndexByTaskId.get(sourceId) ?? 0) % PALETTE.length].edge;
+      edges.push({
+        id: `blocking:${sourceId}:${task.id}`,
+        sourceId,
+        targetId: task.id,
+        color,
+        kind: "blocking"
+      });
+    }
+    if (!showParentChild)
+      continue;
+    for (const ref of readTaskMetadataStringList(task, "parentChildDeps")) {
+      const sourceId = resolve(ref);
+      if (!sourceId)
+        continue;
+      edges.push({
+        id: `parent:${sourceId}:${task.id}`,
+        sourceId,
+        targetId: task.id,
+        color: "#5b6472",
+        kind: "parent-child"
+      });
+    }
+  }
+  const graphTaskIds = new Set(graphTasks.map((task) => task.id));
+  const depths = computeDepths(graphTaskIds, blockingEdgesRaw);
+  const maxStage = Math.max(0, ...depths.values());
+  const rowMaxStack = new Array(orderedRows.length).fill(0);
+  const cells = new Map;
+  for (const task of graphTasks) {
+    const rowIndex = rowIndexByTaskId.get(task.id) ?? 0;
+    const stage = depths.get(task.id) ?? 0;
+    const key = `${rowIndex}:${stage}`;
+    const current = cells.get(key) ?? [];
+    current.push(task);
+    cells.set(key, current);
+  }
+  for (const [cellKey, cellTasks] of cells) {
+    const [rowIndexText] = cellKey.split(":");
+    const rowIndex = Number.parseInt(rowIndexText ?? "0", 10) || 0;
+    cellTasks.sort((left, right) => {
+      const leftFanout = depsOut.get(left.id) ?? 0;
+      const rightFanout = depsOut.get(right.id) ?? 0;
+      if (leftFanout !== rightFanout)
+        return rightFanout - leftFanout;
+      return left.title.localeCompare(right.title);
+    });
+    rowMaxStack[rowIndex] = Math.max(rowMaxStack[rowIndex] ?? 0, cellTasks.length);
+  }
+  const rowHeights = rowMaxStack.map((count) => Math.max(count, 1) * (CARD_HEIGHT + CELL_V_PAD) - CELL_V_PAD + CELL_V_PAD * 2);
+  const colWidths = new Array(maxStage + 1).fill(CARD_WIDTH + CELL_H_PAD * 2);
+  const colX = [];
+  let currentX = LANE_LABEL_W;
+  for (let index = 0;index <= maxStage; index += 1) {
+    colX.push(currentX);
+    currentX += (colWidths[index] ?? 0) + COL_GAP;
+  }
+  const totalWidth = currentX - COL_GAP;
+  const rowY = [];
+  let currentY = STAGE_HDR_H;
+  for (let rowIndex = 0;rowIndex < orderedRows.length; rowIndex += 1) {
+    rowY.push(currentY);
+    currentY += (rowHeights[rowIndex] ?? 0) + ROW_GAP;
+  }
+  const totalHeight = currentY - ROW_GAP;
+  const lanes = orderedRows.map(([rowKey, rowTasks], rowIndex) => ({
+    key: rowKey,
+    label: rowLabelByKey.get(rowKey) ?? rowKey,
+    rowIndex,
+    x: LANE_LABEL_W - 6,
+    y: (rowY[rowIndex] ?? 0) - 6,
+    width: totalWidth - LANE_LABEL_W + 12,
+    height: (rowHeights[rowIndex] ?? 0) + 12,
+    color: PALETTE[rowIndex % PALETTE.length].border,
+    taskCount: rowTasks.length
+  }));
+  const stages = Array.from({ length: maxStage + 1 }, (_, index) => ({
+    index,
+    label: index === 0 ? "Roots" : `Stage ${index}`,
+    x: (colX[index] ?? 0) + CELL_H_PAD,
+    width: CARD_WIDTH
+  }));
+  const pendingApprovalRunIds = new Set(selectPendingApprovals(snapshot).map((approval) => approval.runId));
+  const nodes = [];
+  for (const [rowIndex, [rowKey]] of orderedRows.entries()) {
+    for (let stage = 0;stage <= maxStage; stage += 1) {
+      const cellTasks = cells.get(`${rowIndex}:${stage}`) ?? [];
+      const baseX = (colX[stage] ?? 0) + CELL_H_PAD;
+      const baseY = (rowY[rowIndex] ?? 0) + CELL_V_PAD;
+      const palette = PALETTE[rowIndex % PALETTE.length];
+      for (const [stackIndex, task] of cellTasks.entries()) {
+        const runs = selectRunsByTask(snapshot, task.id);
+        const runIds = new Set(runs.map((run) => run.id));
+        const hasApprovals = runs.some((run) => pendingApprovalRunIds.has(run.id));
+        const hasPendingUserInput = runs.some((run) => selectUserInputsForRun(snapshot, run.id).some((request) => request.status === "pending"));
+        const hasRejectedReview = (snapshot?.reviews ?? []).some((review) => runIds.has(review.runId) && review.status === "rejected");
+        const hasFailedValidations = (snapshot?.validations ?? []).some((validation) => runIds.has(validation.runId) && validation.status === "failed");
+        const artifactCount = (snapshot?.artifacts ?? []).filter((artifact) => runIds.has(artifact.runId)).length;
+        nodes.push({
+          id: task.id,
+          task,
+          rowKey,
+          rowLabel: rowLabelByKey.get(rowKey) ?? rowKey,
+          rowIndex,
+          stage,
+          x: baseX,
+          y: baseY + stackIndex * (CARD_HEIGHT + CELL_V_PAD),
+          width: CARD_WIDTH,
+          height: CARD_HEIGHT,
+          color: palette.border,
+          taskCode: extractTaskCode(task.title),
+          strippedTitle: stripTaskCode(task.title),
+          depsIn: depsIn.get(task.id) ?? 0,
+          depsOut: depsOut.get(task.id) ?? 0,
+          runCount: runs.length,
+          hasApprovals,
+          hasPendingUserInput,
+          hasRejectedReview,
+          hasFailedValidations,
+          artifactCount
+        });
+      }
+    }
+  }
+  return {
+    lanes,
+    stages,
+    nodes,
+    edges,
+    totalWidth,
+    totalHeight,
+    taskCount: graphTasks.length
+  };
+}
+export {
+  buildTaskGraphLayout
+};

package/package.json

@@ -0,0 +1,37 @@
+{
+  "name": "@h-rig/core",
+  "version": "0.0.6-alpha.0",
+  "type": "module",
+  "description": "Rig package",
+  "license": "UNLICENSED",
+  "files": [
+    "dist",
+    "README.md"
+  ],
+  "exports": {
+    ".": {
+      "import": "./dist/src/index.js"
+    },
+    "./load-config": {
+      "import": "./dist/src/load-config.js"
+    },
+    "./engineReadModelReducer": {
+      "import": "./dist/src/engineReadModelReducer.js"
+    },
+    "./rigSelectors": {
+      "import": "./dist/src/rigSelectors.js"
+    },
+    "./taskGraph": {
+      "import": "./dist/src/taskGraph.js"
+    }
+  },
+  "engines": {
+    "bun": ">=1.3.11"
+  },
+  "main": "./dist/src/index.js",
+  "module": "./dist/src/index.js",
+  "dependencies": {
+    "@rig/contracts": "npm:@h-rig/contracts@0.0.6-alpha.0",
+    "effect": "4.0.0-beta.78"
+  }
+}