discord.js/src/Util/TokenCacher.js

"use strict";
/* global process */

import fs from "fs-extra";
import EventEmitter from "events";
import crypto from "crypto";

var savePaths = [
	process.env.APPDATA || (process.platform == "darwin" ? process.env.HOME + "Library/Preference" : "/var/local"),
	process.env[(process.platform == "win32") ? "USERPROFILE" : "HOME"],
	process.cwd()
];

var algo = "aes-256-ctr";

function secureEmail(email, password) {
	return new Buffer(crypto.createHash("sha256").update(email + password, "utf8").digest()).toString("hex");
}

export default class TokenCacher extends EventEmitter {

	constructor(client, options) {
		super();
		this.client = client;
		this.savePath = null;
		this.error = false;
		this.done = false;
		this.data = {};
	}

	setToken(email="", password="", token="") {
		email = secureEmail(email, password);
		var cipher = crypto.createCipher(algo, password)
		var crypted = cipher.update("valid" + token, "utf8", "hex")
		crypted += cipher.final("hex");
		this.data[email] = crypted;
		this.save();
	}

	save() {
		fs.writeJson(this.savePath, this.data);
	}

	getToken(email="", password="") {

		email = secureEmail(email, password);

		if (this.data[email]) {

			try {
				var decipher = crypto.createDecipher(algo, password)
				var dec = decipher.update(this.data[email], "hex", "utf8");
				dec += decipher.final("utf8");
				return (dec.indexOf("valid") === 0 ? dec.substr(5) : false);
			} catch (e) {
				// not a valid token
				return null;
			}

		} else {
			return null;
		}

	}

	init(ind) {

		var self = this;
		var savePath = savePaths[ind];

		fs.ensureDir(savePath, err => {

			if (err) {
				error(err);
			} else {
				//good to go

				fs.ensureFile(savePath + "/.discordjs/tokens.json", err => {
					if (err) {
						error(err);
					} else {
						//file exists

						fs.readFile(savePath + "/.discordjs/tokens.json", (err, data) => {

							if (err) {
								error(err);
							} else {
								// can read file, is it valid JSON?
								try {

									this.data = JSON.parse(data);
									// good to go!
									this.savePath = savePath + "/.discordjs/tokens.json";
									this.emit("ready");
									this.done = true;

								} catch (e) {
									// not valid JSON, make it valid and then write
									fs.writeJson(savePath + "/.discordjs/tokens.json", {}, err => {
										if (err) {
											error(err);
										} else {
											// good to go!
											this.savePath = savePath + "/.discordjs/tokens.json";
											this.emit("ready");
											this.done = true;
										}
									});
								}
							}

						});

					}
				})

			}

		});

		function error(e) {
			ind++;
			if (!savePaths[ind]) {
				self.emit("error");
				self.error = e;
				self.done = true;
			} else {
				self.init(ind);
			}
		}

	}
}