lydia.js

/* Copyright (C) 2020 Yusuf Usta.

Licensed under the  GPL-3.0 License;
you may not use this file except in compliance with the License.

WhatsAsena - Yusuf Usta
*/

const Asena = require('../events');
const Config = require('../config');
const {MessageType} = require('@adiwajshing/baileys');
const {LydiaDB} = require('./sql/lydia');
const {LydiaAI} = require('coffeehouse');

const Language = require('../language');
const Lang = Language.getString('lydia');

Asena.addCommand({pattern: 'addlydia$', fromMe: true, desc: Lang.ADDLYDIA_DESC}, (async (message, match) => {
    if (!message.reply_message) return await message.reply(Lang.NEED_REPLY);
    if (!Config.COFFEEHOUSE_API_KEY) return await message.reply(Lang.COFFEEHOUSE);
    var unix = Date.now() / 1000 | 0;

    var veriler = await LydiaDB.findAll();
    var lydia = new LydiaAI(Config.COFFEEHOUSE_API_KEY);

    if (veriler.length < 1) {
        var session = await lydia.createSession();
        await LydiaDB.create({session_id: session.id, expires: session.expires, users: message.reply_message.jid.split('@')[0]});
    } else {
        if (unix > veriler[0].dataValues.expires) {
            var session = await lydia.createSession();
            await veriler[0].update({session_id: session.id, expires: session.expires, users: veriler[0].dataValues.users + ',' + message.reply_message.jid.split('@')[0]});
        } else {
            await veriler[0].update({users: veriler[0].dataValues.users + (veriler[0].dataValues.users != '' ? ',' : '') + message.reply_message.jid.split('@')[0]});
        }
    }

    return await message.reply(Lang.ENABLED_LYDIA);
}));

Asena.addCommand({pattern: 'rmlydia$', fromMe: true, desc: Lang.RMLYDIA_DESC}, (async (message, match) => {
    if (!message.reply_message) return await message.reply(Lang.NEED_REPLY);
    if (!Config.COFFEEHOUSE_API_KEY) return await message.reply(Lang.COFFEEHOUSE);
    var unix = Date.now() / 1000 | 0;

    var veriler = await LydiaDB.findAll();
    var lydia = new LydiaAI(Config.COFFEEHOUSE_API_KEY);

    if (veriler.length < 1) {
        return await message.reply(Lang.ALREADY_EMPTY)
    } else {
        if (veriler[0].dataValues.users == '') return await message.reply(Lang.ALREADY_EMPTY)
        var users = veriler[0].dataValues.users.split(',');
        if (users.includes(message.reply_message.jid.split('@')[0])) { users = users.remove(message.reply_message.jid.split('@')[0]).join(','); } else {
            return await message.reply(Lang.NOT_ADDED);
        }

        if (unix > veriler[0].dataValues.expires) {
            var session = await lydia.createSession();
            await veriler[0].update({'session_id': session.id, 'expires': session.expires, 'users': users});
        } else {
            await veriler[0].update({'users': users});
        }
    }

    return await message.reply(Lang.DISABLED);
}));

Asena.addCommand({on: 'text', fromMe: false, deleteCommand: false}, (async (message, match) => {   
    if (Config.COFFEEHOUSE_API_KEY === false) return;
    var unix = Date.now() / 1000 | 0;

    var fromId;
    if (message.jid.includes('-')) {
        if (message.reply_message === false && message.mention === false) return;
        if (message.reply_message && (message.reply_message.jid.split('@')[0] != message.client.user.jid.split('@')[0])) return;
        if (message.mention !== false && message.mention.length >= 1 && (message.mention[0].split('@')[0] != message.client.user.jid.split('@')[0])) return;
    }

    if (message.data.participant) { fromId = message.data.participant.split('@')[0]; } else { fromId = message.jid.split('@')[0]; }
    var veriler = await LydiaDB.findAll();
    var lydia = new LydiaAI(Config.COFFEEHOUSE_API_KEY);
    var session;

    if (veriler.length < 1) {
        return;
    } else {
        if (unix > veriler[0].dataValues.expires) {
            session = await lydia.createSession();
            await veriler[0].update({session_id: session.id, expires: session.expires});
        } else {
            session = await lydia.getSession(veriler[0].dataValues.session_id);
        }
        
        var isLydiaEnabled = false;
        if (veriler[0].users.includes(',')) {
            veriler[0].users.split(',').map((user) => {
                if (user == fromId) isLydiaEnabled = true;
            });
        } else {
            if (veriler[0].users == fromId) isLydiaEnabled = true;
        }

        if (isLydiaEnabled) {
            await message.sendTyping();

            var mesaj = await session.think_thought(message.message);
            await message.client.sendMessage(message.jid,mesaj.data.payload.output, MessageType.text, {quoted: message.data});
        }
    }
}));