fix(kube-run): handle SIGINT/SIGTERM properly (#6888)

Signed-off-by: Alexandr Zaytsev <alexandr.zaytsev@flant.com>

Author: nervgh

cmd/werf/kube_run/kube_run.go

@@ -297,8 +297,6 @@ func runMain(ctx context.Context) error {
 				return err
 			}
 
-			cleanupResources(ctx, pod, secret, namespace)
-
 			return nil
 		})
 	} else {
@@ -680,6 +678,8 @@ func isPodReady(ctx context.Context, namespace, pod string, extraArgs []string)
 }
 
 func copyFromPod(ctx context.Context, namespace, pod, container string, copyFrom copyFromTo, extraArgs []string) {
+	ctx = context.WithoutCancel(ctx)
+
 	logboek.Context(ctx).LogF("Copying %q from pod to %q ...\n", copyFrom.Src, copyFrom.Dst)
 
 	args := []string{
@@ -730,6 +730,8 @@ func copyToPod(ctx context.Context, namespace, pod, container string, copyFrom c
 }
 
 func stopContainer(ctx context.Context, namespace, pod, container string, extraArgs []string) {
+	ctx = context.WithoutCancel(ctx)
+
 	logboek.Context(ctx).LogF("Stopping container %q in pod ...\n", container)
 
 	args := []string{
@@ -775,6 +777,7 @@ func execCommandInPod(ctx context.Context, namespace, pod, container string, com
 	args = append(args, command...)
 
 	cmd := werfExec.PrepareGracefulCancellation(util.ExecKubectlCmdContext(ctx, args...))
+	cmd.Stdin = os.Stdin
 
 	if *commonCmdData.DryRun {
 		fmt.Println(cmd.String())
@@ -792,6 +795,8 @@ func execCommandInPod(ctx context.Context, namespace, pod, container string, com
 }
 
 func cleanupResources(ctx context.Context, pod, secret, namespace string) {
+	ctx = context.WithoutCancel(ctx)
+
 	if !cmdData.Rm || *commonCmdData.DryRun {
 		return
 	}

cmd/werf/kubectl/kubectl.go

@@ -33,6 +33,7 @@ func NewCmd(ctx context.Context) *cobra.Command {
 		ConfigFlags:   configFlags,
 		IOStreams:     genericclioptions.IOStreams{In: os.Stdin, Out: os.Stdout, ErrOut: os.Stderr},
 	})
+	kubectlCmd.SetContext(ctx)
 
 	logs.AddFlags(kubectlCmd.PersistentFlags())
 
@@ -59,18 +60,16 @@ func NewCmd(ctx context.Context) *cobra.Command {
 		}
 	}
 
-	wrapPreRun(kubectlCmd)
+	wrapPreRun(ctx, kubectlCmd)
 
 	return kubectlCmd
 }
 
-func wrapPreRun(kubectlCmd *cobra.Command) {
+func wrapPreRun(ctx context.Context, kubectlCmd *cobra.Command) {
 	switch {
 	case kubectlCmd.PersistentPreRunE != nil:
 		oldFunc := kubectlCmd.PersistentPreRunE
 		kubectlCmd.PersistentPreRunE = func(cmd *cobra.Command, args []string) error {
-			ctx := cmd.Context()
-
 			if err := prePreRun(ctx); err != nil {
 				return err
 			}
@@ -79,17 +78,13 @@ func wrapPreRun(kubectlCmd *cobra.Command) {
 	case kubectlCmd.PersistentPreRun != nil:
 		oldFunc := kubectlCmd.PersistentPreRun
 		kubectlCmd.PersistentPreRun = func(cmd *cobra.Command, args []string) {
-			ctx := cmd.Context()
-
 			if err := prePreRun(ctx); err != nil {
 				util.CheckErr(err)
 			}
 			oldFunc(cmd, args)
 		}
 	default:
 		kubectlCmd.PersistentPreRunE = func(cmd *cobra.Command, args []string) error {
-			ctx := cmd.Context()
-
 			if err := prePreRun(ctx); err != nil {
 				return err
 			}

cmd/werf/main.go

@@ -63,7 +63,7 @@ func main() {
 		return
 	}
 
-	if err := rootCmd.Execute(); err != nil {
+	if err := rootCmd.ExecuteContext(ctx); err != nil {
 		if helm_v3.IsPluginError(err) {
 			common.ShutdownTelemetry(ctx, helm_v3.PluginErrorCode(err))
 			graceful.Terminate(ctx, err, helm_v3.PluginErrorCode(err))