# shroudBNC - an object-oriented framework for IRC
# Copyright (C) 2005 Gunnar Beutner
#
# This program is free software; you can redistribute it and/or
# modify it under the terms of the GNU General Public License
# as published by the Free Software Foundation; either version 2
# of the License, or (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.

# configuration variables: ::account354

foreach user [split $::account354] {
	setctx $user
	bind join - * auth:join
	bind raw - 315 auth:raw315
	bind raw - 354 auth:raw354
	bind evnt - init-server auth:connect
}

internaltimer 180 1 auth:pulse 180
internaltimer 240 1 auth:pulse 240

proc auth:connect {type} {
	namespace eval [getns] {
		if {![info exists inwho]} { set inwho 0 }
		if {![info exists accounttimer]} { set accounttimer 0 }
		if {![info exists accountwait]} { set accountwait 0 }
		if {![info exists authchans]} { set authchans {} }
	}

	upvar [getns]::inwho inwho
	upvar [getns]::accounttimer accounttimer
	upvar [getns]::accountwait accountwait
	upvar [getns]::authchans authchans

	set inwho 0
	set authchans [list]
	set accounttimer 0
	set accountwait 0

	internalkilltimer auth:jointimer [getctx]
}

proc auth:join {nick host hand chan} {
	namespace eval [getns] {
		if {![info exists inwho]} { set inwho 0 }
		if {![info exists whonicks]} { set whonicks {} }
		if {![info exists accounttimer]} { set accounttimer 0 }
		if {![info exists accountwait]} { set accountwait 0 }
	}

	upvar [getns]::inwho inwho
	upvar [getns]::accounttimer accounttimer
	upvar [getns]::accountwait accountwait

	if {[isbotnick $nick]} {
		auth:enqueuewho $chan
	} else {
		set account [getchanlogin $nick]

		if {$account == ""} {
			if {$accounttimer != 0} {
				incr accountwait

				if {$accountwait > 2} {
					return
				}
			}

			internalkilltimer auth:jointimer [getctx]
			internaltimer 3 0 auth:jointimer [getctx]

			set accounttimer 1
		} else {
			bncsettag $chan $nick account $account

			sbnc:callbinds "account" - $chan "$chan $host" $nick $host $hand $chan
		}
	}
}

proc auth:enqueuewho {names} {
	namespace eval [getns] {
		if {![info exists authchans]} { set authchans [list] }
		if {![info exists inwho]} { set inwho 0 }
	}

	upvar [getns]::authchans authchans
	upvar [getns]::inwho inwho

	set current [list]
	set outnames [list]

	foreach cur $authchans {
		foreach item [split $cur ","] {
			lappend current [string tolower $item]
		}
	}

	set allowedchans [split [string tolower [getbncuser [getctx] tag whochans]] ","]

	foreach name [split $names ","] {
		set name [string tolower $name]

		if {[llength $allowedchans] == 0 || ([string first [string index $name 0] [getisupport CHANTYPES]] == -1 || [lsearch -exact $allowedchans $name] != -1)} {
			if {[lsearch -exact $current $name] == -1} {
				lappend outnames $name
			}
		}
	}

	if {[llength $outnames] == 0} { return }

	if {$inwho} {
		lappend authchans [join $outnames ","]
	} else {
		set inwho 1
		puthelp "WHO [join $outnames ","] n%nat,23"
	}
}

proc auth:dequeuewho {} {
	namespace eval [getns] {
		if {![info exists authchans]} { set authchans [list] }
	}

	upvar [getns]::authchans authchans

	set chan [lindex $authchans 0]
	set authchans [lrange $authchans 1 end]

	if {$chan == ""} { return }

	set result [list]

	foreach item [split $chan] {
		set account [getchanlogin $item]

		if {$account != "" && $account != 0} {
			lappend result $item
		}
	}

	return $result
}

proc auth:raw315 {source raw text} {
	namespace eval [getns] {
		if {![info exists inwho]} { set inwho 0 }
	}

	upvar [getns]::inwho inwho

	if {$inwho} {
		haltoutput
	}

	set chan [auth:dequeuewho]

	if {$chan != ""} {
		puthelp "WHO $chan n%nat,23"
		set inwho 1
	} else {
		set inwho 0
	}
}

proc auth:raw354 {source raw text} {
	set pieces [split $text]

	if {[lindex $pieces 1] != 23} { return }

	set nick [lindex $pieces 2]
	set site [getchanhost $nick]
	set host [lindex $pieces 2]!$site
	set hand [finduser $host]

	haltoutput

	foreach chan [internalchannels] {
		if {[onchan $nick $chan]} {
			set old [bncgettag $chan $nick account]
			bncsettag $chan $nick account [lindex $pieces 3]

			if {$old == "" || ($old == 0 && [lindex $pieces 3] != 0)} {
				sbnc:callbinds "account" - $chan "$chan $host" $nick $site $hand $chan
			}
		}
	}
}

# it's possible that the same nick gets /who'd twice (during the same call of auth:jointimer). fix?
proc auth:jointimer {context} {
	setctx $context

	namespace eval [getns] {
		if {![info exists accounttimer]} { set accounttimer 0 }
		if {![info exists accountwait]} { set accountwait 0 }
		if {![info exists inwho]} { set inwho 0 }
	}

	upvar [getns]::accounttimer accounttimer
	upvar [getns]::accountwait accountwait
	upvar [getns]::inwho inwho

	set accounttimer 0
	set accountwait 0

	set nicks ""
	foreach chan [internalchannels] {
		foreach nick [internalchanlist $chan] {
			if {[bncgettag $chan $nick account] == ""} {
				lappend nicks $nick
			}

			set l [join [sbnc:uniq $nicks] ","]

			if {[string length $l] > 350} {
				auth:enqueuewho $l
				set nicks [list]

			}
		}
	}

	if {$nicks != ""} {
		auth:enqueuewho [join [sbnc:uniq $nicks] ","]
	}
}

proc auth:pulse {reason} {
	global account354

	foreach user [split $account354] {
		if {![bncvaliduser $user]} {
			continue
		}

		setctx $user

		set nicks ""
		set allowedchans [split [string tolower [getbncuser [getctx] tag whochans]] ","]

		namespace eval [getns] {
			if {![info exists inwho]} { set inwho 0 }
		}

		upvar [getns]::inwho inwho

		foreach chan [internalchannels] {
			if {$allowedchans != "" && [lsearch -exact $allowedchans [string tolower $chan]] == -1} {
				continue
			}

			set count 0

			foreach nick [internalchanlist $chan] {
				set account [bncgettag $chan $nick account]

				if {$account == 0 || $account == ""} {
					incr count
				}
			}

			if {$count > 100 || ($count > 20 && $count > [llength [internalchanlist $chan]] * 3 / 4)} {
				auth:enqueuewho $chan
				continue
			}

			foreach nick [internalchanlist $chan] {
				set account [bncgettag $chan $nick account]

				if {($reason == 180 && $account == "") || ($account == 0 && $reason == 240)} {
					lappend nicks $nick
				}

				if {[string length [join [sbnc:uniq $nicks] ","]] > 350} {
					auth:enqueuewho [join [sbnc:uniq $nicks] ","]
					set nicks [list]
				}
			}
		}

		if {$nicks != ""} {
			auth:enqueuewho [join [sbnc:uniq $nicks] ","]
		}
	}
}

proc getchanlogin {nick {channel ""}} {
	foreach chan [internalchannels] {
		set acc [bncgettag $chan $nick account]

		if {$acc != ""} {
			return $acc
		}
	}

	return ""
}