feat: add zerolog logger

cmd/root.go

@@ -6,6 +6,7 @@ package cmd
 import (
 	"os"
 
+	"github.com/0xtux/trok/internal/config"
 	"github.com/spf13/cobra"
 )
 
@@ -15,6 +16,10 @@ var rootCmd = &cobra.Command{
 	Long:   "Simple TCP tunnel in Go that exposes local ports to internet, bypassing NAT firewalls.",
 }
 
+func init() {
+	config.InitLogger()
+}
+
 func Execute() {
 	rootCmd.CompletionOptions.HiddenDefaultCmd = true
 	err := rootCmd.Execute()

internal/client/init.go

@@ -5,9 +5,9 @@ Copyright © 2024 tux <0xtux@pm.me>
 package client
 
 import (
-	"log"
+	"github.com/rs/zerolog/log"
 )
 
 func Start(port uint16) {
-	log.Printf("Hello from client: %d", port)
+	log.Info().Msgf("Hello from client: %d", port)
 }

internal/config/logger.go

@@ -0,0 +1,30 @@
+/*
+Copyright © 2024 tux <0xtux@pm.me>
+*/
+
+package config
+
+import (
+	"os"
+	"time"
+
+	"github.com/rs/zerolog"
+	"github.com/rs/zerolog/log"
+)
+
+func InitLogger() {
+	output := zerolog.ConsoleWriter{
+		Out:        os.Stdout,
+		TimeFormat: time.RFC3339,
+		NoColor:    false,
+	}
+
+	log.Logger = zerolog.New(output).
+		With().
+		Timestamp().
+		Caller().
+		Stack().
+		Logger()
+
+	zerolog.SetGlobalLevel(zerolog.DebugLevel)
+}

internal/server/init.go

@@ -5,9 +5,9 @@ Copyright © 2024 tux <0xtux@pm.me>
 package server
 
 import (
-	"log"
+	"github.com/rs/zerolog/log"
 )
 
 func Start(port uint16) {
-	log.Printf("Hello from server: %d", port)
+	log.Info().Msgf("Hello from server: %d", port)
 }