Pins

.deploy/lambda/bin/JProfByBotStack.ts

@@ -1,25 +1,25 @@
 #!/usr/bin/env node
 import 'source-map-support/register';
 import * as cdk from '@aws-cdk/core';
-import { JProfByBotStack } from '../lib/JProfByBotStack';
+import {JProfByBotStack} from '../lib/JProfByBotStack';
 
 if (process.env.TOKEN_TELEGRAM_BOT == null) {
-  throw new Error('Undefined TOKEN_TELEGRAM_BOT')
+    throw new Error('Undefined TOKEN_TELEGRAM_BOT')
 }
 
 if (process.env.TOKEN_YOUTUBE_API == null) {
-  throw new Error('Undefined TOKEN_YOUTUBE_API')
+    throw new Error('Undefined TOKEN_YOUTUBE_API')
 }
 
 const app = new cdk.App();
 new JProfByBotStack(
-  app,
-  'JProfByBotStack',
-  {
-    telegramToken: process.env.TOKEN_TELEGRAM_BOT,
-    youtubeToken: process.env.TOKEN_YOUTUBE_API,
-    env: {
-      region: 'us-east-1'
+    app,
+    'JProfByBotStack',
+    {
+        telegramToken: process.env.TOKEN_TELEGRAM_BOT,
+        youtubeToken: process.env.TOKEN_YOUTUBE_API,
+        env: {
+            region: 'us-east-1'
+        }
     }
-  }
 );

.deploy/lambda/lib/JProfByBotStack.ts

@@ -1,98 +1,169 @@
 import * as cdk from '@aws-cdk/core';
-import { Duration, RemovalPolicy } from '@aws-cdk/core';
-import { JProfByBotStackProps } from './JProfByBotStackProps';
+import {JProfByBotStackProps} from './JProfByBotStackProps';
 import * as dynamodb from '@aws-cdk/aws-dynamodb';
 import * as lambda from '@aws-cdk/aws-lambda';
 import * as apigateway from '@aws-cdk/aws-apigateway';
+import * as sfn from '@aws-cdk/aws-stepfunctions';
+import * as tasks from '@aws-cdk/aws-stepfunctions-tasks';
 
 export class JProfByBotStack extends cdk.Stack {
-  constructor(scope: cdk.Construct, id: string, props: JProfByBotStackProps) {
-    super(scope, id, props);
-
-    const votesTable = new dynamodb.Table(this, 'jprof-by-bot-table-votes', {
-      tableName: 'jprof-by-bot-table-votes',
-      partitionKey: { name: 'id', type: dynamodb.AttributeType.STRING },
-      billingMode: dynamodb.BillingMode.PAY_PER_REQUEST,
-      removalPolicy: RemovalPolicy.DESTROY,
-    });
-    const youtubeChannelsWhitelistTable = new dynamodb.Table(this, 'jprof-by-bot-table-youtube-channels-whitelist', {
-      tableName: 'jprof-by-bot-table-youtube-channels-whitelist',
-      partitionKey: { name: 'id', type: dynamodb.AttributeType.STRING },
-      billingMode: dynamodb.BillingMode.PAY_PER_REQUEST,
-    });
-    const kotlinMentionsTable = new dynamodb.Table(this, 'jprof-by-bot-table-kotlin-mentions', {
-      tableName: 'jprof-by-bot-table-kotlin-mentions',
-      partitionKey: { name: 'chat', type: dynamodb.AttributeType.NUMBER },
-      billingMode: dynamodb.BillingMode.PAY_PER_REQUEST,
-    });
-    const dialogStatesTable = new dynamodb.Table(this, 'jprof-by-bot-table-dialog-states', {
-      tableName: 'jprof-by-bot-table-dialog-states',
-      partitionKey: { name: 'userId', type: dynamodb.AttributeType.NUMBER },
-      sortKey: { name: 'chatId', type: dynamodb.AttributeType.NUMBER },
-      billingMode: dynamodb.BillingMode.PAY_PER_REQUEST,
-    });
-    const quizojisTable = new dynamodb.Table(this, 'jprof-by-bot-table-quizojis', {
-      tableName: 'jprof-by-bot-table-quizojis',
-      partitionKey: { name: 'id', type: dynamodb.AttributeType.STRING },
-      billingMode: dynamodb.BillingMode.PAY_PER_REQUEST,
-    });
-
-    const layerLibGL = new lambda.LayerVersion(this, 'jprof-by-bot-lambda-layer-libGL', {
-      code: lambda.Code.fromAsset('layers/libGL.zip'),
-      compatibleRuntimes: [lambda.Runtime.JAVA_11],
-    });
-    const layerLibfontconfig = new lambda.LayerVersion(this, 'jprof-by-bot-lambda-layer-libfontconfig', {
-      code: lambda.Code.fromAsset('layers/libfontconfig.zip'),
-      compatibleRuntimes: [lambda.Runtime.JAVA_11],
-    });
-
-    const lambdaWebhook = new lambda.Function(this, 'jprof-by-bot-lambda-webhook', {
-      functionName: 'jprof-by-bot-lambda-webhook',
-      runtime: lambda.Runtime.JAVA_11,
-      layers: [
-        layerLibGL,
-        layerLibfontconfig,
-      ],
-      timeout: Duration.seconds(30),
-      memorySize: 1024,
-      code: lambda.Code.fromAsset('../../runners/lambda/build/libs/jprof_by_bot-runners-lambda-all.jar'),
-      handler: 'by.jprof.telegram.bot.runners.lambda.JProf',
-      environment: {
-        'LOG_THRESHOLD': 'DEBUG',
-        'TABLE_VOTES': votesTable.tableName,
-        'TABLE_YOUTUBE_CHANNELS_WHITELIST': youtubeChannelsWhitelistTable.tableName,
-        'TABLE_KOTLIN_MENTIONS': kotlinMentionsTable.tableName,
-        'TABLE_DIALOG_STATES': dialogStatesTable.tableName,
-        'TABLE_QUIZOJIS': quizojisTable.tableName,
-        'TOKEN_TELEGRAM_BOT': props.telegramToken,
-        'TOKEN_YOUTUBE_API': props.youtubeToken,
-      },
-    });
-
-    votesTable.grantReadWriteData(lambdaWebhook);
-    youtubeChannelsWhitelistTable.grantReadData(lambdaWebhook);
-    kotlinMentionsTable.grantReadWriteData(lambdaWebhook);
-    dialogStatesTable.grantReadWriteData(lambdaWebhook);
-    quizojisTable.grantReadWriteData(lambdaWebhook);
-
-    const api = new apigateway.RestApi(this, 'jprof-by-bot-api', {
-      restApiName: 'jprof-by-bot-api',
-      cloudWatchRole: false,
-      endpointTypes: [apigateway.EndpointType.REGIONAL],
-      deployOptions: {
-        loggingLevel: apigateway.MethodLoggingLevel.INFO,
-        dataTraceEnabled: true,
-        metricsEnabled: true,
-        tracingEnabled: true,
-      },
-    });
-
-    api.root
-      .addResource(props.telegramToken.replace(':', '_'))
-      .addMethod('POST', new apigateway.LambdaIntegration(lambdaWebhook));
-
-    new cdk.CfnOutput(this, 'URL', {
-      value: api.deploymentStage.urlForPath()
-    });
-  }
+    constructor(scope: cdk.Construct, id: string, props: JProfByBotStackProps) {
+        super(scope, id, props);
+
+        const votesTable = new dynamodb.Table(this, 'jprof-by-bot-table-votes', {
+            tableName: 'jprof-by-bot-table-votes',
+            partitionKey: {name: 'id', type: dynamodb.AttributeType.STRING},
+            billingMode: dynamodb.BillingMode.PAY_PER_REQUEST,
+            removalPolicy: cdk.RemovalPolicy.DESTROY,
+        });
+        const youtubeChannelsWhitelistTable = new dynamodb.Table(this, 'jprof-by-bot-table-youtube-channels-whitelist', {
+            tableName: 'jprof-by-bot-table-youtube-channels-whitelist',
+            partitionKey: {name: 'id', type: dynamodb.AttributeType.STRING},
+            billingMode: dynamodb.BillingMode.PAY_PER_REQUEST,
+            removalPolicy: cdk.RemovalPolicy.DESTROY,
+        });
+        const kotlinMentionsTable = new dynamodb.Table(this, 'jprof-by-bot-table-kotlin-mentions', {
+            tableName: 'jprof-by-bot-table-kotlin-mentions',
+            partitionKey: {name: 'chat', type: dynamodb.AttributeType.NUMBER},
+            billingMode: dynamodb.BillingMode.PAY_PER_REQUEST,
+            removalPolicy: cdk.RemovalPolicy.DESTROY,
+        });
+        const dialogStatesTable = new dynamodb.Table(this, 'jprof-by-bot-table-dialog-states', {
+            tableName: 'jprof-by-bot-table-dialog-states',
+            partitionKey: {name: 'userId', type: dynamodb.AttributeType.NUMBER},
+            sortKey: {name: 'chatId', type: dynamodb.AttributeType.NUMBER},
+            billingMode: dynamodb.BillingMode.PAY_PER_REQUEST,
+            removalPolicy: cdk.RemovalPolicy.DESTROY,
+        });
+        const quizojisTable = new dynamodb.Table(this, 'jprof-by-bot-table-quizojis', {
+            tableName: 'jprof-by-bot-table-quizojis',
+            partitionKey: {name: 'id', type: dynamodb.AttributeType.STRING},
+            billingMode: dynamodb.BillingMode.PAY_PER_REQUEST,
+            removalPolicy: cdk.RemovalPolicy.DESTROY,
+        });
+        const moniesTable = new dynamodb.Table(this, 'jprof-by-bot-table-monies', {
+            tableName: 'jprof-by-bot-table-monies',
+            partitionKey: {name: 'user', type: dynamodb.AttributeType.NUMBER},
+            sortKey: {name: 'chat', type: dynamodb.AttributeType.NUMBER},
+            billingMode: dynamodb.BillingMode.PAY_PER_REQUEST,
+            removalPolicy: cdk.RemovalPolicy.DESTROY,
+        });
+        const pinsTable = new dynamodb.Table(this, 'jprof-by-bot-table-pins', {
+            tableName: 'jprof-by-bot-table-pins',
+            partitionKey: {name: 'messageId', type: dynamodb.AttributeType.NUMBER},
+            sortKey: {name: 'chatId', type: dynamodb.AttributeType.NUMBER},
+            billingMode: dynamodb.BillingMode.PAY_PER_REQUEST,
+            removalPolicy: cdk.RemovalPolicy.DESTROY,
+        });
+
+        pinsTable.addGlobalSecondaryIndex({
+            indexName: 'chatId',
+            partitionKey: {name: 'chatId', type: dynamodb.AttributeType.NUMBER},
+            projectionType: dynamodb.ProjectionType.ALL,
+        });
+        pinsTable.addGlobalSecondaryIndex({
+            indexName: 'userId',
+            partitionKey: {name: 'userId', type: dynamodb.AttributeType.NUMBER},
+            projectionType: dynamodb.ProjectionType.ALL,
+        });
+
+        const lambdaUnpin = new lambda.Function(this, 'jprof-by-bot-lambda-unpin', {
+            functionName: 'jprof-by-bot-lambda-unpin',
+            runtime: lambda.Runtime.JAVA_11,
+            timeout: cdk.Duration.seconds(30),
+            memorySize: 512,
+            code: lambda.Code.fromAsset('../../pins/unpin/build/libs/jprof_by_bot-pins-unpin-all.jar'),
+            handler: 'by.jprof.telegram.bot.pins.unpin.Handler',
+            environment: {
+                'LOG_THRESHOLD': 'DEBUG',
+                'TABLE_PINS': pinsTable.tableName,
+                'TOKEN_TELEGRAM_BOT': props.telegramToken,
+            },
+        });
+
+        const stateMachineUnpin = new sfn.StateMachine(this, 'jprof-by-bot-state-machine-unpin', {
+            stateMachineName: 'jprof-by-bot-state-machine-unpin',
+            stateMachineType: sfn.StateMachineType.STANDARD,
+            definition: new sfn.Wait(this, 'jprof-by-bot-state-machine-unpin-wait', {
+                time: sfn.WaitTime.secondsPath('$.ttl'),
+            }).next(new tasks.LambdaInvoke(this, 'jprof-by-bot-state-machine-unpin-unpin', {
+                lambdaFunction: lambdaUnpin,
+                invocationType: tasks.LambdaInvocationType.EVENT,
+                retryOnServiceExceptions: false,
+            })),
+            tracingEnabled: false,
+        });
+
+        const layerLibGL = new lambda.LayerVersion(this, 'jprof-by-bot-lambda-layer-libGL', {
+            code: lambda.Code.fromAsset('layers/libGL.zip'),
+            compatibleRuntimes: [lambda.Runtime.JAVA_11],
+        });
+        const layerLibfontconfig = new lambda.LayerVersion(this, 'jprof-by-bot-lambda-layer-libfontconfig', {
+            code: lambda.Code.fromAsset('layers/libfontconfig.zip'),
+            compatibleRuntimes: [lambda.Runtime.JAVA_11],
+        });
+
+        const lambdaWebhook = new lambda.Function(this, 'jprof-by-bot-lambda-webhook', {
+            functionName: 'jprof-by-bot-lambda-webhook',
+            runtime: lambda.Runtime.JAVA_11,
+            layers: [
+                layerLibGL,
+                layerLibfontconfig,
+            ],
+            timeout: cdk.Duration.seconds(30),
+            memorySize: 1024,
+            code: lambda.Code.fromAsset('../../runners/lambda/build/libs/jprof_by_bot-runners-lambda-all.jar'),
+            handler: 'by.jprof.telegram.bot.runners.lambda.JProf',
+            environment: {
+                'LOG_THRESHOLD': 'DEBUG',
+                'TABLE_VOTES': votesTable.tableName,
+                'TABLE_YOUTUBE_CHANNELS_WHITELIST': youtubeChannelsWhitelistTable.tableName,
+                'TABLE_KOTLIN_MENTIONS': kotlinMentionsTable.tableName,
+                'TABLE_DIALOG_STATES': dialogStatesTable.tableName,
+                'TABLE_QUIZOJIS': quizojisTable.tableName,
+                'TABLE_MONIES': moniesTable.tableName,
+                'TABLE_PINS': pinsTable.tableName,
+                'STATE_MACHINE_UNPINS': stateMachineUnpin.stateMachineArn,
+                'TOKEN_TELEGRAM_BOT': props.telegramToken,
+                'TOKEN_YOUTUBE_API': props.youtubeToken,
+            },
+        });
+
+        votesTable.grantReadWriteData(lambdaWebhook);
+
+        youtubeChannelsWhitelistTable.grantReadData(lambdaWebhook);
+
+        kotlinMentionsTable.grantReadWriteData(lambdaWebhook);
+
+        dialogStatesTable.grantReadWriteData(lambdaWebhook);
+
+        quizojisTable.grantReadWriteData(lambdaWebhook);
+
+        moniesTable.grantReadWriteData(lambdaWebhook);
+
+        pinsTable.grantReadWriteData(lambdaWebhook);
+        pinsTable.grantReadWriteData(lambdaUnpin);
+
+        stateMachineUnpin.grantStartExecution(lambdaWebhook)
+
+        const api = new apigateway.RestApi(this, 'jprof-by-bot-api', {
+            restApiName: 'jprof-by-bot-api',
+            cloudWatchRole: false,
+            endpointTypes: [apigateway.EndpointType.REGIONAL],
+            deployOptions: {
+                loggingLevel: apigateway.MethodLoggingLevel.INFO,
+                dataTraceEnabled: true,
+                metricsEnabled: true,
+                tracingEnabled: true,
+            },
+        });
+
+        api.root
+            .addResource(props.telegramToken.replace(':', '_'))
+            .addMethod('POST', new apigateway.LambdaIntegration(lambdaWebhook));
+
+        new cdk.CfnOutput(this, 'URL', {
+            value: api.deploymentStage.urlForPath()
+        });
+    }
 }

.deploy/lambda/lib/JProfByBotStackProps.ts

@@ -1,6 +1,6 @@
 import * as cdk from '@aws-cdk/core';
 
 export interface JProfByBotStackProps extends cdk.StackProps {
-  readonly telegramToken: string;
-  readonly youtubeToken: string;
+    readonly telegramToken: string;
+    readonly youtubeToken: string;
 }