phoenix/hugo
1
0
Fork 0
mirror of https://github.com/gohugoio/hugo.git synced 2024-07-05 08:42:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add readFile template func

Browse source 
This also includes a refactor of the hugofs package and its usage.

The motivation for that is:

The Afero filesystems are brilliant. Hugo's way of adding a dozen of global variables for the different filesystems was a mistake. In readFile (and also in some other places in Hugo today) we need a way to restrict the access inside the working dir. We could use ioutil.ReadFile and implement the path checking, checking the base path and the dots ("..") etc. But it is obviously better to use an Afero BasePathFs combined witha ReadOnlyFs. We could create a use-once-filesystem and handle the initialization ourselves, but since this is also useful to others and the initialization depends on some other global state (which would mean to create a new file system on every invocation), we might as well do it properly and encapsulate the predefined set of filesystems. This change also leads the way, if needed, to encapsulate the file systems in a struct, making it possible to have several file system sets in action at once (parallel multilanguage site building? With Moore's law and all...)

Fixes #1551
This commit is contained in:
Bjørn Erik Pedersen 2016-03-22 00:28:42 +01:00
parent a89035bdaa
commit 4f66f790b1
32 changed files with 400 additions and 152 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
commands/gendoc.go
View file

@ -1,4 +1,4 @@
// Copyright 2015 The Hugo Authors. All rights reserved. // Copyright 2016 The Hugo Authors. All rights reserved.
// //
// Licensed under the Apache License, Version 2.0 (the "License"); // Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License. // you may not use this file except in compliance with the License.
@ -51,9 +51,9 @@ for rendering in Hugo.`,
if !strings.HasSuffix(gendocdir, helpers.FilePathSeparator) { if !strings.HasSuffix(gendocdir, helpers.FilePathSeparator) {
gendocdir += helpers.FilePathSeparator gendocdir += helpers.FilePathSeparator
} }
if found, _ := helpers.Exists(gendocdir, hugofs.OsFs); !found { if found, _ := helpers.Exists(gendocdir, hugofs.Os()); !found {
jww.FEEDBACK.Println("Directory", gendocdir, "does not exist, creating...") jww.FEEDBACK.Println("Directory", gendocdir, "does not exist, creating...")
hugofs.OsFs.MkdirAll(gendocdir, 0777) hugofs.Os().MkdirAll(gendocdir, 0777)
} }
now := time.Now().Format(time.RFC3339) now := time.Now().Format(time.RFC3339)
prepender := func(filename string) string { prepender := func(filename string) string {

6
commands/genman.go
View file

@ -1,4 +1,4 @@
// Copyright 2015 The Hugo Authors. All rights reserved. // Copyright 2016 The Hugo Authors. All rights reserved.
// //
// Licensed under the Apache License, Version 2.0 (the "License"); // Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License. // you may not use this file except in compliance with the License.
@ -41,9 +41,9 @@ in the "man" directory under the current directory.`,
if !strings.HasSuffix(genmandir, helpers.FilePathSeparator) { if !strings.HasSuffix(genmandir, helpers.FilePathSeparator) {
genmandir += helpers.FilePathSeparator genmandir += helpers.FilePathSeparator
} }
if found, _ := helpers.Exists(genmandir, hugofs.OsFs); !found { if found, _ := helpers.Exists(genmandir, hugofs.Os()); !found {
jww.FEEDBACK.Println("Directory", genmandir, "does not exist, creating...") jww.FEEDBACK.Println("Directory", genmandir, "does not exist, creating...")
hugofs.OsFs.MkdirAll(genmandir, 0777) hugofs.Os().MkdirAll(genmandir, 0777)
} }
cmd.Root().DisableAutoGenTag = true cmd.Root().DisableAutoGenTag = true

37
commands/hugo.go
View file

