pissbot-9000/commands/message/search/wolfram-alpha.js

87 lines
No EOL
3.2 KiB
JavaScript

const { paginator } = require('#client');
const { PERMISSION_GROUPS } = require('#constants');
const { WolframQueryCompute } = require('#obelisk');
const { createEmbed, formatPaginationEmbeds, page } = require('#utils/embed');
const { acknowledge } = require('#utils/interactions');
const { citation, smallIconPill } = require('#utils/markdown');
const { editOrReply } = require('#utils/message')
const { STATICS } = require('#utils/statics')
function createWolframPage(context, pod, query, sources) {
let res = page(createEmbed("default", context, {
author: {
name: pod.title,
url: `https://www.wolframalpha.com/input?i=${encodeURIComponent(query)}`
},
description: undefined,
footer: {
iconUrl: STATICS.wolframalpha,
text: `Wolfram|Alpha • ${context.application.name}`
}
}))
if (pod.icon) res.embeds[0].author.iconUrl = pod.icon
if (pod.value) res.embeds[0].description = pod.value.substr(0, 1000)
if (pod.value && pod.refs) {
for (const r of pod.refs) {
let src = Object.values(sources).filter((s) => s.ref == r)[0]
if (!src) continue;
// Only add a direct source if one is available
if (src.collections) {
res.embeds[0].description += citation(r, src.url, src.title + ' | ' + src.collections[0].text)
continue;
}
if (src.url) res.embeds[0].description += citation(r, src.url, src.title)
}
}
if (pod.image) res.embeds[0].image = { url: pod.image };
return res;
}
module.exports = {
name: 'wolframalpha',
label: 'query',
aliases: ['wa', 'wolfram-alpha', 'wolfram'],
metadata: {
description: `${smallIconPill("reply", "Supports Replies")}\n\nComputes a query using Wolfram|Alpha.`,
description_short: 'Compute via Wolfram|Alpha',
examples: ['wa x^2+5x+6=0', 'wa 5€ to $', 'wa 5\'11 to cm'],
category: 'search',
usage: 'wolframalpha <query>',
slashCommand: "wolframalpha"
},
permissionsClient: [...PERMISSION_GROUPS.baseline],
run: async (context, args) => {
await acknowledge(context);
if(context.message.messageReference) {
let msg = await context.message.channel.fetchMessage(context.message.messageReference.messageId);
if(msg.content && msg.content.length) args.query = msg.content
if(msg.embeds?.length) for(const e of msg.embeds) if(e[1].description?.length) { args.query += '\n' + e[1].description; break; }
}
if (!args.query) return editOrReply(context, createEmbed("warning", context, `Missing Parameter (query).`))
try {
let search = await WolframQueryCompute(context, args.query)
search = search.response
if (search.body.status == 1) return editOrReply(context, createEmbed("warning", context, search.body.message))
let pages = []
for (const res of search.body.data) {
pages.push(createWolframPage(context, res, args.query, search.body.sources))
}
await paginator.createPaginator({
context,
pages: formatPaginationEmbeds(pages)
});
} catch (e) {
if(e.response.body?.error) return editOrReply(context, createEmbed("warning", context, e.response.body.error.message))
console.log(e)
return editOrReply(context, createEmbed("error", context, `Unable to perform Wolfram|Alpha search.`))
}
},
};