Update to [email protected] and refactor

Author: vrumger

functions.js

@@ -0,0 +1,42 @@
+const codebottle = require(`codebottle`);
+const { inlineButtons, formatText } = require(`../helpers`);
+
+module.exports = bot => {
+    bot.action(/^code_(.+)$/, async ctx => {
+        try {
+            const [, id] = ctx.match;
+            const data = await codebottle.fetch(id);
+
+            await ctx.editMessageText(
+                formatText(data, true),
+                Object.assign({
+                    parse_mode: `html`,
+                }, inlineButtons(id))
+            );
+
+            ctx.answerCbQuery();
+        } catch (e) {
+            ctx.answerCbQuery(`There was an error.`, true);
+            console.log(e);
+        }
+    });
+
+    bot.action(/^desc_(.+)$/, async ctx => {
+        try {
+            const [, id] = ctx.match;
+            const data = await codebottle.fetch(id);
+
+            await ctx.editMessageText(
+                formatText(data),
+                Object.assign({
+                    parse_mode: `html`,
+                }, inlineButtons(id, true))
+            );
+
+            ctx.answerCbQuery();
+        } catch (e) {
+            ctx.answerCbQuery(`There was an error.`, true);
+            console.log(e);
+        }
+    });
+};

handlers/callbacks.js

@@ -0,0 +1,24 @@
+const codebottle = require(`codebottle`);
+const { inlineButtons, formatText } = require(`../helpers`);
+
+module.exports = bot =>
+    bot.command(`fetch`, async ctx => {
+        try {
+            let { text, entities } = ctx.message;
+            text = text.slice(entities[0].length + 1);
+
+            if (text) {
+                const data = await codebottle.fetch(text);
+
+                ctx.reply(
+                    formatText(data),
+                    Object.assign({
+                        parse_mode: `html`,
+                    }, inlineButtons(data.id, true))
+                );
+            }
+        } catch (e) {
+            ctx.reply(`There was an error.`);
+            console.log(e);
+        }
+    });

handlers/fetch.js

@@ -0,0 +1,13 @@
+const handlers = [
+    `start`,
+    `search`,
+    `latest`,
+    `fetch`,
+    `inline`,
+    `callbacks`,
+];
+
+module.exports = bot =>
+    handlers.forEach(handler =>
+        require(`./${handler}`)(bot)
+    );

handlers/index.js

@@ -0,0 +1,41 @@
+const codebottle = require(`codebottle`);
+const { inlineButtons, formatText } = require(`../helpers`);
+
+module.exports = bot =>
+    bot.on(`inline_query`, async ctx => {
+        const results = [];
+        const { query } = ctx.inlineQuery;
+
+        try {
+            const data = await (query && query.length > 2 ?
+                codebottle.search({ query }) :
+                codebottle.latest);
+
+            data.forEach(snippet => {
+                results.push({
+                    type: `article`,
+                    id: snippet.id,
+                    title: snippet.title,
+                    input_message_content: {
+                        message_text: formatText(snippet),
+                        parse_mode: `html`,
+                        disable_web_page_preview: true,
+                    },
+                    url: `https://codebottle.io/s/${snippet.id}`,
+                    hide_url: true,
+                    ...inlineButtons(snippet.id, true),
+                });
+            });
+        } catch (e) {
+            results.push({
+                type: `article`,
+                id: `error`,
+                title: `There was an error.`,
+                message_text: `¯\\_(ツ)_/¯`,
+            });
+
+            console.log(e);
+        } finally {
+            ctx.answerInlineQuery(results.slice(0, 50));
+        }
+    });

handlers/inline.js

@@ -0,0 +1,18 @@
+const codebottle = require(`codebottle`);
+const { inlineButtons, formatText } = require(`../helpers`);
+
+module.exports = bot =>
+    bot.command(`latest`, async ctx => {
+        try {
+            const data = await codebottle.latest;
+            ctx.reply(
+                formatText(data[0]),
+                Object.assign({
+                    parse_mode: `html`,
+                }, inlineButtons(data[0].id, true))
+            );
+        } catch (e) {
+            ctx.reply(`There was an error.`);
+            console.log(e);
+        }
+    });

