first working version

pkg/fs/fs.go

@@ -0,0 +1,120 @@
+package fs
+
+import (
+	"errors"
+	"git.giftfish.de/ston1th/docstore/pkg/core"
+	"os"
+	"path/filepath"
+	"sort"
+	"strings"
+)
+
+type FileMode int
+
+const (
+	Undefined FileMode = iota
+	IsDir
+	IsFile
+)
+
+// Directory represents a directory and its contents
+type Directory struct {
+	Dirs  core.Paths
+	Files core.Paths
+}
+
+type Filesystem struct {
+	Base string
+}
+
+func NewFilesystem(base string) (fs *Filesystem, err error) {
+	fi, err := os.Stat(base)
+	if err != nil {
+		return
+	}
+	if !fi.IsDir() {
+		err = errors.New("base dir '" + base + "' is not a directory")
+	}
+	fs = &Filesystem{base}
+	return
+}
+
+func (fs *Filesystem) ReadDir(path string) (d *Directory) {
+	path = filepath.Clean(path)
+	combined := filepath.Join(fs.Base, path)
+	f, err := os.Open(combined)
+	if err != nil {
+		return
+	}
+	names, err := f.Readdirnames(0)
+	if err != nil {
+		return
+	}
+	d = new(Directory)
+	for _, n := range names {
+		fi, err := os.Stat(filepath.Join(combined, n))
+		if err != nil {
+			continue
+		}
+		m := fi.Mode()
+		if m.IsDir() {
+			d.Dirs = append(d.Dirs, core.Path{Name: n, Abs: filepath.Join(path, n)})
+			continue
+		}
+		if m.IsRegular() {
+			d.Files = append(d.Files, core.Path{Name: n, Abs: filepath.Join(path, n)})
+		}
+	}
+	sort.Sort(d.Dirs)
+	sort.Sort(d.Files)
+	return
+}
+
+func (fs *Filesystem) Recursive() (p core.Paths) {
+	filepath.Walk(fs.Base, func(path string, info os.FileInfo, err error) error {
+		if err != nil {
+			return nil
+		}
+		if info.IsDir() {
+			a := strings.TrimPrefix(path, fs.Base)
+			if a == "" {
+				a = "/"
+			}
+			p = append(p, core.Path{Abs: a})
+		}
+		return nil
+	})
+	sort.Sort(p)
+	return
+}
+
+func (fs *Filesystem) RecursiveFiles(p string) (files []string) {
+	p = filepath.Clean(p)
+	fullpath := filepath.Join(fs.Base, p)
+	filepath.Walk(fullpath, func(path string, info os.FileInfo, err error) error {
+		if err != nil {
+			return nil
+		}
+		if info.Mode().IsRegular() {
+			files = append(files, strings.TrimPrefix(path, fullpath))
+		}
+		return nil
+	})
+	return
+}
+
+func Paths(path string) (p core.Paths) {
+	path = filepath.Clean(path)
+	s := strings.Split(path, string(filepath.Separator))
+	for i, v := range s[1:] {
+		if v == "" {
+			continue
+		}
+		if i == 0 {
+			p = append(p, core.Path{Name: v, Abs: "/" + v})
+			continue
+		}
+		p = append(p, core.Path{Name: v, Abs: filepath.Join(p[i-1].Abs, v)})
+	}
+	return
+}