#!/bin/sh
set -eu

usage() {
	cat <<-'EOF'
		Usage:
		  html FILENAME
		  html -h
	EOF
}

help() {
	cat <<-'EOF'


		Options:
		  -h, --help    show this message

		  FILENAME      the name of the input file .md file


		Process the FILENAME, and generate a full HTML page.


		Examples:

		  Generate the HTML for a pastebin:

		    $ html src/a-paste.md > src/a-paste.html
	EOF
}


for f in "$@"; do
	case "$f" in
		--)
			break
			;;
		--help)
			usage
			help
			exit
			;;
		*)
			;;
	esac
done

while getopts 'h' flag; do
	case "$flag" in
		h)
			usage
			help
			exit
			;;
		*)
			usage >&2
			exit 2
			;;
	esac
done
shift $((OPTIND - 1))

. src/lib.sh

FILENAME="${1:-}"
eval "$(assert_arg "$FILENAME" 'FILENAME')"


# shellcheck source=/dev/null
. "${FILENAME%.md}.conf"

#
# Utility functions
#

INDENT='        '
markdown_to_html() {
	md2html | awk -vINDENT="$INDENT" '
		BEGIN {
			in_block = 0
		}

		{
			if (in_block == 0) {
				printf "%s", INDENT
			}
			print
		}

		/^<\/code><\/pre>$/ {
			in_block = 0
		}

		/^<pre><code/ {
			in_block = 1
		}
	'
}

extract_plaintext_snippets() {
	SNIPPETS="${FILENAME%.md}.snippets"
	printf '' > "$SNIPPETS"
	F="$(mkstemp)"
	cat > "$F"
	(
		IFS=''
		BLOCK_NUMBER=0
		IN_BLOCK=
		while read -r line; do
			if [ "$line" = '</code></pre>' ]; then
				IN_BLOCK=
			fi

			if [ -n "$IN_BLOCK" ]; then
				printf '%s\n' "$line" | htmlesc -d >> "$OUT"
			fi

			if printf '%s' "$line" | grep -q "^$INDENT<pre><code"; then
				IN_BLOCK=1
				OUT="${FILENAME%.md}.html.$BLOCK_NUMBER.txt"
				BLOCK_NUMBER=$((BLOCK_NUMBER + 1))
				printf '%s\n' "$line" |
					sed "s|^\($INDENT<pre><code.*>\)\(.*\)$|\2|" |
					htmlesc -d > "$OUT"
				printf '%s\n' "$OUT" >> "$SNIPPETS"
			fi
		done < "$F"

		BLOCK_NUMBER=0
		while read -r line; do
			printf '%s\n' "$line"

			if [ "$line" = '</code></pre>' ]; then
				printf '%s<p class="plaintext-link"><a href="%s">plaintext</a></p>\n' \
					"$INDENT" \
					"$(basename "${url_part:?}").$BLOCK_NUMBER.txt"
				BLOCK_NUMBER=$((BLOCK_NUMBER + 1))
			fi
		done < "$F"
	)

}

add_line_numbers() {
	awk '
		/^<\/code><\/pre>$/ {
			in_block = 0
			printf "</tbody></table>%s\n", $0
			next
		}

		match($0, /^( +<pre><code.*>)(.*)$/, a) {
			printf "%s<table rules=columns class=\"code-block\"><tbody>", a[1]

			n = 1
			block_count++
			printf "<tr><td class=\"line-number\"><a id=\"B%s-L%s\" href=\"#B%s-L%s\">%s</a></td><td class=\"code-line\">%s</td></tr>\n", block_count, n, block_count, n, n, a[2]
			in_block = 1
			next
		}

		in_block == 1 {
			n++
			printf "<tr><td class=\"line-number\"><a id=\"B%s-L%s\" href=\"#B%s-L%s\">%s</a></td><td class=\"code-line\">%s</td></tr>\n", block_count, n, block_count, n, n, $0
			next
		}

		{ print }
	'
}

add_headings_anchors() {
	(
		IFS=''
		while read -r line; do
			if ! printf '%s' "$line" | grep -q "^$INDENT<h[2-6]>"; then
				printf '%s\n' "$line"
				continue
			fi
			LVL="$(printf '%s' "$line" | sed "s|^$INDENT<h\(.\)>.*|\1|")"
			HEADING="$(printf '%s' "$line" | sed "s|^$INDENT<h.>\(.*\)</h.>$|\1|")"
			SLUG="$(slugify "$HEADING")"
			printf '%s<h%s class="header-anchor" id="%s">%s<a href="#%s" aria-hidden="true"><img class="svg-icon" src="%s" /></a></h%s>\n' \
				"$INDENT"  \
				"$LVL"     \
				"$SLUG"    \
				"$HEADING" \
				"$SLUG"    \
				"${icon_link_url:?}" \
				"$LVL"
		done
	)
}


warn_duplicate_ids() {
	F="$(mkstemp)"
	tee "$F"
	{
		grep "^$INDENT<h[2-6] class=\"header-anchor\" id=\"" |
			sed "s|^$INDENT<h[2-6] class=\"header-anchor\" id=\"\(.*\)\">.*<a href=.*$|\1|" |
			sort |
			uniq -c |
			awk -v F="$FILENAME" '$1 != 1 {
				printf "WARNING: duplicate header id: %s: %s\n", F, $2
			}'
	} >&2 < "$F"
	rm "$F"
}

emit_body() {
	< "${FILENAME%.md}.content" \
		markdown_to_html           |
		extract_plaintext_snippets |
		add_line_numbers           |
		add_headings_anchors       |
		warn_duplicate_ids
}

envsubst < src/lib/preamble.html
emit_body | tee "${FILENAME%.md}.htmlbody"
envsubst < src/lib/postamble.html