feat: gen plugin file with hrp version

This commit is contained in:
debugtalk
2022-05-28 09:40:49 +08:00
parent cbdc7464c6
commit 52ce77efa8
3 changed files with 16 additions and 7 deletions

View File

@@ -14,9 +14,11 @@ import (
"text/template"
"github.com/httprunner/funplugin/shared"
"github.com/httprunner/httprunner/v4/hrp/internal/builtin"
"github.com/pkg/errors"
"github.com/rs/zerolog/log"
"github.com/httprunner/httprunner/v4/hrp/internal/builtin"
"github.com/httprunner/httprunner/v4/hrp/internal/version"
)
const (
@@ -36,6 +38,7 @@ var pyTemplate string
var goTemplate string
type TemplateContent struct {
Version string // hrp version
Fun string // funplugin package
Regexps *Regexps // match import/function
Imports []string // python/go import
@@ -178,7 +181,8 @@ func (t *TemplateContent) genDebugTalk(path string, templ string) error {
// buildGo builds debugtalk.go to debugtalk.bin
func buildGo(path string, output string) error {
templateContent := &TemplateContent{
Fun: fungo,
Version: version.VERSION,
Fun: fungo,
Regexps: &Regexps{
Import: regexp.MustCompile(regexGoImport),
Imports: regexp.MustCompile(regexGoImports),
@@ -248,7 +252,8 @@ func buildGo(path string, output string) error {
// buildPy completes funppy information in debugtalk.py
func buildPy(path string, output string) error {
templateContent := &TemplateContent{
Fun: funppy,
Version: version.VERSION,
Fun: funppy,
Regexps: &Regexps{
FunctionName: regexp.MustCompile(regexPythonFunctionName),
},

View File

@@ -9,8 +9,9 @@ import (
{{ range $function := .Functions }}
{{ $function }}
{{ end }}
func main() {
{{- range $idx, $functionName := .FunctionNames }}
{{- range $functionName := .FunctionNames }}
fungo.Register("{{ $functionName }}", {{ $functionName }})
{{- end }}
fungo.Serve()

View File

@@ -1,13 +1,16 @@
{{- range $import := .Imports }}
# NOTE: Generated By hrp {{ .Version }}, DO NOT EDIT!
{{ range $import := .Imports }}
{{- $import}}
{{ end }}
{{ range $fromImport := .FromImports }}
{{- $fromImport}}
{{ end }}
{{ range $function := .Functions }}
{{ $function }}
{{ range $function := .Functions }}
{{- $function }}
{{ end }}
if __name__ == "__main__":
{{- range $functionName := .FunctionNames }}
funppy.register("{{ $functionName }}", {{ $functionName }})