/**
 * Скрипт получения метрик по балансу абонента МТС
 * 1.0
 *
 * Параметры запуска:
 *   phantomjs --web-security=no getMtsBalance.js "<папка для вывода результатов>" "телефон в формате (XXX) XXX-XX-XX" "<пароль>"
 * Пример:
 *   phantomjs --web-security=no getMtsBalance.js "/tmp/getMtsBalance" "(916) 123-45-67" "P@ssw0rd"
 *
 * (c) Jet/ДСУ 2016
 */

// PhantomJs stuff
var fs      = require('fs');
var system  = require('system');
var webpage = require('webpage');
var args    = system.args;

var TRAFFIC_REGEX = /Доступно.{1,100}?([0-9.]+).{0,50}?(ГБ|МБ).{0,50}?на (\d+) дн/i;
var DATE_REGEX = /Дата обновления интернет-пакета ([0-9]{1,2})\.([0-9]{1,2})\.([0-9]{2}) ([0-2][0-9]):([0-5][0-9])/;
var SCRIPT_TIMEOUT = 40000;

var config = {
	lkUrl  : 'https://lk.ssl.mts.ru/',
	lkLogin: null,
	lkPass : null,
	outDir : null,
	debugDir: null,
	formattedStartTime: null
};

var timer = {
	lastActionStartTime: null,
	lastActionTimeMs : null,
	
	currentTimeMillis: function() {
		return Date.now();
	},
	
	startAction: function() {
		this.lastActionStartTime = this.currentTimeMillis();
		this.lastActionTimeMs = null;
	},
	
	stopAction: function() {
		lastActionTimeMs = this.currentTimeMillis() - this.lastActionStartTime;
	},
	
	getLastActionTimeMs: function() {
		return lastActionTimeMs;
	},
	
	getLastActionTimeSec: function() {
		return lastActionTimeMs === null ? null : lastActionTimeMs/1000;
	}
}

var metrics = {
	trafLeftMb: null,
	daysLeft  : null,
	balance   : null,
	pages: {
		login: {
			availability   : null,
			responseTimeSec: null
		},
		lk: {
			availability   : null,
			responseTimeSec: null			
		}
	}
};

////////// HELPER FUNCTIONS //////////
var func = {
	log: function(s) {
		console.log(this.formatDateTimeForLog(new Date()) + " " + s);
	},
	
	roundToTwo: function(num) {    
	    return +(Math.round(num + "e+2")  + "e-2");
	},

	zero: function(i) {
		return i < 10 ? '0' + i : i;
	},

	formatDateTimeForLog: function(date) {
		var dd  = date.getDate();
		var mm  = date.getMonth() + 1;
		var yy  = date.getFullYear();  
		var hh  = date.getHours();
		var min = date.getMinutes();
		var ss  = date.getSeconds();
		var ms  = date.getMilliseconds();
		ms = ('00' + ms).slice(-3);
	  
		return yy + '-' + this.zero(mm) + '-' + this.zero(dd) + ' ' + this.zero(hh) + ':' + this.zero(min) + ':' + this.zero(ss) + '.' + ms;
	},
	
	formatDateTimeForFileName: function(date) {	  
		return date.getFullYear()
			+ this.zero(date.getMonth() + 1) 
			+ this.zero(date.getDate())
			+ '-' 
			+ this.zero(date.getHours()) 
		 	+ this.zero(date.getMinutes())
		;
	},
	
	writeMetricToFileAndLog: function(filePrefix, metricName, metricValue) {
		if ( metricValue == null ) {
			metricValue = 0;
		}
		fs.write(config.outDir + filePrefix + config.formattedStartTime + '.log', this.roundToTwo(metricValue), 'w');
		this.log('  ' + metricName + ' = ' + metricValue);	
	}
}

// Разбираем аргументы запуска скрипта
config.outDir  = args[1] + '/';
config.lkLogin = args[2];
config.lkPass  = args[3];

config.debugDir = config.outDir + 'debug/';
fs.makeDirectory(config.debugDir);

func.log("Папка с результатами работы: " + config.outDir);

