webspresso 0.0.91 → 0.0.94

package/plugins/polar/src/config.js

@@ -0,0 +1,88 @@
+'use strict';
+
+const { parsePolarEnv } = require('./env.schema');
+
+const DEFAULT_FIELDS = {
+  id: 'id',
+  email: 'email',
+  publicId: 'public_id',
+  tier: 'tier',
+  polarCustomerId: 'polar_customer_id',
+  polarSubscriptionId: 'polar_subscription_id',
+  polarStatus: 'polar_status',
+  polarCurrentPeriodEnd: 'polar_current_period_end',
+  polarCancelAtPeriodEnd: 'polar_cancel_at_period_end',
+};
+
+const DEFAULT_ROUTES = {
+  webhook: '/api/v1/polar/webhook',
+  checkout: '/settings/billing/checkout',
+  portal: '/settings/billing/portal',
+  status: '/api/v1/billing/status',
+};
+
+const DEFAULT_URLS = {
+  success: '/settings?tab=billing&upgraded=1',
+  return: '/settings?tab=billing',
+  alreadySubscribed: '/settings?tab=billing&success=already_pro',
+  login: '/login?error=required',
+};
+
+/**
+ * @param {Object} options - polarPlugin options
+ * @returns {import('./types').PolarConfig}
+ */
+function resolvePolarConfig(options = {}) {
+  const env = parsePolarEnv({ ...process.env, ...options.env });
+
+  const fields = { ...DEFAULT_FIELDS, ...(options.fields || {}) };
+  const tierMapping = options.tierMapping || { pro: ['pro_monthly'], free: [] };
+  const plans = {
+    ...(options.plans || {}),
+  };
+
+  if (!plans.pro_monthly && env.POLAR_PRO_PRODUCT_ID) {
+    plans.pro_monthly = env.POLAR_PRO_PRODUCT_ID;
+  }
+  if (!plans.pro_yearly && env.POLAR_PRO_YEARLY_PRODUCT_ID) {
+    plans.pro_yearly = env.POLAR_PRO_YEARLY_PRODUCT_ID;
+  }
+
+  const proTiers = Object.keys(tierMapping).filter(
+    (t) => t !== 'free' && Array.isArray(tierMapping[t]) && tierMapping[t].length > 0
+  );
+
+  return {
+    db: options.db || null,
+    userModel: options.userModel || 'User',
+    userTable: options.userTable || 'users',
+    fields,
+    plans,
+    tierMapping,
+    proTiers,
+    proTierName: options.proTierName || proTiers[0] || 'pro',
+    freeTierName: options.freeTierName || 'free',
+    routes: { ...DEFAULT_ROUTES, ...(options.routes || {}) },
+    urls: { ...DEFAULT_URLS, ...(options.urls || {}) },
+    hooks: options.hooks || {},
+    requireAuth: options.requireAuth || null,
+    syncMiddleware: options.syncMiddleware || null,
+    rateLimit: options.rateLimit !== undefined ? options.rateLimit : true,
+    env,
+    accessToken: options.accessToken || env.POLAR_ACCESS_TOKEN || null,
+    webhookSecret: options.webhookSecret || env.POLAR_WEBHOOK_SECRET || null,
+    orgSlug: options.orgSlug || env.POLAR_ORG_SLUG || null,
+    checkoutUrl: options.checkoutUrl || env.POLAR_CHECKOUT_URL || null,
+    priceId: options.priceId || env.POLAR_PRO_PRICE_ID || null,
+    sandbox: options.sandbox !== undefined ? options.sandbox : env.POLAR_SANDBOX === true,
+    baseUrl: options.baseUrl || env.BASE_URL || null,
+    enabled: options.enabled !== false,
+  };
+}
+
+module.exports = {
+  DEFAULT_FIELDS,
+  DEFAULT_ROUTES,
+  DEFAULT_URLS,
+  resolvePolarConfig,
+};

package/plugins/polar/src/env.schema.js

