hugo/target/file.go

package target

import (
	"fmt"
	"io"
	"os"
	"path"
	"path/filepath"
)

type Publisher interface {
	Publish(string, io.Reader) error
}

type Translator interface {
	Translate(string) (string, error)
}

type Filesystem struct {
	UglyUrls         bool
	DefaultExtension string
	PublishDir       string
}

func (fs *Filesystem) Publish(path string, r io.Reader) (err error) {

	translated, err := fs.Translate(path)
	if err != nil {
		return
	}

	path, _ = filepath.Split(translated)
	dest := filepath.Join(fs.PublishDir, path)
	ospath := filepath.FromSlash(dest)

	err = os.MkdirAll(ospath, 0764) // rwx, rw, r
	if err != nil {
		return
	}

	file, err := os.Create(filepath.Join(fs.PublishDir, translated))
	if err != nil {
		return
	}
	defer file.Close()

	_, err = io.Copy(file, r)
	return
}

func (fs *Filesystem) Translate(src string) (dest string, err error) {
	if src == "/" {
		return "index.html", nil
	}
	if fs.UglyUrls {
		return src, nil
	}

	dir, file := path.Split(src)
	ext := fs.extension(path.Ext(file))
	name := filename(file)

	return path.Join(dir, name, fmt.Sprintf("index%s", ext)), nil
}

func (fs *Filesystem) extension(ext string) string {
	if ext != "" {
		return ext
	}

	if fs.DefaultExtension != "" {
		return fs.DefaultExtension
	}

	return ".html"
}

func filename(f string) string {
	ext := path.Ext(f)
	if ext == "" {
		return f
	}

	return f[:len(f)-len(ext)]
}