mstar-adsperformance/lib/RenderResult.js

auto monitoring morningstar web pages ads performance

'use strict';

import fs from 'fs';
import cheerio from 'cheerio';
import handlebars from 'handlebars';
import AdTypes from './AdType'
import path from 'path';

class RenderResult {
	constructor (data, config) {
		this.data = data;
		this.config = config;
	}

	render() {
		let source = fs.readFileSync(path.join(__dirname, '/html/result.html'), 'utf8');
		const $ = cheerio.load(source);

		// this.addMDSStyle($);

		// foreach Url
		Object.keys(this.data).forEach((key) => {
			this.addPageHead($, key);
			this.addPageTables($, this.data[key]);
		});

		this.writeResult($);
		return this;
	}
	addMDSStyle($) {
		$('html >head').append(`<link rel="stylesheet" type="text/css" href="${process.env.INIT_CWD}/dist/css/mds_doc_library.css">`);
	}

	addPageHead($, title) {
		$('.col-md-12').append(`<h1 class="title">${title}</h1>`);
	}

	addTableHead($, title) {
		$('.col-md-12').append(`<h4 class="title">${title}</h4>`);
	}

	addPageTables($, dataObject) {
		let tableSource = fs.readFileSync(path.join(__dirname, '/html/table.html'), 'utf8');
		this.registerHandlebarsHelper();
		let template = handlebars.compile(tableSource);

		let adTypes = AdTypes.getInstance().getAdTypes();
		for(let type of adTypes) {
			let data = dataObject[type];
			let resultTableHtml = template(data);
			this.addTableHead($, type.toUpperCase() + ' Result')
			$('.col-md-12').append(resultTableHtml);			
		}
	}

	registerHandlebarsHelper() {
		handlebars.registerHelper('isNumber', function (value, options) {
			if (typeof value === 'number') {
				return options.fn(this);
			} else if (typeof value === 'string' && (value.match(/^-{0,1}\d+$/) || value.match(/^\d+\.\d+$/))) {
				return options.fn(this);
			}
			return options.inverse(this);
		});
	}
	writeResult($) {
		let outFileName = process.env.HOME + '/Documents/performance.html';
		if (this.config.outfile !== null && this.config.outfile !== '') {
			outFileName = path.resolve(this.config.outfile);
		}
		fs.writeFile(outFileName, $.html(), 'utf8', function (err) {
			if (err) {
				console.log(err);
				throw err
			}
		});
	}

	writeRealData(data) {
		let folderPath = process.env.HOME + '/Documents/nativeData.json';				
		if (this.config.outfile !== null && this.config.outfile !== '') {
			const outFileName = path.resolve(this.config.outfile);
			folderPath = path.dirname(outFileName) + '/nativeData.json';
		}

		fs.writeFile(folderPath, JSON.stringify(data, null, 4), 'utf8', function (err) {
			if (err) {
				console.log(err);
				throw err
			}
		});
	}
}

export default RenderResult;