github.com/avfs/avfs@v0.33.1-0.20240303173310-c6ba67c33eb7/pathiterator.go

//
//  Copyright 2021 The AVFS authors
//
//  Licensed under the Apache License, Version 2.0 (the "License");
//  you may not use this file except in compliance with the License.
//  You may obtain a copy of the License at
//
//  	http://www.apache.org/licenses/LICENSE-2.0
//
//  Unless required by applicable law or agreed to in writing, software
//  distributed under the License is distributed on an "AS IS" BASIS,
//  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
//  See the License for the specific language governing permissions and
//  limitations under the License.
//

package avfs

import "strings"

// PathIterator iterates through an absolute path.
// It returns each part of the path in successive calls to Next.
// The volume name (for Windows) is not considered as part of the path
// it is returned by VolumeName.
//
// Sample code :
//
//	pi := NewPathIterator(vfs, path)
//	for pi.Next() {
//	  fmt.Println(pi.Part())
//	}
//
// The path below shows the different results of the PathIterator methods
// when thirdPart is the current Part :
//
//	/firstPart/secondPart/thirdPart/fourthPart/fifthPart
//	                     |- Part --|
//	                   Start      End
//	|------- Left -------|         |------ Right ------|
//	|----- LeftPart ---------------|
//	                     |----------- RightPart -------|
type PathIterator[T VFSBase] struct {
	vfs           T
	path          string
	start         int
	end           int
	volumeNameLen int
	pathSeparator uint8
}

// NewPathIterator creates a new path iterator from an absolute path.
func NewPathIterator[T VFSBase](vfs T, path string) *PathIterator[T] {
	pi := PathIterator[T]{path: path}
	pi.vfs = vfs
	pi.volumeNameLen = VolumeNameLen(vfs, path)
	pi.pathSeparator = vfs.PathSeparator()
	pi.Reset()

	return &pi
}

// End returns the end position of the current Part.
func (pi *PathIterator[_]) End() int {
	return pi.end
}

// IsLast returns true if the current Part is the last one.
func (pi *PathIterator[_]) IsLast() bool {
	return pi.end == len(pi.path)
}

// Left returns the left path of the current Part.
func (pi *PathIterator[_]) Left() string {
	return pi.path[:pi.start]
}

// LeftPart returns the left path and current Part.
func (pi *PathIterator[_]) LeftPart() string {
	return pi.path[:pi.end]
}

// Next iterates through the next Part of the path.
// It returns false if there's no more parts.
func (pi *PathIterator[_]) Next() bool {
	pi.start = pi.end + 1
	if pi.start >= len(pi.path) {
		pi.end = pi.start

		return false
	}

	pos := strings.IndexByte(pi.path[pi.start:], pi.pathSeparator)
	if pos == -1 {
		pi.end = len(pi.path)
	} else {
		pi.end = pi.start + pos
	}

	return true
}

// Part returns the current Part.
func (pi *PathIterator[_]) Part() string {
	return pi.path[pi.start:pi.end]
}

// Path returns the path to iterate.
func (pi *PathIterator[_]) Path() string {
	return pi.path
}

// ReplacePart replaces the current Part of the path with the new path.
// If the path iterator has been reset it returns true.
// It can be used in symbolic link replacement.
func (pi *PathIterator[_]) ReplacePart(newPath string) bool {
	vfs := pi.vfs
	oldPath := pi.path

	if vfs.IsAbs(newPath) {
		pi.path = vfs.Join(newPath, oldPath[pi.end:])
	} else {
		pi.path = vfs.Join(oldPath[:pi.start], newPath, oldPath[pi.end:])
	}

	// If the old path before the current part is different, the iterator must be reset.
	if pi.start >= len(pi.path) || pi.path[:pi.start] != oldPath[:pi.start] {
		pi.Reset()

		return true
	}

	// restart from the part before the symbolic link part.
	pi.end = pi.start - 1

	return false
}

// Reset resets the iterator.
func (pi *PathIterator[_]) Reset() {
	pi.end = pi.volumeNameLen
}

// Right returns the right path of the current Part.
func (pi *PathIterator[_]) Right() string {
	return pi.path[pi.end:]
}

// RightPart returns the right path and the current Part.
func (pi *PathIterator[_]) RightPart() string {
	return pi.path[pi.start:]
}

// Start returns the start position of the current Part.
func (pi *PathIterator[_]) Start() int {
	return pi.start
}

// VolumeName returns leading volume name.
func (pi *PathIterator[_]) VolumeName() string {
	return pi.path[:pi.volumeNameLen]
}

// VolumeNameLen returns length of the leading volume name on Windows.
// It returns 0 elsewhere.
func (pi *PathIterator[_]) VolumeNameLen() int {
	return pi.volumeNameLen
}