1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
|
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include "mcp/server.h"
#include "storage/store.h"
#include "sys/system.h"
#include "base/allocator.h"
static void
usage(FILE *stream)
{
fprintf(stream, "Usage: tiny-convmem-mcp --data-dir <path>\n");
}
static int
run(const tiny_sys_arguments_t *arguments)
{
char *absolute;
tiny_store_t *store = NULL;
tiny_store_status_t status;
int result;
if (arguments->count == 2 && strcmp(arguments->values[1], "--help") == 0)
{
usage(stdout);
return 0;
}
if (arguments->count != 3 || strcmp(arguments->values[1], "--data-dir") != 0
|| arguments->values[2][0] == '\0')
{
usage(stderr);
return 2;
}
absolute = tiny_sys_path_absolute(arguments->values[2]);
if (absolute == NULL)
{
fprintf(stderr, "unable to resolve --data-dir\n");
return 2;
}
status = tiny_store_open(&store, absolute);
free(absolute);
if (status != TINY_STORE_OK)
{
fprintf(stderr, "unable to open conversation store: %s\n", tiny_store_status_name(status));
return status == TINY_STORE_BUSY ? 3 : 1;
}
result = tiny_convmem_server_run(stdin, stdout, store);
tiny_store_close(store);
if (result != 0)
{
fprintf(stderr, "tiny-convmem-mcp stopped with error code %d\n", result);
}
return result;
}
int
main(int argc, char **argv)
{
tiny_sys_arguments_t arguments;
int result;
if (!tiny_sys_arguments_init(&arguments, argc, argv))
{
fprintf(stderr, "unable to read process arguments\n");
return 2;
}
result = run(&arguments);
tiny_sys_arguments_free(&arguments);
return result;
}
|
