Cognitoのユーザー登録をトリガーに、APIキーの発行とDynamoDBへの登録を行う

import boto3
import os

# 環境変数
REGION_NAME = os.environ['REGION_NAME']
DYNAMODB_TABLE = os.environ['DYNAMODB_TABLE']
REST_API_ID = os.environ['REST_API_ID']
USAGE_PLAN_ID = os.environ['USAGE_PLAN_ID']

dynamodb = boto3.resource('dynamodb', region_name=REGION_NAME)
table = dynamodb.Table(DYNAMODB_TABLE)
apigateway_cli = boto3.client('apigateway')


def lambda_handler(event, context):
    # ユーザー名をキーとする
    user_name = event['userName']

    # APIキーを発行
    result = apigateway_cli.create_api_key(
        name='fm_mail_free_' + user_name,
        enabled=True,
        stageKeys=[
            {
                'restApiId': REST_API_ID,
                'stageName': 'api'
            }
        ]
    )

    # 発行したAPIキーの値とIDを取得
    api_key = result['value']
    api_key_id = result['id']

    # APIキーに使用量プランを適用
    apigateway_cli.create_usage_plan_key(
        usagePlanId=USAGE_PLAN_ID,
        keyId=api_key_id,
        keyType='API_KEY'
    )

    # DynamoDBにAPIキーを登録
    with table.batch_writer() as batch:
        batch.put_item(Item={"UserID": user_name, "Type": "FREE", "ApiKey": api_key})

    # eventを返さないとCognito側で「Unrecognizable lambda output」というエラーになる
    return event

# 環境変数
REGION_NAME = os.environ['REGION_NAME']
DYNAMODB_TABLE = os.environ['DYNAMODB_TABLE']
REST_API_ID = os.environ['REST_API_ID']
USAGE_PLAN_ID = os.environ['USAGE_PLAN_ID']

    # APIキーを発行
    result = apigateway_cli.create_api_key(
        name='fm_mail_free_' + user_name,
        enabled=True,
        stageKeys=[
            {
                'restApiId': REST_API_ID,
                'stageName': 'api'
            }
        ]
    )

    # APIキーに使用量プランを適用
    apigateway_cli.create_usage_plan_key(
        usagePlanId=USAGE_PLAN_ID,
        keyId=api_key_id,
        keyType='API_KEY'
    )

    # DynamoDBにAPIキーを登録
    with table.batch_writer() as batch:
        batch.put_item(Item={"UserID": user_name, "Type": "FREE", "ApiKey": api_key})

    # eventを返さないとCognito側で「Unrecognizable lambda output」というエラーになる
    return event

import boto3
from boto3.dynamodb.conditions import Key
import json
import os
from chalice import Chalice, CognitoUserPoolAuthorizer

app = Chalice(app_name='dynamodb_api')

# 環境変数
USER_POOL_ARN = os.environ.get('USER_POOL_ARN')
USER_POOL_NAME = os.environ.get('USER_POOL_NAME')
DYNAMODB_TABLE = os.environ.get('DYNAMODB_TABLE')
REGION_NAME = os.environ.get('REGION_NAME')

# Cognitoで認証する
authorizer = CognitoUserPoolAuthorizer(
    USER_POOL_NAME,
    provider_arns=[USER_POOL_ARN]
)

# DynamoDBに接続
dynamodb = boto3.resource('dynamodb', region_name=REGION_NAME)
table = dynamodb.Table(DYNAMODB_TABLE)

@app.route('/apikey', authorizer=authorizer, cors=True)
def get_my_api_key():
    # 認証情報からUserNameを取り出す
    context = app.current_request.context
    user_name = context['authorizer']['claims']['cognito:username']

    # DynamoDBからユーザーに紐づくAPIキーの情報を取り出す
    result = table.query(KeyConditionExpression=Key('UserID').eq(user_name))

    resp = {
        'status': 'OK',
        'result': result,
        'UserName': user_name
    }

    # 結果を返す
    return json.dumps(resp, ensure_ascii=False)

    # 認証情報からUserNameを取り出す
    context = app.current_request.context
    user_name = context['authorizer']['claims']['cognito:username']

    # DynamoDBからユーザーに紐づくAPIキーの情報を取り出す
    result = table.query(KeyConditionExpression=Key('UserID').eq(user_name))

    # 認証情報からUserIDを取り出す
    context = app.current_request.context
    user_id = context['authorizer']['claims']['identities'][0]['userId']

chalice deploy

