flutter/apidash
1
0
Fork 0
mirror of https://github.com/foss42/apidash.git synced 2025-05-28 20:27:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

fixes

Browse Source
This commit is contained in:
Ashita Prasad
2024-10-02 13:07:53 +05:30
parent 3eff5ca64a
commit 5b1f8fdbea
2 changed files with 590 additions and 710 deletions
Download Patch File Download Diff File Expand all files Collapse all files

116
lib/codegen/rust/hyper.dart
View File

@ -1,45 +1,38 @@
import 'dart:core';
import 'package:jinja/jinja.dart' as jj;
import 'package:apidash/consts.dart';
import 'package:apidash/utils/utils.dart'
show getValidRequestUri;
import 'package:apidash/utils/utils.dart' show getValidRequestUri;
import 'package:apidash/models/models.dart';
class RustHyperCodeGen {
final String kTemplateStart = """
{% if hasForm %}extern crate hyper_multipart_rfc7578 as hyper_multipart;{% endif %}
use hyper::{Body, Client, Request, Uri};
use hyper::client::HttpConnector;
use hyper_tls::HttpsConnector;
{% if hasForm %}use hyper_multipart::client::multipart;{% endif %}
{% if hasJsonBody %}use serde_json::json;{% endif %}
use tokio;
{% if hasForm %}extern crate hyper_multipart_rfc7578 as hyper_multipart;
{% endif %}use hyper::{Body, Client, Request, Uri};
{% if isHttps %}use hyper_tls::HttpsConnector;
{% else %}use hyper::client::HttpConnector;
{% endif %}{% if hasForm %}use hyper_multipart::client::multipart;
{% endif %}{% if hasJsonBody %}use serde_json::json;
{% endif %}use tokio;
#[tokio::main]
async fn main() -> Result<(), Box<dyn std::error::Error>> {
let https = HttpsConnector::new();
let client = Client::builder().build::<_, hyper::Body>(https);
let http{% if isHttps %}s{% endif %} = Http{% if isHttps %}s{% endif %}Connector::new();
let client = Client::builder().build::<_, hyper::Body>(http{% if isHttps %}s{% endif %});
let url = "{{ url }}".parse::<Uri>().unwrap();
""";
final String kTemplateMethod = """
let reqBuilder = Request::builder()
let req_builder = Request::builder()
.method("{{ method }}")
.uri(url)
""";
final String kTemplateMethodNoHeadersButForm = """
let reqBuilder = Request::builder()
let req_builder = Request::builder()
.method("{{ method }}")
.uri(url);
""";
final String kTemplateHeaders = """
{% for key, val in headers %}
.header("{{ key }}", "{{ val }}")
@ -51,7 +44,6 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
{% endfor %}
""";
final String kTemplateBody = """
.body(Body::from(r#"{{ body }}"#))?;\n
@ -67,11 +59,7 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
.body(Body::empty())?;\n
""";
final String kTemplateFormData = """
final String kTemplateFormData = """
let mut form = multipart::Form::default();
{%- for field in fields_list %}
@ -82,12 +70,11 @@ final String kTemplateFormData = """
{%- endif %}
{%- endfor %}
let req = form.set_body_convert::<Body, multipart::Body>(reqBuilder).unwrap();
let req = form.set_body_convert::<Body, multipart::Body>(req_builder).unwrap();
""";
final String kTemplateEndForm = """
final String kTemplateEndForm = """
let res = client.request(req).await?;
let status = res.status();
let body_bytes = hyper::body::to_bytes(res).await?;
@ -100,7 +87,7 @@ final String kTemplateFormData = """
""";
final String kTemplateRequestEnd = """
let res = client.request(reqBuilder).await?;
let res = client.request(req_builder).await?;
let status = res.status();
let body_bytes = hyper::body::to_bytes(res).await?;
let body = String::from_utf8(body_bytes.to_vec())?;
@ -110,83 +97,76 @@ final String kTemplateFormData = """
""";
final String kTemplateEnd = """
final String kTemplateEnd = """
Ok(())
}
""";
String? getCode(HttpRequestModel requestModel) {
try {
String result = "";
String url = requestModel.url;
var rec = getValidRequestUri(url, requestModel.enabledParams);
Uri? uri = rec.$1;
if (uri != null) {
var headers = requestModel.enabledHeadersMap;
result += jj.Template(kTemplateStart).render({
"url": uri,
'hasJsonBody': requestModel.bodyContentType == ContentType.json,
"isHttps": uri.scheme == "https" ? true : false,
'hasJsonBody': requestModel.hasJsonData,
'hasForm': requestModel.hasFormData,
});
if(requestModel.hasFormData && headers.isEmpty){
result += jj.Template(kTemplateMethodNoHeadersButForm).render({
"method": requestModel.method.name.toUpperCase(),
});
}else{
result += jj.Template(kTemplateMethod).render({
"method": requestModel.method.name.toUpperCase(),
});
if (requestModel.hasFormData && headers.isEmpty) {
result += jj.Template(kTemplateMethodNoHeadersButForm).render({
"method": requestModel.method.name.toUpperCase(),
});
} else {
result += jj.Template(kTemplateMethod).render({
"method": requestModel.method.name.toUpperCase(),
});
}
// Add headers if available
if (headers.isNotEmpty) {
if(requestModel.hasFormData){
result += jj.Template(kTemplateHeadersFormData).render({"headers": headers});
}else{
result += jj.Template(kTemplateHeaders).render({"headers": headers});
if (requestModel.hasFormData) {
result += jj.Template(kTemplateHeadersFormData)
.render({"headers": headers});
} else {
result +=
jj.Template(kTemplateHeaders).render({"headers": headers});
}
}
// Handle body (JSON or raw)
var requestBody = requestModel.body;
if (requestModel.hasFormData) {
result += jj.Template(kTemplateFormData).render({
"fields_list": requestModel.formDataMapList,
});
}else if (requestBody == "" || requestBody == null|| requestModel.method ==HTTPVerb.get || requestModel.method == HTTPVerb.head) {
} else if (requestBody == "" ||
requestBody == null ||
requestModel.method == HTTPVerb.get ||
requestModel.method == HTTPVerb.head) {
result += kTemplateEmptyBody;
}else if(requestModel.hasJsonData){
result += jj.Template(kTemplateJsonBody).render({"body": requestBody});
}else if(requestModel.hasTextData){
} else if (requestModel.hasJsonData) {
result +=
jj.Template(kTemplateJsonBody).render({"body": requestBody});
} else if (requestModel.hasTextData) {
result += jj.Template(kTemplateBody).render({"body": requestBody});
}
// End request
if(requestModel.hasFormData && requestModel.method!=HTTPVerb.get){
result+=kTemplateEndForm;
}else{
result+=kTemplateRequestEnd;
if (requestModel.hasFormData && requestModel.method != HTTPVerb.get) {
result += kTemplateEndForm;
} else {
result += kTemplateRequestEnd;
}
result+=kTemplateEnd;
result += kTemplateEnd;
}
return result;