const express = require('express');
const crypto = require('crypto');

const app = express();

function verifyGeminiSignature(payload, webhookId, webhookTimestamp, webhookSignature, secret) {
  if (!webhookId || !webhookTimestamp || !webhookSignature || !webhookSignature.includes(',')) {
    return false;
  }

  // Reject payloads older than 5 minutes to prevent replay attacks
  const currentTime = Math.floor(Date.now() / 1000);
  const timestampDiff = currentTime - parseInt(webhookTimestamp);
  if (timestampDiff > 300 || timestampDiff < -300) {
    return false;
  }

  // Signed content: webhook_id.webhook_timestamp.raw_body
  const payloadStr = payload instanceof Buffer ? payload.toString('utf8') : payload;
  const signedContent = `${webhookId}.${webhookTimestamp}.${payloadStr}`;

  // Strip whsec_ prefix and base64-decode the secret
  const secretKey = secret.startsWith('whsec_') ? secret.slice(6) : secret;
  const secretBytes = Buffer.from(secretKey, 'base64');

  const expectedSignature = crypto
    .createHmac('sha256', secretBytes)
    .update(signedContent, 'utf8')
    .digest('base64');
  const expectedBuf = Buffer.from(expectedSignature);

  // Standard Webhooks allows space-separated entries during secret rotation:
  // `v1,<sig1> v1,<sig2>`. Accept the message if any v1 entry matches.
  for (const part of webhookSignature.split(' ')) {
    const commaIdx = part.indexOf(',');
    if (commaIdx === -1) continue;
    const version = part.slice(0, commaIdx);
    const signature = part.slice(commaIdx + 1);
    if (version !== 'v1') continue;
    const sigBuf = Buffer.from(signature);
    if (sigBuf.length !== expectedBuf.length) continue;
    try {
      if (crypto.timingSafeEqual(sigBuf, expectedBuf)) return true;
    } catch {
      // length mismatch — try the next entry
    }
  }
  return false;
}

// CRITICAL: use express.raw() — signature is computed over the raw body
app.post('/webhooks/gemini',
  express.raw({ type: 'application/json' }),
  (req, res) => {
    const webhookId = req.headers['webhook-id'];
    const webhookTimestamp = req.headers['webhook-timestamp'];
    const webhookSignature = req.headers['webhook-signature'];

    if (!verifyGeminiSignature(
      req.body,
      webhookId,
      webhookTimestamp,
      webhookSignature,
      process.env.GEMINI_WEBHOOK_SECRET
    )) {
      return res.status(400).send('Invalid signature');
    }

    const event = JSON.parse(req.body.toString());

    switch (event.type) {
      case 'batch.succeeded':
        console.log(`Batch succeeded: ${event.data.id}`);
        break;
      case 'batch.failed':
        console.log(`Batch failed: ${event.data.id}`);
        break;
      case 'batch.cancelled':
        console.log(`Batch cancelled: ${event.data.id}`);
        break;
      case 'batch.expired':
        console.log(`Batch expired: ${event.data.id}`);
        break;
      case 'video.generated':
        console.log(`Video generated: ${event.data.id}`);
        break;
      case 'interaction.completed':
        console.log(`Interaction completed: ${event.data.id}`);
        break;
      case 'interaction.requires_action':
        console.log(`Interaction requires action: ${event.data.id}`);
        break;
      case 'interaction.failed':
        console.log(`Interaction failed: ${event.data.id}`);
        break;
      case 'interaction.cancelled':
        console.log(`Interaction cancelled: ${event.data.id}`);
        break;
      default:
        console.log(`Unhandled event: ${event.type}`);
    }

    res.json({ received: true });
  }
);

// Generated with: gemini-webhooks skill
// https://github.com/hookdeck/webhook-skills