handlers/latest.js

@@ -0,0 +1,24 @@
+const codebottle = require(`codebottle`);
+const { inlineButtons, formatText } = require(`../helpers`);
+
+module.exports = bot =>
+    bot.command(`search`, async ctx => {
+        try {
+            let { text, entities } = ctx.message;
+            text = text.slice(entities[0].length + 1);
+
+            if (text) {
+                const data = await codebottle.search({ query: text });
+
+                ctx.reply(
+                    formatText(data[0]),
+                    Object.assign({
+                        parse_mode: `html`,
+                    }, inlineButtons(data[0].id, true))
+                );
+            }
+        } catch (e) {
+            ctx.reply(`There was an error.`);
+            console.log(e);
+        }
+    });

handlers/search.js

@@ -0,0 +1,2 @@
+module.exports = bot =>
+    bot.command(`start`, async ctx => ctx.reply(`Hi I'm CodeBottle Bot.`));

handlers/start.js

@@ -0,0 +1,15 @@
+const htmlEntities = {
+    '<': `&lt;`,
+    '>': `&gt;`,
+    '&': `&amp;`,
+    '"': `&quot;`,
+};
+
+const escapeHtml = html => {
+    return html.replace(
+        new RegExp(`[${Object.keys(htmlEntities).join(``)}]`, `g`),
+        char => htmlEntities[char]
+    );
+};
+
+module.exports = escapeHtml;

helpers/escapeHtml.js

@@ -0,0 +1,24 @@
+const escapeHtml = require(`./escapeHtml`);
+const truncate = require(`./truncate`);
+const md2html = require(`./md2html`);
+
+const formatText = (data, code) => {
+    let message = `<a href="https://codebottle.io/s/${data.id}">${data.title}</a> `;
+    message += `by <a href="https://codebottle.io/users/${data.username}">${data.username}</a>\n\n`;
+
+    if (code) {
+        message += `<pre>${escapeHtml(truncate(data.code))}</pre>`;
+    } else {
+        message += md2html(
+            escapeHtml(
+                truncate(
+                    data.description || `No description.`
+                )
+            )
+        );
+    }
+
+    return message;
+};
+
+module.exports = formatText;

helpers/formatText.js

@@ -0,0 +1,4 @@
+module.exports = {
+    inlineButtons: require(`./inlineButtons`),
+    formatText: require(`./formatText`),
+};

helpers/index.js

@@ -0,0 +1,11 @@
+const Markup = require(`telegraf/markup`);
+
+const inlineButtons = (id, code) =>
+    Markup.inlineKeyboard([
+        Markup.callbackButton(code ? `Code` : `Description`, `${code ? `code` : `desc`}_${id}`),
+        Markup.urlButton(`Open`, `https://codebottle.io/s/${id}`),
+    ]).extra({
+        disable_web_page_preview: true,
+    });
+
+module.exports = inlineButtons;

helpers/inlineButtons.js

@@ -0,0 +1,10 @@
+const md2html = text => {
+    return text
+        .replace(/\*{2}(.+?)\*{2}/g, `<i>$1</i>`)
+        .replace(/\*(.+?)\*/g, `<b>$1</b>`)
+        .replace(/`{3}\w*\n(.+?)\n`{3}/gs, `<pre>$1</pre>`)
+        .replace(/`(.+?)`/g, `<code>$1</code>`)
+        .replace(/\[(.+?)\]\((.+?)\)/g, `<a href="$2">$1</a>`);
+};
+
+module.exports = md2html;

helpers/md2html.js

@@ -0,0 +1,3 @@
+const truncate = code => code.length > 1000 ? `${code.slice(0, 1000)}...` : code;
+
+module.exports = truncate;