mirror of
https://github.com/lancatlin/mdsh.git
synced 2026-11-06 03:30:13 +08:00
35 lines
715 B
Go
35 lines
715 B
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"os/exec"
|
|
"text/template"
|
|
)
|
|
|
|
func sh(ctx TemplateContext) func(string) string {
|
|
return func(command string) string {
|
|
cmd := exec.Command("sh", "-c", command)
|
|
for k, v := range ctx {
|
|
cmd.Env = append(cmd.Env, k+"="+v)
|
|
}
|
|
fmt.Println(cmd.Environ())
|
|
out, err := cmd.Output()
|
|
if err != nil {
|
|
return fmt.Sprintf("> Failed to execute the command: %s\n> %s\n", cmd.Args, err.Error())
|
|
}
|
|
return fmt.Sprintf("\n```\n%s\n```\n", string(out))
|
|
}
|
|
}
|
|
|
|
func render(tmpl string, ctx TemplateContext) {
|
|
funcMap := template.FuncMap{
|
|
"sh": sh(ctx),
|
|
}
|
|
t, err := template.New("foo").Funcs(funcMap).Parse(tmpl)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
t.Execute(os.Stdout, ctx)
|
|
}
|