mirror of
https://github.com/foss42/apidash.git
synced 2025-05-31 22:33:59 +08:00
Added curl rust code generator
This commit is contained in:
144
lib/codegen/rust/curl_rust.dart
Normal file
144
lib/codegen/rust/curl_rust.dart
Normal file
@ -0,0 +1,144 @@
|
||||
import 'package:jinja/jinja.dart' as jj;
|
||||
import 'package:apidash/utils/utils.dart' show getValidRequestUri, requestModelToHARJsonRequest;
|
||||
import 'package:apidash/models/models.dart' show RequestModel;
|
||||
import 'package:apidash/consts.dart';
|
||||
|
||||
class RustCurlCodeGen {
|
||||
final String kTemplateStart = """use curl::easy::Easy;
|
||||
{% if hasJsonBody %}use serde_json::json;
|
||||
{% endif %}{% if hasHeaders %}use curl::easy::List;
|
||||
{% endif %}
|
||||
fn main() {
|
||||
let mut easy = Easy::new();
|
||||
let mut data = Vec::new();
|
||||
|
||||
""";
|
||||
|
||||
String kTemplateUrl = """
|
||||
easy.url("{{url}}").unwrap();
|
||||
""";
|
||||
|
||||
String kTemplateMethod = """
|
||||
{% if method == 'get' or method == 'post' or method == 'put' %}
|
||||
easy.{{ method }}(true).unwrap();
|
||||
{% elif method == 'delete' %}
|
||||
easy.custom_request("DELETE").unwrap();
|
||||
{% elif method == 'patch' %}
|
||||
easy.custom_request("PATCH").unwrap();
|
||||
{% elif method == 'head' %}
|
||||
easy.nobody(true).unwrap();
|
||||
{% endif %}
|
||||
|
||||
""";
|
||||
|
||||
String kTemplateRawBody ="""
|
||||
easy.post_fields_copy(r#"{{body}}"#.as_bytes()).unwrap();
|
||||
|
||||
|
||||
""";
|
||||
|
||||
String kTemplateJsonBody ="""
|
||||
easy.post_fields_copy(json!({{body}}).to_string().as_bytes()).unwrap();
|
||||
|
||||
|
||||
""";
|
||||
|
||||
String kTemplateFormData = """
|
||||
let mut form = curl::easy::Form::new();
|
||||
{% for field in fields %}
|
||||
form.part("{{field.name}}")
|
||||
{% if field.type == "file" %}.file("{{field.value}}"){% else %}.contents(b"{{field.value}}"){% endif %}
|
||||
.add();
|
||||
{% endfor %}
|
||||
easy.httppost(form);
|
||||
""";
|
||||
|
||||
String kTemplateHeader = """
|
||||
{% if headers %}let mut list = List::new();{% for header, value in headers %}
|
||||
list.append("{{header}}: {{value}}");{% endfor %}
|
||||
easy.http_headers(list).unwrap();
|
||||
{% endif %}
|
||||
|
||||
""";
|
||||
|
||||
final String kTemplateEnd = """
|
||||
{
|
||||
let mut transfer = easy.transfer();
|
||||
transfer.write_function(|new_data| {
|
||||
data.extend_from_slice(new_data);
|
||||
Ok(new_data.len())
|
||||
}).unwrap();
|
||||
transfer.perform().unwrap();
|
||||
}
|
||||
|
||||
let response_body = String::from_utf8_lossy(&data);
|
||||
|
||||
println!("Response body:{}", response_body);
|
||||
println!("Response code: {}", easy.response_code().unwrap());
|
||||
}""";
|
||||
|
||||
String? getCode(RequestModel requestModel) {
|
||||
try {
|
||||
String result = "";
|
||||
var requestBody = requestModel.requestBody;
|
||||
|
||||
String url = requestModel.url;
|
||||
|
||||
result += jj.Template(kTemplateStart).render({
|
||||
"hasJsonBody": requestModel.hasJsonData,
|
||||
"hasHeaders": requestModel.enabledRequestHeaders != null || requestModel.hasBody
|
||||
});
|
||||
|
||||
var rec = getValidRequestUri(
|
||||
url,
|
||||
requestModel.enabledRequestParams,
|
||||
);
|
||||
|
||||
Uri? uri = rec.$1;
|
||||
var harJson =
|
||||
requestModelToHARJsonRequest(requestModel, useEnabled: true);
|
||||
|
||||
var templateUrl = jj.Template(kTemplateUrl);
|
||||
result += templateUrl.render({"url": harJson["url"]});
|
||||
|
||||
var methodTemplate = jj.Template(kTemplateMethod);
|
||||
result += methodTemplate.render({"method": requestModel.method.name});
|
||||
|
||||
if (uri != null) {
|
||||
if (requestModel.hasTextData) {
|
||||
var templateBody = jj.Template(kTemplateRawBody);
|
||||
result += templateBody.render({"body": requestBody});
|
||||
} else if (requestModel.hasJsonData) {
|
||||
var templateBody = jj.Template(kTemplateJsonBody);
|
||||
result += templateBody.render({"body": requestBody});
|
||||
} else if (requestModel.hasFormData) {
|
||||
var templateFormData = jj.Template(kTemplateFormData);
|
||||
result += templateFormData.render({
|
||||
"fields": requestModel.formDataMapList,
|
||||
});
|
||||
}
|
||||
|
||||
var headersList = requestModel.enabledRequestHeaders;
|
||||
if (headersList != null || requestModel.hasBody) {
|
||||
var headers = requestModel.enabledHeadersMap;
|
||||
if (requestModel.hasJsonData || requestModel.hasTextData) {
|
||||
headers.putIfAbsent(kHeaderContentType,
|
||||
() => requestModel.requestBodyContentType.header);
|
||||
}
|
||||
if (headers.isNotEmpty) {
|
||||
var templateHeader = jj.Template(kTemplateHeader);
|
||||
result += templateHeader.render({
|
||||
"headers": headers,
|
||||
});
|
||||
}
|
||||
}
|
||||
|
||||
result += kTemplateEnd;
|
||||
}
|
||||
|
||||
return result;
|
||||
} catch (e) {
|
||||
return null;
|
||||
}
|
||||
}
|
||||
}
|
Reference in New Issue
Block a user