add !foto command to take screenshot

.github/workflows/main_twitch-mz-bot.yml

@@ -29,6 +29,7 @@ jobs:
           envkey_PORT: 3000
           envkey_TELEGRAM_TOKEN: ${{ secrets.TELEGRAM_TOKEN }}
           envKey_TELEGRAM_CHAT_ID: ${{ secrets.TELEGRAM_CHAT_ID }}
+          envKey_EXTERNAL_URL: ${{ secrets.EXTERNAL_URL }}
 
       - name: Set up Node.js version
         uses: actions/setup-node@v4

app.js

@@ -10,14 +10,16 @@ mongoose.connect(config.database).then(() => {
         .then(() => {
             console.log('Connected')
 
-            const app = express();
+            const app = express()
+
+            app.use(express.static('public'))
+            app.use('/images', express.static('images'))
+
             const listener = app.listen(process.env.PORT, ()=>  {
                 console.log('Listening on port ', + listener.address().port)
                 app.get('/', (req, res) => res.send('Hello World!'))
             })
         })
-    const notifier = new Notifier()
-    notifier.notify()
 })
 
 

config/index.js

@@ -1,4 +1,5 @@
 module.exports = {
+    externalUrl: process.env.EXTERNAL_URL,
     database: process.env.MONGODB_URI,
     twitch: {
         channels: process.env.TWITCH_CHANNELS,

handlers/stream.js

@@ -1,12 +1,25 @@
 const TwitchService = require('../services/twitch')
-const config = require("../config");
+const BrowserService = require('../services/browser')
+const config = require("../config")
 const moment = require('moment')
 require('moment-precise-range-plugin')
-var math = require('mathjs');
+require('mathjs')
+
 
 const twitchUrl = 'https://www.twitch.tv/'
 
 class Stream {
+    async captureScreenshot(target, bot, notifierBot, user) {
+        const image = await BrowserService.getScreenshot()
+        await bot.say(target, `Captura de ${user}: ${config.externalUrl}/images/${image.fileName}`)
+        const channel = await TwitchService.getChannel()
+        await notifierBot.sendPhoto(config.telegram.chatId, image.buffer, {
+            caption: `Captura del directo _${channel.title}_ \n por *${user}*`,
+            parse_mode: 'Markdown'
+        })
+
+    }
+
     async catchStream (bot) {
         const result = await TwitchService.getStream()
 

lib/inputParser.js

@@ -18,6 +18,10 @@ class InputParser {
     isAskingForNextAveTrain (text) {
         return text.toLowerCase().startsWith('!ave')
     }
+
+    isAskingForTakeScreenshot (text) {
+        return text.toLowerCase().startsWith('!foto')
+    }
 }
 
 module.exports = InputParser

lib/messenger.js

@@ -6,6 +6,7 @@ const inputParser = new InputParser()
 const tmi = require('@twurple/auth-tmi')
 const { RefreshingAuthProvider } = require('@twurple/auth')
 const TokenService = require("../services/token");
+const Notifier = require("./notifier");
 
 class Messenger {
     constructor () {}
@@ -20,6 +21,9 @@ class Messenger {
         }
         this.bot = new tmi.client(opts)
 
+        this.notifier = new Notifier()
+        await this.notifier.notify()
+
         return this.listen()
     }
 
@@ -72,6 +76,9 @@ class Messenger {
         if (textSplit.length > 2 && inputParser.isAskingForNextAveTrain(textSplit[0])) {
             return handlers.train.getNextAVE(target, textSplit.slice(1).join(' '), this.bot)
         }
+
+        if (textSplit.length > 0 && inputParser.isAskingForTakeScreenshot(textSplit[0]))
+            return handlers.stream.captureScreenshot(target, this.bot, this.notifier.bot, context['display-name'])
     }
 
     handleHosting (channel, target, viewers) {

lib/notifier.js

@@ -15,6 +15,10 @@ class Notifier {
 
         return Promise.resolve()
     }
+
+    getBot () {
+        return this.bot
+    }
 }
 
 module.exports = Notifier

package.json

@@ -17,6 +17,7 @@
     "mssql": "^10.0.2",
     "node-cron": "^3.0.3",
     "node-telegram-bot-api": "^0.64.0",
+    "puppeteer": "^21.11.0",
     "tmi.js": "^1.8.5"
   },
   "scripts": {

services/browser.js

@@ -0,0 +1,31 @@
+const config = require('../config')
+const puppeteer = require("puppeteer");
+require('mathjs')
+
+async function getScreenshot() {
+    const browser = await puppeteer.launch({headless: true})
+    const page = await browser.newPage()
+    const session = await page.target().createCDPSession()
+    await session.send("Page.enable")
+
+    await page.setViewport({ width: 1920, height: 1080 })
+
+    await page.goto("https://www.twitch.tv/" + config.twitch.channels, { waitUntil: 'networkidle0' })
+    await page.waitForSelector('div.persistent-player')
+    await page.$eval('.video-player__default-player', el => el.remove())
+    const svgImage = await page.$('div.persistent-player')
+    const name = Math.random().toString(36).substring(2,8)
+    const bufferImage = await svgImage.screenshot({
+        path: `public/images/${name}.png`,
+        omitBackground: true
+    })
+    await browser.close()
+
+    return {buffer: bufferImage, fileName: `${name}.png` }
+}
+
+
+
+module.exports = {
+    getScreenshot
+}

services/twitch.js

@@ -32,6 +32,10 @@ async function getStream() {
     return { ...result, lastUpdate: channel.lastUpdate}
 }
 
+async function getChannel () {
+    return dbManager.getChannel(config.twitch.channels).lean()
+}
+
 async function saveLastMessage (msg) {
     await dbManager.updateChannel(config.twitch.channels, { lastMessageId: msg.message_id })
 }
@@ -49,5 +53,6 @@ module.exports = {
     getStream,
     saveLastMessage,
     deleteLastMessage,
-    saveTitle
+    saveTitle,
+    getChannel
 }