add the mod support

main.go

@@ -43,13 +43,19 @@ func main() {
 			Value: "json",
 			Usage: "the type of swagger file (json or yaml)",
 		},
+		cli.BoolFlag{
+			Name:  "mod, m",
+			Usage: "whether use the mod ",
+		},
 	}
 	app.Action = func(c *cli.Context) error {
 		if err := parser.Parse(c.String("project"),
 			c.String("swagger"),
 			c.String("output"),
 			c.String("type"),
-			c.Bool("dev")); err != nil {
+			c.Bool("dev"),
+			c.Bool("mod"),
+		); err != nil {
 			return err
 		}
 		return nil

parser/app.go

@@ -20,6 +20,7 @@ var (
 	goPaths = []string{}
 	goRoot  = ""
 	devMode bool
+	useMod  bool
 )
 
 func init() {
@@ -34,13 +35,14 @@ func init() {
 }
 
 // Parse the project by args
-func Parse(projectPath, swaggerGo, output, t string, dev bool) (err error) {
+func Parse(projectPath, swaggerGo, output, t string, dev bool, mod bool) (err error) {
 	absPPath, err := filepath.Abs(projectPath)
 	if err != nil {
 		return err
 	}
 	vendor = filepath.Join(absPPath, "vendor")
 	devMode = dev
+	useMod = mod
 
 	sw := swagger.NewV2()
 	if err = doc2Swagger(projectPath, swaggerGo, dev, sw); err != nil {
@@ -54,7 +56,7 @@ func Parse(projectPath, swaggerGo, output, t string, dev bool) (err error) {
 	switch t {
 	case "json":
 		filename = jsonFile
-		data, err = json.Marshal(sw)
+		data, err = json.MarshalIndent(sw, "", "    ")
 	case "yaml":
 		filename = yamlFile
 		data, err = yaml.Marshal(sw)

parser/utils.go

@@ -2,6 +2,7 @@ package parser
 
 import (
 	"go/ast"
+	"golang.org/x/tools/go/packages"
 	"os"
 	"path/filepath"
 	"strconv"
@@ -36,6 +37,20 @@ func str2RealType(s string, typ string) (ret interface{}, err error) {
 }
 
 func absPathFromGoPath(importPath string) (string, bool) {
+	if useMod {
+		//modPath, err := module.EscapePath(importPath)
+		x1, err := packages.Load(&packages.Config{
+			Mode: packages.LoadFiles,
+		}, importPath)
+		if err != nil {
+
+		}
+		realPath := x1[0].GoFiles[0]
+		x2 := strings.Split(realPath, "/")
+		x2 = x2[0 : len(x2)-1]
+		return strings.Join(x2, "/"), true
+		//return importPath, true
+	}
 	if vendor != "" {
 		vendorImport := filepath.Join(vendor, importPath)
 		if fileExists(vendorImport) {