OSDN Git Service

Hulk did something
[bytom/vapor.git] / vendor / github.com / spf13 / afero / path.go
diff --git a/vendor/github.com/spf13/afero/path.go b/vendor/github.com/spf13/afero/path.go
new file mode 100644 (file)
index 0000000..1d90e46
--- /dev/null
@@ -0,0 +1,108 @@
+// Copyright ©2015 The Go Authors
+// Copyright ©2015 Steve Francia <spf@spf13.com>
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package afero
+
+import (
+       "os"
+       "path/filepath"
+       "sort"
+)
+
+// readDirNames reads the directory named by dirname and returns
+// a sorted list of directory entries.
+// adapted from https://golang.org/src/path/filepath/path.go
+func readDirNames(fs Fs, dirname string) ([]string, error) {
+       f, err := fs.Open(dirname)
+       if err != nil {
+               return nil, err
+       }
+       names, err := f.Readdirnames(-1)
+       f.Close()
+       if err != nil {
+               return nil, err
+       }
+       sort.Strings(names)
+       return names, nil
+}
+
+// walk recursively descends path, calling walkFn
+// adapted from https://golang.org/src/path/filepath/path.go
+func walk(fs Fs, path string, info os.FileInfo, walkFn filepath.WalkFunc) error {
+       err := walkFn(path, info, nil)
+       if err != nil {
+               if info.IsDir() && err == filepath.SkipDir {
+                       return nil
+               }
+               return err
+       }
+
+       if !info.IsDir() {
+               return nil
+       }
+
+       names, err := readDirNames(fs, path)
+       if err != nil {
+               return walkFn(path, info, err)
+       }
+
+       for _, name := range names {
+               filename := filepath.Join(path, name)
+               fileInfo, err := lstatIfOs(fs, filename)
+               if err != nil {
+                       if err := walkFn(filename, fileInfo, err); err != nil && err != filepath.SkipDir {
+                               return err
+                       }
+               } else {
+                       err = walk(fs, filename, fileInfo, walkFn)
+                       if err != nil {
+                               if !fileInfo.IsDir() || err != filepath.SkipDir {
+                                       return err
+                               }
+                       }
+               }
+       }
+       return nil
+}
+
+// if the filesystem is OsFs use Lstat, else use fs.Stat
+func lstatIfOs(fs Fs, path string) (info os.FileInfo, err error) {
+       _, ok := fs.(*OsFs)
+       if ok {
+               info, err = os.Lstat(path)
+       } else {
+               info, err = fs.Stat(path)
+       }
+       return
+}
+
+// Walk walks the file tree rooted at root, calling walkFn for each file or
+// directory in the tree, including root. All errors that arise visiting files
+// and directories are filtered by walkFn. The files are walked in lexical
+// order, which makes the output deterministic but means that for very
+// large directories Walk can be inefficient.
+// Walk does not follow symbolic links.
+
+func (a Afero) Walk(root string, walkFn filepath.WalkFunc) error {
+       return Walk(a.Fs, root, walkFn)
+}
+
+func Walk(fs Fs, root string, walkFn filepath.WalkFunc) error {
+       info, err := lstatIfOs(fs, root)
+       if err != nil {
+               return walkFn(root, nil, err)
+       }
+       return walk(fs, root, info, walkFn)
+}