@shoru/kitten 0.0.1-beta

package/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 Aymane Shoru
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

package/README.md

@@ -0,0 +1,24 @@
+<div align="left">
+
+<img src="https://files.catbox.moe/j7dpni.png" align="right" width="200" alt="Kitten Logo">
+
+<a name="-kitten"></a>
+# **Kitten Framework**
+
+### A powerful `Node.js` framework to simplify making WhatsApp Bots. built on top of the [Baileys](https://github.com/WhiskeySockets/Baileys) library. 😺
+
+![State](https://img.shields.io/badge/State-BETA-5a67d8?style=for-the-badge&logo=activity)
+[![npm](https://img.shields.io/npm/v/@shoru/kitten.svg?style=for-the-badge)](https://www.npmjs.com/package/@shoru/kitten)
+[![License](https://img.shields.io/github/license/Mangaka-bot/kitten-wa?style=for-the-badge)](https://github.com/Mangaka-bot/kitten-wa/blob/main/LICENSE)
+
+</div>
+
+---
+
+<br clear="left"/>
+
+<div align="center">
+
+
+**Made with ❤️ for Whatsapp community**
+</div>

package/package.json

@@ -0,0 +1,41 @@
+{
+  "name": "@shoru/kitten",
+  "type": "module",
+  "version": "0.0.1-beta",
+  "description": "A powerful Node.js framework to simplify making WhatsApp Bots",
+  "main": "src/index.js",
+  "repository": {
+    "url": "git+https://github.com/Mangaka-bot/kitten-wa.git"
+  },
+  "scripts": {
+    "start": "node --no-warnings src/index.js",
+    "test": "echo \"Error: no test specified\" && exit 1"
+  },
+  "imports": {
+    "#utils.js": "./src/utils/index.js",
+    "#auth.js": "./src/auth/index.js",
+    "#internals.js": "./src/internals/index.js",
+    "#client.js": "./src/client/index.js",
+    "#formatter.js": "./src/formatter/index.js"
+  },
+  "exports": {
+    ".": "./src/index.js"
+  },
+  "author": "Aymane Shoru",
+  "license": "MIT",
+  "dependencies": {
+    "@hapi/boom": "^10.0.1",
+    "@inquirer/prompts": "^8.1.0",
+    "@shoru/spindle": "^1.0.6",
+    "async-mutex": "^0.5.0",
+    "baileys": "^7.0.0-rc.9",
+    "chalk": "^5.6.2",
+    "chokidar": "^5.0.0",
+    "cosmiconfig": "^9.0.0",
+    "defu": "^6.1.4",
+    "lmdb": "^3.4.4",
+    "pino": "^10.1.0",
+    "pino-pretty": "^13.1.3",
+    "qrcode-terminal": "^0.12.0"
+  }
+}

package/src/auth/index.js

@@ -0,0 +1,2 @@
+export * from "./lmdb-auth-state.js";
+export * from "./init-session.js";

package/src/auth/init-session.js

@@ -0,0 +1,21 @@
+import { makeWASocket } from "baileys";
+import { useLMDBAuthState } from "./lmdb-auth-state.js";
+import { config } from "#internals.js";
+
+export const initSession = async ({ socketConfig, id } = {}) => {
+  try {
+    const { state, saveCreds, session } = await useLMDBAuthState(id);
+  
+    const sock = makeWASocket({ 
+      auth: state,
+      ...config.socket, 
+      ...socketConfig
+    });
+  
+    sock.ev.on("creds.update", saveCreds);
+
+    return { sock, session };
+  } catch (err) {
+    throw new Error(`[INIT_SESSION] Failed to initialize session: ${err.message}`, { cause: err });
+  }
+};

package/src/auth/lmdb-auth-state.js

@@ -0,0 +1,171 @@
+import { proto, initAuthCreds } from "baileys";
+import { LMDBManager, logger } from "#internals.js";
+import { serialize, deserialize } from "#utils.js";
+
+const KEY_PREFIX = "baileys";
+const COUNTER_KEY = `${KEY_PREFIX}:__meta__:counter`;
+const SESSION_PREFIX = `${KEY_PREFIX}:__sessions__:`;
+
+const keyBuilder = (sessionId) => {
+  const prefix = `${KEY_PREFIX}:${sessionId}:`;
+  return {
+    sessionId,
+    sessionPrefix: prefix,
+    creds: `${prefix}creds`,
+    forKey: (type, id) => `${prefix}${type}:${id}`,
+  };
+};
+
+const genID = async (db) => {
+  return db.transaction(() => {
+    const id = (db.get(COUNTER_KEY) ?? 0) + 1;
+    db.put(COUNTER_KEY, id);
+    return id;
+  });
+};
+
+const getSessionId = async (db, input) => {
+  if (input == null) return genID(db);
+  if (Number.isInteger(input) && input >= 0) return input;
+  throw new TypeError(
+    'Invalid sessionId: expected null/undefined or non-negative integer'
+  );
+};
+
+export async function useLMDBAuthState(inputSessionId) {
+  const { db } = LMDBManager;
+  const sessionId = await getSessionId(db, inputSessionId);
+  const keys = keyBuilder(sessionId);
+
+  const writeCreds = async (credsData) => {
+    await db.put(keys.creds, serialize(credsData));
+  };
+
+  const getKeys = (type, ids) => {
+    if (!ids.length) return {};
+
+    const keyList = ids.map((id) => keys.forKey(type, id));
+    const values = db.getMany(keyList);
+
+    const result = {};
+    for (let i = 0; i < ids.length; i++) {
+      const rawValue = values[i];
+
+      if (rawValue) {
+        try {
+          let parsed = deserialize(rawValue);
+          if (type === "app-state-sync-key" && parsed) {
+            parsed = proto.Message.AppStateSyncKeyData.fromObject(parsed);
+          }
+          result[ids[i]] = parsed;
+        } catch (err) {
+          logger.error(
+            err,
+            `[LMDBAuthState] Deserialize error: ${type}:${ids[i]}`
+          );
+          db.remove(keys.forKey(type, ids[i]));
+        }
+      }
+    }
+    return result;
+  };
+
+  const setKeys = async (data) => {
+    await db.batch(() => {
+      for (const [category, categoryData] of Object.entries(data)) {
+        if (!categoryData) continue;
+        for (const [id, value] of Object.entries(categoryData)) {
+          const key = keys.forKey(category, id);
+          if (value != null) {
+            db.put(key, serialize(value));
+          } else {
+            db.remove(key);
+          }
+        }
+      }
+    });
+  };
+
+  const clearKeys = async () => {
+    let count = 0;
+    await db.batch(() => {
+      for (const { key } of db.getRange({
+        start: keys.sessionPrefix,
+        end: `${keys.sessionPrefix}\xFF`,
+      })) {
+        if (key !== keys.creds) {
+          db.remove(key);
+          count++;
+        }
+      }
+    });
+    logger.debug(`[LMDBAuthState] Cleared ${count} keys`);
+  };
+
+  const deleteSession = async () => {
+    await db.batch(() => {
+      for (const { key } of db.getRange({
+        start: keys.sessionPrefix,
+        end: `${keys.sessionPrefix}\xFF`,
+      })) {
+        db.remove(key);
+      }
+      db.remove(`${SESSION_PREFIX}${sessionId}`);
+    });
+
+    logger.debug(`[LMDBAuthState] Deleted session ${sessionId}`);
+  };
+
+  const creds = await db.transaction(() => {
+    const existing = db.get(keys.creds);
+    if (existing != null) {
+      return deserialize(existing);
+    }
+
+    const newCreds = initAuthCreds();
+    db.put(keys.creds, serialize(newCreds));
+    db.put(`${SESSION_PREFIX}${sessionId}`, true);
+    return newCreds;
+  });
+
+  return {
+    state: {
+      creds,
+      keys: {
+        get: getKeys,
+        set: setKeys,
+        clear: clearKeys,
+      },
+    },
+    saveCreds: () => writeCreds(creds),
+    session: {
+      delete: deleteSession,
+      clear: clearKeys,
+      id: sessionId,
+    },
+  };
+}
+
+export function listSessions() {
+  if (!LMDBManager.isOpen) return [];
+
+  const { db } = LMDBManager;
+  const sessions = [];
+
+  for (const { key } of db.getRange({
+    start: SESSION_PREFIX,
+    end: `${SESSION_PREFIX}\xFF`,
+  })) {
+    const id = parseInt(key.slice(SESSION_PREFIX.length), 10);
+    if (!isNaN(id)) sessions.push(id);
+  }
+
+  return sessions.sort((a, b) => a - b);
+}
+
+export function sessionExists(sessionId) {
+  if (!Number.isInteger(sessionId) || sessionId < 0 || !LMDBManager.isOpen)
+    return false;
+  const { db } = LMDBManager;
+  return db.get(`${SESSION_PREFIX}${sessionId}`) != null;
+}