@bretwardjames/tw-bridge 0.2.0 → 0.4.0

package/dist/extensions/bridge.js

@@ -0,0 +1,219 @@
+#!/usr/bin/env node
+
+// src/extensions/bridge.ts
+import fs from "fs";
+import { spawnSync } from "child_process";
+function parseInput() {
+  const raw = fs.readFileSync("/dev/stdin", "utf-8");
+  const lines = raw.split("\n");
+  let i = 0;
+  while (i < lines.length && lines[i].trim() !== "") {
+    i++;
+  }
+  i++;
+  const json = lines.slice(i).join("\n").trim();
+  if (!json) return [];
+  try {
+    return JSON.parse(json);
+  } catch {
+    return [];
+  }
+}
+function parseTimewDate(s) {
+  return new Date(
+    s.replace(/(\d{4})(\d{2})(\d{2})T(\d{2})(\d{2})(\d{2})Z/, "$1-$2-$3T$4:$5:$6Z")
+  );
+}
+function intervalDuration(iv) {
+  const start = parseTimewDate(iv.start).getTime();
+  const end = iv.end ? parseTimewDate(iv.end).getTime() : Date.now();
+  return end - start;
+}
+function formatDuration(ms) {
+  const totalSecs = Math.round(ms / 1e3);
+  const h = Math.floor(totalSecs / 3600);
+  const m = Math.floor(totalSecs % 3600 / 60);
+  const s = totalSecs % 60;
+  return `${h}:${String(m).padStart(2, "0")}:${String(s).padStart(2, "0")}`;
+}
+function formatTime(d) {
+  return `${String(d.getHours()).padStart(2, "0")}:${String(d.getMinutes()).padStart(2, "0")}`;
+}
+function lookupDescriptions(taskIds) {
+  const descriptions = /* @__PURE__ */ new Map();
+  if (taskIds.length === 0) return descriptions;
+  const filter = taskIds.map((id) => `backend_id:${id}`).join(" or ");
+  const result = spawnSync("task", ["rc.verbose=nothing", filter, "export"], {
+    encoding: "utf-8",
+    stdio: ["pipe", "pipe", "pipe"]
+  });
+  if (result.status !== 0 || !result.stdout?.trim()) return descriptions;
+  try {
+    const tasks = JSON.parse(result.stdout);
+    for (const t of tasks) {
+      if (t.backend_id) {
+        descriptions.set(t.backend_id, t.description ?? "");
+      }
+    }
+  } catch {
+  }
+  return descriptions;
+}
+function computeTaskTime(intervals) {
+  const taskMap = /* @__PURE__ */ new Map();
+  for (const iv of intervals) {
+    const duration = intervalDuration(iv);
+    const idTag = iv.tags?.find((t) => t.startsWith("#"));
+    const id = idTag?.slice(1) ?? "";
+    const key = id || iv.tags?.join(" ") || "untagged";
+    const project = iv.tags?.find((t) => !t.startsWith("#")) ?? "";
+    const existing = taskMap.get(key);
+    if (existing) {
+      existing.duration += duration;
+    } else {
+      taskMap.set(key, { project, duration });
+    }
+  }
+  const ids = [...taskMap.keys()].filter((k) => k !== "untagged" && /^\d+$/.test(k));
+  const descriptions = lookupDescriptions(ids);
+  const tasks = [];
+  for (const [id, info] of taskMap) {
+    tasks.push({
+      id,
+      project: info.project,
+      description: descriptions.get(id) ?? "",
+      duration: info.duration
+    });
+  }
+  return tasks.sort((a, b) => b.duration - a.duration);
+}
+function computeWallTime(intervals) {
+  if (intervals.length === 0) return 0;
+  const events = [];
+  for (const iv of intervals) {
+    const start = parseTimewDate(iv.start).getTime();
+    const end = iv.end ? parseTimewDate(iv.end).getTime() : Date.now();
+    events.push({ time: start, type: "start" });
+    events.push({ time: end, type: "end" });
+  }
+  events.sort((a, b) => a.time - b.time || (a.type === "end" ? -1 : 1));
+  let wallTime = 0;
+  let depth = 0;
+  let segmentStart = 0;
+  for (const event of events) {
+    if (event.type === "start") {
+      if (depth === 0) segmentStart = event.time;
+      depth++;
+    } else {
+      depth--;
+      if (depth === 0) {
+        wallTime += event.time - segmentStart;
+      }
+    }
+  }
+  return wallTime;
+}
+function computeTimeline(intervals) {
+  if (intervals.length === 0) return [];
+  const points = /* @__PURE__ */ new Set();
+  for (const iv of intervals) {
+    points.add(parseTimewDate(iv.start).getTime());
+    points.add(iv.end ? parseTimewDate(iv.end).getTime() : Date.now());
+  }
+  const sorted = [...points].sort((a, b) => a - b);
+  const segments = [];
+  for (let i = 0; i < sorted.length - 1; i++) {
+    const segStart = sorted[i];
+    const segEnd = sorted[i + 1];
+    const midpoint = (segStart + segEnd) / 2;
+    const activeTags = [];
+    for (const iv of intervals) {
+      const ivStart = parseTimewDate(iv.start).getTime();
+      const ivEnd = iv.end ? parseTimewDate(iv.end).getTime() : Date.now();
+      if (ivStart <= midpoint && ivEnd > midpoint) {
+        for (const tag of iv.tags ?? []) {
+          if (!activeTags.includes(tag)) activeTags.push(tag);
+        }
+      }
+    }
+    if (activeTags.length > 0) {
+      segments.push({
+        start: new Date(segStart),
+        end: new Date(segEnd),
+        tags: activeTags
+      });
+    }
+  }
+  return segments;
+}
+var BOLD = "\x1B[1m";
+var DIM = "\x1B[2m";
+var RESET = "\x1B[0m";
+var LINE = "\u2500".repeat(60);
+function printTaskTime(tasks, totalTask, wallTime) {
+  console.log(`${BOLD}Task Time${RESET}`);
+  console.log(LINE);
+  for (const t of tasks) {
+    const desc = t.description ? `  ${t.description}` : "";
+    const idLabel = t.id ? `#${t.id}` : t.project;
+    console.log(
+      `  ${t.project.padEnd(14)} ${idLabel.padEnd(8)}${desc.padEnd(28)} ${formatDuration(t.duration).padStart(10)}`
+    );
+  }
+  console.log(LINE);
+  console.log(`  ${"Total task time:".padEnd(50)} ${BOLD}${formatDuration(totalTask).padStart(10)}${RESET}`);
+  const overlap = totalTask - wallTime;
+  if (overlap > 1e3) {
+    console.log(`  ${"Wall time:".padEnd(50)} ${formatDuration(wallTime).padStart(10)}`);
+    console.log(`  ${DIM}${"Overlap:".padEnd(50)} ${formatDuration(overlap).padStart(10)}${RESET}`);
+  }
+}
+function printWallTime(intervals) {
+  const timeline = computeTimeline(intervals);
+  const wallTime = computeWallTime(intervals);
+  console.log(`${BOLD}Wall Time${RESET}`);
+  console.log(LINE);
+  for (const seg of timeline) {
+    const start = formatTime(seg.start);
+    const end = formatTime(seg.end);
+    const tags = seg.tags.join(", ");
+    const duration = seg.end.getTime() - seg.start.getTime();
+    console.log(
+      `  ${start}\u2013${end}  ${tags.padEnd(36)} ${DIM}${formatDuration(duration).padStart(10)}${RESET}`
+    );
+  }
+  console.log(LINE);
+  console.log(`  ${"Total wall time:".padEnd(50)} ${BOLD}${formatDuration(wallTime).padStart(10)}${RESET}`);
+}
+function main() {
+  const args = process.argv.slice(2);
+  const intervals = parseInput();
+  let mode = "both";
+  const filters = [];
+  for (const arg of args) {
+    if (arg === "task-time") mode = "task-time";
+    else if (arg === "wall-time") mode = "wall-time";
+    else if (!arg.startsWith(":")) filters.push(arg);
+  }
+  let filtered = intervals;
+  if (filters.length > 0) {
+    filtered = intervals.filter(
+      (iv) => filters.some((f) => iv.tags?.includes(f))
+    );
+  }
+  if (filtered.length === 0) {
+    console.log("No intervals found.");
+    return;
+  }
+  if (mode === "both" || mode === "task-time") {
+    const tasks = computeTaskTime(filtered);
+    const totalTask = tasks.reduce((a, b) => a + b.duration, 0);
+    const wallTime = computeWallTime(filtered);
+    printTaskTime(tasks, totalTask, wallTime);
+  }
+  if (mode === "both") console.log("");
+  if (mode === "both" || mode === "wall-time") {
+    printWallTime(filtered);
+  }
+}
+main();