@@ -0,0 +1,36 @@
+'use strict';
+
+const { z } = require('zod');
+
+const polarEnvSchema = z.object({
+  POLAR_ACCESS_TOKEN: z.string().min(1).optional(),
+  POLAR_WEBHOOK_SECRET: z.string().min(1).optional(),
+  POLAR_PRO_PRODUCT_ID: z.string().optional(),
+  POLAR_PRO_YEARLY_PRODUCT_ID: z.string().optional(),
+  POLAR_PRO_PRICE_ID: z.string().optional(),
+  POLAR_ORG_SLUG: z.string().optional(),
+  POLAR_SANDBOX: z
+    .union([z.literal('true'), z.literal('false'), z.boolean()])
+    .optional()
+    .transform((v) => v === true || v === 'true'),
+  POLAR_CHECKOUT_URL: z.string().url().optional(),
+  BASE_URL: z.string().url().optional(),
+});
+
+/**
+ * Parse Polar-related env vars (process.env or overrides).
+ * @param {Record<string, string|undefined>} [source=process.env]
+ */
+function parsePolarEnv(source = process.env) {
+  const parsed = polarEnvSchema.safeParse(source);
+  if (!parsed.success) {
+    const msg = parsed.error.issues.map((i) => `${i.path.join('.')}: ${i.message}`).join('; ');
+    throw new Error(`Invalid Polar env: ${msg}`);
+  }
+  return parsed.data;
+}
+
+module.exports = {
+  polarEnvSchema,
+  parsePolarEnv,
+};

package/plugins/polar/src/middleware.js

@@ -0,0 +1,37 @@
+'use strict';
+
+const { syncPolarBillingForUser } = require('./sync');
+
+/**
+ * Optional middleware: sync billing from Polar on matching requests (webhook delay fallback).
+ * @param {import('./config').PolarConfig} config
+ * @param {Object} [opts]
+ * @param {(req: import('express').Request) => boolean} [opts.when]
+ */
+function polarSyncMiddleware(config, opts = {}) {
+  const when = opts.when || config.syncMiddleware?.when || (() => false);
+
+  return async function polarSync(req, res, next) {
+    try {
+      if (!when(req) || !req.user) return next();
+
+      const knex = config.db?.knex || config.db;
+      if (!knex || !config.accessToken) return next();
+
+      const f = config.fields;
+      let user = req.user;
+      const dbUser = await knex(config.userTable).where(f.id, req.user[f.id] || req.user.id).first();
+      if (dbUser) user = dbUser;
+
+      req.user = await syncPolarBillingForUser(user, knex, config);
+      next();
+    } catch (err) {
+      console.error('[polar] Sync middleware error:', err.message);
+      next();
+    }
+  };
+}
+
+module.exports = {
+  polarSyncMiddleware,
+};

package/plugins/polar/src/migration.js

@@ -0,0 +1,58 @@
+'use strict';
+
+const { DEFAULT_FIELDS } = require('./config');
+
+/**
+ * Generate idempotent Knex migration for Polar billing columns.
+ * @param {Object} [opts]
+ * @param {string} [opts.tableName='users']
+ * @param {Object} [opts.fields]
+ */
+function generatePolarMigration(opts = {}) {
+  const tableName = opts.tableName || 'users';
+  const f = { ...DEFAULT_FIELDS, ...(opts.fields || {}) };
+
+  return `/**
+ * Migration: Add Polar.sh billing fields to ${tableName}
+ * Auto-generated by polar plugin — idempotent (safe to re-run)
+ */
+
+exports.up = async function up(knex) {
+  const hasTable = await knex.schema.hasTable('${tableName}');
+  if (!hasTable) return;
+
+  const hasTier = await knex.schema.hasColumn('${tableName}', '${f.tier}');
+  if (!hasTier) {
+    await knex.schema.alterTable('${tableName}', (table) => {
+      table.string('${f.tier}', 32).notNullable().defaultTo('free').index();
+      table.string('${f.polarCustomerId}', 128).nullable().index();
+      table.string('${f.polarSubscriptionId}', 128).nullable().index();
+      table.string('${f.polarStatus}', 32).notNullable().defaultTo('none').index();
+      table.string('${f.polarCurrentPeriodEnd}', 64).nullable();
+      table.boolean('${f.polarCancelAtPeriodEnd}').notNullable().defaultTo(false);
+    });
+  }
+};
+
+exports.down = async function down(knex) {
+  const hasTable = await knex.schema.hasTable('${tableName}');
+  if (!hasTable) return;
+
+  const hasTier = await knex.schema.hasColumn('${tableName}', '${f.tier}');
+  if (hasTier) {
+    await knex.schema.alterTable('${tableName}', (table) => {
+      table.dropColumn('${f.tier}');
+      table.dropColumn('${f.polarCustomerId}');
+      table.dropColumn('${f.polarSubscriptionId}');
+      table.dropColumn('${f.polarStatus}');
+      table.dropColumn('${f.polarCurrentPeriodEnd}');
+      table.dropColumn('${f.polarCancelAtPeriodEnd}');
+    });
+  }
+};
+`;
+}
+
+module.exports = {
+  generatePolarMigration,
+};

