feat(shared:log): Improve unit logging

1 files changed, 11 insertions, 7 deletions

diff --git a/packages/shared/log.ts b/packages/shared/log.ts
index ce50dd5..965f7f5 100644
--- a/packages/shared/log.ts
+++ b/packages/shared/log.ts
@@ -8,11 +8,7 @@ export enum LogLevel {
   Trace = "TRACE",
 }
 
-export const log = (
-  unit: string,
-  message: string,
-  level: LogLevel = LogLevel.Info,
-) => {
+export const log = (message: string, level: LogLevel = LogLevel.Info) => {
   const timestamp = new Date().toISOString();
   const levelColors = {
     [LogLevel.Error]: "\x1b[31m",
@@ -22,15 +18,23 @@ export const log = (
     [LogLevel.Trace]: "\x1b[35m",
   };
   const levelColor = levelColors[level];
+  const callingLine = new Error().stack?.split("\n")[2].trim();
+  const unit =
+    callingLine
+      ?.match(/packages\/(.*?):\d+/)?.[1]
+      ?.replace(/\//g, "::")
+      ?.replace(/::src::/, "::")
+      ?.replace(/\.ts$/, "") +
+    "::" +
+    callingLine?.match(/:(\d+):/)?.[1];
 
   console.log(
     `\x1b[37m${timestamp}\x1b[0m ${levelColor}${level}\x1b[0m \x1b[1m${unit}\x1b[22m \x1b[37m> ${message}\x1b[0m`,
   );
 };
 
-export const logUnexpectedDiscordAPIError = (unit: string, error: unknown) => {
+export const logUnexpectedDiscordAPIError = (error: unknown) => {
   log(
-    unit,
     `Unexpected Discord API error: ${error instanceof Error ? error.message : String(error)}`,
     LogLevel.Error,
   );