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
}