podman

1
// Copyright 2017 Manu Martinez-Almeida. All rights reserved.
2
// Use of this source code is governed by a MIT style
3
// license that can be found in the LICENSE file.
4

5
package gin
6

7
import (
8
	"net/http"
9
	"os"
10
)
11

12
type onlyFilesFS struct {
13
	fs http.FileSystem
14
}
15

16
type neuteredReaddirFile struct {
17
	http.File
18
}
19

20
// Dir returns a http.FileSystem that can be used by http.FileServer(). It is used internally
21
// in router.Static().
22
// if listDirectory == true, then it works the same as http.Dir() otherwise it returns
23
// a filesystem that prevents http.FileServer() to list the directory files.
24
func Dir(root string, listDirectory bool) http.FileSystem {
25
	fs := http.Dir(root)
26
	if listDirectory {
27
		return fs
28
	}
29
	return &onlyFilesFS{fs}
30
}
31

32
// Open conforms to http.Filesystem.
33
func (fs onlyFilesFS) Open(name string) (http.File, error) {
34
	f, err := fs.fs.Open(name)
35
	if err != nil {
36
		return nil, err
37
	}
38
	return neuteredReaddirFile{f}, nil
39
}
40

41
// Readdir overrides the http.File default implementation.
42
func (f neuteredReaddirFile) Readdir(_ int) ([]os.FileInfo, error) {
43
	// this disables directory listing
44
	return nil, nil
45
}
46