mirror of
https://github.com/kubeshark/kubeshark.git
synced 2025-06-04 22:21:14 +00:00
85 lines
3.0 KiB
Go
85 lines
3.0 KiB
Go
package cmd
|
||
|
||
import (
|
||
"context"
|
||
"encoding/json"
|
||
"errors"
|
||
"fmt"
|
||
"github.com/up9inc/mizu/cli/apiserver"
|
||
"github.com/up9inc/mizu/cli/mizu"
|
||
"github.com/up9inc/mizu/cli/mizu/fsUtils"
|
||
"github.com/up9inc/mizu/cli/resources"
|
||
"github.com/up9inc/mizu/cli/telemetry"
|
||
"github.com/up9inc/mizu/shared"
|
||
"path"
|
||
"time"
|
||
|
||
"github.com/up9inc/mizu/cli/config"
|
||
"github.com/up9inc/mizu/cli/config/configStructs"
|
||
"github.com/up9inc/mizu/cli/errormessage"
|
||
"github.com/up9inc/mizu/cli/uiUtils"
|
||
"github.com/up9inc/mizu/shared/kubernetes"
|
||
"github.com/up9inc/mizu/shared/logger"
|
||
)
|
||
|
||
func GetApiServerUrl() string {
|
||
return fmt.Sprintf("http://%s", kubernetes.GetMizuApiServerProxiedHostAndPath(config.Config.Tap.GuiPort))
|
||
}
|
||
|
||
func startProxyReportErrorIfAny(kubernetesProvider *kubernetes.Provider, cancel context.CancelFunc) {
|
||
err := kubernetes.StartProxy(kubernetesProvider, config.Config.Tap.ProxyHost, config.Config.Tap.GuiPort, config.Config.MizuResourcesNamespace, kubernetes.ApiServerPodName)
|
||
if err != nil {
|
||
logger.Log.Errorf(uiUtils.Error, fmt.Sprintf("Error occured while running k8s proxy %v\n"+
|
||
"Try setting different port by using --%s", errormessage.FormatError(err), configStructs.GuiPortTapName))
|
||
cancel()
|
||
}
|
||
|
||
logger.Log.Debugf("proxy ended")
|
||
}
|
||
|
||
func getKubernetesProviderForCli() (*kubernetes.Provider, error) {
|
||
kubernetesProvider, err := kubernetes.NewProvider(config.Config.KubeConfigPath())
|
||
if err != nil {
|
||
handleKubernetesProviderError(err)
|
||
return nil, err
|
||
}
|
||
return kubernetesProvider, nil
|
||
}
|
||
|
||
func handleKubernetesProviderError(err error) {
|
||
var clusterBehindProxyErr *kubernetes.ClusterBehindProxyError
|
||
if ok := errors.As(err, &clusterBehindProxyErr); ok {
|
||
logger.Log.Errorf("cannot establish http-proxy connection to the Kubernetes cluster. If you’re using Lens or similar tool, please run mizu with regular kubectl config using --%v %v=$HOME/.kube/config flag", config.SetCommandName, config.KubeConfigPathConfigName)
|
||
} else {
|
||
logger.Log.Error(err)
|
||
}
|
||
}
|
||
|
||
func finishMizuExecution(kubernetesProvider *kubernetes.Provider, apiProvider *apiserver.Provider, isNsRestrictedMode bool, mizuResourcesNamespace string) {
|
||
telemetry.ReportAPICalls(apiProvider)
|
||
removalCtx, cancel := context.WithTimeout(context.Background(), cleanupTimeout)
|
||
defer cancel()
|
||
dumpLogsIfNeeded(removalCtx, kubernetesProvider)
|
||
resources.CleanUpMizuResources(removalCtx, cancel, kubernetesProvider, isNsRestrictedMode, mizuResourcesNamespace)
|
||
}
|
||
|
||
func dumpLogsIfNeeded(ctx context.Context, kubernetesProvider *kubernetes.Provider) {
|
||
if !config.Config.DumpLogs {
|
||
return
|
||
}
|
||
mizuDir := mizu.GetMizuFolderPath()
|
||
filePath := path.Join(mizuDir, fmt.Sprintf("mizu_logs_%s.zip", time.Now().Format("2006_01_02__15_04_05")))
|
||
if err := fsUtils.DumpLogs(ctx, kubernetesProvider, filePath); err != nil {
|
||
logger.Log.Errorf("Failed dump logs %v", err)
|
||
}
|
||
}
|
||
|
||
func getSerializedMizuAgentConfig(mizuAgentConfig *shared.MizuAgentConfig) (string, error) {
|
||
serializedConfig, err := json.Marshal(mizuAgentConfig)
|
||
if err != nil {
|
||
return "", err
|
||
}
|
||
|
||
return string(serializedConfig), nil
|
||
}
|