You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
41 lines
824 B
Go
41 lines
824 B
Go
package gen
|
|
|
|
import (
|
|
"path/filepath"
|
|
"strings"
|
|
"text/template"
|
|
|
|
"github.com/tal-tech/go-zero/tools/goctl/util"
|
|
)
|
|
|
|
func GenerateDockerfile(goFile string, args ...string) error {
|
|
projPath, err := getFilePath(filepath.Dir(goFile))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
pos := strings.IndexByte(projPath, '/')
|
|
if pos >= 0 {
|
|
projPath = projPath[pos+1:]
|
|
}
|
|
|
|
out, err := util.CreateIfNotExist("Dockerfile")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer out.Close()
|
|
|
|
var builder strings.Builder
|
|
for _, arg := range args {
|
|
builder.WriteString(`, "` + arg + `"`)
|
|
}
|
|
|
|
t := template.Must(template.New("dockerfile").Parse(dockerTemplate))
|
|
return t.Execute(out, map[string]string{
|
|
"goRelPath": projPath,
|
|
"goFile": goFile,
|
|
"exeFile": util.FileNameWithoutExt(filepath.Base(goFile)),
|
|
"argument": builder.String(),
|
|
})
|
|
}
|