chiark / gitweb /
Do not trash errorcode &c with unset calling_nick
[ircbot] / bot.tcl
diff --git a/bot.tcl b/bot.tcl
index f03c1832e6f86a361aa68a357e13db9d8b2397e7..09c4ca2caf446eaaeb8bda422ee99fce636ff0c8 100755 (executable)
--- a/bot.tcl
+++ b/bot.tcl
@@ -3,8 +3,8 @@
 set host chiark
 set port 6667
 if {![info exists nick]} { set nick Blight }
+if {![info exists ownfullname]} { set ownfullname "here to Help" }
 set ownmailaddr blight@chiark.greenend.org.uk
-set ownfullname "here to Help"
 
 if {![info exists globalsecret]} {
     set gsfile [open /dev/urandom r]
@@ -15,6 +15,29 @@ if {![info exists globalsecret]} {
     unset gsfile
 }
 
+proc try_except_finally {try except finally} {
+    global errorInfo errorCode
+    set er [catch { uplevel 1 $try } emsg]
+    if {$er} {
+       set ei $errorInfo
+       set ec $errorCode
+       if {[catch { uplevel 1 $except } emsg3]} {
+           append ei "\nALSO ERROR HANDLING ERROR:\n$emsg3"
+       }
+    }
+    set er2 [catch { uplevel 1 $finally } emsg2]
+    if {$er} {
+       if {$er2} {
+           append ei "\nALSO ERROR CLEANING UP:\n$emsg2"
+       }
+       return -code $er -errorinfo $ei -errorcode $ec $emsg
+    } elseif {$er2} {
+       return -code $er2 -errorinfo $errorInfo -errorcode $errorCode $emsg2
+    } else {
+       return $emsg
+    }
+}
+
 proc sendout {command args} {
     global sock
     if {[llength $args]} {
@@ -57,13 +80,24 @@ proc bgerror {msg} {
 }
 
 proc onread {args} {
-    global sock
+    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]} {
+           set calling_nick $maybenick
+           if {"[irctolower $maybenick]" == "[irctolower $nick]"} return
+       }
     } else {
        set prefix {}
     }
@@ -145,10 +179,16 @@ proc prefix_nick {} {
     upvar 1 n n
     if {![regexp {^([^!]+)!} $p dummy n]} { error "not from nick" }
     check_nick $n
-    if {"[irctolower $n]" == "[irctolower $nick]"} { error "from myself" }
+    if {"[irctolower $n]" == "[irctolower $nick]"} {
+       error "from myself" {} {}
+    }
 }
 
 proc showintervalsecs {howlong} {
+    return [showintervalsecs/[opt timeformat] $howlong]
+}
+
+proc showintervalsecs/ks {howlong} {
     if {$howlong < 1000} {
        return "${howlong}s"
     } else {
@@ -167,6 +207,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}
@@ -313,15 +378,89 @@ 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} {
+    set chans [split $chans ,]
+    set users [split $users ,]
+    if {[llength $chans] > 1} {
+       foreach chan $chans user $users { process_kickpart $chan $user }
+    } else {
+       foreach user $users { process_kickpart [lindex $chans 0] $user }
+    }
+}
+
+proc msg_KILL {p c user why} {
+    nick_forget $user
+}
+
+set nick_arys {onchans username}
+
+proc nick_forget {n} {
+    global nick_arys
+    foreach ary $nick_arys {
+       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 nick_case
     prefix_nick
     recordlastseen_n $n "changing nicks to $newnick" 0
     recordlastseen_n $newnick "changing nicks from $n" 1
+    foreach ary $nick_arys {
+       upvar #0 nick_${ary}($n) old
+       upvar #0 nick_${ary}($newnick) new
+       if {[info exists new]} { error "nick collision ?! $ary $n $newnick" }
+       if {[info exists old]} { set new $old; unset old }
+    }
+    nick_case $newnick
 }
 
-proc msg_JOIN {p c chan} { recordlastseen_p $p "joining $chan" 1 }
-proc msg_PART {p c chan} { recordlastseen_p $p "leaving $chan" 1 }
-proc msg_QUIT {p c why} { recordlastseen_p $p "leaving ($why)" 0 }
+proc msg_JOIN {p c chan} {
+    prefix_nick
+    recordlastseen_n $n "joining $chan" 1
+    upvar #0 nick_onchans($n) oc
+    lappend oc [irctolower $chan]
+    nick_case $n
+}
+proc msg_PART {p c chan} {
+    prefix_nick
+    recordlastseen_n $n "leaving $chan" 1
+    process_kickpart $chan $n
+}
+proc msg_QUIT {p c why} {
+    prefix_nick
+    recordlastseen_n $n "leaving ($why)" 0
+    nick_forget $n
+}
 
 proc msg_PRIVMSG {p c dest text} {
     prefix_nick
@@ -332,6 +471,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
@@ -345,19 +485,65 @@ proc msg_PRIVMSG {p c dest text} {
        sendprivmsg $n "error: $rv"
     } else {
        manyset $rv priv_msgs pub_msgs priv_acts pub_acts
-       foreach {td val} [list $n $priv_msgs $output $pub_msgs] {
+       foreach {td val} [list $n $priv_acts $output $pub_acts] {
            foreach l [split $val "\n"] {
-               sendprivmsg $td $l
+               sendaction $td $l
            }
        }
-       foreach {td val} [list $n $priv_acts $output $pub_acts] {
+       foreach {td val} [list $n $priv_msgs $output $pub_msgs] {
            foreach l [split $val "\n"] {
-               sendaction $td $l
+               sendprivmsg $td $l
            }
        }
     }
 }
 
+proc msg_INVITE {p c n chan} {
+    after 1000 [list sendout JOIN $chan]
+}
+
+proc grep {var predicate list} {
+    set o {}
+    upvar 1 $var v
+    foreach v $list {
+       if {[uplevel 1 [list expr $predicate]]} { lappend o $v }
+    }
+    return $o
+}
+
+proc msg_353 {p c dest type chan nicklist} {
+    global names_chans nick_onchans
+    if {![info exists names_chans]} { set names_chans {} }
+    set chan [irctolower $chan]
+    lappend names_chans $chan
+    channel_noone_seen $chan
+    foreach n [split $nicklist { }] {
+       regsub {^[@+]} $n {} n
+       if {![string length $n]} continue
+       check_nick $n
+       upvar #0 nick_onchans($n) oc
+       lappend oc $chan
+       nick_case $n
+    }
+}
+
+proc msg_366 {p c args} {
+    global names_chans nick_onchans
+    if {[llength names_chans] > 1} {
+       foreach n [array names nick_onchans] {
+           upvar #0 nick_onchans($n) oc
+           set oc [grep tc {[lsearch -exact $tc $names_chans] >= 0} $oc]
+           if {![llength $oc]} { nick_forget $n }
+       }
+    }
+    unset names_chans
+}
+
+proc ta_anymore {} {
+    upvar 1 text text
+    return [expr {!![string length $text]}]
+}
+
 proc ta_nomore {} {
     upvar 1 text text
     if {[string length $text]} { error "too many parameters" }
@@ -386,18 +572,51 @@ proc ucmdr {priv pub args} {
     return -code return [concat [list $priv $pub] $args]
 }
 
-proc ucmd_sendhelp {} {
-    ucmdr \
-{Commands currently understood:
- help              get this list of commands
- seen <nick>       ask after someone (I'll tell them you asked)
- summon <username> invite a logged-on user onto IRC
-Send commands to be by /msg, or say them in channel with ! in front.} {}
+proc loadhelp {} {
+    global help_topics
+
+    catch { unset help_topics }
+    set f [open helpinfos r]
+    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"
+           }
+       }
+       if {[info exists topic]} { error "unfinished topic $topic" }
+    } {} {
+       close $f
+    }
 }
 
-def_ucmd help { ta_nomore; ucmd_sendhelp }
+def_ucmd help {
+    upvar #0 help_topics([irctolower [string trim $text]]) info
+    if {![info exists info]} { ucmdr "No help on $text, sorry." {} }
+    ucmdr $info {}
+}
 
-def_ucmd ? { ta_nomore; ucmd_sendhelp }
+def_ucmd ? {
+    global help_topics
+    ucmdr $help_topics() {}
+}
 
 proc manyset {list args} {
     foreach val $list var $args {
@@ -406,14 +625,270 @@ proc manyset {list args} {
     }
 }
 
-def_ucmd summon {
-    set target [ta_word]
-    ta_nomore
+proc check_username {target} {
     if {
        [string length $target] > 8 ||
         [regexp {[^-0-9a-z]} $target] ||
         ![regexp {^[a-z]} $target]
     } { error "invalid username" }
+}
+
+proc nickdb__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]
+           try_except_finally { set newval [read $f] } {} { close $f }
+           if {[llength $newval] % 2} { error "invalid length" }
+           set ndbe $newval
+       }
+    }
+}
+
+proc def_nickdb {name arglist body} {
+    proc nickdb_$name $arglist "nickdb__head; $body"
+}
+
+def_nickdb exists {n} {
+    return [info exists ndbe]
+}
+
+def_nickdb delete {n} {
+    catch { unset ndbe }
+    file delete $nfn
+}
+
+set default_settings {timeformat ks}
+
+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 }
+    set newval {}
+    foreach {key value} [array get a] { lappend newval $key $value }
+    set f [open $nfn.new w]
+    try_except_finally {
+       puts $f $newval
+       close $f
+       file rename -force $nfn.new $nfn
+    } {
+    } {
+       catch { close $f }
+    }
+    set ndbe $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
+    } else {
+       set l $default_settings
+    }
+    foreach {tkey value} $l {
+       if {"$tkey" == "$key"} { return $value }
+    }
+    error "unset setting $key"
+}
+
+proc check_notonchan {} {
+    upvar 1 dest dest
+    if {[ischan $dest]} { error "that command must be sent privately" }
+}
+
+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]
+    if {![string length $wantu]} {
+       if {$strict} {
+           error "that feature is only available to secure users, sorry."
+       } else {
+           return
+       }
+    }
+    upvar #0 nick_username($n) nu
+    if {![info exists nu]} {
+       error "nick $n is secure, you must identify yourself first."
+    }
+    if {"$wantu" != "$nu"} {
+       error "you are the wrong user - the nick $n belongs to $wantu, not $nu"
+    }
+}
+
+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_opt $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
+    set old [nickdb_exists $n]
+    if {$old} { nick_securitycheck 0 }
+    switch -exact [string tolower [string trim $text]] {
+       {} {
+           upvar #0 nick_username($n) nu
+           if {![info exists nu]} {
+               ucmdr {} \
+ "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." {}
+       }
+       delete {
+           nickdb_delete $n
+           ucmdr {} {} "forgets your nickname." {}
+       }
+       insecure {
+           nickdb_set $n username {}
+           if {$old} {
+               ucmdr {} "Security is now disabled for your nickname !"
+           } else {
+               ucmdr {} "This is fine, but bear in mind that people will be able to mess with your settings.  Channel management features need a secure registration." "makes an insecure registration for your nick."
+           }
+       }
+    }
+}
+
+proc timeformat_desc {tf} {
+    switch -exact $tf {
+       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" }
+    }
+}
+
+proc def_setting {opt show_body set_body} {
+    proc set_show/$opt {} "
+        upvar 1 n n
+        set opt $opt
+        $show_body"
+    if {![string length $set_body]} return
+    proc set_set/$opt {} "
+        upvar 1 n n
+        upvar 1 text text
+        set opt $opt
+        $set_body"
+}
+
+def_setting timeformat {
+    set tf [nickdb_opt $n timeformat]
+    return "$tf: [timeformat_desc $tf]"
+} {
+    set tf [string tolower [ta_word]]
+    ta_nomore
+    set desc [timeformat_desc $tf]
+    nickdb_set $n timeformat $tf
+    ucmdr {} $desc
+}
+
+def_setting security {
+    set s [nickdb_opt $n username]
+    if {[string length $s]} {
+       return "Your nick, $n, is controlled by the user $s."
+    } else {
+       return "Your nick, $n, is not secure."
+    }
+} {}
+
+def_ucmd set {
+    prefix_nick
+    check_notonchan
+    if {![nickdb_exists $n]} {
+       ucmdr {} "You are unknown to me and so have no settings.  (Use `register'.)"
+    }
+    if {![ta_anymore]} {
+       set ol {}
+       foreach proc [lsort [info procs]] {
+           if {![regexp {^set_show/(.*)$} $proc dummy opt]} continue
+           lappend ol [format "%-10s %s" $opt [set_show/$opt]]
+       }
+       ucmdr {} [join $ol "\n"]
+    } else {
+       set opt [ta_word]
+       if {[catch { info body set_show/$opt }]} {
+           error "no setting $opt"
+       }
+       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'"
+           }
+           set_set/$opt
+       }
+    }
+}
+
+def_ucmd identpass {
+    set username [ta_word]
+    set passmd5 [md5sum "[ta_word]\n"]
+    ta_nomore
+    prefix_nick
+    check_notonchan
+    upvar #0 nick_onchans($n) onchans
+    if {![info exists onchans] || ![llength $onchans]} {
+       ucmdr "You must be on a channel with me to identify yourself." {}
+    }
+    check_username $username
+    exec userv --timeout 3 $username << "$passmd5\n" > /dev/null \
+           irc-identpass $n
+    upvar #0 nick_username($n) rec_username
+    set rec_username $username
+    ucmdr "Pleased to see you, $username." {}
+}
+
+def_ucmd summon {
+    set target [ta_word]
+    ta_nomore
+    check_username $target
     prefix_nick
 
     upvar #0 lastsummon($target) ls
@@ -449,61 +924,7 @@ def_ucmd summon {
     }
 }
 
