Skip to content

Commit

Permalink
Revert "Revert "Fix PostProcess regression for hugo server""
Browse files Browse the repository at this point in the history
This reverts commit 6c35a1a.

Updates #9794
  • Loading branch information
bep committed Apr 18, 2022
1 parent 5de6f8a commit e66e2e9
Show file tree
Hide file tree
Showing 4 changed files with 66 additions and 38 deletions.
28 changes: 10 additions & 18 deletions commands/commandeer.go
Expand Up @@ -395,23 +395,23 @@ func (c *commandeer) loadConfig() error {
}

c.fsCreate.Do(func() {
fs := hugofs.NewFrom(sourceFs, config)
// Assume both source and destination are using same filesystem.
fs := hugofs.NewFromSourceAndDestination(sourceFs, sourceFs, config)

if c.publishDirFs != nil {
// Need to reuse the destination on server rebuilds.
fs.PublishDir = c.publishDirFs
fs.PublishDirServer = c.publishDirServerFs
} else {
publishDir := config.GetString("publishDir")
publishDirStatic := config.GetString("publishDirStatic")
workingDir := config.GetString("workingDir")
absPublishDir := paths.AbsPathify(workingDir, publishDir)
absPublishDirStatic := paths.AbsPathify(workingDir, publishDirStatic)

if c.renderStaticToDisk {
// Writes the dynamic output oton memory,
publishDirStatic := config.GetString("publishDirStatic")
workingDir := config.GetString("workingDir")
absPublishDirStatic := paths.AbsPathify(workingDir, publishDirStatic)

fs = hugofs.NewFromSourceAndDestination(sourceFs, afero.NewMemMapFs(), config)
// Writes the dynamic output to memory,
// while serve others directly from /public on disk.
dynamicFs := afero.NewMemMapFs()
dynamicFs := fs.PublishDir
staticFs := afero.NewBasePathFs(afero.NewOsFs(), absPublishDirStatic)

// Serve from both the static and dynamic fs,
Expand All @@ -427,18 +427,10 @@ func (c *commandeer) loadConfig() error {
},
},
)
fs.PublishDir = dynamicFs
fs.PublishDirStatic = staticFs
} else if createMemFs {
// Hugo writes the output to memory instead of the disk.
fs.PublishDir = new(afero.MemMapFs)
fs.PublishDirServer = fs.PublishDir
fs.PublishDirStatic = fs.PublishDir
} else {
// Write everything to disk.
fs.PublishDir = afero.NewBasePathFs(afero.NewOsFs(), absPublishDir)
fs.PublishDirServer = fs.PublishDir
fs.PublishDirStatic = fs.PublishDir
fs = hugofs.NewFromSourceAndDestination(sourceFs, afero.NewMemMapFs(), config)
}
}

Expand Down
4 changes: 4 additions & 0 deletions commands/commands_test.go
Expand Up @@ -375,6 +375,10 @@ Single: {{ .Title }}
List: {{ .Title }}
Environment: {{ hugo.Environment }}
For issue 9788:
{{ $foo :="abc" | resources.FromString "foo.css" | minify | resources.PostProcess }}
PostProcess: {{ $foo.RelPermalink }}
`)

return dir
Expand Down
46 changes: 36 additions & 10 deletions commands/server_test.go
Expand Up @@ -31,16 +31,6 @@ import (
qt "github.com/frankban/quicktest"
)

func TestServer(t *testing.T) {
c := qt.New(t)

r := runServerTest(c, true, "")

c.Assert(r.err, qt.IsNil)
c.Assert(r.homeContent, qt.Contains, "List: Hugo Commands")
c.Assert(r.homeContent, qt.Contains, "Environment: development")
}

// Issue 9518
func TestServerPanicOnConfigError(t *testing.T) {
c := qt.New(t)
Expand Down Expand Up @@ -101,6 +91,42 @@ baseURL="https://example.org"

}

func TestServerBugs(t *testing.T) {
c := qt.New(t)

for _, test := range []struct {
name string
flag string
assert func(c *qt.C, r serverTestResult)
}{
// Issue 9788
{"PostProcess, memory", "", func(c *qt.C, r serverTestResult) {
c.Assert(r.err, qt.IsNil)
c.Assert(r.homeContent, qt.Contains, "PostProcess: /foo.min.css")
}},
{"PostProcess, disk", "--renderToDisk", func(c *qt.C, r serverTestResult) {
c.Assert(r.err, qt.IsNil)
c.Assert(r.homeContent, qt.Contains, "PostProcess: /foo.min.css")
}},
} {
c.Run(test.name, func(c *qt.C) {
config := `
baseURL="https://example.org"
`

var args []string
if test.flag != "" {
args = strings.Split(test.flag, "=")
}
r := runServerTest(c, true, config, args...)
test.assert(c, r)

})

}

}

type serverTestResult struct {
err error
homeContent string
Expand Down
26 changes: 16 additions & 10 deletions hugofs/fs.go
Expand Up @@ -65,53 +65,59 @@ type Fs struct {
// as source and destination file systems.
func NewDefault(cfg config.Provider) *Fs {
fs := Os
return newFs(fs, cfg)
return newFs(fs, fs, cfg)
}

// NewMem creates a new Fs with the MemMapFs
// as source and destination file systems.
// Useful for testing.
func NewMem(cfg config.Provider) *Fs {
fs := &afero.MemMapFs{}
return newFs(fs, cfg)
return newFs(fs, fs, cfg)
}

// NewFrom creates a new Fs based on the provided Afero Fs
// as source and destination file systems.
// Useful for testing.
func NewFrom(fs afero.Fs, cfg config.Provider) *Fs {
return newFs(fs, cfg)
return newFs(fs, fs, cfg)
}

func newFs(base afero.Fs, cfg config.Provider) *Fs {
// NewFrom creates a new Fs based on the provided Afero Fss
// as the source and destination file systems.
func NewFromSourceAndDestination(source, destination afero.Fs, cfg config.Provider) *Fs {
return newFs(source, destination, cfg)
}

func newFs(source, destination afero.Fs, cfg config.Provider) *Fs {
workingDir := cfg.GetString("workingDir")
publishDir := cfg.GetString("publishDir")
if publishDir == "" {
panic("publishDir is empty")
}

// Sanity check
if IsOsFs(base) && len(workingDir) < 2 {
if IsOsFs(source) && len(workingDir) < 2 {
panic("workingDir is too short")
}

absPublishDir := paths.AbsPathify(workingDir, publishDir)

// Make sure we always have the /public folder ready to use.
if err := base.MkdirAll(absPublishDir, 0777); err != nil && !os.IsExist(err) {
if err := source.MkdirAll(absPublishDir, 0777); err != nil && !os.IsExist(err) {
panic(err)
}

pubFs := afero.NewBasePathFs(base, absPublishDir)
pubFs := afero.NewBasePathFs(destination, absPublishDir)

return &Fs{
Source: base,
Source: source,
PublishDir: pubFs,
PublishDirServer: pubFs,
PublishDirStatic: pubFs,
Os: &afero.OsFs{},
WorkingDirReadOnly: getWorkingDirFsReadOnly(base, workingDir),
WorkingDirWritable: getWorkingDirFsWritable(base, workingDir),
WorkingDirReadOnly: getWorkingDirFsReadOnly(source, workingDir),
WorkingDirWritable: getWorkingDirFsWritable(source, workingDir),
}
}

Expand Down

0 comments on commit e66e2e9

Please sign in to comment.