wip: updated http and requests header handle

This commit is contained in:
Vidya Sagar
2023-12-31 23:25:35 +05:30
parent 68c69d37fe
commit f0047aa087
2 changed files with 37 additions and 43 deletions

View File

@ -1,11 +1,10 @@
import 'dart:io'; import 'dart:io';
import 'dart:convert'; import 'dart:convert';
import 'package:apidash/utils/convert_utils.dart';
import 'package:apidash/utils/extensions/request_model_extension.dart'; import 'package:apidash/utils/extensions/request_model_extension.dart';
import 'package:jinja/jinja.dart' as jj; import 'package:jinja/jinja.dart' as jj;
import 'package:apidash/consts.dart'; import 'package:apidash/consts.dart';
import 'package:apidash/utils/utils.dart' import 'package:apidash/utils/utils.dart'
show getValidRequestUri, padMultilineString; show getNewUuid, getValidRequestUri, padMultilineString, rowsToFormDataMap;
import 'package:apidash/models/models.dart' show FormDataModel, RequestModel; import 'package:apidash/models/models.dart' show FormDataModel, RequestModel;
class PythonHttpClientCodeGen { class PythonHttpClientCodeGen {
@ -13,11 +12,8 @@ class PythonHttpClientCodeGen {
{% if isFormDataRequest %} {% if isFormDataRequest %}
import mimetypes import mimetypes
from codecs import encode from codecs import encode
import uuid
headers = {}
boundary = str(uuid.uuid4())
{% endif %} {% endif %}
"""; """;
String kTemplateParams = """ String kTemplateParams = """
@ -40,6 +36,8 @@ body = r'''{{body}}'''
headers = {{headers}} headers = {{headers}}
"""; """;
String kTemplateFormHeaderContentType = '''
multipart/form-data; boundary={{boundary}}''';
int kHeadersPadding = 10; int kHeadersPadding = 10;
@ -51,10 +49,6 @@ conn = http.client.HTTP{{isHttps}}Connection("{{authority}}")""";
conn.request("{{method}}", "{{path}}"{{queryParamsStr}}"""; conn.request("{{method}}", "{{path}}"{{queryParamsStr}}""";
String kTemplateFormDataRequest = """
conn.request("{{method}}", "{{path}}"{{queryParamsStr}},body=payload, headers=headers
""";
String kStringRequestBody = """, String kStringRequestBody = """,
body= body"""; body= body""";
@ -76,8 +70,7 @@ def build_data_list(fields):
name = field.get('name', '') name = field.get('name', '')
value = field.get('value', '') value = field.get('value', '')
type_ = field.get('type', 'text') type_ = field.get('type', 'text')
dataList.append(encode('--{{boundary}}'))
dataList.append(encode(f'--{boundary}'))
if type_ == 'text': if type_ == 'text':
dataList.append(encode(f'Content-Disposition: form-data; name="{name}"')) dataList.append(encode(f'Content-Disposition: form-data; name="{name}"'))
dataList.append(encode('Content-Type: text/plain')) dataList.append(encode('Content-Type: text/plain'))
@ -88,23 +81,18 @@ def build_data_list(fields):
dataList.append(encode(f'Content-Type: {mimetypes.guess_type(value)[0] or "application/octet-stream"}')) dataList.append(encode(f'Content-Type: {mimetypes.guess_type(value)[0] or "application/octet-stream"}'))
dataList.append(encode('')) dataList.append(encode(''))
dataList.append(open(value, 'rb').read()) dataList.append(open(value, 'rb').read())
dataList.append(encode(f'--{{boundary}}--'))
dataList.append(encode(f'--{boundary}--'))
dataList.append(encode('')) dataList.append(encode(''))
return dataList return dataList
dataList = build_data_list({{fields_list}}) dataList = build_data_list({{fields_list}})
body = b'\r\n'.join(dataList) body = b'\r\n'.join(dataList)
'''; ''';
String kFormDataHeaders = '''
headers['Content-type'] = f'multipart/form-data; boundary={boundary}';
''';
String? getCode( String? getCode(
RequestModel requestModel, RequestModel requestModel,
String defaultUriScheme, String defaultUriScheme,
) { ) {
List<FormDataModel> formDataList = requestModel.formDataList ?? []; List<FormDataModel> formDataList = requestModel.formDataList ?? [];
String uuid = getNewUuid();
try { try {
String result = ""; String result = "";
@ -116,6 +104,7 @@ headers['Content-type'] = f'multipart/form-data; boundary={boundary}';
if (!url.contains("://") && url.isNotEmpty) { if (!url.contains("://") && url.isNotEmpty) {
url = "$defaultUriScheme://$url"; url = "$defaultUriScheme://$url";
} }
var templateStartUrl = jj.Template(kTemplateStart); var templateStartUrl = jj.Template(kTemplateStart);
result += templateStartUrl.render( result += templateStartUrl.render(
{ {
@ -151,6 +140,13 @@ headers['Content-type'] = f'multipart/form-data; boundary={boundary}';
var headersList = requestModel.requestHeaders; var headersList = requestModel.requestHeaders;
if (headersList != null || hasBody) { if (headersList != null || hasBody) {
var headers = requestModel.headersMap; var headers = requestModel.headersMap;
if (requestModel.isFormDataRequest) {
var formHeaderTemplate =
jj.Template(kTemplateFormHeaderContentType);
headers[HttpHeaders.contentTypeHeader] = formHeaderTemplate.render({
"boundary": uuid,
});
}
if (headers.isNotEmpty || hasBody) { if (headers.isNotEmpty || hasBody) {
hasHeaders = true; hasHeaders = true;
if (hasBody) { if (hasBody) {
@ -164,11 +160,11 @@ headers['Content-type'] = f'multipart/form-data; boundary={boundary}';
} }
} }
if (requestModel.isFormDataRequest) { if (requestModel.isFormDataRequest) {
result += kFormDataHeaders;
var formDataBodyData = jj.Template(kStringFormDataBody); var formDataBodyData = jj.Template(kStringFormDataBody);
result += formDataBodyData.render( result += formDataBodyData.render(
{ {
"fields_list": json.encode(rowsToFormDataMap(formDataList)), "fields_list": json.encode(rowsToFormDataMap(formDataList)),
"boundary": uuid,
}, },
); );
} }
@ -177,6 +173,7 @@ headers['Content-type'] = f'multipart/form-data; boundary={boundary}';
"isHttps": uri.scheme == "https" ? "S" : "", "isHttps": uri.scheme == "https" ? "S" : "",
"authority": uri.authority "authority": uri.authority
}); });
var templateRequest = jj.Template(kTemplateRequest); var templateRequest = jj.Template(kTemplateRequest);
result += templateRequest.render({ result += templateRequest.render({
"method": method.name.toUpperCase(), "method": method.name.toUpperCase(),
@ -196,7 +193,6 @@ headers['Content-type'] = f'multipart/form-data; boundary={boundary}';
} }
return result; return result;
} catch (e) { } catch (e) {
print(e);
return null; return null;
} }
} }

View File

@ -1,11 +1,15 @@
import 'dart:io'; import 'dart:io';
import 'dart:convert'; import 'dart:convert';
import 'package:apidash/utils/convert_utils.dart';
import 'package:apidash/utils/extensions/request_model_extension.dart'; import 'package:apidash/utils/extensions/request_model_extension.dart';
import 'package:jinja/jinja.dart' as jj; import 'package:jinja/jinja.dart' as jj;
import 'package:apidash/consts.dart'; import 'package:apidash/consts.dart';
import 'package:apidash/utils/utils.dart' import 'package:apidash/utils/utils.dart'
show getValidRequestUri, padMultilineString, stripUriParams; show
getNewUuid,
getValidRequestUri,
padMultilineString,
rowsToFormDataMap,
stripUriParams;
import 'package:apidash/models/models.dart' show FormDataModel, RequestModel; import 'package:apidash/models/models.dart' show FormDataModel, RequestModel;
class PythonRequestsCodeGen { class PythonRequestsCodeGen {
@ -13,23 +17,9 @@ class PythonRequestsCodeGen {
{% if isFormDataRequest %} {% if isFormDataRequest %}
import mimetypes import mimetypes
from codecs import encode from codecs import encode
import uuid
headers = {}
boundary = str(uuid.uuid4())
{% endif %} {% endif %}
url = '{{url}}' url = '{{url}}'
""";
final String kTemplateFormDataImports = """
import mimetypes
from codecs import encode
import uuid
headers = {}
boundary = str(uuid.uuid4())
"""; """;
String kTemplateParams = """ String kTemplateParams = """
@ -56,6 +46,8 @@ payload = {{body}}
headers = {{headers}} headers = {{headers}}
"""; """;
String kTemplateFormHeaderContentType = '''
multipart/form-data; boundary={{boundary}}''';
int kHeadersPadding = 10; int kHeadersPadding = 10;
@ -72,7 +64,7 @@ def build_data_list(fields):
value = field.get('value', '') value = field.get('value', '')
type_ = field.get('type', 'text') type_ = field.get('type', 'text')
dataList.append(encode(f'--{boundary}')) dataList.append(encode('--{{boundary}}'))
if type_ == 'text': if type_ == 'text':
dataList.append(encode(f'Content-Disposition: form-data; name="{name}"')) dataList.append(encode(f'Content-Disposition: form-data; name="{name}"'))
dataList.append(encode('Content-Type: text/plain')) dataList.append(encode('Content-Type: text/plain'))
@ -84,12 +76,13 @@ def build_data_list(fields):
dataList.append(encode('')) dataList.append(encode(''))
dataList.append(open(value, 'rb').read()) dataList.append(open(value, 'rb').read())
dataList.append(encode(f'--{boundary}--')) dataList.append(encode('--{{boundary}}--'))
dataList.append(encode('')) dataList.append(encode(''))
return dataList return dataList
dataList = build_data_list({{fields_list}}) dataList = build_data_list({{fields_list}})
payload = b'\r\n'.join(dataList) payload = b'\r\n'.join(dataList)
'''; ''';
String kStringRequestParams = """, params=params"""; String kStringRequestParams = """, params=params""";
String kStringRequestBody = """, data=payload"""; String kStringRequestBody = """, data=payload""";
@ -97,9 +90,6 @@ payload = b'\r\n'.join(dataList)
String kStringRequestJson = """, json=payload"""; String kStringRequestJson = """, json=payload""";
String kStringRequestHeaders = """, headers=headers"""; String kStringRequestHeaders = """, headers=headers""";
String kFormDataHeaders = '''
headers['Content-type'] = f'multipart/form-data; boundary={boundary}';
''';
final String kStringRequestEnd = """) final String kStringRequestEnd = """)
@ -118,6 +108,7 @@ print('Response Body:', response.text)
bool hasBody = false; bool hasBody = false;
bool hasJsonBody = false; bool hasJsonBody = false;
List<FormDataModel> formDataList = requestModel.formDataList ?? []; List<FormDataModel> formDataList = requestModel.formDataList ?? [];
String uuid = getNewUuid();
String url = requestModel.url; String url = requestModel.url;
if (!url.contains("://") && url.isNotEmpty) { if (!url.contains("://") && url.isNotEmpty) {
@ -164,6 +155,13 @@ print('Response Body:', response.text)
var headersList = requestModel.requestHeaders; var headersList = requestModel.requestHeaders;
if (headersList != null || hasBody) { if (headersList != null || hasBody) {
var headers = requestModel.headersMap; var headers = requestModel.headersMap;
if (requestModel.isFormDataRequest) {
var formHeaderTemplate =
jj.Template(kTemplateFormHeaderContentType);
headers[HttpHeaders.contentTypeHeader] = formHeaderTemplate.render({
"boundary": uuid,
});
}
if (headers.isNotEmpty || hasBody) { if (headers.isNotEmpty || hasBody) {
hasHeaders = true; hasHeaders = true;
if (hasBody) { if (hasBody) {
@ -177,11 +175,11 @@ print('Response Body:', response.text)
} }
} }
if (requestModel.isFormDataRequest) { if (requestModel.isFormDataRequest) {
result += kFormDataHeaders;
var formDataBodyData = jj.Template(kStringFormDataBody); var formDataBodyData = jj.Template(kStringFormDataBody);
result += formDataBodyData.render( result += formDataBodyData.render(
{ {
"fields_list": json.encode(rowsToFormDataMap(formDataList)), "fields_list": json.encode(rowsToFormDataMap(formDataList)),
"boundary": uuid,
}, },
); );
} }