stateManager.js

// stateManager.js
//
// Projects are persisted in Postgres and mirrored in an in-memory cache for
// synchronous reads. Mutations must call `updateProject()` to persist.
//
// User states (the conversational step a user is on) are in-memory only —
// losing them on restart just means the user re-enters whatever multi-step
// flow they were in the middle of. Acceptable trade-off for simplicity.

const db = require('./db.js');

// ──────────────────────────────────────────────────────────────────────────
// In-memory caches
// ──────────────────────────────────────────────────────────────────────────
let projectsCache = new Map();
const userStates = new Map();
let initialized = false;

// ──────────────────────────────────────────────────────────────────────────
// Initialization — call once at startup
// ──────────────────────────────────────────────────────────────────────────
async function init() {
  // Ensure schema exists BEFORE we try to query it (handles fresh databases).
  await db.createTables();
  projectsCache = await db.loadAllProjects();
  initialized = true;
  console.log(`📦 Loaded ${projectsCache.size} project(s) from database`);
}

function assertInitialized() {
  if (!initialized) {
    throw new Error('stateManager.init() must be called before use');
  }
}

// ──────────────────────────────────────────────────────────────────────────
// Project API
// ──────────────────────────────────────────────────────────────────────────
async function createProject(projectName, adminChatId) {
  assertInitialized();
  if (projectsCache.has(projectName)) {
    throw new Error(`Project "${projectName}" already exists.`);
  }
  const project = {
    projectName,
    adminChatId,
    adminPassword: null,
    websites: [],
  };
  await db.upsertProject(projectName, project);
  projectsCache.set(projectName, project);
  return project;
}

function getProject(projectName) {
  return projectsCache.get(projectName);
}

async function updateProject(projectName, updatedProject) {
  // Ensure the project carries its own name so callers can save-back without
  // tracking the key separately.
  updatedProject.projectName = updatedProject.projectName || projectName;
  await db.upsertProject(projectName, updatedProject);
  projectsCache.set(projectName, updatedProject);
}

async function deleteProject(projectName) {
  await db.deleteProjectRow(projectName);
  projectsCache.delete(projectName);
}

// ──────────────────────────────────────────────────────────────────────────
// User state API (in-memory only, no async needed)
// ──────────────────────────────────────────────────────────────────────────
function setUserState(chatId, state) {
  userStates.set(chatId, state);
}
function getUserState(chatId) {
  return userStates.get(chatId);
}
function deleteUserState(chatId) {
  userStates.delete(chatId);
}

// ──────────────────────────────────────────────────────────────────────────
// Module exports
// ──────────────────────────────────────────────────────────────────────────
module.exports = {
  init,
  // project methods
  createProject,
  getProject,
  updateProject,
  deleteProject,
  // user state methods
  setUserState,
  getUserState,
  deleteUserState,
  // direct cache access (for legacy callers that iterate / check existence)
  get projectsByName() {
    return projectsCache;
  },
  userStates,
};