const colors = require("colors");
require("dotenv").config();
const fs = require("fs");
const express = require("express");
const app = express();
const port = process.env.PORT || 3000;
const { exec } = require("child_process");
const sqlite3 = require("sqlite3").verbose();
const db = new sqlite3.Database("./database.db");
// Find all migrations .sql files in ./migrations, execute them in order
db.on("open", () => {
	console.log(`${colors.cyan("[DB]")} Connected to the database`);
	fs.readdirSync("./migrations").forEach((file) => {
		if (file.endsWith(".sql")) {
			const migration = fs.readFileSync(`./migrations/${file}`, "utf8");
			db.run(migration);
			console.log(`${colors.cyan("[DB]")} ${file} ${colors.green("executed")}`);
		}
	});
});

const Discord = require("discord.js");
const {
	REST,
	Routes
} = require('discord.js');
const { title } = require("process");
const rest = new REST({
	version: '10'
}).setToken(process.env.DISCORD_TOKEN);
const client = new Discord.Client({
	intents: [
		"Guilds"
	]
});

// Vars

var handledTransactions = [];

// Funcs

// runCommand(command) // Run a shell command and return the output
function runCommand(command) {
	return new Promise((resolve, reject) => {
		exec(command, (error, stdout, stderr) => {
			if (error) {
				console.error(`exec error: ${error}`);
				reject(error);
			} else {
				resolve(stdout);
			}
		});
	});
}



// generateAccountNumber() // Returns a random 10 digit number, checks if it already exists in the database, and loops until it gets one that doesn't exist
function generateAccountNumber() {
	let accountNumber = Math.floor(Math.random() * 10000000000);
	db.get("SELECT * FROM accounts WHERE id = ?", accountNumber, (err, row) => {
		if (err) {
			console.error(err);
		} else if (row) {
			accountNumber = generateAccountNumber();
		}
	});
	return accountNumber;
}

// generateTransactionNumber() // Returns a random 10 digit number
function generateTransactionNumber() {
	return Math.floor(Math.random() * 10000000000);
}

function sendAlert(accountNumber, transaction, placeName, systemName, zoneNumber, zoneName, event) {
	if (handledTransactions.includes(transaction)) {
		return; // Duplicate transaction
	}
	handledTransactions.push(transaction);
	// Check if the account exists and is verified
	db.get("SELECT * FROM accounts WHERE id = ? AND verified = 1", accountNumber, (err, row) => {
		if (err) {
			console.error(err);
		} else if (row) {
			// Account exists and is verified
			// Send the alert
			runCommand(`flite -t "Hello. This is an automated call from KCA SecuriNet Monitoring. ${systemName} has reported an ${event}, ZONE ${zoneNumber}, ${zoneName}, at ${placeName}" -o /tmp/${transaction}.wav`).then((output) => {
				runCommand(`ffmpeg -y -i /tmp/${transaction}.wav -ar 8000 -ac 1 -c:a pcm_s16le /tmp/${transaction}-alert.wav`).then(() => {
					runCommand(`rm /tmp/${transaction}.wav`)
					// strip extension from filename

					runCommand(`/var/lib/asterisk/bin/originate ${row.phone} roblox.s.1 0 0 /tmp/${transaction}-alert "Ik5vb24gQ2hpbWUiIDw+"`).then(() => {
						console.log(`Alert sent to ${row.phone}`);
					})
				})
			})
		} else {
			return;
			// Account does not exist or is not verified
		}
	});
}

function sendVerificationCode(account) {
	// Get verification code from database
	db.get("SELECT * FROM accounts WHERE id = ?", account, (err, row) => {
		if (err) {
			console.error(err);
		} else if (row) {
			// Send verification code to phone number
			runCommand(`flite -t "Hello. This is an automated call from KCA SecuriNet Monitoring. Your verification code is ${row.verification_code.split("").join(", ")}. Repeating, your code is ${row.verification_code}. Once again, your code is ${row.verification_code}" -o /tmp/${account}-code.wav`).then((output) => {
				runCommand(`ffmpeg -y -i /tmp/${account}-code.wav -ar 8000 -ac 1 -c:a pcm_s16le /tmp/${account}-verification.wav`).then(() => {
					runCommand(`rm /tmp/${account}-code.wav`)
					// strip extension from filename

					runCommand(`/var/lib/asterisk/bin/originate ${row.phone} roblox.s.1 0 0 /tmp/${account}-verification "Ik5vb24gQ2hpbWUiIDw+"`).then(() => {
						console.log(`Verification code sent to ${row.phone}`);
					})
				})
			})
		} else {
			return;
			// Account does not exist or is not verified
		}
	});
}

function generatePhoneCode() {
	// generate 6 digit
	return Math.floor(100000 + Math.random() * 900000).toString();
}

client.on("ready", async () => {
	console.log(`${colors.cyan("[Discord]")} Logged in as ${client.user.tag}`);

	const commands = require("./commands.json");
	//Register commands
	await (async () => {
		try {
			console.log(`${colors.cyan("[Discord]")} Registering Commands...`)
			//Global
			await rest.put(Routes.applicationCommands(client.user.id), { body: commands })
			console.log(`${colors.cyan("[Discord]")} Successfully registered commands. Took ${colors.green((Date.now() - startTime) / 1000)} seconds.`);
		} catch (error) {
			console.error(error);
		}
	})();
	app.listen(port, () => {
		console.log(`${colors.cyan("[EXPRESS]")} Listening on ${port}`);
	});
});

