git.meshlink.io Git - meshlink/blob - examples/chat.c

   1 #include <stdio.h>
   2 #include <stdlib.h>
   3 #include <string.h>
   4 #include "../src/meshlink.h"
   5
   6 static void log_message(meshlink_handle_t *mesh, meshlink_log_level_t level, const char *text) {
   7         const char *levelstr[] = {"DEBUG", "INFO", "WARNING", "ERROR", "CRITICAL"};
   8         fprintf(stderr, "%s: %s\n", levelstr[level], text);
   9 }
  10
  11 static void receive(meshlink_handle_t *mesh, meshlink_node_t *source, const void *data, size_t len) {
  12         const char *msg = data;
  13
  14         if(!len || msg[len - 1]) {
  15                 fprintf(stderr, "Received invalid data from %s\n", source->name);
  16                 return;
  17         }
  18
  19         printf("%s says: %s\n", source->name, msg);
  20 }
  21
  22 static void node_status(meshlink_handle_t *mesh, meshlink_node_t *node, bool reachable) {
  23         if(reachable)
  24                 printf("%s joined.\n", node->name);
  25         else
  26                 printf("%s left.\n", node->name);
  27 }
  28
  29 static void parse_command(meshlink_handle_t *mesh, char *buf) {
  30         char *arg = strchr(buf, ' ');
  31         if(arg)
  32                 *arg++ = 0;
  33
  34         if(!strcasecmp(buf, "invite")) {
  35                 char *invitation;
  36
  37                 if(!arg) {
  38                         fprintf(stderr, "/invite requires an argument!\n");
  39                         return;
  40                 }
  41
  42                 invitation = meshlink_invite(mesh, arg);
  43                 if(!invitation) {
  44                         fprintf(stderr, "Could not invite '%s': %s\n", arg, mesh->errstr);
  45                         return;
  46                 }
  47
  48                 printf("Invitation for %s: %s\n", arg, invitation);
  49                 free(invitation);
  50         } else if(!strcasecmp(buf, "join")) {
  51                 if(!arg) {
  52                         fprintf(stderr, "/join requires an argument!\n");
  53                         return;
  54                 }
  55
  56                 if(!meshlink_join(mesh, arg))
  57                         fprintf(stderr, "Could not join using invitation: %s\n", mesh->errstr);
  58                 else
  59                         fprintf(stderr, "Invitation accepted!\n");
  60         } else if(!strcasecmp(buf, "kick")) {
  61                 if(!arg) {
  62                         fprintf(stderr, "/kick requires an argument!\n");
  63                         return;
  64                 }
  65
  66                 meshlink_node_t *node = meshlink_get_node(mesh, arg);
  67                 if(!node) {
  68                         fprintf(stderr, "Unknown node '%s'\n", arg);
  69                         return;
  70                 }
  71
  72                 meshlink_blacklist(mesh, node);
  73
  74                 printf("Node '%s' blacklisted.\n", arg);
  75         } else if(!strcasecmp(buf, "quit")) {
  76                 printf("Bye!\n");
  77                 fclose(stdin);
  78         } else {
  79                 fprintf(stderr, "Unknown command '/%s'\n", buf);
  80         }
  81 }
  82
  83 static void parse_input(meshlink_handle_t *mesh, char *buf) {
  84         static meshlink_node_t *destination;
  85         size_t len;
  86
  87         if(!buf)
  88                 return;
  89
  90         // Remove newline.
  91
  92         len = strlen(buf);
  93
  94         if(len && buf[len - 1] == '\n')
  95                 buf[--len] = 0;
  96
  97         if(len && buf[len - 1] == '\r')
  98                 buf[--len] = 0;
  99
 100         // Ignore empty lines.
 101
 102         if(!len)
 103                 return;
 104
 105         // Commands start with '/'
 106
 107         if(*buf == '/')
 108                 return parse_command(mesh, buf + 1);
 109
 110         // Lines in the form "name: message..." set the destination node.
 111
 112         char *msg = buf;
 113         char *colon = strchr(buf, ':');
 114
 115         if(colon) {
 116                 *colon = 0;
 117                 msg = colon + 1;
 118                 if(*msg == ' ')
 119                         msg++;
 120
 121                 destination = meshlink_get_node(mesh, buf);
 122                 if(!destination) {
 123                         fprintf(stderr, "Unknown node '%s'\n", buf);
 124                         return;
 125                 }
 126         }
 127
 128         if(!destination) {
 129                 fprintf(stderr, "Who are you talking to? Write 'name: message...'\n");
 130                 return;
 131         }
 132
 133         if(!meshlink_send(mesh, destination, msg, strlen(msg) + 1)) {
 134                 fprintf(stderr, "Could not send message to '%s': %s\n", destination->name, mesh->errstr);
 135                 return;
 136         }
 137
 138         printf("Message sent to '%s'.\n", destination->name);
 139 }
 140
 141 int main(int argc, char *argv[]) {
 142         const char *confbase = ".chat";
 143         const char *nick = NULL;
 144         char buf[1024];
 145
 146         if(argc > 1)
 147                 confbase = argv[1];
 148
 149         if(argc > 2)
 150                 nick = argv[2];
 151
 152         meshlink_handle_t *mesh = meshlink_open(confbase, nick);
 153         if(!mesh) {
 154                 fprintf(stderr, "Could not open MeshLink: %s\n", mesh->errstr);
 155                 return 1;
 156         }
 157
 158         meshlink_set_receive_cb(mesh, receive);
 159         meshlink_set_node_status_cb(mesh, node_status);
 160         meshlink_set_log_cb(mesh, MESHLINK_INFO, log_message);
 161
 162         if(!meshlink_start(mesh)) {
 163                 fprintf(stderr, "Could not start MeshLink: %s\n", mesh->errstr);
 164                 return 1;
 165         }
 166
 167         printf("Chat started.\n");
 168
 169         while(fgets(buf, sizeof buf, stdin))
 170                 parse_input(mesh, buf);
 171
 172         printf("Chat stopping.\n");
 173
 174         meshlink_stop(mesh);
 175         meshlink_close(mesh);
 176
 177         return 0;
 178 }