#!/bin/bash
source config/master.sh
source src/mime.sh
source src/misc.sh
source src/account.sh
source src/mail.sh
[[ -f "${cfg[namespace]}/config.sh" ]] && source "${cfg[namespace]}/config.sh"


declare -A r # current request / response
declare -A meta # metadata for templates
declare -A cookies # cookies!

r[status]=210 # Mommy always said that I was special
post_length=0
post=false
get=false

while read param; do
	if [[ "$param" == $'\015' ]]; then
		break
		
	elif [[ "$param" == *"Content-Length:"* ]]; then
		r[content_length]=$(echo -n $param | sed 's/Content-Length: //;s/\r//')

	elif [[ "$param" == *"Content-Type:"* ]]; then
		r[content_type]="$(echo -n $param | sed 's/Content-Type: //;s/\r//')"
		if [[ "${r[content_type]}" == *"multipart/form-data"* ]]; then
			tmpdir=$(mktemp -d)
		fi
		if [[ "${r[content_type]}" == *"boundary="* ]]; then
			r[content_boundary]="$(echo -n ${r[content_type]} | sed -E 's/(.*)boundary=//;s/\r//;s/ //')"
		fi
		
	elif [[ "$param" == *"Host:"* ]]; then
		r[host]="$(printf "$param" | sed 's/Host: //;s/\r//;s/\\//g')"
		r[host_portless]="$(echo "${r[host]}" | sed -E 's/:(.*)$//')"
		if [[ -f "config/${r[host]}" ]]; then
			source "config/${r[host]}"
		elif [[ -f "config/${r[host_portless]}" ]]; then
			source "config/${r[host_portless]}"
		fi
		
	elif [[ "$param" == *"Upgrade:"* && $(printf "$param" | sed 's/Upgrade: //;s/\r//') == "websocket" ]]; then
		r[status]=101
		
	elif [[ "$param" == *"Sec-WebSocket-Key:"* ]]; then
		r[websocket_key]="$(printf "$param" | sed 's/Sec-WebSocket-Key: //;s/\r//')"
		
	elif [[ "$param" == *"Authorization: Basic"* ]]; then
		login_simple "$param"
		
	elif [[ "$param" == *"Authorization: Bearer"* ]]; then
		r[authorization]="$(printf "$param" | sed 's/Authorization: Bearer //;s/\r//')"
		
	elif [[ "$param" == *"Cookie: "* ]]; then
		IFS=';'
		for i in $(IFS=' '; echo "$param" | sed -E 's/Cookie: //;;s/%/\\x/g'); do
			name="$((grep -Poh ".*?(?==)" | head -1) <<< $i)"
			value="$(sed "s/$name=//" <<< $i)"
			cookies[$name]="$(echo -e $value)"
		done
		
	elif [[ "$param" == *"GET "* ]]; then
		r[url]="$(echo -ne "$(echo -n $param | sed -E 's/GET //;s/HTTP\/[0-9]+\.[0-9]+//;s/ //g;s/\%/\\x/g;s/\/*\r//g;s/\/\/*/\//g')")"
		data="$(echo ${r[url]} | sed -E 's/^(.*)\?//;s/\&/ /g')"
		if [[ "$data" != "${r[url]}" ]]; then
			data="$(echo ${r[url]} | sed -E 's/^(.*)\?//')"
			declare -A get_data
			IFS='&'
			for i in $data; do
				name="$(echo $i | sed -E 's/\=(.*)$//')"
				value="$(echo $i | sed "s/$name\=//")"
				get_data[$name]="$value"
			done
		fi
		
	elif [[ "$param" == *"POST "* ]]; then
		r[url]="$(echo -ne "$(echo -n $param | sed -E 's/POST //;s/HTTP\/[0-9]+\.[0-9]+//;s/ //g;s/\%/\\x/g;s/\/*\r//g;s/\/\/*/\//g')")"
		r[post]=true
		# below shamelessly copied from GET, should be moved to a function
		data="$(echo ${r[url]} | sed -E 's/^(.*)\?//;s/\&/ /g')"
		if [[ "$data" != "${r[url]}" ]]; then
			data="$(echo ${r[url]} | sed -E 's/^(.*)\?//')"
			declare -A post_data
			IFS='&'
			for i in $data; do
				name="$(echo $i | sed -E 's/\=(.*)$//')"
				value="$(echo $i | sed "s/$name\=//")"
				post_data[$name]="$value"
			done
		fi
		
	fi
