wagent 1.0.0

package/dist/db/client.js

@@ -0,0 +1,160 @@
+// ============================================================
+// WA MCP — Database Client
+// ============================================================
+import { existsSync, mkdirSync } from "node:fs";
+import { dirname, join } from "node:path";
+import { homedir } from "node:os";
+import Database from "better-sqlite3";
+import { drizzle } from "drizzle-orm/better-sqlite3";
+import * as schema from "./schema.js";
+function getDataDir() {
+    if (process.env.WAGENT_HOME)
+        return process.env.WAGENT_HOME;
+    if (process.env.WAGENT_DATA_DIR)
+        return process.env.WAGENT_DATA_DIR;
+    if (process.platform === "win32")
+        return join(process.env.APPDATA ?? homedir(), "wagent");
+    return join(process.env.XDG_CONFIG_HOME ?? join(homedir(), ".config"), "wagent");
+}
+const DB_PATH = join(getDataDir(), "wagent.db");
+// Ensure the data directory exists
+const dir = dirname(DB_PATH);
+if (!existsSync(dir)) {
+    mkdirSync(dir, { recursive: true });
+}
+const sqlite = new Database(DB_PATH);
+// Enable WAL mode for better concurrent read performance
+sqlite.pragma("journal_mode = WAL");
+sqlite.pragma("foreign_keys = ON");
+export const db = drizzle(sqlite, { schema });
+// Run table creation on init (push-based, no migration files needed)
+function initializeDatabase() {
+    sqlite.exec(`
+    CREATE TABLE IF NOT EXISTS instances (
+      id              TEXT PRIMARY KEY,
+      name            TEXT NOT NULL,
+      channel         TEXT NOT NULL DEFAULT 'baileys',
+      phone_number    TEXT,
+      status          TEXT NOT NULL DEFAULT 'disconnected',
+      wa_version      TEXT,
+      cloud_access_token    TEXT,
+      cloud_phone_number_id TEXT,
+      cloud_business_id     TEXT,
+      created_at      INTEGER NOT NULL,
+      updated_at      INTEGER NOT NULL,
+      last_connected  INTEGER,
+      last_disconnected INTEGER
+    );
+
+    CREATE TABLE IF NOT EXISTS auth_keys (
+      instance_id     TEXT NOT NULL REFERENCES instances(id) ON DELETE CASCADE,
+      key_id          TEXT NOT NULL,
+      key_data        TEXT NOT NULL,
+      PRIMARY KEY (instance_id, key_id)
+    );
+    CREATE INDEX IF NOT EXISTS idx_auth_keys_instance ON auth_keys(instance_id);
+
+    CREATE TABLE IF NOT EXISTS messages (
+      id              TEXT NOT NULL,
+      instance_id     TEXT NOT NULL REFERENCES instances(id) ON DELETE CASCADE,
+      chat_id         TEXT NOT NULL,
+      sender_id       TEXT NOT NULL,
+      type            TEXT NOT NULL,
+      content         TEXT,
+      media_url       TEXT,
+      media_local     TEXT,
+      media_mimetype  TEXT,
+      quoted_id       TEXT,
+      is_from_me      INTEGER NOT NULL DEFAULT 0,
+      is_forwarded    INTEGER NOT NULL DEFAULT 0,
+      status          TEXT NOT NULL DEFAULT 'received',
+      timestamp       INTEGER NOT NULL,
+      raw_data        TEXT,
+      PRIMARY KEY (instance_id, id)
+    );
+    CREATE INDEX IF NOT EXISTS idx_messages_chat ON messages(instance_id, chat_id, timestamp DESC);
+    CREATE INDEX IF NOT EXISTS idx_messages_timestamp ON messages(instance_id, timestamp DESC);
+
+    CREATE TABLE IF NOT EXISTS processed_messages (
+      message_id      TEXT NOT NULL,
+      instance_id     TEXT NOT NULL REFERENCES instances(id) ON DELETE CASCADE,
+      processed_at    INTEGER NOT NULL,
+      PRIMARY KEY (instance_id, message_id)
+    );
+
+    CREATE TABLE IF NOT EXISTS contacts (
+      instance_id     TEXT NOT NULL REFERENCES instances(id) ON DELETE CASCADE,
+      jid             TEXT NOT NULL,
+      name            TEXT,
+      notify_name     TEXT,
+      phone           TEXT,
+      lid             TEXT,
+      profile_pic_url TEXT,
+      is_business     INTEGER NOT NULL DEFAULT 0,
+      is_blocked      INTEGER NOT NULL DEFAULT 0,
+      updated_at      INTEGER NOT NULL,
+      PRIMARY KEY (instance_id, jid)
+    );
+    CREATE INDEX IF NOT EXISTS idx_contacts_phone ON contacts(instance_id, phone);
+    CREATE INDEX IF NOT EXISTS idx_contacts_lid ON contacts(instance_id, lid);
+
+    CREATE TABLE IF NOT EXISTS groups_cache (
+      instance_id     TEXT NOT NULL REFERENCES instances(id) ON DELETE CASCADE,
+      jid             TEXT NOT NULL,
+      subject         TEXT,
+      description     TEXT,
+      owner_jid       TEXT,
+      participants    TEXT NOT NULL DEFAULT '[]',
+      participant_count INTEGER NOT NULL DEFAULT 0,
+      is_announce      INTEGER NOT NULL DEFAULT 0,
+      is_locked        INTEGER NOT NULL DEFAULT 0,
+      ephemeral_duration INTEGER,
+      invite_code     TEXT,
+      created_at      INTEGER,
+      updated_at      INTEGER NOT NULL,
+      PRIMARY KEY (instance_id, jid)
+    );
+
+    CREATE TABLE IF NOT EXISTS chats (
+      instance_id     TEXT NOT NULL REFERENCES instances(id) ON DELETE CASCADE,
+      jid             TEXT NOT NULL,
+      name            TEXT,
+      is_group        INTEGER NOT NULL DEFAULT 0,
+      unread_count    INTEGER NOT NULL DEFAULT 0,
+      is_pinned       INTEGER NOT NULL DEFAULT 0,
+      is_muted        INTEGER NOT NULL DEFAULT 0,
+      mute_until      INTEGER,
+      is_archived     INTEGER NOT NULL DEFAULT 0,
+      last_message_id TEXT,
+      last_message_at INTEGER,
+      updated_at      INTEGER NOT NULL,
+      PRIMARY KEY (instance_id, jid)
+    );
+    CREATE INDEX IF NOT EXISTS idx_chats_recent ON chats(instance_id, last_message_at DESC);
+
+    CREATE TABLE IF NOT EXISTS wa_version_cache (
+      id              INTEGER PRIMARY KEY DEFAULT 1,
+      version_json    TEXT NOT NULL,
+      fetched_at      INTEGER NOT NULL,
+      is_latest       INTEGER NOT NULL DEFAULT 1
+    );
+
+    CREATE TABLE IF NOT EXISTS queue_stats (
+      instance_id     TEXT NOT NULL PRIMARY KEY REFERENCES instances(id) ON DELETE CASCADE,
+      messages_sent   INTEGER NOT NULL DEFAULT 0,
+      messages_failed INTEGER NOT NULL DEFAULT 0,
+      last_sent_at    INTEGER,
+      rate_limited_until INTEGER
+    );
+  `);
+}
+// Inline migration: add `lid` column for existing databases (pre-v7 upgrade)
+// Must run BEFORE initializeDatabase() so the index creation succeeds
+try {
+    sqlite.exec(`ALTER TABLE contacts ADD COLUMN lid TEXT`);
+}
+catch {
+    // column already exists or table doesn't exist yet — ignore
+}
+initializeDatabase();
+export { sqlite };

