mirror of
https://github.com/gohugoio/hugo.git
synced 2024-11-14 20:37:55 -05:00
be1a566203
`404.html` needs to be that and not `/404/` in the root. There seem to be content pages in the wild with the name `404` (Hugo docs), so this commit adds an extra check so only root 404 pages ignore the `uglifyURLs`setting. Fixes #1140
72 lines
1.4 KiB
Go
72 lines
1.4 KiB
Go
package target
|
|
|
|
import (
|
|
"fmt"
|
|
"html/template"
|
|
"io"
|
|
"path/filepath"
|
|
|
|
"github.com/spf13/hugo/helpers"
|
|
"github.com/spf13/hugo/hugofs"
|
|
)
|
|
|
|
type PagePublisher interface {
|
|
Translator
|
|
Publish(string, template.HTML) error
|
|
}
|
|
|
|
type PagePub struct {
|
|
UglyURLs bool
|
|
DefaultExtension string
|
|
PublishDir string
|
|
}
|
|
|
|
func (pp *PagePub) Publish(path string, r io.Reader) (err error) {
|
|
|
|
translated, err := pp.Translate(path)
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
return helpers.WriteToDisk(translated, r, hugofs.DestinationFS)
|
|
}
|
|
|
|
func (pp *PagePub) Translate(src string) (dest string, err error) {
|
|
if src == helpers.FilePathSeparator {
|
|
if pp.PublishDir != "" {
|
|
return filepath.Join(pp.PublishDir, "index.html"), nil
|
|
}
|
|
return "index.html", nil
|
|
}
|
|
|
|
dir, file := filepath.Split(src)
|
|
isRoot := dir == ""
|
|
ext := pp.extension(filepath.Ext(file))
|
|
name := filename(file)
|
|
if pp.PublishDir != "" {
|
|
dir = filepath.Join(pp.PublishDir, dir)
|
|
}
|
|
|
|
if pp.UglyURLs || file == "index.html" || (isRoot && file == "404.html") {
|
|
return filepath.Join(dir, fmt.Sprintf("%s%s", name, ext)), nil
|
|
}
|
|
|
|
return filepath.Join(dir, name, fmt.Sprintf("index%s", ext)), nil
|
|
}
|
|
|
|
func (pp *PagePub) extension(ext string) string {
|
|
switch ext {
|
|
case ".md", ".rst": // TODO make this list configurable. page.go has the list of markup types.
|
|
return ".html"
|
|
}
|
|
|
|
if ext != "" {
|
|
return ext
|
|
}
|
|
|
|
if pp.DefaultExtension != "" {
|
|
return pp.DefaultExtension
|
|
}
|
|
|
|
return ".html"
|
|
}
|