Skip to content

Commit 9cdfafb

Browse files
committed
Added code generator for curl-rust
1 parent 9841f48 commit 9cdfafb

File tree

3 files changed

+119
-1
lines changed

3 files changed

+119
-1
lines changed

lib/codegen/codegen.dart

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
1+
import 'package:apidash/codegen/rust/curl-rust.dart';
12
import 'package:apidash/models/models.dart' show RequestModel;
23
import 'package:apidash/consts.dart';
34
import 'dart/http.dart';
@@ -42,6 +43,8 @@ class Codegen {
4243
.getCode(requestModel, defaultUriScheme);
4344
case CodegenLanguage.pythonRequests:
4445
return PythonRequestsCodeGen().getCode(requestModel, defaultUriScheme);
46+
case CodegenLanguage.rustCurl:
47+
return RustCurlCodeGen().getCode(requestModel, defaultUriScheme);
4548
}
4649
}
4750
}

lib/codegen/rust/curl-rust.dart

Lines changed: 113 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,113 @@
1+
import 'dart:convert';
2+
import 'package:jinja/jinja.dart' as jj;
3+
import 'package:apidash/utils/utils.dart'
4+
show getNewUuid, getValidRequestUri, padMultilineString;
5+
import 'package:apidash/models/models.dart' show RequestModel;
6+
import 'package:apidash/consts.dart';
7+
8+
class RustCurlCodeGen {
9+
final String kTemplateStart = """use curl::easy::Easy;
10+
use serde_json::json;
11+
12+
fn main() {
13+
""";
14+
15+
String kTemplateUrl = """
16+
let mut easy = Easy::new();
17+
easy.url("{{url}}").unwrap();
18+
""";
19+
20+
String kTemplateMethod = """easy.{{method}}(true).unwrap();
21+
""";
22+
23+
String kTemplateRawBody =
24+
"""easy.post_field_copy({{body}}.as_bytes()).unwrap();
25+
""";
26+
27+
String kTemplateJsonBody =
28+
"""easy.post_fields_copy(json!({{body}}).to_string().as_bytes()).unwrap();
29+
""";
30+
31+
String kTemplateFormData = """
32+
let mut form = curl::easy::Form::new();
33+
{% for field in fields %}
34+
form.part("{{field.name}}")
35+
{% if field.type == "file" %}.file("{{field.value}}"){% else %}.contents(b"{{field.value}}"){% endif %}
36+
.add();
37+
{% endfor %}
38+
easy.httppost(form);
39+
""";
40+
41+
String kTemplateHeader = """easy.header("{{header}}", "{{value}}");
42+
""";
43+
44+
final String kTemplateEnd = """easy.perform().unwrap();
45+
}
46+
""";
47+
48+
String? getCode(
49+
RequestModel requestModel,
50+
String defaultUriScheme,
51+
) {
52+
try {
53+
String result = "";
54+
55+
String url = requestModel.url;
56+
if (!url.contains("://") && url.isNotEmpty) {
57+
url = "$defaultUriScheme://$url";
58+
}
59+
60+
result += kTemplateStart;
61+
62+
var rec = getValidRequestUri(
63+
url,
64+
requestModel.enabledRequestParams,
65+
);
66+
67+
Uri? uri = rec.$1;
68+
69+
if (uri != null) {
70+
var templateUrl = jj.Template(kTemplateUrl);
71+
result += templateUrl.render({"url": uri.toString()});
72+
73+
var method = requestModel.method;
74+
var templateMethod = jj.Template(kTemplateMethod);
75+
result += templateMethod.render({"method": method.name.toLowerCase()});
76+
77+
var requestBody = requestModel.requestBody;
78+
if (kMethodsWithBody.contains(method) && requestBody != null) {
79+
if (requestModel.requestBodyContentType == ContentType.text) {
80+
var templateRawBody = jj.Template(kTemplateRawBody);
81+
result += templateRawBody.render({"body": requestBody});
82+
} else if (requestModel.requestBodyContentType == ContentType.json) {
83+
var templateJsonBody = jj.Template(kTemplateJsonBody);
84+
result += templateJsonBody.render({"body": requestBody});
85+
} else if (requestModel.isFormDataRequest) {
86+
var templateFormData = jj.Template(kTemplateFormData);
87+
result += templateFormData.render({
88+
"fields": requestModel.formDataMapList,
89+
});
90+
}
91+
}
92+
93+
var headersList = requestModel.enabledRequestHeaders;
94+
if (headersList != null) {
95+
var headers = requestModel.enabledHeadersMap;
96+
for (var header in headers.keys) {
97+
var templateHeader = jj.Template(kTemplateHeader);
98+
result += templateHeader.render({
99+
"header": header,
100+
"value": headers[header],
101+
});
102+
}
103+
}
104+
105+
result += kTemplateEnd;
106+
}
107+
108+
return result;
109+
} catch (e) {
110+
return null;
111+
}
112+
}
113+
}

lib/consts.dart

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -271,7 +271,9 @@ enum CodegenLanguage {
271271
nodejsFetch("node.js (fetch)", "javascript", "js"),
272272
kotlinOkHttp("Kotlin (okhttp3)", "java", "kt"),
273273
pythonHttpClient("Python (http.client)", "python", "py"),
274-
pythonRequests("Python (requests)", "python", "py");
274+
pythonRequests("Python (requests)", "python", "py"),
275+
rustCurl("Rust (curl-rust)", "rust", "rs");
276+
275277

276278
const CodegenLanguage(this.label, this.codeHighlightLang, this.ext);
277279
final String label;

0 commit comments

Comments
 (0)