write method

String write()

Implementation

String write() {
  final sb = StringBuffer();
  final imports = <String>{}
    ..add("import 'package:json_annotation/json_annotation.dart';");
  for (final element
      in (jsonModel.extraImports ?? pubspecConfig.extraImports)) {
    imports.add('import \'$element\';');
  }
  final extendsModel = jsonModel.extendsModel;

  if (extendsModel != null) {
    if (!TypeChecker.isKnownDartType(extendsModel)) {
      imports.addAll(_getImportsFromPath(extendsModel));
    }
  }

  for (final field in jsonModel.fields) {
    final type = field.type;
    if (!TypeChecker.isKnownDartType(type.name) &&
        type.name != jsonModel.name) {
      imports.addAll(_getImportsFromPath(type.name));
    }
    if (type is MapType && !TypeChecker.isKnownDartType(type.valueName)) {
      imports.addAll(_getImportsFromPath(type.valueName));
    }
  }
  for (final field in extendsFields) {
    imports.addAll(_getImportsFromField(field));
  }

  for (final converter in jsonModel.converters) {
    imports.addAll(_getImportsFromPath(converter));
  }
  (imports.toList()..sort((i1, i2) => i1.compareTo(i2))).forEach(sb.writeln);

  sb
    ..writeln()
    ..writeln("part '${jsonModel.fileName}.g.dart';")
    ..writeln();

  final modelDescription = jsonModel.description?.trim();
  if (modelDescription != null && modelDescription.isNotEmpty) {
    sb.writeln("///$modelDescription");
  }

  if (jsonModel.explicitToJson ?? pubspecConfig.explicitToJson) {
    sb.writeln('@JsonSerializable(explicitToJson: true)');
  } else {
    sb.writeln('@JsonSerializable()');
  }
  (jsonModel.extraAnnotations ?? pubspecConfig.extraAnnotations)
      .forEach(sb.writeln);

  for (final converter in jsonModel.converters) {
    sb.writeln('@$converter()');
  }

  if (extendsModel != null) {
    sb.writeln('class ${jsonModel.name} extends $extendsModel {');
  } else {
    sb.writeln('class ${jsonModel.name} {');
  }

  jsonModel.fields.sort((a, b) {
    final b1 = a.isRequired ? 1 : 0;
    final b2 = b.isRequired ? 1 : 0;
    return b2 - b1;
  });

  for (final key in jsonModel.fields) {
    final description = key.description;
    if (description != null) {
      sb.writeln('  ///$description');
    }
    sb.write("  @JsonKey(name: '${key.serializedName}'");
    if (key.isRequired) {
      if (key.hasDefaultValue) {
        sb.write(', required: false, disallowNullValue: true');
      } else {
        sb.write(', required: true');
      }
    }

    if (!key.includeIfNull) {
      sb.write(', includeIfNull: false');
    }

    if (key.ignore) {
      sb.write(', ignore: true');
    }

    if (key.unknownEnumValue != null) {
      sb.write(
          ', unknownEnumValue: ${key.type.name}.${key.unknownEnumValue}');
    }

    final fieldModel = yamlConfig.getModelByName(key.type);
    if (key.fromJson != null) {
      sb.write(', fromJson: ${key.fromJson}');
    } else if (fieldModel is CustomFromToJsonModel) {
      sb.write(', fromJson: handle${fieldModel.name}FromJson');
    }
    if (key.toJson != null) {
      sb.write(', toJson: ${key.toJson}');
    } else if (fieldModel is CustomFromToJsonModel) {
      sb.write(', toJson: handle${fieldModel.name}ToJson');
    }
    sb.writeln(')');
    if (key.nonFinal) {
      sb.write('  ');
    } else {
      sb.write('  final ');
    }
    sb.writeln('${_getKeyType(key)} ${key.name};');
  }

  final anyNonFinal = jsonModel.fields.any((element) => element.nonFinal) ||
      extendsFields.any((element) => element.nonFinal);
  sb
    ..writeln()
    ..writeln('  ${anyNonFinal ? '' : 'const '}${jsonModel.name}({');

  for (final key in jsonModel.fields
      .where((key) => (key.isRequired && !key.hasDefaultValue))) {
    sb.writeln('    required this.${key.name}${_fillDefaulValue(key)},');
  }
  for (final key in extendsFields
      .where((key) => (key.isRequired && !key.hasDefaultValue))) {
    sb.writeln(
        '    required ${_getKeyType(key)} ${key.name}${_fillDefaulValue(key)},');
  }
  for (final key in jsonModel.fields
      .where((key) => !(key.isRequired && !key.hasDefaultValue))) {
    sb.writeln('    this.${key.name}${_fillDefaulValue(key)},');
  }
  for (final key in extendsFields
      .where((key) => !(key.isRequired && !key.hasDefaultValue))) {
    sb.writeln(
        '    ${_getKeyType(key)} ${key.name}${_fillDefaulValue(key)},');
  }
  if (extendsModel != null) {
    sb.writeln('  }) : super(');
    for (final key in extendsFields) {
      sb.writeln('          ${key.name}: ${key.name},');
    }
    sb
      ..writeln('        );')
      ..writeln();
  } else {
    sb
      ..writeln('  });')
      ..writeln();
  }
  if (jsonModel.generateForGenerics) {
    sb.writeln(
        '  factory ${jsonModel.name}.fromJson(Object? json) => _\$${jsonModel.name}FromJson(json as Map<String, dynamic>); // ignore: avoid_as');
  } else {
    sb.writeln(
        '  factory ${jsonModel.name}.fromJson(Map<String, dynamic> json) => _\$${jsonModel.name}FromJson(json);');
  }
  sb.writeln();
  if (extendsModel != null) {
    sb.writeln('  @override');
  }
  sb.writeln(
      '  Map<String, dynamic> toJson() => _\$${jsonModel.name}ToJson(this);');

  if (jsonModel.staticCreate ?? pubspecConfig.staticCreate) {
    sb
      ..writeln()
      ..writeln('  // ignore: prefer_constructors_over_static_methods')
      ..writeln(
          '  static ${jsonModel.name} create(${jsonModel.generateForGenerics ? 'Object? json' : 'Map<String, dynamic> json'}) => ${jsonModel.name}.fromJson(json);');
  }

  if (jsonModel.equalsAndHashCode ?? pubspecConfig.equalsHashCode) {
    sb
      ..writeln()
      ..writeln('  @override')
      ..writeln('  bool operator ==(Object other) =>')
      ..writeln('      identical(this, other) ||')
      ..writeln('      other is ${jsonModel.name} &&')
      ..write('          runtimeType == other.runtimeType');
    for (final field
        in jsonModel.fields.where((field) => !field.ignoreEquality)) {
      sb.write(' &&\n          ${field.name} == other.${field.name}');
    }
    if (extendsModel != null) {
      sb.write(' &&\n          super == other');
    }
    sb
      ..writeln(';')
      ..writeln()
      ..writeln('  @override')
      ..writeln('  int get hashCode =>');
    var c = 0;
    for (final field
        in jsonModel.fields.where((field) => !field.ignoreEquality)) {
      if (c++ > 0) sb.write(' ^\n');
      sb.write('      ${field.name}.hashCode');
    }
    if (extendsModel != null) {
      sb.write(' ^ \n      super.hashCode');
    }
    if (c == 0) sb.write('      0');
    sb.writeln(';');
  }
  if (jsonModel.generateToString ?? pubspecConfig.generateToString) {
    sb
      ..writeln()
      ..writeln('  @override')
      ..writeln('  String toString() =>')
      ..writeln('      \'${jsonModel.name}{\'');

    var c = 0;
    for (final field in jsonModel.fields) {
      if (c++ > 0) sb.writeln(', \'');
      sb.write('      \'${field.name}: \$${field.name}');
    }
    for (final field in extendsFields) {
      if (c++ > 0) sb.writeln(', \'');
      sb.write('      \'${field.name}: \$${field.name}');
    }
    sb.writeln('\'\n      \'}\';');
  }

  sb
    ..writeln()
    ..writeln('}');

  if (pubspecConfig.retrofitMappers) {
    createRetrofitMappers(sb);
  }

  return sb.toString();
}