httprunner/hrp/cmd/har2case.go

package cmd

import (
	"github.com/spf13/cobra"

	"github.com/httprunner/hrp"
	"github.com/httprunner/hrp/har2case"
)

// har2caseCmd represents the har2case command
var har2caseCmd = &cobra.Command{
	Use:   "har2case harPath...",
	Short: "Convert HAR to json/yaml testcase files",
	Long:  `Convert HAR to json/yaml testcase files`,
	Args:  cobra.MinimumNArgs(1),
	RunE: func(cmd *cobra.Command, args []string) error {
		var outputFiles []string
		for _, arg := range args {
			var outputPath string
			var err error

			har := har2case.NewHAR(arg)

			// specify output dir
			if outputDir != "" {
				har.SetOutputDir(outputDir)
			}

			// generate json/yaml files
			if genYAMLFlag {
				outputPath, err = har.GenYAML()
			} else {
				outputPath, err = har.GenJSON()
			}
			if err != nil {
				return err
			}
			outputFiles = append(outputFiles, outputPath)
		}
		log := hrp.GetLogger()
		log.Info().Strs("output", outputFiles).Msg("convert testcase success")
		return nil
	},
}

var (
	genJSONFlag bool
	genYAMLFlag bool
	outputDir   string
)

func init() {
	RootCmd.AddCommand(har2caseCmd)
	har2caseCmd.Flags().BoolVarP(&genJSONFlag, "to-json", "j", false, "convert to JSON format (default)")
	har2caseCmd.Flags().BoolVarP(&genYAMLFlag, "to-yaml", "y", false, "convert to JSON format")
	har2caseCmd.Flags().StringVarP(&outputDir, "output-dir", "d", "", "specify output directory, default to the same dir with har file")
}