// Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
// SPDX-License-Identifier: Apache-2.0

import 'package:code_builder/code_builder.dart';
import 'package:smithy/ast.dart';
import 'package:smithy_codegen/smithy_codegen.dart';
import 'package:smithy_codegen/src/generator/generator.dart';

/// Generates the top-level library file which simply exports all the public
/// types generated in the service's closure. This includes all operations,
/// models, serializers, and the service client -- basically everything except
/// the generated payload types which are meant to be an internal abstraction.
class ServiceGenerator extends LibraryGenerator<ServiceShape> {
  ServiceGenerator(
    super.shape,
    CodegenContext context, {
    super.smithyLibrary,
  }) : super(
          context: context,
        );

  @override
  Library generate() {
    builder.directives.addAll([
      for (final type in context.generatedTypes.entries)
        Directive.export(type.key.url!, hide: type.value.toList()),
      Directive.export(context.serviceClientLibrary.libraryUrl),
      if (context.generateServer)
        Directive.export(context.serviceServerLibrary.libraryUrl),
    ]);

    return builder.build();
  }
}