CodeGen class

This commit is contained in:
Ankit Mahato
2023-03-22 23:57:55 +05:30
parent 5c39ffd34f
commit a8fbbc96d2
2 changed files with 109 additions and 113 deletions

View File

@ -2,77 +2,79 @@ import 'dart:io';
import 'dart:convert'; import 'dart:convert';
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' show padMultilineString;
import 'package:apidash/models/models.dart' show RequestModel, rowsToMap; import 'package:apidash/models/models.dart' show RequestModel, rowsToMap;
String kTemplateUrl = """import 'package:http/http.dart' as http; class DartHttpCodeGen {
String kTemplateUrl = """import 'package:http/http.dart' as http;
void main() async { void main() async {
var uri = Uri.parse('{{url}}'); var uri = Uri.parse('{{url}}');
"""; """;
String kTemplateParams = """ String kTemplateParams = """
var queryParams = {{params}}; var queryParams = {{params}};
"""; """;
int kParamsPadding = 20; int kParamsPadding = 20;
String kStringUrlParams = """ String kStringUrlParams = """
var urlQueryParams = Map<String,String>.from(uri.queryParameters); var urlQueryParams = Map<String,String>.from(uri.queryParameters);
urlQueryParams.addAll(queryParams); urlQueryParams.addAll(queryParams);
uri = uri.replace(queryParameters: urlQueryParams); uri = uri.replace(queryParameters: urlQueryParams);
"""; """;
String kStringNoUrlParams = """ String kStringNoUrlParams = """
uri = uri.replace(queryParameters: queryParams); uri = uri.replace(queryParameters: queryParams);
"""; """;
String kTemplateBody = """ String kTemplateBody = """
String body = r'''{{body}}'''; String body = r'''{{body}}''';
"""; """;
String kBodyImportDartConvert = """ String kBodyImportDartConvert = """
import 'dart:convert'; import 'dart:convert';
"""; """;
String kBodyLength = """ String kBodyLength = """
var contentLength = utf8.encode(body).length; var contentLength = utf8.encode(body).length;
"""; """;
String kTemplateHeaders = """ String kTemplateHeaders = """
var headers = {{headers}}; var headers = {{headers}};
"""; """;
int kHeadersPadding = 16; int kHeadersPadding = 16;
String kTemplateRequest = """ String kTemplateRequest = """
final response = await http.{{method}}(uri"""; final response = await http.{{method}}(uri""";
String kStringRequestHeaders = """, String kStringRequestHeaders = """,
headers: headers"""; headers: headers""";
String kStringRequestBody = """, String kStringRequestBody = """,
body: body"""; body: body""";
String kStringRequestEnd = """); String kStringRequestEnd = """);
"""; """;
String kTemplateSingleSuccess = """ String kTemplateSingleSuccess = """
if (response.statusCode == {{code}}) { if (response.statusCode == {{code}}) {
"""; """;
String kTemplateMultiSuccess = """ String kTemplateMultiSuccess = """
if ({{codes}}.contains(response.statusCode)) {\n"""; if ({{codes}}.contains(response.statusCode)) {\n""";
String kStringResult = r""" String kStringResult = r"""
print('Status Code: ${response.statusCode}'); print('Status Code: ${response.statusCode}');
print('Result: ${response.body}'); print('Result: ${response.body}');
@ -83,107 +85,99 @@ String kStringResult = r"""
} }
"""; """;
String padMultilineString(String text, int padding, String? getCode(RequestModel requestModel) {
{bool firstLinePadded = false}) { try {
var lines = kSplitter.convert(text); String result = "";
int start = firstLinePadded ? 0 : 1; bool hasHeaders = false;
for (start; start < lines.length; start++) { bool hasBody = false;
lines[start] = ' ' * padding + lines[start];
}
return lines.join("\n");
}
String? getDartHttpCode(RequestModel requestModel) { String url = requestModel.url;
try { if (!url.contains("://") && url.isNotEmpty) {
String result = ""; url = kDefaultUriScheme + url;
bool hasHeaders = false;
bool hasBody = false;
String url = requestModel.url;
if (!url.contains("://") && url.isNotEmpty) {
url = kDefaultUriScheme + url;
}
var templateUrl = jj.Template(kTemplateUrl);
result += templateUrl.render({"url": url});
var paramsList = requestModel.requestParams;
if (paramsList != null) {
var params = rowsToMap(requestModel.requestParams) ?? {};
if (params.isNotEmpty) {
var templateParams = jj.Template(kTemplateParams);
var paramsString = kEncoder.convert(params);
paramsString = padMultilineString(paramsString, kParamsPadding);
result += templateParams.render({"params": paramsString});
Uri uri = Uri.parse(url);
if (uri.hasQuery) {
result += kStringUrlParams;
} else {
result += kStringNoUrlParams;
}
} }
} var templateUrl = jj.Template(kTemplateUrl);
result += templateUrl.render({"url": url});
var method = requestModel.method; var paramsList = requestModel.requestParams;
if (kMethodsWithBody.contains(method) && requestModel.requestBody != null) { if (paramsList != null) {
var contentLength = utf8.encode(requestModel.requestBody).length; var params = rowsToMap(requestModel.requestParams) ?? {};
if (contentLength > 0) { if (params.isNotEmpty) {
hasBody = true; var templateParams = jj.Template(kTemplateParams);
var body = requestModel.requestBody; var paramsString = kEncoder.convert(params);
var templateBody = jj.Template(kTemplateBody); paramsString = padMultilineString(paramsString, kParamsPadding);
result += templateBody.render({"body": body}); result += templateParams.render({"params": paramsString});
result = kBodyImportDartConvert + result; Uri uri = Uri.parse(url);
result += kBodyLength; if (uri.hasQuery) {
} result += kStringUrlParams;
} } else {
result += kStringNoUrlParams;
var headersList = requestModel.requestHeaders;
if (headersList != null || hasBody) {
var headers = rowsToMap(requestModel.requestHeaders) ?? {};
if (headers.isNotEmpty || hasBody) {
hasHeaders = true;
if (hasBody) {
headers[HttpHeaders.contentLengthHeader] = r"$contentLength";
switch (requestModel.requestBodyContentType) {
case ContentType.json:
headers[HttpHeaders.contentTypeHeader] = 'application/json';
break;
case ContentType.text:
headers[HttpHeaders.contentTypeHeader] = 'text/plain';
break;
} }
} }
var headersString = kEncoder.convert(headers);
headersString = padMultilineString(headersString, kHeadersPadding);
var templateHeaders = jj.Template(kTemplateHeaders);
result += templateHeaders.render({"headers": headersString});
} }
var method = requestModel.method;
if (kMethodsWithBody.contains(method) &&
requestModel.requestBody != null) {
var contentLength = utf8.encode(requestModel.requestBody).length;
if (contentLength > 0) {
hasBody = true;
var body = requestModel.requestBody;
var templateBody = jj.Template(kTemplateBody);
result += templateBody.render({"body": body});
result = kBodyImportDartConvert + result;
result += kBodyLength;
}
}
var headersList = requestModel.requestHeaders;
if (headersList != null || hasBody) {
var headers = rowsToMap(requestModel.requestHeaders) ?? {};
if (headers.isNotEmpty || hasBody) {
hasHeaders = true;
if (hasBody) {
headers[HttpHeaders.contentLengthHeader] = r"$contentLength";
switch (requestModel.requestBodyContentType) {
case ContentType.json:
headers[HttpHeaders.contentTypeHeader] = 'application/json';
break;
case ContentType.text:
headers[HttpHeaders.contentTypeHeader] = 'text/plain';
break;
}
}
var headersString = kEncoder.convert(headers);
headersString = padMultilineString(headersString, kHeadersPadding);
var templateHeaders = jj.Template(kTemplateHeaders);
result += templateHeaders.render({"headers": headersString});
}
}
var templateRequest = jj.Template(kTemplateRequest);
result += templateRequest.render({"method": method.name});
if (hasHeaders) {
result += kStringRequestHeaders;
}
if (hasBody) {
result += kStringRequestBody;
}
result += kStringRequestEnd;
var success = kCodegenSuccessStatusCodes[method]!;
if (success.length > 1) {
var templateMultiSuccess = jj.Template(kTemplateMultiSuccess);
result += templateMultiSuccess.render({"codes": success});
} else {
var templateSingleSuccess = jj.Template(kTemplateSingleSuccess);
result += templateSingleSuccess.render({"code": success[0]});
}
result += kStringResult;
return result;
} catch (e) {
return null;
} }
var templateRequest = jj.Template(kTemplateRequest);
result += templateRequest.render({"method": method.name});
if (hasHeaders) {
result += kStringRequestHeaders;
}
if (hasBody) {
result += kStringRequestBody;
}
result += kStringRequestEnd;
var success = kCodegenSuccessStatusCodes[method]!;
if (success.length > 1) {
var templateMultiSuccess = jj.Template(kTemplateMultiSuccess);
result += templateMultiSuccess.render({"codes": success});
} else {
var templateSingleSuccess = jj.Template(kTemplateSingleSuccess);
result += templateSingleSuccess.render({"code": success[0]});
}
result += kStringResult;
return result;
} catch (e) {
return null;
} }
} }

View File

@ -13,6 +13,8 @@ class CodePane extends ConsumerStatefulWidget {
} }
class _CodePaneState extends ConsumerState<CodePane> { class _CodePaneState extends ConsumerState<CodePane> {
final DartHttpCodeGen dartHttpCodeGen = DartHttpCodeGen();
@override @override
void initState() { void initState() {
super.initState(); super.initState();
@ -39,7 +41,7 @@ class _CodePaneState extends ConsumerState<CodePane> {
borderRadius: kBorderRadius8, borderRadius: kBorderRadius8,
); );
final code = getDartHttpCode(requestModel); final code = dartHttpCodeGen.getCode(requestModel);
if (code == null) { if (code == null) {
return const ErrorMessage( return const ErrorMessage(
message: "An error was encountered while generating code. $kRaiseIssue", message: "An error was encountered while generating code. $kRaiseIssue",