done

r[uri]="$(realpath "${cfg[namespace]}/${cfg[root]}$(echo ${r[url]} | sed -E 's/\?(.*)$//')")"
[[ -d "${r[uri]}/" ]] && pwd="${r[uri]}" || pwd=$(dirname "${r[uri]}")

if [[ $NCAT_LOCAL_PORT == '' ]]; then
	r[proto]='http'
	r[ip]="NCAT_IS_BORK"
else
	r[proto]='https'
	r[ip]="$NCAT_REMOTE_ADDR:$NCAT_REMOTE_PORT"
fi

echo "$(date) - IP: ${r[ip]}, PROTO: ${r[proto]}, URL: ${r[url]}, GET_data: ${get_data[@]}, POST_data: ${post_data[@]}, POST_multipart: ${post_multipart[@]}" >> "${cfg[namespace]}/${cfg[log]}"

if [[ ${r[status]} != 101 ]]; then
	if [[ -a ${r[uri]} && ! -r ${r[uri]} ]]; then
		r[status]=403
	elif [[ "$(echo -n ${r[uri]})" != "$(realpath "${cfg[namespace]}/${cfg[root]}")"* ]]; then
		r[status]=403
	elif [[ -f ${r[uri]} ]]; then
		r[status]=200
	elif [[ -d ${r[uri]} ]]; then
		for name in ${cfg[index]}; do
			if [[ -f "${r[uri]}/$name" ]]; then
				r[uri]="${r[uri]}/$name"
				r[status]=200
			fi
		done
	else
		r[status]=404
	fi
fi

if [[ ${cfg[auth_required]} == true && ${r[authorized]} != true ]]; then
	echo "Auth failed." >> ${cfg[log_misc]}
	r[status]=401
fi

if [[ ${cfg[proxy]} == true ]]; then
	r[status]=211
fi

if [[ ${r[post]} == true && ${r[status]} == 200 ]]; then

	# This whole ordeal is here to prevent passing binary data as a variable.
	# I could have done it as an array, but this solution works, and it's
	# speedy enough so I don't care.

	if [[ $tmpdir ]]; then
		declare post_multipart
		tmpfile=$(mktemp -p $tmpdir)
		dd iflag=fullblock of=$tmpfile ibs=${r[content_length]} count=1 obs=1M
		
		delimeter_len=$(echo -n "$content_boundary"$'\015' | wc -c)
		boundaries_list=$(echo -ne $(grep $tmpfile -ao -e ${r[content_boundary]} --byte-offset | sed -E 's/:(.*)//g') | sed -E 's/ [0-9]+$//')
		
		for i in $boundaries_list; do
			tmpout=$(mktemp -p $tmpdir)
			dd iflag=fullblock if=$tmpfile ibs=$(($i+$delimeter_len)) obs=1M skip=1 | while true; do
				read line
				if [[ $line == $'\015' ]]; then
					cat - > $tmpout
					break
				fi
			done
			length=$(grep $tmpout --byte-offset -ae ${r[content_boundary]} | sed -E 's/:(.*)//' | head -n 1)
			outfile=$(mktemp -p $tmpdir)
			post_multipart+=($outfile)
			dd iflag=fullblock if=$tmpout ibs=$length count=1 obs=1M of=$outfile
			rm $tmpout
		done
		rm $tmpfile
	else
		read -N ${r[content_length]} data
		declare -A post_data
		
		for i in $(echo "$data" | sed -s 's/\&/ /g;'); do
			name="$(echo $i | sed -E 's/\=(.*)$//')"
			param="$(echo $i | sed "s/$name\=//")"
			post_data[$name]="$param"
		done
	fi
fi

if [[ ${r[status]} == 210 && ${cfg[autoindex]} == true ]]; then
	source "src/response/listing.sh"
elif [[ ${r[status]} == 211 ]]; then
	source "src/response/proxy.sh"
elif [[ ${r[status]} == 200 ]]; then
	source "src/response/200.sh"
elif [[ ${r[status]} == 401 ]]; then
	source "src/response/401.sh"
elif [[ ${r[status]} == 404 ]]; then
	source "src/response/404.sh"
elif [[ ${r[status]} == 101 ]]; then
	source "src/response/101.sh"
else
	source "src/response/403.sh"
fi