1 files changed, 46 insertions, 31 deletions
diff --git a/u/server/activity/health.ts b/u/server/activity/health.ts
index b9efa3a..0f54a99 100644
--- a/u/server/activity/health.ts
+++ b/u/server/activity/health.ts
@@ -1,4 +1,5 @@
 import {
+  type IActivity,
   type IEither,
   type ITraceable,
   JsonResponse,
@@ -14,39 +15,53 @@ export enum HealthCheckInput {
   CHECK,
 }
 export enum HealthCheckOutput {
-  YAASQUEEN,
+  YAASSSLAYQUEEN,
+}
+
+export interface IHealthCheckActivity {
+  checkHealth: IActivity;
 }
 
 const healthCheckMetric = Metric.fromName("Health");
-export const HealthCheckActivity = (
-  check: Mapper<
+export interface HealthChecker extends
+  Mapper<
     ITraceable<HealthCheckInput, ServerTrace>,
     Promise<IEither<Error, HealthCheckOutput>>
-  >,
-) =>
-(req: ITraceable<PenguenoRequest, ServerTrace>) =>
-  req
-    .bimap(TraceUtil.withFunctionTrace(HealthCheckActivity))
-    .bimap(TraceUtil.withMetricTrace(healthCheckMetric))
-    .flatMap((r) => r.move(HealthCheckInput.CHECK).map(check))
-    .map(TraceUtil.promiseify((h) => {
-      const health = h.get();
-      health.mapBoth((e) => {
-        h.trace.trace(healthCheckMetric.failure);
-        h.trace.addTrace(LogLevel.ERROR).trace(`${e}`);
-        return new JsonResponse(
-          req,
-          "oh no, i need to eat more vegetables (｡•́︿•̀｡)...",
-          { status: 500 },
-        );
-      }, (_healthy) => {
-        h.trace.trace(healthCheckMetric.success);
-        const msg = `think im healthy!! (✿˘◡˘) ready to do work~`;
-        h.trace.trace(msg);
-        return new JsonResponse(
-          req,
-          msg,
-          { status: 200 },
-        );
-      });
-    }));
+  > {}
+export class HealthCheckActivityImpl implements IHealthCheckActivity {
+  constructor(
+    private readonly check: HealthChecker,
+  ) {}
+
+  public checkHealth(req: ITraceable<PenguenoRequest, ServerTrace>) {
+    return req
+      .bimap(TraceUtil.withFunctionTrace(this.checkHealth))
+      .bimap(TraceUtil.withMetricTrace(healthCheckMetric))
+      .flatMap((r) => r.move(HealthCheckInput.CHECK).map(this.check))
+      .peek(TraceUtil.promiseify((h) =>
+        h.get().fold((err) => {
+          if (err) {
+            h.trace.trace(healthCheckMetric.failure);
+            h.trace.addTrace(LogLevel.ERROR).trace(`${err}`);
+            return;
+          }
+          h.trace.trace(healthCheckMetric.success);
+        })
+      ))
+      .map(TraceUtil.promiseify((h) =>
+        h.get()
+          .mapBoth(
+            () => "oh no, i need to eat more vegetables (｡•́︿•̀｡)...",
+            () => "think im healthy!! (✿˘◡˘) ready to do work~",
+          )
+          .fold((errMsg, okMsg) =>
+            new JsonResponse(
+              req,
+              errMsg ?? okMsg,
+              { status: errMsg ? 500 : 200 },
+            )
+          )
+      ))
+      .get();
+  }
+}