96 lines
2.2 KiB
Go
96 lines
2.2 KiB
Go
|
package vfs
|
||
|
|
|
||
|
|
import (
|
||
|
|
"io/fs"
|
||
|
|
"path/filepath"
|
||
|
|
"strings"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
var (
|
||
|
|
configExtensions = map[string]struct{}{
|
||
|
|
"toml": {}, "yaml": {}, "yml": {}, "json": {}, "jsonc": {}, "ini": {}, "conf": {},
|
||
|
|
"config": {}, "env": {}, "properties": {}, "xml": {}, "mod": {}, "sum": {}, "lock": {},
|
||
|
|
}
|
||
|
|
textExtensions = map[string]struct{}{
|
||
|
|
"txt": {}, "md": {}, "rst": {}, "go": {}, "rs": {}, "c": {}, "h": {}, "cpp": {}, "hpp": {},
|
||
|
|
"py": {}, "js": {}, "ts": {}, "tsx": {}, "jsx": {}, "java": {}, "kt": {}, "swift": {},
|
||
|
|
"html": {}, "css": {}, "scss": {}, "sh": {}, "bash": {}, "zsh": {}, "fish": {}, "sql": {},
|
||
|
|
"log": {}, "csv": {},
|
||
|
|
}
|
||
|
|
imageExtensions = map[string]struct{}{
|
||
|
|
"png": {}, "jpg": {}, "jpeg": {}, "gif": {}, "webp": {}, "bmp": {}, "svg": {}, "ico": {},
|
||
|
|
}
|
||
|
|
archiveExtensions = map[string]struct{}{
|
||
|
|
"zip": {}, "tar": {}, "gz": {}, "tgz": {}, "xz": {}, "bz2": {}, "7z": {}, "rar": {},
|
||
|
|
}
|
||
|
|
)
|
||
|
|
|
||
|
|
type Entry struct {
|
||
|
|
Name string
|
||
|
|
Path string
|
||
|
|
Extension string
|
||
|
|
Mode fs.FileMode
|
||
|
|
Size int64
|
||
|
|
ModifiedAt time.Time
|
||
|
|
CreatedAt time.Time
|
||
|
|
CreatedKnown bool
|
||
|
|
IsDir bool
|
||
|
|
IsParent bool
|
||
|
|
IsHidden bool
|
||
|
|
DirSizeKnown bool
|
||
|
|
}
|
||
|
|
|
||
|
|
func (e Entry) DisplayName() string {
|
||
|
|
if e.IsParent {
|
||
|
|
return ".."
|
||
|
|
}
|
||
|
|
if e.IsDir {
|
||
|
|
return e.Name + "/"
|
||
|
|
}
|
||
|
|
return e.Name
|
||
|
|
}
|
||
|
|
|
||
|
|
func (e Entry) IsFile() bool {
|
||
|
|
return !e.IsDir && !e.IsParent
|
||
|
|
}
|
||
|
|
|
||
|
|
func (e Entry) MatchKey() string {
|
||
|
|
return strings.ToLower(e.Name)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (e Entry) IsExecutable() bool {
|
||
|
|
return !e.IsDir && !e.IsParent && e.Mode&0o111 != 0
|
||
|
|
}
|
||
|
|
|
||
|
|
func (e Entry) Category() string {
|
||
|
|
switch {
|
||
|
|
case e.IsParent:
|
||
|
|
return "parent"
|
||
|
|
case e.IsDir:
|
||
|
|
return "directory"
|
||
|
|
case e.IsExecutable():
|
||
|
|
return "executable"
|
||
|
|
case hasExt(configExtensions, e.Extension):
|
||
|
|
return "config"
|
||
|
|
case hasExt(imageExtensions, e.Extension):
|
||
|
|
return "image"
|
||
|
|
case hasExt(textExtensions, e.Extension):
|
||
|
|
return "text"
|
||
|
|
case hasExt(archiveExtensions, e.Extension):
|
||
|
|
return "archive"
|
||
|
|
default:
|
||
|
|
return "binary"
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func ext(name string) string {
|
||
|
|
value := strings.TrimPrefix(filepath.Ext(name), ".")
|
||
|
|
return strings.ToLower(value)
|
||
|
|
}
|
||
|
|
|
||
|
|
func hasExt(set map[string]struct{}, ext string) bool {
|
||
|
|
_, ok := set[strings.ToLower(ext)]
|
||
|
|
return ok
|
||
|
|
}
|