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