mirror of
https://gitlab.com/bignutty/labscore.git
synced 2025-06-08 14:13:02 -04:00
85 lines
No EOL
3.2 KiB
JavaScript
85 lines
No EOL
3.2 KiB
JavaScript
const { paginator } = require('#client');
|
|
const { WolframQueryCompute } = require('#obelisk');
|
|
|
|
const { createEmbed, formatPaginationEmbeds, page } = require('#utils/embed')
|
|
const { citation, smallIconPill } = require('#utils/markdown');
|
|
const { editOrReply } = require('#utils/message')
|
|
const { STATICS } = require('#utils/statics')
|
|
|
|
// TODO: Turn this into a general purpose permissions constant
|
|
const { Permissions } = require("detritus-client/lib/constants");
|
|
|
|
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>'
|
|
},
|
|
permissionsClient: [Permissions.EMBED_LINKS, Permissions.SEND_MESSAGES, Permissions.USE_EXTERNAL_EMOJIS, Permissions.READ_MESSAGE_HISTORY],
|
|
run: async (context, args) => {
|
|
context.triggerTyping();
|
|
|
|
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) {
|
|
console.log(e)
|
|
return editOrReply(context, createEmbed("error", context, `Unable to perform Wolfram|Alpha search.`))
|
|
}
|
|
},
|
|
}; |