refactor: move hrp/ to root folder

2026-06-02 14:29:44 +08:00 · 2025-02-06 10:52:08 +08:00
parent 9376692b71
commit 1f063dd6f7
221 changed files with 206 additions and 211 deletions
--- a/cmd/convert.go
+++ b/cmd/convert.go
@@ -0,0 +1,122 @@
+package cmd
+
+import (
+	"os"
+	"path/filepath"
+
+	"github.com/httprunner/funplugin/myexec"
+	"github.com/pkg/errors"
+	"github.com/rs/zerolog/log"
+	"github.com/spf13/cobra"
+
+	"github.com/httprunner/httprunner/v5/code"
+	"github.com/httprunner/httprunner/v5/internal/builtin"
+	"github.com/httprunner/httprunner/v5/pkg/convert"
+)
+
+var convertCmd = &cobra.Command{
+	Use:          "convert $path...",
+	Short:        "convert multiple source format to HttpRunner JSON/YAML/gotest/pytest cases",
+	Args:         cobra.MinimumNArgs(1),
+	SilenceUsage: false,
+	RunE: func(cmd *cobra.Command, args []string) error {
+		caseConverter := convert.NewConverter(outputDir, profilePath)
+
+		var fromType convert.FromType
+		if fromYAMLFlag {
+			fromType = convert.FromTypeYAML
+		} else if fromPostmanFlag {
+			fromType = convert.FromTypePostman
+		} else if fromHARFlag {
+			fromType = convert.FromTypeHAR
+		} else if fromCurlFlag {
+			fromType = convert.FromTypeCurl
+		} else {
+			fromType = convert.FromTypeJSON
+			log.Info().Str("fromType", fromType.String()).Msg("set default")
+		}
+
+		var outputType convert.OutputType
+		if toYAMLFlag {
+			outputType = convert.OutputTypeYAML
+		} else if toPyTestFlag {
+			packages := []string{"httprunner"}
+			_, err := myexec.EnsurePython3Venv(venv, packages...)
+			if err != nil {
+				log.Error().Err(err).Msg("python3 venv is not ready")
+				return errors.Wrap(code.InvalidPython3Venv, err.Error())
+			}
+
+			outputType = convert.OutputTypePyTest
+		} else {
+			outputType = convert.OutputTypeJSON
+			log.Info().Str("outputType", outputType.String()).Msg("set default")
+		}
+
+		var files []string
+		for _, arg := range args {
+			if builtin.IsFolderPathExists(arg) {
+				fs, err := os.ReadDir(arg)
+				if err != nil {
+					log.Error().Err(err).Str("path", arg).Msg("read dir failed")
+					continue
+				}
+				for _, f := range fs {
+					files = append(files, filepath.Join(arg, f.Name()))
+				}
+			} else {
+				files = append(files, arg)
+			}
+		}
+
+		for _, file := range files {
+			extName := filepath.Ext(file)
+			if !builtin.Contains(fromType.Extensions(), extName) {
+				log.Warn().Str("path", file).
+					Strs("expectExtensions", fromType.Extensions()).
+					Msg("skip file")
+				continue
+			}
+
+			if err := caseConverter.Convert(file, fromType, outputType); err != nil {
+				log.Error().Err(err).Str("path", file).
+					Str("outputType", outputType.String()).
+					Msg("convert case failed")
+			}
+		}
+
+		return nil
+	},
+}
+
+var (
+	outputDir   string
+	profilePath string
+
+	fromJSONFlag    bool
+	fromYAMLFlag    bool
+	fromPostmanFlag bool
+	fromHARFlag     bool
+	fromCurlFlag    bool
+
+	toJSONFlag   bool
+	toYAMLFlag   bool
+	toPyTestFlag bool
+)
+
+func init() {
+	rootCmd.AddCommand(convertCmd)
+
+	convertCmd.Flags().BoolVar(&fromJSONFlag, "from-json", true, "load from json case format")
+	convertCmd.Flags().BoolVar(&fromYAMLFlag, "from-yaml", false, "load from yaml case format")
+	convertCmd.Flags().BoolVar(&fromHARFlag, "from-har", false, "load from HAR format")
+	convertCmd.Flags().BoolVar(&fromPostmanFlag, "from-postman", false, "load from postman format")
+	convertCmd.Flags().BoolVar(&fromCurlFlag, "from-curl", false, "load from curl format")
+
+	convertCmd.Flags().BoolVar(&toJSONFlag, "to-json", true, "convert to JSON case scripts")
+	convertCmd.Flags().BoolVar(&toYAMLFlag, "to-yaml", false, "convert to YAML case scripts")
+	convertCmd.Flags().BoolVar(&toPyTestFlag, "to-pytest", false, "convert to pytest scripts")
+
+	convertCmd.Flags().StringVarP(&outputDir, "output-dir", "d", "", "specify output directory")
+	convertCmd.Flags().StringVarP(&profilePath, "profile", "p", "", "specify profile path to override headers and cookies")
+}