hobots 1.3.5 → 1.5.0

package/dist/index.js

@@ -58,7 +58,7 @@ function parseStacktrace(stack) {
 
 // src/utils.ts
 var SDK_NAME = "hobots";
-var SDK_VERSION = true ? "1.3.4" : "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";
 }
@@ -88,11 +88,26 @@ function sleep(ms) {
 var PREFIX = "[hobots-monitor]";
 function makeLogger(enabled, prefix = PREFIX) {
   return {
-    debug: (...args) => enabled && console.debug(prefix, ...args),
-    warn: (...args) => enabled && console.warn(prefix, ...args),
-    error: (...args) => console.error(prefix, ...args)
+    debug: (...args) => enabled && console.debug(colorize(prefix, "gray"), ...args),
+    warn: (...args) => enabled && console.warn(colorize(prefix, "yellow"), ...args),
+    error: (...args) => console.error(colorize(prefix, "red"), ...args)
   };
 }
+var ANSI = {
+  reset: "\x1B[0m",
+  gray: "\x1B[90m",
+  cyan: "\x1B[36m",
+  yellow: "\x1B[33m",
+  red: "\x1B[31m"
+};
+function colorEnabled() {
+  if (!isNode()) return false;
+  return Boolean(process.stdout?.isTTY) && !process.env.NO_COLOR;
+}
+function colorize(text, color) {
+  if (!colorEnabled()) return text;
+  return `${ANSI[color]}${text}${ANSI.reset}`;
+}
 
 // src/eventbuilder.ts
 function safeString(value) {
@@ -564,11 +579,12 @@ var Span = class {
   }
 };
 var Transaction = class {
-  constructor(client2, name, op, sampled = true) {
+  constructor(client2, name, op, sampled = true, runId) {
     this.client = client2;
     this.name = name;
     this.op = op;
     this.sampled = sampled;
+    this.runId = runId;
   }
   eventId = uuid4();
   spanId = spanId();
@@ -611,7 +627,8 @@ var Transaction = class {
       start_timestamp: this.startTimestamp,
       timestamp: nowISO(),
       spans: this.spans.map((span) => span.toJSON()),
-      tags: Object.keys(this.tags).length > 0 ? this.tags : void 0
+      tags: Object.keys(this.tags).length > 0 ? this.tags : void 0,
+      run_id: this.runId
     };
     this.client.sendTransaction(payload);
   }
@@ -746,6 +763,17 @@ var ManagerApi = class {
 };
 
 // src/manager/context.ts
+var LOG_LEVEL_COLOR = {
+  debug: "gray",
+  info: "cyan",
+  warning: "yellow",
+  error: "red"
+};
+function printRunLog(task, level, message) {
+  const line = colorize(`[hobots:${task}] ${level.toUpperCase()}`, LOG_LEVEL_COLOR[level]);
+  const method = level === "warning" ? "warn" : level;
+  console[method](line, message);
+}
 function runCanceledError() {
   return Object.assign(new Error("execu\xE7\xE3o cancelada"), { name: "RunCanceled" });
 }
@@ -775,7 +803,7 @@ var MAX_LOGS_PER_RUN = 500;
 var MAX_LOGS_PER_PATCH = 100;
 var FLUSH_AT_BUFFERED = 20;
 var RunReporter = class {
-  constructor(api, runId, task, logger, flushIntervalMs, cancelCheckIntervalMs) {
+  constructor(api, runId, task, logger, tx, flushIntervalMs, cancelCheckIntervalMs) {
     this.api = api;
     this.runId = runId;
     this.logger = logger;
@@ -795,6 +823,7 @@ var RunReporter = class {
       runId,
       task,
       log,
+      tx,
       signal,
       throwIfCanceled: () => {
         if (signal.aborted) throw runCanceledError();
@@ -835,6 +864,7 @@ var RunReporter = class {
   }
   log(level, message) {
     if (this.dead) return;
+    printRunLog(this.context.task, level, message);
     if (this.loggedCount >= MAX_LOGS_PER_RUN) {
       this.droppedCount++;
       return;
@@ -948,6 +978,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) {
@@ -1009,6 +1057,7 @@ var ManagerAgent = class {
     this.logger = makeLogger(options.debug ?? false, "[hobots-manager]");
     const config = resolveConfig(options, "[hobots-manager]");
     this.api = new ManagerApi({ baseUrl: config.managerBaseUrl, key: config.key, logger: this.logger });
+    this.client = options.client;
     this.pollIntervalMs = Math.max(MIN_POLL_INTERVAL_MS, options.pollIntervalMs ?? DEFAULT_POLL_INTERVAL_MS);
     this.concurrency = Math.min(MAX_CONCURRENCY, Math.max(1, options.concurrency ?? 1));
     this.instanceId = options.instanceId;
@@ -1019,6 +1068,7 @@ var ManagerAgent = class {
   }
   logger;
   api;
+  client;
   pollIntervalMs;
   concurrency;
   instanceId;
@@ -1156,7 +1206,8 @@ var ManagerAgent = class {
     this.activeRuns.add(promise);
   }
   async executeRun(run) {
-    const reporter = new RunReporter(this.api, run.id, run.task, this.logger, this.logFlushIntervalMs, this.cancelCheckIntervalMs);
+    const tx = new Transaction(this.client, `run:${run.task}`, "manager.run", true, run.id);
+    const reporter = new RunReporter(this.api, run.id, run.task, this.logger, tx, this.logFlushIntervalMs, this.cancelCheckIntervalMs);
     this.reporters.add(reporter);
     try {
       const handler = this.registry.get(run.task);
@@ -1166,22 +1217,29 @@ var ManagerAgent = class {
           message: `tarefa n\xE3o registrada no agente: ${run.task}`
         });
         await reporter.finalize();
+        tx.setStatus("error");
+        tx.finish();
         return;
       }
       const ok = await reporter.reportRunning();
       if (!ok) {
         this.logger.debug(`run ${run.id} cancelada antes de iniciar \u2014 handler n\xE3o executado`);
+        tx.setStatus("cancelled");
+        tx.finish();
         return;
       }
       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`);
       } catch (error) {
         reporter.reportFailure(toRunError(error));
+        tx.setStatus("error");
         this.logger.debug(`run ${run.id} falhou: ${String(error)}`);
       }
+      tx.finish();
       await reporter.finalize();
     } finally {
       this.reporters.delete(reporter);
@@ -1252,7 +1310,7 @@ function init(options) {
   if (options.manager) {
     const managerOptions = options.manager === true ? {} : options.manager;
     agent = new ManagerAgent(
-      { ...managerOptions, clientSecret: options.clientSecret, instanceId: options.instanceId, debug: options.debug },
+      { ...managerOptions, clientSecret: options.clientSecret, instanceId: options.instanceId, debug: options.debug, client },
       registry
     );
   }
@@ -1297,7 +1355,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;