// Таймаут - чтобы процесс навечно не завис
setTimeout(function() {
	func.log("Сработал таймаут после " + SCRIPT_TIMEOUT + " мс");
	if ( metrics.pages.login.availability == null ) {
		metrics.pages.login.availability = 0;
		metrics.pages.login.responseTimeSec = 0;
	}
	if ( metrics.pages.lk.availability == null ) {
		metrics.pages.lk.availability = 0;
		metrics.pages.lk.responseTimeSec = 0;
	}
	outMetricsAndExit();
}, SCRIPT_TIMEOUT);

// Настраиваем наш "браузер"
var page = webpage.create();
page.settings.userAgent = 'Mozilla/4.0';

// Отмечаем время начала процесса для логов
config.formattedStartTime = func.formatDateTimeForFileName(new Date());

// Открываем страницу личного кабинета
func.log("Загружаем " + config.lkUrl);
timer.startAction();

page.open(config.lkUrl, function (status) {
	timer.stopAction();
	metrics.pages.login.responseTimeSec = timer.getLastActionTimeSec();
	
	if (status !== "success" ) {
		func.log("Страница " + config.lkUrl + " недоступна");
		metrics.pages.login.availability = 0;		
		outMetricsAndExit();
	} else {
		func.log("Страница " + config.lkUrl + " успешно получена");
		metrics.pages.login.availability = 1;
		page.render(config.debugDir + 'login.png');
						
		// Страница будет подгружать iframe'ы, будем их обрабатывать
		var contentN = 0;
		page.onLoadFinished = function(status) {
			// Останавливаем таймер, чтобы замерять время получения страницы личного кабинета
			// Если мы будем получать несколько страниц (iframe'ов),
			// то время таймера просто будет расти, т.к. мы его стартанули только один раз,
			// перед отправкой логина-пароля. Личный кабинет открывается не сразу - 
			// сначала он открывает страницу "Подождите", и только через некоторое время
			// показывает контент. Соотв. парсер корректно замеряет время от отправки логина до
			// получения страницы с реальными данными 
			timer.stopAction();
			
			contentN++;
			func.log('Загружен контент N' + contentN + ':' + status);
			
			page.render(config.debugDir + contentN + '.png');
			fs.write(config.debugDir + contentN + '.html', page.content, 'w');
			
			if ( status === 'success') {
				getMtsMetrics(page, contentN);
			}
		};

		func.log("Заполняем поля формы, логин: " + config.lkLogin);				
		
		timer.startAction();
		page.evaluate(function(config) {
			var form = document.forms[0];
	
			form.phone.value    = config.lkLogin;
			form.password.value = config.lkPass;	
				
			form.elements[2].click();
		}, config);		
	}
});

function getMtsMetrics(page, contentN) {	
	if ( page.content.match('подозрительную активность') ) {
		func.log("Ответ страницы МТС: замечена подозрительная активность. Слишком частое обращение к странице личного кабинета");
		metrics.pages.lk.availability = 0;
		metrics.pages.lk.responseTimeSec = 0;
		outMetricsAndExit();
	}
	
	// Ищем информацию о балансе внутри iFrame'ов
	findBalanceInPage(page, contentN);
	
	// Ищем остаток трафика и кол-во дней до оплаты
	findTrafficInfoInPage(page);
	
	// Если собрали все метрики, заканчиваем скрипт
	if ( checkGotMetricsAlready() ) {
		outMetricsAndExit();
	}
}

/**
 * Ищет информацию о балансе на полученной странице
 */
