X-Git-Url: https://git.distorted.org.uk/~mdw/ircbot/blobdiff_plain/868921287e5f8bfbab6ac285e987e845c660be0d..7818b6afb539ce627b9887e73e43ee684928fd53:/bot.tcl?ds=sidebyside

diff --git a/bot.tcl b/bot.tcl
index dff9ab3..66ab9c2 100755
--- a/bot.tcl
+++ b/bot.tcl
@@ -6,6 +6,19 @@ if {![info exists nick]} { set nick Blight }
 if {![info exists ownfullname]} { set ownfullname "here to Help" }
 set ownmailaddr blight@chiark.greenend.org.uk
 
+set out_maxburst 6
+set out_interval 2100
+set out_lag_lag 5000
+set out_lag_very 15000
+
+if {![info exists out_queue]} {
+    set out_creditms [expr {$out_maxburst*$out_interval}]
+    set out_creditat [clock seconds]
+    set out_queue {}
+    set out_lag_reported 0
+    set out_lag_reportwhen $out_creditat
+}
+
 if {![info exists globalsecret]} {
     set gsfile [open /dev/urandom r]
     fconfigure $gsfile -translation binary
@@ -15,6 +28,13 @@ if {![info exists globalsecret]} {
     unset gsfile
 }
 
+proc manyset {list args} {
+    foreach val $list var $args {
+	upvar 1 $var my
+	set my $val
+    }
+}
+
 proc try_except_finally {try except finally} {
     global errorInfo errorCode
     set er [catch { uplevel 1 $try } emsg]
@@ -38,8 +58,69 @@ proc try_except_finally {try except finally} {
     }
 }
 
-proc sendout {command args} {
+proc out__vars {} {
+    uplevel 1 {
+	global out_queue out_creditms out_creditat out_interval out_maxburst
+	global out_lag_lag out_lag_very
+#set pr [lindex [info level 0] 0]
+#puts $pr>[clock seconds]|$out_creditat|$out_creditms|[llength $out_queue]<
+    }
+}
+
+proc out_lagged {} {
+    out__vars
+    if {[llength $out_queue]*$out_interval > $out_lag_very} {
+	return 2
+    } elseif {[llength $out_queue]*$out_interval > $out_lag_lag} {
+	return 1
+    } else {
+	return 0
+    }
+}
+
+proc out_restart {} {
+    out__vars
+    
+    set now [clock seconds]
+    incr out_creditms [expr {($now - $out_creditat) * 1000}]
+    set out_creditat $now
+    if {$out_creditms > $out_maxburst*$out_interval} {
+	set out_creditms [expr {$out_maxburst*$out_interval}]
+    }
+    out_runqueue $now
+}
+
+proc out_runqueue {now} {
     global sock
+    out__vars
+    
+    while {[llength $out_queue] && $out_creditms >= $out_interval} {
+#puts rq>$now|$out_creditat|$out_creditms|[llength $out_queue]<
+	manyset [lindex $out_queue 0] orgwhen msg
+	set out_queue [lrange $out_queue 1 end]
+	if {[llength $out_queue]} {
+	    append orgwhen "+[expr {$now - $orgwhen}]"
+	    append orgwhen ([llength $out_queue])"
+	}
+	puts "$orgwhen -> $msg"
+	puts $sock $msg
+	incr out_creditms -$out_interval
+    }
+    if {[llength $out_queue]} {
+	after $out_interval out_nextmessage
+    }
+}
+
+proc out_nextmessage {} {
+    out__vars
+    set now [clock seconds]
+    incr out_creditms $out_interval
+    set out_creditat $now
+    out_runqueue $now
+}
+
+proc sendout_priority {priority command args} {
+    global sock out_queue
     if {[llength $args]} {
 	set la [lindex $args end]
 	set args [lreplace $args end end]
@@ -52,10 +133,20 @@ proc sendout {command args} {
     }
     set args [lreplace $args 0 -1 $command]
     set string [join $args { }]
-    puts "[clock seconds] -> $string"
-    puts $sock $string
+    set now [clock seconds]
+    set newe [list $now $string]
+    if {$priority} {
+	set out_queue [concat [list $newe] $out_queue]
+    } else {
+	lappend out_queue $newe
+    }
+    if {[llength $out_queue] == 1} {
+	out_restart
+    }
 }
 
+proc sendout {command args} { eval sendout_priority [list 0 $command] $args }
+    
 proc log {data} {
     puts $data
 }
@@ -80,19 +171,26 @@ proc bgerror {msg} {
 }
 
 proc onread {args} {
-    global sock nick calling_nick
+    global sock nick calling_nick errorInfo errorCode
     
     if {[gets $sock line] == -1} { set terminate 1; return }
     regsub -all "\[^ -\176\240-\376\]" $line ? line
     set org $line
+    
+    set ei $errorInfo
+    set ec $errorCode
+    catch { unset calling_nick }
+    set errorInfo $ei
+    set errorCode $ec
+    
     if {[regexp -nocase {^:([^ ]+) (.*)} $line dummy prefix remain]} {
 	set line $remain
-	if {[regexp {^([^!]+)!} $prefix dummy maybenick] &&
-	    "[irctolower $maybenick]" == "[irctolower $nick]"} return
-	set calling_nick $maybenick
+	if {[regexp {^([^!]+)!} $prefix dummy maybenick]} {
+	    set calling_nick $maybenick
+	    if {"[irctolower $maybenick]" == "[irctolower $nick]"} return
+	}
     } else {
 	set prefix {}
-	catch { unset calling_nick }
     }
     if {![string length $line]} { return }
     if {![regexp -nocase {^([0-9a-z]+) *(.*)} $line dummy command line]} {
@@ -131,9 +229,13 @@ proc onread {args} {
 }
 
 proc sendprivmsg {dest l} {
-    sendout [expr {[ischan $dest] ? "PRIVMSG" : "NOTICE"}] $dest $l
+    foreach v [split $l "\n"] {
+	sendout [expr {[ischan $dest] ? "PRIVMSG" : "NOTICE"}] $dest $v
+    }
+}
+proc sendaction_priority {priority dest what} {
+    sendout_priority $priority PRIVMSG $dest "\001ACTION $what\001"
 }
-proc sendaction {dest what} { sendout PRIVMSG $dest "\001ACTION $what\001" }
 proc msendprivmsg {dest ll} { foreach l $ll { sendprivmsg $dest $l } }
 proc msendprivmsg_delayed {delay dest ll} { after $delay [list msendprivmsg $dest $ll] }
 
@@ -333,12 +435,26 @@ proc chanmode_arg {} {
 }
 
 proc chanmode_o1 {m g p chan} {
-    global nick
+    global nick chan_initialop
     prefix_nick
     set who [chanmode_arg]
     recordlastseen_n $n "being nice to $who" 1
     if {"[irctolower $who]" == "[irctolower $nick]"} {
-	sendprivmsg $n Thanks.
+	set nl [irctolower $n]
+	upvar #0 nick_unique($n) u
+	if {[chandb_exists $chan]} {
+	    sendprivmsg $n Thanks.
+	} elseif {![info exists u]} {
+	    sendprivmsg $n {Op me while not on the channel, why don't you ?}
+	} else {
+	    set chan_initialop([irctolower $chan]) $u
+	    sendprivmsg $n \
+ "Thanks. You can use `channel manager ...' to register this channel."
+	    if {![nickdb_exists $n] || ![string length [nickdb_get $n username]]} {
+		sendprivmsg $n \
+ "(But to do that you must register your nick securely first.)"
+	    }
+	}
     }
 }
 
@@ -371,14 +487,26 @@ proc msg_MODE {p c dest modelist args} {
     }
 }
 
+proc channel_noone_seen {chan} {
+    global nick_onchans
+    foreach n [array names nick_onchans] {
+	upvar #0 nick_onchans($n) oc
+	set oc [grep tc {"$tc" != "$chan"} $oc]
+    }
+}
+
 proc process_kickpart {chan user} {
+    global nick
     check_nick $user
     if {![ischan $chan]} { error "not a channel" }
-
+    if {"[irctolower $user]" == "[irctolower $nick]"} {
+	channel_noone_seen $chan
+    }
     upvar #0 nick_onchans($user) oc
     set lc [irctolower $chan]
     set oc [grep tc {"$tc" != "$lc"} $oc]
     if {![llength $oc]} { nick_forget $user }
+    nick_case $user
 }    
 
 proc msg_KICK {p c chans users comment} {
@@ -395,7 +523,8 @@ proc msg_KILL {p c user why} {
     nick_forget $user
 }
 
-set nick_arys {onchans username}
+set nick_counter 0
+set nick_arys {onchans username unique}
 
 proc nick_forget {n} {
     global nick_arys
@@ -403,10 +532,16 @@ proc nick_forget {n} {
 	upvar #0 nick_${ary}($n) av
 	catch { unset av }
     }
+    nick_case $n
+}
+
+proc nick_case {n} {
+    global nick_case
+    set nick_case([irctolower $n]) $n
 }
 
 proc msg_NICK {p c newnick} {
-    global nick_arys
+    global nick_arys nick_case
     prefix_nick
     recordlastseen_n $n "changing nicks to $newnick" 0
     recordlastseen_n $newnick "changing nicks from $n" 1
@@ -416,6 +551,14 @@ proc msg_NICK {p c newnick} {
 	if {[info exists new]} { error "nick collision ?! $ary $n $newnick" }
 	if {[info exists old]} { set new $old; unset old }
     }
+    nick_case $newnick
+}
+
+proc nick_ishere {n} {
+    global nick_counter
+    upvar #0 nick_unique($n) u
+    if {![info exists u]} { set u [incr nick_counter].$n.[clock seconds] }
+    nick_case $n
 }
 
 proc msg_JOIN {p c chan} {
@@ -423,6 +566,7 @@ proc msg_JOIN {p c chan} {
     recordlastseen_n $n "joining $chan" 1
     upvar #0 nick_onchans($n) oc
     lappend oc [irctolower $chan]
+    nick_ishere $n
 }
 proc msg_PART {p c chan} {
     prefix_nick
@@ -444,6 +588,7 @@ proc msg_PRIVMSG {p c dest text} {
 	recordlastseen_n $n "talking to me" 1
 	set output $n
     }
+    nick_case $n
 
     if {[catch {
 	regsub {^! *} $text {} text
@@ -459,7 +604,7 @@ proc msg_PRIVMSG {p c dest text} {
 	manyset $rv priv_msgs pub_msgs priv_acts pub_acts
 	foreach {td val} [list $n $priv_acts $output $pub_acts] {
 	    foreach l [split $val "\n"] {
-		sendaction $td $l
+		sendaction_priority 0 $td $l
 	    }
 	}
 	foreach {td val} [list $n $priv_msgs $output $pub_msgs] {
@@ -488,16 +633,14 @@ proc msg_353 {p c dest type chan nicklist} {
     if {![info exists names_chans]} { set names_chans {} }
     set chan [irctolower $chan]
     lappend names_chans $chan
-    foreach n [array names nick_onchans] {
-	upvar #0 nick_onchans($n) oc
-	set oc [grep tc {"$tc" != "$chan"} $oc]
-    }
+    channel_noone_seen $chan
     foreach n [split $nicklist { }] {
 	regsub {^[@+]} $n {} n
-	check_nick $n
 	if {![string length $n]} continue
+	check_nick $n
 	upvar #0 nick_onchans($n) oc
 	lappend oc $chan
+	nick_ishere $n
     }
 }
 
@@ -582,6 +725,18 @@ proc loadhelp {} {
 }
 
 def_ucmd help {
+    if {[set lag [out_lagged]]} {
+	if {[ischan $dest]} { set replyto $dest } else { set replyto $n }
+	if {$lag > 1} {
+	    sendaction_priority 1 $replyto \
+		"is very lagged.  Please ask for help again later."
+	    ucmdr {} {}
+	} else {
+	    sendaction_priority 1 $replyto \
+		"is lagged.  Your help will arrive shortly ..."
+	}
+    }
+    
     upvar #0 help_topics([irctolower [string trim $text]]) info
     if {![info exists info]} { ucmdr "No help on $text, sorry." {} }
     ucmdr $info {}
@@ -592,13 +747,6 @@ def_ucmd ? {
     ucmdr $help_topics() {}
 }
 
-proc manyset {list args} {
-    foreach val $list var $args {
-	upvar 1 $var my
-	set my $val
-    }
-}
-
 proc check_username {target} {
     if {
 	[string length $target] > 8 ||
@@ -607,66 +755,64 @@ proc check_username {target} {
     } { error "invalid username" }
 }
 
-proc nickdb__head {} {
+proc somedb__head {} {
     uplevel 1 {
-	set nl [irctolower $n]
-	upvar #0 nickdb($nl) ndbe
-	binary scan $nl H* nh
-	set nfn users/n$nh
-	if {![info exists ndbe] && [file exists $nfn]} {
-	    set f [open $nfn r]
+	set idl [irctolower $id]
+	upvar #0 ${nickchan}db($idl) ndbe
+	binary scan $idl H* idh
+	set idfn $fprefix$idh
+	if {![info exists iddbe] && [file exists $idfn]} {
+	    set f [open $idfn r]
 	    try_except_finally { set newval [read $f] } {} { close $f }
 	    if {[llength $newval] % 2} { error "invalid length" }
-	    set ndbe $newval
+	    set iddbe $newval
 	}
     }
 }
 
-proc def_nickdb {name arglist body} {
-    proc nickdb_$name $arglist "nickdb__head; $body"
+proc def_somedb {name arglist body} {
+    foreach {nickchan fprefix} {nick users/n chan chans/c} {
+	proc ${nickchan}db_$name $arglist \
+		"set nickchan $nickchan; set fprefix $fprefix; somedb__head; $body"
+    }
 }
 
-def_nickdb exists {n} {
-    return [info exists ndbe]
+def_somedb exists {id} {
+    return [info exists iddbe]
 }
 
-def_nickdb delete {n} {
-    catch { unset ndbe }
-    file delete $nfn
+def_somedb delete {id} {
+    catch { unset iddbe }
+    file delete $idfn
 }
 
-set default_settings {timeformat ks}
+set default_settings_nick {timeformat ks}
+set default_settings_chan {autojoin 1}
 
-def_nickdb set {n args} {
-    global default_settings
-    if {![info exists ndbe]} { set ndbe $default_settings }
-    foreach {key value} [concat $ndbe $args] { set a($key) $value }
+def_somedb set {id args} {
+    upvar #0 default_settings_$nickchan def
+    if {![info exists iddbe]} { set iddbe $def }
+    foreach {key value} [concat $iddbe $args] { set a($key) $value }
     set newval {}
     foreach {key value} [array get a] { lappend newval $key $value }
-    set f [open $nfn.new w]
+    set f [open $idfn.new w]
     try_except_finally {
 	puts $f $newval
 	close $f
-	file rename -force $nfn.new $nfn
+	file rename -force $idfn.new $idfn
     } {
     } {
 	catch { close $f }
     }
-    set ndbe $newval
+    set iddbe $newval
 }
 
-proc opt {key} {
-    global calling_nick
-    if {[info exists calling_nick]} { set n $calling_nick } { set n {} }
-    return [nickdb_opt $n $key]
-}
-
-def_nickdb opt {n key} {
-    global default_settings
-    if {[info exists ndbe]} {
-	set l $ndbe
+def_somedb get {id key} {
+    upvar #0 default_settings_$nickchan def
+    if {[info exists iddbe]} {
+	set l $iddbe
     } else {
-	set l $default_settings
+	set l $def
     }
     foreach {tkey value} $l {
 	if {"$tkey" == "$key"} { return $value }
@@ -674,6 +820,12 @@ def_nickdb opt {n key} {
     error "unset setting $key"
 }
 
+proc opt {key} {
+    global calling_nick
+    if {[info exists calling_nick]} { set n $calling_nick } { set n {} }
+    return [nickdb_get $n $key]
+}
+
 proc check_notonchan {} {
     upvar 1 dest dest
     if {[ischan $dest]} { error "that command must be sent privately" }
@@ -682,7 +834,7 @@ proc check_notonchan {} {
 proc nick_securitycheck {strict} {
     upvar 1 n n
     if {![nickdb_exists $n]} { error "you are unknown to me, use `register'." }
-    set wantu [nickdb_opt $n username]
+    set wantu [nickdb_get $n username]
     if {![string length $wantu]} {
 	if {$strict} {
 	    error "that feature is only available to secure users, sorry."
@@ -699,6 +851,163 @@ proc nick_securitycheck {strict} {
     }
 }
 
+proc channel_securitycheck {channel n} {
+    # You must also call `nick_securitycheck 1'
+    set mgrs [chandb_get $channel managers]
+    if {[lsearch -exact [irctolower $mgrs] [irctolower $n]] < 0} {
+	error "you are not a manager of $channel"
+    }
+}
+
+proc def_chancmd {name body} {
+    proc channel/$name {} \
+	    "    upvar 1 target chan; upvar 1 n n; upvar 1 text text; $body"
+}
+
+def_chancmd manager {
+    set opcode [ta_word]
+    switch -exact _$opcode {
+	_= { set ml {} }
+	_+ - _- {
+	    if {[chandb_exists $chan]} {
+		set ml [chandb_get $chan managers]
+	    } else {
+		set ml [list [irctolower $n]]
+	    }
+	}
+	default {
+	    error "`channel manager' opcode must be one of + - ="
+	}
+    }
+    foreach nn [split $text " "] {
+	if {![string length $nn]} continue
+	check_nick $nn
+	set nn [irctolower $nn]
+	if {"$opcode" != "-"} {
+	    lappend ml $nn
+	} else {
+	    set ml [grep nq {"$nq" != "$nn"} $ml]
+	}
+    }
+    if {[llength $ml]} {
+	chandb_set $chan managers $ml
+	ucmdr "Managers of $chan: $ml" {}
+    } else {
+	chandb_delete $chan
+	ucmdr {} {} "forgets about managing $chan." {}
+    }
+}
+
+def_chancmd autojoin {
+    set yesno [ta_word]
+    switch -exact [string tolower $yesno] {
+	no { set nv 0 }
+	yes { set nv 1 }
+	default { error "channel autojoin must be `yes' or `no' }
+    }
+    chandb_set $chan autojoin $nv
+}
+
+def_chancmd show {
+    if {[chandb_exists $chan]} {
+	set l "Settings for $chan: autojoin "
+	append l [lindex {no yes} [chandb_get $chan autojoin]]
+	append l "\nManagers: "
+	append l [join [chandb_get $chan managers] " "]
+	ucmdr {} $l
+    } else {
+	ucmdr {} "The channel $chan is not managed."
+    }
+}
+
+def_ucmd op {
+    if {[ischan $dest]} { set target $dest }
+    if {[ta_anymore]} { set target [ta_word] }
+    ta_nomore
+    if {![info exists target]} { error "you must specify, or !... on, the channel" }
+    if {![ischan $target]} { error "not a valid channel" }
+    if {![chandb_exists $target]} { error "$target is not a managed channel." }
+    prefix_nick
+    nick_securitycheck 1
+    channel_securitycheck $target $n
+    sendout MODE $target +o $n
+}
+
+def_ucmd channel {
+    if {[ischan $dest]} { set target $dest }
+    if {![ta_anymore]} {
+	set subcmd show
+    } else {
+	set subcmd [ta_word]
+    }
+    if {[ischan $subcmd]} {
+	set target $subcmd
+	if {![ta_anymore]} {
+	    set subcmd show
+	} else {
+	    set subcmd [ta_word]
+	}
+    }
+    if {![info exists target]} { error "privately, you must specify a channel" }
+    set procname channel/$subcmd
+    if {"$subcmd" != "show"} {
+	if {[catch { info body $procname }]} { error "unknown channel setting $subcmd" }
+	prefix_nick
+	nick_securitycheck 1
+	if {[chandb_exists $target]} {
+	    channel_securitycheck $target $n
+	} else {
+	    upvar #0 chan_initialop([irctolower $target]) io
+	    upvar #0 nick_unique($n) u
+	    if {![info exists io]} { error "$target is not a managed channel" }
+	    if {"$io" != "$u"} { error "you are not the interim manager of $target" }
+	    if {"$subcmd" != "manager"} { error "use `channel manager' first" }
+	}
+    }
+    channel/$subcmd
+}
+
+def_ucmd who {
+    if {[ta_anymore]} {
+	set target [ta_word]; ta_nomore
+	set myself 1
+    } else {
+	prefix_nick
+	set target $n
+	set myself [expr {"$target" != "$n"}]
+    }
+    upvar #0 nick_case([irctolower $target]) nc
+    set nshow $target
+    if {[info exists nc]} {
+	upvar #0 nick_onchans($nc) oc
+	upvar #0 nick_username($nc) nu
+	if {[info exists oc]} { set nshow $nc }
+    }
+    if {![nickdb_exists $target]} {
+	set ol "$nshow is not a registered nick."
+    } elseif {[string length [set username [nickdb_get $target username]]]} {
+	set ol "The nick $nshow belongs to the user $username."
+    } else {
+	set ol "The nick $nshow is registered (but not to a username)."
+    }
+    if {![info exists nc] || ![info exists oc]} {
+	if {$myself} {
+	    append ol "\nI can't see $nshow on anywhere."
+	} else {
+	    append ol "\nYou aren't on any channels with me."
+	}
+    } elseif {![info exists nu]} {
+	append ol "\n$nshow has not identified themselves."
+    } elseif {![info exists username]} {
+	append ol "\n$nshow has identified themselves as the user $nu."
+    } elseif {"$nu" != "$username"} {
+	append ol "\nHowever, $nshow is being used by the user $nu."
+    } else {
+	append ol "\n$nshow has identified themselves to me."
+    }
+    ucmdr {} $ol
+}
+
 def_ucmd register {
     prefix_nick
     check_notonchan
@@ -709,7 +1018,7 @@ def_ucmd register {
 	    upvar #0 nick_username($n) nu
 	    if {![info exists nu]} {
 		ucmdr {} \
- "You must identify yourself before using `register'.  See `help identify'."
+ "You must identify yourself before using `register'.  See `help identify', or use `register insecure'."
 	    }
 	    nickdb_set $n username $nu
 	    ucmdr {} {} "makes a note of your username." {}
@@ -751,7 +1060,7 @@ proc def_setting {opt show_body set_body} {
 }
 
 def_setting timeformat {
-    set tf [nickdb_opt $n timeformat]
+    set tf [nickdb_get $n timeformat]
     return "$tf: [timeformat_desc $tf]"
 } {
     set tf [string tolower [ta_word]]
@@ -762,7 +1071,7 @@ def_setting timeformat {
 }
 
 def_setting security {
-    set s [nickdb_opt $n username]
+    set s [nickdb_get $n username]
     if {[string length $s]} {
 	return "Your nick, $n, is controlled by the user $s."
     } else {