Adding --show-tree flag to both "./kubestr browse pvc" & "./kubestr browse snapshot" commands

cmd/rootCmd.go

@@ -95,6 +95,8 @@ var (
 		},
 	}
 
+	showTree bool
+
 	browsePvcCmd = &cobra.Command{
 		Use:   "pvc [PVC name]",
 		Short: "Browse the contents of a CSI PVC via file browser",
@@ -106,6 +108,7 @@ var (
 				csiCheckVolumeSnapshotClass,
 				csiCheckRunAsUser,
 				browseLocalPort,
+				showTree,
 			)
 		},
 	}
@@ -120,6 +123,7 @@ var (
 				namespace,
 				csiCheckRunAsUser,
 				browseLocalPort,
+				showTree,
 			)
 		},
 	}
@@ -195,6 +199,7 @@ func init() {
 	browseCmd.PersistentFlags().StringVarP(&namespace, "namespace", "n", fio.DefaultNS, "The namespace of the resource to browse.")
 	browseCmd.PersistentFlags().Int64VarP(&csiCheckRunAsUser, "runAsUser", "u", 0, "Runs the inspector pod as a user (int)")
 	browseCmd.PersistentFlags().IntVarP(&browseLocalPort, "localport", "l", 8080, "The local port to expose the inspector")
+	browseCmd.PersistentFlags().BoolVarP(&showTree, "show-tree", "t", false, "Prints the contents of given PVC or VolumeSnapshot")
 
 	browseCmd.AddCommand(browsePvcCmd)
 	browsePvcCmd.Flags().StringVarP(&csiCheckVolumeSnapshotClass, "volumesnapshotclass", "v", "", "The name of a VolumeSnapshotClass. (Required)")
@@ -361,6 +366,7 @@ func CsiPvcBrowse(ctx context.Context,
 	volumeSnapshotClass string,
 	runAsUser int64,
 	localPort int,
+	showTree bool,
 ) error {
 	kubecli, err := kubestr.LoadKubeCli()
 	if err != nil {
@@ -382,6 +388,7 @@ func CsiPvcBrowse(ctx context.Context,
 		VolumeSnapshotClass: volumeSnapshotClass,
 		RunAsUser:           runAsUser,
 		LocalPort:           localPort,
+		ShowTree:            showTree,
 	})
 	if err != nil {
 		fmt.Printf("Failed to run PVC browser (%s)\n", err.Error())
@@ -394,6 +401,7 @@ func CsiSnapshotBrowse(ctx context.Context,
 	namespace string,
 	runAsUser int64,
 	localPort int,
+	showTree bool,
 ) error {
 	kubecli, err := kubestr.LoadKubeCli()
 	if err != nil {
@@ -414,6 +422,7 @@ func CsiSnapshotBrowse(ctx context.Context,
 		Namespace:    namespace,
 		RunAsUser:    runAsUser,
 		LocalPort:    localPort,
+		ShowTree:     showTree,
 	})
 	if err != nil {
 		fmt.Printf("Failed to run Snapshot browser (%s)\n", err.Error())

pkg/csi/csi_ops.go

@@ -568,3 +568,20 @@ func (p *portforward) PortForwardAPod(req *types.PortForwardAPodRequest) error {
 func (p *portforward) FetchRestConfig() (*rest.Config, error) {
 	return kube.LoadConfig()
 }
+
+//go:generate go run github.com/golang/mock/mockgen -destination=mocks/mock_kube_executor.go -package=mocks . KubeExecutor
+type KubeExecutor interface {
+	Exec(ctx context.Context, namespace string, podName string, ContainerName string, command []string) (string, error)
+}
+
+type kubeExec struct {
+	kubeCli kubernetes.Interface
+}
+
+func (k *kubeExec) Exec(ctx context.Context, namespace string, podName string, ContainerName string, command []string) (string, error) {
+	if k.kubeCli == nil {
+		return "", fmt.Errorf("kubeCli not initialized")
+	}
+	stdout, _, err := kankube.Exec(ctx, k.kubeCli, namespace, podName, ContainerName, command, nil)
+	return stdout, err
+}

pkg/csi/pvc_inspector.go

@@ -49,11 +49,18 @@ func (r *PVCBrowseRunner) RunPVCBrowse(ctx context.Context, args *types.PVCBrows
 			dynCli:  r.DynCli,
 		},
 		portForwardOps: &portforward{},
+		kubeExecutor: &kubeExec{
+			kubeCli: r.KubeCli,
+		},
 		cleanerOps: &cleanse{
 			kubeCli: r.KubeCli,
 			dynCli:  r.DynCli,
 		},
 	}
+	if args.ShowTree {
+		fmt.Println("Show Tree works for PVC!")
+		return nil
+	}
 	return r.RunPVCBrowseHelper(ctx, args)
 }
 
@@ -70,19 +77,29 @@ func (r *PVCBrowseRunner) RunPVCBrowseHelper(ctx context.Context, args *types.PV
 		return errors.Wrap(err, "Failed to validate arguments.")
 	}
 
-	fmt.Println("Taking a snapshot")
+	fmt.Println("Taking a snapshot.")
 	snapName := snapshotPrefix + time.Now().Format("20060102150405")
 	r.snapshot, err = r.browserSteps.SnapshotPVC(ctx, args, snapName)
 	if err != nil {
 		return errors.Wrap(err, "Failed to snapshot PVC.")
 	}
 
-	fmt.Println("Creating the file browser application.")
+	fmt.Println("Creating the browser pod.")
 	r.pod, r.pvc, err = r.browserSteps.CreateInspectorApplication(ctx, args, r.snapshot, sc)
 	if err != nil {
 		return errors.Wrap(err, "Failed to create inspector application.")
 	}
 
+	if args.ShowTree {
+		fmt.Println("Printing the tree structure from root directory.")
+		stdout, err := r.browserSteps.ExecuteTreeCommand(ctx, args, r.pod)
+		if err != nil {
+			return errors.Wrap(err, "Failed to execute tree command in pod.")
+		}
+		fmt.Printf("\n%s\n\n", stdout)
+		return nil
+	}
+
 	fmt.Println("Forwarding the port.")
 	err = r.browserSteps.PortForwardAPod(ctx, r.pod, args.LocalPort)
 	if err != nil {
@@ -97,6 +114,7 @@ type PVCBrowserStepper interface {
 	ValidateArgs(ctx context.Context, args *types.PVCBrowseArgs) (*sv1.StorageClass, error)
 	SnapshotPVC(ctx context.Context, args *types.PVCBrowseArgs, snapshotName string) (*snapv1.VolumeSnapshot, error)
 	CreateInspectorApplication(ctx context.Context, args *types.PVCBrowseArgs, snapshot *snapv1.VolumeSnapshot, storageClass *sv1.StorageClass) (*v1.Pod, *v1.PersistentVolumeClaim, error)
+	ExecuteTreeCommand(ctx context.Context, args *types.PVCBrowseArgs, pod *v1.Pod) (string, error)
 	PortForwardAPod(ctx context.Context, pod *v1.Pod, localPort int) error
 	Cleanup(ctx context.Context, pvc *v1.PersistentVolumeClaim, pod *v1.Pod, snapshot *snapv1.VolumeSnapshot)
 }
@@ -108,6 +126,7 @@ type pvcBrowserSteps struct {
 	snapshotCreateOps    SnapshotCreator
 	portForwardOps       PortForwarder
 	cleanerOps           Cleaner
+	kubeExecutor         KubeExecutor
 	SnapshotGroupVersion *metav1.GroupVersionForDiscovery
 }
 
@@ -194,19 +213,40 @@ func (p *pvcBrowserSteps) CreateInspectorApplication(ctx context.Context, args *
 		Namespace:      args.Namespace,
 		RunAsUser:      args.RunAsUser,
 		ContainerImage: "filebrowser/filebrowser:v2",
-		ContainerArgs:  []string{"--noauth", "-r", "/data"},
-		MountPath:      "/data",
+		ContainerArgs:  []string{"--noauth", "-r", "/pvc-data"},
+		MountPath:      "/pvc-data",
+	}
+	if args.ShowTree {
+		podArgs = &types.CreatePodArgs{
+			GenerateName:   clonedPodGenerateName,
+			PVCName:        pvc.Name,
+			Namespace:      args.Namespace,
+			RunAsUser:      args.RunAsUser,
+			ContainerImage: "alpine:3.19",
+			Command:        []string{"/bin/sh"},
+			ContainerArgs:  []string{"-c", "while true; do sleep 3600; done"},
+			MountPath:      "/pvc-data",
+		}
 	}
 	pod, err := p.createAppOps.CreatePod(ctx, podArgs)
 	if err != nil {
-		return nil, pvc, errors.Wrap(err, "Failed to create restored Pod")
+		return nil, pvc, errors.Wrap(err, "Failed to create browse Pod")
 	}
 	if err = p.createAppOps.WaitForPodReady(ctx, args.Namespace, pod.Name); err != nil {
 		return pod, pvc, errors.Wrap(err, "Pod failed to become ready")
 	}
 	return pod, pvc, nil
 }
 
+func (p *pvcBrowserSteps) ExecuteTreeCommand(ctx context.Context, args *types.PVCBrowseArgs, pod *v1.Pod) (string, error) {
+	command := []string{"tree", "/pvc-data"}
+	stdout, err := p.kubeExecutor.Exec(ctx, args.Namespace, pod.Name, pod.Spec.Containers[0].Name, command)
+	if err != nil {
+		return "", errors.Wrapf(err, "Error running command:(%v)", command)
+	}
+	return stdout, nil
+}
+
 func (p *pvcBrowserSteps) PortForwardAPod(ctx context.Context, pod *v1.Pod, localPort int) error {
 	var wg sync.WaitGroup
 	wg.Add(1)

pkg/csi/pvc_inspector_steps_test.go

@@ -541,8 +541,8 @@ func (s *CSITestSuite) TestCreateInspectorApplicationForPVC(c *C) {
 						GenerateName:   clonedPodGenerateName,
 						PVCName:        "pvc1",
 						Namespace:      "ns",
-						ContainerArgs:  []string{"--noauth", "-r", "/data"},
-						MountPath:      "/data",
+						ContainerArgs:  []string{"--noauth", "-r", "/pvc-data"},
+						MountPath:      "/pvc-data",
 						RunAsUser:      100,
 						ContainerImage: "filebrowser/filebrowser:v2",
 					}).Return(&v1.Pod{
@@ -596,8 +596,8 @@ func (s *CSITestSuite) TestCreateInspectorApplicationForPVC(c *C) {
 						GenerateName:   clonedPodGenerateName,
 						PVCName:        "pvc1",
 						Namespace:      "ns",
-						ContainerArgs:  []string{"--noauth", "-r", "/data"},
-						MountPath:      "/data",
+						ContainerArgs:  []string{"--noauth", "-r", "/pvc-data"},
+						MountPath:      "/pvc-data",
 						RunAsUser:      100,
 						ContainerImage: "filebrowser/filebrowser:v2",
 					}).Return(&v1.Pod{