feat: Add first few stats

Overridden elements, overridden components, properties, values

bin/antbear.js

@@ -5,8 +5,15 @@ const path = require('path');
 const minimist = require('minimist');
 const glob = require('glob');
 const kleur = require('kleur');
-const { flatMap } = require('lodash');
-const { antbear } = require('../src');
+const longest = require('longest');
+const { flatMap, sortBy } = require('lodash');
+const {
+	getInstances,
+	getElementsStats,
+	getComponentsStats,
+	getPropsStats,
+	getValuesStats,
+} = require('../src');
 
 function getBinaryName() {
 	const binaryPath = process.env._;
@@ -41,13 +48,37 @@ function printInstances(instances) {
 	});
 }
 
+function printObject(object, caption) {
+	const keyColWidth = longest(Object.keys(object)).length;
+	const valueColWidth = longest(Object.values(object)).length;
+	const rows = sortBy(Object.entries(object), ([, value]) => value).reverse();
+
+	console.log(kleur.underline(caption));
+	console.log();
+
+	rows.forEach(([key, value]) => {
+		console.log(
+			kleur.cyan(key.padEnd(keyColWidth)),
+			' ',
+			String(value).padStart(valueColWidth)
+		);
+	});
+
+	console.log();
+}
+
 const argv = minimist(process.argv.slice(2));
 const patterns = argv._;
 const isVerbose = argv.verbose;
 
 if (patterns.length > 0) {
 	const files = flatMap(patterns, (pattern) => glob.sync(pattern));
-	const instances = antbear(files);
+	const instances = getInstances(files);
+
+	printObject(getElementsStats(instances), 'Overridden elements');
+	printObject(getComponentsStats(instances), 'Overridden components');
+	printObject(getPropsStats(instances), 'Properties');
+	printObject(getValuesStats(instances), 'Values');
 
 	if (isVerbose) {
 		printInstances(instances);

package.json

@@ -40,6 +40,7 @@
     "glob": "^7.1.6",
     "kleur": "^4.0.2",
     "lodash": "^4.17.19",
+    "longest": "^2.0.1",
     "minimist": "^1.2.5",
     "stylis": "^4.0.2"
   },

src/index.js

@@ -1,10 +1,24 @@
 const fs = require('fs');
 const { flatMap } = require('lodash');
 const { parseJavaScript } = require('./util/parseJavaScript');
+const {
+	getElementsStats,
+	getComponentsStats,
+	getPropsStats,
+	getValuesStats,
+} = require('../src/util/calculateStats');
 
-module.exports.antbear = function (files) {
+function getInstances(files) {
 	return flatMap(files, (filename) => {
 		const code = fs.readFileSync(filename, 'utf8');
 		return parseJavaScript(code, filename);
 	});
+}
+
+module.exports = {
+	getInstances,
+	getElementsStats,
+	getComponentsStats,
+	getPropsStats,
+	getValuesStats,
 };

src/util/__tests__/calculateStats.spec.js

@@ -0,0 +1,24 @@
+const { getComponentsStats } = require('../calculateStats');
+
+test('getComponentsStats', () => {
+	const result = getComponentsStats([
+		{
+			component: 'Box',
+			styles: [],
+		},
+		{
+			component: 'div',
+			styles: [],
+		},
+		{
+			component: 'Box',
+			styles: [],
+		},
+	]);
+	expect(result).toMatchInlineSnapshot(`
+		Object {
+		  "Box": 2,
+		  "div": 1,
+		}
+	`);
+});

src/util/calculateStats.js

@@ -0,0 +1,33 @@
+const { countBy, flatMap } = require('lodash');
+
+// TODO: Custom colors
+// TODO: Custom spacing
+
+const isComponent = (name) => name.match(/^[A-Z]/);
+
+const getAllStyles = (instances) =>
+	flatMap(instances, (instance) => instance.styles);
+
+module.exports.getElementsStats = function (instances) {
+	const elementsOnly = instances.filter(
+		(instance) => !isComponent(instance.component)
+	);
+	return countBy(elementsOnly, (instance) => instance.component);
+};
+
+module.exports.getComponentsStats = function (instances) {
+	const componentsOnly = instances.filter((instance) =>
+		isComponent(instance.component)
+	);
+	return countBy(componentsOnly, (instance) => instance.component);
+};
+
+module.exports.getPropsStats = function (instances) {
+	const styles = getAllStyles(instances);
+	return countBy(styles, (style) => style.name);
+};
+
+module.exports.getValuesStats = function (instances) {
+	const styles = getAllStyles(instances);
+	return countBy(styles, (style) => style.value);
+};