Voting for YouTube videos

Author: madhead

.deploy/lambda/bin/JProfByBotStack.ts

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

.deploy/lambda/lib/JProfByBotStack.ts

@@ -15,6 +15,11 @@ export class JProfByBotStack extends cdk.Stack {
       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 lambdaWebhook = new lambda.Function(this, 'jprof-by-bot-lambda-webhook', {
       functionName: 'jprof-by-bot-lambda-webhook',
@@ -26,11 +31,14 @@ export class JProfByBotStack extends cdk.Stack {
       environment: {
         'LOG_THRESHOLD': 'DEBUG',
         'TABLE_VOTES': votesTable.tableName,
-        'TELEGRAM_BOT_TOKEN': props.telegramToken,
+        'TABLE_YOUTUBE_CHANNELS_WHITELIST': youtubeChannelsWhitelistTable.tableName,
+        'TOKEN_TELEGRAM_BOT': props.telegramToken,
+        'TOKEN_YOUTUBE_API': props.youtubeToken,
       },
     });
 
     votesTable.grantReadWriteData(lambdaWebhook);
+    youtubeChannelsWhitelistTable.grantReadData(lambdaWebhook);
 
     const api = new apigateway.RestApi(this, 'jprof-by-bot-api', {
       restApiName: 'jprof-by-bot-api',

.deploy/lambda/lib/JProfByBotStackProps.ts

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

.github/workflows/default.yml

@@ -55,6 +55,7 @@ jobs:
           job-id: db-test
           read-only: ${{ github.ref != 'refs/heads/master' }}
       - run: votes/dynamodb/src/test/resources/seed.sh
+      - run: youtube/dynamodb/src/test/resources/seed.sh
       - run: ./gradlew clean dbTest
       - uses: actions/upload-artifact@v2
         if: always()

.github/workflows/deploy.yml

@@ -39,7 +39,8 @@ jobs:
       - run: npx cdk deploy --outputs-file=cdk.out/outputs.json --require-approval=never
         working-directory: .deploy/lambda
         env:
-          TELEGRAM_BOT_TOKEN: ${{ secrets.TELEGRAM_BOT_TOKEN }}
+          TOKEN_TELEGRAM_BOT: ${{ secrets.TOKEN_TELEGRAM_BOT }}
+          TOKEN_YOUTUBE_API: ${{ secrets.TOKEN_YOUTUBE_API }}
       - id: URL
         run: echo "::set-output name=URL::$(jq -r '.JProfByBotStack.URL' cdk.out/outputs.json)"
         working-directory: .deploy/lambda

gradle/libs.versions.toml

@@ -14,6 +14,8 @@ tgbotapi = "0.33.3"
 
 jsoup = "1.13.1"
 
+google-api-services-youtube = "v3-rev20210410-1.31.0"
+
 log4j = "2.14.1"
 
 junit5 = "5.7.1"
@@ -34,9 +36,12 @@ jackson-databind = { group = "com.fasterxml.jackson.core", name = "jackson-datab
 
 tgbotapi-core = { group = "dev.inmo", name = "tgbotapi.core", version.ref = "tgbotapi" }
 tgbotapi-extensions-api = { group = "dev.inmo", name = "tgbotapi.extensions.api", version.ref = "tgbotapi" }
+tgbotapi-extensions-utils = { group = "dev.inmo", name = "tgbotapi.extensions.utils", version.ref = "tgbotapi" }
 
 jsoup = { group = "org.jsoup", name = "jsoup", version.ref = "jsoup" }
 
+google-api-services-youtube = { group = "com.google.apis", name = "google-api-services-youtube", version.ref = "google-api-services-youtube" }
+
 log4j-api = { group = "org.apache.logging.log4j", name = "log4j-api", version.ref = "log4j" }
 log4j-core = { group = "org.apache.logging.log4j", name = "log4j-core", version.ref = "log4j" }
 log4j-slf4j-impl = { group = "org.apache.logging.log4j", name = "log4j-slf4j-impl", version.ref = "log4j" }

runners/lambda/build.gradle.kts

@@ -11,4 +11,5 @@ dependencies {
     implementation(project.projects.core)
     implementation(project.projects.votes.dynamodb)
     implementation(project.projects.jep)
+    implementation(project.projects.youtube.dynamodb)
 }

runners/lambda/src/main/kotlin/by/jprof/telegram/bot/runners/lambda/JProf.kt

@@ -36,7 +36,8 @@ class JProf : RequestHandler<APIGatewayV2HTTPEvent, APIGatewayV2HTTPResponse>, K
                 databaseModule,
                 jsonModule,
                 telegramModule,
-                pipelineModule
+                youtubeModule,
+                pipelineModule,
             )
         }
     }

runners/lambda/src/main/kotlin/by/jprof/telegram/bot/runners/lambda/config/database.kt

@@ -1,10 +1,12 @@
 package by.jprof.telegram.bot.runners.lambda.config
 
 import by.jprof.telegram.bot.votes.dao.VotesDAO
+import by.jprof.telegram.bot.youtube.dao.YouTubeChannelsWhitelistDAO
 import org.koin.core.qualifier.named
 import org.koin.dsl.module
 import software.amazon.awssdk.services.dynamodb.DynamoDbAsyncClient
 import by.jprof.telegram.bot.votes.dynamodb.dao.VotesDAO as DynamoDBVotesDAO
+import by.jprof.telegram.bot.youtube.dynamodb.dao.YouTubeChannelsWhitelistDAO as DynamoDBYouTubeChannelsWhitelistDAO
 
 val databaseModule = module {
     single {
@@ -17,4 +19,11 @@ val databaseModule = module {
             get(named(TABLE_VOTES))
         )
     }
+
+    single<YouTubeChannelsWhitelistDAO> {
+        DynamoDBYouTubeChannelsWhitelistDAO(
+            get(),
+            get(named(TABLE_YOUTUBE_CHANNELS_WHITELIST))
+        )
+    }
 }

runners/lambda/src/main/kotlin/by/jprof/telegram/bot/runners/lambda/config/env.kt

@@ -3,13 +3,17 @@ package by.jprof.telegram.bot.runners.lambda.config
 import org.koin.core.qualifier.named
 import org.koin.dsl.module
 
-const val TELEGRAM_BOT_TOKEN = "TELEGRAM_BOT_TOKEN"
+const val TOKEN_TELEGRAM_BOT = "TOKEN_TELEGRAM_BOT"
+const val TOKEN_YOUTUBE_API = "TOKEN_YOUTUBE_API"
 const val TABLE_VOTES = "TABLE_VOTES"
+const val TABLE_YOUTUBE_CHANNELS_WHITELIST = "TABLE_YOUTUBE_CHANNELS_WHITELIST"
 
 val envModule = module {
     listOf(
-        TELEGRAM_BOT_TOKEN,
+        TOKEN_TELEGRAM_BOT,
+        TOKEN_YOUTUBE_API,
         TABLE_VOTES,
+        TABLE_YOUTUBE_CHANNELS_WHITELIST,
     ).forEach { variable ->
         single(named(variable)) {
             System.getenv(variable)!!