Merge pull request #342 from apoorvdwi/add-rust-curl-codegen

Add Curl Rust Code Generator

Author: ashitaprasad

lib/codegen/codegen.dart

@@ -9,6 +9,7 @@ import 'php/guzzle.dart';
 import 'python/http_client.dart';
 import 'python/requests.dart';
 import 'rust/actix.dart';
+import 'rust/curl_rust.dart';
 import 'rust/reqwest.dart';
 import 'rust/ureq.dart';
 import 'js/axios.dart';
@@ -73,6 +74,8 @@ class Codegen {
         return PythonRequestsCodeGen().getCode(rM, boundary: boundary);
       case CodegenLanguage.rustActix:
         return RustActixCodeGen().getCode(rM, boundary: boundary);
+      case CodegenLanguage.rustCurl:
+        return RustCurlCodeGen().getCode(rM);
       case CodegenLanguage.rustReqwest:
         return RustReqwestCodeGen().getCode(rM);
       case CodegenLanguage.rustUreq:

lib/codegen/rust/curl_rust.dart

@@ -0,0 +1,147 @@
+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().unwrap();
+  {% endfor %}
+  easy.httppost(form).unwrap();
+  """;
+
+  String kTemplateHeader = """
+  {% if headers %}let mut list = List::new();{% for header, value in headers %}
+  list.append("{{header}}: {{value}}").unwrap();{% 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.enabledRequestHeaders!.isNotEmpty) ||
+            (requestModel.hasJsonData || requestModel.hasTextData)
+      });
+
+      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;
+    }
+  }
+}

lib/consts.dart

@@ -284,6 +284,7 @@ enum CodegenLanguage {
   pythonHttpClient("Python (http.client)", "python", "py"),
   rustActix("Rust (Actix Client)", "rust", "rs"),
   rustReqwest("Rust (reqwest)", "rust", "rs"),
+  rustCurl("Rust (curl-rust)", "rust", "rs"),
   rustUreq("Rust (ureq)", "rust", "rs"),
   javaOkHttp("Java (okhttp3)", "java", 'java'),
   javaAsyncHttpClient("Java (asynchttpclient)", "java", "java"),