web-agent-bridge 2.3.0 → 2.4.0

package/server/migrations/003_ads_integer_cents.sql

@@ -0,0 +1,33 @@
+-- Migration 003: Convert ads financial columns from REAL to INTEGER (cents)
+-- This avoids floating-point precision issues in billing calculations.
+--
+-- NOTE: The wab_ads table in db.js now creates with INTEGER columns directly.
+-- This migration only matters for databases created before this change.
+-- On fresh databases, db.js already has the correct schema, so this is a no-op.
+-- On existing databases, this migration was already applied.
+
+-- Ensure the table and index exist (idempotent)
+CREATE TABLE IF NOT EXISTS wab_ads (
+    id TEXT PRIMARY KEY,
+    title TEXT NOT NULL,
+    description TEXT,
+    image_url TEXT,
+    target_url TEXT NOT NULL,
+    advertiser_name TEXT NOT NULL,
+    advertiser_email TEXT NOT NULL,
+    status TEXT DEFAULT 'pending' CHECK(status IN ('pending','approved','rejected','paused','expired')),
+    position TEXT DEFAULT 'new-tab' CHECK(position IN ('new-tab','sidebar','search')),
+    budget_cents INTEGER DEFAULT 0,
+    spent_cents INTEGER DEFAULT 0,
+    cpc_cents INTEGER DEFAULT 5,
+    cpi_cents INTEGER DEFAULT 1,
+    impressions INTEGER DEFAULT 0,
+    clicks INTEGER DEFAULT 0,
+    created_at TEXT DEFAULT (datetime('now')),
+    approved_by TEXT,
+    approved_at TEXT,
+    expires_at TEXT,
+    FOREIGN KEY (approved_by) REFERENCES admins(id)
+);
+
+CREATE INDEX IF NOT EXISTS idx_wab_ads_status ON wab_ads(status);

package/server/models/db.js

