summaryrefslogtreecommitdiffstats
path: root/stories/react/utils/BeautifyHTML.js
blob: 1a29b0017c6dd421a3af9bd7fcf3a636b4d49845 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
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);
}