mirror of
https://github.com/odin-lang/Odin.git
synced 2025-12-29 01:14:40 +00:00
`dir` will leak memory if u use it with allocators that don´t care in freeing the memory at the end ( like arenas or the temp_allocator ) , because `strings.clone` and `strings.concatenate` are not using the passed allocator.
405 lines
7.8 KiB
Odin
405 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 {
|
|
context.allocator = allocator
|
|
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])
|
|
defer delete(dir)
|
|
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^ = {}
|
|
}
|