// If you aren't running this script inside of the repository, replace the following line with:
// const SteamCommunity = require('steamcommunity');
const SteamCommunity = require('../index.js');
const SteamSession = require('steam-session');
const SteamTotp = require('steam-totp');
const ReadLine = require('readline');

const EConfirmationType = SteamCommunity.ConfirmationType;

let g_AbortPromptFunc = null;

let community = new SteamCommunity();

main();
async function main() {
	let accountName = await promptAsync('Username: ');
	let password = await promptAsync('Password (hidden): ', true);

	// Create a LoginSession for us to use to attempt to log into steam
	let session = new SteamSession.LoginSession(SteamSession.EAuthTokenPlatformType.MobileApp);

	// Go ahead and attach our event handlers before we do anything else.
	session.on('authenticated', async () => {
		abortPrompt();

		let cookies = await session.getWebCookies();
		community.setCookies(cookies);

		doConfirmations();
	});

	session.on('timeout', () => {
		abortPrompt();
		console.log('This login attempt has timed out.');
	});

	session.on('error', (err) => {
		abortPrompt();

		// This should ordinarily not happen. This only happens in case there's some kind of unexpected error while
		// polling, e.g. the network connection goes down or Steam chokes on something.

		console.log(`ERROR: This login attempt has failed! ${err.message}`);
	});

	// Start our login attempt
	let startResult = await session.startWithCredentials({accountName, password});
	if (startResult.actionRequired) {
		// Some Steam Guard action is required. We only care about email and device codes; in theory an
		// EmailConfirmation and/or DeviceConfirmation action could be possible, but we're just going to ignore those.
		// If the user does receive a confirmation and accepts it, LoginSession will detect and handle that automatically.
		// The only consequence of ignoring it here is that we don't print a message to the user indicating that they
		// could accept an email or device confirmation.

		let codeActionTypes = [SteamSession.EAuthSessionGuardType.EmailCode, SteamSession.EAuthSessionGuardType.DeviceCode];
		let codeAction = startResult.validActions.find(action => codeActionTypes.includes(action.type));
		if (codeAction) {
			if (codeAction.type == SteamSession.EAuthSessionGuardType.EmailCode) {
				// We wouldn't expect this to happen since mobile confirmations are only possible with 2FA enabled, but just in case...
				console.log(`A code has been sent to your email address at ${codeAction.detail}.`);
			} else {
				console.log('You need to provide a Steam Guard Mobile Authenticator code.');
			}

			let code = await promptAsync('Code or Shared Secret: ');
			if (code) {
				// The code might've been a shared secret
				if (code.length > 10) {
					code = SteamTotp.getAuthCode(code);
				}
				await session.submitSteamGuardCode(code);
			}

			// If we fall through here without submitting a Steam Guard code, that means one of two things:
			//   1. The user pressed enter without providing a code, in which case the script will simply exit
			//   2. The user approved a device/email confirmation, in which case 'authenticated' was emitted and the prompt was canceled
		}
	}
}

async function doConfirmations() {
	let identitySecret = await promptAsync('Identity Secret: ');

	let confs = await new Promise((resolve, reject) => {
		let time = SteamTotp.time();
		let key = SteamTotp.getConfirmationKey(identitySecret, time, 'conf');
		community.getConfirmations(time, key, (err, confs) => {
			if (err) {
				return reject(err);
			}

			resolve(confs);
		});
	});

	console.log(`Found ${confs.length} outstanding confirmations.`);

	// We need to track the previous timestamp we used, as we cannot reuse timestamps.
	let previousTime = 0;

	for (let i = 0; i < confs.length; i++) {
		let conf = confs[i];

		process.stdout.write(`Accepting confirmation for ${EConfirmationType[conf.type]} - ${conf.title}... `);

		try {
			await new Promise((resolve, reject) => {
				let time = SteamTotp.time();
				if (time == previousTime) {
					time++;
				}

				previousTime = time;
				let key = SteamTotp.getConfirmationKey(identitySecret, time, 'allow');
				conf.respond(time, key, true, (err) => {
					err ? reject(err) : resolve();
				});
			});

			console.log('success');
		} catch (ex) {
			console.log(`error: ${ex.message}`);
		}

		// sleep 500ms so we don't run too far away from the current timestamp
		await new Promise(resolve => setTimeout(resolve, 500));
	}

	console.log('Finished processing confirmations');
	process.exit(0);
}

// Nothing interesting below here, just code for prompting for input from the console.

function promptAsync(question, sensitiveInput = false) {
	return new Promise((resolve) => {
		let rl = ReadLine.createInterface({
			input: process.stdin,
			output: sensitiveInput ? null : process.stdout,
			terminal: true
		});

		g_AbortPromptFunc = () => {
			rl.close();
			resolve('');
		};

		if (sensitiveInput) {
			// We have to write the question manually if we didn't give readline an output stream
			process.stdout.write(question);
		}

		rl.question(question, (result) => {
			if (sensitiveInput) {
				// We have to manually print a newline
				process.stdout.write('\n');
			}

			g_AbortPromptFunc = null;
			rl.close();
			resolve(result);
		});
	});
}

function abortPrompt() {
	if (!g_AbortPromptFunc) {
		return;
	}

	g_AbortPromptFunc();
	process.stdout.write('\n');
}