Support version command

buildAllPlatforms.sh

@@ -4,6 +4,10 @@ OUTPUT="bin/"
 PROGRAM="mnh"
 LDFLAGS="-s -w"
 
+ver="$(git describe --tags --match "v*" --dirty="" 2>/dev/null || git log -1 --pretty=format:"v0.0.0-%h" 2>/dev/null || echo "v0.0.0")"
+[ -n "$(git status --porcelain |& grep -Ev '^\?\?')" ] && ver="$ver-$(date +"%Y%M%d-%H%m%S")"
+LDFLAGS="$LDFLAGS -X main.version=$ver"
+
 mkdir -p "$OUTPUT"
 rm -f "$OUTPUT/${PROGRAM}-"*
 

main.go

@@ -20,13 +20,22 @@ import (
 	"github.com/hzyitc/mnh/routerForward"
 )
 
+var version = "v0.0.0"
+
 var rootCmd = &cobra.Command{
 	Use:   "mnh",
 	Short: "A NAT hole punching tool that allows peers directly connect to your NATed server without client.",
 	Long: "mnh is a tool that makes exposing a port behind NAT possible.\n" +
 		"mnh client will produce an ip:port pair for your NATed server which can be used for public access.",
 }
 
+var versionCmd = &cobra.Command{
+	Use: "version",
+	Run: func(cmd *cobra.Command, args []string) {
+		print(version)
+	},
+}
+
 var tcpCmd = &cobra.Command{
 	Use: "tcp",
 	Run: func(cmd *cobra.Command, args []string) {
@@ -118,6 +127,7 @@ func runHook(event string, errmsg string, port string, addr string) {
 }
 
 func main() {
+	rootCmd.AddCommand(versionCmd)
 	tcpCmdRegister(rootCmd)
 	udpCmdRegister(rootCmd)