io.codemodder.providers.sarif.appscan.AppScanModule Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of codemodder-plugin-appscan Show documentation

Plugin to enable the use of appscan in codemods

There is a newer version: 0.98.0

Show newest version

package io.codemodder.providers.sarif.appscan;

import com.google.inject.AbstractModule;
import io.codemodder.CodeChanger;
import io.codemodder.RuleSarif;
import java.lang.reflect.Constructor;
import java.util.List;
import java.util.Map;
import java.util.Objects;
import java.util.Optional;
import java.util.stream.Collectors;
import java.util.stream.Stream;

/** Responsible for distributing the SARIFS to AppScan based codemods based on rules. */
public final class AppScanModule extends AbstractModule {

  private final List> codemodTypes;
  private final List allAppScanRuleSarifs;

  public AppScanModule(
      final List> codemodTypes, final List sarifs) {
    this.codemodTypes = Objects.requireNonNull(codemodTypes);
    this.allAppScanRuleSarifs = sarifs;
  }

  @Override
  protected void configure() {
    final Map map =
        allAppScanRuleSarifs.stream()
            .collect(Collectors.toUnmodifiableMap(RuleSarif::getRule, rs -> rs));

    for (final Class codemodType : codemodTypes) {
      final Constructor[] constructors = codemodType.getDeclaredConstructors();

      final Optional annotation =
          Stream.of(constructors)
              .filter(constructor -> constructor.getAnnotation(javax.inject.Inject.class) != null)
              .flatMap(constructor -> Stream.of(constructor.getParameters()))
              .map(parameter -> parameter.getAnnotation(ProvidedAppScanScan.class))
              .filter(Objects::nonNull)
              .findFirst();

      annotation.ifPresent(
          providedAppScanScan -> {
            if (!providedAppScanScan.ruleName().isEmpty()) {
              bind(RuleSarif.class)
                  .annotatedWith(providedAppScanScan)
                  .toInstance(map.getOrDefault(providedAppScanScan.ruleName(), RuleSarif.EMPTY));
            } else if (providedAppScanScan.ruleNames().length > 0) {

              RuleSarif ruleSarif = RuleSarif.EMPTY;
              for (final String ruleName : providedAppScanScan.ruleNames()) {
                final var result = map.get(ruleName);
                if (result != null) {
                  ruleSarif = result;
                  break;
                }
              }

              bind(RuleSarif.class).annotatedWith(providedAppScanScan).toInstance(ruleSarif);
            } else {
              throw new IllegalStateException("No rule name provided in " + providedAppScanScan);
            }
          });
    }
  }
}