halbot 1989.6.4 → 1989.6.6

package/bin/halbot.mjs

@@ -0,0 +1,34 @@
+#!/usr/bin/env node
+
+import { storage, utilitas } from 'utilitas';
+import halbot from '../index.mjs';
+
+const debug = utilitas.humanReadableBoolean(process.env['DEBUG']);
+const log = content => utilitas.log(content, import.meta.url);
+
+// Disabled for current version.
+// const args = await (async (options) => {
+//     const { parseArgs } = await import('node:util');
+//     if (parseArgs) { // https://kgrz.io/node-has-native-arg-parsing.html
+//         const { values } = parseArgs({
+//             options: {
+//                 // xxx: { type: 'string', short: 'x', default: '' }
+//             },
+//             ...options || {},
+//         });
+//         return values;
+//     }
+//     let args = {}; // simple fallback for node version < 19
+//     process.argv.map(arg => {
+//         const item = arg.replace(/^\-*([^=]*)=(.*)$/ig, '$1<,>$2').split('<,>');
+//         item.length > 1 && (args[item[0]] = item[1]);
+//     });
+//     return args;
+// })();
+
+try {
+    const { config } = await storage.getConfig(
+        ...debug ? [{ pack: './package.json' }] : [], // utilitas debug
+    );
+    await halbot(config);
+} catch (err) { debug ? utilitas.throwError(err) : log(err); }

package/config.json

@@ -0,0 +1,6 @@
+{
+    "telegramToken": "// Telegram Bot API Token",
+    "chatGptKey": "// ChatGPT API Key",
+    "bingToken": "// Bing Usertoken from cookies",
+    "googleApiKey": "// Google Cloud API Key"
+}

package/index.mjs

@@ -0,0 +1,36 @@
+import { bot, hal, speech, utilitas } from 'utilitas';
+
+const skillPath = utilitas.__(import.meta.url, 'skills');
+
+const init = async (options) => {
+    assert(options?.telegramToken, 'Telegram Bot API Token is required.', 400);
+    const [ai, _speech] = [{}, {}];
+    if (options?.googleApiKey) {
+        await speech.init({ apiKey: options?.googleApiKey, tts: true, stt: true });
+        Object.assign(_speech, { stt: speech.stt, tts: speech.tts });
+    }
+    if (options?.chatGptKey) {
+        ai['ChatGPT'] = await hal.init({
+            provider: 'CHATGPT', clientOptions: { apiKey: options.chatGptKey },
+        });
+    }
+    if (options?.bingToken) {
+        ai['Bing'] = await hal.init({
+            provider: 'BING', clientOptions: { userToken: options.bingToken },
+        });
+    }
+    assert(utilitas.countKeys(ai), 'No AI provider is configured.', 400);
+    return await bot.init({
+        ai, auth: options?.auth,
+        botToken: options?.telegramToken,
+        chatType: options?.chatType,
+        homeGroup: options?.homeGroup,
+        magicWord: options?.magicWord,
+        private: options?.private,
+        provider: 'telegram', speech: _speech,
+        skillPath: options?.skillPath || skillPath,
+    });
+};
+
+export default init;
+export { bot, hal, speech, utilitas };

package/package.json

@@ -1,11 +1,39 @@
 {
   "name": "halbot",
-  "version": "1989.6.4",
-  "description": "",
+  "description": "Just another ChatGPT/Bing Telegram bob.",
+  "version": "1989.6.6",
+  "private": false,
+  "homepage": "https://github.com/Leask/halbot",
+  "type": "module",
+  "engines": {
+    "node": ">=18.x"
+  },
   "main": "index.js",
+  "bin": {
+    "halbot": "bin/halbot.mjs"
+  },
   "scripts": {
-    "test": "echo \"Error: no test specified\" && exit 1"
+    "start": "node bin/halbot.mjs",
+    "debug": "node --inspect --trace-warnings debug.mjs",
+    "test": "node --inspect --trace-warnings test.mjs",
+    "updep": "npx npm-check-updates -u && npm install",
+    "gitsync": "( git commit -am \"Released @ `date`\" || true ) && git pull && git push",
+    "build": "npm run updep && ( git commit -am 'update dependencies' || true )",
+    "pub": "npm run build && npm run gitsync",
+    "beta": "npm publish --tag beta"
+  },
+  "author": "Leask Wong <i@leaskh.com>",
+  "license": "MIT",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/Leask/halbot.git"
+  },
+  "dependencies": {
+    "@google-cloud/speech": "^5.4.0",
+    "@google-cloud/text-to-speech": "^4.2.1",
+    "@waylaidwanderer/chatgpt-api": "^1.28.10",
+    "telegraf": "^4.12.2",
+    "utilitas": "^1992.3.43"
   },
-  "author": "",
-  "license": "MIT"
+  "devDependencies": {}
 }

package/skills/ai.mjs