curl -X GET \
-H 'Content-Type:application/json' \
-H 'Authorization:Bearer {CognitoのidToken}' \
https://xxxxxxxxxxxxxxxxxxxxxxxx.amazonaws.com/api/apikey

import { useEffect, useState, VFC } from 'react';
import { useAtom } from 'jotai';
import ky from 'ky';
import stateCurrentUser from '../atom/User';
import Spinner from './Spinner';
import Image1 from '../svg/undraw_content_team_3epn.svg';
import Image2 from '../svg/undraw_real_time_analytics_re_yliv.svg';
import Image3 from '../svg/undraw_react_y-7-wq.svg';
import Image4 from '../svg/undraw_remotely_-2-j6y.svg';
import APIKeyList from './APIKeyList';

type ApiKeyInfo = {
  Type: string;
  ApiKey: string;
};
type Result = { Items: ApiKeyInfo[] };
type Resp = { result: Result };

// 表示するコンテンツが少ないので画像を表示
// 同じ画像ではつまらないので、表示するたびにランダムに切り替える
const randomImage = [Image1, Image2, Image3, Image4][
  Math.floor(Math.random() * 4)
];

const APIKey: VFC = () => {
  // サインイン中のユーザー情報
  const [user] = useAtom(stateCurrentUser);

  // 読込中フラグ
  const [isLoading, setIsLoading] = useState<boolean>(true);

  // APIキー
  const [apiKeys, setApiKeys] = useState<ApiKeyInfo[] | null>(null);

  // DynamoDBからAPIキーを取得する
  useEffect(() => {
    // awaitを扱うため、いったん非同期関数を作ってから呼び出している
    const getApiKeys = async () => {
      try {
        if (user) {
          // Lambda経由でDynamoDBにアクセスする
          const res: Resp = await ky
            .get(
              ' https://xxxxxxxxxxxxxxxxxxxxxxxx.amazonaws.com/api/apikey',
              {
                headers: {
                  Authorization: `Bearer ${user.signInUserSession.idToken.jwtToken}`,
                },
              },
            )
            .json();
          if (res.result.Items) setApiKeys(res.result.Items);
        }
      } catch (e) {
        // APIキー取得に失敗したらnullをセット
        setApiKeys(null);
      }
    };

    // Promiseを無視して呼び出すことを明示するためvoidを付けている
    void getApiKeys();
  }, [user]);

  // APIキーを取得できたらローディング表示をやめる
  useEffect(() => {
    if (apiKeys) setIsLoading(false);
  }, [apiKeys]);

  // ローディング表示
  if (isLoading) {
    return (
      <section className="bg-white py-6 sm:py-8 lg:py-12">
        <div className="mx-auto max-w-screen-md px-4 md:px-8">
          <h1 className="mb-8 text-4xl font-bold">APIキーの確認</h1>
          <p className="mb-4">ご利用可能なAPIキーはこちらです。</p>
          <Spinner />
          <div className="w-5/6 md:w-1/2 lg:w-full lg:max-w-lg">
            <img
              className="rounded object-cover object-center"
              src={randomImage}
              alt="APIキーの確認"
            />
          </div>
        </div>
      </section>
    );
  }

  return (
    <section className="bg-white py-6 sm:py-8 lg:py-12">
      <div className="mx-auto max-w-screen-md px-4 md:px-8">
        <h1 className="mb-8 text-4xl font-bold">APIキーの確認</h1>
        <p className="mb-4">ご利用可能なAPIキーはこちらです。</p>
        <table className="mb-8 rounded-lg bg-white p-4 shadow">
          <thead>
            <tr>
              <th className="dark:border-dark-5 whitespace-nowrap border-b-2 p-4 font-normal text-gray-900">
                No.
              </th>
              <th className="dark:border-dark-5 whitespace-nowrap border-b-2 p-4 font-normal text-gray-900">
                プラン種別
              </th>
              <th className="dark:border-dark-5 whitespace-nowrap border-b-2 p-4 font-normal text-gray-900">
                APIキー
              </th>
              <th className="dark:border-dark-5 whitespace-nowrap border-b-2 p-4 font-normal text-gray-900">
                コピー
              </th>
            </tr>
          </thead>
          <tbody>
            {apiKeys &&
              apiKeys.map((item: ApiKeyInfo, index: number) => (
                <APIKeyList item={item} index={index} key={item.ApiKey} />
              ))}
          </tbody>
        </table>
        <div className="w-5/6 md:w-1/2 lg:w-full lg:max-w-lg">
          <img
            className="rounded object-cover object-center"
            src={randomImage}
            alt="APIキーの確認"
          />
        </div>
      </div>
    </section>
  );
};
export default APIKey;