test_that("Document link works", {
    skip_on_cran()

    dir <- tempdir()
    client <- language_client(dir)

    src_file1 <- withr::local_tempfile(tmpdir = dir, fileext = ".R")
    src_file2 <- withr::local_tempfile(tmpdir = dir, fileext = ".R")
    temp_file <- withr::local_tempfile(tmpdir = dir, fileext = ".R")
    src_file1 <- format(fs::path(src_file1))
    src_file2 <- format(fs::path(src_file2))

    writeLines(
        c(
            "x <- 1"
        ),
        src_file1
    )

    writeLines(
        c(
            "x <- 2"
        ),
        src_file2
    )

    writeLines(
        c(
            sprintf("source('%s')", src_file1),
            sprintf("source('%s')", basename(src_file1)),
            sprintf("source('%s')", src_file2),
            sprintf("source('%s')", basename(src_file2)),
            "source('non_exist_file.R')"
        ),
        temp_file
    )

    client %>% did_open(temp_file)

    result <- client %>% respond_document_link(temp_file)

    expect_length(result, 4)
    expect_equal(result[[1]]$range$start, list(line = 0, character = 8))
    expect_equal(result[[1]]$range$end, list(line = 0, character = 8 + nchar(src_file1)))
    expect_equal(result[[1]]$data$path, src_file1)

    expect_equal(result[[2]]$range$start, list(line = 1, character = 8))
    expect_equal(result[[2]]$range$end, list(line = 1, character = 8 + nchar(basename(src_file1))))
    expect_equal(result[[2]]$data$path, src_file1)

    expect_equal(result[[3]]$range$start, list(line = 2, character = 8))
    expect_equal(result[[3]]$range$end, list(line = 2, character = 8 + nchar(src_file2)))
    expect_equal(result[[3]]$data$path, src_file2)

    expect_equal(result[[4]]$range$start, list(line = 3, character = 8))
    expect_equal(result[[4]]$range$end, list(line = 3, character = 8 + nchar(basename(src_file2))))
    expect_equal(result[[4]]$data$path, src_file2)

    resolve <- client %>% respond_document_link_resolve(result[[1]])
    expect_equal(resolve$target, path_to_uri(src_file1))

    resolve <- client %>% respond_document_link_resolve(result[[2]])
    expect_equal(resolve$target, path_to_uri(src_file1))

    resolve <- client %>% respond_document_link_resolve(result[[3]])
    expect_equal(resolve$target, path_to_uri(src_file2))

    resolve <- client %>% respond_document_link_resolve(result[[4]])
    expect_equal(resolve$target, path_to_uri(src_file2))
})

test_that("Document link works with raw strings", {
    skip_on_cran()
    skip_if(getRversion() < "4.0.0")

    dir <- tempdir()
    client <- language_client(dir)

    src_file1 <- withr::local_tempfile(tmpdir = dir, fileext = ".R")
    src_file2 <- withr::local_tempfile(tmpdir = dir, fileext = ".R")
    temp_file <- withr::local_tempfile(tmpdir = dir, fileext = ".R")
    src_file1 <- format(fs::path(src_file1))
    src_file2 <- format(fs::path(src_file2))

    writeLines(
        c(
            "x <- 1"
        ),
        src_file1
    )

    writeLines(
        c(
            "x <- 2"
        ),
        src_file2
    )

    writeLines(
        c(
            sprintf("source(r'(%s)')", src_file1),
            sprintf("source(R'{%s}')", basename(src_file1)),
            sprintf("source(r'-(%s)-')", src_file2),
            sprintf("source(R'--{%s}--')", basename(src_file2)),
            "source(r'(non_exist_file.R)')"
        ),
        temp_file
    )

    client %>% did_open(temp_file)

    result <- client %>% respond_document_link(temp_file)

    expect_length(result, 4)
    expect_equal(result[[1]]$range$start, list(line = 0, character = 9))
    expect_equal(result[[1]]$range$end, list(line = 0, character = 9 + nchar(src_file1) + 2))
    expect_equal(result[[1]]$data$path, src_file1)

    expect_equal(result[[2]]$range$start, list(line = 1, character = 9))
    expect_equal(result[[2]]$range$end, list(line = 1, character = 9 + nchar(basename(src_file1)) + 2))
    expect_equal(result[[2]]$data$path, src_file1)

    expect_equal(result[[3]]$range$start, list(line = 2, character = 9))
    expect_equal(result[[3]]$range$end, list(line = 2, character = 9 + nchar(src_file2) + 4))
    expect_equal(result[[3]]$data$path, src_file2)

    expect_equal(result[[4]]$range$start, list(line = 3, character = 9))
    expect_equal(result[[4]]$range$end, list(line = 3, character = 9 + nchar(basename(src_file2)) + 6))
    expect_equal(result[[4]]$data$path, src_file2)

    resolve <- client %>% respond_document_link_resolve(result[[1]])
    expect_equal(resolve$target, path_to_uri(src_file1))

    resolve <- client %>% respond_document_link_resolve(result[[2]])
    expect_equal(resolve$target, path_to_uri(src_file1))

    resolve <- client %>% respond_document_link_resolve(result[[3]])
    expect_equal(resolve$target, path_to_uri(src_file2))

    resolve <- client %>% respond_document_link_resolve(result[[4]])
    expect_equal(resolve$target, path_to_uri(src_file2))
})

