1 files changed, 33 insertions, 0 deletions
diff --git a/stories/react/utils/BeautifyHTML.js b/stories/react/utils/BeautifyHTML.js
new file mode 100644
index 0000000..1a29b00
--- /dev/null
+++ b/stories/react/utils/BeautifyHTML.js
@@ -0,0 +1,33 @@
+export default function beautifyHTML({html, indentChar = '    ', startingIndentCount = 0}) {
+	html = html.replace(/[ ]{2,}/g, ' ');
+
+	let result = '', indentCount = startingIndentCount, parsingText = false;
+	for (let i = 0; i < html.length; i++) {
+
+		let startOfTag, endOfTag, closingTag, upcomingTag, afterTag, numTabs;
+		if (html[i] === '<') { startOfTag = true; }
+		else if (html[i] === '>') { endOfTag = true; }
+		else if (html[i - 1] === '>') { afterTag = true; }
+		if (html[i + 1] === '/') { closingTag = true; }
+		else if (html[i + 1 ] === '<') { upcomingTag = true; }
+
+		if (startOfTag) {
+			if (closingTag) { numTabs = --indentCount; }
+			else { numTabs = indentCount++; }
+		}
+
+		if (parsingText && afterTag) {
+			numTabs = indentCount;
+		}
+
+		result += indentChar.repeat(numTabs) + html[i];
+
+		if (endOfTag || parsingText && upcomingTag) {
+			result += '\n';
+			parsingText = false;
+			if (!upcomingTag) { parsingText = true; }
+		}
+	}
+
+	return result.slice(0, -1);
+}