const Discord = require("discord.js");
const fetch = require("node-fetch");
const fs = require('fs');

const server = "192.168.1.213";

class BotChat {

        async get_response(history, input_text) {
	      let historyNoEOL = history;
	      if(history.endsWith("\n")) { //get rid of extra newline if it's there
		      historyNoEOL = history.slice(0,-1);
	      }
	      const complete_prompt = this.bot_prompt.replace("<CONVHISTORY>",historyNoEOL).replace("<INTEXT>",input_text);

	      console.log("PROMPT:" + complete_prompt);

	      const response = await fetch(`http://${server}:7860/run/textgen`, {
		method: "POST",
		headers: { "Content-Type": "application/json" },
		body: JSON.stringify({
			data: [
				complete_prompt,
				100, //max tokens
				true, //do_sample
				0.44, //temperature
				1, //top_p
				1, //typical_p
				1.15, //rep penalty
				1.0, //encoder rep penalty
				0, //top_k
				0, //min_length
				16, //no_repeat_ngram_size
				1, //num_beams
				0, //penalty_alpha
				1, //length_penalty
				false, //early_stopping
				-1 //seed
			]
		})
	});

                const response_json = await response.json();
		const full_text = response_json.data[0];
		const completion = full_text.slice(complete_prompt.length);
		console.log(`RESPONSE: ${completion}`);
        	//return response_json.text;
        	return completion;
        }

	constructor() {
		this.matches = [ "botchat", "botmention", "botrandom" ];
		this.bot_prompt = fs.readFileSync("bot-modules/botchat/prompts/default.txt").toString();
	}

	async getRecentChannelHistory(chan) {
		let history = "";
		// Fetch the last 10 messages in the channel
		const messages = await chan.messages.fetch({ limit: 10 });
		// Reverse the order of the messages so that the oldest one is first
		const reversed = messages.reverse();
		// Loop through the messages and print their content and author
		for (let msg of reversed) {
		  const user = msg[1].author.username;
		  const content = msg[1].content;
		  history += `${user}: ${content}\n`;
		}
		const fixedString = this.inputFixup(chan, history);
		return fixedString;
	}

	// Replace all the user IDs with their actual usernames
	async inputFixup(chan,str) {
		const client = chan.client;

		const userStrs = str.match(/<@.+?>/g);
		const uniqueIDs = [...new Set(userStrs)];
		const translationTable = {};

		for (let id of uniqueIDs) {
			// Get the user object from the ID
			const flake = id.slice(2,-1);
			const user = await client.users.fetch(`${flake}`);
			// Check if the user exists
			if (user) {
				translationTable[id] = user.username;
			}
		}

		console.log(JSON.stringify(translationTable));
		let new_input = str.replace("!botchat","bot,");

		for (let t in translationTable) {
			new_input = new_input.replaceAll(t, translationTable[t]);
		}

		return new_input;
	}

	async outputFixup(str) {
		const outLines = str.trim().split("\n");
		let outStr = "";
		
		for (let ln of outLines) {
			if(ln.startsWith("dank-bot:")) {
				const trucStr = ln.replace("dank-bot:","");
				outStr += "\n" + trucStr.trim();
			} else if (ln.indexOf(":") > 0 && ln.indexOf(":") < 20 ) {
				break;
			} else {
				outStr += ln;
			}
		}

		return outStr.trim();
	}

	async action(msg, botrandom) {
	    let chan = msg.channel;
	    try {
		    //const input_text = msg.content.replace("!botchat","").replace(/<@.+?>/,"bot,").trim();
		    const input_text = await this.inputFixup(chan, msg.content.trim());
		    await chan.sendTyping();
		    console.log(`Input Text: ${input_text}`);
		    const history = await this.getRecentChannelHistory(chan);
		    const bot_response = await this.get_response(history,input_text);
		    const response_only = await this.outputFixup(bot_response);
		    chan.send(response_only);
	    } catch (e) {
		    chan.send(`Error: ${e}`);
	    }
	}

}

module.exports = BotChat;