cn.taketoday.aot.test.generate.CompilerFiles Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of today-core-test Show documentation
Infra Core Test
There is a newer version: 5.0.0-Draft.1
/*
 * Original Author -> Harry Yang ([email protected]) https://taketoday.cn
 * Copyright © Harry Yang & 2017 - 2023 All Rights Reserved.
 *
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see [http://www.gnu.org/licenses/]
 */

package cn.taketoday.aot.test.generate;

import java.util.ArrayList;
import java.util.List;
import java.util.function.BiFunction;
import java.util.function.UnaryOperator;

import cn.taketoday.aot.generate.GeneratedFiles;
import cn.taketoday.aot.generate.GeneratedFiles.Kind;
import cn.taketoday.aot.generate.InMemoryGeneratedFiles;
import cn.taketoday.core.io.InputStreamSource;
import cn.taketoday.core.test.tools.ClassFile;
import cn.taketoday.core.test.tools.ResourceFile;
import cn.taketoday.core.test.tools.SourceFile;
import cn.taketoday.core.test.tools.TestCompiler;

/**
 * Adapter class that can be used to apply AOT {@link GeneratedFiles} to the
 * {@link TestCompiler}.
 *
 * @author Stephane Nicoll
 * @author Phillip Webb
 * @since 4.0
 */
public final class CompilerFiles implements UnaryOperator {

  private final InMemoryGeneratedFiles generatedFiles;

  private CompilerFiles(InMemoryGeneratedFiles generatedFiles) {
    this.generatedFiles = generatedFiles;
  }

  public static UnaryOperator from(
          InMemoryGeneratedFiles generatedFiles) {
    return new CompilerFiles(generatedFiles);
  }

  @Override
  public TestCompiler apply(TestCompiler testCompiler) {
    return testCompiler
            .withSources(adapt(Kind.SOURCE, (path, inputStreamSource) ->
                    SourceFile.of(inputStreamSource)))
            .withResources(adapt(Kind.RESOURCE, ResourceFile::of))
            .withClasses(adapt(Kind.CLASS, (path, inputStreamSource) ->
                    ClassFile.of(ClassFile.toClassName(path), inputStreamSource)));
  }

  private  List adapt(Kind kind,
          BiFunction adapter) {
    List result = new ArrayList<>();
    this.generatedFiles.getGeneratedFiles(kind)
            .forEach((k, v) -> result.add(adapter.apply(k, v)));
    return result;
  }

}