feat: add template-related template funcs

README.md

@@ -55,7 +55,14 @@ docker run --rm muxiu1997/goet -h
 
 - [Sprig Function Documentation](http://masterminds.github.io/sprig/)
 
-### OS/FS/PATH
+### Template
+
+- `templateFilePath` - templateFilePath returns the path of the template file. If the template is stdin, "/dev/stdin" is returned. If the template is a URL, the URL is returned.
+- `templateDirPath` - templateDirPath returns the directory of the template file. If the template is stdin, "/dev" is returned. If the template is a URL, "" is returned.
+- `outputFilePath` - outputFilePath returns the path of the output file. If the output is stdout, "/dev/stdout" is returned.
+- `outputDirPath` - outputDirPath returns the directory of the output file. If the output is stdout, "/dev" is returned.
+
+### OS / FS / PATH
 
 - `lookPath` - lookPath runs exec.LookPath on name and returns the path or an empty string if the name is not found.
 - `lstat` - lstat runs os.Lstat on name and returns structured data that contains the
@@ -86,4 +93,4 @@ docker run --rm muxiu1997/goet -h
 
 ## License
 
-[MIT](https://github.com/MuXiu1997/goet/blob/main/LICENSE)
+[MIT](https://github.com/MuXiu1997/goet/blob/main/LICENSE)

pkg/templatefunc/template/template.go

@@ -0,0 +1,53 @@
+package template
+
+import "path/filepath"
+import ttemplate "text/template"
+
+import tc "github.com/MuXiu1997/goet/pkg/templatecontext"
+
+func FuncMap(templateContext *tc.TemplateContext) ttemplate.FuncMap {
+	return ttemplate.FuncMap{
+		"templateFilePath": templateFilePath(templateContext),
+		"templateDirPath":  templateDirPath(templateContext),
+		"outputFilePath":   outputFilePath(templateContext),
+		"outputDirPath":    outputDirPath(templateContext),
+	}
+}
+
+// templateFilePath returns the path of the template file.
+// If the template is stdin, "/dev/stdin" is returned.
+// If the template is a URL, the URL is returned.
+func templateFilePath(templateContext *tc.TemplateContext) func() string {
+	return func() string {
+		return templateContext.Template
+	}
+}
+
+// templateDirPath returns the directory of the template file.
+// If the template is stdin, "/dev" is returned.
+// If the template is a URL, "" is returned.
+func templateDirPath(templateContext *tc.TemplateContext) func() string {
+	return func() string {
+		if templateContext.TemplateType == tc.TemplateTypeFile {
+			return filepath.Dir(templateContext.Template)
+		} else {
+			return ""
+		}
+	}
+}
+
+// outputFilePath returns the path of the output file.
+// If the output is stdout, "/dev/stdout" is returned.
+func outputFilePath(templateContext *tc.TemplateContext) func() string {
+	return func() string {
+		return templateContext.Output
+	}
+}
+
+// outputDirPath returns the directory of the output file.
+// If the output is stdout, "/dev" is returned.
+func outputDirPath(templateContext *tc.TemplateContext) func() string {
+	return func() string {
+		return filepath.Dir(templateContext.Output)
+	}
+}

pkg/templaterenderer/renderer.go

@@ -4,6 +4,7 @@ import (
 	"bytes"
 	"github.com/Masterminds/sprig/v3"
 	"github.com/MuXiu1997/goet/pkg/templatefunc/osfspath"
+	"github.com/MuXiu1997/goet/pkg/templatefunc/template"
 	"github.com/MuXiu1997/goet/pkg/templatefunc/toml"
 	"github.com/MuXiu1997/goet/pkg/templatefunc/yaml"
 	ttemplate "text/template"
@@ -17,6 +18,7 @@ func Render(templateContext *tc.TemplateContext) (string, error) {
 		Funcs(osfspath.FuncMap()).
 		Funcs(yaml.FuncMap()).
 		Funcs(toml.FuncMap()).
+		Funcs(template.FuncMap(templateContext)).
 		Parse(templateContext.TemplateContent)
 	if err != nil {
 		return "", err