mirror of
https://github.com/foss42/apidash.git
synced 2025-06-03 16:27:06 +08:00
Fixed the java http client codegen
This commit is contained in:
@ -60,7 +60,7 @@ class Codegen {
|
|||||||
case CodegenLanguage.javaAsyncHttpClient:
|
case CodegenLanguage.javaAsyncHttpClient:
|
||||||
return JavaAsyncHttpClientGen().getCode(rM);
|
return JavaAsyncHttpClientGen().getCode(rM);
|
||||||
case CodegenLanguage.javaHttpClient:
|
case CodegenLanguage.javaHttpClient:
|
||||||
return JavaHttpClientCodeGen().getCode(rM);
|
return JavaHttpClientCodeGen().getCode(rM, boundary: boundary);
|
||||||
case CodegenLanguage.javaOkHttp:
|
case CodegenLanguage.javaOkHttp:
|
||||||
return JavaOkHttpCodeGen().getCode(rM);
|
return JavaOkHttpCodeGen().getCode(rM);
|
||||||
case CodegenLanguage.juliaHttp:
|
case CodegenLanguage.juliaHttp:
|
||||||
|
@ -1,183 +1,200 @@
|
|||||||
import 'dart:convert';
|
|
||||||
import 'package:jinja/jinja.dart' as jj;
|
import 'package:jinja/jinja.dart' as jj;
|
||||||
import 'package:apidash/utils/utils.dart'
|
import 'package:apidash/utils/utils.dart'
|
||||||
show getValidRequestUri, requestModelToHARJsonRequest, stripUriParams;
|
show getValidRequestUri, requestModelToHARJsonRequest;
|
||||||
import '../../models/request_model.dart';
|
import 'package:apidash/models/models.dart' show RequestModel;
|
||||||
import 'package:apidash/consts.dart';
|
import 'package:apidash/consts.dart';
|
||||||
|
|
||||||
class JavaHttpClientCodeGen {
|
class JavaHttpClientCodeGen {
|
||||||
final String kTemplateStart = """
|
final String kTemplateStart = """import java.net.URI;
|
||||||
import java.io.IOException;
|
|
||||||
import java.net.URI;
|
|
||||||
import java.net.URISyntaxException;
|
|
||||||
import java.net.http.HttpClient;
|
import java.net.http.HttpClient;
|
||||||
import java.net.http.HttpRequest;
|
import java.net.http.HttpRequest;
|
||||||
import java.net.http.HttpResponse;
|
import java.net.http.HttpResponse;
|
||||||
import java.net.http.HttpHeaders;
|
import java.io.IOException;
|
||||||
import java.net.http.HttpRequest.BodyPublishers;
|
{% if hasFormData %}import java.util.HashMap;
|
||||||
import java.net.http.HttpResponse.BodyHandlers;
|
import java.util.Map;
|
||||||
|
import java.nio.file.Files;
|
||||||
|
import java.nio.file.Path;
|
||||||
|
import java.nio.file.Paths;
|
||||||
|
import java.nio.charset.StandardCharsets;
|
||||||
|
import java.util.ArrayList;{% endif %}
|
||||||
|
{% if hasJsonBody %}import com.google.gson.Gson;
|
||||||
|
import java.nio.charset.StandardCharsets;{% endif %}
|
||||||
|
|
||||||
public class JavaHttpClientExample {
|
public class Main {
|
||||||
|
public static void main(String[] args) {
|
||||||
|
try {
|
||||||
|
HttpClient client = HttpClient.newHttpClient();
|
||||||
|
|
||||||
public static void main(String[] args) throws IOException, InterruptedException {
|
|
||||||
HttpClient client = HttpClient.newHttpClient();
|
|
||||||
""";
|
|
||||||
final String kTemplateUrl = '''
|
|
||||||
|
|
||||||
String url = "{{url}}";
|
|
||||||
|
|
||||||
''';
|
|
||||||
|
|
||||||
final String kTemplateUrlQuery = '''
|
|
||||||
|
|
||||||
String url = "{{url}}";
|
|
||||||
try {
|
|
||||||
URI uri = new URI(url);
|
|
||||||
url = uri.resolve("{{params}}").toString();
|
|
||||||
} catch (URISyntaxException e) {
|
|
||||||
e.printStackTrace();
|
|
||||||
}
|
|
||||||
|
|
||||||
''';
|
|
||||||
|
|
||||||
String kTemplateRequestBody = '''
|
|
||||||
|
|
||||||
String body = "{{body}}";
|
|
||||||
|
|
||||||
''';
|
|
||||||
|
|
||||||
final String kStringRequestStart = """
|
|
||||||
|
|
||||||
HttpRequest request = HttpRequest.newBuilder()
|
|
||||||
.uri(URI.create(url))
|
|
||||||
""";
|
""";
|
||||||
|
|
||||||
final String kTemplateRequestEnd = """
|
String kTemplateUrl = """
|
||||||
.{{method}}({{body}})
|
URI uri = URI.create("{{url}}");
|
||||||
.build();
|
|
||||||
|
|
||||||
HttpResponse<String> response = client.send(request, BodyHandlers.ofString());
|
""";
|
||||||
System.out.println(response.statusCode());
|
|
||||||
System.out.println(response.body());
|
String kTemplateFormHeaderContentType = '''
|
||||||
|
multipart/form-data; boundary={{boundary}}''';
|
||||||
|
|
||||||
|
String kTemplateMethod = """
|
||||||
|
{% if method == 'get' %}
|
||||||
|
HttpRequest.Builder requestBuilder = HttpRequest.newBuilder(uri).GET();
|
||||||
|
{% elif method == 'post' %}
|
||||||
|
HttpRequest.Builder requestBuilder = HttpRequest.newBuilder(uri).POST({% if hasBody %}bodyPublisher{% else %}HttpRequest.BodyPublishers.noBody(){% endif %});
|
||||||
|
{% elif method == 'put' %}
|
||||||
|
HttpRequest.Builder requestBuilder = HttpRequest.newBuilder(uri).PUT({% if hasBody %}bodyPublisher{% else %}HttpRequest.BodyPublishers.noBody(){% endif %});
|
||||||
|
{% elif method == 'delete' %}
|
||||||
|
HttpRequest.Builder requestBuilder = HttpRequest.newBuilder(uri).method("DELETE", {% if hasBody %}bodyPublisher{% else %}HttpRequest.BodyPublishers.noBody(){% endif %});
|
||||||
|
{% elif method == 'patch' %}
|
||||||
|
HttpRequest.Builder requestBuilder = HttpRequest.newBuilder(uri).method("PATCH", {% if hasBody %}bodyPublisher{% else %}HttpRequest.BodyPublishers.noBody(){% endif %});
|
||||||
|
{% elif method == 'head' %}
|
||||||
|
HttpRequest.Builder requestBuilder = HttpRequest.newBuilder(uri).method("HEAD", HttpRequest.BodyPublishers.noBody());
|
||||||
|
{% endif %}
|
||||||
|
""";
|
||||||
|
|
||||||
|
String kTemplateRawBody = """
|
||||||
|
HttpRequest.BodyPublisher bodyPublisher = HttpRequest.BodyPublishers.ofString(\"\"\"
|
||||||
|
{{body}}\"\"\");
|
||||||
|
""";
|
||||||
|
|
||||||
|
String kTemplateJsonBody = """
|
||||||
|
HttpRequest.BodyPublisher bodyPublisher = HttpRequest.BodyPublishers.ofString(new Gson().toJson(\"\"\"
|
||||||
|
{{body}}\"\"\"), StandardCharsets.UTF_8);
|
||||||
|
""";
|
||||||
|
|
||||||
|
String kTemplateFormData = """
|
||||||
|
String boundary = "{{boundary}}";
|
||||||
|
Map<Object, Object> data = new HashMap<>();
|
||||||
|
{% for field in fields %}
|
||||||
|
{% if field.type == "file" %}data.put("{{field.name}}", Paths.get("{{field.value}}"));{% else %}data.put("{{field.name}}", "{{field.value}}");{% endif %}{% endfor %}
|
||||||
|
HttpRequest.BodyPublisher bodyPublisher = buildMultipartFormData(data, boundary);
|
||||||
|
""";
|
||||||
|
|
||||||
|
String kTemplateHeader = """
|
||||||
|
requestBuilder = requestBuilder.headers({% for header, value in headers %}
|
||||||
|
"{{header}}", "{{value}}"{% if not loop.last %},{% endif %}{% endfor %}
|
||||||
|
);
|
||||||
|
|
||||||
|
""";
|
||||||
|
|
||||||
|
final String kTemplateEnd = """
|
||||||
|
HttpResponse<String> response = client.send(requestBuilder.build(), HttpResponse.BodyHandlers.ofString());
|
||||||
|
System.out.println("Response body: " + response.body());
|
||||||
|
System.out.println("Response code: " + response.statusCode());
|
||||||
|
} catch (IOException | InterruptedException e) {
|
||||||
|
System.out.println("An error occurred: " + e.getMessage());
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
{% if hasFormData %}
|
||||||
|
private static HttpRequest.BodyPublisher buildMultipartFormData(Map<Object, Object> data, String boundary) throws IOException {
|
||||||
|
var byteArrays = new ArrayList<byte[]>();
|
||||||
|
var CRLF = "\\r\\n".getBytes(StandardCharsets.UTF_8);
|
||||||
|
|
||||||
|
for (Map.Entry<Object, Object> entry : data.entrySet()) {
|
||||||
|
byteArrays.add(("--" + boundary + "\\r\\n").getBytes(StandardCharsets.UTF_8));
|
||||||
|
if (entry.getValue() instanceof Path) {
|
||||||
|
var file = (Path) entry.getValue();
|
||||||
|
var fileName = file.getFileName().toString();
|
||||||
|
byteArrays.add(("Content-Disposition: form-data; name=\\"" + entry.getKey() + "\\"; filename=\\"" + fileName + "\\"\\r\\n").getBytes(StandardCharsets.UTF_8));
|
||||||
|
byteArrays.add(("Content-Type: " + Files.probeContentType(file) + "\\r\\n\\r\\n").getBytes(StandardCharsets.UTF_8));
|
||||||
|
byteArrays.add(Files.readAllBytes(file));
|
||||||
|
byteArrays.add(CRLF);
|
||||||
|
} else {
|
||||||
|
byteArrays.add(("Content-Disposition: form-data; name=\\"" + entry.getKey() + "\\"\\r\\n\\r\\n").getBytes(StandardCharsets.UTF_8));
|
||||||
|
byteArrays.add(entry.getValue().toString().getBytes(StandardCharsets.UTF_8));
|
||||||
|
byteArrays.add(CRLF);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
byteArrays.add(("--" + boundary + "--\\r\\n").getBytes(StandardCharsets.UTF_8));
|
||||||
|
|
||||||
|
return HttpRequest.BodyPublishers.ofByteArrays(byteArrays);
|
||||||
|
}{% endif %}
|
||||||
}
|
}
|
||||||
\n
|
|
||||||
""";
|
""";
|
||||||
|
|
||||||
String? getCode(
|
String? getCode(
|
||||||
RequestModel requestModel,
|
RequestModel requestModel, {
|
||||||
) {
|
String? boundary,
|
||||||
|
}) {
|
||||||
try {
|
try {
|
||||||
String result = "";
|
String result = "";
|
||||||
bool hasQuery = false;
|
var requestBody = requestModel.requestBody;
|
||||||
bool hasBody = false;
|
String url = requestModel.url;
|
||||||
bool hasJsonBody = false;
|
|
||||||
|
result += jj.Template(kTemplateStart).render({
|
||||||
|
"hasJsonBody": requestModel.hasJsonData,
|
||||||
|
"hasFormData": requestModel.hasFormData,
|
||||||
|
});
|
||||||
|
|
||||||
var rec = getValidRequestUri(
|
var rec = getValidRequestUri(
|
||||||
requestModel.url,
|
url,
|
||||||
requestModel.enabledRequestParams,
|
requestModel.enabledRequestParams,
|
||||||
);
|
);
|
||||||
|
|
||||||
Uri? uri = rec.$1;
|
Uri? uri = rec.$1;
|
||||||
|
|
||||||
|
var harJson =
|
||||||
|
requestModelToHARJsonRequest(requestModel, useEnabled: true);
|
||||||
|
|
||||||
if (uri != null) {
|
if (uri != null) {
|
||||||
String url = stripUriParams(uri);
|
var templateUrl = jj.Template(kTemplateUrl);
|
||||||
|
result += templateUrl.render({"url": harJson["url"]});
|
||||||
|
|
||||||
if (uri.hasQuery) {
|
String? bodyPublisher = "";
|
||||||
var params = uri.queryParameters;
|
if (requestModel.hasTextData) {
|
||||||
if (params.isNotEmpty) {
|
var templateBody = jj.Template(kTemplateRawBody);
|
||||||
hasQuery = true;
|
bodyPublisher = templateBody.render({"body": requestBody});
|
||||||
var templateParams = jj.Template(kTemplateUrlQuery);
|
} else if (requestModel.hasJsonData) {
|
||||||
result += templateParams.render({"url": url, "params": uri.query});
|
var templateBody = jj.Template(kTemplateJsonBody);
|
||||||
}
|
bodyPublisher = templateBody.render({"body": requestBody});
|
||||||
|
} else if (requestModel.hasFormData) {
|
||||||
|
var templateFormData = jj.Template(kTemplateFormData);
|
||||||
|
bodyPublisher = templateFormData.render({
|
||||||
|
"fields": requestModel.formDataMapList,
|
||||||
|
"boundary": boundary,
|
||||||
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
if (!hasQuery) {
|
result += bodyPublisher;
|
||||||
var templateUrl = jj.Template(kTemplateUrl);
|
|
||||||
result += templateUrl.render({"url": url});
|
|
||||||
}
|
|
||||||
var rM = requestModel.copyWith(url: url);
|
|
||||||
|
|
||||||
var harJson = requestModelToHARJsonRequest(rM, useEnabled: true);
|
var methodTemplate = jj.Template(kTemplateMethod);
|
||||||
|
result += methodTemplate.render({
|
||||||
var method = requestModel.method;
|
"method": requestModel.method.name,
|
||||||
var requestBody = requestModel.requestBody;
|
"hasBody": requestModel.hasBody,
|
||||||
if (requestModel.hasFormData &&
|
});
|
||||||
requestModel.formDataMapList.isNotEmpty &&
|
|
||||||
kMethodsWithBody.contains(method)) {
|
|
||||||
var formDataList = requestModel.formDataMapList;
|
|
||||||
result += """
|
|
||||||
StringBuilder formData = new StringBuilder();
|
|
||||||
formData.append(""";
|
|
||||||
|
|
||||||
for (var formDataMap in formDataList) {
|
|
||||||
result += '"""${formDataMap['name']}=${formDataMap['value']}&""",';
|
|
||||||
}
|
|
||||||
|
|
||||||
result = result.substring(0, result.length - 1);
|
|
||||||
result += ");\n";
|
|
||||||
hasBody = true;
|
|
||||||
} else if (kMethodsWithBody.contains(method) && requestBody != null) {
|
|
||||||
var contentLength = utf8.encode(requestBody).length;
|
|
||||||
if (contentLength > 0) {
|
|
||||||
var templateBody = jj.Template(kTemplateRequestBody);
|
|
||||||
hasBody = true;
|
|
||||||
hasJsonBody =
|
|
||||||
requestBody.startsWith("{") && requestBody.endsWith("}");
|
|
||||||
if (harJson["postData"]?["text"] != null) {
|
|
||||||
result += templateBody.render({
|
|
||||||
"body": kEncoder.convert(harJson["postData"]["text"]).substring(
|
|
||||||
1, kEncoder.convert(harJson["postData"]["text"]).length - 1)
|
|
||||||
});
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
result = kTemplateStart + result;
|
|
||||||
result += kStringRequestStart;
|
|
||||||
|
|
||||||
var headersList = requestModel.enabledRequestHeaders;
|
var headersList = requestModel.enabledRequestHeaders;
|
||||||
var contentType = requestModel.requestBodyContentType.header;
|
if (headersList != null || requestModel.hasBody) {
|
||||||
if (hasBody &&
|
|
||||||
!requestModel.enabledHeadersMap.containsKey('Content-Type')) {
|
|
||||||
result =
|
|
||||||
"""$result .header("Content-Type", "$contentType")\n""";
|
|
||||||
}
|
|
||||||
if (headersList != null) {
|
|
||||||
var headers = requestModel.enabledHeadersMap;
|
var headers = requestModel.enabledHeadersMap;
|
||||||
|
if (requestModel.hasJsonData || requestModel.hasTextData) {
|
||||||
|
headers.putIfAbsent(kHeaderContentType,
|
||||||
|
() => requestModel.requestBodyContentType.header);
|
||||||
|
}
|
||||||
|
if (requestModel.hasFormData) {
|
||||||
|
var formDataHeader = jj.Template(kTemplateFormHeaderContentType);
|
||||||
|
headers.putIfAbsent(
|
||||||
|
kHeaderContentType,
|
||||||
|
() => formDataHeader.render({
|
||||||
|
"boundary": boundary,
|
||||||
|
}));
|
||||||
|
}
|
||||||
if (headers.isNotEmpty) {
|
if (headers.isNotEmpty) {
|
||||||
result += getHeaders(headers, hasJsonBody);
|
var templateHeader = jj.Template(kTemplateHeader);
|
||||||
|
result += templateHeader.render({
|
||||||
|
"headers": headers,
|
||||||
|
});
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
var templateRequestEnd = jj.Template(kTemplateRequestEnd);
|
var templateEnd = jj.Template(kTemplateEnd);
|
||||||
|
result += templateEnd.render({
|
||||||
if (kMethodsWithBody.contains(method)) {
|
"hasFormData": requestModel.hasFormData,
|
||||||
result += templateRequestEnd.render({
|
"boundary": boundary,
|
||||||
"method": method.name.toUpperCase(),
|
});
|
||||||
"body": hasBody
|
|
||||||
? "BodyPublishers.ofString(body)"
|
|
||||||
: "BodyPublishers.noBody()"
|
|
||||||
});
|
|
||||||
} else {
|
|
||||||
result += templateRequestEnd
|
|
||||||
.render({"method": method.name.toUpperCase(), "body": ""});
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
return result;
|
|
||||||
|
return result.replaceAll(RegExp('\\n\\n+'), '\n\n');
|
||||||
} catch (e) {
|
} catch (e) {
|
||||||
return null;
|
return null;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
String getHeaders(Map<String, String> headers, hasJsonBody) {
|
|
||||||
String result = "";
|
|
||||||
for (final k in headers.keys) {
|
|
||||||
if (k.toLowerCase() == 'authorization') {
|
|
||||||
result = """$result .header("$k", "${headers[k]}")\n""";
|
|
||||||
} else {
|
|
||||||
result = """$result .header("$k", "${headers[k]}")\n""";
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return result;
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
Reference in New Issue
Block a user