1 #include <u.h>
2 #include <libc.h>
3 #include <bio.h>
4 #include "msgdb.h"
5
6 void
usage(void)7 usage(void)
8 {
9 fprint(2, "usage: msgdb [-c] file\n");
10 exits("usage");
11 }
12
13 void
main(int argc,char ** argv)14 main(int argc, char **argv)
15 {
16 int create = 0;
17 Msgdb *db;
18 char *tok, *p;
19 long val;
20 int input;
21 Biobuf b;
22
23 input = 0;
24 ARGBEGIN{
25 case 'c':
26 create = 1;
27 break;
28 case 'i':
29 input = 1;
30 break;
31 default:
32 usage();
33 }ARGEND
34
35 if(argc != 1)
36 usage();
37
38 if((db = mdopen(argv[0], create)) == nil)
39 sysfatal("open db: %r");
40
41 if(input){
42 Binit(&b, 0, OREAD);
43 while((tok = Brdline(&b, '\n')) != nil){
44 tok[Blinelen(&b)-1] = '\0';
45 p = strrchr(tok, ' ');
46 if(p == nil)
47 val = mdget(db, tok)+1;
48 else{
49 *p++ = 0;
50 val = atoi(p);
51 }
52 mdput(db, tok, val);
53 }
54 }else{
55 mdenum(db);
56 Binit(&b, 1, OWRITE);
57 while(mdnext(db, &tok, &val) >= 0)
58 Bprint(&b, "%s %ld\n", tok, val);
59 Bterm(&b);
60 }
61 mdclose(db);
62 exits(nil);
63 }
64