diff options
| -rw-r--r-- | litterbox.c | 36 |
1 files changed, 33 insertions, 3 deletions
diff --git a/litterbox.c b/litterbox.c index 1eb663a..4171427 100644 --- a/litterbox.c +++ b/litterbox.c @@ -106,9 +106,19 @@ static void require(const struct Message *msg, bool nick, size_t len) { } } -static char *self; static const char *join; +static char *self; +static char *network; +static char *chanTypes; +static char *prefixes; + +static void set(char **field, const char *value) { + free(*field); + *field = strdup(value); + if (!*field) err(EX_OSERR, "strdup"); +} + typedef void Handler(struct Message *msg); static void handleCap(struct Message *msg) { @@ -118,11 +128,27 @@ static void handleCap(struct Message *msg) { static void handleReplyWelcome(struct Message *msg) { require(msg, false, 1); - self = strdup(msg->params[0]); - if (!self) err(EX_OSERR, "strdup"); + set(&self, msg->params[0]); format("JOIN :%s\r\n", join); } +static void handleReplyISupport(struct Message *msg) { + for (size_t i = 0; i < ParamCap; ++i) { + if (!msg->params[i]) break; + char *key = strsep(&msg->params[i], "="); + if (!msg->params[i]) continue; + if (!strcmp(key, "NETWORK")) { + set(&network, msg->params[i]); + } else if (!strcmp(key, "CHANTYPES")) { + set(&chanTypes, msg->params[i]); + } else if (!strcmp(key, "PREFIX")) { + strsep(&msg->params[i], ")"); + if (!msg->params[i]) continue; + set(&prefixes, msg->params[i]); + } + } +} + static void handlePing(struct Message *msg) { require(msg, false, 1); format("PONG :%s\r\n", msg->params[0]); @@ -133,6 +159,7 @@ static const struct { Handler *fn; } Handlers[] = { { "001", handleReplyWelcome }, + { "005", handleReplyISupport }, { "CAP", handleCap }, { "PING", handlePing }, }; @@ -197,6 +224,9 @@ int main(int argc, char *argv[]) { } if (!host) errx(EX_USAGE, "host required"); + set(&network, host); + set(&chanTypes, "#&"); + set(&prefixes, "@+"); client = tls_client(); if (!client) errx(EX_SOFTWARE, "tls_client"); |