1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117
|
package filter_test
import (
"fmt"
"io"
"log"
"net/http"
"os"
pathpkg "path"
"strings"
"github.com/shurcooL/httpfs/filter"
"github.com/shurcooL/httpfs/vfsutil"
"golang.org/x/tools/godoc/vfs/httpfs"
"golang.org/x/tools/godoc/vfs/mapfs"
)
func ExampleKeep() {
var srcFS http.FileSystem
// Keep only "/target/dir" and its contents.
fs := filter.Keep(srcFS, func(path string, fi os.FileInfo) bool {
return path == "/" ||
path == "/target" ||
path == "/target/dir" ||
strings.HasPrefix(path, "/target/dir/")
})
_ = fs
}
func ExampleSkip() {
var srcFS http.FileSystem
// Skip all files named ".DS_Store".
fs := filter.Skip(srcFS, func(path string, fi os.FileInfo) bool {
return !fi.IsDir() && fi.Name() == ".DS_Store"
})
_ = fs
}
func Example_detailed() {
srcFS := httpfs.New(mapfs.New(map[string]string{
"zzz-last-file.txt": "It should be visited last.",
"a-file.txt": "It has stuff.",
"another-file.txt": "Also stuff.",
"some-file.html": "<html>and stuff</html>",
"folderA/entry-A.txt": "Alpha.",
"folderA/entry-B.txt": "Beta.",
"folderA/main.go": "package main\n",
"folderA/folder-to-skip/many.txt": "Entire folder can be skipped.",
"folderA/folder-to-skip/files.txt": "Entire folder can be skipped.",
"folder-to-skip": "This is a file, not a folder, and shouldn't be skipped.",
}))
// Skip files with .go and .html extensions, and directories named "folder-to-skip" (but
// not files named "folder-to-skip").
fs := filter.Skip(srcFS, func(path string, fi os.FileInfo) bool {
return pathpkg.Ext(fi.Name()) == ".go" || pathpkg.Ext(fi.Name()) == ".html" ||
(fi.IsDir() && fi.Name() == "folder-to-skip")
})
err := vfsutil.Walk(fs, "/", func(path string, fi os.FileInfo, err error) error {
if err != nil {
log.Printf("can't stat file %s: %v\n", path, err)
return nil
}
fmt.Println(path)
return nil
})
if err != nil {
panic(err)
}
fmt.Println()
// This file should be filtered out, even if accessed directly.
_, err = fs.Open("/folderA/main.go")
fmt.Println("os.IsNotExist(err):", os.IsNotExist(err))
fmt.Println(err)
fmt.Println()
// This folder should be filtered out, even if accessed directly.
_, err = fs.Open("/folderA/folder-to-skip")
fmt.Println("os.IsNotExist(err):", os.IsNotExist(err))
fmt.Println(err)
fmt.Println()
// This file should not be filtered out.
f, err := fs.Open("/folder-to-skip")
if err != nil {
panic(err)
}
io.Copy(os.Stdout, f)
f.Close()
// Output:
// /
// /a-file.txt
// /another-file.txt
// /folder-to-skip
// /folderA
// /folderA/entry-A.txt
// /folderA/entry-B.txt
// /zzz-last-file.txt
//
// os.IsNotExist(err): true
// open /folderA/main.go: file does not exist
//
// os.IsNotExist(err): true
// open /folderA/folder-to-skip: file does not exist
//
// This is a file, not a folder, and shouldn't be skipped.
}
|