The big refactor TM

1 files changed, 82 insertions, 0 deletions

diff --git a/u/trace/trace.ts b/u/trace/trace.ts
new file mode 100644
index 0000000..e942066
--- /dev/null
+++ b/u/trace/trace.ts
@@ -0,0 +1,82 @@
+import {
+  isMetricsTraceSupplier,
+  type ITrace,
+  type ITraceWith,
+  LogTrace,
+  type LogTraceSupplier,
+  MetricsTrace,
+  type MetricsTraceSupplier,
+  type MetricValue,
+  TraceableImpl,
+} from "@emprespresso/pengueno";
+
+export class LogTraceable<T> extends TraceableImpl<T, LogTraceSupplier> {
+  public static LogTrace = new LogTrace();
+  static from<T>(t: T) {
+    return new LogTraceable(t, LogTraceable.LogTrace);
+  }
+}
+
+const getEmbeddedMetricConsumer =
+  (logTrace: LogTrace) => (metrics: Array<MetricValue>) =>
+    logTrace.addTrace("<metrics>").trace(
+      JSON.stringify(metrics, null, 2) + "</metrics>",
+    );
+export class EmbeddedMetricsTraceable<T>
+  extends TraceableImpl<T, MetricsTraceSupplier> {
+  public static MetricsTrace = new MetricsTrace(
+    getEmbeddedMetricConsumer(LogTraceable.LogTrace),
+  );
+
+  static from<T>(t: T) {
+    return new EmbeddedMetricsTraceable(
+      t,
+      EmbeddedMetricsTraceable.MetricsTrace,
+    );
+  }
+}
+
+export type LogMetricTraceSupplier = ITraceWith<
+  LogTraceSupplier | MetricsTraceSupplier
+>;
+export class LogMetricTrace implements ITrace<LogMetricTraceSupplier> {
+  constructor(
+    private logTrace: ITrace<LogTraceSupplier>,
+    private metricsTrace: ITrace<MetricsTraceSupplier>,
+  ) {}
+
+  public addTrace(
+    trace: LogTraceSupplier | MetricsTraceSupplier,
+  ): LogMetricTrace {
+    if (isMetricsTraceSupplier(trace)) {
+      this.metricsTrace = this.metricsTrace.addTrace(trace);
+      return this;
+    }
+    this.logTrace = this.logTrace.addTrace(trace);
+    return this;
+  }
+
+  public trace(trace: LogTraceSupplier | MetricsTraceSupplier) {
+    if (isMetricsTraceSupplier(trace)) {
+      this.metricsTrace.trace(trace);
+      return this;
+    }
+    this.logTrace.trace(trace);
+    return this;
+  }
+}
+
+export class LogMetricTraceable<T>
+  extends TraceableImpl<T, MetricsTraceSupplier | LogTraceSupplier> {
+  public static LogMetricTrace = new LogMetricTrace(
+    LogTraceable.LogTrace,
+    EmbeddedMetricsTraceable.MetricsTrace,
+  );
+
+  static from<T>(t: T) {
+    return new LogMetricTraceable(
+      t,
+      LogMetricTraceable.LogMetricTrace,
+    );
+  }
+}