io.codemodder.remediation.javadeserialization.JavaDeserializationRemediator Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of codemodder-base Show documentation
Base framework for writing codemods in Java
The newest version!
package io.codemodder.remediation.javadeserialization;

import com.github.javaparser.ast.CompilationUnit;
import com.github.javaparser.ast.expr.MethodCallExpr;
import com.github.javaparser.ast.expr.VariableDeclarationExpr;
import io.codemodder.CodemodFileScanningResult;
import io.codemodder.codetf.DetectorRule;
import io.codemodder.remediation.*;
import java.util.Collection;
import java.util.Optional;
import java.util.function.Function;

/**
 * Fixes issues of related to object deserialization
 *
 * @param 
 */
public final class JavaDeserializationRemediator implements Remediator {

  private final SearcherStrategyRemediator searchStrategyRemediator;

  public JavaDeserializationRemediator() {
    this.searchStrategyRemediator =
        new SearcherStrategyRemediator.Builder()
            .withSearcherStrategyPair(
                // matches declarations
                new FixCandidateSearcher.Builder()
                    .withMatcher(
                        n ->
                            Optional.empty()
                                .or(
                                    () ->
                                        Optional.of(n)
                                            .map(
                                                m ->
                                                    m instanceof VariableDeclarationExpr vde
                                                        ? vde
                                                        : null)
                                            .filter(JavaDeserializationFixStrategy::match))
                                .or(
                                    () ->
                                        Optional.of(n)
                                            .map(m -> m instanceof MethodCallExpr mce ? mce : null)
                                            .filter(JavaDeserializationFixStrategy::match))
                                .isPresent())
                    .build(),
                new JavaDeserializationFixStrategy())
            .build();
  }

  @Override
  public CodemodFileScanningResult remediateAll(
      CompilationUnit cu,
      String path,
      DetectorRule detectorRule,
      Collection findingsForPath,
      Function findingIdExtractor,
      Function findingStartLineExtractor,
      Function> findingEndLineExtractor,
      Function> findingColumnExtractor) {
    return searchStrategyRemediator.remediateAll(
        cu,
        path,
        detectorRule,
        findingsForPath,
        findingIdExtractor,
        findingStartLineExtractor,
        findingEndLineExtractor,
        findingColumnExtractor);
  }
}