pkgs/messages_builder/lib/code_generation/import_generation.dart

// Copyright (c) 2023, the Dart project authors.  Please see the AUTHORS file
// for details. All rights reserved. Use of this source code is governed by a
// BSD-style license that can be found in the LICENSE file.

import 'package:code_builder/code_builder.dart';

import '../generation_options.dart';

class ImportGeneration {
  final GenerationOptions options;

  ImportGeneration(this.options);

  List<Directive> generate() {
    final serializationImports = switch (options.deserialization) {
      DeserializationType.web => [
          Directive.import('package:messages/messages_json.dart')
        ],
    };
    final pluralImports = switch (options.pluralSelector) {
      PluralSelectorType.intl => [Directive.import('package:intl/intl.dart')],
      PluralSelectorType.intl4x => [
          Directive.import('package:intl4x/intl4x.dart')
        ],
      PluralSelectorType.custom => <Directive>[],
    };
    return [
      ...serializationImports,
      ...pluralImports,
      if (options.findById ||
          options.messageCalls ||
          options.indexType == IndexType.enumerate)
        Directive.import('package:meta/meta.dart'),
      Directive.import('dart:ffi'),
    ];
  }
}