refactor: Clean up log & version flags

Signed-off-by: Justin Toh <tohjustin@hotmail.com>

cmd/kube-lineage/flags.go

@@ -0,0 +1,19 @@
+package main
+
+import (
+	"fmt"
+
+	"github.com/spf13/cobra"
+
+	"github.com/tohjustin/kube-lineage/internal/log"
+	"github.com/tohjustin/kube-lineage/internal/version"
+)
+
+func addLogFlags(cmd *cobra.Command) {
+	log.AddFlags(cmd.Flags())
+}
+
+func addVersionFlags(cmd *cobra.Command) {
+	cmd.SetVersionTemplate("{{printf \"%s\" .Version}}\n")
+	cmd.Version = fmt.Sprintf("%#v", version.Get())
+}

cmd/kube-lineage/main.go

@@ -6,15 +6,14 @@ import (
 	"github.com/spf13/cobra"
 	"github.com/spf13/pflag"
 	"k8s.io/cli-runtime/pkg/genericclioptions"
-	"k8s.io/klog/v2"
 
 	"github.com/tohjustin/kube-lineage/pkg/cmd/lineage"
 )
 
 func New(streams genericclioptions.IOStreams) *cobra.Command {
 	cmd := lineage.New(streams)
-	addVersionFlag(cmd)
-
+	addLogFlags(cmd)
+	addVersionFlags(cmd)
 	return cmd
 }
 
@@ -25,7 +24,6 @@ func main() {
 	streams := genericclioptions.IOStreams{In: os.Stdin, Out: os.Stdout, ErrOut: os.Stderr}
 	rootCmd := New(streams)
 
-	klog.V(4).Infof("Version: %s", getVersion())
 	if err := rootCmd.Execute(); err != nil {
 		os.Exit(1)
 	}

pkg/cmd/lineage/lineage.go

@@ -27,7 +27,6 @@ import (
 	"k8s.io/kubectl/pkg/util/templates"
 
 	"github.com/tohjustin/kube-lineage/internal/graph"
-	"github.com/tohjustin/kube-lineage/internal/log"
 	lineageprinters "github.com/tohjustin/kube-lineage/internal/printers"
 )
 
@@ -143,6 +142,7 @@ func New(streams genericclioptions.IOStreams) *cobra.Command {
 		DisableSuggestions:    true,
 		SilenceUsage:          true,
 		Run: func(c *cobra.Command, args []string) {
+			klog.V(4).Infof("Version: %s", c.Version)
 			cmdutil.CheckErr(o.Complete(c, args))
 			cmdutil.CheckErr(o.Validate())
 			cmdutil.CheckErr(o.Run())
@@ -151,7 +151,6 @@ func New(streams genericclioptions.IOStreams) *cobra.Command {
 
 	o.ConfigFlags.AddFlags(cmd.Flags())
 	o.PrintFlags.AddFlags(cmd.Flags())
-	log.AddFlags(cmd.Flags())
 
 	return cmd
 }