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 });
  }
);

import os
import hmac
import hashlib
import base64
import time
from fastapi import FastAPI, Request, HTTPException, Header

app = FastAPI()

def verify_gemini_signature(
    payload: bytes,
    webhook_id: str,
    webhook_timestamp: str,
    webhook_signature: str,
    secret: str
) -> bool:
    if not webhook_id or not webhook_timestamp or not webhook_signature or ',' not in webhook_signature:
        return False

    current_time = int(time.time())
    try:
        timestamp_diff = current_time - int(webhook_timestamp)
    except ValueError:
        return False
    if timestamp_diff > 300 or timestamp_diff < -300:
        return False

    signed_content = f"{webhook_id}.{webhook_timestamp}.{payload.decode('utf-8')}"

    secret_key = secret[6:] if secret.startswith('whsec_') else secret
    secret_bytes = base64.b64decode(secret_key)

    expected_signature = base64.b64encode(
        hmac.new(secret_bytes, signed_content.encode('utf-8'), hashlib.sha256).digest()
    ).decode('utf-8')

    # Standard Webhooks allows space-separated entries during secret rotation:
    # `v1,<sig1> v1,<sig2>`. Accept the message if any v1 entry matches.
    for part in webhook_signature.split(' '):
        if ',' not in part:
            continue
        version, _, signature = part.partition(',')
        if version != 'v1':
            continue
        if hmac.compare_digest(signature, expected_signature):
            return True
    return False


@app.post("/webhooks/gemini")
async def gemini_webhook(
    request: Request,
    webhook_id: str = Header(None, alias="webhook-id"),
    webhook_timestamp: str = Header(None, alias="webhook-timestamp"),
    webhook_signature: str = Header(None, alias="webhook-signature"),
):
    payload = await request.body()

    if not verify_gemini_signature(
        payload,
        webhook_id,
        webhook_timestamp,
        webhook_signature,
        os.environ.get("GEMINI_WEBHOOK_SECRET", "")
    ):
        raise HTTPException(status_code=400, detail="Invalid signature")

    event = await request.json()
    # Handle event...
    return {"received": True}

GEMINI_API_KEY=your-api-key                # Your Gemini API key
GEMINI_WEBHOOK_SECRET=whsec_xxxxxxxxxxxxxx # Static webhook signing secret (whsec_-prefixed)

# Tunnel localhost to a public URL Gemini can reach (no account required)
npx hookdeck-cli listen 3000 gemini --path /webhooks/gemini

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