mirror of
https://github.com/gohugoio/hugo.git
synced 2024-07-04 16:21:57 +00:00
e6541c45ab
filepath.Walk does not follow symbolic links. There's no easy fix for that outside of Go, so the best we can do for now is to give notice to the end user by ERROR log statements. This commit also fixes a related panic situation in GenerateTemplateNameFrom when the layout dir was a symbolic link. Fixes #283
138 lines
2.7 KiB
Go
138 lines
2.7 KiB
Go
// Copyright © 2014 Steve Francia <spf@spf13.com>.
|
|
//
|
|
// Licensed under the Simple Public 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://opensource.org/licenses/Simple-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 source
|
|
|
|
import (
|
|
"bytes"
|
|
"github.com/spf13/hugo/helpers"
|
|
jww "github.com/spf13/jwalterweatherman"
|
|
"io"
|
|
"io/ioutil"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
)
|
|
|
|
type Input interface {
|
|
Files() []*File
|
|
}
|
|
|
|
type Filesystem struct {
|
|
files []*File
|
|
Base string
|
|
AvoidPaths []string
|
|
}
|
|
|
|
func (f *Filesystem) FilesByExts(exts ...string) []*File {
|
|
var newFiles []*File
|
|
|
|
if len(exts) == 0 {
|
|
return f.Files()
|
|
}
|
|
|
|
for _, x := range f.Files() {
|
|
for _, e := range exts {
|
|
if x.Ext() == strings.TrimPrefix(e, ".") {
|
|
newFiles = append(newFiles, x)
|
|
}
|
|
}
|
|
}
|
|
return newFiles
|
|
}
|
|
|
|
func (f *Filesystem) Files() []*File {
|
|
if len(f.files) < 1 {
|
|
f.captureFiles()
|
|
}
|
|
return f.files
|
|
}
|
|
|
|
func (f *Filesystem) add(name string, reader io.Reader) (err error) {
|
|
var file *File
|
|
|
|
//if f.Base == "" {
|
|
//file = NewFileWithContents(name, reader)
|
|
//} else {
|
|
file, err = NewFileFromAbs(f.Base, name, reader)
|
|
//}
|
|
|
|
if err == nil {
|
|
f.files = append(f.files, file)
|
|
}
|
|
return err
|
|
}
|
|
|
|
func (f *Filesystem) getRelativePath(name string) (final string, err error) {
|
|
return helpers.GetRelativePath(name, f.Base)
|
|
}
|
|
|
|
func (f *Filesystem) captureFiles() {
|
|
|
|
walker := func(filePath string, fi os.FileInfo, err error) error {
|
|
if err != nil {
|
|
return nil
|
|
}
|
|
|
|
if fi.Mode()&os.ModeSymlink == os.ModeSymlink {
|
|
jww.ERROR.Printf("Symbolic links not supported, skipping '%s'", filePath)
|
|
return nil
|
|
}
|
|
|
|
if fi.IsDir() {
|
|
if f.avoid(filePath) || isNonProcessablePath(filePath) {
|
|
return filepath.SkipDir
|
|
}
|
|
return nil
|
|
} else {
|
|
if isNonProcessablePath(filePath) {
|
|
return nil
|
|
}
|
|
data, err := ioutil.ReadFile(filePath)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
f.add(filePath, bytes.NewBuffer(data))
|
|
return nil
|
|
}
|
|
}
|
|
|
|
filepath.Walk(f.Base, walker)
|
|
}
|
|
|
|
func (f *Filesystem) avoid(filePath string) bool {
|
|
for _, avoid := range f.AvoidPaths {
|
|
if avoid == filePath {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
func isNonProcessablePath(filePath string) bool {
|
|
base := filepath.Base(filePath)
|
|
if base[0] == '.' {
|
|
return true
|
|
}
|
|
|
|
if base[0] == '#' {
|
|
return true
|
|
}
|
|
|
|
if base[len(base)-1] == '~' {
|
|
return true
|
|
}
|
|
|
|
return false
|
|
}
|