follow invites

irc/README.md

@@ -11,12 +11,18 @@ that a user would be able to pwn your system though.
 In an IRC channel, prefix a message with `rink: ` and then put your
 query after. Direct messages are always interpreted as queries.
 
+To add rink to a new channel, you can invite it and it will follow. It
+won't remember this if the bot is restarted though.
+
 ## How to operate it
 
 You'll need to create a `config.toml` for the bot. It uses the following
 format (the values show the defaults):
 
 ```toml
+[behavior]
+follow_invites = true
+
 [limits]
 # Not yet implemented
 enabled = true

irc/src/config.rs

@@ -49,11 +49,26 @@ impl Default for Currency {
     }
 }
 
+#[derive(Serialize, Deserialize, Clone)]
+#[serde(default, deny_unknown_fields)]
+pub struct Behavior {
+    pub follow_invites: bool,
+}
+
+impl Default for Behavior {
+    fn default() -> Self {
+        Behavior {
+            follow_invites: true,
+        }
+    }
+}
+
 #[derive(Serialize, Deserialize, Clone, Default)]
 #[serde(default, deny_unknown_fields)]
 pub struct Config {
     pub limits: Limits,
     pub currency: Currency,
+    pub behavior: Behavior,
     pub servers: Vec<IrcConfig>,
 }
 

irc/src/fmt.rs

@@ -23,7 +23,7 @@ fn write_irc_string(out: &mut String, config: &Config, spans: &[Span]) {
     }
 }
 
-fn write_irc_token(out: &mut String, config: &Config, text: &str, token: FmtToken) {
+fn write_irc_token(out: &mut String, _config: &Config, text: &str, token: FmtToken) {
     let (prefix, postfix) = match token {
         FmtToken::Plain => ("", ""),
         FmtToken::Error => ("\x0304", "\x03"),

irc/src/main.rs

@@ -58,6 +58,10 @@ async fn server_task(config: config::Config, index: usize) {
                 &channel
             };
             client.send_notice(where_to, result).unwrap();
+        } else if let Command::INVITE(_nickname, channel) = message.command {
+            if config.behavior.follow_invites {
+                client.send_join(channel).unwrap();
+            }
         }
     }
 }