goldsmith/saver.go

package goldsmith

import (
	"os"
	"path/filepath"
)

type saver struct {
	clean  bool
	tokens map[string]bool
}

func (*saver) Name() string {
	return "saver"
}

func (saver *saver) Initialize(context *Context) error {
	saver.tokens = make(map[string]bool)
	context.Threads(1)
	return nil
}

func (saver *saver) Process(context *Context, file *File) error {
	for token := cleanPath(file.sourcePath); token != "."; token = filepath.Dir(token) {
		saver.tokens[token] = true
	}

	return file.export(context.goldsmith.targetDir)
}

func (saver *saver) Finalize(context *Context) error {
	if !saver.clean {
		return nil
	}

	infos := make(chan fileInfo)
	go scanDir(context.goldsmith.targetDir, infos)

	for info := range infos {
		if info.path != context.goldsmith.targetDir {
			relPath, _ := filepath.Rel(context.goldsmith.targetDir, info.path)
			if contained, _ := saver.tokens[relPath]; !contained {
				os.RemoveAll(info.path)
			}
		}
	}

	return nil
}