-def_ucmd newuser {
-    global ownmailaddr ownfullname nick globalsecret
-    prefix_nick
-    if {[ischan $dest]} {
-       error "You must register privately."
-    }
-    binary scan [irctolower $n] H* nhex
-    if {[file exists users/$nhex]} {
-       error "You (or someone else) have already registered the nick $n."
-    }
-    set ownermail [ta_word]
-
-    set now [clock seconds]
-    set small 100000
-    set mult 6
-    set ksecs [expr {$now / $small}]
-    set kmod [expr {$ksecs % $mult}]
-
-    if {[string length $text]} {
-       if {![regexp -nocase {^([0-5])([0-9a-f]+)$} $text pass_sup kmod]} {
-       }
-       set ksecs [expr {(($ksecs - $kmod) / $mult) * 6 + $kmod}]
-    }
-    set tohash "$ksecs\n$ownermail\n"
-    set hash [exec md5sum << $tohash]
-    set passwd "$kmod[string range $hash 0 15]"
-
-    if {[info exists pass_sup]} {
-       if {"$passwd" != "$pass_sup"} {
-           error "Incorrect registration password."
-       }
-       sendaction $n "ignores your ok"
-    } else {
-       set mailmsg \
-"From: $ownmailaddr ($ownfullname)
-Subject: $nick registration
-To: $ownermail
-
-Thanks for starting the registration process.  You must now issue the
-`newuser' command with both the same email address again, and your
-registration password from this mail.
-
-Nick:          $n
-Email address: $ownermail
-Password:      $passwd
-
-This password will be valid for approximately the next 600ks
-(or until I am restarted).
-
-For example,
- /msg $nick newuser $ownermail $passwd"
-        exec /usr/sbin/sendmail -odi -oee -oi -t << $mailmsg
-        sendaction $n "has sent your registration mail to $ownermail."
-    }
-}
+proc md5sum {value} { exec md5sum << $value }
 
 def_ucmd seen {
     global lastseen nick
@@ -543,11 +964,13 @@ if {![info exists sock]} {
     #fconfigure $sock -translation binary
     fconfigure $sock -translation crlf
 
-    sendout USER guest 0 * $ownfullname
+    sendout USER blight 0 * $ownfullname
     sendout NICK $nick
     fileevent $sock readable onread
 }
 
+loadhelp
+
 #if {![regexp {tclsh} $argv0]} {
 #    vwait terminate
 #}