127 lines
		
	
	
		
			3.7 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
	
	
		
		
			
		
	
	
			127 lines
		
	
	
		
			3.7 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
	
	
|  | /* | ||
|  |  Copyright 2012-2015, Yahoo Inc. | ||
|  |  Copyrights licensed under the New BSD License. See the accompanying LICENSE file for terms. | ||
|  |  */ | ||
|  | const fs = require('fs'); | ||
|  | const FileWriter = require('./file-writer'); | ||
|  | const XMLWriter = require('./xml-writer'); | ||
|  | const tree = require('./tree'); | ||
|  | const watermarks = require('./watermarks'); | ||
|  | 
 | ||
|  | function defaultSourceLookup(path) { | ||
|  |     try { | ||
|  |         return fs.readFileSync(path, 'utf8'); | ||
|  |     } catch (ex) { | ||
|  |         throw new Error( | ||
|  |             'Unable to lookup source: ' + path + '(' + ex.message + ')' | ||
|  |         ); | ||
|  |     } | ||
|  | } | ||
|  | 
 | ||
|  | function mergeWatermarks(specified, defaults) { | ||
|  |     specified = specified || {}; | ||
|  |     Object.keys(defaults).forEach(k => { | ||
|  |         const specValue = specified[k]; | ||
|  |         if ( | ||
|  |             !(specValue && Array.isArray(specValue) && specValue.length === 2) | ||
|  |         ) { | ||
|  |             specified[k] = defaults[k]; | ||
|  |         } | ||
|  |     }); | ||
|  |     return specified; | ||
|  | } | ||
|  | /** | ||
|  |  * A reporting context that is passed to report implementations | ||
|  |  * @param {Object} [opts=null] opts options | ||
|  |  * @param {String} [opts.dir='coverage'] opts.dir the reporting directory | ||
|  |  * @param {Object} [opts.watermarks=null] opts.watermarks watermarks for | ||
|  |  *  statements, lines, branches and functions | ||
|  |  * @param {Function} [opts.sourceFinder=fsLookup] opts.sourceFinder a | ||
|  |  *  function that returns source code given a file path. Defaults to | ||
|  |  *  filesystem lookups based on path. | ||
|  |  * @constructor | ||
|  |  */ | ||
|  | function Context(opts) { | ||
|  |     opts = opts || {}; | ||
|  |     this.dir = opts.dir || 'coverage'; | ||
|  |     this.watermarks = mergeWatermarks(opts.watermarks, watermarks.getDefault()); | ||
|  |     this.sourceFinder = opts.sourceFinder || defaultSourceLookup; | ||
|  |     this.data = {}; | ||
|  | } | ||
|  | 
 | ||
|  | Object.defineProperty(Context.prototype, 'writer', { | ||
|  |     enumerable: true, | ||
|  |     get() { | ||
|  |         if (!this.data.writer) { | ||
|  |             this.data.writer = new FileWriter(this.dir); | ||
|  |         } | ||
|  |         return this.data.writer; | ||
|  |     } | ||
|  | }); | ||
|  | 
 | ||
|  | /** | ||
|  |  * returns a FileWriter implementation for reporting use. Also available | ||
|  |  * as the `writer` property on the context. | ||
|  |  * @returns {Writer} | ||
|  |  */ | ||
|  | Context.prototype.getWriter = function() { | ||
|  |     return this.writer; | ||
|  | }; | ||
|  | 
 | ||
|  | /** | ||
|  |  * returns the source code for the specified file path or throws if | ||
|  |  * the source could not be found. | ||
|  |  * @param {String} filePath the file path as found in a file coverage object | ||
|  |  * @returns {String} the source code | ||
|  |  */ | ||
|  | Context.prototype.getSource = function(filePath) { | ||
|  |     return this.sourceFinder(filePath); | ||
|  | }; | ||
|  | 
 | ||
|  | /** | ||
|  |  * returns the coverage class given a coverage | ||
|  |  * types and a percentage value. | ||
|  |  * @param {String} type - the coverage type, one of `statements`, `functions`, | ||
|  |  *  `branches`, or `lines` | ||
|  |  * @param {Number} value - the percentage value | ||
|  |  * @returns {String} one of `high`, `medium` or `low` | ||
|  |  */ | ||
|  | Context.prototype.classForPercent = function(type, value) { | ||
|  |     const watermarks = this.watermarks[type]; | ||
|  |     if (!watermarks) { | ||
|  |         return 'unknown'; | ||
|  |     } | ||
|  |     if (value < watermarks[0]) { | ||
|  |         return 'low'; | ||
|  |     } | ||
|  |     if (value >= watermarks[1]) { | ||
|  |         return 'high'; | ||
|  |     } | ||
|  |     return 'medium'; | ||
|  | }; | ||
|  | /** | ||
|  |  * returns an XML writer for the supplied content writer | ||
|  |  * @param {ContentWriter} contentWriter the content writer to which the returned XML writer | ||
|  |  *  writes data | ||
|  |  * @returns {XMLWriter} | ||
|  |  */ | ||
|  | Context.prototype.getXMLWriter = function(contentWriter) { | ||
|  |     return new XMLWriter(contentWriter); | ||
|  | }; | ||
|  | /** | ||
|  |  * returns a full visitor given a partial one. | ||
|  |  * @param {Object} partialVisitor a partial visitor only having the functions of | ||
|  |  *  interest to the caller. These functions are called with a scope that is the | ||
|  |  *  supplied object. | ||
|  |  * @returns {Visitor} | ||
|  |  */ | ||
|  | Context.prototype.getVisitor = function(partialVisitor) { | ||
|  |     return new tree.Visitor(partialVisitor); | ||
|  | }; | ||
|  | 
 | ||
|  | module.exports = { | ||
|  |     create(opts) { | ||
|  |         return new Context(opts); | ||
|  |     } | ||
|  | }; |