function findBalanceInPage(page, contentN) {
	// Информация о балансе находится в iframe'ах
	if ( page.framesCount == 0 ) {
		return;
	}
		
	func.log("Анализируем полученные iframe'ы");
	var balanceResult = page.evaluate(function() {
		var result = {
			iframes: [],
			balance: null
		};
		
		$("iframe").each(function(i, iframe) {
			var iframeBody = $(iframe).contents().find('body');
			
			if ( iframeBody.size() > 0 ) {
				result.iframes.push( iframeBody.html() );
				
				// Вариант поиска баланса 1 - через DOM
				if ( result.balance === null ) {
					iframeBody.find(".b-header_balance").each(function() {
						var m = $(this).text().match(/([0-9.]+) руб/i);
						if ( m ) {
							result.balance = m[1];
						}
					});
				}
				
				// Вариант поиска баланса 2 - через regex
				if ( result.balance === null ) {
					var m = iframeBody.text().match(/баланс\s*:\s*-?([0-9.]+)\s*руб/i);
					if ( m ) {
						result.balance = m[1];
					}
				}
			}
		});
		
		return result;
	});
	
	var iframesAnalyzed = balanceResult.iframes.length;
	func.log("Проанализировано iframe'ов:" + iframesAnalyzed);
	
	if ( iframesAnalyzed > 0 ) {
		
		// Сохраняем iFrame'ы на диск
		for (var i = 0; i < iframesAnalyzed; i++) {
			var iframeContent = balanceResult.iframes[i];
			func.log("  Сохраняем iframe " + config.debugDir + contentN + '_iframe' + i + '.html');
			fs.write(config.debugDir + contentN + '_iframe' + i + '.html', iframeContent, 'w');
		}
		
		// Проверяем, была ли найдена информация о балансе
		if ( balanceResult.balance !== null ) {
			if ( metrics.pages.lk.availability === null ) {
				// Если мы получили баланс, то страница личного кабинета корректно загрузилась
				metrics.pages.lk.availability = 1;
				metrics.pages.lk.responseTimeSec = timer.getLastActionTimeSec();
			}		
			func.log("Найдена информация о балансе: " + balanceResult.balance);
			metrics.balance = balanceResult.balance;
		}
	}
}

/**
 * Ищет информацию об интернет-трафике на странице
 */
function findTrafficInfoInPage(page) {
	var traf = page.content.match(TRAFFIC_REGEX);
	if ( traf ) {
		func.log("Остаток трафика - строка найдена: " + traf);
		metrics.trafLeftMb = traf[1];
		var trafUnits  = traf[2];
		if ( trafUnits.toLowerCase() == 'гб' ) {
			metrics.trafLeftMb *= 1024;
		}	
		metrics.daysLeft = traf[3];
	}
	else if (page.content.match("ревышена квота трафика") ) {
		func.log("Найдена строка: превышена квота трафика");
		metrics.trafLeftMb = 0;
			
		if ( page.injectJs("jquery.min.js") ) {
			metrics.daysLeft = page.evaluate(function() {
				var p = $("p:contains('Интернет-пакет будет обновлен')");
				var pText = p.find("b").text();
				console.log( "Найден текст: " + pText);
				return pText.replace(/\D/g, '');
			});
		}
	}
}

/**
 * Проверяет, собраны ли уже все метрики
 */
function checkGotMetricsAlready() {
	if ( metrics.pages.login.availability == 0 || metrics.pages.lk.availability == 0 ) {
	    // Мы определили недоступность одной из страниц (логин или страница ЛК)
	    // В любом из этих случаев метрики собрать не удастся, инициируем окончание скрипта
		return true;
	}
	
	if ( metrics.balance != null && metrics.daysLeft != null && metrics.trafLeftMb != null ) {
		// Все метрики собраны, инициируем окончание скрипта
		return true;
	}

	return false;
}

/**
 * Выводит значения всех метрик на консоль и в файлы, 
 * инициирует завершение скрипта
 */
function outMetricsAndExit() {
	func.log("Метрики:");
	func.writeMetricToFileAndLog('traffic',   'metrics.trafLeftMb', metrics.trafLeftMb);
	func.writeMetricToFileAndLog('money',     'metrics.balance',    metrics.balance);
	func.writeMetricToFileAndLog('daysLeft',  'metrics.daysLeft',   metrics.daysLeft);
	func.writeMetricToFileAndLog('status-initialpageload', 'metrics.pages.login.availability',    metrics.pages.login.availability);
	func.writeMetricToFileAndLog('time-initialpageload',   'metrics.pages.login.responseTimeSec', metrics.pages.login.responseTimeSec);
	func.writeMetricToFileAndLog('status-lkpageload',      'metrics.pages.lk.availability',       metrics.pages.lk.availability);
	func.writeMetricToFileAndLog('time-lkpageload',        'metrics.pages.lk.responseTimeSec',    metrics.pages.lk.responseTimeSec);
	
	phantom.exit();
}