feat: added wakatime, dotenv reload and update mem monitor

Author: mrepol742

.env.example

@@ -75,4 +75,8 @@ QUERY_CACHING_TTL=3600
 
 # the shell to use for executing commands
 # this should be a valid shell path, e.g., /bin/bash, /bin/zsh
-EXEC_SHELL=/bin/bash
+EXEC_SHELL=/bin/bash
+
+# Your Wakatime API keys
+# (optional) else the wakatime command aint going to work
+WAKATIME_API_KEY=

src/commands/env.ts

@@ -0,0 +1,21 @@
+import { Message } from "../../types/message";
+import log from "../components/utils/log";
+import dotenv from "dotenv";
+
+export const info = {
+  command: "env",
+  description:
+    "Get all new process.env and append them into the project without restarting.",
+  usage: "env",
+  example: "env",
+  role: "user",
+  cooldown: 5000,
+};
+
+export default async function (msg: Message) {
+  if (!/^env/i.test(msg.body)) return;
+
+  dotenv.config({ override: true });
+
+  await msg.reply("Dotenv override successfully.");
+}

src/commands/reload.ts

@@ -11,7 +11,7 @@ export const info = {
   description: "Reload a specific command or all commands.",
   usage: "reload [command]",
   example: "reload ai",
-  role: "admin",
+  role: "user",
   cooldown: 5000,
 };
 

src/commands/update.ts

@@ -11,22 +11,18 @@ export const info = {
   description: "Pull changes from the remote repository.",
   usage: "update",
   example: "update",
-  role: "admin",
+  role: "user",
   cooldown: 5000,
 };
 
 const execPromise = util.promisify(exec);
 
 export default async function (msg: Message) {
-  try {
-    const { stdout, stderr } = await execPromise("git pull");
+  if (!/^update/i.test(msg.body)) return;
+  const { stdout, stderr } = await execPromise("git pull");
 
-    if (stdout) log.info("Update", `git pull stdout:\n${stdout}`);
-    if (stderr) log.warn("Update", `git pull stderr:\n${stderr}`);
+  if (stdout) log.info("Update", `git pull stdout:\n${stdout}`);
+  if (stderr) log.warn("Update", `git pull stderr:\n${stderr}`);
 
-    await msg.reply(stdout || stderr);
-  } catch (error: any) {
-    log.error("Update", `git pull failed: ${error.message}`);
-    await msg.reply("Failed to update repository. Check logs.");
-  }
+  await msg.reply(stdout || stderr);
 }

src/commands/wakatime.ts

@@ -0,0 +1,78 @@
+import { Message } from "../../types/message";
+import log from "../components/utils/log";
+import axios from "axios";
+
+export const info = {
+  command: "wakatime",
+  description: "Shows current WakaTime tasks for today.",
+  usage: "wakatime",
+  example: "wakatime",
+  role: "user",
+  cooldown: 5000,
+};
+
+export default async function (msg: Message) {
+  if (!/^wakatime/i.test(msg.body)) return;
+
+  const apiKey = process.env.WAKATIME_API_KEY;
+  if (!apiKey) {
+    await msg.reply("WakaTime API key is not configured.");
+    return;
+  }
+
+  const today = new Date().toISOString().split("T")[0]; // YYYY-MM-DD
+
+  const response = await axios.get(
+    `https://wakatime.com/api/v1/users/current/heartbeats`,
+    {
+      params: { date: today },
+      headers: {
+        Authorization: `Basic ${Buffer.from(apiKey + ":").toString("base64")}`,
+      },
+    },
+  );
+
+  const heartbeats = response.data.data;
+  if (!heartbeats || heartbeats.length === 0) {
+    await msg.reply("The author didn't do any job today unfortunately 😢");
+    return;
+  }
+
+  // Sort by time
+  heartbeats.sort((a: any, b: any) => a.time - b.time);
+
+  const projectDurations: Record<string, number> = {};
+
+  for (let i = 0; i < heartbeats.length; i++) {
+    const hb = heartbeats[i];
+    const nextHb = heartbeats[i + 1];
+
+    if (!hb.project) continue;
+
+    let duration = 0;
+    if (nextHb) {
+      duration = Math.min(120, nextHb.time - hb.time); // max 2 minutes per heartbeat
+    } else {
+      duration = 60; // assume 1 minute for last heartbeat
+    }
+
+    projectDurations[hb.project] =
+      (projectDurations[hb.project] || 0) + duration;
+  }
+
+  const projects = Object.keys(projectDurations);
+  if (projects.length === 0) {
+    await msg.reply("The author didn't do any job today unfortunately 😢");
+    return;
+  }
+
+  const projectList = projects.map((proj) => {
+    const seconds = projectDurations[proj];
+    const hours = Math.floor(seconds / 3600);
+    const minutes = Math.floor((seconds % 3600) / 60);
+    return `- ${proj}: ${hours}h ${minutes}m`;
+  });
+
+  const replyText = `\`Today's WakaTime projects\`\n${projectList.join("\n")}`;
+  await msg.reply(replyText);
+}

src/components/utils/memMonitor.ts

@@ -47,7 +47,7 @@ export class MemoryMonitor {
 
       // Oh shit!
       if (this.history.length > 5) {
-        const lastFive = this.history.slice(-5).map((h) => h.usedMB);
+        const lastFive = this.history.slice(-5).map((h) => h.usedMB && h.usedMB > this.thresholdMB);
         if (lastFive.every((val, i, arr) => i === 0 || val > arr[i - 1])) {
           log.warn(
             "MemoryMonitor",