X-Git-Url: https://git.distorted.org.uk/~mdw/ircbot/blobdiff_plain/11d9bff953ec37d62ab4f6b706da368257b659aa..28c8ab788e2733e81f35d586bdbbe1678ce0398f:/bot.tcl diff --git a/bot.tcl b/bot.tcl index b52e139..55d93c7 100755 --- a/bot.tcl +++ b/bot.tcl @@ -6,13 +6,17 @@ if {![info exists nick]} { set nick Blight } if {![info exists ownfullname]} { set ownfullname "here to Help" } set ownmailaddr blight@chiark.greenend.org.uk -if {![info exists globalsecret]} { - set gsfile [open /dev/urandom r] - fconfigure $gsfile -translation binary - set globalsecret [read $gsfile 32] - binary scan $globalsecret H* globalsecret - close $gsfile - unset gsfile +set musthaveping_ms 10000 +set out_maxburst 6 +set out_interval 2100 +set out_lag_lag 5000 +set out_lag_very 25000 + +proc manyset {list args} { + foreach val $list var $args { + upvar 1 $var my + set my $val + } } proc try_except_finally {try except finally} { @@ -38,8 +42,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 +117,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,15 +155,24 @@ proc bgerror {msg} { } proc onread {args} { - global sock nick + global sock nick calling_nick errorInfo errorCode - if {[gets $sock line] == -1} { set terminate 1; return } + if {[gets $sock line] == -1} { fail "EOF/error on input" } 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 + if {[regexp {^([^!]+)!} $prefix dummy maybenick]} { + set calling_nick $maybenick + if {"[irctolower $maybenick]" == "[irctolower $nick]"} return + } } else { set prefix {} } @@ -129,9 +213,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] } @@ -141,8 +229,10 @@ proc prefix_none {} { } proc msg_PING {p c s1} { + global musthaveping_after prefix_none sendout PONG $s1 + if {[info exists musthaveping_after]} { after cancel $musthaveping_after] } } proc check_nick {n} { @@ -176,6 +266,10 @@ proc prefix_nick {} { } proc showintervalsecs {howlong} { + return [showintervalsecs/[opt timeformat] $howlong] +} + +proc showintervalsecs/ks {howlong} { if {$howlong < 1000} { return "${howlong}s" } else { @@ -194,6 +288,31 @@ proc showintervalsecs {howlong} { } } +proc format_qty {qty unit} { + set o $qty + append o " " + append o $unit + if {$qty != 1} { append o s } + return $o +} + +proc showintervalsecs/hms {qty} { + set ul {second 60 minute 60 hour 24 day 7 week} + set remainv 0 + while {[llength $ul] > 1 && $qty >= [set uv [lindex $ul 1]]} { + set remainu [lindex $ul 0] + set remainv [expr {$qty % $uv}] + set qty [expr {($qty-$remainv)/$uv}] + set ul [lreplace $ul 0 1] + } + set o [format_qty $qty [lindex $ul 0]] + if {$remainv} { + append o " " + append o [format_qty $remainv $remainu] + } + return $o +} + proc showinterval {howlong} { if {$howlong <= 0} { return {just now} @@ -302,12 +421,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.)" + } + } } } @@ -340,13 +473,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} { @@ -363,7 +509,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 @@ -371,10 +518,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 @@ -384,6 +537,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} { @@ -391,6 +552,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 @@ -412,6 +574,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 @@ -427,7 +590,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] { @@ -456,16 +619,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 } } @@ -519,31 +680,49 @@ proc loadhelp {} { catch { unset help_topics } set f [open helpinfos r] - set lno 0 - while {[gets $f l] >= 0} { - incr lno - if {[regexp {^#.*} $l]} { - } elseif {[regexp {^ *$} $l]} { - if {[info exists topic]} { - set help_topics($topic) [join $lines "\n"] - unset topic - unset lines + try_except_finally { + set lno 0 + while {[gets $f l] >= 0} { + incr lno + if {[regexp {^#.*} $l]} { + } elseif {[regexp {^ *$} $l]} { + if {[info exists topic]} { + set help_topics($topic) [join $lines "\n"] + unset topic + unset lines + } + } elseif {[regexp {^!([-+._0-9a-z]*)$} $l dummy newtopic]} { + if {[info exists topic]} { + error "help $newtopic while in $topic" + } + set topic $newtopic + set lines {} + } elseif {[regexp {^[^!#]} $l]} { + set topic + lappend lines [string trimright $l] + } else { + error "eh ? $lno: $l" } - } elseif {[regexp {^!([-+._0-9a-z]*)$} $l dummy newtopic]} { - if {[info exists topic]} { error "help $newtopic while in $topic" } - set topic $newtopic - set lines {} - } elseif {[regexp {^[^!#]} $l]} { - set topic - lappend lines [string trimright $l] - } else { - error "eh ? $lno: $l" } + if {[info exists topic]} { error "unfinished topic $topic" } + } {} { + close $f } - if {[info exists topic]} { error "unfinished topic $topic" } } 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 {} @@ -554,13 +733,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 || @@ -569,60 +741,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/$nh - if {![info exists ndbe] && [file exists $nfn]} { - set f [file 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 } } { + catch { close $f } } - set ndbe $newval + set iddbe $newval } -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 } @@ -630,6 +806,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" } @@ -638,7 +820,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." @@ -655,6 +837,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 @@ -665,7 +1004,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." {} @@ -687,7 +1026,7 @@ def_ucmd register { proc timeformat_desc {tf} { switch -exact $tf { - ks { return "Times will be displayed in kiloseconds or seconds." } + ks { return "Times will be displayed in seconds or kiloseconds." } hms { return "Times will be displayed in hours, minutes, etc." } default { error "invalid timeformat: $v" } } @@ -707,7 +1046,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]] @@ -718,7 +1057,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 { @@ -730,7 +1069,7 @@ def_ucmd set { prefix_nick check_notonchan if {![nickdb_exists $n]} { - ucmdr {} "You are unknown to me and so have no settings." + ucmdr {} "You are unknown to me and so have no settings. (Use `register'.)" } if {![ta_anymore]} { set ol {} @@ -747,6 +1086,7 @@ def_ucmd set { if {![ta_anymore]} { ucmdr {} "$opt [set_show/$opt]" } else { + nick_securitycheck 0 if {[catch { info body set_set/$opt }]} { error "setting $opt cannot be set with `set'" } @@ -757,7 +1097,7 @@ def_ucmd set { def_ucmd identpass { set username [ta_word] - set passmd5 [md5sum [ta_word]] + set passmd5 [md5sum "[ta_word]\n"] ta_nomore prefix_nick check_notonchan @@ -846,18 +1186,54 @@ def_ucmd seen { ucmdr {} $rstr } -if {![info exists sock]} { +proc ensure_globalsecret {} { + global globalsecret + + if {[info exists globalsecret]} return + set gsfile [open /dev/urandom r] + fconfigure $gsfile -translation binary + set globalsecret [read $gsfile 32] + binary scan $globalsecret H* globalsecret + close $gsfile + unset gsfile +} + +proc ensure_outqueue {} { + out__vars + if {[info exists out_queue]} return + 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 +} + +proc fail {msg} { + logerror "failing: $msg" + exit 1 +} + +proc ensure_connecting {} { + global sock ownfullname host port nick + global musthaveping_ms musthaveping_after + + if {[info exists sock]} return set sock [socket $host $port] fconfigure $sock -buffering line - #fconfigure $sock -translation binary fconfigure $sock -translation crlf sendout USER blight 0 * $ownfullname sendout NICK $nick fileevent $sock readable onread + + set musthaveping_after [after $musthaveping_ms \ + {fail "no ping within timeout"}] } +ensure_globalsecret +ensure_outqueue loadhelp +ensure_connecting #if {![regexp {tclsh} $argv0]} { # vwait terminate