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
|
/*
* Created by yuuta on 4/1/22.
*/
#include "botd.h"
#include "log.h"
#include <stdlib.h>
#include <string.h>
#include <stdio.h>
#include <limits.h>
#include <inttypes.h>
#include <errno.h>
static inline void parse_api_id(const char *str) {
char *endptr;
intmax_t num = strtoimax(str, &endptr, 10);
if (strcmp(endptr, "") != 0 || (num == INTMAX_MAX && errno == ERANGE) ||
num > INT_MAX || num < INT_MIN) {
LOGFEV("Invalid API Hash: %s.", 64, str);
}
cmd.api_id = (int) num;
}
void parse_cmdline(int argc, char **argv) {
for (int i = 1; i < argc; i++) {
const char *arg = argv[i];
if (arg[0] != '-' || arg[1] != '-')
LOGFEV("Unexpected argument: %s.", 64, arg);
const char *a = &arg[2];
if (!strcmp(a, "test")) {
LOGD("Test DC is in use.");
cmd.test_dc = true;
} else if (!strcmp(a, "logout")) {
cmd.logout = true;
} else if (!strcmp(a, "api_id")) {
if (i == (argc - 1)) LOGFEV("%s expects an argument.", 64, arg);
parse_api_id(argv[++i]);
} else if (!strcmp(a, "api_hash")) {
if (i == (argc - 1)) LOGFEV("%s expects an argument.", 64, arg);
cmd.api_hash = argv[++i];
} else if (!strcmp(a, "bot_token")) {
if (i == (argc - 1)) LOGFEV("%s expects an argument.", 64, arg);
cmd.bot_token = argv[++i];
} else if (!strcmp(a, "td_path")) {
if (i == (argc - 1)) LOGFEV("%s expects an argument.", 64, arg);
cmd.td_path = argv[++i];
} else if (!strcmp(a, "db_path")) {
if (i == (argc - 1)) LOGFEV("%s expects an argument.", 64, arg);
cmd.db_path = argv[++i];
} else if (!strcmp(a, "help")) {
printf("Usage: "
"[TD_API_ID=] "
"[TD_API_HASH=] "
"[TD_PATH=] "
"[BOT_TOKEN=] "
"[DB_PATH=] "
"%s "
"[--help] "
"[--logout] "
"[--test] "
"[--api_id TD_API_ID] "
"[--api_hash TD_API_HASH] "
"[--td_path TD_PATH] "
"[--bot_token BOT_TOKEN] "
"[--db_path DB_PATH] ",
argv[0]);
exit(0);
} else {
LOGFEV("Invalid command: %s. Use %s --help to view usage.",
64,
arg,
argv[0]);
}
}
if (!cmd.api_hash && !(cmd.api_hash = getenv("TD_API_HASH")))
LOGFEV("Required environment variable %s is missing.", 64, "TD_API_HASH");
if (!cmd.bot_token && !(cmd.bot_token = getenv("BOT_TOKEN")))
LOGFEV("Required environment variable %s is missing.", 64, "BOT_TOKEN");
if (!cmd.td_path && !(cmd.td_path = getenv("TD_PATH")))
LOGFEV("Required environment variable %s is missing.", 64, "TD_PATH");
if (!cmd.db_path && !(cmd.db_path = getenv("DB_PATH")))
LOGFEV("Required environment variable %s is missing.", 64, "DB_PATH");
if (!cmd.api_id) {
const char *a;
if (!(a = getenv("TD_API_ID")))
LOGFEV("Required environment variable %s is missing.", 64, "TD_API_ID");
parse_api_id(a);
}
}
|