client.on("interactionCreate", async (interaction) => {
	if (!interaction.isCommand()) return;

	switch (interaction.commandName) {
		case "register":
			phone_number = interaction.options.getString("phone_number");
			// Check that phone_number is either a 4 digit number starting with 1, a 7 digit number, a 10 digit number, or an 11 digit number starting with 1
			if (!/^(1\d{3}|\d{7}|\d{10}|1\d{10})$/.test(phone_number)) {
				interaction.reply("Invalid phone number. Please enter one of the following:\n- A 4 digit LiteNet extension.\n- A 7 digit TandmX number\n- An 10 digit US phone number.\n- An 11 digit US phone number");
				return;
			}
			// check that the user doesnt have any unverified accounts already (check discord_id and verified)
			db.get("SELECT * FROM accounts WHERE discord_id = ? AND verified = 0", interaction.user.id, (err, row) => {
				if (err) {
					console.error(err);
				} else if (row) {
					interaction.reply("You already have an unverified account. Please verify it before creating a new one.");
				} else {
					accountNumber = generateAccountNumber();
					verification_code = generatePhoneCode();
					db.run("INSERT INTO accounts (id, discord_id, verification_code, phone) VALUES (?, ?, ?, ?)", accountNumber, interaction.user.id, verification_code, phone_number, (err) => {
						if (err) {
							console.error(err);
						} else {
							sendVerificationCode(accountNumber);
							interaction.reply({
								content: `Account created. Please verify your account by entering the verification code sent to ${phone_number}`,
								ephemeral: true
							});
						}
					});
				}
			});
			break;
		case "verify":
			verification_code = interaction.options.getString("verification_code");
			db.get("SELECT * FROM accounts WHERE verification_code = ? AND verified = 0", verification_code, (err, row) => {
				if (err) {
					console.error(err);
				} else if (row) {
					db.run("UPDATE accounts SET verified = 1 WHERE verification_code = ?", verification_code, (err) => {
						if (err) {
							console.error(err);
						} else {
							interaction.reply({
								content: `Account Verified! Your account number is \`${row.id}\`.\nFor help setting up the dialer, feel free to contact a member of staff!`,
							})
						}
					});
				} else {
					interaction.reply("Invalid verification code.");
				}
			});
			break;
		case "resend":
			// Find the account thats unverified owned by the user
			db.get("SELECT * FROM accounts WHERE discord_id = ? AND verified = 0", interaction.user.id, (err, row) => {
				if (err) {
					console.error(err);
				} else if (row) {
					sendVerificationCode(row.id);
					interaction.reply({
						content: `Verification code resent to ${row.phone}`,
						ephemeral: true
					});
				} else {
					interaction.reply("You don't have an unverified account.");
				}
			});
			break;
		case "list":
			// list all active accounts owned by the user
			db.all("SELECT * FROM accounts WHERE discord_id = ? AND verified = 1", interaction.user.id, (err, rows) => {
				if (err) {
					console.error(err);
				} else if (rows) {
					let accountList = "";
					rows.forEach((row) => {
						accountList += `\`${row.id}\` - \`${row.phone}\`\n`;
					});
					interaction.reply({
						content: `Active accounts:\n${accountList}`,
						ephemeral: true
					});
				} else {
					interaction.reply("You don't have any active accounts.");
				}
			});
			break;
		case "deactivate": // Deactivate an account
			// Check that account_number is owned by the user, if it is, delete the row
			accountNumber = interaction.options.getInteger("account_number");
			db.get("SELECT * FROM accounts WHERE discord_id = ? AND id = ?", interaction.user.id, accountNumber, (err, row) => {
				if (err) {
					console.error(err);
				} else if (row) {
					db.run("DELETE FROM accounts WHERE id = ?", accountNumber, (err) => {
						if (err) {
							console.error(err);
						} else {
							interaction.reply({
								content: `Account \`${accountNumber}\` deactivated.`,
								ephemeral: true
							});
						}
					});
				} else {
					interaction.reply("You don't own that account.");
				}
			});
			break;
		case "update": // If the account is owned by the user and verified, update the phone number, unverify it, and send verification to the new number. Do the same checks as register
			accountNumber = interaction.options.getString("account_number");
			phone_number = interaction.options.getString("phone_number");
			db.get("SELECT * FROM accounts WHERE discord_id = ? AND id = ? AND verified = 1", interaction.user.id, accountNumber, (err, row) => {
				if (err) {
					console.error(err);
				} else if (row) {
					if (!/^(1\d{3}|\d{7}|\d{10}|1\d{10})$/.test(phone_number)) {
						interaction.reply("Invalid phone number. Please enter one of the following:\n- A 4 digit LiteNet extension.\n- A 7 digit TandmX number\n- An 10 digit US phone number.\n- An 11 digit US phone number");
						return;
					}
					verification_code = generatePhoneCode();
					db.run("UPDATE accounts SET phone = ?, verified = 0, verification_code = ? WHERE id = ?", phone_number, verification_code, accountNumber, (err) => {
						if (err) {
							console.error(err);
						} else {
							sendVerificationCode(accountNumber);
							interaction.reply({
								content: `Account updated. Please verify your account by entering the verification code sent to ${phone_number}`,
								ephemeral: true
							});
						}
					});
				} else {
					interaction.reply("You don't own that account.");
				}
			});
			break;
	}
});


startTime = new Date();
client.login(process.env.DISCORD_TOKEN);