test_that("Document link works in Rmarkdown", {
    skip_on_cran()

    dir <- tempdir()
    client <- language_client(working_dir = dir)

    src_file1 <- withr::local_tempfile(tmpdir = dir, fileext = ".R")
    src_file2 <- withr::local_tempfile(tmpdir = dir, fileext = ".R")
    temp_file <- withr::local_tempfile(tmpdir = dir, fileext = ".Rmd")
    src_file1 <- format(fs::path(src_file1))
    src_file2 <- format(fs::path(src_file2))

    writeLines(
        c(
            "x <- 1"
        ),
        src_file1
    )

    writeLines(
        c(
            "x <- 2"
        ),
        src_file2
    )

    writeLines(
        c(
            "---",
            "title: r markdown",
            "---",
            sprintf("will source file '%s' and '%s'", src_file1, src_file2),
            "```{r}",
            sprintf("source('%s')", src_file1),
            sprintf("source('%s')", basename(src_file1)),
            sprintf("source('%s')", src_file2),
            sprintf("source('%s')", basename(src_file2)),
            "source('non_exist_file.R')",
            "```"
        ),
        temp_file
    )

    client %>% did_open(temp_file)

    result <- client %>% respond_document_link(temp_file)

    expect_length(result, 4)
    expect_equal(result[[1]]$range$start, list(line = 5, character = 8))
    expect_equal(result[[1]]$range$end, list(line = 5, character = 8 + nchar(src_file1)))
    expect_equal(result[[1]]$data$path, src_file1)

    expect_equal(result[[2]]$range$start, list(line = 6, character = 8))
    expect_equal(result[[2]]$range$end, list(line = 6, character = 8 + nchar(basename(src_file1))))
    expect_equal(result[[2]]$data$path, src_file1)

    expect_equal(result[[3]]$range$start, list(line = 7, character = 8))
    expect_equal(result[[3]]$range$end, list(line = 7, character = 8 + nchar(src_file2)))
    expect_equal(result[[3]]$data$path, src_file2)

    expect_equal(result[[4]]$range$start, list(line = 8, character = 8))
    expect_equal(result[[4]]$range$end, list(line = 8, character = 8 + nchar(basename(src_file2))))
    expect_equal(result[[4]]$data$path, src_file2)

    resolve <- client %>% respond_document_link_resolve(result[[1]])
    expect_equal(resolve$target, path_to_uri(src_file1))

    resolve <- client %>% respond_document_link_resolve(result[[2]])
    expect_equal(resolve$target, path_to_uri(src_file1))

    resolve <- client %>% respond_document_link_resolve(result[[3]])
    expect_equal(resolve$target, path_to_uri(src_file2))

    resolve <- client %>% respond_document_link_resolve(result[[4]])
    expect_equal(resolve$target, path_to_uri(src_file2))
})

