hobots 1.3.6 → 1.5.1

package/dist/index.cjs

@@ -1,7 +1,9 @@
 "use strict";
+var __create = Object.create;
 var __defProp = Object.defineProperty;
 var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
 var __getOwnPropNames = Object.getOwnPropertyNames;
+var __getProtoOf = Object.getPrototypeOf;
 var __hasOwnProp = Object.prototype.hasOwnProperty;
 var __export = (target, all) => {
   for (var name in all)
@@ -15,6 +17,14 @@ var __copyProps = (to, from, except, desc) => {
   }
   return to;
 };
+var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__getProtoOf(mod)) : {}, __copyProps(
+  // If the importer is in node compatibility mode or this is not an ESM
+  // file that has been converted to a CommonJS file using a Babel-
+  // compatible transform (i.e. "__esModule" has not been set), then set
+  // "default" to the CommonJS "module.exports" for node compatibility.
+  isNodeMode || !mod || !mod.__esModule ? __defProp(target, "default", { value: mod, enumerable: true }) : target,
+  mod
+));
 var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
 
 // src/index.ts
@@ -111,7 +121,7 @@ function parseStacktrace(stack) {
 
 // src/utils.ts
 var SDK_NAME = "hobots";
-var SDK_VERSION = true ? "1.3.5" : "0.0.0-dev";
+var SDK_VERSION = true ? "1.5.0" : "0.0.0-dev";
 function isNode() {
   return typeof process !== "undefined" && typeof process.versions?.node === "string";
 }
@@ -151,7 +161,8 @@ var ANSI = {
   gray: "\x1B[90m",
   cyan: "\x1B[36m",
   yellow: "\x1B[33m",
-  red: "\x1B[31m"
+  red: "\x1B[31m",
+  green: "\x1B[32m"
 };
 function colorEnabled() {
   if (!isNode()) return false;
@@ -591,6 +602,15 @@ var Scope = class _Scope {
 function spanId() {
   return uuid4().slice(0, 16);
 }
+function printSpanLog(kind, phase, label, startTs, endTs) {
+  const durationMs = startTs && endTs ? new Date(endTs).getTime() - new Date(startTs).getTime() : void 0;
+  const tag = colorize(`[hobots:${kind}] ${phase.toUpperCase()}`, phase === "finish" ? "green" : "cyan");
+  const suffix = durationMs != null ? ` (${durationMs}ms)` : "";
+  console.log(`${tag} ${label}${suffix}`);
+}
+function spanLabel(op, description) {
+  return [op, description].filter(Boolean).join(" \u2014 ") || "(sem nome)";
+}
 var Span = class {
   constructor(transaction, parentSpanId, op, description) {
     this.transaction = transaction;
@@ -616,7 +636,9 @@ var Span = class {
     return this.transaction.createSpan(this.spanId, op, description);
   }
   finish() {
-    if (!this.endTimestamp) this.endTimestamp = nowISO();
+    if (this.endTimestamp) return;
+    this.endTimestamp = nowISO();
+    printSpanLog("span", "finish", spanLabel(this.op, this.description), this.startTimestamp, this.endTimestamp);
   }
   toJSON() {
     return {
@@ -638,6 +660,7 @@ var Transaction = class {
     this.op = op;
     this.sampled = sampled;
     this.runId = runId;
+    printSpanLog("transaction", "start", op ? `${name} (${op})` : name);
   }
   eventId = uuid4();
   spanId = spanId();
@@ -649,6 +672,7 @@ var Transaction = class {
   createSpan(parentSpanId, op, description) {
     const span = new Span(this, parentSpanId, op, description);
     this.spans.push(span);
+    printSpanLog("span", "start", spanLabel(op, description));
     return span;
   }
   /** Span direto da transaction. */
@@ -671,6 +695,14 @@ var Transaction = class {
     if (this.finished) return;
     this.finished = true;
     for (const span of this.spans) span.finish();
+    const endTimestamp = nowISO();
+    printSpanLog(
+      "transaction",
+      "finish",
+      this.op ? `${this.name} (${this.op})` : this.name,
+      this.startTimestamp,
+      endTimestamp
+    );
     if (!this.sampled || !this.client) return;
     const payload = {
       event_id: this.eventId,
@@ -678,7 +710,7 @@ var Transaction = class {
       op: this.op,
       status: this.status,
       start_timestamp: this.startTimestamp,
-      timestamp: nowISO(),
+      timestamp: endTimestamp,
       spans: this.spans.map((span) => span.toJSON()),
       tags: Object.keys(this.tags).length > 0 ? this.tags : void 0,
       run_id: this.runId
@@ -820,11 +852,12 @@ var LOG_LEVEL_COLOR = {
   debug: "gray",
   info: "cyan",
   warning: "yellow",
-  error: "red"
+  error: "red",
+  success: "green"
 };
 function printRunLog(task, level, message) {
   const line = colorize(`[hobots:${task}] ${level.toUpperCase()}`, LOG_LEVEL_COLOR[level]);
-  const method = level === "warning" ? "warn" : level;
+  const method = level === "warning" ? "warn" : level === "success" ? "info" : level;
   console[method](line, message);
 }
 function runCanceledError() {
@@ -869,7 +902,8 @@ var RunReporter = class {
       debug: (message) => this.log("debug", message),
       info: (message) => this.log("info", message),
       warn: (message) => this.log("warning", message),
-      error: (message) => this.log("error", message)
+      error: (message) => this.log("error", message),
+      success: (message) => this.log("success", message)
     };
     const signal = this.abortController.signal;
     this.context = {
@@ -1031,6 +1065,24 @@ var RunReporter = class {
   }
 };
 
+// src/manager/run-context-store.ts
+var storage;
+var initPromise;
+async function ensureStorage() {
+  if (storage || !isNode()) return;
+  initPromise ??= import("async_hooks").then(({ AsyncLocalStorage: ALS }) => {
+    storage = new ALS();
+  });
+  await initPromise;
+}
+async function runWithRunId(runId, fn) {
+  await ensureStorage();
+  return storage ? storage.run({ runId }, fn) : fn();
+}
+function currentRunId() {
+  return storage?.getStore()?.runId;
+}
+
 // src/manager/shutdown.ts
 var WATCHED_SIGNALS = ["SIGINT", "SIGTERM"];
 function installShutdownHandlers(agent2) {
@@ -1265,7 +1317,7 @@ var ManagerAgent = class {
       }
       this.logger.debug(`executando run ${run.id} (${run.task})`);
       try {
-        await this.invokeWithTimeout(handler, run, reporter);
+        await runWithRunId(run.id, () => this.invokeWithTimeout(handler, run, reporter));
         reporter.reportSuccess();
         tx.setStatus("ok");
         this.logger.debug(`run ${run.id} conclu\xEDda`);
@@ -1390,7 +1442,7 @@ function withScope(callback) {
 function startTransaction(name, op) {
   const rate = client?.options.tracesSampleRate ?? 1;
   const sampled = rate >= 1 || Math.random() < rate;
-  return new Transaction(client, name, op, sampled);
+  return new Transaction(client, name, op, sampled, currentRunId());
 }
 function checkIn(options) {
   return client ? sendCheckIn(client, options) : void 0;