go-zero/tools/goctl/docker/template.go

package docker

import (
	_ "embed"

	"github.com/zeromicro/go-zero/tools/goctl/util/pathx"
)

const (
	category           = "docker"
	dockerTemplateFile = "docker.tpl"
)

//go:embed docker.tpl
var dockerTemplate string

// Clean deletes all templates files
func Clean() error {
	return pathx.Clean(category)
}

// GenTemplates creates docker template files
func GenTemplates() error {
	return initTemplate()
}

// Category returns the const string of docker category
func Category() string {
	return category
}

// RevertTemplate recovers the deleted template files
func RevertTemplate(name string) error {
	return pathx.CreateTemplate(category, name, dockerTemplate)
}

// Update deletes and creates new template files
func Update() error {
	err := Clean()
	if err != nil {
		return err
	}

	return initTemplate()
}

func initTemplate() error {
	return pathx.InitTemplates(category, map[string]string{
		dockerTemplateFile: dockerTemplate,
	})
}