test_that("Document link works with null workspace folder", {
    skip_on_cran()

    dir <- tempdir()
    client <- language_client(NULL)

    src_file1 <- withr::local_tempfile(tmpdir = dir, fileext = ".R")
    src_file2 <- withr::local_tempfile(tmpdir = dir, fileext = ".R")
    temp_file <- withr::local_tempfile(tmpdir = dir, fileext = ".R")
    src_file1 <- format(fs::path(src_file1))
    src_file2 <- format(fs::path(src_file2))

    writeLines(
        c(
            "x <- 1"
        ),
        src_file1
    )

    writeLines(
        c(
            "x <- 2"
        ),
        src_file2
    )

    writeLines(c(
        sprintf("source('%s')", src_file1),
        sprintf("source('%s')", basename(src_file1)),
        sprintf("source('%s')", src_file2),
        sprintf("source('%s')", basename(src_file2)),
        "source('non_exist_file.R')"
    ), temp_file)

    client %>% did_open(temp_file)

    result <- client %>% respond_document_link(temp_file)

    expect_length(result, 4)
    expect_equal(result[[1]]$range$start, list(line = 0, character = 8))
    expect_equal(result[[1]]$range$end, list(line = 0, character = 8 + nchar(src_file1)))
    expect_equal(result[[1]]$data$path, src_file1)

    expect_equal(result[[2]]$range$start, list(line = 1, character = 8))
    expect_equal(result[[2]]$range$end, list(line = 1, character = 8 + nchar(basename(src_file1))))
    expect_equal(result[[2]]$data$path, src_file1)

    expect_equal(result[[3]]$range$start, list(line = 2, character = 8))
    expect_equal(result[[3]]$range$end, list(line = 2, character = 8 + nchar(src_file2)))
    expect_equal(result[[3]]$data$path, src_file2)

    expect_equal(result[[4]]$range$start, list(line = 3, character = 8))
    expect_equal(result[[4]]$range$end, list(line = 3, character = 8 + nchar(basename(src_file2))))
    expect_equal(result[[4]]$data$path, src_file2)

    resolve <- client %>% respond_document_link_resolve(result[[1]])
    expect_equal(resolve$target, path_to_uri(src_file1))

    resolve <- client %>% respond_document_link_resolve(result[[2]])
    expect_equal(resolve$target, path_to_uri(src_file1))

    resolve <- client %>% respond_document_link_resolve(result[[3]])
    expect_equal(resolve$target, path_to_uri(src_file2))

    resolve <- client %>% respond_document_link_resolve(result[[4]])
    expect_equal(resolve$target, path_to_uri(src_file2))
})

test_that("Document link works with untitled document", {
    skip_on_cran()

    dir <- tempdir()
    client <- language_client(dir)

    src_file1 <- withr::local_tempfile(tmpdir = dir, fileext = ".R")
    src_file2 <- withr::local_tempfile(tmpdir = dir, fileext = ".R")
    temp_file <- withr::local_tempfile(tmpdir = dir, fileext = ".R")
    src_file1 <- format(fs::path(src_file1))
    src_file2 <- format(fs::path(src_file2))

    writeLines(
        c(
            "x <- 1"
        ),
        src_file1
    )

    writeLines(
        c(
            "x <- 2"
        ),
        src_file2
    )

    uri <- "untitled:Untitled-1"

    client %>% did_open(uri = uri, text = c(
        sprintf("source('%s')", src_file1),
        sprintf("source('%s')", basename(src_file1)),
        sprintf("source('%s')", src_file2),
        sprintf("source('%s')", basename(src_file2)),
        "source('non_exist_file.R')"
    ))

    result <- client %>% respond_document_link(uri = uri)

    expect_length(result, 4)
    expect_equal(result[[1]]$range$start, list(line = 0, character = 8))
    expect_equal(result[[1]]$range$end, list(line = 0, character = 8 + nchar(src_file1)))
    expect_equal(result[[1]]$data$path, src_file1)

    expect_equal(result[[2]]$range$start, list(line = 1, character = 8))
    expect_equal(result[[2]]$range$end, list(line = 1, character = 8 + nchar(basename(src_file1))))
    expect_equal(result[[2]]$data$path, src_file1)

    expect_equal(result[[3]]$range$start, list(line = 2, character = 8))
    expect_equal(result[[3]]$range$end, list(line = 2, character = 8 + nchar(src_file2)))
    expect_equal(result[[3]]$data$path, src_file2)

    expect_equal(result[[4]]$range$start, list(line = 3, character = 8))
    expect_equal(result[[4]]$range$end, list(line = 3, character = 8 + nchar(basename(src_file2))))
    expect_equal(result[[4]]$data$path, src_file2)

    resolve <- client %>% respond_document_link_resolve(result[[1]])
    expect_equal(resolve$target, path_to_uri(src_file1))

    resolve <- client %>% respond_document_link_resolve(result[[2]])
    expect_equal(resolve$target, path_to_uri(src_file1))

    resolve <- client %>% respond_document_link_resolve(result[[3]])
    expect_equal(resolve$target, path_to_uri(src_file2))

    resolve <- client %>% respond_document_link_resolve(result[[4]])
    expect_equal(resolve$target, path_to_uri(src_file2))
})