From 03ae187f3ba2873aec3abb2996e86df66d3adde6 Mon Sep 17 00:00:00 2001
From: lqqyt2423 <974923609@qq.com>
Date: Tue, 28 Sep 2021 13:56:34 +0800
Subject: [PATCH] show version

---
 cmd/go-mitmproxy/main.go | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/cmd/go-mitmproxy/main.go b/cmd/go-mitmproxy/main.go
index 53b2cb7..62fb654 100644
--- a/cmd/go-mitmproxy/main.go
+++ b/cmd/go-mitmproxy/main.go
@@ -2,6 +2,7 @@ package main
 
 import (
 	"flag"
+	"fmt"
 	"os"
 
 	"github.com/lqqyt2423/go-mitmproxy/addon"
@@ -13,6 +14,7 @@ import (
 const version = "0.0.5"
 
 type Config struct {
+	version   bool
 	addr      string
 	webAddr   string
 	dump      string // dump filename
@@ -22,6 +24,7 @@ type Config struct {
 func loadConfig() *Config {
 	config := new(Config)
 
+	flag.BoolVar(&config.version, "version", false, "show version")
 	flag.StringVar(&config.addr, "addr", ":9080", "proxy listen addr")
 	flag.StringVar(&config.webAddr, "web_addr", ":9081", "web interface listen addr")
 	flag.StringVar(&config.dump, "dump", "", "dump filename")
@@ -32,6 +35,13 @@ func loadConfig() *Config {
 }
 
 func main() {
+	config := loadConfig()
+
+	if config.version {
+		fmt.Println("go-mitmproxy: " + version)
+		os.Exit(0)
+	}
+
 	log.SetLevel(log.InfoLevel)
 	log.SetReportCaller(false)
 	log.SetOutput(os.Stdout)
@@ -41,8 +51,6 @@ func main() {
 
 	log.Infof("go-mitmproxy version %v\n", version)
 
-	config := loadConfig()
-
 	opts := &proxy.Options{
 		Addr:              config.addr,
 		StreamLargeBodies: 1024 * 1024 * 5,