lightshot-parser/index.js

Parsing screenshots

const fetch = require("node-fetch")
const cheerio = require("cheerio")
const fs = require("fs")
const buffer = require("buffer")
const path = require("path")
/**
 *@param {Number} [quantity] Number of screenshots
 *@param {String} [src] Where to download directory
 *@param {Number} [delay] Delay in sending the request
 *@param {String} [template] String pattern
 **/
function imgParser(quantity, src, delay = 1000, template = null) {
  let c = quantity
  console.log("In progress")
  async function getImg() {
    if (c > 0) {
      try {
        const randomString = "abcdefghijklmnopqrstuvwxyz1234567890"
        let url = ""
        if (template) {
          url = template
          for (let i = 0; i < 6 - template.length; i++) {
            url += randomString[Math.floor(Math.random() * randomString.length)]
          }
        } else {
          for (let i = 0; i < 6; i++) {
            url += randomString[Math.floor(Math.random() * randomString.length)]
          }
        }

        const response = await fetch(`https://prnt.sc/${url}`)
        const bufferResponse = await response.arrayBuffer()
        const bf = buffer.Buffer.from(bufferResponse)
        const decoder = new TextDecoder("utf-8")
        const decode = decoder.decode(bf)
        const $ = cheerio.load(decode)
        const rows = $(".screenshot-image")
        if (rows[0].attribs.src &&!rows[0].attribs.src.startsWith("/") && !rows[0].attribs.src.match(/i.imgur/g)){
          c--
          let img = await fetch(rows[0].attribs.src)
          const response = await img.arrayBuffer()
          const imgBuffer = buffer.Buffer.from(response)
          fs.writeFileSync(path.join(src, `${url}.png`),imgBuffer,"binary")
        }
      } catch (e) {}

      setTimeout(() => {
        getImg()
      }, delay)
    }
    else {
      console.log("Done")
    }
  }
  getImg()
}

module.exports = imgParser