Files
Odin/core/path/filepath/path.odin
2021-08-31 22:21:13 +01:00

404 lines
7.8 KiB
Odin

// The path/filepath package uses either forward slashes or backslashes depending on the operating system
// To process paths usch as URLs that depend on forward slashes regardless of the OS, use the path package
package filepath
import "core:strings"
// is_separator checks whether the byte is a valid separator character
is_separator :: proc(c: byte) -> bool {
switch c {
case '/': return true
case '\\': return ODIN_OS == "windows"
}
return false
}
@(private)
is_slash :: proc(c: byte) -> bool {
return c == '\\' || c == '/'
}
split :: proc(path: string) -> (dir, file: string) {
vol := volume_name(path)
i := len(path) - 1
for i >= len(vol) && !is_separator(path[i]) {
i -= 1
}
return path[:i+1], path[i+1:]
}
volume_name :: proc(path: string) -> string {
return path[:volume_name_len(path)]
}
volume_name_len :: proc(path: string) -> int {
if ODIN_OS == "windows" {
if len(path) < 2 {
return 0
}
c := path[0]
if path[1] == ':' {
switch c {
case 'a'..='z', 'A'..='Z':
return 2
}
}
// URL: https://msdn.microsoft.com/en-us/library/windows/desktop/aa365247(v=vs.85).aspx
if l := len(path); l >= 5 && is_slash(path[0]) && is_slash(path[1]) &&
!is_slash(path[2]) && path[2] != '.' {
for n := 3; n < l-1; n += 1 {
if is_slash(path[n]) {
n += 1
if !is_slash(path[n]) {
if path[n] == '.' {
break
}
}
for ; n < l; n += 1 {
if is_slash(path[n]) {
break
}
}
return n
}
break
}
}
}
return 0
}
base :: proc(path: string) -> string {
if path == "" {
return "."
}
path := path
for len(path) > 0 && is_separator(path[len(path)-1]) {
path = path[:len(path)-1]
}
path = path[volume_name_len(path):]
i := len(path)-1
for i >= 0 && !is_separator(path[i]) {
i -= 1
}
if i >= 0 {
path = path[i+1:]
}
if path == "" {
return SEPARATOR_STRING
}
return path
}
clean :: proc(path: string, allocator := context.allocator) -> string {
context.allocator = allocator
path := path
original_path := path
vol_len := volume_name_len(path)
path = path[vol_len:]
if path == "" {
if vol_len > 1 && original_path[1] != ':' {
s, ok := from_slash(original_path)
if !ok {
s = strings.clone(s)
}
return s
}
return strings.concatenate({original_path, "."})
}
rooted := is_separator(path[0])
n := len(path)
out := &Lazy_Buffer{
s = path,
vol_and_path = original_path,
vol_len = vol_len,
}
r, dot_dot := 0, 0
if rooted {
lazy_buffer_append(out, SEPARATOR)
r, dot_dot = 1, 1
}
for r < n {
switch {
case is_separator(path[r]):
r += 1
case path[r] == '.' && (r+1 == n || is_separator(path[r+1])):
r += 1
case path[r] == '.' && path[r+1] == '.' && (r+2 == n || is_separator(path[r+2])):
r += 2
switch {
case out.w > dot_dot:
out.w -= 1
for out.w > dot_dot && !is_separator(lazy_buffer_index(out, out.w)) {
out.w -= 1
}
case !rooted:
if out.w > 0 {
lazy_buffer_append(out, SEPARATOR)
}
lazy_buffer_append(out, '.')
lazy_buffer_append(out, '.')
dot_dot = out.w
}
case:
if rooted && out.w != 1 || !rooted && out.w != 0 {
lazy_buffer_append(out, SEPARATOR)
}
for ; r < n && !is_separator(path[r]); r += 1 {
lazy_buffer_append(out, path[r])
}
}
}
if out.w == 0 {
lazy_buffer_append(out, '.')
}
s := lazy_buffer_string(out)
cleaned, new_allocation := from_slash(s)
if new_allocation {
delete(s)
lazy_buffer_destroy(out)
}
return cleaned
}
from_slash :: proc(path: string, allocator := context.allocator) -> (new_path: string, new_allocation: bool) {
if SEPARATOR == '/' {
return path, false
}
return strings.replace_all(path, "/", SEPARATOR_STRING, allocator)
}
to_slash :: proc(path: string, allocator := context.allocator) -> (new_path: string, new_allocation: bool) {
if SEPARATOR == '/' {
return path, false
}
return strings.replace_all(path, SEPARATOR_STRING, "/", allocator)
}
ext :: proc(path: string) -> string {
for i := len(path)-1; i >= 0 && !is_separator(path[i]); i -= 1 {
if path[i] == '.' {
return path[i:]
}
}
return ""
}
Relative_Error :: enum {
None,
Cannot_Relate,
}
rel :: proc(base_path, target_path: string, allocator := context.allocator) -> (string, Relative_Error) {
context.allocator = allocator
base_clean, target_clean := clean(base_path), clean(target_path)
delete_target := true
defer {
if delete_target {
delete(target_clean)
}
delete(base_clean)
}
if strings.equal_fold(target_clean, base_clean) {
return strings.clone("."), .None
}
base_vol, target_vol := volume_name(base_path), volume_name(target_path)
base := base_clean[len(base_vol):]
target := target_clean[len(target_vol):]
if base == "." {
base = ""
}
base_slashed := len(base) > 0 && base[0] == SEPARATOR
target_slashed := len(target) > 0 && target[0] == SEPARATOR
if base_slashed != target_slashed || !strings.equal_fold(base_vol, target_vol) {
return "", .Cannot_Relate
}
bl, tl := len(base), len(target)
b0, bi, t0, ti: int
for {
for bi < bl && base[bi] != SEPARATOR {
bi += 1
}
for ti < tl && target[ti] != SEPARATOR {
ti += 1
}
if !strings.equal_fold(target[t0:ti], base[b0:bi]) {
break
}
if bi < bl {
bi += 1
}
if ti < tl {
ti += 1
}
b0, t0 = bi, ti
}
if base[b0:bi] == ".." {
return "", .Cannot_Relate
}
if b0 != bl {
seps := strings.count(base[b0:bl], SEPARATOR_STRING)
size := 2 + seps*3
if tl != t0 {
size += 1 + tl - t0
}
buf := make([]byte, size)
n := copy(buf, "..")
for in 0..<seps {
buf[n] = SEPARATOR
copy(buf[n+1:], "..")
n += 3
}
if t0 != tl {
buf[n] = SEPARATOR
copy(buf[n+1:], target[t0:])
}
return string(buf), .None
}
delete_target = false
return target[t0:], .None
}
dir :: proc(path: string, allocator := context.allocator) -> string {
vol := volume_name(path)
i := len(path) - 1
for i >= len(vol) && !is_separator(path[i]) {
i -= 1
}
dir := clean(path[len(vol) : i+1], allocator)
defer delete(dir, allocator)
if dir == "." && len(vol) > 2 {
return strings.clone(vol)
}
return strings.concatenate({vol, dir})
}
split_list :: proc(path: string, allocator := context.allocator) -> []string {
if path == "" {
return nil
}
start: int
quote: bool
start, quote = 0, false
count := 0
for i := 0; i < len(path); i += 1 {
c := path[i]
switch {
case c == '"':
quote = !quote
case c == LIST_SEPARATOR && !quote:
count += 1
}
}
start, quote = 0, false
list := make([]string, count, allocator)
index := 0
for i := 0; i < len(path); i += 1 {
c := path[i]
switch {
case c == '"':
quote = !quote
case c == LIST_SEPARATOR && !quote:
list[index] = path[start:i]
index += 1
start = i + 1
}
}
assert(index == count)
for s0, i in list {
s, new := strings.replace_all(s0, `"`, ``, allocator)
if !new {
s = strings.clone(s, allocator)
}
list[i] = s
}
return list
}
/*
Lazy_Buffer is a lazily made path buffer
When it does allocate, it uses the context.allocator
*/
@(private)
Lazy_Buffer :: struct {
s: string,
b: []byte,
w: int, // write index
vol_and_path: string,
vol_len: int,
}
@(private)
lazy_buffer_index :: proc(lb: ^Lazy_Buffer, i: int) -> byte {
if lb.b != nil {
return lb.b[i]
}
return lb.s[i]
}
@(private)
lazy_buffer_append :: proc(lb: ^Lazy_Buffer, c: byte) {
if lb.b == nil {
if lb.w < len(lb.s) && lb.s[lb.w] == c {
lb.w += 1
return
}
lb.b = make([]byte, len(lb.s))
copy(lb.b, lb.s[:lb.w])
}
lb.b[lb.w] = c
lb.w += 1
}
@(private)
lazy_buffer_string :: proc(lb: ^Lazy_Buffer) -> string {
if lb.b == nil {
return strings.clone(lb.vol_and_path[:lb.vol_len+lb.w])
}
x := lb.vol_and_path[:lb.vol_len]
y := string(lb.b[:lb.w])
z := make([]byte, len(x)+len(y))
copy(z, x)
copy(z[len(x):], y)
return string(z)
}
@(private)
lazy_buffer_destroy :: proc(lb: ^Lazy_Buffer) {
delete(lb.b)
lb^ = {}
}