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