-
Notifications
You must be signed in to change notification settings - Fork 3
/
mini_settings_set_main.c
122 lines (109 loc) · 3.4 KB
/
mini_settings_set_main.c
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
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
#include <errno.h>
#include <stdbool.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include "mini_settings_read_file.h"
#include "mini_settings_set.h"
#ifndef DEFAULT_CONFIG_IN_PATH
#ifdef DEFAULT_CONFIG_PATH
#define DEFAULT_CONFIG_IN_PATH DEFAULT_CONFIG_PATH
#else
#define DEFAULT_CONFIG_IN_PATH "/dev/stdin"
#endif
#endif
#ifndef DEFAULT_CONFIG_OUT_PATH
#ifdef DEFAULT_CONFIG_PATH
#define DEFAULT_CONFIG_OUT_PATH DEFAULT_CONFIG_PATH
#else
#define DEFAULT_CONFIG_OUT_PATH "/dev/stdout"
#endif
#endif
static const char kUsage[] =
"Usage: %s [-f CONFIG_FILE] [--validate] <KEY> <VALUE> [KEY "
"VALUE]...\n\n"
"Sets value(s) in the config file.\n"
"If a commented key exists, will uncomment it and set the value in-place"
" instead of appending a new key\n\n"
"If --validate is passed, checks that the file is in the key=value"
" format.\n\n"
"By default, reads from " DEFAULT_CONFIG_IN_PATH
" and writes to " DEFAULT_CONFIG_OUT_PATH "\n";
int main(int argc, char *argv[]) {
if (argc < 2 || strcmp(argv[1], "--help") == 0) {
fprintf(stderr, kUsage, argv[0]);
return EXIT_FAILURE;
}
bool validate = false;
const char *config_in_path = DEFAULT_CONFIG_IN_PATH;
const char *config_out_path = DEFAULT_CONFIG_OUT_PATH;
unsigned i = 1;
for (; i < argc && argv[i][0] == '-'; ++i) {
if (strcmp(argv[i], "-f") == 0) {
if (i + 1 >= argc) {
const char message[] = "-f requires an argument\n";
fwrite(message, sizeof(char), sizeof(message), stderr);
return EXIT_FAILURE;
}
config_in_path = config_out_path = argv[++i];
} else if (strcmp(argv[i], "--validate") == 0) {
validate = true;
} else if (strcmp(argv[i], "--") == 0) {
++i;
break;
} else {
fprintf(stderr, "unknown argument: %s\n", argv[i]);
return EXIT_FAILURE;
}
}
const char **kvs = (const char **)argv + i;
const size_t kvs_size = argc - i;
if (kvs_size == 0) {
const char message[] = "Pass at least 1 key and 1 value\n";
fwrite(message, sizeof(char), sizeof(message), stderr);
return EXIT_FAILURE;
}
if (kvs_size % 2 != 0) {
const char message[] = "the number of key-value arguments must be even\n";
fwrite(message, sizeof(char), sizeof(message), stderr);
return EXIT_FAILURE;
}
char *config_contents;
size_t config_contents_size;
if (!mini_settings_try_read_file(config_in_path, &config_contents,
&config_contents_size)) {
if (errno == ENOENT) {
config_contents = NULL;
config_contents_size = 0;
} else {
perror(config_in_path);
return EXIT_FAILURE;
}
}
struct mini_settings_set_result_t result = mini_settings_set(
config_contents, config_contents_size, kvs, kvs_size, validate);
if (result.error != NULL) {
fwrite(result.error, sizeof(char), result.error_size, stderr);
free(result.error);
fputc('\n', stderr);
goto set_failed;
}
FILE *config_file = fopen(config_out_path, "w");
if (config_file == NULL) {
perror(config_out_path);
goto set_failed;
}
fwrite(result.contents, sizeof(char), result.contents_size, config_file);
if (ferror(config_file)) {
perror("write error");
goto set_failed;
}
fclose(config_file);
free(result.contents);
free(config_contents);
return EXIT_SUCCESS;
set_failed:
free(result.contents);
free(config_contents);
return EXIT_FAILURE;
}