package org.javacs;

import static org.hamcrest.Matchers.*;

import static org.junit.Assert.assertThat;

import java.net.URI;

import java.nio.file.Path;

import java.nio.file.Paths;

import java.util.ArrayList;

import java.util.List;

import org.javacs.lsp.*;

import org.junit.Test;

public class GotoTest {

private static final String file = "/org/javacs/example/Goto.java";

private static final String defaultConstructorFile = "/org/javacs/example/GotoDefaultConstructor.java";

@Test

public void localVariable() {

var suggestions = doGoto(file, 10, 9);

assertThat(suggestions, hasItem("Goto.java:5"));

}

@Test

public void defaultConstructor() {

var suggestions = doGoto(defaultConstructorFile, 5, 46);

assertThat(suggestions, hasItem("GotoDefaultConstructor.java:3"));

}

@Test

public void constructor() {

var suggestions = doGoto(file, 11, 21);

assertThat(suggestions, hasItem("Goto.java:3"));

}

@Test

public void className() {

var suggestions = doGoto(file, 16, 9);

assertThat(suggestions, hasItem("Goto.java:3"));

}

@Test

public void staticField() {

var suggestions = doGoto(file, 13, 22);

assertThat(suggestions, hasItem("Goto.java:36"));

}

@Test

public void field() {

var suggestions = doGoto(file, 14, 22);

assertThat(suggestions, hasItem("Goto.java:37"));

}

@Test

public void staticMethod() {

var suggestions = doGoto(file, 16, 14);

assertThat(suggestions, hasItem("Goto.java:38"));

}

@Test

public void method() {

var suggestions = doGoto(file, 17, 14);

assertThat(suggestions, hasItem("Goto.java:41"));

}

@Test

public void staticMethodReference() {

var suggestions = doGoto(file, 19, 27);

assertThat(suggestions, hasItem("Goto.java:38"));

}

@Test

public void methodReference() {

var suggestions = doGoto(file, 20, 27);

assertThat(suggestions, hasItem("Goto.java:41"));

}

@Test

public void otherStaticMethod() {

var suggestions = doGoto(file, 29, 25);

assertThat(suggestions, hasItem(startsWith("GotoOther.java:")));

}

@Test

public void otherMethod() {

var suggestions = doGoto(file, 30, 18);

assertThat(suggestions, hasItem(startsWith("GotoOther.java:")));

}

@Test

public void otherCompiledFile() {

var suggestions = doGoto(file, 29, 25);

assertThat(suggestions, hasItem(startsWith("GotoOther.java:")));

}

@Test

public void constructorInOtherFile() {

var suggestions = doGoto(file, 24, 17);

assertThat(suggestions, hasItem("GotoOther.java:12"));

}

@Test

public void typeParam() {

var suggestions = doGoto(file, 46, 12);

assertThat(suggestions, hasItem("Goto.java:3"));

}

@Test

public void gotoEnum() {

var file = "/org/javacs/example/GotoEnum.java";

assertThat(doGoto(file, 5, 30), hasItem("GotoEnum.java:8"));

assertThat(doGoto(file, 5, 35), hasItem("GotoEnum.java:9"));

}

@Test

public void gotoOverload() {

var file = "/org/javacs/example/GotoOverload.java";

assertThat(doGoto(file, 7, 12), hasItem("GotoOverload.java:4"));

assertThat(doGoto(file, 8, 12), hasItem("GotoOverload.java:12"));

assertThat(doGoto(file, 9, 12), hasItem("GotoOverload.java:16"));

}

@Test

public void gotoOverloadInOtherFile() {

var file = "/org/javacs/example/GotoOverloadInOtherFile.java";

assertThat(doGoto(file, 5, 25), hasItem("GotoOverload.java:4"));

assertThat(doGoto(file, 6, 25), hasItem("GotoOverload.java:12"));

assertThat(doGoto(file, 7, 25), hasItem("GotoOverload.java:16"));

}

@Test

public void gotoImplementation() {

var file = "/org/javacs/example/GotoImplementation.java";

assertThat(doGoto(file, 5, 18), hasItems("GotoImplementation.java:9"));

// assertThat(doGoto(file, 5, 18), hasItems("GotoImplementation.java:9", "GotoImplementation.java:14"));

}

@Test

public void gotoImplementsRunnable() {

var file = "/org/javacs/example/GotoOtherPackageMethod.java";

assertThat(doGoto(file, 7, 12), empty());

// assertThat(doGoto(file, 7, 12), hasItem("ImplementsRunnable.java:5"));

}

@Test

public void gotoImplementsConsumer() {

var file = "/org/javacs/example/GotoOtherPackageMethod.java";

assertThat(doGoto(file, 11, 12), empty());

// assertThat(doGoto(file, 11, 12), hasItem("ImplementsConsumer.java:7"));

}

@Test

public void gotoError() {

var file = "/org/javacs/example/GotoError.java";

assertThat(doGoto(file, 5, 22), empty());

}

@Test

public void gotoSingleChar() {

var file = "/org/javacs/example/GotoSingleChar.java";

assertThat(doGoto(file, 6, 28, true), hasItem("GotoSingleChar.java:5,16"));

}

@Test

public void gotoInterface() {

var file = "/org/javacs/example/GotoInterface.java";

assertThat(doGoto(file, 3, 40, false), hasItem("GotoInterfaceInterface.java:3"));

}

@Test

public void gotoMismatchedMethod() {

var file = "/org/javacs/example/GotoMismatchedMethod.java";

assertThat(doGoto(file, 5, 12, false), hasItems("GotoMismatchedMethod.java:7", "GotoMismatchedMethod.java:8"));

}

@Test

public void packagePrivate() {

// There is a separate bug where javac doesn't find package-private classes in files with different names.

// This is tested in WarningsTest#referencePackagePrivateClassInFileWithDifferentName

var warmup = doGoto("/org/javacs/example/ContainsGotoPackagePrivate.java", 4, 29);

assertThat(warmup, not(empty()));

var suggestions = doGoto(file, 50, 42);

assertThat(suggestions, hasItem("ContainsGotoPackagePrivate.java:4"));

}

@Test

public void gsonSourceJar() {

var file = "/org/javacs/example/GotoGuava.java";

assertThat(doGoto(file, 7, 15, false), hasItem("Gson.java:105"));

}

private static final JavaLanguageServer server = LanguageServerFixture.getJavaLanguageServer();

private List<String> doGoto(String file, int row, int column) {

return doGoto(file, row, column, false);

}

private List<String> doGoto(String file, int row, int column, boolean includeColumn) {

TextDocumentIdentifier document = new TextDocumentIdentifier();

document.uri = FindResource.uri(file);

Position position = new Position();

position.line = row - 1;

position.character = column - 1;

TextDocumentPositionParams p = new TextDocumentPositionParams();

p.textDocument = document;

p.position = position;

var locations = server.gotoDefinition(p).orElse(List.of());

var strings = new ArrayList<String>();

for (var l : locations) {

var fileName = path(l.uri).getFileName();

var start = l.range.start;

if (includeColumn) {

strings.add(String.format("%s:%d,%d", fileName, start.line + 1, start.character + 1));

} else {

strings.add(String.format("%s:%d", fileName, start.line + 1));

}

}

return strings;

}

private Path path(URI uri) {

switch (uri.getScheme()) {

case "file":

return Paths.get(uri);

case "jar":

return Paths.get(uri.getSchemeSpecificPart().substring("file://".length()));

default:

throw new RuntimeException("Don't know what to do with " + uri.getScheme());

}

}

}