screenshot

This commit is contained in:
derpystuff 2024-01-18 21:34:06 +01:00
parent 3c180acdf1
commit 809409920e
3 changed files with 16 additions and 17 deletions

View file

@ -1,10 +1,10 @@
const { screenshot } = require("../../../labscore/api");
const { createEmbed } = require("../../../labscore/utils/embed"); const { createEmbed } = require("../../../labscore/utils/embed");
const { editOrReply } = require("../../../labscore/utils/message"); const { editOrReply } = require("../../../labscore/utils/message");
const superagent = require('superagent') const superagent = require('superagent')
const { Permissions } = require("detritus-client/lib/constants"); const { Permissions } = require("detritus-client/lib/constants");
const { webshot } = require("../../../labscore/api/obelisk");
async function processJob(jobUrl) { async function processJob(jobUrl) {
let job = await superagent.get(jobUrl) let job = await superagent.get(jobUrl)
@ -28,12 +28,12 @@ module.exports = {
run: async (context, args) => { run: async (context, args) => {
if (!args.url) return editOrReply(context, createEmbed("warning", context, "No url supplied.")) if (!args.url) return editOrReply(context, createEmbed("warning", context, "No url supplied."))
let response = await editOrReply(context, createEmbed("loading", context, `Creating website screenshot...`)) await editOrReply(context, createEmbed("loading", context, `Creating website screenshot...`))
try { try {
const t = Date.now(); const t = Date.now();
let ss = await screenshot(context, args.url, context.channel.nsfw) let ss = await webshot(context, args.url, context.channel.nsfw)
if (ss.response.body.status && ss.response.body.status !== 3) { if (ss.response.body.status && ss.response.body.status !== 3) {
if (ss.response.body.image) return await editOrReply(context, if (ss.response.body.image) return await editOrReply(context,
@ -45,28 +45,17 @@ module.exports = {
return await editOrReply(context, createEmbed("error", context, "Unable to create screenshot.")) return await editOrReply(context, 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 editOrReply(context, createEmbed("image", context, {
url: job.image,
time: ((Date.now() - t) / 1000).toFixed(2)
}))
return await editOrReply(context, createEmbed("error", context, "Unable to create screenshot."))
}
return await editOrReply(context, { return await editOrReply(context, {
embeds: [createEmbed("image", context, { embeds: [createEmbed("image", context, {
url: "screenshot.png", url: "screenshot.png",
time: ((Date.now() - t) / 1000).toFixed(2) time: ((Date.now() - t) / 1000).toFixed(2)
})], })],
files: [{ filename: "screenshot.png", value: job }] files: [{ filename: "screenshot.png", value: ss.response.body }]
}) })
} catch (e) { } catch (e) {
console.log(e) console.log(e)
return await editOrReply(context, createEmbed("image", context, { return await editOrReply(context, createEmbed("image", context, {
url: "https://bignutty.gitlab.io/webstorage4/v2/assets/screenshot/screenshot_error.png" url: "https://bignutty.gitlab.io/webstorage4/v2/assets/screenshot/brand-update-2024/scr_unavailable.png"
})) }))
} }
} }

View file

@ -16,7 +16,9 @@ const ObeliskApi = Object.freeze({
SUMMARIZE_WEBPAGES: "/flamingo/v1/web:summarize", SUMMARIZE_WEBPAGES: "/flamingo/v1/web:summarize",
AI_WALLPAPER: "/robin/v1/wallpaper:generate" AI_WALLPAPER: "/robin/v1/wallpaper:generate",
WEBSHOT: "/peacock/v1/screenshot",
}) })
module.exports = { module.exports = {

View file

@ -94,4 +94,12 @@ module.exports.aiWallpaper = async function(context, prompt, style){
prompt, prompt,
style style
}) })
}
// PEACOCK
module.exports.webshot = async function(context, url, allowAdultContent = false){
return await request(ObeliskApi.WEBSHOT, "POST", {}, {
url,
allow_adult: allowAdultContent
})
} }