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
123
124
125
|
#include "common/printing.h"
#include "common/jsonconfig.h"
#include "common/strutil.h"
#include "modules/publicip/publicip.h"
#include "detection/publicip/publicip.h"
#define FF_PUBLICIP_DISPLAY_NAME "Public IP"
bool ffPrintPublicIp(FFPublicIPOptions* options) {
FFPublicIpResult result;
ffStrbufInit(&result.ip);
ffStrbufInit(&result.location);
const char* error = ffDetectPublicIp(options, &result);
if (error) {
ffPrintError(FF_PUBLICIP_DISPLAY_NAME, 0, &options->moduleArgs, FF_PRINT_TYPE_DEFAULT, "%s", error);
return false;
}
if (options->moduleArgs.outputFormat.length == 0) {
ffPrintLogoAndKey(FF_PUBLICIP_DISPLAY_NAME, 0, &options->moduleArgs, FF_PRINT_TYPE_DEFAULT);
if (result.location.length) {
printf("%s (%s)\n", result.ip.chars, result.location.chars);
} else {
ffStrbufPutTo(&result.ip, stdout);
}
} else {
FF_PRINT_FORMAT_CHECKED(FF_PUBLICIP_DISPLAY_NAME, 0, &options->moduleArgs, FF_PRINT_TYPE_DEFAULT, ((FFformatarg[]) {
FF_ARG(result.ip, "ip"),
FF_ARG(result.location, "location"),
}));
}
ffStrbufDestroy(&result.ip);
ffStrbufDestroy(&result.location);
return true;
}
void ffParsePublicIpJsonObject(FFPublicIPOptions* options, yyjson_val* module) {
yyjson_val *key, *val;
size_t idx, max;
yyjson_obj_foreach (module, idx, max, key, val) {
if (ffJsonConfigParseModuleArgs(key, val, &options->moduleArgs)) {
continue;
}
if (unsafe_yyjson_equals_str(key, "url")) {
ffStrbufSetJsonVal(&options->url, val);
continue;
}
if (unsafe_yyjson_equals_str(key, "timeout")) {
options->timeout = (uint32_t) yyjson_get_uint(val);
continue;
}
if (unsafe_yyjson_equals_str(key, "ipv6")) {
options->ipv6 = yyjson_get_bool(val);
continue;
}
ffPrintError(FF_PUBLICIP_MODULE_NAME, 0, &options->moduleArgs, FF_PRINT_TYPE_DEFAULT, "Unknown JSON key %s", unsafe_yyjson_get_str(key));
}
}
void ffGeneratePublicIpJsonConfig(FFPublicIPOptions* options, yyjson_mut_doc* doc, yyjson_mut_val* module) {
ffJsonConfigGenerateModuleArgsConfig(doc, module, &options->moduleArgs);
yyjson_mut_obj_add_strbuf(doc, module, "url", &options->url);
yyjson_mut_obj_add_uint(doc, module, "timeout", options->timeout);
yyjson_mut_obj_add_bool(doc, module, "ipv6", options->ipv6);
}
bool ffGeneratePublicIpJsonResult(FFPublicIPOptions* options, yyjson_mut_doc* doc, yyjson_mut_val* module) {
FFPublicIpResult result;
ffStrbufInit(&result.ip);
ffStrbufInit(&result.location);
const char* error = ffDetectPublicIp(options, &result);
if (error) {
yyjson_mut_obj_add_str(doc, module, "error", error);
return false;
}
yyjson_mut_val* obj = yyjson_mut_obj_add_obj(doc, module, "result");
yyjson_mut_obj_add_strbuf(doc, obj, "ip", &result.ip);
yyjson_mut_obj_add_strbuf(doc, obj, "location", &result.location);
ffStrbufDestroy(&result.ip);
ffStrbufDestroy(&result.location);
return true;
}
void ffInitPublicIpOptions(FFPublicIPOptions* options) {
ffOptionInitModuleArg(&options->moduleArgs, "");
ffStrbufInit(&options->url);
options->timeout = 0;
options->ipv6 = false;
}
void ffDestroyPublicIpOptions(FFPublicIPOptions* options) {
ffOptionDestroyModuleArg(&options->moduleArgs);
ffStrbufDestroy(&options->url);
}
FFModuleBaseInfo ffPublicIPModuleInfo = {
.name = FF_PUBLICIP_MODULE_NAME,
.description = "Print your public IP address and related information",
.initOptions = (void*) ffInitPublicIpOptions,
.destroyOptions = (void*) ffDestroyPublicIpOptions,
.parseJsonObject = (void*) ffParsePublicIpJsonObject,
.printModule = (void*) ffPrintPublicIp,
.generateJsonResult = (void*) ffGeneratePublicIpJsonResult,
.generateJsonConfig = (void*) ffGeneratePublicIpJsonConfig,
.formatArgs = FF_FORMAT_ARG_LIST(((FFModuleFormatArg[]) {
{ "Public IP address", "ip" },
{ "Location", "location" },
}))
};
|