package/dist/db/schema.js

@@ -0,0 +1,189 @@
+// ============================================================
+// WA MCP — Drizzle ORM Schema Definitions
+// ============================================================
+import { sqliteTable, text, integer, primaryKey, index } from "drizzle-orm/sqlite-core";
+// ============================================================
+// INSTANCES
+// Core entity: each instance = one WhatsApp number connection
+// ============================================================
+export const instances = sqliteTable("instances", {
+    id: text("id").primaryKey(),
+    name: text("name").notNull(),
+    channel: text("channel", { enum: ["baileys", "cloud"] })
+        .notNull()
+        .default("baileys"),
+    phoneNumber: text("phone_number"),
+    status: text("status", {
+        enum: ["connected", "disconnected", "connecting", "qr_pending"],
+    })
+        .notNull()
+        .default("disconnected"),
+    waVersion: text("wa_version"),
+    // Cloud API specific
+    cloudAccessToken: text("cloud_access_token"),
+    cloudPhoneNumberId: text("cloud_phone_number_id"),
+    cloudBusinessId: text("cloud_business_id"),
+    // Metadata
+    createdAt: integer("created_at").notNull(),
+    updatedAt: integer("updated_at").notNull(),
+    lastConnected: integer("last_connected"),
+    lastDisconnected: integer("last_disconnected"),
+});
+// ============================================================
+// AUTH KEYS
+// Baileys session persistence (Signal protocol keys)
+// ============================================================
+export const authKeys = sqliteTable("auth_keys", {
+    instanceId: text("instance_id")
+        .notNull()
+        .references(() => instances.id, { onDelete: "cascade" }),
+    keyId: text("key_id").notNull(),
+    keyData: text("key_data").notNull(),
+}, (table) => [
+    primaryKey({ columns: [table.instanceId, table.keyId] }),
+    index("idx_auth_keys_instance").on(table.instanceId),
+]);
+// ============================================================
+// MESSAGES
+// Message store for agent context and history
+// ============================================================
+export const messages = sqliteTable("messages", {
+    id: text("id").notNull(),
+    instanceId: text("instance_id")
+        .notNull()
+        .references(() => instances.id, { onDelete: "cascade" }),
+    chatId: text("chat_id").notNull(),
+    senderId: text("sender_id").notNull(),
+    type: text("type", {
+        enum: [
+            "text",
+            "image",
+            "video",
+            "audio",
+            "document",
+            "location",
+            "contact",
+            "poll",
+            "reaction",
+            "sticker",
+        ],
+    }).notNull(),
+    content: text("content"),
+    mediaUrl: text("media_url"),
+    mediaLocal: text("media_local"),
+    mediaMimetype: text("media_mimetype"),
+    quotedId: text("quoted_id"),
+    isFromMe: integer("is_from_me").notNull().default(0),
+    isForwarded: integer("is_forwarded").notNull().default(0),
+    status: text("status", {
+        enum: ["received", "sent", "delivered", "read", "played"],
+    })
+        .notNull()
+        .default("received"),
+    timestamp: integer("timestamp").notNull(),
+    rawData: text("raw_data"),
+}, (table) => [
+    primaryKey({ columns: [table.instanceId, table.id] }),
+    index("idx_messages_chat").on(table.instanceId, table.chatId, table.timestamp),
+    index("idx_messages_timestamp").on(table.instanceId, table.timestamp),
+]);
+// ============================================================
+// PROCESSED MESSAGES
+// Deduplication: track which messages have been processed
+// ============================================================
+export const processedMessages = sqliteTable("processed_messages", {
+    messageId: text("message_id").notNull(),
+    instanceId: text("instance_id")
+        .notNull()
+        .references(() => instances.id, { onDelete: "cascade" }),
+    processedAt: integer("processed_at").notNull(),
+}, (table) => [primaryKey({ columns: [table.instanceId, table.messageId] })]);
+// ============================================================
+// CONTACTS
+// Cached contact information
+// ============================================================
+export const contacts = sqliteTable("contacts", {
+    instanceId: text("instance_id")
+        .notNull()
+        .references(() => instances.id, { onDelete: "cascade" }),
+    jid: text("jid").notNull(),
+    name: text("name"),
+    notifyName: text("notify_name"),
+    phone: text("phone"),
+    lid: text("lid"),
+    profilePicUrl: text("profile_pic_url"),
+    isBusiness: integer("is_business").notNull().default(0),
+    isBlocked: integer("is_blocked").notNull().default(0),
+    updatedAt: integer("updated_at").notNull(),
+}, (table) => [
+    primaryKey({ columns: [table.instanceId, table.jid] }),
+    index("idx_contacts_phone").on(table.instanceId, table.phone),
+    index("idx_contacts_lid").on(table.instanceId, table.lid),
+]);
+// ============================================================
+// GROUPS CACHE
+// Cached group metadata
+// ============================================================
+export const groupsCache = sqliteTable("groups_cache", {
+    instanceId: text("instance_id")
+        .notNull()
+        .references(() => instances.id, { onDelete: "cascade" }),
+    jid: text("jid").notNull(),
+    subject: text("subject"),
+    description: text("description"),
+    ownerJid: text("owner_jid"),
+    participants: text("participants").notNull().default("[]"),
+    participantCount: integer("participant_count").notNull().default(0),
+    isAnnounce: integer("is_announce").notNull().default(0),
+    isLocked: integer("is_locked").notNull().default(0),
+    ephemeralDuration: integer("ephemeral_duration"),
+    inviteCode: text("invite_code"),
+    createdAt: integer("created_at"),
+    updatedAt: integer("updated_at").notNull(),
+}, (table) => [primaryKey({ columns: [table.instanceId, table.jid] })]);
+// ============================================================
+// CHATS
+// Cached chat list with metadata
+// ============================================================
+export const chats = sqliteTable("chats", {
+    instanceId: text("instance_id")
+        .notNull()
+        .references(() => instances.id, { onDelete: "cascade" }),
+    jid: text("jid").notNull(),
+    name: text("name"),
+    isGroup: integer("is_group").notNull().default(0),
+    unreadCount: integer("unread_count").notNull().default(0),
+    isPinned: integer("is_pinned").notNull().default(0),
+    isMuted: integer("is_muted").notNull().default(0),
+    muteUntil: integer("mute_until"),
+    isArchived: integer("is_archived").notNull().default(0),
+    lastMessageId: text("last_message_id"),
+    lastMessageAt: integer("last_message_at"),
+    updatedAt: integer("updated_at").notNull(),
+}, (table) => [
+    primaryKey({ columns: [table.instanceId, table.jid] }),
+    index("idx_chats_recent").on(table.instanceId, table.lastMessageAt),
+]);
+// ============================================================
+// VERSION CACHE
+// WhatsApp Web version auto-update (daily check)
+// ============================================================
+export const waVersionCache = sqliteTable("wa_version_cache", {
+    id: integer("id").primaryKey().default(1),
+    versionJson: text("version_json").notNull(),
+    fetchedAt: integer("fetched_at").notNull(),
+    isLatest: integer("is_latest").notNull().default(1),
+});
+// ============================================================
+// QUEUE STATS
+// Track outbound message queue state for monitoring
+// ============================================================
+export const queueStats = sqliteTable("queue_stats", {
+    instanceId: text("instance_id")
+        .primaryKey()
+        .references(() => instances.id, { onDelete: "cascade" }),
+    messagesSent: integer("messages_sent").notNull().default(0),
+    messagesFailed: integer("messages_failed").notNull().default(0),
+    lastSentAt: integer("last_sent_at"),
+    rateLimitedUntil: integer("rate_limited_until"),
+});