# Connect with:
#   socat 'TCP:localhost:12345' 'EXEC:awk -f $0'
#
# Pre-defined variables:
#
# When connected:
#   SERVER:  IRC Server that is connected to
#   NICK:    Nickname given by the bot
#   CHANNEL: Channel the bot is currently in
#
# When a messages is recieved:
#   CMD:     Message, e.g. PRIVMSG
#   SRC:     Source of the message
#   DST:     Destination of the message, e.g. the channel
#   TO:      Nickname the message was addressed to
#   FROM:    Nickname of the user who sent the message
#   MSG:     Message sent
#   $0:      Message sans TO

# Debugging
function send(msg) {
	print "  > " msg > "/dev/stderr"
	print msg
	fflush()
}

// {
	#print ""         > "/dev/stderr"
	print "  < " $0  > "/dev/stderr"
}

function debug(msg) {
	print "  # " msg > "/dev/stderr"
	fflush()
}

function set(i) {
	debug("CMD:  [" CMD  "]")
	debug("SRC:  [" SRC  "]")
	debug("DST:  [" DST  "]")
	debug("FROM: [" FROM "]")
	debug("TO:   [" TO   "]")
	debug("MSG:  [" MSG  "]")
	debug("$0:   [" $0   "]")
	for (i in ARG)
		debug("ARG"i": [" ARG[i] "]")
}

# Functions
function connect(server, nick, channel, auth, pass) {
	SERVER  = server
	NICK    = nick
	CHANNEL = channel
	if (FIRST) {
		"whoami"   | getline _name
		"hostname" | getline _host
		send("USER " _name " " _host " " server " :" nick)
		send("NICK " nick)
		send("CAP REQ :account-notify")
		send("CAP REQ :extended-join")
		send("CAP END")
		say("NickServ", "IDENTIFY " pass)
	} else {
		send("WHOIS " nick)
	}
}

function say(to, msg) {
	if (msg == "") {
		msg = to
		if (DST ~ "^#")
			to = DST
		else if (DST == NICK && FROM)
			to = FROM
		else
			to = CHANNEL
	}
	send("PRIVMSG " to " :" msg)
	if (!DEBUG && NR > 1)
		system("sleep 1")
}

function action(to, msg)
{
	if (msg)
		say(to, "\001ACTION " msg "\001")
	else
		say("\001ACTION " to "\001")
}

function reply(msg) {
	say(FROM ": " msg)
}

function join(chan) {
	send("JOIN " chan)
	send("TOPIC " chan)
	send("WHO " chan " %uhnar")
}

function part(chan) {
	send("PART " chan)
}

function topic(chan, msg) {
	send("TOPIC " chan " :" msg)
}

# Reloading
BEGIN {
	if (CHILD == "") {
		debug("Starting server");
		cmd = "awk -f rhawk" \
		      " -v CHILD=1" \
		      " -v START=" systime() \
		      " -v DEBUG=" !!DEBUG
		status = system(cmd " -v FIRST=1");
		while (status)
			status = system(cmd);
		exit(0);
	} else {
		debug("Starting child:" \
		      " DEBUG=" DEBUG   \
		      " CHILD=" CHILD   \
		      " START=" START   \
		      " FIRST=" FIRST);
	}
}

function quit() {
	exit(0)
}

function reload() {
	exit(1)
}

# Input parsing
// {
	gsub(/\s+/, " ")
	gsub(/^ | $/, "")
	gsub(/[\2\11\17\23\25\26\37]/, "")
	gsub(/[\3\13](1[0-5]|0?[0-9])?(,(1[0-5]|0?[0-9]))?/, "")
	match($0, /(:([^ ]+) )?([A-Z0-9]+)(( [^:][^ ]*)*)( :(.*))?/, arr);
	sub(/^ /, "", arr[4])
	SRC = arr[2]
	CMD = arr[3]
	MSG = arr[7]

	split(arr[4], ARG)
	DST = ARG[1]

	match(SRC, /([^! ]+)!([^@ ]+)@([^ ]+\/[^ ]+)?/, arr);
	FROM = arr[1]
	USER = arr[2]
	HOST = arr[3]

	match(MSG, /(([^ :,]*)[:,] *)?(.*)/, arr);
	TO  = arr[2]
	$0  = TO ? arr[3] : MSG

	if (CMD == "PRIVMSG" && DST == NICK && FROM && !TO)
		TO = DST

	if (FROM in USERS)
		AUTH = USERS[FROM]["auth"]
	else
		AUTH = ""

	#set()
}

# IRC client
CMD == "001" && MSG ~ /Welcome/ {
	join(CHANNEL)
}

CMD == "PING" {
	send("PING " MSG)
}

CMD == "332" {
	TOPICS[ARG[2]] = MSG
}

CMD == "TOPIC" {
	TOPICS[ARG[1]] = MSG
}

# Authentication
# todo - netsplits
CMD == "319" {
	gsub(/[@+]/, "")
	for (i=1; i<=NF; i++)
		send("WHO " $i " %uhnar")
	for (i=1; i<=NF; i++)
		send("TOPIC " $i)
}

CMD == "ACCOUNT" {
	_auth = ARG[1] == "*" ? 0 : ARG[1]
	USERS[FROM]["auth"] = _auth
}

CMD == "354" {
	_auth = ARG[5] == "*" ? 0 : ARG[5]
	USERS[ARG[4]]["user"] = ARG[2]
	USERS[ARG[4]]["host"] = ARG[3]
	USERS[ARG[4]]["nick"] = ARG[4]
	USERS[ARG[4]]["auth"] = _auth
	USERS[ARG[4]]["real"] = MSG
}

CMD == "JOIN" {
	_auth = ARG[2] == "*" ? 0 : ARG[2]
	USERS[FROM]["user"] = USER
	USERS[FROM]["host"] = HOST
	USERS[FROM]["nick"] = FROM
	USERS[FROM]["auth"] = _auth
	USERS[FROM]["real"] = MSG
}

CMD == "NICK" {
	USERS[MSG]["user"] = USERS[FROM]["user"]
	USERS[MSG]["host"] = USERS[FROM]["host"]
	USERS[MSG]["nick"] = MSG
	USERS[MSG]["auth"] = USERS[FROM]["auth"]
	USERS[MSG]["real"] = USERS[FROM]["real"]
	delete USERS[FROM]
}