SbmAdapterRecipe.java

/*
 * Copyright 2021 - 2023 the original author or 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
 *
 *      https://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 example.recipe;


import freemarker.template.Configuration;
import org.openrewrite.*;
import org.openrewrite.internal.lang.Nullable;
import org.openrewrite.maven.MavenSettings;
import org.openrewrite.maven.cache.LocalMavenArtifactCache;
import org.springframework.context.ApplicationContext;
import org.springframework.context.annotation.AnnotationConfigApplicationContext;
import org.springframework.rewrite.parser.JavaParserBuilder;
import org.springframework.rewrite.parser.maven.MavenSettingsInitializer;
import org.springframework.rewrite.parser.maven.RewriteMavenArtifactDownloader;
import org.springframework.rewrite.resource.*;
import org.springframework.rewrite.scopes.ProjectMetadata;
import org.springframework.sbm.build.impl.MavenBuildFileRefactoringFactory;
import org.springframework.sbm.build.impl.RewriteMavenParser;
import org.springframework.sbm.build.resource.BuildFileResourceWrapper;
import org.springframework.sbm.engine.context.ProjectContext;
import org.springframework.sbm.engine.context.ProjectContextFactory;
import org.springframework.sbm.engine.recipe.RewriteRecipeLoader;
import org.springframework.sbm.java.JavaSourceProjectResourceWrapper;
import org.springframework.sbm.java.refactoring.JavaRefactoringFactory;
import org.springframework.sbm.java.refactoring.JavaRefactoringFactoryImpl;
import org.springframework.sbm.java.util.BasePackageCalculator;
import org.springframework.sbm.jee.jaxrs.MigrateJaxRsRecipe;
import org.springframework.sbm.jee.jaxrs.actions.ConvertJaxRsAnnotations;
import org.springframework.sbm.project.resource.ProjectResourceSetHolder;
import org.springframework.sbm.project.resource.ProjectResourceWrapper;
import org.springframework.sbm.project.resource.ProjectResourceWrapperRegistry;
import org.springframework.sbm.project.resource.SbmApplicationProperties;

import java.nio.file.Path;
import java.util.ArrayList;
import java.util.Collection;
import java.util.List;

/**
 * @author Fabian Krüger
 */
public class SbmAdapterRecipe extends ScanningRecipe<List<SourceFile>> {

    private ProjectResourceSetFactory projectResourceSetFactory;
    private ProjectContextFactory projectContextFactory;

    @Override
    public @NlsRewrite.DisplayName String getDisplayName() {
        return "";
    }

    @Override
    public @NlsRewrite.Description String getDescription() {
        return "";
    }

    public SbmAdapterRecipe() {

    }

    @Override
    public List<SourceFile> getInitialValue(ExecutionContext ctx) {
        return new ArrayList<>();
    }

    @Override
    public TreeVisitor<?, ExecutionContext> getScanner(List<SourceFile> acc) {
        return new TreeVisitor<Tree, ExecutionContext>() {
            @Override
            public @Nullable Tree visit(@Nullable Tree tree, ExecutionContext executionContext) {
                if(tree instanceof SourceFile) {
                    acc.add((SourceFile) tree);
                }
                return super.visit(tree, executionContext);
            }
        };
    }

    @Override
    public Collection<? extends SourceFile> generate(List<SourceFile> generate, ExecutionContext executionContext) {
        // create the required classes
        initBeans();

        // transform nodes to SourceFiles
        List<SourceFile> sourceFiles = generate;

        // FIXME: base dir calculation is fake
        Path baseDir = Path.of(".").resolve("testcode/jee/jaxrs/bootify-jaxrs/given").toAbsolutePath().normalize(); //executionContext.getMessage("base.dir");

        // Create the SBM resource set abstraction
        ProjectResourceSet projectResourceSet = projectResourceSetFactory.create(baseDir, sourceFiles);
        // Create the SBM ProjectContext
        ProjectContext pc = projectContextFactory.createProjectContext(baseDir, projectResourceSet);

        // Execute the SBM Action = the JAXRS Recipe
//        new ConvertJaxRsAnnotations().apply(pc);

        RewriteRecipeLoader recipeLoader = new RewriteRecipeLoader();
        new MigrateJaxRsRecipe().jaxRs(recipeLoader).apply(pc);

        List<? extends SourceFile> list = pc.getProjectResources().stream()
                .map(pr -> pr.getSourceFile())
                .toList();

        return list;
    }

    private void initBeans() {
        // ExecutionContext is not retrieved from OpenRewrite here
        AnnotationConfigApplicationContext ctx = new AnnotationConfigApplicationContext("org.springframework.freemarker", "org.springframework.sbm", "org.springframework.rewrite");
        ctx.register(Configuration.class);
        this.projectContextFactory = ctx.getBean(ProjectContextFactory.class);
        this.projectResourceSetFactory = ctx.getBean(ProjectResourceSetFactory.class);
    }

}