mirror of
https://github.com/foss42/apidash.git
synced 2025-06-23 08:09:07 +08:00
Merge pull request #190 from Tanish2002/add-rust-ureq-codegen
feat: add codegen for rust(ureq)
This commit is contained in:
@ -8,6 +8,7 @@ import 'kotlin/okhttp.dart';
|
||||
import 'python/http_client.dart';
|
||||
import 'python/requests.dart';
|
||||
import 'rust/actix.dart';
|
||||
import 'rust/ureq.dart';
|
||||
import 'js/axios.dart';
|
||||
import 'js/fetch.dart';
|
||||
import 'others/har.dart';
|
||||
@ -54,6 +55,8 @@ class Codegen {
|
||||
.getCode(rM, boundary: boundary ?? getNewUuid());
|
||||
case CodegenLanguage.pythonRequests:
|
||||
return PythonRequestsCodeGen().getCode(rM, boundary: boundary);
|
||||
case CodegenLanguage.rustUreq:
|
||||
return RustUreqCodeGen().getCode(rM, boundary: boundary);
|
||||
case CodegenLanguage.rustActix:
|
||||
return RustActixCodeGen().getCode(rM, boundary: boundary);
|
||||
case CodegenLanguage.goHttp:
|
||||
|
211
lib/codegen/rust/ureq.dart
Normal file
211
lib/codegen/rust/ureq.dart
Normal file
@ -0,0 +1,211 @@
|
||||
import 'dart:io';
|
||||
import 'dart:convert';
|
||||
import 'package:jinja/jinja.dart' as jj;
|
||||
import 'package:apidash/consts.dart';
|
||||
import 'package:apidash/utils/utils.dart'
|
||||
show getNewUuid, getValidRequestUri, stripUriParams;
|
||||
import 'package:apidash/models/models.dart' show RequestModel;
|
||||
|
||||
class RustUreqCodeGen {
|
||||
final String kTemplateStart = """
|
||||
{%- if isFormDataRequest -%}
|
||||
use std::io::Read;
|
||||
{% endif -%}
|
||||
fn main() -> Result<(), ureq::Error> {
|
||||
let url = "{{url}}";
|
||||
""";
|
||||
|
||||
// String kTemplateParams = """\n .query_pairs({{ params }})""";
|
||||
String kTemplateParams =
|
||||
"""\n {% for key, val in params -%}.query("{{key}}", "{{val}}"){% if not loop.last %}{{ '\n ' }}{% endif %}{%- endfor -%}""";
|
||||
|
||||
String kTemplateBody = """
|
||||
|
||||
let payload = r#"{{body}}"#;
|
||||
|
||||
""";
|
||||
|
||||
String kTemplateJson = """
|
||||
|
||||
let payload = ureq::json!({{body}});
|
||||
|
||||
""";
|
||||
|
||||
String kTemplateHeaders =
|
||||
"""\n {% for key, val in headers -%}.set("{{key}}", "{{val}}"){% if not loop.last %}{{ '\n ' }}{% endif %}{%- endfor -%}""";
|
||||
|
||||
String kTemplateFormHeaderContentType = '''
|
||||
multipart/form-data; boundary={{boundary}}''';
|
||||
|
||||
String kTemplateRequest = """
|
||||
|
||||
let response = ureq::{{method}}(url)
|
||||
""";
|
||||
|
||||
final String kStringFormDataBody = r"""
|
||||
|
||||
struct FormDataItem {
|
||||
name: String,
|
||||
value: String,
|
||||
field_type: String,
|
||||
}
|
||||
|
||||
let form_data_items: Vec<FormDataItem> = vec![
|
||||
{%- for formitem in fields_list %}
|
||||
FormDataItem {
|
||||
{%- for key, val in formitem %}
|
||||
{% if key == "type" %}field_type: "{{ val }}".to_string(),{% else %}{{ key }}: "{{ val }}".to_string(),{% endif %}
|
||||
{%- endfor %}
|
||||
},
|
||||
{%- endfor %}
|
||||
];
|
||||
|
||||
fn build_data_list(fields: Vec<FormDataItem>) -> Vec<u8> {
|
||||
let mut data_list = Vec::new();
|
||||
|
||||
for field in fields {
|
||||
data_list.extend_from_slice(b"--{{boundary}}\r\n");
|
||||
|
||||
if field.field_type == "text" {
|
||||
data_list.extend_from_slice(format!("Content-Disposition: form-data; name=\"{}\"\r\n", field.name).as_bytes());
|
||||
data_list.extend_from_slice(b"Content-Type: text/plain\r\n\r\n");
|
||||
data_list.extend_from_slice(field.value.as_bytes());
|
||||
data_list.extend_from_slice(b"\r\n");
|
||||
} else if field.field_type == "file" {
|
||||
data_list.extend_from_slice(format!("Content-Disposition: form-data; name=\"{}\"; filename=\"{}\"\r\n", field.name, field.value).as_bytes());
|
||||
|
||||
let mime_type = mime_guess::from_path(&field.value).first_or(mime_guess::mime::APPLICATION_OCTET_STREAM);
|
||||
data_list.extend_from_slice(format!("Content-Type: {}\r\n\r\n", mime_type).as_bytes());
|
||||
|
||||
let mut file = std::fs::File::open(&field.value).unwrap();
|
||||
let mut file_contents = Vec::new();
|
||||
file.read_to_end(&mut file_contents).unwrap();
|
||||
data_list.extend_from_slice(&file_contents);
|
||||
data_list.extend_from_slice(b"\r\n");
|
||||
}
|
||||
}
|
||||
|
||||
data_list.extend_from_slice(b"--{{boundary}}--\r\n");
|
||||
data_list
|
||||
}
|
||||
|
||||
let payload = build_data_list(form_data_items);
|
||||
""";
|
||||
|
||||
String kStringRequestBody = """\n .send_string(payload)?;""";
|
||||
|
||||
String kStringRequestForm = """\n .send_bytes(&payload)?;""";
|
||||
|
||||
String kStringRequestJson = """\n .send_json(payload)?;""";
|
||||
|
||||
String kStringRequestNormal = """\n .call()?;""";
|
||||
|
||||
final String kStringRequestEnd = """\n
|
||||
println!("Response Status: {}", response.status());
|
||||
println!("Response: {}", response.into_string()?);
|
||||
|
||||
Ok(())
|
||||
}
|
||||
""";
|
||||
|
||||
String? getCode(
|
||||
RequestModel requestModel, {
|
||||
String? boundary,
|
||||
}) {
|
||||
try {
|
||||
String result = "";
|
||||
bool hasBody = false;
|
||||
bool hasJsonBody = false;
|
||||
String uuid = getNewUuid();
|
||||
|
||||
String url = requestModel.url;
|
||||
|
||||
var rec = getValidRequestUri(
|
||||
url,
|
||||
requestModel.enabledRequestParams,
|
||||
);
|
||||
Uri? uri = rec.$1;
|
||||
if (uri != null) {
|
||||
var templateStartUrl = jj.Template(kTemplateStart);
|
||||
result += templateStartUrl.render({
|
||||
"url": stripUriParams(uri),
|
||||
'isFormDataRequest': requestModel.hasFormData,
|
||||
"method": requestModel.method.name.toLowerCase()
|
||||
});
|
||||
|
||||
var method = requestModel.method;
|
||||
var requestBody = requestModel.requestBody;
|
||||
if (kMethodsWithBody.contains(method) && requestBody != null) {
|
||||
var contentLength = utf8.encode(requestBody).length;
|
||||
if (contentLength > 0) {
|
||||
if (requestModel.requestBodyContentType == ContentType.json) {
|
||||
hasJsonBody = true;
|
||||
var templateBody = jj.Template(kTemplateJson);
|
||||
result += templateBody.render({"body": requestBody});
|
||||
} else if (!requestModel.hasFormData) {
|
||||
hasBody = true;
|
||||
var templateBody = jj.Template(kTemplateBody);
|
||||
result += templateBody.render({"body": requestBody});
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
if (requestModel.hasFormData) {
|
||||
var formDataBodyData = jj.Template(kStringFormDataBody);
|
||||
result += formDataBodyData.render(
|
||||
{
|
||||
"fields_list": requestModel.formDataMapList,
|
||||
"boundary": boundary ?? uuid,
|
||||
},
|
||||
);
|
||||
}
|
||||
var templateRequest = jj.Template(kTemplateRequest);
|
||||
result += templateRequest.render({
|
||||
"method": method.name.toLowerCase(),
|
||||
});
|
||||
|
||||
if (uri.hasQuery) {
|
||||
var params = uri.queryParameters;
|
||||
if (params.isNotEmpty) {
|
||||
var templateParms = jj.Template(kTemplateParams);
|
||||
result += templateParms.render({"params": params});
|
||||
}
|
||||
}
|
||||
|
||||
var headersList = requestModel.enabledRequestHeaders;
|
||||
if (headersList != null || hasBody || requestModel.hasFormData) {
|
||||
var headers = requestModel.enabledHeadersMap;
|
||||
if (requestModel.hasFormData) {
|
||||
var formHeaderTemplate =
|
||||
jj.Template(kTemplateFormHeaderContentType);
|
||||
headers[HttpHeaders.contentTypeHeader] = formHeaderTemplate.render({
|
||||
"boundary": boundary ?? uuid,
|
||||
});
|
||||
} else if (hasBody) {
|
||||
headers[HttpHeaders.contentTypeHeader] =
|
||||
requestModel.requestBodyContentType.header;
|
||||
}
|
||||
|
||||
if (headers.isNotEmpty) {
|
||||
var templateHeaders = jj.Template(kTemplateHeaders);
|
||||
result += templateHeaders.render({"headers": headers});
|
||||
}
|
||||
}
|
||||
if (requestModel.hasFormData) {
|
||||
result += kStringRequestForm;
|
||||
} else if (hasBody) {
|
||||
result += kStringRequestBody;
|
||||
} else if (hasJsonBody) {
|
||||
result += kStringRequestJson;
|
||||
} else {
|
||||
result += kStringRequestNormal;
|
||||
}
|
||||
|
||||
result += kStringRequestEnd;
|
||||
}
|
||||
return result;
|
||||
} catch (e) {
|
||||
return null;
|
||||
}
|
||||
}
|
||||
}
|
@ -274,7 +274,8 @@ enum CodegenLanguage {
|
||||
kotlinOkHttp("Kotlin (okhttp3)", "java", "kt"),
|
||||
rustActix("Rust (Actix Client)", "rust", "rs"),
|
||||
pythonHttpClient("Python (http.client)", "python", "py"),
|
||||
pythonRequests("Python (requests)", "python", "py");
|
||||
pythonRequests("Python (requests)", "python", "py"),
|
||||
rustUreq("Rust (ureq)", "rust", "rs");
|
||||
|
||||
const CodegenLanguage(this.label, this.codeHighlightLang, this.ext);
|
||||
final String label;
|
||||
|
871
test/codegen/rust_ureq_codegen_test.dart
Normal file
871
test/codegen/rust_ureq_codegen_test.dart
Normal file
@ -0,0 +1,871 @@
|
||||
import 'package:apidash/codegen/codegen.dart';
|
||||
import 'package:apidash/consts.dart';
|
||||
import 'package:test/test.dart';
|
||||
import '../request_models.dart';
|
||||
|
||||
void main() {
|
||||
final codeGen = Codegen();
|
||||
|
||||
group('GET Request', () {
|
||||
test('GET 1', () {
|
||||
const expectedCode = r"""fn main() -> Result<(), ureq::Error> {
|
||||
let url = "https://api.apidash.dev";
|
||||
let response = ureq::get(url)
|
||||
.call()?;
|
||||
|
||||
println!("Response Status: {}", response.status());
|
||||
println!("Response: {}", response.into_string()?);
|
||||
|
||||
Ok(())
|
||||
}
|
||||
""";
|
||||
expect(
|
||||
codeGen.getCode(CodegenLanguage.rustUreq, requestModelGet1, "https"),
|
||||
expectedCode);
|
||||
});
|
||||
|
||||
test('GET 2', () {
|
||||
const expectedCode = r"""fn main() -> Result<(), ureq::Error> {
|
||||
let url = "https://api.apidash.dev/country/data";
|
||||
let response = ureq::get(url)
|
||||
.query("code", "US")
|
||||
.call()?;
|
||||
|
||||
println!("Response Status: {}", response.status());
|
||||
println!("Response: {}", response.into_string()?);
|
||||
|
||||
Ok(())
|
||||
}
|
||||
""";
|
||||
expect(
|
||||
codeGen.getCode(CodegenLanguage.rustUreq, requestModelGet2, "https"),
|
||||
expectedCode);
|
||||
});
|
||||
|
||||
test('GET 3', () {
|
||||
const expectedCode = r"""fn main() -> Result<(), ureq::Error> {
|
||||
let url = "https://api.apidash.dev/country/data";
|
||||
let response = ureq::get(url)
|
||||
.query("code", "IND")
|
||||
.call()?;
|
||||
|
||||
println!("Response Status: {}", response.status());
|
||||
println!("Response: {}", response.into_string()?);
|
||||
|
||||
Ok(())
|
||||
}
|
||||
""";
|
||||
expect(
|
||||
codeGen.getCode(CodegenLanguage.rustUreq, requestModelGet3, "https"),
|
||||
expectedCode);
|
||||
});
|
||||
|
||||
test('GET 4', () {
|
||||
const expectedCode = r"""fn main() -> Result<(), ureq::Error> {
|
||||
let url = "https://api.apidash.dev/humanize/social";
|
||||
let response = ureq::get(url)
|
||||
.query("num", "8700000")
|
||||
.query("digits", "3")
|
||||
.query("system", "SS")
|
||||
.query("add_space", "true")
|
||||
.query("trailing_zeros", "true")
|
||||
.call()?;
|
||||
|
||||
println!("Response Status: {}", response.status());
|
||||
println!("Response: {}", response.into_string()?);
|
||||
|
||||
Ok(())
|
||||
}
|
||||
""";
|
||||
expect(
|
||||
codeGen.getCode(CodegenLanguage.rustUreq, requestModelGet4, "https"),
|
||||
expectedCode);
|
||||
});
|
||||
|
||||
test('GET 5', () {
|
||||
const expectedCode = r"""fn main() -> Result<(), ureq::Error> {
|
||||
let url = "https://api.github.com/repos/foss42/apidash";
|
||||
let response = ureq::get(url)
|
||||
.set("User-Agent", "Test Agent")
|
||||
.call()?;
|
||||
|
||||
println!("Response Status: {}", response.status());
|
||||
println!("Response: {}", response.into_string()?);
|
||||
|
||||
Ok(())
|
||||
}
|
||||
""";
|
||||
expect(
|
||||
codeGen.getCode(CodegenLanguage.rustUreq, requestModelGet5, "https"),
|
||||
expectedCode);
|
||||
});
|
||||
|
||||
test('GET 6', () {
|
||||
const expectedCode = r"""fn main() -> Result<(), ureq::Error> {
|
||||
let url = "https://api.github.com/repos/foss42/apidash";
|
||||
let response = ureq::get(url)
|
||||
.query("raw", "true")
|
||||
.set("User-Agent", "Test Agent")
|
||||
.call()?;
|
||||
|
||||
println!("Response Status: {}", response.status());
|
||||
println!("Response: {}", response.into_string()?);
|
||||
|
||||
Ok(())
|
||||
}
|
||||
""";
|
||||
expect(
|
||||
codeGen.getCode(CodegenLanguage.rustUreq, requestModelGet6, "https"),
|
||||
expectedCode);
|
||||
});
|
||||
|
||||
test('GET 7', () {
|
||||
const expectedCode = r"""fn main() -> Result<(), ureq::Error> {
|
||||
let url = "https://api.apidash.dev";
|
||||
let response = ureq::get(url)
|
||||
.call()?;
|
||||
|
||||
println!("Response Status: {}", response.status());
|
||||
println!("Response: {}", response.into_string()?);
|
||||
|
||||
Ok(())
|
||||
}
|
||||
""";
|
||||
expect(
|
||||
codeGen.getCode(CodegenLanguage.rustUreq, requestModelGet7, "https"),
|
||||
expectedCode);
|
||||
});
|
||||
|
||||
test('GET 8', () {
|
||||
const expectedCode = r"""fn main() -> Result<(), ureq::Error> {
|
||||
let url = "https://api.github.com/repos/foss42/apidash";
|
||||
let response = ureq::get(url)
|
||||
.query("raw", "true")
|
||||
.set("User-Agent", "Test Agent")
|
||||
.call()?;
|
||||
|
||||
println!("Response Status: {}", response.status());
|
||||
println!("Response: {}", response.into_string()?);
|
||||
|
||||
Ok(())
|
||||
}
|
||||
""";
|
||||
expect(
|
||||
codeGen.getCode(CodegenLanguage.rustUreq, requestModelGet8, "https"),
|
||||
expectedCode);
|
||||
});
|
||||
|
||||
test('GET 9', () {
|
||||
const expectedCode = r"""fn main() -> Result<(), ureq::Error> {
|
||||
let url = "https://api.apidash.dev/humanize/social";
|
||||
let response = ureq::get(url)
|
||||
.query("num", "8700000")
|
||||
.query("add_space", "true")
|
||||
.call()?;
|
||||
|
||||
println!("Response Status: {}", response.status());
|
||||
println!("Response: {}", response.into_string()?);
|
||||
|
||||
Ok(())
|
||||
}
|
||||
""";
|
||||
expect(
|
||||
codeGen.getCode(CodegenLanguage.rustUreq, requestModelGet9, "https"),
|
||||
expectedCode);
|
||||
});
|
||||
|
||||
test('GET 10', () {
|
||||
const expectedCode = r"""fn main() -> Result<(), ureq::Error> {
|
||||
let url = "https://api.apidash.dev/humanize/social";
|
||||
let response = ureq::get(url)
|
||||
.set("User-Agent", "Test Agent")
|
||||
.call()?;
|
||||
|
||||
println!("Response Status: {}", response.status());
|
||||
println!("Response: {}", response.into_string()?);
|
||||
|
||||
Ok(())
|
||||
}
|
||||
""";
|
||||
expect(
|
||||
codeGen.getCode(
|
||||
CodegenLanguage.rustUreq,
|
||||
requestModelGet10,
|
||||
"https",
|
||||
),
|
||||
expectedCode);
|
||||
});
|
||||
|
||||
test('GET 11', () {
|
||||
const expectedCode = r"""fn main() -> Result<(), ureq::Error> {
|
||||
let url = "https://api.apidash.dev/humanize/social";
|
||||
let response = ureq::get(url)
|
||||
.query("num", "8700000")
|
||||
.query("digits", "3")
|
||||
.set("User-Agent", "Test Agent")
|
||||
.call()?;
|
||||
|
||||
println!("Response Status: {}", response.status());
|
||||
println!("Response: {}", response.into_string()?);
|
||||
|
||||
Ok(())
|
||||
}
|
||||
""";
|
||||
expect(
|
||||
codeGen.getCode(CodegenLanguage.rustUreq, requestModelGet11, "https"),
|
||||
expectedCode);
|
||||
});
|
||||
|
||||
test('GET 12', () {
|
||||
const expectedCode = r"""fn main() -> Result<(), ureq::Error> {
|
||||
let url = "https://api.apidash.dev/humanize/social";
|
||||
let response = ureq::get(url)
|
||||
.call()?;
|
||||
|
||||
println!("Response Status: {}", response.status());
|
||||
println!("Response: {}", response.into_string()?);
|
||||
|
||||
Ok(())
|
||||
}
|
||||
""";
|
||||
expect(
|
||||
codeGen.getCode(CodegenLanguage.rustUreq, requestModelGet12, "https"),
|
||||
expectedCode);
|
||||
});
|
||||
});
|
||||
|
||||
group('HEAD Request', () {
|
||||
test('HEAD 1', () {
|
||||
const expectedCode = r"""fn main() -> Result<(), ureq::Error> {
|
||||
let url = "https://api.apidash.dev";
|
||||
let response = ureq::head(url)
|
||||
.call()?;
|
||||
|
||||
println!("Response Status: {}", response.status());
|
||||
println!("Response: {}", response.into_string()?);
|
||||
|
||||
Ok(())
|
||||
}
|
||||
""";
|
||||
expect(
|
||||
codeGen.getCode(CodegenLanguage.rustUreq, requestModelHead1, "https"),
|
||||
expectedCode);
|
||||
});
|
||||
|
||||
test('HEAD 2', () {
|
||||
const expectedCode = r"""fn main() -> Result<(), ureq::Error> {
|
||||
let url = "http://api.apidash.dev";
|
||||
let response = ureq::head(url)
|
||||
.call()?;
|
||||
|
||||
println!("Response Status: {}", response.status());
|
||||
println!("Response: {}", response.into_string()?);
|
||||
|
||||
Ok(())
|
||||
}
|
||||
""";
|
||||
expect(
|
||||
codeGen.getCode(CodegenLanguage.rustUreq, requestModelHead2, "http"),
|
||||
expectedCode);
|
||||
});
|
||||
});
|
||||
|
||||
group('POST Request', () {
|
||||
test('POST 1', () {
|
||||
const expectedCode = r'''fn main() -> Result<(), ureq::Error> {
|
||||
let url = "https://api.apidash.dev/case/lower";
|
||||
let payload = r#"{
|
||||
"text": "I LOVE Flutter"
|
||||
}"#;
|
||||
|
||||
let response = ureq::post(url)
|
||||
.set("content-type", "text/plain")
|
||||
.send_string(payload)?;
|
||||
|
||||
println!("Response Status: {}", response.status());
|
||||
println!("Response: {}", response.into_string()?);
|
||||
|
||||
Ok(())
|
||||
}
|
||||
''';
|
||||
expect(
|
||||
codeGen.getCode(CodegenLanguage.rustUreq, requestModelPost1, "https"),
|
||||
expectedCode);
|
||||
});
|
||||
|
||||
test('POST 2', () {
|
||||
const expectedCode = r'''fn main() -> Result<(), ureq::Error> {
|
||||
let url = "https://api.apidash.dev/case/lower";
|
||||
let payload = ureq::json!({
|
||||
"text": "I LOVE Flutter",
|
||||
"flag": null,
|
||||
"male": true,
|
||||
"female": false,
|
||||
"no": 1.2,
|
||||
"arr": ["null", "true", "false", null]
|
||||
});
|
||||
|
||||
let response = ureq::post(url)
|
||||
.send_json(payload)?;
|
||||
|
||||
println!("Response Status: {}", response.status());
|
||||
println!("Response: {}", response.into_string()?);
|
||||
|
||||
Ok(())
|
||||
}
|
||||
''';
|
||||
expect(
|
||||
codeGen.getCode(CodegenLanguage.rustUreq, requestModelPost2, "https"),
|
||||
expectedCode);
|
||||
});
|
||||
|
||||
test('POST 3', () {
|
||||
const expectedCode = r'''fn main() -> Result<(), ureq::Error> {
|
||||
let url = "https://api.apidash.dev/case/lower";
|
||||
let payload = ureq::json!({
|
||||
"text": "I LOVE Flutter"
|
||||
});
|
||||
|
||||
let response = ureq::post(url)
|
||||
.set("User-Agent", "Test Agent")
|
||||
.send_json(payload)?;
|
||||
|
||||
println!("Response Status: {}", response.status());
|
||||
println!("Response: {}", response.into_string()?);
|
||||
|
||||
Ok(())
|
||||
}
|
||||
''';
|
||||
expect(
|
||||
codeGen.getCode(CodegenLanguage.rustUreq, requestModelPost3, "https"),
|
||||
expectedCode);
|
||||
});
|
||||
|
||||
test('POST 4', () {
|
||||
const expectedCode = r"""use std::io::Read;
|
||||
fn main() -> Result<(), ureq::Error> {
|
||||
let url = "https://api.apidash.dev/io/form";
|
||||
struct FormDataItem {
|
||||
name: String,
|
||||
value: String,
|
||||
field_type: String,
|
||||
}
|
||||
|
||||
let form_data_items: Vec<FormDataItem> = vec![
|
||||
FormDataItem {
|
||||
name: "text".to_string(),
|
||||
value: "API".to_string(),
|
||||
field_type: "text".to_string(),
|
||||
},
|
||||
FormDataItem {
|
||||
name: "sep".to_string(),
|
||||
value: "|".to_string(),
|
||||
field_type: "text".to_string(),
|
||||
},
|
||||
FormDataItem {
|
||||
name: "times".to_string(),
|
||||
value: "3".to_string(),
|
||||
field_type: "text".to_string(),
|
||||
},
|
||||
];
|
||||
|
||||
fn build_data_list(fields: Vec<FormDataItem>) -> Vec<u8> {
|
||||
let mut data_list = Vec::new();
|
||||
|
||||
for field in fields {
|
||||
data_list.extend_from_slice(b"--test\r\n");
|
||||
|
||||
if field.field_type == "text" {
|
||||
data_list.extend_from_slice(format!("Content-Disposition: form-data; name=\"{}\"\r\n", field.name).as_bytes());
|
||||
data_list.extend_from_slice(b"Content-Type: text/plain\r\n\r\n");
|
||||
data_list.extend_from_slice(field.value.as_bytes());
|
||||
data_list.extend_from_slice(b"\r\n");
|
||||
} else if field.field_type == "file" {
|
||||
data_list.extend_from_slice(format!("Content-Disposition: form-data; name=\"{}\"; filename=\"{}\"\r\n", field.name, field.value).as_bytes());
|
||||
|
||||
let mime_type = mime_guess::from_path(&field.value).first_or(mime_guess::mime::APPLICATION_OCTET_STREAM);
|
||||
data_list.extend_from_slice(format!("Content-Type: {}\r\n\r\n", mime_type).as_bytes());
|
||||
|
||||
let mut file = std::fs::File::open(&field.value).unwrap();
|
||||
let mut file_contents = Vec::new();
|
||||
file.read_to_end(&mut file_contents).unwrap();
|
||||
data_list.extend_from_slice(&file_contents);
|
||||
data_list.extend_from_slice(b"\r\n");
|
||||
}
|
||||
}
|
||||
|
||||
data_list.extend_from_slice(b"--test--\r\n");
|
||||
data_list
|
||||
}
|
||||
|
||||
let payload = build_data_list(form_data_items);
|
||||
let response = ureq::post(url)
|
||||
.set("content-type", "multipart/form-data; boundary=test")
|
||||
.send_bytes(&payload)?;
|
||||
|
||||
println!("Response Status: {}", response.status());
|
||||
println!("Response: {}", response.into_string()?);
|
||||
|
||||
Ok(())
|
||||
}
|
||||
""";
|
||||
expect(
|
||||
codeGen.getCode(
|
||||
CodegenLanguage.rustUreq,
|
||||
requestModelPost4,
|
||||
"https",
|
||||
boundary: "test",
|
||||
),
|
||||
expectedCode);
|
||||
});
|
||||
test('POST 5', () {
|
||||
const expectedCode = r"""use std::io::Read;
|
||||
fn main() -> Result<(), ureq::Error> {
|
||||
let url = "https://api.apidash.dev/io/form";
|
||||
struct FormDataItem {
|
||||
name: String,
|
||||
value: String,
|
||||
field_type: String,
|
||||
}
|
||||
|
||||
let form_data_items: Vec<FormDataItem> = vec![
|
||||
FormDataItem {
|
||||
name: "text".to_string(),
|
||||
value: "API".to_string(),
|
||||
field_type: "text".to_string(),
|
||||
},
|
||||
FormDataItem {
|
||||
name: "sep".to_string(),
|
||||
value: "|".to_string(),
|
||||
field_type: "text".to_string(),
|
||||
},
|
||||
FormDataItem {
|
||||
name: "times".to_string(),
|
||||
value: "3".to_string(),
|
||||
field_type: "text".to_string(),
|
||||
},
|
||||
];
|
||||
|
||||
fn build_data_list(fields: Vec<FormDataItem>) -> Vec<u8> {
|
||||
let mut data_list = Vec::new();
|
||||
|
||||
for field in fields {
|
||||
data_list.extend_from_slice(b"--test\r\n");
|
||||
|
||||
if field.field_type == "text" {
|
||||
data_list.extend_from_slice(format!("Content-Disposition: form-data; name=\"{}\"\r\n", field.name).as_bytes());
|
||||
data_list.extend_from_slice(b"Content-Type: text/plain\r\n\r\n");
|
||||
data_list.extend_from_slice(field.value.as_bytes());
|
||||
data_list.extend_from_slice(b"\r\n");
|
||||
} else if field.field_type == "file" {
|
||||
data_list.extend_from_slice(format!("Content-Disposition: form-data; name=\"{}\"; filename=\"{}\"\r\n", field.name, field.value).as_bytes());
|
||||
|
||||
let mime_type = mime_guess::from_path(&field.value).first_or(mime_guess::mime::APPLICATION_OCTET_STREAM);
|
||||
data_list.extend_from_slice(format!("Content-Type: {}\r\n\r\n", mime_type).as_bytes());
|
||||
|
||||
let mut file = std::fs::File::open(&field.value).unwrap();
|
||||
let mut file_contents = Vec::new();
|
||||
file.read_to_end(&mut file_contents).unwrap();
|
||||
data_list.extend_from_slice(&file_contents);
|
||||
data_list.extend_from_slice(b"\r\n");
|
||||
}
|
||||
}
|
||||
|
||||
data_list.extend_from_slice(b"--test--\r\n");
|
||||
data_list
|
||||
}
|
||||
|
||||
let payload = build_data_list(form_data_items);
|
||||
let response = ureq::post(url)
|
||||
.set("User-Agent", "Test Agent")
|
||||
.set("content-type", "multipart/form-data; boundary=test")
|
||||
.send_bytes(&payload)?;
|
||||
|
||||
println!("Response Status: {}", response.status());
|
||||
println!("Response: {}", response.into_string()?);
|
||||
|
||||
Ok(())
|
||||
}
|
||||
""";
|
||||
expect(
|
||||
codeGen.getCode(CodegenLanguage.rustUreq, requestModelPost5, "https",
|
||||
boundary: "test"),
|
||||
expectedCode);
|
||||
});
|
||||
test('POST 6', () {
|
||||
const expectedCode = r"""use std::io::Read;
|
||||
fn main() -> Result<(), ureq::Error> {
|
||||
let url = "https://api.apidash.dev/io/img";
|
||||
struct FormDataItem {
|
||||
name: String,
|
||||
value: String,
|
||||
field_type: String,
|
||||
}
|
||||
|
||||
let form_data_items: Vec<FormDataItem> = vec![
|
||||
FormDataItem {
|
||||
name: "token".to_string(),
|
||||
value: "xyz".to_string(),
|
||||
field_type: "text".to_string(),
|
||||
},
|
||||
FormDataItem {
|
||||
name: "imfile".to_string(),
|
||||
value: "/Documents/up/1.png".to_string(),
|
||||
field_type: "file".to_string(),
|
||||
},
|
||||
];
|
||||
|
||||
fn build_data_list(fields: Vec<FormDataItem>) -> Vec<u8> {
|
||||
let mut data_list = Vec::new();
|
||||
|
||||
for field in fields {
|
||||
data_list.extend_from_slice(b"--test\r\n");
|
||||
|
||||
if field.field_type == "text" {
|
||||
data_list.extend_from_slice(format!("Content-Disposition: form-data; name=\"{}\"\r\n", field.name).as_bytes());
|
||||
data_list.extend_from_slice(b"Content-Type: text/plain\r\n\r\n");
|
||||
data_list.extend_from_slice(field.value.as_bytes());
|
||||
data_list.extend_from_slice(b"\r\n");
|
||||
} else if field.field_type == "file" {
|
||||
data_list.extend_from_slice(format!("Content-Disposition: form-data; name=\"{}\"; filename=\"{}\"\r\n", field.name, field.value).as_bytes());
|
||||
|
||||
let mime_type = mime_guess::from_path(&field.value).first_or(mime_guess::mime::APPLICATION_OCTET_STREAM);
|
||||
data_list.extend_from_slice(format!("Content-Type: {}\r\n\r\n", mime_type).as_bytes());
|
||||
|
||||
let mut file = std::fs::File::open(&field.value).unwrap();
|
||||
let mut file_contents = Vec::new();
|
||||
file.read_to_end(&mut file_contents).unwrap();
|
||||
data_list.extend_from_slice(&file_contents);
|
||||
data_list.extend_from_slice(b"\r\n");
|
||||
}
|
||||
}
|
||||
|
||||
data_list.extend_from_slice(b"--test--\r\n");
|
||||
data_list
|
||||
}
|
||||
|
||||
let payload = build_data_list(form_data_items);
|
||||
let response = ureq::post(url)
|
||||
.set("content-type", "multipart/form-data; boundary=test")
|
||||
.send_bytes(&payload)?;
|
||||
|
||||
println!("Response Status: {}", response.status());
|
||||
println!("Response: {}", response.into_string()?);
|
||||
|
||||
Ok(())
|
||||
}
|
||||
""";
|
||||
expect(
|
||||
codeGen.getCode(CodegenLanguage.rustUreq, requestModelPost6, "https",
|
||||
boundary: "test"),
|
||||
expectedCode);
|
||||
});
|
||||
test('POST 7', () {
|
||||
const expectedCode = r"""use std::io::Read;
|
||||
fn main() -> Result<(), ureq::Error> {
|
||||
let url = "https://api.apidash.dev/io/img";
|
||||
struct FormDataItem {
|
||||
name: String,
|
||||
value: String,
|
||||
field_type: String,
|
||||
}
|
||||
|
||||
let form_data_items: Vec<FormDataItem> = vec![
|
||||
FormDataItem {
|
||||
name: "token".to_string(),
|
||||
value: "xyz".to_string(),
|
||||
field_type: "text".to_string(),
|
||||
},
|
||||
FormDataItem {
|
||||
name: "imfile".to_string(),
|
||||
value: "/Documents/up/1.png".to_string(),
|
||||
field_type: "file".to_string(),
|
||||
},
|
||||
];
|
||||
|
||||
fn build_data_list(fields: Vec<FormDataItem>) -> Vec<u8> {
|
||||
let mut data_list = Vec::new();
|
||||
|
||||
for field in fields {
|
||||
data_list.extend_from_slice(b"--test\r\n");
|
||||
|
||||
if field.field_type == "text" {
|
||||
data_list.extend_from_slice(format!("Content-Disposition: form-data; name=\"{}\"\r\n", field.name).as_bytes());
|
||||
data_list.extend_from_slice(b"Content-Type: text/plain\r\n\r\n");
|
||||
data_list.extend_from_slice(field.value.as_bytes());
|
||||
data_list.extend_from_slice(b"\r\n");
|
||||
} else if field.field_type == "file" {
|
||||
data_list.extend_from_slice(format!("Content-Disposition: form-data; name=\"{}\"; filename=\"{}\"\r\n", field.name, field.value).as_bytes());
|
||||
|
||||
let mime_type = mime_guess::from_path(&field.value).first_or(mime_guess::mime::APPLICATION_OCTET_STREAM);
|
||||
data_list.extend_from_slice(format!("Content-Type: {}\r\n\r\n", mime_type).as_bytes());
|
||||
|
||||
let mut file = std::fs::File::open(&field.value).unwrap();
|
||||
let mut file_contents = Vec::new();
|
||||
file.read_to_end(&mut file_contents).unwrap();
|
||||
data_list.extend_from_slice(&file_contents);
|
||||
data_list.extend_from_slice(b"\r\n");
|
||||
}
|
||||
}
|
||||
|
||||
data_list.extend_from_slice(b"--test--\r\n");
|
||||
data_list
|
||||
}
|
||||
|
||||
let payload = build_data_list(form_data_items);
|
||||
let response = ureq::post(url)
|
||||
.set("content-type", "multipart/form-data; boundary=test")
|
||||
.send_bytes(&payload)?;
|
||||
|
||||
println!("Response Status: {}", response.status());
|
||||
println!("Response: {}", response.into_string()?);
|
||||
|
||||
Ok(())
|
||||
}
|
||||
""";
|
||||
expect(
|
||||
codeGen.getCode(CodegenLanguage.rustUreq, requestModelPost7, "https",
|
||||
boundary: "test"),
|
||||
expectedCode);
|
||||
});
|
||||
test('POST 8', () {
|
||||
const expectedCode = r"""use std::io::Read;
|
||||
fn main() -> Result<(), ureq::Error> {
|
||||
let url = "https://api.apidash.dev/io/form";
|
||||
struct FormDataItem {
|
||||
name: String,
|
||||
value: String,
|
||||
field_type: String,
|
||||
}
|
||||
|
||||
let form_data_items: Vec<FormDataItem> = vec![
|
||||
FormDataItem {
|
||||
name: "text".to_string(),
|
||||
value: "API".to_string(),
|
||||
field_type: "text".to_string(),
|
||||
},
|
||||
FormDataItem {
|
||||
name: "sep".to_string(),
|
||||
value: "|".to_string(),
|
||||
field_type: "text".to_string(),
|
||||
},
|
||||
FormDataItem {
|
||||
name: "times".to_string(),
|
||||
value: "3".to_string(),
|
||||
field_type: "text".to_string(),
|
||||
},
|
||||
];
|
||||
|
||||
fn build_data_list(fields: Vec<FormDataItem>) -> Vec<u8> {
|
||||
let mut data_list = Vec::new();
|
||||
|
||||
for field in fields {
|
||||
data_list.extend_from_slice(b"--test\r\n");
|
||||
|
||||
if field.field_type == "text" {
|
||||
data_list.extend_from_slice(format!("Content-Disposition: form-data; name=\"{}\"\r\n", field.name).as_bytes());
|
||||
data_list.extend_from_slice(b"Content-Type: text/plain\r\n\r\n");
|
||||
data_list.extend_from_slice(field.value.as_bytes());
|
||||
data_list.extend_from_slice(b"\r\n");
|
||||
} else if field.field_type == "file" {
|
||||
data_list.extend_from_slice(format!("Content-Disposition: form-data; name=\"{}\"; filename=\"{}\"\r\n", field.name, field.value).as_bytes());
|
||||
|
||||
let mime_type = mime_guess::from_path(&field.value).first_or(mime_guess::mime::APPLICATION_OCTET_STREAM);
|
||||
data_list.extend_from_slice(format!("Content-Type: {}\r\n\r\n", mime_type).as_bytes());
|
||||
|
||||
let mut file = std::fs::File::open(&field.value).unwrap();
|
||||
let mut file_contents = Vec::new();
|
||||
file.read_to_end(&mut file_contents).unwrap();
|
||||
data_list.extend_from_slice(&file_contents);
|
||||
data_list.extend_from_slice(b"\r\n");
|
||||
}
|
||||
}
|
||||
|
||||
data_list.extend_from_slice(b"--test--\r\n");
|
||||
data_list
|
||||
}
|
||||
|
||||
let payload = build_data_list(form_data_items);
|
||||
let response = ureq::post(url)
|
||||
.query("size", "2")
|
||||
.query("len", "3")
|
||||
.set("content-type", "multipart/form-data; boundary=test")
|
||||
.send_bytes(&payload)?;
|
||||
|
||||
println!("Response Status: {}", response.status());
|
||||
println!("Response: {}", response.into_string()?);
|
||||
|
||||
Ok(())
|
||||
}
|
||||
""";
|
||||
expect(
|
||||
codeGen.getCode(CodegenLanguage.rustUreq, requestModelPost8, "https",
|
||||
boundary: "test"),
|
||||
expectedCode);
|
||||
});
|
||||
test('POST 9', () {
|
||||
const expectedCode = r"""use std::io::Read;
|
||||
fn main() -> Result<(), ureq::Error> {
|
||||
let url = "https://api.apidash.dev/io/img";
|
||||
struct FormDataItem {
|
||||
name: String,
|
||||
value: String,
|
||||
field_type: String,
|
||||
}
|
||||
|
||||
let form_data_items: Vec<FormDataItem> = vec![
|
||||
FormDataItem {
|
||||
name: "token".to_string(),
|
||||
value: "xyz".to_string(),
|
||||
field_type: "text".to_string(),
|
||||
},
|
||||
FormDataItem {
|
||||
name: "imfile".to_string(),
|
||||
value: "/Documents/up/1.png".to_string(),
|
||||
field_type: "file".to_string(),
|
||||
},
|
||||
];
|
||||
|
||||
fn build_data_list(fields: Vec<FormDataItem>) -> Vec<u8> {
|
||||
let mut data_list = Vec::new();
|
||||
|
||||
for field in fields {
|
||||
data_list.extend_from_slice(b"--test\r\n");
|
||||
|
||||
if field.field_type == "text" {
|
||||
data_list.extend_from_slice(format!("Content-Disposition: form-data; name=\"{}\"\r\n", field.name).as_bytes());
|
||||
data_list.extend_from_slice(b"Content-Type: text/plain\r\n\r\n");
|
||||
data_list.extend_from_slice(field.value.as_bytes());
|
||||
data_list.extend_from_slice(b"\r\n");
|
||||
} else if field.field_type == "file" {
|
||||
data_list.extend_from_slice(format!("Content-Disposition: form-data; name=\"{}\"; filename=\"{}\"\r\n", field.name, field.value).as_bytes());
|
||||
|
||||
let mime_type = mime_guess::from_path(&field.value).first_or(mime_guess::mime::APPLICATION_OCTET_STREAM);
|
||||
data_list.extend_from_slice(format!("Content-Type: {}\r\n\r\n", mime_type).as_bytes());
|
||||
|
||||
let mut file = std::fs::File::open(&field.value).unwrap();
|
||||
let mut file_contents = Vec::new();
|
||||
file.read_to_end(&mut file_contents).unwrap();
|
||||
data_list.extend_from_slice(&file_contents);
|
||||
data_list.extend_from_slice(b"\r\n");
|
||||
}
|
||||
}
|
||||
|
||||
data_list.extend_from_slice(b"--test--\r\n");
|
||||
data_list
|
||||
}
|
||||
|
||||
let payload = build_data_list(form_data_items);
|
||||
let response = ureq::post(url)
|
||||
.query("size", "2")
|
||||
.query("len", "3")
|
||||
.set("User-Agent", "Test Agent")
|
||||
.set("Keep-Alive", "true")
|
||||
.set("content-type", "multipart/form-data; boundary=test")
|
||||
.send_bytes(&payload)?;
|
||||
|
||||
println!("Response Status: {}", response.status());
|
||||
println!("Response: {}", response.into_string()?);
|
||||
|
||||
Ok(())
|
||||
}
|
||||
""";
|
||||
expect(
|
||||
codeGen.getCode(CodegenLanguage.rustUreq, requestModelPost9, "https",
|
||||
boundary: "test"),
|
||||
expectedCode);
|
||||
});
|
||||
});
|
||||
|
||||
group('PUT Request', () {
|
||||
test('PUT 1', () {
|
||||
const expectedCode = r'''fn main() -> Result<(), ureq::Error> {
|
||||
let url = "https://reqres.in/api/users/2";
|
||||
let payload = ureq::json!({
|
||||
"name": "morpheus",
|
||||
"job": "zion resident"
|
||||
});
|
||||
|
||||
let response = ureq::put(url)
|
||||
.send_json(payload)?;
|
||||
|
||||
println!("Response Status: {}", response.status());
|
||||
println!("Response: {}", response.into_string()?);
|
||||
|
||||
Ok(())
|
||||
}
|
||||
''';
|
||||
expect(
|
||||
codeGen.getCode(CodegenLanguage.rustUreq, requestModelPut1, "https"),
|
||||
expectedCode);
|
||||
});
|
||||
});
|
||||
|
||||
group('PATCH Request', () {
|
||||
test('PATCH 1', () {
|
||||
const expectedCode = r'''fn main() -> Result<(), ureq::Error> {
|
||||
let url = "https://reqres.in/api/users/2";
|
||||
let payload = ureq::json!({
|
||||
"name": "marfeus",
|
||||
"job": "accountant"
|
||||
});
|
||||
|
||||
let response = ureq::patch(url)
|
||||
.send_json(payload)?;
|
||||
|
||||
println!("Response Status: {}", response.status());
|
||||
println!("Response: {}", response.into_string()?);
|
||||
|
||||
Ok(())
|
||||
}
|
||||
''';
|
||||
expect(
|
||||
codeGen.getCode(
|
||||
CodegenLanguage.rustUreq, requestModelPatch1, "https"),
|
||||
expectedCode);
|
||||
});
|
||||
});
|
||||
|
||||
group('DELETE Request', () {
|
||||
test('DELETE 1', () {
|
||||
const expectedCode = r"""fn main() -> Result<(), ureq::Error> {
|
||||
let url = "https://reqres.in/api/users/2";
|
||||
let response = ureq::delete(url)
|
||||
.call()?;
|
||||
|
||||
println!("Response Status: {}", response.status());
|
||||
println!("Response: {}", response.into_string()?);
|
||||
|
||||
Ok(())
|
||||
}
|
||||
""";
|
||||
expect(
|
||||
codeGen.getCode(
|
||||
CodegenLanguage.rustUreq, requestModelDelete1, "https"),
|
||||
expectedCode);
|
||||
});
|
||||
|
||||
test('DELETE 2', () {
|
||||
const expectedCode = r'''fn main() -> Result<(), ureq::Error> {
|
||||
let url = "https://reqres.in/api/users/2";
|
||||
let payload = ureq::json!({
|
||||
"name": "marfeus",
|
||||
"job": "accountant"
|
||||
});
|
||||
|
||||
let response = ureq::delete(url)
|
||||
.send_json(payload)?;
|
||||
|
||||
println!("Response Status: {}", response.status());
|
||||
println!("Response: {}", response.into_string()?);
|
||||
|
||||
Ok(())
|
||||
}
|
||||
''';
|
||||
expect(
|
||||
codeGen.getCode(
|
||||
CodegenLanguage.rustUreq, requestModelDelete2, "https"),
|
||||
expectedCode);
|
||||
});
|
||||
});
|
||||
}
|
Reference in New Issue
Block a user