mirror of
https://github.com/kubeshark/kubeshark.git
synced 2026-05-09 10:47:44 +00:00
0eb7883a47
* Rename `mizu` to `kubeshark` * Rename `up9inc` to `kubeshark` * Change the logo, title, motto and the main color * Replace the favicon * Update the docs link * Change the copyright text in C files * Remove a comment * Rewrite the `README.md` and update the logo and screenshots used in it * Add a `TODO` * Fix the grammar * Fix the bottom text in the filtering guide * Change the Docker Hub username of cross-compilation intermediate images * Add an install script * Fix `docker/login-action` in the CI * Delete `build-custom-branch.yml` GitHub workflow * Update `README.md` * Remove `install.sh` * Change the motto back to "Traffic viewer for Kubernetes"
65 lines
1.9 KiB
Go
65 lines
1.9 KiB
Go
package cmd
|
|
|
|
import (
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/creasty/defaults"
|
|
"github.com/kubeshark/kubeshark/cli/config"
|
|
"github.com/kubeshark/kubeshark/cli/kubeshark"
|
|
"github.com/kubeshark/kubeshark/cli/kubeshark/fsUtils"
|
|
"github.com/kubeshark/kubeshark/cli/kubeshark/version"
|
|
"github.com/kubeshark/kubeshark/cli/uiUtils"
|
|
"github.com/kubeshark/kubeshark/logger"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
var rootCmd = &cobra.Command{
|
|
Use: "kubeshark",
|
|
Short: "A web traffic viewer for kubernetes",
|
|
Long: `A web traffic viewer for kubernetes
|
|
Further info is available at https://github.com/kubeshark/kubeshark`,
|
|
PersistentPreRunE: func(cmd *cobra.Command, args []string) error {
|
|
if err := config.InitConfig(cmd); err != nil {
|
|
logger.Log.Fatal(err)
|
|
}
|
|
|
|
return nil
|
|
},
|
|
}
|
|
|
|
func init() {
|
|
defaultConfig := config.ConfigStruct{}
|
|
if err := defaults.Set(&defaultConfig); err != nil {
|
|
logger.Log.Debug(err)
|
|
}
|
|
|
|
rootCmd.PersistentFlags().StringSlice(config.SetCommandName, []string{}, fmt.Sprintf("Override values using --%s", config.SetCommandName))
|
|
rootCmd.PersistentFlags().String(config.ConfigFilePathCommandName, defaultConfig.ConfigFilePath, fmt.Sprintf("Override config file path using --%s", config.ConfigFilePathCommandName))
|
|
}
|
|
|
|
func printNewVersionIfNeeded(versionChan chan string) {
|
|
select {
|
|
case versionMsg := <-versionChan:
|
|
if versionMsg != "" {
|
|
logger.Log.Infof(uiUtils.Yellow, versionMsg)
|
|
}
|
|
case <-time.After(2 * time.Second):
|
|
}
|
|
}
|
|
|
|
// Execute adds all child commands to the root command and sets flags appropriately.
|
|
// This is called by main.main(). It only needs to happen once to the tapCmd.
|
|
func Execute() {
|
|
if err := fsUtils.EnsureDir(kubeshark.GetKubesharkFolderPath()); err != nil {
|
|
logger.Log.Errorf("Failed to use kubeshark folder, %v", err)
|
|
}
|
|
logger.InitLogger(fsUtils.GetLogFilePath())
|
|
|
|
versionChan := make(chan string)
|
|
defer printNewVersionIfNeeded(versionChan)
|
|
go version.CheckNewerVersion(versionChan)
|
|
|
|
cobra.CheckErr(rootCmd.Execute())
|
|
}
|