Changeset 263 in code for trunk/service.go

Timestamp:

Apr 16, 2020, 3:38:47 PM (5 years ago)

Author:

delthas

Message:

Add support for custom network on-connect commands

Some servers use custom IRC bots with custom commands for registering to
specific services after connection.

This adds support for setting custom raw IRC messages, that will be
sent after registering to a network.

It also adds support for a custom flag.Value type for string
slice flags (flags taking several string values).

File:

• trunk/service.go (modified) (5 diffs)

trunk/service.go

@@ -105 +105 @@
                         children: serviceCommandSet{
                                 "create": {
-                                        usage:  "-addr <addr> [-name name] [-username username] [-pass pass] [-realname realname] [-nick nick]",
+                                        usage:  "-addr <addr> [-name name] [-username username] [-pass pass] [-realname realname] [-nick nick] [[-connect-command command] ...]",
                                         desc:   "add a new network",
                                         handle: handleServiceCreateNetwork,
@@ -175 +175 @@
 }
 
+type stringSliceVar []string
+
+func (v *stringSliceVar) String() string {
+        return fmt.Sprint([]string(*v))
+}
+
+func (v *stringSliceVar) Set(s string) error {
+        *v = append(*v, s)
+        return nil
+}
+
 func handleServiceCreateNetwork(dc *downstreamConn, params []string) error {
         fs := newFlagSet()
@@ -183 +194 @@
         realname := fs.String("realname", "", "")
         nick := fs.String("nick", "", "")
+        var connectCommands stringSliceVar
+        fs.Var(&connectCommands, "connect-command", "")
 
         if err := fs.Parse(params); err != nil {
@@ -191 +204 @@
         }
 
+        for _, command := range connectCommands {
+                _, err := irc.ParseMessage(command)
+                if err != nil {
+                        return fmt.Errorf("flag -connect-command must be a valid raw irc command string: %q: %v", command, err)
+                }
+        }
+
         if *nick == "" {
                 *nick = dc.nick
@@ -197 +217 @@
         var err error
         network, err := dc.user.createNetwork(&Network{
-                Addr:     *addr,
-                Name:     *name,
-                Username: *username,
-                Pass:     *pass,
-                Realname: *realname,
-                Nick:     *nick,
+                Addr:            *addr,
+                Name:            *name,
+                Username:        *username,
+                Pass:            *pass,
+                Realname:        *realname,
+                Nick:            *nick,
+                ConnectCommands: connectCommands,
         })
         if err != nil {