@@ -0,0 +1,70 @@
+import { utilitas } from 'utilitas';
+
+const onProgress = { onProgress: true };
+const [YOU, BOT, LN2] = ['😸 You:', '🤖️ ', '\n\n'];
+const [joinL1, joinL2] = [a => a.join('${LN2}---${LN2}'), a => a.join(LN2)];
+
+const action = async (bot) => {
+    bot.use(async (ctx, next) => {
+        if (ctx.end || !ctx.text) { return await next(); }
+        const [multiAi, msgs, tts, pms] = [ctx.session.ai.size > 1, {}, {}, []];
+        let [lastMsg, lastSent] = ['', 0];
+        const packMsg = (options) => {
+            const packed = [...ctx.stt ? joinL2([YOU, ctx.stt]) : []];
+            const source = options?.tts ? tts : msgs;
+            for (let name of ctx.session.ai.size ? ctx.session.ai : [ctx.firstAi]) {
+                const defaultAi = name === ctx.firstAi;
+                packed.push(joinL2([
+                    ...multiAi || !defaultAi || ctx.stt ? [`${BOT}${name}:`] : [],
+                    options?.onProgress ? (
+                        source[name] ? `${source[name].trim()} |` : '...'
+                    ) : (source[name] || ''),
+                ]));
+            }
+            return joinL1(packed);
+        };
+        const ok = async (options) => {
+            const [curTime, curMsg] = [Date.now(), packMsg(options)];
+            if (options?.onProgress && (
+                curTime - lastSent < ctx.limit || lastMsg === curMsg
+            )) { return; }
+            [lastSent, lastMsg] = [curTime, curMsg];
+            return await ctx.ok(curMsg, options);
+        };
+        await ok(onProgress);
+        for (let name of ctx.session.ai.size ? ctx.session.ai : [ctx.firstAi]) {
+            if (utilitas.insensitiveCompare('/clear', ctx.text)) {
+                bot.ai[name].clear(ctx.chatId);
+                ctx.text = bot.hello;
+            }
+            pms.push((async () => {
+                try {
+                    const resp = await bot.ai[name].send(
+                        ctx.text, { session: ctx.chatId },
+                        tkn => {
+                            msgs[name] = `${(msgs[name] || '')}${tkn}`;
+                            ok(onProgress);
+                        }
+                    );
+                    msgs[name] = resp.responseRendered;
+                    tts[name] = resp.spokenText;
+                } catch (err) {
+                    msgs[name] = `[ERROR] ${err?.message || err}`;
+                    tts[name] = msgs[name];
+                    utilitas.log(err);
+                }
+            })());
+        }
+        await Promise.all(pms);
+        await ok();
+        ctx.tts = packMsg({ tts: true });
+        await next();
+    });
+};
+
+export const { run, priority, func } = {
+    run: true,
+    priority: 30,
+    name: 'ai',
+    func: action,
+};

package/skills/cmd.mjs

@@ -0,0 +1,29 @@
+import { utilitas } from 'utilitas';
+
+const matchReg = /^\/([^\ ]*)(.*)$/ig;
+
+const action = async (bot) => {
+    bot.use(async (ctx, next) => {
+        if (ctx.end || !ctx.text) { return await next(); }
+        ctx.session.ai || (ctx.session.ai = new Set());
+        const curAi = new Set();
+        ctx.cmd = ctx.text.split('\n')?.[0]?.replace(matchReg, '$1');
+        for (let name in bot.ai) {
+            ctx.firstAi || (ctx.firstAi = name);
+            (utilitas.insensitiveCompare(ctx.cmd, name) || ctx.cmd === '*')
+                && curAi.add(name);
+        }
+        curAi.size && (ctx.session.ai = curAi);
+        ctx.session.ai.size && (
+            ctx.text = ctx.text.replace(matchReg, '$2').trim() || bot.hello
+        );
+        await next();
+    });
+};
+
+export const { run, priority, func } = {
+    run: true,
+    priority: 10,
+    name: 'ai',
+    func: action,
+};

package/skills/translate.mjs

@@ -0,0 +1,40 @@
+const getTranslatePrompt = (lang) => // https://github.com/yetone/bob-plugin-openai-translator/blob/main/src/main.js
+    'You are a translation engine that can only translate text and cannot interpret it.'
+    + ` Translate the following text into ${lang}: `;
+const getPolishPrompt = () => // https://github.com/yetone/bob-plugin-openai-polisher/blob/main/src/main.js
+    'Revise the following text to make them more clear, concise, and coherent.'
+    + ' Please note that you need to list the changes and briefly explain why: ';
+
+const action = async (bot) => {
+    bot.use(async (ctx, next) => {
+        if (ctx.end || !ctx.text) { return await next(); }
+        switch (ctx.cmd) {
+            case '2en':
+                ctx.text = ctx.text.replace(`/${ctx.cmd}`, getTranslatePrompt('English'));
+                break;
+            case '2zh':
+                ctx.text = ctx.text.replace(`/${ctx.cmd}`, getTranslatePrompt('Traditional Chinese'));
+                break;
+            case '2zht':
+                ctx.text = ctx.text.replace(`/${ctx.cmd}`, getTranslatePrompt('Traditional Chinese'));
+                break;
+            case '2zhs':
+                ctx.text = ctx.text.replace(`/${ctx.cmd}`, getTranslatePrompt('Simplified Chinese'));
+                break;
+            case '2fr':
+                ctx.text = ctx.text.replace(`/${ctx.cmd}`, getTranslatePrompt('French'));
+                break;
+            case 'p':
+                ctx.text = ctx.text.replace(`/${ctx.cmd}`, getPolishPrompt());
+                break;
+        }
+        await next();
+    });
+};
+
+export const { run, priority, func } = {
+    run: true,
+    priority: 20,
+    name: 'ai',
+    func: action,
+};