12345678910111213141516171819202122232425262728293031323334353637383940414243444546 |
- // Copyright 2017 Manu Martinez-Almeida. All rights reserved.
- // Use of this source code is governed by a MIT style
- // license that can be found in the LICENSE file.
-
- package gin
-
- import (
- "net/http"
- "os"
- )
-
- type onlyfilesFS struct {
- fs http.FileSystem
- }
-
- type neuteredReaddirFile struct {
- http.File
- }
-
- // Dir returns a http.Filesystem that can be used by http.FileServer(). It is used internally
- // in router.Static().
- // if listDirectory == true, then it works the same as http.Dir() otherwise it returns
- // a filesystem that prevents http.FileServer() to list the directory files.
- func Dir(root string, listDirectory bool) http.FileSystem {
- fs := http.Dir(root)
- if listDirectory {
- return fs
- }
- return &onlyfilesFS{fs}
- }
-
- // Open conforms to http.Filesystem.
- func (fs onlyfilesFS) Open(name string) (http.File, error) {
- f, err := fs.fs.Open(name)
- if err != nil {
- return nil, err
- }
- return neuteredReaddirFile{f}, nil
- }
-
- // Readdir overrides the http.File default implementation.
- func (f neuteredReaddirFile) Readdir(count int) ([]os.FileInfo, error) {
- // this disables directory listing
- return nil, nil
- }
|