@@ -168,6 +168,43 @@ db.exec(`
   CREATE INDEX IF NOT EXISTS idx_stripe_subs_user ON stripe_subscriptions(user_id);
   CREATE INDEX IF NOT EXISTS idx_payments_user ON payments(user_id);
   CREATE INDEX IF NOT EXISTS idx_notifications_user ON notifications_log(user_id);
+
+  CREATE TABLE IF NOT EXISTS wab_ads (
+    id TEXT PRIMARY KEY,
+    title TEXT NOT NULL,
+    description TEXT,
+    image_url TEXT,
+    target_url TEXT NOT NULL,
+    advertiser_name TEXT NOT NULL,
+    advertiser_email TEXT NOT NULL,
+    status TEXT DEFAULT 'pending' CHECK(status IN ('pending','approved','rejected','paused','expired')),
+    position TEXT DEFAULT 'new-tab' CHECK(position IN ('new-tab','sidebar','search')),
+    budget_cents INTEGER DEFAULT 0,
+    spent_cents INTEGER DEFAULT 0,
+    cpc_cents INTEGER DEFAULT 5,
+    cpi_cents INTEGER DEFAULT 1,
+    impressions INTEGER DEFAULT 0,
+    clicks INTEGER DEFAULT 0,
+    created_at TEXT DEFAULT (datetime('now')),
+    approved_by TEXT,
+    approved_at TEXT,
+    expires_at TEXT,
+    FOREIGN KEY (approved_by) REFERENCES admins(id)
+  );
+
+  CREATE TABLE IF NOT EXISTS ad_events (
+    id INTEGER PRIMARY KEY AUTOINCREMENT,
+    ad_id TEXT NOT NULL,
+    event_type TEXT NOT NULL CHECK(event_type IN ('impression','click')),
+    platform TEXT DEFAULT 'browser',
+    ip_hash TEXT,
+    created_at TEXT DEFAULT (datetime('now')),
+    FOREIGN KEY (ad_id) REFERENCES wab_ads(id) ON DELETE CASCADE
+  );
+
+  CREATE INDEX IF NOT EXISTS idx_wab_ads_status ON wab_ads(status);
+  CREATE INDEX IF NOT EXISTS idx_ad_events_ad ON ad_events(ad_id);
+  CREATE INDEX IF NOT EXISTS idx_ad_events_created ON ad_events(created_at);
 `);
 
 function generateLicenseKey() {
@@ -218,6 +255,7 @@ const findSitesByUser = db.prepare(`SELECT * FROM sites WHERE user_id = ? ORDER
 const findSiteById = db.prepare(`SELECT * FROM sites WHERE id = ?`);
 const findSiteByLicense = db.prepare(`SELECT * FROM sites WHERE license_key = ? AND active = 1`);
 const findSiteByDomainAndLicense = db.prepare(`SELECT * FROM sites WHERE domain = ? AND license_key = ? AND active = 1`);
+const findSiteByDomain = db.prepare(`SELECT * FROM sites WHERE domain = ? AND active = 1 LIMIT 1`);
 const updateSiteConfig = db.prepare(`UPDATE sites SET config = ?, updated_at = datetime('now') WHERE id = ? AND user_id = ?`);
 const updateSiteTier = db.prepare(`UPDATE sites SET tier = ?, updated_at = datetime('now') WHERE id = ? AND user_id = ?`);
 const deleteSite = db.prepare(`UPDATE sites SET active = 0, updated_at = datetime('now') WHERE id = ? AND user_id = ?`);
@@ -506,6 +544,77 @@ function setPlatformSetting(key, value) {
   db.prepare(`INSERT OR REPLACE INTO platform_settings (key, value, updated_at) VALUES (?, ?, datetime('now'))`).run(key, value);
 }
 
+// ─── Ads Operations ──────────────────────────────────────────────────
+function submitAd({ title, description, imageUrl, targetUrl, advertiserName, advertiserEmail, position, budgetCents, cpcCents, cpiCents, expiresAt }) {
+  const id = uuidv4();
+  db.prepare(`INSERT INTO wab_ads (id, title, description, image_url, target_url, advertiser_name, advertiser_email, position, budget_cents, cpc_cents, cpi_cents, expires_at) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`).run(id, title, description || '', imageUrl || '', targetUrl, advertiserName, advertiserEmail, position || 'new-tab', budgetCents || 0, cpcCents || 5, cpiCents || 1, expiresAt || null);
+  return { id, title, advertiserName, status: 'pending' };
+}
+
+function getActiveAds(position) {
+  let q = `SELECT id, title, description, image_url, target_url, advertiser_name, position FROM wab_ads WHERE status = 'approved' AND (expires_at IS NULL OR expires_at > datetime('now')) AND (budget_cents <= 0 OR spent_cents < budget_cents)`;
+  const params = [];
+  if (position) { q += ` AND position = ?`; params.push(position); }
+  q += ` ORDER BY created_at DESC LIMIT 10`;
+  return db.prepare(q).all(...params);
+}
+
+function getAllAds() {
+  return db.prepare(`SELECT * FROM wab_ads ORDER BY created_at DESC`).all();
+}
+
+function getPendingAds() {
+  return db.prepare(`SELECT * FROM wab_ads WHERE status = 'pending' ORDER BY created_at ASC`).all();
+}
+
+function getAdById(id) {
+  return db.prepare(`SELECT * FROM wab_ads WHERE id = ?`).get(id);
+}
+
+function updateAdStatus(id, status, adminId) {
+  const sets = ['status = ?'];
+  const params = [status];
+  if (status === 'approved') {
+    sets.push('approved_by = ?', 'approved_at = datetime(\'now\')');
+    params.push(adminId);
+  }
+  params.push(id);
+  db.prepare(`UPDATE wab_ads SET ${sets.join(', ')} WHERE id = ?`).run(...params);
+}
+
+function deleteAd(id) {
+  db.prepare(`DELETE FROM ad_events WHERE ad_id = ?`).run(id);
+  db.prepare(`DELETE FROM wab_ads WHERE id = ?`).run(id);
+}
+
+function recordAdEvent(adId, eventType, ipHash) {
+  // Deduplicate: skip if same ip+ad+event in last 60s
+  const recent = db.prepare(`SELECT 1 FROM ad_events WHERE ad_id = ? AND event_type = ? AND ip_hash = ? AND created_at > datetime('now', '-60 seconds') LIMIT 1`).get(adId, eventType, ipHash || '');
+  if (recent) return;
+  db.prepare(`INSERT INTO ad_events (ad_id, event_type, ip_hash) VALUES (?, ?, ?)`).run(adId, eventType, ipHash || null);
+  if (eventType === 'click') {
+    const ad = db.prepare(`SELECT cpc_cents FROM wab_ads WHERE id = ?`).get(adId);
+    if (ad) {
+      db.prepare(`UPDATE wab_ads SET clicks = clicks + 1, spent_cents = spent_cents + ? WHERE id = ?`).run(ad.cpc_cents, adId);
+    }
+  } else {
+    const ad = db.prepare(`SELECT cpi_cents FROM wab_ads WHERE id = ?`).get(adId);
+    if (ad) {
+      db.prepare(`UPDATE wab_ads SET impressions = impressions + 1, spent_cents = spent_cents + ? WHERE id = ?`).run(ad.cpi_cents, adId);
+    }
+  }
+}
+
+function getAdStats() {
+  const total = db.prepare(`SELECT COUNT(*) as c FROM wab_ads`).get().c;
+  const pending = db.prepare(`SELECT COUNT(*) as c FROM wab_ads WHERE status = 'pending'`).get().c;
+  const approved = db.prepare(`SELECT COUNT(*) as c FROM wab_ads WHERE status = 'approved'`).get().c;
+  const totalImpressions = db.prepare(`SELECT COALESCE(SUM(impressions), 0) as c FROM wab_ads`).get().c;
+  const totalClicks = db.prepare(`SELECT COALESCE(SUM(clicks), 0) as c FROM wab_ads`).get().c;
+  const totalRevenueCents = db.prepare(`SELECT COALESCE(SUM(spent_cents), 0) as c FROM wab_ads`).get().c;
+  return { total, pending, approved, totalImpressions, totalClicks, totalRevenueCents };
+}
+
 module.exports = {
   db,
   registerUser,
@@ -516,6 +625,7 @@ module.exports = {
   findSitesByUser,
   findSiteById,
   findSiteByLicense,
+  findSiteByDomain,
   updateSiteConfig,
   updateSiteTier,
   deleteSite,
@@ -557,5 +667,15 @@ module.exports = {
   getNotificationLogs,
   // Platform
   getPlatformSetting,
-  setPlatformSetting
+  setPlatformSetting,
+  // Ads
+  submitAd,
+  getActiveAds,
+  getAllAds,
+  getPendingAds,
+  getAdById,
+  updateAdStatus,
+  deleteAd,
+  recordAdEvent,
+  getAdStats
 };