From d73674c106b7f993b39e4f551eae67120da1f766 Mon Sep 17 00:00:00 2001
From: huiyifyj <jxfengyijie@gmail.com>
Date: Wed, 18 Dec 2024 00:34:00 +0800
Subject: [PATCH] Add support for specifying version of documents

Add `--version` flag for `init` subcommand, to specify the version of the API in the generated files.
This can reset the version of API docs with external command lines, such as the following:

$ swag init --version $(git describe --tags `git rev-list --tags --max-count=1`)
---
 cmd/swag/main.go | 7 +++++++
 gen/gen.go       | 6 ++++++
 2 files changed, 13 insertions(+)

diff --git a/cmd/swag/main.go b/cmd/swag/main.go
index fd12c7f8a..a713eb55b 100644
--- a/cmd/swag/main.go
+++ b/cmd/swag/main.go
@@ -43,6 +43,7 @@ const (
 	packagePrefixFlag        = "packagePrefix"
 	stateFlag                = "state"
 	parseFuncBodyFlag        = "parseFuncBody"
+	versionFlag              = "version"
 )
 
 var initFlags = []cli.Flag{
@@ -186,6 +187,11 @@ var initFlags = []cli.Flag{
 		// Value: false,
 		Usage: "Parse API info within body of functions in go files, disabled by default (default: false)",
 	},
+	&cli.StringFlag{
+		Name:  versionFlag,
+		Value: "",
+		Usage: "Specify the version of the version of the API in the generated files",
+	},
 }
 
 func initAction(ctx *cli.Context) error {
@@ -268,6 +274,7 @@ func initAction(ctx *cli.Context) error {
 		PackagePrefix:       ctx.String(packagePrefixFlag),
 		State:               ctx.String(stateFlag),
 		ParseFuncBody:       ctx.Bool(parseFuncBodyFlag),
+		Version:             ctx.String(versionFlag),
 	})
 }
 
diff --git a/gen/gen.go b/gen/gen.go
index 417d51c36..45597126e 100644
--- a/gen/gen.go
+++ b/gen/gen.go
@@ -149,6 +149,9 @@ type Config struct {
 
 	// ParseFuncBody whether swag should parse api info inside of funcs
 	ParseFuncBody bool
+
+	// Version represents the version of the API in the generated swagger documents
+	Version string
 }
 
 // Build builds swagger json file  for given searchDir and mainAPIFile. Returns json.
@@ -223,6 +226,9 @@ func (g *Gen) Build(config *Config) error {
 	}
 
 	swagger := p.GetSwagger()
+	if config.Version != "" {
+		swagger.Info.Version = config.Version
+	}
 
 	if err := os.MkdirAll(config.OutputDir, os.ModePerm); err != nil {
 		return err