Twilio Functions + Momento で実現する、日ごとの SMS 送信 API 呼び出し制限の導入例

// /send-sms
const { CacheClient, Configurations, CredentialProvider } = require('@gomomento/sdk');

exports.handler = async function(context, event, callback) {
  try {
    // Private Asset から Momento Auth Token を読み込み
    const openTokenFile = Runtime.getAssets()['/momento-token'].open;
    const authToken = openTokenFile().trim();

    // Momento クライアントの初期化
    const cacheClient = new CacheClient({
      configuration: Configurations.Laptop.v1(),
      credentialProvider: CredentialProvider.fromString({
        authToken: authToken
      }),
      defaultTtlSeconds: 25 * 60 * 60 // 25時間（日付変更を確実にカバー）
    });

    const cacheName = 'sms-limit';

    // 今日の日付をキーとして使用
    const today = new Date().toISOString().split('T')[0]; // YYYY-MM-DD 形式
    const countKey = `sms_count_${today}`;

    // 現在の送信回数を取得
    const currentCountResponse = await cacheClient.get(cacheName, countKey);
    let currentCount = 0;

    if (currentCountResponse.type === 'Hit') {
      currentCount = parseInt(currentCountResponse.valueString()) || 0;
    }

    // 送信制限をチェック（今回は1日1件）
    const dailyLimit = 1;
    if (currentCount >= dailyLimit) {
      const response = new Twilio.Response();
      response.setStatusCode(429);
      response.setBody(JSON.stringify({
        success: false,
        error: 'Daily SMS limit exceeded',
        currentCount: currentCount,
        limit: dailyLimit,
        resetTime: `${today}T23:59:59Z`
      }));
      response.appendHeader('Content-Type', 'application/json');
      return callback(null, response);
    }

    // リクエストパラメータの取得
    const { to, body: messageBody } = event;

    if (!to || !messageBody) {
      const response = new Twilio.Response();
      response.setStatusCode(400);
      response.setBody(JSON.stringify({
        success: false,
        error: 'Missing required parameters: to, body'
      }));
      response.appendHeader('Content-Type', 'application/json');
      return callback(null, response);
    }

    // SMS 送信
    const client = context.getTwilioClient();
    const message = await client.messages.create({
      body: messageBody,
      from: context.TWILIO_PHONE_NUMBER, // Environment Variables で設定
      to: to
    });

    // 送信成功後、カウンターをインクリメント
    await cacheClient.set(cacheName, countKey, (currentCount + 1).toString());

    const response = new Twilio.Response();
    response.setStatusCode(200);
    response.setBody(JSON.stringify({
      success: true,
      messageSid: message.sid,
      currentCount: currentCount + 1,
      remainingCount: dailyLimit - (currentCount + 1)
    }));
    response.appendHeader('Content-Type', 'application/json');

    callback(null, response);

  } catch (error) {
    console.error('Error:', error);

    const response = new Twilio.Response();
    response.setStatusCode(500);
    response.setBody(JSON.stringify({
      success: false,
      error: error.message
    }));
    response.appendHeader('Content-Type', 'application/json');

    callback(null, response);
  }
};

// test-sms-limit.js
const axios = require('axios');
require('dotenv').config();

const functionUrl = process.env.FUNCTION_ENDPOINT;

async function testWithAxios() {
    console.log('=== SMS 送信制限テスト ===\n');

    const testParams = {
        to: process.env.TO_NUMBER,
        body: 'テストメッセージです'
    };

    console.log('送信パラメータ:', testParams);
    console.log('Function URL:', functionUrl);

    try {
        console.log('1回目の送信テスト...');

        const response1 = await axios.post(functionUrl, testParams, {
            headers: {
                'Content-Type': 'application/x-www-form-urlencoded'
            }
        });

        console.log('✅ 1回目の送信結果:');
        console.log('Status:', response1.status);
        console.log('Data:', typeof response1.data === 'string' ? JSON.parse(response1.data) : response1.data);
        console.log('');

        // 少し待機
        await new Promise(resolve => setTimeout(resolve, 1000));

        console.log('2回目の送信テスト...');

        const response2 = await axios.post(functionUrl, testParams, {
            headers: {
                'Content-Type': 'application/x-www-form-urlencoded'
            }
        });

        console.log('2回目の送信結果:');
        console.log('Status:', response2.status);
        console.log('Data:', typeof response2.data === 'string' ? JSON.parse(response2.data) : response2.data);

    } catch (error) {
        if (error.response) {
            console.log('HTTP エラー:');
            console.log('Status:', error.response.status);

            let errorData = error.response.data;
            if (typeof errorData === 'string') {
                try {
                    errorData = JSON.parse(errorData);
                } catch (e) {
                    // パースできない場合はそのまま
                }
            }

            console.log('Data:', errorData);

            if (error.response.status === 429) {
                console.log('✅ 制限エラーが正常に発生');
            }
        } else {
            console.error('❌ 予期しないエラー:', error.message);
        }
    }

    console.log('\n=== テスト完了 ===');
}

if (require.main === module) {
    testWithAxios().catch(console.error);
}