package jadx.tests.functional;

import java.util.ArrayList;
import java.util.HashSet;
import java.util.List;
import java.util.Set;

import org.junit.jupiter.api.Test;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import jadx.api.JadxArgs;
import jadx.core.Jadx;
import jadx.core.dex.visitors.IDexTreeVisitor;
import jadx.core.dex.visitors.JadxVisitor;

import static jadx.tests.api.utils.assertj.JadxAssertions.assertThat;

public class JadxVisitorsOrderTest {
	private static final Logger LOG = LoggerFactory.getLogger(JadxVisitorsOrderTest.class);

	@Test
	public void testOrder() {
		checkPassList(Jadx.getPassesList(new JadxArgs()));
		checkPassList(Jadx.getPreDecompilePassesList());
		checkPassList(Jadx.getFallbackPassesList());
	}

	private void checkPassList(List<IDexTreeVisitor> passes) {
		List<String> errors = check(passes);
		for (String str : errors) {
			LOG.error(str);
		}
		assertThat(errors).isEmpty();
	}

	private static List<String> check(List<IDexTreeVisitor> passes) {
		List<Class<?>> classList = new ArrayList<>(passes.size());
		for (IDexTreeVisitor pass : passes) {
			classList.add(pass.getClass());
		}
		List<String> errors = new ArrayList<>();

		Set<String> names = new HashSet<>();
		Set<Class<?>> passClsSet = new HashSet<>();
		for (int i = 0; i < passes.size(); i++) {
			IDexTreeVisitor pass = passes.get(i);
			Class<? extends IDexTreeVisitor> passClass = pass.getClass();
			JadxVisitor info = passClass.getAnnotation(JadxVisitor.class);
			if (info == null) {
				LOG.warn("No JadxVisitor annotation for visitor: {}", passClass.getName());
				continue;
			}
			boolean firstOccurrence = passClsSet.add(passClass);
			String passName = passClass.getSimpleName();
			if (firstOccurrence && !names.add(passName)) {
				errors.add("Visitor name conflict: " + passName + ", class: " + passClass.getName());
			}
			for (Class<? extends IDexTreeVisitor> cls : info.runBefore()) {
				int beforeIndex = classList.indexOf(cls);
				if (beforeIndex != -1 && beforeIndex < i) {
					errors.add("Pass " + passName + " must be before " + cls.getSimpleName());
				}
			}
			for (Class<? extends IDexTreeVisitor> cls : info.runAfter()) {
				if (classList.indexOf(cls) > i) {
					errors.add("Pass " + passName + " must be after " + cls.getSimpleName());
				}
			}
		}
		return errors;
	}
}
