Add version flag.

1 files changed, 10 insertions, 2 deletions

diff --git a/main.go b/main.go
index 74e208bc..0c6d4ee9 100644
--- a/main.go
+++ b/main.go
@@ -15,6 +15,7 @@ var VERSION = "dev"
 var REVISION = "HEAD"
 
 func appMain() {
+	var showVersion = flag.Bool("version", false, "Show version")
 	var listenHTTP = flag.String("listen-http", "", "The address to listen for HTTP requests")
 	var listenHTTPS = flag.String("listen-https", "", "The address to listen for HTTPS requests")
 	var listenProxy = flag.String("listen-proxy", "", "The address to listen for proxy requests")
@@ -27,10 +28,17 @@ func appMain() {
 	var daemonUID = flag.Uint("daemon-uid", 0, "Drop privileges to this user")
 	var daemonGID = flag.Uint("daemon-gid", 0, "Drop privileges to this group")
 
-	log.Printf("GitLab Pages Daemon %s (%s)", VERSION, REVISION)
-	log.Printf("URL: https://gitlab.com/gitlab-org/gitlab-pages\n")
 	flag.Parse()
 
+	version := VERSION + "-" + REVISION
+	if *showVersion {
+		log.SetFlags(0)
+		log.Printf(version)
+		os.Exit(0)
+	}
+
+	log.Printf("GitLab Pages Daemon %s (%s)", VERSION, REVISION)
+	log.Printf("URL: https://gitlab.com/gitlab-org/gitlab-pages\n")
 	err := os.Chdir(*pagesRoot)
 	if err != nil {
 		log.Fatalln(err)