From 215b736fdb6b67d4681b8aec0851d1f909be7c8b Mon Sep 17 00:00:00 2001
From: Mikhail Osipov <mike.osipov@gmail.com>
Date: Tue, 25 Feb 2020 23:26:06 +0300
Subject: [cmd/tunneld] -d option

---
 cmd/tunnel/main.go  |  4 ++++
 cmd/tunneld/main.go | 20 ++++++++++++++++++++
 2 files changed, 24 insertions(+)

(limited to 'cmd')

diff --git a/cmd/tunnel/main.go b/cmd/tunnel/main.go
index 49d9452..302b60e 100644
--- a/cmd/tunnel/main.go
+++ b/cmd/tunnel/main.go
@@ -8,6 +8,10 @@ import (
 	"tunnel/pkg/client"
 )
 
+func init() {
+	log.SetFlags(0)
+}
+
 func main() {
 	var args = os.Args
 
diff --git a/cmd/tunneld/main.go b/cmd/tunneld/main.go
index f713dea..ef274c8 100644
--- a/cmd/tunneld/main.go
+++ b/cmd/tunneld/main.go
@@ -1,6 +1,7 @@
 package main
 
 import (
+	"flag"
 	"log"
 	"os"
 	"os/signal"
@@ -8,7 +9,26 @@ import (
 	"tunnel/pkg/server"
 )
 
+var debugVar bool
+
+func init() {
+	flag.BoolVar(&debugVar, "d", false, "debug: print time and source info")
+}
+
+func initLog() {
+	var logFlags int
+
+	if debugVar {
+		logFlags |= log.Ldate | log.Ltime | log.Lshortfile
+	}
+
+	log.SetFlags(logFlags)
+}
+
 func main() {
+	flag.Parse()
+	initLog()
+
 	var c = make(chan os.Signal)
 
 	signal.Notify(c, syscall.SIGINT, syscall.SIGTERM)
-- 
cgit v1.2.3-70-g09d2