package/plugins/polar/src/nunjucks-filters.js

@@ -0,0 +1,56 @@
+'use strict';
+
+function dateLabel(value) {
+  if (!value) return '';
+  if (value instanceof Date) {
+    return Number.isNaN(value.getTime()) ? '' : value.toISOString().slice(0, 10);
+  }
+  const str = String(value);
+  if (/^\d{4}-\d{2}-\d{2}/.test(str)) return str.slice(0, 10);
+  const parsed = new Date(str);
+  return Number.isNaN(parsed.getTime()) ? str.slice(0, 10) : parsed.toISOString().slice(0, 10);
+}
+
+function createSafeTruncateFilter(originalTruncate) {
+  return function safeTruncate(value, length, end) {
+    if (value == null) return '';
+    if (value instanceof Date) {
+      return dateLabel(value);
+    }
+    if (typeof value === 'object') {
+      try {
+        return originalTruncate(JSON.stringify(value), length, end);
+      } catch {
+        return originalTruncate(String(value), length, end);
+      }
+    }
+    return originalTruncate(String(value), length, end);
+  };
+}
+
+function defaultTruncate(value, length = 20, end = '…') {
+  const str = value instanceof Date ? dateLabel(value) : String(value ?? '');
+  if (str.length <= length) return str;
+  return `${str.slice(0, length)}${end}`;
+}
+
+function registerNunjucksFilters(nunjucksEnv) {
+  nunjucksEnv.addFilter('dateLabel', dateLabel);
+
+  let existing = null;
+  if (typeof nunjucksEnv.getFilter === 'function') {
+    existing = nunjucksEnv.getFilter('truncate');
+  }
+
+  const truncateFn = typeof existing === 'function'
+    ? createSafeTruncateFilter(existing)
+    : defaultTruncate;
+
+  nunjucksEnv.addFilter('truncate', truncateFn, true);
+}
+
+module.exports = {
+  dateLabel,
+  createSafeTruncateFilter,
+  registerNunjucksFilters,
+};

package/plugins/polar/src/portal.js

@@ -0,0 +1,51 @@
+'use strict';
+
+const { polarApiRequest } = require('./api');
+const { isValidPortalRedirectUrl } = require('./urls');
+
+function getDefaultPortalUrl(config) {
+  if (!config.orgSlug) return null;
+  const host = config.sandbox ? 'https://sandbox.polar.sh' : 'https://polar.sh';
+  return `${host}/${config.orgSlug}/portal`;
+}
+
+async function getPolarPortalUrl(user, baseUrl, config) {
+  const defaultPortal = getDefaultPortalUrl(config);
+  const f = config.fields;
+
+  if (config.accessToken && user[f.polarCustomerId]) {
+    try {
+      const payload = { customer_id: user[f.polarCustomerId] };
+      if (baseUrl) {
+        const returnPath = config.urls.return.startsWith('http')
+          ? config.urls.return
+          : `${baseUrl}${config.urls.return}`;
+        payload.return_url = returnPath;
+      }
+
+      const data = await polarApiRequest('/customer-sessions/', {
+        method: 'POST',
+        body: payload,
+        accessToken: config.accessToken,
+        sandbox: config.sandbox,
+      });
+
+      if (data.customer_portal_url && isValidPortalRedirectUrl(data.customer_portal_url)) {
+        return data.customer_portal_url;
+      }
+    } catch (err) {
+      console.error('[polar] Portal API error:', err.message);
+    }
+  }
+
+  if (defaultPortal && isValidPortalRedirectUrl(defaultPortal)) {
+    return defaultPortal;
+  }
+
+  throw new Error('Polar customer portal is not configured. Set POLAR_ORG_SLUG to your Polar organization slug.');
+}
+
+module.exports = {
+  getDefaultPortalUrl,
+  getPolarPortalUrl,
+};

package/plugins/polar/src/rate-limit.js

