From 4650c1b3c83c53a226cd82e5c983c9717d0323ef Mon Sep 17 00:00:00 2001
From: Andhieka Putra <andhieka.putra@gmail.com>
Date: Wed, 8 Jan 2020 13:37:49 +0800
Subject: [PATCH] Added copy constructor to model generator
 (non-private-fields)

Create copy constructor to help users who want to use generated models
as immutable value types (copying object instead of directly modifying
variables).
---
 lib/syntax.dart | 29 +++++++++++++++++++++++++----
 1 file changed, 25 insertions(+), 4 deletions(-)

diff --git a/lib/syntax.dart b/lib/syntax.dart
index d4571f8..b00ab32 100644
--- a/lib/syntax.dart
+++ b/lib/syntax.dart
@@ -214,19 +214,23 @@ class ClassDefinition {
     }
   }
 
-  String get _fieldList {
+  String _generateFieldList({int indentLevel = 1, String delimiter = ";"}) {
     return fields.keys.map((key) {
       final f = fields[key];
       final fieldName =
           fixFieldName(key, typeDef: f, privateField: privateFields);
       final sb = new StringBuffer();
-      sb.write('\t');
+      sb.write('\t' * indentLevel);
       _addTypeDef(f, sb);
-      sb.write(' $fieldName;');
+      sb.write(' $fieldName$delimiter');
       return sb.toString();
     }).join('\n');
   }
 
+  String get _fieldList {
+    return _generateFieldList();
+  }
+
   String get _gettersSetters {
     return fields.keys.map((key) {
       final f = fields[key];
@@ -293,6 +297,23 @@ class ClassDefinition {
     return sb.toString();
   }
 
+  String get _copyConstructor {
+    final sb = new StringBuffer();
+    sb.write('\t$name copyWithUpdate({');
+    sb.write(_generateFieldList(indentLevel: 2, delimiter: ","));
+    sb.write('\t}) {');
+    sb.write('\t\treturn $name(');
+    fields.keys.forEach((key) {
+      final f = fields[key];
+      final fieldName =
+          fixFieldName(key, typeDef: f, privateField: privateFields);
+      sb.write('$fieldName: $fieldName ?? this.$fieldName,');
+    });
+    sb.write(');'); // return $name(...
+    sb.write('}');
+    return sb.toString();
+  }
+
   String get _jsonParseFunc {
     final sb = new StringBuffer();
     sb.write('\t$name');
@@ -320,7 +341,7 @@ class ClassDefinition {
     if (privateFields) {
       return 'class $name {\n$_fieldList\n\n$_defaultPrivateConstructor\n\n$_gettersSetters\n\n$_jsonParseFunc\n\n$_jsonGenFunc\n}\n';
     } else {
-      return 'class $name {\n$_fieldList\n\n$_defaultConstructor\n\n$_jsonParseFunc\n\n$_jsonGenFunc\n}\n';
+      return 'class $name {\n$_fieldList\n\n$_defaultConstructor\n\n$_copyConstructor\n\n$_jsonParseFunc\n\n$_jsonGenFunc\n}\n';
     }
   }
 }