pissbot-9000/commands/message/utils/screenshot.js

const { screenshot } = require("../../../labscore/api");
const { createEmbed } = require("../../../labscore/utils/embed");
const { editOrReply } = require("../../../labscore/utils/message");

const superagent = require('superagent')

async function processJob(jobUrl){
  let job = await superagent.get(jobUrl)
    .set('User-Agent', 'labscore/1.0')

  return job.body;
} 

module.exports = {
  label: "url",
  name: "screenshot",
  aliases: ['ss'],
  metadata: {
    description: 'Takes screenshots of a website.',
    description_short: 'Screenshot websites.',
    examples: ['ss google.com'],
    category: 'utils',
    usage: 'screenshot <url>'
  },
  run: async (context, args) => {
    if(!args.url) return editOrReply(context, { embeds: [createEmbed("warning", context, "No url supplied.")] })
    
    let response = await editOrReply(context, createEmbed("loading", context, `Creating website screenshot...`))

    try{
      const t = Date.now();

      let ss = await screenshot(context, args.url, context.channel.nsfw)

      if(ss.response.body.status && ss.response.body.status !== 3){
        if(ss.response.body.image) return await response.edit({
          embeds: [createEmbed("image", context, {
            url: ss.response.body.image,
            time: ((Date.now() - t) / 1000).toFixed(2)
          })]
        })
        return await response.edit({ embeds: [createEmbed("error", context, "Unable to create screenshot.")] })
      }

      let job = await processJob(ss.response.body.job)

      if(job.status){
        if(!job.image) job = await processJob(ss.response.body.job)
        if(job.image) return await response.edit({
          embeds: [createEmbed("image", context, {
            url: job.image,
            time: ((Date.now() - t) / 1000).toFixed(2)
          })]
        })
        return await response.edit({ embeds: [createEmbed("error", context, "Unable to create screenshot.")] })
      }

      return await response.edit({
        embeds: [createEmbed("image", context, {
          url: "screenshot.png",
          time: ((Date.now() - t) / 1000).toFixed(2)
        })],
        files: [{ filename: "screenshot.png", value: job }]
      })
    } catch(e){
      console.log(e)
      return await response.edit({
        embeds: [createEmbed("image", context, {
          url: "https://derpystuff.gitlab.io/webstorage4/v2/assets/screenshot/screenshot_error.png"
        })]
      })
    }
  }
};