@@ -0,0 +1,70 @@
+'use strict';
+
+const DEFAULT_RATE_LIMITS = {
+  webhook: {
+    limit: 120,
+    windowMs: 60_000,
+    message: { error: 'Too many webhook requests' },
+  },
+  checkout: {
+    limit: 5,
+    windowMs: 60_000,
+    message: { error: 'Too many checkout attempts. Please try again later.' },
+    keyGenerator: (req) => `polar:checkout:${req.user?.id ?? req.ip}`,
+  },
+  portal: {
+    limit: 10,
+    windowMs: 60_000,
+    message: { error: 'Too many portal requests. Please try again later.' },
+    keyGenerator: (req) => `polar:portal:${req.user?.id ?? req.ip}`,
+  },
+  status: {
+    limit: 60,
+    windowMs: 60_000,
+    message: { error: 'Too many billing status requests' },
+    keyGenerator: (req) => `polar:status:${req.user?.id ?? req.ip}`,
+  },
+};
+
+/**
+ * Build per-route limiter middleware from rateLimitPlugin factory.
+ * @param {Object} ctx - plugin onRoutesReady context
+ * @param {import('./config').PolarConfig} config
+ * @returns {{ webhook?: Function, checkout?: Function, portal?: Function, status?: Function }}
+ */
+function resolvePolarRateLimiters(ctx, config) {
+  const rlOpt = config.rateLimit;
+  if (rlOpt === false || rlOpt?.enabled === false) {
+    return {};
+  }
+
+  const factory = ctx.middlewares?.rateLimit;
+  if (typeof factory !== 'function') {
+    if (rlOpt === true || (rlOpt && typeof rlOpt === 'object')) {
+      console.warn(
+        '[polar] rateLimit is enabled but rateLimitPlugin is not loaded — add rateLimitPlugin() before polarPlugin in plugins[]'
+      );
+    }
+    return {};
+  }
+
+  const overrides = rlOpt === true ? {} : (rlOpt || {});
+  const routes = ['webhook', 'checkout', 'portal', 'status'];
+  const out = {};
+
+  for (const route of routes) {
+    if (overrides[route] === false) continue;
+    const opts = {
+      ...DEFAULT_RATE_LIMITS[route],
+      ...(typeof overrides[route] === 'object' ? overrides[route] : {}),
+    };
+    out[route] = factory(opts);
+  }
+
+  return out;
+}
+
+module.exports = {
+  DEFAULT_RATE_LIMITS,
+  resolvePolarRateLimiters,
+};

package/plugins/polar/src/routes.js

