Serverless frameworkでSlackのスラッシュコマンドを作ろう２ 〜スナップショットを収集しよう〜

npm install -g serverless serverless-step-functions serverless-pseudo-parameters
serverless create --template aws-nodejs --name serverless-snapshot-slack-app --path serverless-snapshot-slack-app

 npm i aws-sdk axios chrome-aws-lambda puppeteer-core query-string dayjs

module.exports.accept = async (event) => {
  const params = queryString.parse(event.toString());

  await axios.post(params.response_url, {
    response_type: "in_channel",
    text: `処理を受け付けました。`,
  });

  return {
    channel_name: params.channel_name,
    response_url: params.response_url,
  };
};

const chromeLambda = require("chrome-aws-lambda");
const puppeteer = chromeLambda.puppeteer;

module.exports.capture = async (event) => {
  const browser = await puppeteer.launch({
    args: chromeLambda.args,
    defaultViewport: chromeLambda.defaultViewport,
    executablePath: await chromeLambda.executablePath,
    headless: chromeLambda.headless,
    ignoreHTTPSErrors: true,
  });

  const page = await browser.newPage();

  // スクリーンショットを撮りたいページにアクセスし、pngデータを取得する。
  await page.goto(`https://www.google.com/`, {
    waitUntil: "networkidle0",
  });
  const png = await page.screenshot({ fullPage: true });

  // S3にアップロードする
  await s3
    .putObject({
      Bucket: bucket_name,
      Key: filename,
      Body: png,
    })
    .promise();

  const download_url = await s3.getSignedUrlPromise("getObject", {
    Bucket: bucket_name,
    Key: filename,
    Expires: 604800,
  });

  await axios.post(event.response_url, {
    response_type: "in_channel",
    text: download_url,
  });
};

service: serverless-capture-slack-app

frameworkVersion: '2'

provider:
  name: aws
  runtime: nodejs12.x
  region: ap-northeast-1
  stage: ${opt:stage, 'dev'}
  # プロファイルは各自のものを使おう。
  profile: rhythm191
  memorySize: 2048
  timeout: 90
  layers:
    - arn:aws:lambda:ap-northeast-1:764866452798:layer:chrome-aws-lambda:19
  iamRoleStatements:
    - Effect: "Allow"
      Action:
        - s3:PutObject
        - s3:PutObjectAcl
      Resource: arn:aws:s3:::serverless-capture-slack-app/*

package:
  exclude:
    - .prettier*

functions:
  accept:
    handler: handler.accept
  capture:
    handler: handler.capture
    timeout: 300

stepFunctions:
  stateMachines:
    captureSteps:
      name: serverless-capture-slack-app-${opt:stage, 'dev'}
      events:
        - http:
            path: serverless-capture-slack-app
            method: post
      definition:
        Comment: "capture function"
        StartAt: accept
        States:
          accept:
            Type: Task
            Resource:
              Fn::GetAtt: [accept, Arn]
            Next: capture
          capture:
            Type: Task
            Resource:
              Fn::GetAtt: [capture, Arn]
            End: true


# plugins
plugins:
  - serverless-step-functions
  - serverless-pseudo-parameters

serverless deploy