/*
 * Copyright (C) 2018 The Dagger Authors.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package dagger.internal.codegen;

import dagger.Module;
import dagger.Provides;
import dagger.internal.GenerationOptions;

/** Adds bindings for serializing and rereading {@link GenerationOptions}. */
@Module
interface GenerationOptionsModule {
  @Provides
  @PerComponentImplementation
  @GenerationCompilerOptions
  static CompilerOptions generationOptions(
      CompilerOptions defaultOptions,
      ComponentImplementation componentImplementation,
      DaggerElements elements) {
    return componentImplementation
        .baseImplementation()
        .map(baseImplementation -> elements.getTypeElement(baseImplementation.name()))
        .map(typeElement -> typeElement.getAnnotation(GenerationOptions.class))
        .map(defaultOptions::withGenerationOptions)
        .orElse(defaultOptions);
  }
}

@@ -718,16 +718,43 @@
 Options;
+%0Aimport java.util.Optional;
 %0A%0A/** Ad
@@ -1118,155 +1118,1068 @@
     
-return componentImplementation%0A        .baseImplementation()%0A        .map(baseImplementation -%3E elements.getTypeElement(baseImplementation.name()))
+// Inspect the base implementation for the @GenerationOptions annotation. Even if%0A    // componentImplementation is the base implementation, inspect it for the case where we are%0A    // recomputing the ComponentImplementation from a previous compilation.%0A    // TODO(b/117833324): consider adding a method that returns baseImplementation.orElse(this).%0A    // The current state of the world is a little confusing and maybe not intuitive: the base%0A    // implementation has no base implementation, but it _is_ a base implementation.%0A    return Optional.of(componentImplementation.baseImplementation().orElse(componentImplementation))%0A        .map(baseImplementation -%3E elements.getTypeElement(baseImplementation.name()))%0A        // If this returns null, the type has not been generated yet and Optional will switch to an%0A        // empty state. This means that we're currently generating componentImplementation, or that%0A        // the base implementation is being generated in this round, and thus the options passed to%0A        // this compilation are applicable
 %0A   

Fix the @GenerationOptions retrieval and enable a build_test for the dagger.functional.aot.fastinit libraries so that we catch this in presubmit.