@ -423,14 +423,14 @@ func InitializeConfig(subCmdVs ...*cobra.Command) error {
if helpers.FilePathSeparator != cacheDir[len(cacheDir)-1:] { if helpers.FilePathSeparator != cacheDir[len(cacheDir)-1:] {
cacheDir = cacheDir + helpers.FilePathSeparator cacheDir = cacheDir + helpers.FilePathSeparator
} }
isDir, err := helpers.DirExists(cacheDir, hugofs.SourceFs) isDir, err := helpers.DirExists(cacheDir, hugofs.Source())
utils.CheckErr(err) utils.CheckErr(err)
if isDir == false { if isDir == false {
mkdir(cacheDir) mkdir(cacheDir)
} }
viper.Set("CacheDir", cacheDir) viper.Set("CacheDir", cacheDir)
} else { } else {
viper.Set("CacheDir", helpers.GetTempDir("hugo_cache", hugofs.SourceFs)) viper.Set("CacheDir", helpers.GetTempDir("hugo_cache", hugofs.Source()))
} }
if verboseLog || logging || (viper.IsSet("LogFile") && viper.GetString("LogFile") != "") { if verboseLog || logging || (viper.IsSet("LogFile") && viper.GetString("LogFile") != "") {
@ -453,6 +453,9 @@ func InitializeConfig(subCmdVs ...*cobra.Command) error {
jww.INFO.Println("Using config file:", viper.ConfigFileUsed()) jww.INFO.Println("Using config file:", viper.ConfigFileUsed())
// Init file systems. This may be changed at a later point.
hugofs.InitDefaultFs()
themeDir := helpers.GetThemeDir() themeDir := helpers.GetThemeDir()
if themeDir != "" { if themeDir != "" {
if _, err := os.Stat(themeDir); os.IsNotExist(err) { if _, err := os.Stat(themeDir); os.IsNotExist(err) {
@ -498,7 +501,7 @@ func build(watches ...bool) error {
// This is only used for benchmark testing. Cause the content is only visible // This is only used for benchmark testing. Cause the content is only visible
// in memory // in memory
if renderToMemory { if renderToMemory {
hugofs.DestinationFS = new(afero.MemMapFs) hugofs.SetDestination(new(afero.MemMapFs))
// Rendering to memoryFS, publish to Root regardless of publishDir. // Rendering to memoryFS, publish to Root regardless of publishDir.
viper.Set("PublishDir", "/") viper.Set("PublishDir", "/")
} }
@ -524,7 +527,7 @@ func build(watches ...bool) error {
} }
func getStaticSourceFs() afero.Fs { func getStaticSourceFs() afero.Fs {
source := hugofs.SourceFs source := hugofs.Source()
themeDir, err := helpers.GetThemeStaticDirPath() themeDir, err := helpers.GetThemeStaticDirPath()
staticDir := helpers.GetStaticDirPath() + helpers.FilePathSeparator staticDir := helpers.GetStaticDirPath() + helpers.FilePathSeparator
@ -563,8 +566,8 @@ func getStaticSourceFs() afero.Fs {
jww.INFO.Println("using a UnionFS for static directory comprised of:") jww.INFO.Println("using a UnionFS for static directory comprised of:")
jww.INFO.Println("Base:", themeDir) jww.INFO.Println("Base:", themeDir)
jww.INFO.Println("Overlay:", staticDir) jww.INFO.Println("Overlay:", staticDir)
base := afero.NewReadOnlyFs(afero.NewBasePathFs(hugofs.SourceFs, themeDir)) base := afero.NewReadOnlyFs(afero.NewBasePathFs(hugofs.Source(), themeDir))
overlay := afero.NewReadOnlyFs(afero.NewBasePathFs(hugofs.SourceFs, staticDir)) overlay := afero.NewReadOnlyFs(afero.NewBasePathFs(hugofs.Source(), staticDir))
return afero.NewCopyOnWriteFs(base, overlay) return afero.NewCopyOnWriteFs(base, overlay)
} }
@ -587,7 +590,7 @@ func copyStatic() error {
syncer := fsync.NewSyncer() syncer := fsync.NewSyncer()
syncer.NoTimes = viper.GetBool("notimes") syncer.NoTimes = viper.GetBool("notimes")
syncer.SrcFs = staticSourceFs syncer.SrcFs = staticSourceFs
syncer.DestFs = hugofs.DestinationFS syncer.DestFs = hugofs.Destination()
// Now that we are using a unionFs for the static directories // Now that we are using a unionFs for the static directories
// We can effectively clean the publishDir on initial sync // We can effectively clean the publishDir on initial sync
syncer.Delete = viper.GetBool("cleanDestinationDir") syncer.Delete = viper.GetBool("cleanDestinationDir")
@ -653,12 +656,12 @@ func getDirList() []string {
return nil return nil
} }
helpers.SymbolicWalk(hugofs.SourceFs, dataDir, walker) helpers.SymbolicWalk(hugofs.Source(), dataDir, walker)
helpers.SymbolicWalk(hugofs.SourceFs, helpers.AbsPathify(viper.GetString("ContentDir")), walker) helpers.SymbolicWalk(hugofs.Source(), helpers.AbsPathify(viper.GetString("ContentDir")), walker)
helpers.SymbolicWalk(hugofs.SourceFs, helpers.AbsPathify(viper.GetString("LayoutDir")), walker) helpers.SymbolicWalk(hugofs.Source(), helpers.AbsPathify(viper.GetString("LayoutDir")), walker)
helpers.SymbolicWalk(hugofs.SourceFs, helpers.AbsPathify(viper.GetString("StaticDir")), walker) helpers.SymbolicWalk(hugofs.Source(), helpers.AbsPathify(viper.GetString("StaticDir")), walker)
if helpers.ThemeSet() { if helpers.ThemeSet() {
helpers.SymbolicWalk(hugofs.SourceFs, helpers.AbsPathify(viper.GetString("themesDir")+"/"+viper.GetString("theme")), walker) helpers.SymbolicWalk(hugofs.Source(), helpers.AbsPathify(viper.GetString("themesDir")+"/"+viper.GetString("theme")), walker)
} }
return a return a
@ -770,8 +773,8 @@ func NewWatcher(port int) error {
// recursively add new directories to watch list // recursively add new directories to watch list
// When mkdir -p is used, only the top directory triggers an event (at least on OSX) // When mkdir -p is used, only the top directory triggers an event (at least on OSX)
if ev.Op&fsnotify.Create == fsnotify.Create { if ev.Op&fsnotify.Create == fsnotify.Create {
if s, err := hugofs.SourceFs.Stat(ev.Name); err == nil && s.Mode().IsDir() { if s, err := hugofs.Source().Stat(ev.Name); err == nil && s.Mode().IsDir() {
helpers.SymbolicWalk(hugofs.SourceFs, ev.Name, walkAdder) helpers.SymbolicWalk(hugofs.Source(), ev.Name, walkAdder)
} }
} }
@ -813,7 +816,7 @@ func NewWatcher(port int) error {
syncer := fsync.NewSyncer() syncer := fsync.NewSyncer()
syncer.NoTimes = viper.GetBool("notimes") syncer.NoTimes = viper.GetBool("notimes")
syncer.SrcFs = staticSourceFs syncer.SrcFs = staticSourceFs
syncer.DestFs = hugofs.DestinationFS syncer.DestFs = hugofs.Destination()
// prevent spamming the log on changes // prevent spamming the log on changes
logger := helpers.NewDistinctFeedbackLogger() logger := helpers.NewDistinctFeedbackLogger()
@ -858,7 +861,7 @@ func NewWatcher(port int) error {
// If file doesn't exist in any static dir, remove it // If file doesn't exist in any static dir, remove it
toRemove := filepath.Join(publishDir, relPath) toRemove := filepath.Join(publishDir, relPath)
logger.Println("File no longer exists in static dir, removing", toRemove) logger.Println("File no longer exists in static dir, removing", toRemove)
hugofs.DestinationFS.RemoveAll(toRemove) hugofs.Destination().RemoveAll(toRemove)
} else if err == nil { } else if err == nil {
// If file still exists, sync it // If file still exists, sync it
logger.Println("Syncing", relPath, "to", publishDir) logger.Println("Syncing", relPath, "to", publishDir)
@ -939,7 +942,7 @@ func isThemeVsHugoVersionMismatch() (mismatch bool, requiredMinVersion string) {
themeDir := helpers.GetThemeDir() themeDir := helpers.GetThemeDir()
fs := hugofs.SourceFs fs := hugofs.Source()
path := filepath.Join(themeDir, "theme.toml") path := filepath.Join(themeDir, "theme.toml")
exists, err := helpers.Exists(path, fs) exists, err := helpers.Exists(path, fs)

10
commands/import_jekyll.go
View file

@ -1,4 +1,4 @@
// Copyright 2015 The Hugo Authors. All rights reserved. // Copyright 2016 The Hugo Authors. All rights reserved.
// //
// Licensed under the Apache License, Version 2.0 (the "License"); // Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License. // you may not use this file except in compliance with the License.
@ -124,7 +124,7 @@ func importFromJekyll(cmd *cobra.Command, args []string) error {
return convertJekyllPost(path, relPath, targetDir, draft) return convertJekyllPost(path, relPath, targetDir, draft)
} }
err = helpers.SymbolicWalk(hugofs.OsFs, jekyllRoot, callback) err = helpers.SymbolicWalk(hugofs.Os(), jekyllRoot, callback)
if err != nil { if err != nil {
return err return err
@ -139,7 +139,7 @@ func importFromJekyll(cmd *cobra.Command, args []string) error {
// TODO: Consider calling doNewSite() instead? // TODO: Consider calling doNewSite() instead?
func createSiteFromJekyll(jekyllRoot, targetDir string, force bool) error { func createSiteFromJekyll(jekyllRoot, targetDir string, force bool) error {
fs := hugofs.SourceFs fs := hugofs.Source()
if exists, _ := helpers.Exists(targetDir, fs); exists { if exists, _ := helpers.Exists(targetDir, fs); exists {
if isDir, _ := helpers.IsDir(targetDir, fs); !isDir { if isDir, _ := helpers.IsDir(targetDir, fs); !isDir {
return errors.New("Target path \"" + targetDir + "\" already exists but not a directory") return errors.New("Target path \"" + targetDir + "\" already exists but not a directory")
@ -187,7 +187,7 @@ func createSiteFromJekyll(jekyllRoot, targetDir string, force bool) error {
} }
func loadJekyllConfig(jekyllRoot string) map[string]interface{} { func loadJekyllConfig(jekyllRoot string) map[string]interface{} {
fs := hugofs.SourceFs fs := hugofs.Source()
path := filepath.Join(jekyllRoot, "_config.yml") path := filepath.Join(jekyllRoot, "_config.yml")
exists, err := helpers.Exists(path, fs) exists, err := helpers.Exists(path, fs)
@ -252,7 +252,7 @@ func createConfigFromJekyll(inpath string, kind string, jekyllConfig map[string]
return err return err
} }
err = helpers.WriteToDisk(filepath.Join(inpath, "config."+kind), bytes.NewReader(by), hugofs.SourceFs) err = helpers.WriteToDisk(filepath.Join(inpath, "config."+kind), bytes.NewReader(by), hugofs.Source())
if err != nil { if err != nil {
return return
} }

26
commands/new.go
View file

@ -1,4 +1,4 @@
// Copyright 2015 The Hugo Authors. All rights reserved. // Copyright 2016 The Hugo Authors. All rights reserved.
// //
// Licensed under the Apache License, Version 2.0 (the "License"); // Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License. // you may not use this file except in compliance with the License.
@ -110,7 +110,7 @@ func NewContent(cmd *cobra.Command, args []string) error {
kind = contentType kind = contentType
} }
return create.NewContent(hugofs.SourceFs, kind, createpath) return create.NewContent(hugofs.Source(), kind, createpath)
} }
func doNewSite(basepath string, force bool) error { func doNewSite(basepath string, force bool) error {
@ -123,12 +123,12 @@ func doNewSite(basepath string, force bool) error {
filepath.Join(basepath, "themes"), filepath.Join(basepath, "themes"),
} }
if exists, _ := helpers.Exists(basepath, hugofs.SourceFs); exists { if exists, _ := helpers.Exists(basepath, hugofs.Source()); exists {
if isDir, _ := helpers.IsDir(basepath, hugofs.SourceFs); !isDir { if isDir, _ := helpers.IsDir(basepath, hugofs.Source()); !isDir {
return errors.New(basepath + " already exists but not a directory") return errors.New(basepath + " already exists but not a directory")
} }
isEmpty, _ := helpers.IsEmpty(basepath, hugofs.SourceFs) isEmpty, _ := helpers.IsEmpty(basepath, hugofs.Source())
switch { switch {
case !isEmpty && !force: case !isEmpty && !force:
@ -137,7 +137,7 @@ func doNewSite(basepath string, force bool) error {
case !isEmpty && force: case !isEmpty && force:
all := append(dirs, filepath.Join(basepath, "config."+configFormat)) all := append(dirs, filepath.Join(basepath, "config."+configFormat))
for _, path := range all { for _, path := range all {
if exists, _ := helpers.Exists(path, hugofs.SourceFs); exists { if exists, _ := helpers.Exists(path, hugofs.Source()); exists {
return errors.New(path + " already exists") return errors.New(path + " already exists")
} }
} }
@ -145,7 +145,7 @@ func doNewSite(basepath string, force bool) error {
} }
for _, dir := range dirs { for _, dir := range dirs {
hugofs.SourceFs.MkdirAll(dir, 0777) hugofs.Source().MkdirAll(dir, 0777)
} }
createConfig(basepath, configFormat) createConfig(basepath, configFormat)
@ -185,7 +185,7 @@ func NewTheme(cmd *cobra.Command, args []string) error {
createpath := helpers.AbsPathify(filepath.Join(viper.GetString("themesDir"), args[0])) createpath := helpers.AbsPathify(filepath.Join(viper.GetString("themesDir"), args[0]))
jww.INFO.Println("creating theme at", createpath) jww.INFO.Println("creating theme at", createpath)
if x, _ := helpers.Exists(createpath, hugofs.SourceFs); x { if x, _ := helpers.Exists(createpath, hugofs.Source()); x {
return newUserError(createpath, "already exists") return newUserError(createpath, "already exists")
} }
@ -204,7 +204,7 @@ func NewTheme(cmd *cobra.Command, args []string) error {
archDefault := []byte("+++\n+++\n") archDefault := []byte("+++\n+++\n")
err := helpers.WriteToDisk(filepath.Join(createpath, "archetypes", "default.md"), bytes.NewReader(archDefault), hugofs.SourceFs) err := helpers.WriteToDisk(filepath.Join(createpath, "archetypes", "default.md"), bytes.NewReader(archDefault), hugofs.Source())
if err != nil { if err != nil {
return err return err
} }
@ -234,7 +234,7 @@ IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
`) `)
err = helpers.WriteToDisk(filepath.Join(createpath, "LICENSE.md"), bytes.NewReader(by), hugofs.SourceFs) err = helpers.WriteToDisk(filepath.Join(createpath, "LICENSE.md"), bytes.NewReader(by), hugofs.Source())
if err != nil { if err != nil {
return err return err
} }
@ -256,7 +256,7 @@ func mkdir(x ...string) {
func touchFile(x ...string) { func touchFile(x ...string) {
inpath := filepath.Join(x...) inpath := filepath.Join(x...)
mkdir(filepath.Dir(inpath)) mkdir(filepath.Dir(inpath))
err := helpers.WriteToDisk(inpath, bytes.NewReader([]byte{}), hugofs.SourceFs) err := helpers.WriteToDisk(inpath, bytes.NewReader([]byte{}), hugofs.Source())
if err != nil { if err != nil {
jww.FATAL.Fatalln(err) jww.FATAL.Fatalln(err)
} }
@ -287,7 +287,7 @@ min_version = 0.15
repo = "" repo = ""
`) `)
err = helpers.WriteToDisk(filepath.Join(inpath, "theme.toml"), bytes.NewReader(by), hugofs.SourceFs) err = helpers.WriteToDisk(filepath.Join(inpath, "theme.toml"), bytes.NewReader(by), hugofs.Source())
if err != nil { if err != nil {
return return
} }
@ -321,7 +321,7 @@ func createConfig(inpath string, kind string) (err error) {
return err return err
} }
err = helpers.WriteToDisk(filepath.Join(inpath, "config."+kind), bytes.NewReader(by), hugofs.SourceFs) err = helpers.WriteToDisk(filepath.Join(inpath, "config."+kind), bytes.NewReader(by), hugofs.Source())
if err != nil { if err != nil {
return return
} }

31
commands/new_test.go
View file

@ -1,4 +1,4 @@
// Copyright 2015 The Hugo Authors. All rights reserved. // Copyright 2016 The Hugo Authors. All rights reserved.
// //
// Licensed under the Apache License, Version 2.0 (the "License"); // Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License. // you may not use this file except in compliance with the License.
@ -18,7 +18,6 @@ import (
"path/filepath" "path/filepath"
"testing" "testing"
"github.com/spf13/afero"
"github.com/spf13/hugo/hugofs" "github.com/spf13/hugo/hugofs"
"github.com/stretchr/testify/assert" "github.com/stretchr/testify/assert"
) )
@ -41,14 +40,14 @@ func checkNewSiteInited(basepath string, t *testing.T) {
} }
for _, path := range paths { for _, path := range paths {
_, err := hugofs.SourceFs.Stat(path) _, err := hugofs.Source().Stat(path)
assert.Nil(t, err) assert.Nil(t, err)
} }
} }
func TestDoNewSite(t *testing.T) { func TestDoNewSite(t *testing.T) {
basepath := filepath.Join(os.TempDir(), "blog") basepath := filepath.Join(os.TempDir(), "blog")
hugofs.SourceFs = new(afero.MemMapFs) hugofs.InitMemFs()
err := doNewSite(basepath, false) err := doNewSite(basepath, false)
assert.Nil(t, err) assert.Nil(t, err)
@ -57,17 +56,17 @@ func TestDoNewSite(t *testing.T) {
func TestDoNewSite_noerror_base_exists_but_empty(t *testing.T) { func TestDoNewSite_noerror_base_exists_but_empty(t *testing.T) {
basepath := filepath.Join(os.TempDir(), "blog") basepath := filepath.Join(os.TempDir(), "blog")
hugofs.SourceFs = new(afero.MemMapFs) hugofs.InitMemFs()
hugofs.SourceFs.MkdirAll(basepath, 777) hugofs.Source().MkdirAll(basepath, 777)
err := doNewSite(basepath, false) err := doNewSite(basepath, false)
assert.Nil(t, err) assert.Nil(t, err)
} }
func TestDoNewSite_error_base_exists(t *testing.T) { func TestDoNewSite_error_base_exists(t *testing.T) {
basepath := filepath.Join(os.TempDir(), "blog") basepath := filepath.Join(os.TempDir(), "blog")
hugofs.SourceFs = new(afero.MemMapFs) hugofs.InitMemFs()
hugofs.SourceFs.MkdirAll(basepath, 777) hugofs.Source().MkdirAll(basepath, 777)
hugofs.SourceFs.Create(filepath.Join(basepath, "foo")) hugofs.Source().Create(filepath.Join(basepath, "foo"))
// Since the directory already exists and isn't empty, expect an error // Since the directory already exists and isn't empty, expect an error
err := doNewSite(basepath, false) err := doNewSite(basepath, false)
assert.NotNil(t, err) assert.NotNil(t, err)
@ -75,8 +74,8 @@ func TestDoNewSite_error_base_exists(t *testing.T) {
func TestDoNewSite_force_empty_dir(t *testing.T) { func TestDoNewSite_force_empty_dir(t *testing.T) {
basepath := filepath.Join(os.TempDir(), "blog") basepath := filepath.Join(os.TempDir(), "blog")
hugofs.SourceFs = new(afero.MemMapFs) hugofs.InitMemFs()
hugofs.SourceFs.MkdirAll(basepath, 777) hugofs.Source().MkdirAll(basepath, 777)
err := doNewSite(basepath, true) err := doNewSite(basepath, true)
assert.Nil(t, err) assert.Nil(t, err)
@ -86,8 +85,8 @@ func TestDoNewSite_force_empty_dir(t *testing.T) {
func TestDoNewSite_error_force_dir_inside_exists(t *testing.T) { func TestDoNewSite_error_force_dir_inside_exists(t *testing.T) {
basepath := filepath.Join(os.TempDir(), "blog") basepath := filepath.Join(os.TempDir(), "blog")
contentPath := filepath.Join(basepath, "content") contentPath := filepath.Join(basepath, "content")
hugofs.SourceFs = new(afero.MemMapFs) hugofs.InitMemFs()
hugofs.SourceFs.MkdirAll(contentPath, 777) hugofs.Source().MkdirAll(contentPath, 777)
err := doNewSite(basepath, true) err := doNewSite(basepath, true)
assert.NotNil(t, err) assert.NotNil(t, err)
} }
@ -95,9 +94,9 @@ func TestDoNewSite_error_force_dir_inside_exists(t *testing.T) {
func TestDoNewSite_error_force_config_inside_exists(t *testing.T) { func TestDoNewSite_error_force_config_inside_exists(t *testing.T) {
basepath := filepath.Join(os.TempDir(), "blog") basepath := filepath.Join(os.TempDir(), "blog")
configPath := filepath.Join(basepath, "config.toml") configPath := filepath.Join(basepath, "config.toml")
hugofs.SourceFs = new(afero.MemMapFs) hugofs.InitMemFs()
hugofs.SourceFs.MkdirAll(basepath, 777) hugofs.Source().MkdirAll(basepath, 777)
hugofs.SourceFs.Create(configPath) hugofs.Source().Create(configPath)
err := doNewSite(basepath, true) err := doNewSite(basepath, true)
assert.NotNil(t, err) assert.NotNil(t, err)
} }

6
commands/server.go
View file

@ -1,4 +1,4 @@
// Copyright 2015 The Hugo Authors. All rights reserved. // Copyright 2016 The Hugo Authors. All rights reserved.
// //
// Licensed under the Apache License, Version 2.0 (the "License"); // Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License. // you may not use this file except in compliance with the License.
@ -152,7 +152,7 @@ func server(cmd *cobra.Command, args []string) error {
// Hugo writes the output to memory instead of the disk // Hugo writes the output to memory instead of the disk
if !renderToDisk { if !renderToDisk {
hugofs.DestinationFS = new(afero.MemMapFs) hugofs.SetDestination(new(afero.MemMapFs))
// Rendering to memoryFS, publish to Root regardless of publishDir. // Rendering to memoryFS, publish to Root regardless of publishDir.
viper.Set("PublishDir", "/") viper.Set("PublishDir", "/")
} }
@ -191,7 +191,7 @@ func serve(port int) {
jww.FEEDBACK.Println("Serving pages from memory") jww.FEEDBACK.Println("Serving pages from memory")
} }
httpFs := afero.NewHttpFs(hugofs.DestinationFS) httpFs := afero.NewHttpFs(hugofs.Destination())
fs := filesOnlyFs{httpFs.Dir(helpers.AbsPathify(viper.GetString("PublishDir")))} fs := filesOnlyFs{httpFs.Dir(helpers.AbsPathify(viper.GetString("PublishDir")))}
fileserver := http.FileServer(fs) fileserver := http.FileServer(fs)

12
create/content_test.go
View file

@ -44,19 +44,19 @@ func TestNewContent(t *testing.T) {
} }
for i, c := range cases { for i, c := range cases {
err = create.NewContent(hugofs.SourceFs, c.kind, c.path) err = create.NewContent(hugofs.Source(), c.kind, c.path)
if err != nil { if err != nil {
t.Errorf("[%d] NewContent: %s", i, err) t.Errorf("[%d] NewContent: %s", i, err)
} }
fname := filepath.Join(os.TempDir(), "content", filepath.FromSlash(c.path)) fname := filepath.Join(os.TempDir(), "content", filepath.FromSlash(c.path))
_, err = hugofs.SourceFs.Stat(fname) _, err = hugofs.Source().Stat(fname)
if err != nil { if err != nil {
t.Errorf("[%d] Stat: %s", i, err) t.Errorf("[%d] Stat: %s", i, err)
} }
for _, v := range c.resultStrings { for _, v := range c.resultStrings {
found, err := afero.FileContainsBytes(hugofs.SourceFs, fname, []byte(v)) found, err := afero.FileContainsBytes(hugofs.Source(), fname, []byte(v))
if err != nil { if err != nil {
t.Errorf("[%d] FileContainsBytes: %s", i, err) t.Errorf("[%d] FileContainsBytes: %s", i, err)
} }
@ -77,7 +77,7 @@ func initViper() {
} }
func initFs() error { func initFs() error {
hugofs.SourceFs = new(afero.MemMapFs) hugofs.SetSource(new(afero.MemMapFs))
perm := os.FileMode(0755) perm := os.FileMode(0755)
var err error var err error
@ -89,7 +89,7 @@ func initFs() error {
} }
for _, dir := range dirs { for _, dir := range dirs {
dir = filepath.Join(os.TempDir(), dir) dir = filepath.Join(os.TempDir(), dir)
err = hugofs.SourceFs.Mkdir(dir, perm) err = hugofs.Source().Mkdir(dir, perm)
if err != nil { if err != nil {
return err return err
} }
@ -109,7 +109,7 @@ func initFs() error {
content: "+++\n+++\n", content: "+++\n+++\n",
}, },
} { } {
f, err := hugofs.SourceFs.Create(v.path) f, err := hugofs.Source().Create(v.path)
if err != nil { if err != nil {
return err return err
} }

8
docs/content/templates/functions.md
View file

@ -83,7 +83,7 @@ e.g. Pass into "foo.html" a map with the keys "important, content"
or create a map on the fly to pass into or create a map on the fly to pass into
{{partial "foo" (dict "important" "Smiles" "content" "You should do more")}} {{partial "foo" (dict "important" "Smiles" "content" "You should do more")}}
### slice ### slice
@ -336,6 +336,12 @@ e.g.
{{ .Content }} {{ .Content }}
{{ end }} {{ end }}
## Files
### readFile
Reads a file from disk and converts it into a string. Note that the filename must be relative to the current project working dir.
So, if you have a file with the name `README.txt` in the root of your project with the content `Hugo Rocks!`:
`{{readFile "README.txt"}}` → `"Hugo Rocks!"`
## Math ## Math

4
helpers/pygments.go
View file

@ -1,4 +1,4 @@
// Copyright 2015 The Hugo Authors. All rights reserved. // Copyright 2016 The Hugo Authors. All rights reserved.
// //
// Licensed under the Apache License, Version 2.0 (the "License"); // Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License. // you may not use this file except in compliance with the License.
@ -60,7 +60,7 @@ func Highlight(code, lang, optsStr string) string {
io.WriteString(hash, lang) io.WriteString(hash, lang)
io.WriteString(hash, options) io.WriteString(hash, options)
fs := hugofs.OsFs fs := hugofs.Os()
cacheDir := viper.GetString("CacheDir") cacheDir := viper.GetString("CacheDir")
var cachefile string var cachefile string

83
hugofs/fs.go
View file

@ -1,4 +1,4 @@
// Copyright 2015 The Hugo Authors. All rights reserved. // Copyright 2016 The Hugo Authors. All rights reserved.
// //
// Licensed under the Apache License, Version 2.0 (the "License"); // Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License. // you may not use this file except in compliance with the License.
@ -13,8 +13,81 @@
package hugofs package hugofs
import "github.com/spf13/afero" import (
"github.com/spf13/afero"
"github.com/spf13/viper"
)
var SourceFs afero.Fs = new(afero.OsFs) var (
var DestinationFS afero.Fs = new(afero.OsFs) sourceFs afero.Fs
var OsFs afero.Fs = new(afero.OsFs) destinationFs afero.Fs
osFs afero.Fs = &afero.OsFs{}
workingDirFs *afero.BasePathFs
)
// Source returns Hugo's source file system.
func Source() afero.Fs {
return sourceFs
}
// SetSource sets Hugo's source file system
// and re-initializes dependent file systems.
func SetSource(fs afero.Fs) {
sourceFs = fs
initSourceDependencies()
}
// Destination returns Hugo's destionation file system.
func Destination() afero.Fs {
return destinationFs
}
// SetDestination sets Hugo's destionation file system
func SetDestination(fs afero.Fs) {
destinationFs = fs
}
// Os returns an OS file system.
func Os() afero.Fs {
return osFs
}
// WorkingDir returns a read-only file system
// restricted to the project working dir.
func WorkingDir() *afero.BasePathFs {
return workingDirFs
}
// InitFs initializes with the OS file system
// as source and destination file systems.
func InitDefaultFs() {
InitFs(&afero.OsFs{})
}
// InitMemFs initializes with a MemMapFs as source and destination file systems.
// Useful for testing.
func InitMemFs() {
InitFs(&afero.MemMapFs{})
}
// InitFs initializes with the given file system
// as source and destination file systems.
func InitFs(fs afero.Fs) {
sourceFs = fs
destinationFs = fs
initSourceDependencies()
}
func initSourceDependencies() {
workingDir := viper.GetString("WorkingDir")
if workingDir != "" {
workingDirFs = afero.NewBasePathFs(afero.NewReadOnlyFs(sourceFs), workingDir).(*afero.BasePathFs)
}
}
func init() {
InitDefaultFs()
}

72
hugofs/fs_test.go Normal file
View file

@ -0,0 +1,72 @@
// Copyright 2016 The Hugo Authors. All rights reserved.
//
// 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 hugofs
import (
"github.com/spf13/afero"
"github.com/spf13/viper"
"github.com/stretchr/testify/assert"
"testing"
)
func TestInitDefault(t *testing.T) {
viper.Reset()
defer viper.Reset()
InitDefaultFs()
assert.IsType(t, new(afero.OsFs), Source())
assert.IsType(t, new(afero.OsFs), Destination())
assert.IsType(t, new(afero.OsFs), Os())
assert.Nil(t, WorkingDir())
}
func TestInitMemFs(t *testing.T) {
viper.Reset()
defer viper.Reset()
InitMemFs()
assert.IsType(t, new(afero.MemMapFs), Source())
assert.IsType(t, new(afero.MemMapFs), Destination())
assert.IsType(t, new(afero.OsFs), Os())
assert.Nil(t, WorkingDir())
}
func TestSetSource(t *testing.T) {
InitMemFs()
SetSource(new(afero.OsFs))
assert.IsType(t, new(afero.OsFs), Source())
}
func TestSetDestination(t *testing.T) {
InitMemFs()
SetDestination(new(afero.OsFs))
assert.IsType(t, new(afero.OsFs), Destination())
}
func TestWorkingDir(t *testing.T) {
viper.Reset()
defer viper.Reset()
viper.Set("WorkingDir", "/a/b/")
InitMemFs()
assert.IsType(t, new(afero.BasePathFs), WorkingDir())
}

5
hugolib/handler_test.go
View file

@ -17,7 +17,6 @@ import (
"path/filepath" "path/filepath"
"testing" "testing"
"github.com/spf13/afero"
"github.com/spf13/hugo/helpers" "github.com/spf13/hugo/helpers"
"github.com/spf13/hugo/hugofs" "github.com/spf13/hugo/hugofs"
"github.com/spf13/hugo/source" "github.com/spf13/hugo/source"
@ -29,7 +28,7 @@ func TestDefaultHandler(t *testing.T) {
viper.Reset() viper.Reset()
defer viper.Reset() defer viper.Reset()
hugofs.DestinationFS = new(afero.MemMapFs) hugofs.InitMemFs()
sources := []source.ByteSource{ sources := []source.ByteSource{
{filepath.FromSlash("sect/doc1.html"), []byte("---\nmarkup: markdown\n---\n# title\nsome *content*")}, {filepath.FromSlash("sect/doc1.html"), []byte("---\nmarkup: markdown\n---\n# title\nsome *content*")},
{filepath.FromSlash("sect/doc2.html"), []byte("<!doctype html><html><body>more content</body></html>")}, {filepath.FromSlash("sect/doc2.html"), []byte("<!doctype html><html><body>more content</body></html>")},
@ -75,7 +74,7 @@ func TestDefaultHandler(t *testing.T) {
} }
for _, test := range tests { for _, test := range tests {
file, err := hugofs.DestinationFS.Open(test.doc) file, err := hugofs.Destination().Open(test.doc)
if err != nil { if err != nil {
t.Fatalf("Did not find %s in target.", test.doc) t.Fatalf("Did not find %s in target.", test.doc)
} }

5
hugolib/menu_test.go
View file

@ -1,4 +1,4 @@
// Copyright 2015 The Hugo Authors. All rights reserved. // Copyright 2016 The Hugo Authors. All rights reserved.
// //
// Licensed under the Apache License, Version 2.0 (the "License"); // Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License. // you may not use this file except in compliance with the License.
@ -22,7 +22,6 @@ import (
"github.com/BurntSushi/toml" "github.com/BurntSushi/toml"
"github.com/kr/pretty" "github.com/kr/pretty"
"github.com/spf13/afero"
"github.com/spf13/hugo/hugofs" "github.com/spf13/hugo/hugofs"
"github.com/spf13/hugo/source" "github.com/spf13/hugo/source"
"github.com/spf13/viper" "github.com/spf13/viper"
@ -684,7 +683,7 @@ func setupMenuTests(t *testing.T, pageSources []source.ByteSource) *Site {
} }
func createTestSite(pageSources []source.ByteSource) *Site { func createTestSite(pageSources []source.ByteSource) *Site {
hugofs.DestinationFS = new(afero.MemMapFs) hugofs.InitMemFs()
s := &Site{ s := &Site{
Source: &source.InMemorySource{ByteSource: pageSources}, Source: &source.InMemorySource{ByteSource: pageSources},

6
hugolib/page.go
View file

@ -1,4 +1,4 @@
// Copyright 2015 The Hugo Authors. All rights reserved. // Copyright 2016 The Hugo Authors. All rights reserved.
// //
// Licensed under the Apache License, Version 2.0 (the "License"); // Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License. // you may not use this file except in compliance with the License.
@ -890,9 +890,9 @@ func (p *Page) saveSource(by []byte, inpath string, safe bool) (err error) {
jww.INFO.Println("creating", inpath) jww.INFO.Println("creating", inpath)
if safe { if safe {
err = helpers.SafeWriteToDisk(inpath, bytes.NewReader(by), hugofs.SourceFs) err = helpers.SafeWriteToDisk(inpath, bytes.NewReader(by), hugofs.Source())
} else { } else {
err = helpers.WriteToDisk(inpath, bytes.NewReader(by), hugofs.SourceFs) err = helpers.WriteToDisk(inpath, bytes.NewReader(by), hugofs.Source())
} }
if err != nil { if err != nil {
return return

18
hugolib/robotstxt_test.go
View file

@ -1,10 +1,22 @@
// Copyright 2016 The Hugo Authors. All rights reserved.
//
// 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 hugolib package hugolib
import ( import (
"bytes" "bytes"
"testing" "testing"
"github.com/spf13/afero"
"github.com/spf13/hugo/helpers" "github.com/spf13/hugo/helpers"
"github.com/spf13/hugo/hugofs" "github.com/spf13/hugo/hugofs"
"github.com/spf13/hugo/source" "github.com/spf13/hugo/source"
@ -21,7 +33,7 @@ func TestRobotsTXTOutput(t *testing.T) {
viper.Reset() viper.Reset()
defer viper.Reset() defer viper.Reset()
hugofs.DestinationFS = new(afero.MemMapFs) hugofs.InitMemFs()
viper.Set("baseurl", "http://auth/bub/") viper.Set("baseurl", "http://auth/bub/")
@ -53,7 +65,7 @@ func TestRobotsTXTOutput(t *testing.T) {
t.Fatalf("Unable to RenderRobotsTXT :%s", err) t.Fatalf("Unable to RenderRobotsTXT :%s", err)
} }
robotsFile, err := hugofs.DestinationFS.Open("robots.txt") robotsFile, err := hugofs.Destination().Open("robots.txt")
if err != nil { if err != nil {
t.Fatalf("Unable to locate: robots.txt") t.Fatalf("Unable to locate: robots.txt")

7
hugolib/rss_test.go
View file

@ -1,4 +1,4 @@
// Copyright 2015 The Hugo Authors. All rights reserved. // Copyright 2016 The Hugo Authors. All rights reserved.
// //
// Licensed under the Apache License, Version 2.0 (the "License"); // Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License. // you may not use this file except in compliance with the License.
@ -17,7 +17,6 @@ import (
"bytes" "bytes"
"testing" "testing"
"github.com/spf13/afero"
"github.com/spf13/hugo/helpers" "github.com/spf13/hugo/helpers"
"github.com/spf13/hugo/hugofs" "github.com/spf13/hugo/hugofs"
"github.com/spf13/hugo/source" "github.com/spf13/hugo/source"
@ -53,7 +52,7 @@ func TestRSSOutput(t *testing.T) {
viper.Set("baseurl", "http://auth/bub/") viper.Set("baseurl", "http://auth/bub/")
viper.Set("RSSUri", rssURI) viper.Set("RSSUri", rssURI)
hugofs.DestinationFS = new(afero.MemMapFs) hugofs.InitMemFs()
s := &Site{ s := &Site{
Source: &source.InMemorySource{ByteSource: WEIGHTED_SOURCES}, Source: &source.InMemorySource{ByteSource: WEIGHTED_SOURCES},
} }
@ -72,7 +71,7 @@ func TestRSSOutput(t *testing.T) {
t.Fatalf("Unable to RenderHomePage: %s", err) t.Fatalf("Unable to RenderHomePage: %s", err)
} }
file, err := hugofs.DestinationFS.Open(rssURI) file, err := hugofs.Destination().Open(rssURI)
if err != nil { if err != nil {
t.Fatalf("Unable to locate: %s", rssURI) t.Fatalf("Unable to locate: %s", rssURI)

2
hugolib/shortcode_test.go
View file

@ -487,7 +487,7 @@ e`,
createAndRenderPages(t, s) createAndRenderPages(t, s)
for _, test := range tests { for _, test := range tests {
file, err := hugofs.DestinationFS.Open(test.outFile) file, err := hugofs.Destination().Open(test.outFile)
if err != nil { if err != nil {
t.Fatalf("Did not find %s in target: %s", test.outFile, err) t.Fatalf("Did not find %s in target: %s", test.outFile, err)

4
hugolib/site.go
View file

@ -529,7 +529,7 @@ func (s *Site) ReBuild(events []fsnotify.Event) error {
// it's been updated // it's been updated
if ev.Op&fsnotify.Rename == fsnotify.Rename { if ev.Op&fsnotify.Rename == fsnotify.Rename {
// If the file is still on disk, it's only been updated, if it's not, it's been moved // If the file is still on disk, it's only been updated, if it's not, it's been moved
if ex, err := afero.Exists(hugofs.SourceFs, ev.Name); !ex || err != nil { if ex, err := afero.Exists(hugofs.Source(), ev.Name); !ex || err != nil {
path, _ := helpers.GetRelativePath(ev.Name, s.absContentDir()) path, _ := helpers.GetRelativePath(ev.Name, s.absContentDir())
s.RemovePageByPath(path) s.RemovePageByPath(path)
continue continue
@ -852,7 +852,7 @@ func (s *Site) absPublishDir() string {
} }
func (s *Site) checkDirectories() (err error) { func (s *Site) checkDirectories() (err error) {
if b, _ := helpers.DirExists(s.absContentDir(), hugofs.SourceFs); !b { if b, _ := helpers.DirExists(s.absContentDir(), hugofs.Source()); !b {
return fmt.Errorf("No source directory found, expecting to find it at " + s.absContentDir()) return fmt.Errorf("No source directory found, expecting to find it at " + s.absContentDir())
} }
return return

37
hugolib/site_test.go
View file

@ -1,4 +1,4 @@
// Copyright 2015 The Hugo Authors. All rights reserved. // Copyright 2016 The Hugo Authors. All rights reserved.
// //
// Licensed under the Apache License, Version 2.0 (the "License"); // Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License. // you may not use this file except in compliance with the License.
@ -25,7 +25,6 @@ import (
"bitbucket.org/pkg/inflect" "bitbucket.org/pkg/inflect"
"github.com/spf13/afero"
"github.com/spf13/hugo/helpers" "github.com/spf13/hugo/helpers"
"github.com/spf13/hugo/hugofs" "github.com/spf13/hugo/hugofs"
"github.com/spf13/hugo/source" "github.com/spf13/hugo/source"
@ -200,7 +199,7 @@ func TestRenderThingOrDefault(t *testing.T) {
{false, TEMPLATE_FUNC, HTML("simple-template")}, {false, TEMPLATE_FUNC, HTML("simple-template")},
} }
hugofs.DestinationFS = new(afero.MemMapFs) hugofs.InitMemFs()
for i, test := range tests { for i, test := range tests {
@ -226,7 +225,7 @@ func TestRenderThingOrDefault(t *testing.T) {
t.Errorf("Unable to render html: %s", err) t.Errorf("Unable to render html: %s", err)
} }
file, err := hugofs.DestinationFS.Open(filepath.FromSlash("out/index.html")) file, err := hugofs.Destination().Open(filepath.FromSlash("out/index.html"))
if err != nil { if err != nil {
t.Errorf("Unable to open html: %s", err) t.Errorf("Unable to open html: %s", err)
} }
@ -240,7 +239,7 @@ func TestDraftAndFutureRender(t *testing.T) {
viper.Reset() viper.Reset()
defer viper.Reset() defer viper.Reset()
hugofs.DestinationFS = new(afero.MemMapFs) hugofs.InitMemFs()
sources := []source.ByteSource{ sources := []source.ByteSource{
{filepath.FromSlash("sect/doc1.md"), []byte("---\ntitle: doc1\ndraft: true\npublishdate: \"2414-05-29\"\n---\n# doc1\n*some content*")}, {filepath.FromSlash("sect/doc1.md"), []byte("---\ntitle: doc1\ndraft: true\npublishdate: \"2414-05-29\"\n---\n# doc1\n*some content*")},
{filepath.FromSlash("sect/doc2.md"), []byte("---\ntitle: doc2\ndraft: true\npublishdate: \"2012-05-29\"\n---\n# doc2\n*some content*")}, {filepath.FromSlash("sect/doc2.md"), []byte("---\ntitle: doc2\ndraft: true\npublishdate: \"2012-05-29\"\n---\n# doc2\n*some content*")},
@ -299,7 +298,7 @@ func TestDraftAndFutureRender(t *testing.T) {
// Issue #957 // Issue #957
func TestCrossrefs(t *testing.T) { func TestCrossrefs(t *testing.T) {
hugofs.DestinationFS = new(afero.MemMapFs) hugofs.InitMemFs()
for _, uglyURLs := range []bool{true, false} { for _, uglyURLs := range []bool{true, false} {
for _, relative := range []bool{true, false} { for _, relative := range []bool{true, false} {
doTestCrossrefs(t, relative, uglyURLs) doTestCrossrefs(t, relative, uglyURLs)
@ -374,7 +373,7 @@ THE END.`, refShortcode))},
} }
for _, test := range tests { for _, test := range tests {
file, err := hugofs.DestinationFS.Open(test.doc) file, err := hugofs.Destination().Open(test.doc)
if err != nil { if err != nil {
t.Fatalf("Did not find %s in target: %s", test.doc, err) t.Fatalf("Did not find %s in target: %s", test.doc, err)
@ -392,7 +391,7 @@ THE END.`, refShortcode))},
// Issue #939 // Issue #939
// Issue #1923 // Issue #1923
func TestShouldAlwaysHaveUglyURLs(t *testing.T) { func TestShouldAlwaysHaveUglyURLs(t *testing.T) {
hugofs.DestinationFS = new(afero.MemMapFs) hugofs.InitMemFs()
for _, uglyURLs := range []bool{true, false} { for _, uglyURLs := range []bool{true, false} {
doTestShouldAlwaysHaveUglyURLs(t, uglyURLs) doTestShouldAlwaysHaveUglyURLs(t, uglyURLs)
} }
@ -462,7 +461,7 @@ func doTestShouldAlwaysHaveUglyURLs(t *testing.T, uglyURLs bool) {
} }
for _, test := range tests { for _, test := range tests {
file, err := hugofs.DestinationFS.Open(test.doc) file, err := hugofs.Destination().Open(test.doc)
if err != nil { if err != nil {
t.Fatalf("Did not find %s in target: %s", test.doc, err) t.Fatalf("Did not find %s in target: %s", test.doc, err)
} }
@ -489,7 +488,7 @@ func TestSectionNaming(t *testing.T) {
} }
func doTestSectionNaming(t *testing.T, canonify, uglify, pluralize bool) { func doTestSectionNaming(t *testing.T, canonify, uglify, pluralize bool) {
hugofs.DestinationFS = new(afero.MemMapFs) hugofs.InitMemFs()
viper.Reset() viper.Reset()
defer viper.Reset() defer viper.Reset()
viper.Set("baseurl", "http://auth/sub/") viper.Set("baseurl", "http://auth/sub/")
@ -539,7 +538,7 @@ func doTestSectionNaming(t *testing.T, canonify, uglify, pluralize bool) {
} }
for _, test := range tests { for _, test := range tests {
file, err := hugofs.DestinationFS.Open(test.doc) file, err := hugofs.Destination().Open(test.doc)
if err != nil { if err != nil {
t.Fatalf("Did not find %s in target: %s", test.doc, err) t.Fatalf("Did not find %s in target: %s", test.doc, err)
} }
@ -560,7 +559,7 @@ func TestSkipRender(t *testing.T) {
viper.Reset() viper.Reset()
defer viper.Reset() defer viper.Reset()
hugofs.DestinationFS = new(afero.MemMapFs) hugofs.InitMemFs()
sources := []source.ByteSource{ sources := []source.ByteSource{
{filepath.FromSlash("sect/doc1.html"), []byte("---\nmarkup: markdown\n---\n# title\nsome *content*")}, {filepath.FromSlash("sect/doc1.html"), []byte("---\nmarkup: markdown\n---\n# title\nsome *content*")},
{filepath.FromSlash("sect/doc2.html"), []byte("<!doctype html><html><body>more content</body></html>")}, {filepath.FromSlash("sect/doc2.html"), []byte("<!doctype html><html><body>more content</body></html>")},
@ -605,7 +604,7 @@ func TestSkipRender(t *testing.T) {
} }
for _, test := range tests { for _, test := range tests {
file, err := hugofs.DestinationFS.Open(test.doc) file, err := hugofs.Destination().Open(test.doc)
if err != nil { if err != nil {
t.Fatalf("Did not find %s in target.", test.doc) t.Fatalf("Did not find %s in target.", test.doc)
} }
@ -624,7 +623,7 @@ func TestAbsURLify(t *testing.T) {
viper.Set("DefaultExtension", "html") viper.Set("DefaultExtension", "html")
hugofs.DestinationFS = new(afero.MemMapFs) hugofs.InitMemFs()
sources := []source.ByteSource{ sources := []source.ByteSource{
{filepath.FromSlash("sect/doc1.html"), []byte("<!doctype html><html><head></head><body><a href=\"#frag1\">link</a></body></html>")}, {filepath.FromSlash("sect/doc1.html"), []byte("<!doctype html><html><head></head><body><a href=\"#frag1\">link</a></body></html>")},
{filepath.FromSlash("content/blue/doc2.html"), []byte("---\nf: t\n---\n<!doctype html><html><body>more content</body></html>")}, {filepath.FromSlash("content/blue/doc2.html"), []byte("---\nf: t\n---\n<!doctype html><html><body>more content</body></html>")},
@ -662,7 +661,7 @@ func TestAbsURLify(t *testing.T) {
for _, test := range tests { for _, test := range tests {
file, err := hugofs.DestinationFS.Open(filepath.FromSlash(test.file)) file, err := hugofs.Destination().Open(filepath.FromSlash(test.file))
if err != nil { if err != nil {
t.Fatalf("Unable to locate rendered content: %s", test.file) t.Fatalf("Unable to locate rendered content: %s", test.file)
} }
@ -730,7 +729,7 @@ func TestOrderedPages(t *testing.T) {
viper.Reset() viper.Reset()
defer viper.Reset() defer viper.Reset()
hugofs.DestinationFS = new(afero.MemMapFs) hugofs.InitMemFs()
viper.Set("baseurl", "http://auth/bub") viper.Set("baseurl", "http://auth/bub")
s := &Site{ s := &Site{
@ -804,7 +803,7 @@ func TestGroupedPages(t *testing.T) {
} }
}() }()
hugofs.DestinationFS = new(afero.MemMapFs) hugofs.InitMemFs()
viper.Set("baseurl", "http://auth/bub") viper.Set("baseurl", "http://auth/bub")
s := &Site{ s := &Site{
@ -984,7 +983,7 @@ func TestWeightedTaxonomies(t *testing.T) {
viper.Reset() viper.Reset()
defer viper.Reset() defer viper.Reset()
hugofs.DestinationFS = new(afero.MemMapFs) hugofs.InitMemFs()
sources := []source.ByteSource{ sources := []source.ByteSource{
{filepath.FromSlash("sect/doc1.md"), PAGE_WITH_WEIGHTED_TAXONOMIES_1}, {filepath.FromSlash("sect/doc1.md"), PAGE_WITH_WEIGHTED_TAXONOMIES_1},
{filepath.FromSlash("sect/doc2.md"), PAGE_WITH_WEIGHTED_TAXONOMIES_2}, {filepath.FromSlash("sect/doc2.md"), PAGE_WITH_WEIGHTED_TAXONOMIES_2},
@ -1039,7 +1038,7 @@ func findPage(site *Site, f string) *Page {
} }
func setupLinkingMockSite(t *testing.T) *Site { func setupLinkingMockSite(t *testing.T) *Site {
hugofs.DestinationFS = new(afero.MemMapFs) hugofs.InitMemFs()
sources := []source.ByteSource{ sources := []source.ByteSource{
{filepath.FromSlash("index.md"), []byte("")}, {filepath.FromSlash("index.md"), []byte("")},
{filepath.FromSlash("rootfile.md"), []byte("")}, {filepath.FromSlash("rootfile.md"), []byte("")},

9
hugolib/site_url_test.go
View file

@ -1,4 +1,4 @@
// Copyright 2015 The Hugo Authors. All rights reserved. // Copyright 2016 The Hugo Authors. All rights reserved.
// //
// Licensed under the Apache License, Version 2.0 (the "License"); // Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License. // you may not use this file except in compliance with the License.
@ -19,7 +19,6 @@ import (
"html/template" "html/template"
"github.com/spf13/afero"
"github.com/spf13/hugo/hugofs" "github.com/spf13/hugo/hugofs"
"github.com/spf13/hugo/source" "github.com/spf13/hugo/source"
"github.com/spf13/hugo/target" "github.com/spf13/hugo/target"
@ -88,7 +87,7 @@ func TestPageCount(t *testing.T) {
viper.Reset() viper.Reset()
defer viper.Reset() defer viper.Reset()
hugofs.DestinationFS = new(afero.MemMapFs) hugofs.InitMemFs()
viper.Set("uglyurls", false) viper.Set("uglyurls", false)
viper.Set("paginate", 10) viper.Set("paginate", 10)
@ -113,7 +112,7 @@ func TestPageCount(t *testing.T) {
t.Errorf("Unable to render site lists: %s", err) t.Errorf("Unable to render site lists: %s", err)
} }
_, err := hugofs.DestinationFS.Open("blue") _, err := hugofs.Destination().Open("blue")
if err != nil { if err != nil {
t.Errorf("No indexed rendered.") t.Errorf("No indexed rendered.")
} }
@ -129,7 +128,7 @@ func TestPageCount(t *testing.T) {
"sd3/index.html", "sd3/index.html",
"sd4.html", "sd4.html",
} { } {
if _, err := hugofs.DestinationFS.Open(filepath.FromSlash(s)); err != nil { if _, err := hugofs.Destination().Open(filepath.FromSlash(s)); err != nil {
t.Errorf("No alias rendered: %s", s) t.Errorf("No alias rendered: %s", s)
} }
} }

7
hugolib/sitemap_test.go
View file

@ -1,4 +1,4 @@
// Copyright 2015 The Hugo Authors. All rights reserved. // Copyright 2016 The Hugo Authors. All rights reserved.
// //
// Licensed under the Apache License, Version 2.0 (the "License"); // Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License. // you may not use this file except in compliance with the License.
@ -17,7 +17,6 @@ import (
"bytes" "bytes"
"testing" "testing"
"github.com/spf13/afero"
"github.com/spf13/hugo/helpers" "github.com/spf13/hugo/helpers"
"github.com/spf13/hugo/hugofs" "github.com/spf13/hugo/hugofs"
"github.com/spf13/hugo/source" "github.com/spf13/hugo/source"
@ -40,7 +39,7 @@ func TestSitemapOutput(t *testing.T) {
viper.Reset() viper.Reset()
defer viper.Reset() defer viper.Reset()
hugofs.DestinationFS = new(afero.MemMapFs) hugofs.InitMemFs()
viper.Set("baseurl", "http://auth/bub/") viper.Set("baseurl", "http://auth/bub/")
@ -72,7 +71,7 @@ func TestSitemapOutput(t *testing.T) {
t.Fatalf("Unable to RenderRobotsTXT :%s", err) t.Fatalf("Unable to RenderRobotsTXT :%s", err)
} }
sitemapFile, err := hugofs.DestinationFS.Open("sitemap.xml") sitemapFile, err := hugofs.Destination().Open("sitemap.xml")
if err != nil { if err != nil {
t.Fatalf("Unable to locate: sitemap.xml") t.Fatalf("Unable to locate: sitemap.xml")

4
source/filesystem.go
View file

@ -1,4 +1,4 @@
// Copyright 2015 The Hugo Authors. All rights reserved. // Copyright 2016 The Hugo Authors. All rights reserved.
// //
// Licensed under the Apache License, Version 2.0 (the "License"); // Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License. // you may not use this file except in compliance with the License.
@ -93,7 +93,7 @@ func (f *Filesystem) captureFiles() {
return err return err
} }
err := helpers.SymbolicWalk(hugofs.SourceFs, f.Base, walker) err := helpers.SymbolicWalk(hugofs.Source(), f.Base, walker)
if err != nil { if err != nil {
jww.ERROR.Println(err) jww.ERROR.Println(err)

4
target/file.go
View file

@ -1,4 +1,4 @@
// Copyright 2015 The Hugo Authors. All rights reserved. // Copyright 2016 The Hugo Authors. All rights reserved.
// //
// Licensed under the Apache License, Version 2.0 (the "License"); // Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License. // you may not use this file except in compliance with the License.
@ -49,7 +49,7 @@ func (fs *Filesystem) Publish(path string, r io.Reader) (err error) {
return return
} }
return helpers.WriteToDisk(translated, r, hugofs.DestinationFS) return helpers.WriteToDisk(translated, r, hugofs.Destination())
} }
func (fs *Filesystem) Translate(src string) (dest string, err error) { func (fs *Filesystem) Translate(src string) (dest string, err error) {

4
target/htmlredirect.go
View file

@ -1,4 +1,4 @@
// Copyright 2015 The Hugo Authors. All rights reserved. // Copyright 2016 The Hugo Authors. All rights reserved.
// //
// Licensed under the Apache License, Version 2.0 (the "License"); // Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License. // you may not use this file except in compliance with the License.
@ -144,5 +144,5 @@ func (h *HTMLRedirectAlias) Publish(path string, permalink string) (err error) {
return return
} }
return helpers.WriteToDisk(path, buffer, hugofs.DestinationFS) return helpers.WriteToDisk(path, buffer, hugofs.Destination())
} }

4
target/page.go
View file

@ -1,4 +1,4 @@
// Copyright 2015 The Hugo Authors. All rights reserved. // Copyright 2016 The Hugo Authors. All rights reserved.
// //
// Licensed under the Apache License, Version 2.0 (the "License"); // Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License. // you may not use this file except in compliance with the License.
@ -41,7 +41,7 @@ func (pp *PagePub) Publish(path string, r io.Reader) (err error) {
return return
} }
return helpers.WriteToDisk(translated, r, hugofs.DestinationFS) return helpers.WriteToDisk(translated, r, hugofs.Destination())
} }
func (pp *PagePub) Translate(src string) (dest string, err error) { func (pp *PagePub) Translate(src string) (dest string, err error) {

14
tpl/template.go
View file

@ -235,7 +235,7 @@ func (t *GoHTMLTemplate) AddTemplateFileWithMaster(name, overlayFilename, master
masterTpl := t.Lookup(masterFilename) masterTpl := t.Lookup(masterFilename)
if masterTpl == nil { if masterTpl == nil {
b, err := afero.ReadFile(hugofs.SourceFs, masterFilename) b, err := afero.ReadFile(hugofs.Source(), masterFilename)
if err != nil { if err != nil {
return err return err
} }
@ -248,7 +248,7 @@ func (t *GoHTMLTemplate) AddTemplateFileWithMaster(name, overlayFilename, master
} }
} }
b, err := afero.ReadFile(hugofs.SourceFs, overlayFilename) b, err := afero.ReadFile(hugofs.Source(), overlayFilename)
if err != nil { if err != nil {
return err return err
} }
@ -309,14 +309,14 @@ func (t *GoHTMLTemplate) AddTemplateFile(name, baseTemplatePath, path string) er
} }
case ".ace": case ".ace":
var innerContent, baseContent []byte var innerContent, baseContent []byte
innerContent, err := afero.ReadFile(hugofs.SourceFs, path) innerContent, err := afero.ReadFile(hugofs.Source(), path)
if err != nil { if err != nil {
return err return err
} }
if baseTemplatePath != "" { if baseTemplatePath != "" {
baseContent, err = afero.ReadFile(hugofs.SourceFs, baseTemplatePath) baseContent, err = afero.ReadFile(hugofs.Source(), baseTemplatePath)
if err != nil { if err != nil {
return err return err
} }
@ -329,7 +329,7 @@ func (t *GoHTMLTemplate) AddTemplateFile(name, baseTemplatePath, path string) er
return t.AddTemplateFileWithMaster(name, path, baseTemplatePath) return t.AddTemplateFileWithMaster(name, path, baseTemplatePath)
} }
b, err := afero.ReadFile(hugofs.SourceFs, path) b, err := afero.ReadFile(hugofs.Source(), path)
if err != nil { if err != nil {
return err return err
@ -414,7 +414,7 @@ func (t *GoHTMLTemplate) loadTemplates(absPath string, prefix string) {
// This may be a view that shouldn't have base template // This may be a view that shouldn't have base template
// Have to look inside it to make sure // Have to look inside it to make sure
needsBase, err := helpers.FileContainsAny(path, innerMarkers, hugofs.OsFs) needsBase, err := helpers.FileContainsAny(path, innerMarkers, hugofs.Os())
if err != nil { if err != nil {
return err return err
} }
@ -442,7 +442,7 @@ func (t *GoHTMLTemplate) loadTemplates(absPath string, prefix string) {
} }
for _, pathToCheck := range pathsToCheck { for _, pathToCheck := range pathsToCheck {
if ok, err := helpers.Exists(pathToCheck, hugofs.OsFs); err == nil && ok { if ok, err := helpers.Exists(pathToCheck, hugofs.Os()); err == nil && ok {
baseTemplatePath = pathToCheck baseTemplatePath = pathToCheck
break break
} }

35
tpl/template_funcs.go
View file

@ -24,6 +24,8 @@ import (
"encoding/json" "encoding/json"
"errors" "errors"
"fmt" "fmt"
"github.com/spf13/afero"
"github.com/spf13/hugo/hugofs"
"html" "html"
"html/template" "html/template"
"math/rand" "math/rand"
@ -1467,6 +1469,38 @@ func index(item interface{}, indices ...interface{}) (interface{}, error) {
return v.Interface(), nil return v.Interface(), nil
} }
// readFile reads the file named by filename relative to the given basepath
// and returns the contents as a string.
// There is a upper size limit set at 1 megabytes.
func readFile(fs *afero.BasePathFs, filename string) (string, error) {
if filename == "" {
return "", errors.New("readFile needs a filename")
}
if info, err := fs.Stat(filename); err == nil {
if info.Size() > 1000000 {
return "", fmt.Errorf("File %q is too big", filename)
}
} else {
return "", err
}
b, err := afero.ReadFile(fs, filename)
if err != nil {
return "", err
}
return string(b), nil
}
// readFileFromWorkingDir reads the file named by filename relative to the
// configured WorkingDir.
// It returns the contents as a string.
// There is a upper size limit set at 1 megabytes.
func readFileFromWorkingDir(i interface{}) (string, error) {
return readFile(hugofs.WorkingDir(), cast.ToString(i))
}
// safeHTMLAttr returns a given string as html/template HTMLAttr content. // safeHTMLAttr returns a given string as html/template HTMLAttr content.
// //
// safeHTMLAttr is currently disabled, pending further discussion // safeHTMLAttr is currently disabled, pending further discussion
@ -1689,6 +1723,7 @@ func init() {
"plainify": plainify, "plainify": plainify,
"pluralize": pluralize, "pluralize": pluralize,
"readDir": readDir, "readDir": readDir,
"readFile": readFileFromWorkingDir,
"ref": ref, "ref": ref,
"relURL": func(a string) template.HTML { return template.HTML(helpers.RelURL(a)) }, "relURL": func(a string) template.HTML { return template.HTML(helpers.RelURL(a)) },
"relref": relRef, "relref": relRef,

64
tpl/template_funcs_test.go
View file

@ -1,4 +1,4 @@
// Copyright 2015 The Hugo Authors. All rights reserved. // Copyright 2016 The Hugo Authors. All rights reserved.
// //
// Licensed under the Apache License, Version 2.0 (the "License"); // Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License. // you may not use this file except in compliance with the License.
@ -18,18 +18,20 @@ import (
"encoding/base64" "encoding/base64"
"errors" "errors"
"fmt" "fmt"
"github.com/spf13/afero"
"github.com/spf13/cast"
"github.com/spf13/hugo/hugofs"
"github.com/spf13/viper"
"github.com/stretchr/testify/assert"
"html/template" "html/template"
"math/rand" "math/rand"
"path" "path"
"path/filepath"
"reflect" "reflect"
"runtime" "runtime"
"strings" "strings"
"testing" "testing"
"time" "time"
"github.com/spf13/cast"
"github.com/spf13/viper"
"github.com/stretchr/testify/assert"
) )
type tstNoStringer struct { type tstNoStringer struct {
@ -63,6 +65,15 @@ func TestFuncsInTemplate(t *testing.T) {
viper.Reset() viper.Reset()
defer viper.Reset() defer viper.Reset()
workingDir := "/home/hugo"
viper.Set("WorkingDir", workingDir)
fs := &afero.MemMapFs{}
hugofs.InitFs(fs)
afero.WriteFile(fs, filepath.Join(workingDir, "README.txt"), []byte("Hugo Rocks!"), 0755)
// Add the examples from the docs: As a smoke test and to make sure the examples work. // Add the examples from the docs: As a smoke test and to make sure the examples work.
// TODO(bep): docs: fix title example // TODO(bep): docs: fix title example
in := in :=
@ -109,6 +120,7 @@ safeCSS: {{ "Bat&Man" | safeCSS | safeCSS }}
safeURL: {{ "http://gohugo.io" | safeURL | safeURL }} safeURL: {{ "http://gohugo.io" | safeURL | safeURL }}
safeJS: {{ "(1*2)" | safeJS | safeJS }} safeJS: {{ "(1*2)" | safeJS | safeJS }}
plainify: {{ plainify "Hello <strong>world</strong>, gophers!" }} plainify: {{ plainify "Hello <strong>world</strong>, gophers!" }}
readFile: {{ readFile "README.txt" }}
` `
expected := `chomp: <p>Blockhead</p> expected := `chomp: <p>Blockhead</p>
dateFormat: Wednesday, Jan 21, 2015 dateFormat: Wednesday, Jan 21, 2015
@ -153,6 +165,7 @@ safeCSS: Bat&amp;Man
safeURL: http://gohugo.io safeURL: http://gohugo.io
safeJS: (1*2) safeJS: (1*2)
plainify: Hello world, gophers! plainify: Hello world, gophers!
readFile: Hugo Rocks!
` `
var b bytes.Buffer var b bytes.Buffer
@ -2182,3 +2195,44 @@ func TestSHA1(t *testing.T) {
} }
} }
} }
func TestReadFile(t *testing.T) {
viper.Reset()
defer viper.Reset()
workingDir := "/home/hugo"
viper.Set("WorkingDir", workingDir)
fs := &afero.MemMapFs{}
hugofs.InitFs(fs)
afero.WriteFile(fs, filepath.Join(workingDir, "/f/f1.txt"), []byte("f1-content"), 0755)
afero.WriteFile(fs, filepath.Join("/home", "f2.txt"), []byte("f2-content"), 0755)
for i, this := range []struct {
filename string
expect interface{}
}{
{"", false},
{"b", false},
{filepath.FromSlash("/f/f1.txt"), "f1-content"},
{filepath.FromSlash("f/f1.txt"), "f1-content"},
{filepath.FromSlash("../f2.txt"), false},
} {
result, err := readFileFromWorkingDir(this.filename)
if b, ok := this.expect.(bool); ok && !b {
if err == nil {
t.Errorf("[%d] readFile didn't return an expected error", i)
}
} else {
if err != nil {
t.Errorf("[%d] readFile failed: %s", i, err)
continue
}
if result != this.expect {
t.Errorf("[%d] readFile got %q but expected %q", i, result, this.expect)
}
}
}
}

10
tpl/template_resources.go
View file

@ -1,4 +1,4 @@
// Copyright 2015 The Hugo Authors. All rights reserved. // Copyright 2016 The Hugo Authors. All rights reserved.
// //
// Licensed under the Apache License, Version 2.0 (the "License"); // Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License. // you may not use this file except in compliance with the License.
@ -177,9 +177,9 @@ func resGetResource(url string) ([]byte, error) {
return nil, nil return nil, nil
} }
if strings.Contains(url, "://") { if strings.Contains(url, "://") {
return resGetRemote(url, hugofs.SourceFs, http.DefaultClient) return resGetRemote(url, hugofs.Source(), http.DefaultClient)
} }
return resGetLocal(url, hugofs.SourceFs) return resGetLocal(url, hugofs.Source())
} }
// getJSON expects one or n-parts of a URL to a resource which can either be a local or a remote one. // getJSON expects one or n-parts of a URL to a resource which can either be a local or a remote one.
@ -201,7 +201,7 @@ func getJSON(urlParts ...string) interface{} {
jww.ERROR.Printf("Cannot read json from resource %s with error message %s", url, err) jww.ERROR.Printf("Cannot read json from resource %s with error message %s", url, err)
jww.ERROR.Printf("Retry #%d for %s and sleeping for %s", i, url, resSleep) jww.ERROR.Printf("Retry #%d for %s and sleeping for %s", i, url, resSleep)
time.Sleep(resSleep) time.Sleep(resSleep)
resDeleteCache(url, hugofs.SourceFs) resDeleteCache(url, hugofs.Source())
continue continue
} }
break break
@ -234,7 +234,7 @@ func getCSV(sep string, urlParts ...string) [][]string {
var clearCacheSleep = func(i int, u string) { var clearCacheSleep = func(i int, u string) {
jww.ERROR.Printf("Retry #%d for %s and sleeping for %s", i, url, resSleep) jww.ERROR.Printf("Retry #%d for %s and sleeping for %s", i, url, resSleep)
time.Sleep(resSleep) time.Sleep(resSleep)
resDeleteCache(url, hugofs.SourceFs) resDeleteCache(url, hugofs.Source())
} }
for i := 0; i <= resRetries; i++ { for i := 0; i <= resRetries; i++ {

4
tpl/template_resources_test.go
View file

@ -1,4 +1,4 @@
// Copyright 2015 The Hugo Authors. All rights reserved. // Copyright 2016 The Hugo Authors. All rights reserved.
// //
// Licensed under the Apache License, Version 2.0 (the "License"); // Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License. // you may not use this file except in compliance with the License.
@ -214,7 +214,7 @@ type wd struct {
func testRetryWhenDone() wd { func testRetryWhenDone() wd {
cd := viper.GetString("CacheDir") cd := viper.GetString("CacheDir")
viper.Set("CacheDir", helpers.GetTempDir("", hugofs.SourceFs)) viper.Set("CacheDir", helpers.GetTempDir("", hugofs.Source()))
var tmpSleep time.Duration var tmpSleep time.Duration
tmpSleep, resSleep = resSleep, time.Millisecond tmpSleep, resSleep = resSleep, time.Millisecond
return wd{func() { return wd{func() {

8
tpl/template_test.go
View file

@ -1,4 +1,4 @@
// Copyright 2015 The Hugo Authors. All rights reserved. // Copyright 2016 The Hugo Authors. All rights reserved.
// //
// Licensed under the Apache License, Version 2.0 (the "License"); // Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License. // you may not use this file except in compliance with the License.
@ -123,17 +123,17 @@ func TestAddTemplateFileWithMaster(t *testing.T) {
{`tpl`, `{{.0.E}}`, 0, false}, {`tpl`, `{{.0.E}}`, 0, false},
} { } {
hugofs.SourceFs = afero.NewMemMapFs() hugofs.InitMemFs()
templ := New() templ := New()
overlayTplName := "ot" overlayTplName := "ot"
masterTplName := "mt" masterTplName := "mt"
finalTplName := "tp" finalTplName := "tp"
if this.writeSkipper != 1 { if this.writeSkipper != 1 {
afero.WriteFile(hugofs.SourceFs, masterTplName, []byte(this.masterTplContent), 0644) afero.WriteFile(hugofs.Source(), masterTplName, []byte(this.masterTplContent), 0644)
} }
if this.writeSkipper != 2 { if this.writeSkipper != 2 {
afero.WriteFile(hugofs.SourceFs, overlayTplName, []byte(this.overlayTplContent), 0644) afero.WriteFile(hugofs.Source(), overlayTplName, []byte(this.overlayTplContent), 0644)
} }
err := templ.AddTemplateFileWithMaster(finalTplName, overlayTplName, masterTplName) err := templ.AddTemplateFileWithMaster(finalTplName, overlayTplName, masterTplName)