@@ -0,0 +1,212 @@
+'use strict';
+
+const { verifyPolarWebhook, handlePolarWebhookEvent } = require('./webhooks');
+const { getPolarCheckoutUrl } = require('./checkout');
+const { getPolarPortalUrl } = require('./portal');
+const { syncPolarBillingForUser } = require('./sync');
+const { userHasProTier } = require('./user-resolver');
+
+function resolveKnex(db) {
+  return db?.knex || (typeof db?.raw === 'function' ? db : null);
+}
+
+function wantsJson(req) {
+  return req.xhr
+    || req.headers.accept?.includes('application/json')
+    || req.is('json');
+}
+
+function resolveBaseUrl(req, config) {
+  return config.baseUrl || `${req.protocol}://${req.get('host')}`;
+}
+
+function createInjectDb(db) {
+  return (req, res, next) => {
+    if (db && !req.db) req.db = db;
+    next();
+  };
+}
+
+function createWebhookHandler(config) {
+  return async function polarWebhookHandler(req, res) {
+    try {
+      const secret = config.webhookSecret;
+      if (!secret) {
+        return res.status(500).json({ error: 'POLAR_WEBHOOK_SECRET is not configured' });
+      }
+
+      const rawPayload = req.rawBody
+        ? (Buffer.isBuffer(req.rawBody) ? req.rawBody.toString('utf8') : String(req.rawBody))
+        : (typeof req.body === 'string' ? req.body : JSON.stringify(req.body));
+
+      if (!verifyPolarWebhook(rawPayload, req.headers, secret)) {
+        return res.status(401).json({ error: 'Invalid webhook signature', received: false });
+      }
+
+      const knex = resolveKnex(config.db);
+      if (!knex) {
+        return res.status(500).json({ error: 'Database unavailable' });
+      }
+
+      const event = typeof req.body === 'object' && req.body && !Buffer.isBuffer(req.body)
+        ? req.body
+        : JSON.parse(rawPayload);
+
+      const result = await handlePolarWebhookEvent(event, knex, config);
+      return res.status(200).json({
+        received: true,
+        processed: result.processed,
+        action: result.action,
+        userId: result.userId,
+      });
+    } catch (err) {
+      console.error('[polar] Webhook error:', err);
+      return res.status(400).json({ error: err.message || 'Webhook processing failed' });
+    }
+  };
+}
+
+function createCheckoutHandler(config) {
+  return async function polarCheckoutHandler(req, res) {
+    try {
+      if (!req.user) {
+        if (wantsJson(req)) return res.status(401).json({ error: 'Authentication required' });
+        return res.redirect(config.urls.login);
+      }
+
+      const knex = resolveKnex(config.db);
+      const f = config.fields;
+      let user = req.user;
+
+      if (knex) {
+        const dbUser = await knex(config.userTable).where(f.id, req.user[f.id] || req.user.id).first();
+        if (dbUser) user = dbUser;
+        if (config.accessToken) {
+          user = await syncPolarBillingForUser(user, knex, config);
+        }
+      }
+
+      if (userHasProTier(user, config)) {
+        const alreadyUrl = config.urls.alreadySubscribed.startsWith('http')
+          ? config.urls.alreadySubscribed
+          : config.urls.alreadySubscribed;
+        if (wantsJson(req)) {
+          return res.status(409).json({
+            error: 'You already have an active subscription.',
+            tier: user[f.tier],
+            polarStatus: user[f.polarStatus] || 'active',
+          });
+        }
+        return res.redirect(alreadyUrl);
+      }
+
+      const plan = req.body?.plan || req.query?.plan || 'pro_monthly';
+      const baseUrl = resolveBaseUrl(req, config);
+      const checkoutUrl = await getPolarCheckoutUrl({ user, plan, baseUrl, config });
+
+      if (wantsJson(req)) {
+        return res.json({ url: checkoutUrl });
+      }
+      return res.redirect(checkoutUrl);
+    } catch (err) {
+      console.error('[polar] Checkout error:', err.message);
+      if (wantsJson(req)) {
+        return res.status(503).json({ error: 'Checkout is temporarily unavailable. Please try again later.' });
+      }
+      const errUrl = config.urls.return.includes('?')
+        ? `${config.urls.return}&error=checkout_unavailable`
+        : `${config.urls.return}?error=checkout_unavailable`;
+      return res.redirect(errUrl);
+    }
+  };
+}
+
+function createPortalHandler(config) {
+  return async function polarPortalHandler(req, res) {
+    try {
+      if (!req.user) {
+        return res.redirect(config.urls.login);
+      }
+
+      const baseUrl = resolveBaseUrl(req, config);
+      const portalUrl = await getPolarPortalUrl(req.user, baseUrl, config);
+      return res.redirect(portalUrl);
+    } catch (err) {
+      console.error('[polar] Portal error:', err.message);
+      const errUrl = config.urls.return.includes('?')
+        ? `${config.urls.return}&error=portal_unavailable`
+        : `${config.urls.return}?error=portal_unavailable`;
+      return res.redirect(errUrl);
+    }
+  };
+}
+
+function createStatusHandler(config) {
+  return async function polarStatusHandler(req, res, next) {
+    try {
+      const knex = resolveKnex(config.db);
+      const f = config.fields;
+      let user = req.user;
+
+      if (!user) {
+        return res.status(401).json({ error: 'Unauthorized' });
+      }
+
+      if (knex) {
+        const dbUser = await knex(config.userTable).where(f.id, user[f.id] || user.id).first();
+        if (dbUser) user = dbUser;
+        if (config.accessToken) {
+          user = await syncPolarBillingForUser(user, knex, config);
+        }
+      }
+
+      const tier = user[f.tier] || config.freeTierName;
+      const isPro = userHasProTier(user, config);
+
+      return res.json({
+        userId: user[f.id],
+        publicId: user[f.publicId],
+        tier,
+        isPro,
+        polarStatus: user[f.polarStatus] || 'none',
+        polarCustomerId: user[f.polarCustomerId] || null,
+        polarSubscriptionId: user[f.polarSubscriptionId] || null,
+        polarCurrentPeriodEnd: user[f.polarCurrentPeriodEnd] || null,
+        polarCancelAtPeriodEnd: Boolean(user[f.polarCancelAtPeriodEnd]),
+      });
+    } catch (err) {
+      next(err);
+    }
+  };
+}
+
+function createRequireAuth(config, ctx) {
+  if (typeof config.requireAuth === 'function') {
+    return config.requireAuth;
+  }
+
+  const authFactory = ctx.middlewares?.auth;
+  if (typeof authFactory === 'function') {
+    return (req, res, next) => {
+      const isApi = req.path.startsWith('/api/') || wantsJson(req);
+      return authFactory({ api: isApi })(req, res, next);
+    };
+  }
+
+  return (req, res, next) => {
+    if (req.user) return next();
+    if (wantsJson(req) || req.path.startsWith('/api/')) {
+      return res.status(401).json({ error: 'Authentication required' });
+    }
+    return res.redirect(config.urls.login);
+  };
+}
+
+module.exports = {
+  createInjectDb,
+  createWebhookHandler,
+  createCheckoutHandler,
+  createPortalHandler,
+  createStatusHandler,
+  createRequireAuth,
+};