90 lines
1.8 KiB
Go
90 lines
1.8 KiB
Go
package disk
|
|
|
|
import (
|
|
"errors"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
|
|
"github.com/google/uuid"
|
|
)
|
|
|
|
type DiskState string
|
|
|
|
const (
|
|
DiskAbsent DiskState = "absent"
|
|
DiskForeign DiskState = "foreign"
|
|
DiskKnown DiskState = "known"
|
|
)
|
|
|
|
type DiskInfo struct {
|
|
State DiskState `json:"state"`
|
|
DiskID string `json:"disk_id"`
|
|
TotalBytes int64 `json:"total_bytes"`
|
|
FreeBytes int64 `json:"free_bytes"`
|
|
MountPath string `json:"mount_path"`
|
|
}
|
|
|
|
const MarkerDir = ".jukebox"
|
|
const idFile = "disk.id"
|
|
|
|
func Probe(mountPath string) (DiskInfo, error) {
|
|
info := DiskInfo{MountPath: mountPath, State: DiskAbsent}
|
|
|
|
if _, err := os.ReadDir(mountPath); err != nil {
|
|
return info, nil
|
|
}
|
|
|
|
total, free, err := DiskUsage(mountPath)
|
|
if err != nil {
|
|
return info, nil
|
|
}
|
|
info.TotalBytes = total
|
|
info.FreeBytes = free
|
|
|
|
idPath := filepath.Join(mountPath, MarkerDir, idFile)
|
|
data, err := os.ReadFile(idPath)
|
|
if errors.Is(err, os.ErrNotExist) {
|
|
info.State = DiskForeign
|
|
return info, nil
|
|
}
|
|
if err != nil {
|
|
info.State = DiskForeign
|
|
return info, nil
|
|
}
|
|
|
|
info.DiskID = strings.TrimSpace(string(data))
|
|
info.State = DiskKnown
|
|
return info, nil
|
|
}
|
|
|
|
func CheckWritable(path string) error {
|
|
f, err := os.CreateTemp(path, ".jukebox-writecheck-*")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
name := f.Name()
|
|
if err := f.Close(); err != nil {
|
|
_ = os.Remove(name)
|
|
return err
|
|
}
|
|
return os.Remove(name)
|
|
}
|
|
|
|
func InitDisk(mountPath string) (string, error) {
|
|
dir := filepath.Join(mountPath, MarkerDir)
|
|
if err := os.MkdirAll(dir, 0o755); err != nil {
|
|
return "", err
|
|
}
|
|
id := uuid.New().String()
|
|
idPath := filepath.Join(dir, idFile)
|
|
if err := os.WriteFile(idPath, []byte(id), 0o644); err != nil {
|
|
return "", err
|
|
}
|
|
return id, nil
|
|
}
|
|
|
|
func DBPath(mountPath string) string {
|
|
return filepath.Join(mountPath, MarkerDir, "history.db")
|
|
}
|