Underconstruction: ガゲァスAdvent Calender 8日目

ガゲマックシャッフラーをセキュアにする

前回の投稿で「APIキーをヘッダに入れて叩きたいが，IFTTT側が未対応でできなかった。」という旨の投稿をした。

今回はカスタムオーソライザーを使ってAPIに認証機構をつける。

認証機構となるLambda関数を作る

認証機構となるLambda関数を作る。

コードは以下のような感じ。

# アクセストークンを適当に生成して貼り付ける
correct_access_token = '***'

def generate_policy(principalId, effect, resource):
    authResponse = {}
    authResponse['principalId'] = principalId
    
    policyDocument = {};
    policyDocument['Version'] = '2012-10-17'
    policyDocument['Statement'] = []
    
    statementOne = {}
    statementOne['Action'] = 'execute-api:Invoke'
    statementOne['Effect'] = effect
    statementOne['Resource'] = resource
    
    policyDocument['Statement'] = [statementOne]
    
    authResponse['policyDocument'] = policyDocument
    
    return authResponse

def lambda_handler(event, context):
    # TODO implement
    access_token = event['queryStringParameters']['access_token']
    if access_token == correct_access_token:
        authResponse = generate_policy('user', 'Allow', event['methodArn'])
    else:
        authResponse = generate_policy('user', 'Deny',  event['methodArn'])
    return authResponse