updates

lib/main/djs/code_djs/code_djs.dart

@@ -4,3 +4,4 @@ export 'function_call.dart';
 export 'function.dart';
 export 'if_else.dart';
 export 'stateless_widget.dart';
+export 'return.dart';

lib/main/djs/code_djs/code_part.dart

@@ -9,6 +9,7 @@ enum CodePartType {
   IfElse,
   Import,
   StatelessWidget,
+  Return,
 }
 
 @JsonSerializable()
@@ -38,6 +39,8 @@ class CodePartDj {
         return ImportDj.fromJson(json);
       case CodePartType.StatelessWidget:
         return StatelessWidgetDj.fromJson(json);
+      case CodePartType.Return:
+        return ReturnDj.fromJson(json);
       default:
         throw Exception(
             'CodePartDj.fromJson not handled for ${codePartDj.type}');

lib/main/djs/code_djs/return.dart

@@ -0,0 +1,34 @@
+import 'package:json_annotation/json_annotation.dart';
+import 'code_part.dart';
+
+part 'return.g.dart';
+
+@JsonSerializable()
+class ReturnDj extends CodePartDj {
+  @JsonKey(name: 'returnStr')
+  final String returnStr;
+
+  ReturnDj({
+    description,
+    this.returnStr = '',
+    CodePartType type = CodePartType.Return,
+  }) : super(
+          description: description,
+          type: type,
+        );
+
+  factory ReturnDj.fromJson(Map<String, dynamic> json) =>
+      _$ReturnDjFromJson(json);
+  @override
+  Map<String, dynamic> toJson() => _$ReturnDjToJson(this);
+
+  @override
+  List<String> lines() {
+    var _lines = super.lines();
+
+    var returnLine = 'return $returnStr;';
+
+    _lines.add(returnLine);
+    return _lines;
+  }
+}

lib/main/djs/code_djs/stateless_widget.dart

@@ -12,7 +12,7 @@ class StatelessWidgetDj extends CodePartDj {
   final List<FunctionArg>? args;
 
   @JsonKey(name: 'body')
-  final CodePartDj? body;
+  final List<CodePartDj>? body;
 
   StatelessWidgetDj({
     description,
@@ -34,7 +34,10 @@ class StatelessWidgetDj extends CodePartDj {
   List<String> lines() {
     var _lines = super.lines();
 
-    var bodyLines = body?.lines() ?? [];
+    var bodyLines = [];
+    body?.forEach((body) {
+      bodyLines += body.lines();
+    });
     var bodyStr = bodyLines.join(' ');
 
     var contructorArgs = args?.map((e) => e.asConstructorArg()).toList() ?? [];
@@ -47,8 +50,7 @@ class StatelessWidgetDj extends CodePartDj {
     var widgetLine = '';
     widgetLine += 'class $name extends StatelessWidget {';
     widgetLine += '$paramtersStr $name($argsLine);';
-    widgetLine +=
-        '@override Widget build(BuildContext context) { $bodyStr return Container();}';
+    widgetLine += '@override Widget build(BuildContext context) { $bodyStr }';
     widgetLine += '}';
 
     _lines.add(widgetLine);