Implement a walk extension function for java.nio.file.Path #KT-52909
This commit is contained in:
committed by
Space
parent
7e5abdb1c0
commit
e7b37b3497
@@ -0,0 +1,177 @@
|
||||
/*
|
||||
* Copyright 2010-2021 JetBrains s.r.o. and Kotlin Programming Language contributors.
|
||||
* Use of this source code is governed by the Apache 2.0 license that can be found in the license/LICENSE.txt file.
|
||||
*/
|
||||
|
||||
package kotlin.io.path
|
||||
|
||||
import java.io.IOException
|
||||
import java.nio.file.*
|
||||
import java.nio.file.attribute.BasicFileAttributes
|
||||
|
||||
/**
|
||||
* This class is intended to implement different file traversal methods.
|
||||
* It allows to iterate through all files inside a given directory.
|
||||
* The order in which sibling files are visited is unspecified.
|
||||
*
|
||||
* If the file located by this path is not a directory, the walker iterates only it.
|
||||
* If the file located by this path does not exist, the walker iterates nothing, i.e. it's equivalent to an empty sequence.
|
||||
*/
|
||||
@ExperimentalPathApi
|
||||
internal class PathTreeWalk(
|
||||
private val start: Path,
|
||||
private val options: Array<out PathWalkOption>
|
||||
) : Sequence<Path> {
|
||||
|
||||
private val followLinks: Boolean
|
||||
get() = options.contains(PathWalkOption.FOLLOW_LINKS)
|
||||
|
||||
private val linkOptions: Array<LinkOption>
|
||||
get() = LinkFollowing.toLinkOptions(followLinks)
|
||||
|
||||
private val includeDirectories: Boolean
|
||||
get() = options.contains(PathWalkOption.INCLUDE_DIRECTORIES)
|
||||
|
||||
private val isBFS: Boolean
|
||||
get() = options.contains(PathWalkOption.BREADTH_FIRST)
|
||||
|
||||
override fun iterator(): Iterator<Path> = if (isBFS) bfsIterator() else dfsIterator()
|
||||
|
||||
private suspend inline fun SequenceScope<Path>.yieldIfNeeded(
|
||||
node: PathNode,
|
||||
entriesReader: DirectoryEntriesReader,
|
||||
entriesAction: (List<PathNode>) -> Unit
|
||||
) {
|
||||
val path = node.path
|
||||
if (path.isDirectory(*linkOptions)) {
|
||||
if (node.createsCycle())
|
||||
throw FileSystemLoopException(path.toString())
|
||||
|
||||
if (includeDirectories)
|
||||
yield(path)
|
||||
|
||||
if (path.isDirectory(*linkOptions)) // make sure the path was not deleted after it was yielded
|
||||
entriesAction(entriesReader.readEntries(node))
|
||||
|
||||
} else if (path.exists(LinkOption.NOFOLLOW_LINKS)) {
|
||||
yield(path)
|
||||
}
|
||||
}
|
||||
|
||||
private fun dfsIterator() = iterator<Path> {
|
||||
// Stack of directory iterators, beginning from the start directory
|
||||
val stack = ArrayDeque<PathNode>()
|
||||
val entriesReader = DirectoryEntriesReader(followLinks)
|
||||
|
||||
val startNode = PathNode(start, keyOf(start, linkOptions), null)
|
||||
yieldIfNeeded(startNode, entriesReader) { entries ->
|
||||
startNode.contentIterator = entries.iterator()
|
||||
stack.addLast(startNode)
|
||||
}
|
||||
|
||||
while (stack.isNotEmpty()) {
|
||||
val topNode = stack.last()
|
||||
val topIterator = topNode.contentIterator!!
|
||||
|
||||
if (topIterator.hasNext()) {
|
||||
val pathNode = topIterator.next()
|
||||
yieldIfNeeded(pathNode, entriesReader) { entries ->
|
||||
pathNode.contentIterator = entries.iterator()
|
||||
stack.addLast(pathNode)
|
||||
}
|
||||
} else {
|
||||
// There is nothing more on the top of the stack, go back
|
||||
stack.removeLast()
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
private fun bfsIterator() = iterator<Path> {
|
||||
// Queue of entries to be visited.
|
||||
val queue = ArrayDeque<PathNode>()
|
||||
val entriesReader = DirectoryEntriesReader(followLinks)
|
||||
|
||||
queue.addLast(PathNode(start, keyOf(start, linkOptions), null))
|
||||
|
||||
while (queue.isNotEmpty()) {
|
||||
val pathNode = queue.removeFirst()
|
||||
yieldIfNeeded(pathNode, entriesReader) { entries ->
|
||||
queue.addAll(entries)
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
private fun keyOf(path: Path, linkOptions: Array<LinkOption>): Any? {
|
||||
return try {
|
||||
path.readAttributes<BasicFileAttributes>(*linkOptions).fileKey()
|
||||
} catch (exception: Throwable) {
|
||||
null
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
private class PathNode(val path: Path, val key: Any?, val parent: PathNode?) {
|
||||
var contentIterator: Iterator<PathNode>? = null
|
||||
}
|
||||
|
||||
private fun PathNode.createsCycle(): Boolean {
|
||||
var ancestor = parent
|
||||
while (ancestor != null) {
|
||||
if (ancestor.key != null && key != null) {
|
||||
if (ancestor.key == key)
|
||||
return true
|
||||
} else {
|
||||
try {
|
||||
if (ancestor.path.isSameFileAs(path))
|
||||
return true
|
||||
} catch (_: IOException) { // ignore
|
||||
} catch (_: SecurityException) { // ignore
|
||||
}
|
||||
}
|
||||
ancestor = ancestor.parent
|
||||
}
|
||||
|
||||
return false
|
||||
}
|
||||
|
||||
|
||||
private object LinkFollowing {
|
||||
private val nofollowLinkOption = arrayOf(LinkOption.NOFOLLOW_LINKS)
|
||||
private val followLinkOption = emptyArray<LinkOption>()
|
||||
|
||||
private val nofollowVisitOption = emptySet<FileVisitOption>()
|
||||
private val followVisitOption = setOf(FileVisitOption.FOLLOW_LINKS)
|
||||
|
||||
fun toLinkOptions(followLinks: Boolean): Array<LinkOption> =
|
||||
if (followLinks) followLinkOption else nofollowLinkOption
|
||||
|
||||
fun toVisitOptions(followLinks: Boolean): Set<FileVisitOption> =
|
||||
if (followLinks) followVisitOption else nofollowVisitOption
|
||||
}
|
||||
|
||||
|
||||
private class DirectoryEntriesReader(val followLinks: Boolean) : SimpleFileVisitor<Path>() {
|
||||
private var directoryNode: PathNode? = null
|
||||
private var entries = ArrayDeque<PathNode>()
|
||||
|
||||
fun readEntries(directoryNode: PathNode): List<PathNode> {
|
||||
this.directoryNode = directoryNode
|
||||
Files.walkFileTree(directoryNode.path, LinkFollowing.toVisitOptions(followLinks), 1, this)
|
||||
entries.removeFirst()
|
||||
return entries.also { entries = ArrayDeque() }
|
||||
}
|
||||
|
||||
override fun preVisitDirectory(dir: Path, attrs: BasicFileAttributes): FileVisitResult {
|
||||
val directoryEntry = PathNode(dir, attrs.fileKey(), directoryNode)
|
||||
entries.add(directoryEntry)
|
||||
return super.preVisitDirectory(dir, attrs)
|
||||
}
|
||||
|
||||
override fun visitFile(file: Path, attrs: BasicFileAttributes): FileVisitResult {
|
||||
val fileEntry = PathNode(file, null, directoryNode)
|
||||
entries.add(fileEntry)
|
||||
return super.visitFile(file, attrs)
|
||||
}
|
||||
}
|
||||
@@ -996,3 +996,22 @@ public inline fun Path(base: String, vararg subpaths: String): Path =
|
||||
@kotlin.internal.InlineOnly
|
||||
public inline fun URI.toPath(): Path =
|
||||
Paths.get(this)
|
||||
|
||||
|
||||
/**
|
||||
* Returns a sequence of paths for visiting this directory and all its content.
|
||||
*
|
||||
* By default, only files are visited, in depth-first order, and symbolic links are not followed.
|
||||
* The combination of [options] overrides the default behavior. See [PathWalkOption].
|
||||
*
|
||||
* The order in which sibling files are visited is unspecified.
|
||||
*
|
||||
* If after calling this function new files get added or deleted from the file tree rooted at this directory,
|
||||
* the changes may or may not appear in the returned sequence.
|
||||
*
|
||||
* If the file located by this path does not exist, an empty sequence is returned.
|
||||
* if the file located by this path is not a directory, a sequence containing only this path is returned.
|
||||
*/
|
||||
@ExperimentalPathApi
|
||||
@SinceKotlin("1.7")
|
||||
public fun Path.walk(vararg options: PathWalkOption): Sequence<Path> = PathTreeWalk(this, options)
|
||||
|
||||
@@ -0,0 +1,27 @@
|
||||
/*
|
||||
* Copyright 2010-2022 JetBrains s.r.o. and Kotlin Programming Language contributors.
|
||||
* Use of this source code is governed by the Apache 2.0 license that can be found in the license/LICENSE.txt file.
|
||||
*/
|
||||
|
||||
package kotlin.io.path
|
||||
|
||||
import java.nio.file.Path
|
||||
|
||||
/**
|
||||
* An enumeration to provide walk options for [Path.walk] function.
|
||||
* The options can be combined to form the walk order and behavior needed.
|
||||
*
|
||||
* Note that this enumeration is not exhaustive and new cases might be added in the future.
|
||||
*/
|
||||
@ExperimentalPathApi
|
||||
@SinceKotlin("1.7")
|
||||
public enum class PathWalkOption {
|
||||
/** Visits directories as well. */
|
||||
INCLUDE_DIRECTORIES,
|
||||
|
||||
/** Walks in breadth-first order. */
|
||||
BREADTH_FIRST,
|
||||
|
||||
/** Follows symbolic links to the directories they point to. */
|
||||
FOLLOW_LINKS
|
||||
}
|
||||
@@ -4,8 +4,12 @@
|
||||
*/
|
||||
|
||||
package kotlin.jdk7.test
|
||||
import java.nio.file.Path
|
||||
|
||||
import java.io.IOException
|
||||
import java.nio.file.*
|
||||
import java.nio.file.attribute.BasicFileAttributes
|
||||
import kotlin.io.path.deleteIfExists
|
||||
import kotlin.io.path.exists
|
||||
import kotlin.test.*
|
||||
|
||||
abstract class AbstractPathTest {
|
||||
@@ -17,10 +21,24 @@ abstract class AbstractPathTest {
|
||||
}
|
||||
|
||||
fun Path.cleanupRecursively(): Path {
|
||||
cleanUpActions.add(this to { it.toFile().deleteRecursively() })
|
||||
cleanUpActions.add(this to {
|
||||
if (it.exists(LinkOption.NOFOLLOW_LINKS)) Files.walkFileTree(it, cleanupVisitor)
|
||||
})
|
||||
return this
|
||||
}
|
||||
|
||||
private val cleanupVisitor = object : SimpleFileVisitor<Path>() {
|
||||
override fun visitFile(file: Path, attrs: BasicFileAttributes): FileVisitResult {
|
||||
file.deleteIfExists()
|
||||
return super.visitFile(file, attrs)
|
||||
}
|
||||
|
||||
override fun postVisitDirectory(dir: Path, exc: IOException?): FileVisitResult {
|
||||
dir.deleteIfExists()
|
||||
return super.postVisitDirectory(dir, exc)
|
||||
}
|
||||
}
|
||||
|
||||
@AfterTest
|
||||
fun cleanUp() {
|
||||
for ((path, action) in cleanUpActions) {
|
||||
@@ -31,4 +49,16 @@ abstract class AbstractPathTest {
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
fun withRestrictedRead(vararg paths: Path, block: () -> Unit) {
|
||||
try {
|
||||
if (paths.all { it.toFile().setReadable(false) }) {
|
||||
block()
|
||||
} else {
|
||||
System.err.println("Couldn't restrict read access")
|
||||
}
|
||||
} finally {
|
||||
paths.forEach { it.toFile().setReadable(true) }
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@@ -374,7 +374,7 @@ class PathExtensionsTest : AbstractPathTest() {
|
||||
(dir / ("link-" + original.fileName)).createLinkPointingTo(original)
|
||||
} catch (e: IOException) {
|
||||
// may require a privilege
|
||||
println("Creating a link failed with ${e.stackTraceToString()}")
|
||||
println("Creating a link failed with $e")
|
||||
return
|
||||
}
|
||||
|
||||
@@ -395,7 +395,7 @@ class PathExtensionsTest : AbstractPathTest() {
|
||||
(dir / ("symlink-" + original.fileName)).createSymbolicLinkPointingTo(original)
|
||||
} catch (e: IOException) {
|
||||
// may require a privilege
|
||||
println("Creating a symlink failed with ${e.stackTraceToString()}")
|
||||
println("Creating a symlink failed with $e")
|
||||
return
|
||||
}
|
||||
|
||||
|
||||
@@ -0,0 +1,408 @@
|
||||
/*
|
||||
* Copyright 2010-2021 JetBrains s.r.o. and Kotlin Programming Language contributors.
|
||||
* Use of this source code is governed by the Apache 2.0 license that can be found in the license/LICENSE.txt file.
|
||||
*/
|
||||
|
||||
package kotlin.jdk7.test
|
||||
|
||||
import java.nio.file.FileSystemLoopException
|
||||
import java.nio.file.Path
|
||||
import kotlin.io.path.*
|
||||
import kotlin.test.*
|
||||
|
||||
class PathTreeWalkTest : AbstractPathTest() {
|
||||
|
||||
companion object {
|
||||
val referenceFilenames = listOf("1", "1/2", "1/3", "1/3/4.txt", "1/3/5.txt", "6", "7.txt", "8", "8/9.txt")
|
||||
val referenceFilesOnly = listOf("1/3/4.txt", "1/3/5.txt", "7.txt", "8/9.txt")
|
||||
|
||||
fun createTestFiles(): Path {
|
||||
val basedir = createTempDirectory()
|
||||
for (name in referenceFilenames) {
|
||||
val file = basedir.resolve(name)
|
||||
if (file.extension.isEmpty())
|
||||
file.createDirectories()
|
||||
else
|
||||
file.createFile()
|
||||
}
|
||||
return basedir
|
||||
}
|
||||
|
||||
fun Path.tryCreateSymbolicLinkTo(original: Path): Path? {
|
||||
return try {
|
||||
this.createSymbolicLinkPointingTo(original)
|
||||
} catch (e: Exception) {
|
||||
// the underlying OS may not support symbolic links or may require a privilege
|
||||
println("Creating a symbolic link failed with $e")
|
||||
null
|
||||
}
|
||||
}
|
||||
|
||||
fun testVisitedFiles(expected: List<String>, walk: Sequence<Path>, basedir: Path, message: (() -> String)? = null) {
|
||||
val actual = walk.map { it.relativeToOrSelf(basedir).invariantSeparatorsPathString }
|
||||
assertEquals(expected.sorted(), actual.toList().sorted(), message?.invoke())
|
||||
}
|
||||
}
|
||||
|
||||
@Test
|
||||
fun visitOnce() {
|
||||
val basedir = createTestFiles().cleanupRecursively()
|
||||
testVisitedFiles(referenceFilesOnly, basedir.walk(), basedir)
|
||||
|
||||
val expected = listOf("") + referenceFilenames
|
||||
testVisitedFiles(expected, basedir.walk(PathWalkOption.INCLUDE_DIRECTORIES), basedir)
|
||||
}
|
||||
|
||||
@Test
|
||||
fun singleFile() {
|
||||
val testFile = createTempFile().cleanup()
|
||||
val nonExistentFile = testFile.resolve("foo")
|
||||
|
||||
assertEquals(testFile, testFile.walk().single())
|
||||
assertEquals(testFile, testFile.walk(PathWalkOption.INCLUDE_DIRECTORIES).single())
|
||||
|
||||
assertTrue(nonExistentFile.walk().none())
|
||||
assertTrue(nonExistentFile.walk(PathWalkOption.INCLUDE_DIRECTORIES).none())
|
||||
}
|
||||
|
||||
@Test
|
||||
fun singleEmptyDirectory() {
|
||||
val testDir = createTempDirectory().cleanup()
|
||||
assertTrue(testDir.walk().none())
|
||||
assertEquals(testDir, testDir.walk(PathWalkOption.INCLUDE_DIRECTORIES).single())
|
||||
}
|
||||
|
||||
@Test
|
||||
fun filterAndMap() {
|
||||
val basedir = createTestFiles().cleanupRecursively()
|
||||
testVisitedFiles(referenceFilesOnly, basedir.walk().filterNot { it.isDirectory() }, basedir)
|
||||
}
|
||||
|
||||
@Test
|
||||
fun deleteTxtChildrenOnVisit() {
|
||||
|
||||
fun visit(path: Path) {
|
||||
if (!path.isDirectory()) return
|
||||
|
||||
for (child in path.listDirectoryEntries()) {
|
||||
if (child.name.endsWith("txt"))
|
||||
child.deleteExisting()
|
||||
}
|
||||
}
|
||||
|
||||
val basedir = createTestFiles().cleanupRecursively()
|
||||
val walk = basedir.walk(PathWalkOption.INCLUDE_DIRECTORIES).onEach { visit(it) }
|
||||
val expected = listOf("", "1", "1/2", "1/3", "6", "8")
|
||||
testVisitedFiles(expected, walk, basedir)
|
||||
}
|
||||
|
||||
@Test
|
||||
fun deleteSubtreeOnVisit() {
|
||||
val basedir = createTestFiles().cleanupRecursively()
|
||||
val walk = basedir.walk(PathWalkOption.INCLUDE_DIRECTORIES).onEach { path ->
|
||||
if (path.name == "1") {
|
||||
path.toFile().deleteRecursively()
|
||||
}
|
||||
}
|
||||
|
||||
val expected = listOf("", "1", "6", "7.txt", "8", "8/9.txt")
|
||||
testVisitedFiles(expected, walk, basedir)
|
||||
}
|
||||
|
||||
@Test
|
||||
fun addChildOnVisit() {
|
||||
val basedir = createTestFiles().cleanupRecursively()
|
||||
val walk = basedir.walk(PathWalkOption.INCLUDE_DIRECTORIES).onEach { path ->
|
||||
if (path.isDirectory()) {
|
||||
path.resolve("a.txt").createFile()
|
||||
}
|
||||
}
|
||||
|
||||
val expected = referenceFilenames + listOf("", "a.txt", "1/a.txt", "1/2/a.txt", "1/3/a.txt", "6/a.txt", "8/a.txt")
|
||||
testVisitedFiles(expected, walk, basedir)
|
||||
}
|
||||
|
||||
@Test
|
||||
fun exceptionOnVisit() {
|
||||
val basedir = createTestFiles().cleanupRecursively()
|
||||
val walk = basedir.walk(PathWalkOption.INCLUDE_DIRECTORIES).onEach { path ->
|
||||
if (path.name == "3") {
|
||||
throw RuntimeException("Test error")
|
||||
}
|
||||
}
|
||||
|
||||
val error = assertFailsWith<RuntimeException> {
|
||||
walk.toList()
|
||||
}
|
||||
assertEquals("Test error", error.message)
|
||||
}
|
||||
|
||||
@Test
|
||||
fun restrictedRead() {
|
||||
val basedir = createTestFiles().cleanupRecursively()
|
||||
val restrictedDir = basedir.resolve("1/3")
|
||||
|
||||
withRestrictedRead(restrictedDir) {
|
||||
val walk = basedir.walk(PathWalkOption.INCLUDE_DIRECTORIES)
|
||||
|
||||
val error = assertFailsWith<java.nio.file.AccessDeniedException> {
|
||||
walk.toList()
|
||||
}
|
||||
assertEquals(restrictedDir.toString(), error.file)
|
||||
}
|
||||
}
|
||||
|
||||
@Test
|
||||
fun depthFirstOrder() {
|
||||
val basedir = createTestFiles().cleanupRecursively()
|
||||
|
||||
val visited = HashSet<Path>()
|
||||
|
||||
fun visit(path: Path) {
|
||||
if (path == basedir) {
|
||||
assertTrue(visited.isEmpty())
|
||||
} else {
|
||||
assertTrue(visited.contains(path.parent))
|
||||
}
|
||||
visited.add(path)
|
||||
}
|
||||
|
||||
val walk = basedir.walk(PathWalkOption.INCLUDE_DIRECTORIES).onEach(::visit)
|
||||
|
||||
val expected = referenceFilenames + listOf("")
|
||||
testVisitedFiles(expected, walk, basedir)
|
||||
assertEquals(expected.sorted(), visited.map { it.relativeToOrSelf(basedir).invariantSeparatorsPathString }.sorted())
|
||||
}
|
||||
|
||||
@Test
|
||||
fun addSiblingOnVisit() {
|
||||
fun makeBackup(file: Path) {
|
||||
val bakFile = Path("$file.bak")
|
||||
file.copyTo(bakFile)
|
||||
}
|
||||
|
||||
val basedir = createTestFiles().cleanupRecursively()
|
||||
|
||||
// added siblings do not appear during iteration
|
||||
testVisitedFiles(referenceFilesOnly, basedir.walk().onEach(::makeBackup), basedir)
|
||||
|
||||
val expected = referenceFilenames + referenceFilesOnly.map { "$it.bak" } + listOf("")
|
||||
testVisitedFiles(expected, basedir.walk(PathWalkOption.INCLUDE_DIRECTORIES), basedir)
|
||||
}
|
||||
|
||||
@Test
|
||||
fun find() {
|
||||
val basedir = createTestFiles().cleanupRecursively()
|
||||
basedir.resolve("8/4.txt").createFile()
|
||||
val count = basedir.walk().count { it.name == "4.txt" }
|
||||
assertEquals(2, count)
|
||||
}
|
||||
|
||||
@Test
|
||||
fun symlinkToFile() {
|
||||
val basedir = createTestFiles().cleanupRecursively()
|
||||
val original = basedir.resolve("8/9.txt")
|
||||
basedir.resolve("1/3/link").tryCreateSymbolicLinkTo(original) ?: return
|
||||
|
||||
for (followLinks in listOf(emptyArray(), arrayOf(PathWalkOption.FOLLOW_LINKS))) {
|
||||
val walk = basedir.walk(*followLinks)
|
||||
testVisitedFiles(referenceFilesOnly + listOf("1/3/link"), walk, basedir)
|
||||
}
|
||||
|
||||
original.deleteExisting()
|
||||
for (followLinks in listOf(emptyArray(), arrayOf(PathWalkOption.FOLLOW_LINKS))) {
|
||||
val walk = basedir.walk(*followLinks)
|
||||
testVisitedFiles(referenceFilesOnly - listOf("8/9.txt") + listOf("1/3/link"), walk, basedir)
|
||||
}
|
||||
}
|
||||
|
||||
@Test
|
||||
fun symlinkToDirectory() {
|
||||
val basedir = createTestFiles().cleanupRecursively()
|
||||
val original = basedir.resolve("8")
|
||||
basedir.resolve("1/3/link").tryCreateSymbolicLinkTo(original) ?: return
|
||||
|
||||
// directory "8" contains "9.txt" file
|
||||
val followWalk = basedir.walk(PathWalkOption.INCLUDE_DIRECTORIES, PathWalkOption.FOLLOW_LINKS)
|
||||
testVisitedFiles(referenceFilenames + listOf("", "1/3/link", "1/3/link/9.txt"), followWalk, basedir)
|
||||
|
||||
val nofollowWalk = basedir.walk(PathWalkOption.INCLUDE_DIRECTORIES)
|
||||
testVisitedFiles(referenceFilenames + listOf("", "1/3/link"), nofollowWalk, basedir)
|
||||
|
||||
original.toFile().deleteRecursively()
|
||||
for (followLinks in listOf(emptyArray(), arrayOf(PathWalkOption.FOLLOW_LINKS))) {
|
||||
val walk = basedir.walk(PathWalkOption.INCLUDE_DIRECTORIES, *followLinks)
|
||||
testVisitedFiles(referenceFilenames - listOf("8", "8/9.txt") + listOf("", "1/3/link"), walk, basedir)
|
||||
}
|
||||
}
|
||||
|
||||
@Test
|
||||
fun symlinkTwoPointingToEachOther() {
|
||||
val basedir = createTempDirectory().cleanupRecursively()
|
||||
val link1 = basedir.resolve("link1")
|
||||
val link2 = basedir.resolve("link2").tryCreateSymbolicLinkTo(link1) ?: return
|
||||
link1.tryCreateSymbolicLinkTo(link2) ?: return
|
||||
|
||||
val walk = basedir.walk(PathWalkOption.FOLLOW_LINKS)
|
||||
|
||||
testVisitedFiles(listOf("link1", "link2"), walk, basedir)
|
||||
}
|
||||
|
||||
@Test
|
||||
fun symlinkPointingToItself() {
|
||||
val basedir = createTempDirectory().cleanupRecursively()
|
||||
val link = basedir.resolve("link")
|
||||
link.tryCreateSymbolicLinkTo(link) ?: return
|
||||
|
||||
val walk = basedir.walk(PathWalkOption.FOLLOW_LINKS)
|
||||
|
||||
testVisitedFiles(listOf("link"), walk, basedir)
|
||||
}
|
||||
|
||||
@Test
|
||||
fun symlinkToSymlink() {
|
||||
val basedir = createTestFiles().cleanupRecursively()
|
||||
val original = basedir.resolve("8")
|
||||
val link = basedir.resolve("1/3/link").tryCreateSymbolicLinkTo(original) ?: return
|
||||
basedir.resolve("1/linkToLink").tryCreateSymbolicLinkTo(link) ?: return
|
||||
|
||||
val walk = basedir.walk(PathWalkOption.INCLUDE_DIRECTORIES, PathWalkOption.FOLLOW_LINKS)
|
||||
|
||||
val depth2ExpectedNames =
|
||||
listOf("", "1", "1/2", "1/3", "1/linkToLink", "6", "7.txt", "8", "8/9.txt") // linkToLink is visited
|
||||
val depth3ExpectedNames = depth2ExpectedNames +
|
||||
listOf("1/3/4.txt", "1/3/5.txt", "1/3/link", "1/linkToLink/9.txt") // "9.txt" is visited once more through linkToLink
|
||||
val depth4ExpectedNames = depth3ExpectedNames +
|
||||
listOf("1/3/link/9.txt") // "9.txt" is visited once more through link
|
||||
testVisitedFiles(depth4ExpectedNames, walk, basedir) // no depth limit
|
||||
}
|
||||
|
||||
@Test
|
||||
fun symlinkBasedir() {
|
||||
val basedir = createTestFiles().cleanupRecursively()
|
||||
val link = createTempDirectory().cleanupRecursively().resolve("link").tryCreateSymbolicLinkTo(basedir) ?: return
|
||||
|
||||
run {
|
||||
val followWalk = link.walk(PathWalkOption.INCLUDE_DIRECTORIES, PathWalkOption.FOLLOW_LINKS)
|
||||
testVisitedFiles(referenceFilenames + listOf(""), followWalk, link)
|
||||
testVisitedFiles(referenceFilesOnly, link.walk(PathWalkOption.FOLLOW_LINKS), link)
|
||||
|
||||
val nofollowWalk = link.walk(PathWalkOption.INCLUDE_DIRECTORIES)
|
||||
assertEquals(link, nofollowWalk.single())
|
||||
assertEquals(link, link.walk().single())
|
||||
}
|
||||
|
||||
run {
|
||||
basedir.toFile().deleteRecursively()
|
||||
|
||||
val followWalk = link.walk(PathWalkOption.INCLUDE_DIRECTORIES, PathWalkOption.FOLLOW_LINKS)
|
||||
assertEquals(link, followWalk.single())
|
||||
|
||||
val nofollowWalk = link.walk(PathWalkOption.INCLUDE_DIRECTORIES)
|
||||
assertEquals(link, nofollowWalk.single())
|
||||
}
|
||||
}
|
||||
|
||||
@Test
|
||||
fun symlinkCyclic() {
|
||||
val basedir = createTestFiles().cleanupRecursively()
|
||||
val original = basedir.resolve("1")
|
||||
val link = original.resolve("2/link").tryCreateSymbolicLinkTo(original) ?: return
|
||||
|
||||
for (order in listOf(arrayOf(), arrayOf(PathWalkOption.BREADTH_FIRST))) {
|
||||
val walk = basedir.walk(PathWalkOption.FOLLOW_LINKS, *order)
|
||||
val error = assertFailsWith<FileSystemLoopException> {
|
||||
walk.toList()
|
||||
}
|
||||
assertEquals(link.toString(), error.file)
|
||||
}
|
||||
}
|
||||
|
||||
@Test
|
||||
fun symlinkCyclicWithTwo() {
|
||||
val basedir = createTestFiles().cleanupRecursively()
|
||||
val dir8 = basedir.resolve("8")
|
||||
val dir2 = basedir.resolve("1/2")
|
||||
dir8.resolve("linkTo2").tryCreateSymbolicLinkTo(dir2) ?: return
|
||||
dir2.resolve("linkTo8").tryCreateSymbolicLinkTo(dir8) ?: return
|
||||
|
||||
for (order in listOf(arrayOf(), arrayOf(PathWalkOption.BREADTH_FIRST))) {
|
||||
val walk = basedir.walk(PathWalkOption.FOLLOW_LINKS, *order)
|
||||
assertFailsWith<FileSystemLoopException> {
|
||||
walk.toList()
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@Test
|
||||
fun breadthFirstOrder() {
|
||||
val basedir = createTestFiles().cleanupRecursively()
|
||||
val walk = basedir.walk(PathWalkOption.BREADTH_FIRST, PathWalkOption.INCLUDE_DIRECTORIES)
|
||||
val depth0 = mutableListOf("")
|
||||
val depth1 = mutableListOf("1", "6", "7.txt", "8")
|
||||
val depth2 = mutableListOf("1/2", "1/3", "8/9.txt")
|
||||
val depth3 = mutableListOf("1/3/4.txt", "1/3/5.txt")
|
||||
|
||||
for (file in walk) {
|
||||
when (val pathString = file.relativeToOrSelf(basedir).invariantSeparatorsPathString) {
|
||||
in depth0 -> {
|
||||
depth0.remove(pathString)
|
||||
}
|
||||
in depth1 -> {
|
||||
assertTrue(depth0.isEmpty())
|
||||
depth1.remove(pathString)
|
||||
}
|
||||
in depth2 -> {
|
||||
assertTrue(depth1.isEmpty())
|
||||
depth2.remove(pathString)
|
||||
}
|
||||
in depth3 -> {
|
||||
assertTrue(depth2.isEmpty())
|
||||
depth3.remove(pathString)
|
||||
}
|
||||
else -> {
|
||||
fail("Unexpected file: $file. It might have appeared for the second time.")
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
assertTrue(
|
||||
depth0.isEmpty() && depth1.isEmpty() && depth2.isEmpty() && depth3.isEmpty(),
|
||||
"The following files were not visited: $depth0, $depth1, $depth2 $depth3"
|
||||
)
|
||||
}
|
||||
|
||||
@Test
|
||||
fun breadthFirstOnlyFiles() {
|
||||
val basedir = createTestFiles().cleanupRecursively()
|
||||
val walk = basedir.walk(PathWalkOption.BREADTH_FIRST)
|
||||
|
||||
val depth1 = mutableListOf("7.txt")
|
||||
val depth2 = mutableListOf("8/9.txt")
|
||||
val depth3 = mutableListOf("1/3/4.txt", "1/3/5.txt")
|
||||
|
||||
for (file in walk) {
|
||||
when (val pathString = file.relativeToOrSelf(basedir).invariantSeparatorsPathString) {
|
||||
in depth1 -> {
|
||||
depth1.remove(pathString)
|
||||
}
|
||||
in depth2 -> {
|
||||
assertTrue(depth1.isEmpty())
|
||||
depth2.remove(pathString)
|
||||
}
|
||||
in depth3 -> {
|
||||
assertTrue(depth2.isEmpty())
|
||||
depth3.remove(pathString)
|
||||
}
|
||||
else -> {
|
||||
fail("Unexpected file: $file. It might have appeared for the second time.")
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
assertTrue(
|
||||
depth1.isEmpty() && depth2.isEmpty() && depth3.isEmpty(),
|
||||
"The following files were not visited: $depth1, $depth2 $depth3"
|
||||
)
|
||||
}
|
||||
}
|
||||
+9
@@ -1,6 +1,14 @@
|
||||
public abstract interface annotation class kotlin/io/path/ExperimentalPathApi : java/lang/annotation/Annotation {
|
||||
}
|
||||
|
||||
public final class kotlin/io/path/PathWalkOption : java/lang/Enum {
|
||||
public static final field BREADTH_FIRST Lkotlin/io/path/PathWalkOption;
|
||||
public static final field FOLLOW_LINKS Lkotlin/io/path/PathWalkOption;
|
||||
public static final field INCLUDE_DIRECTORIES Lkotlin/io/path/PathWalkOption;
|
||||
public static fun valueOf (Ljava/lang/String;)Lkotlin/io/path/PathWalkOption;
|
||||
public static fun values ()[Lkotlin/io/path/PathWalkOption;
|
||||
}
|
||||
|
||||
public final class kotlin/io/path/PathsKt {
|
||||
public static final fun appendText (Ljava/nio/file/Path;Ljava/lang/CharSequence;Ljava/nio/charset/Charset;)V
|
||||
public static synthetic fun appendText$default (Ljava/nio/file/Path;Ljava/lang/CharSequence;Ljava/nio/charset/Charset;ILjava/lang/Object;)V
|
||||
@@ -20,6 +28,7 @@ public final class kotlin/io/path/PathsKt {
|
||||
public static final fun relativeTo (Ljava/nio/file/Path;Ljava/nio/file/Path;)Ljava/nio/file/Path;
|
||||
public static final fun relativeToOrNull (Ljava/nio/file/Path;Ljava/nio/file/Path;)Ljava/nio/file/Path;
|
||||
public static final fun relativeToOrSelf (Ljava/nio/file/Path;Ljava/nio/file/Path;)Ljava/nio/file/Path;
|
||||
public static final fun walk (Ljava/nio/file/Path;[Lkotlin/io/path/PathWalkOption;)Lkotlin/sequences/Sequence;
|
||||
public static final fun writeText (Ljava/nio/file/Path;Ljava/lang/CharSequence;Ljava/nio/charset/Charset;[Ljava/nio/file/OpenOption;)V
|
||||
public static synthetic fun writeText$default (Ljava/nio/file/Path;Ljava/lang/CharSequence;Ljava/nio/charset/Charset;[Ljava/nio/file/OpenOption;ILjava/lang/Object;)V
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user