Reconstruction app_cli.go

Author: Timo

pkg/app/app_cli.go

@@ -0,0 +1,227 @@
+package app
+
+import (
+	"bufio"
+	"fmt"
+	"os"
+	"os/exec"
+	"os/signal"
+	"strings"
+	"syscall"
+	"time"
+
+	"github.com/chzyer/readline"
+	fzf "github.com/junegunn/fzf/src"
+	"github.com/yaogh99123/dcli/pkg/utils"
+)
+
+// runFzfSelect 是一个通用的 FZF 选择器封装
+func (app *App) runFzfSelect(header string, lines []string) string {
+	if len(lines) == 0 {
+		return ""
+	}
+
+	// 1. 设置 fzf 选项
+	fzfArgs := []string{
+		"--height=40%",
+		"--reverse",
+		fmt.Sprintf("--header=%s", header),
+		"--cycle",
+	}
+
+	opts, err := fzf.ParseOptions(true, fzfArgs)
+	if err != nil {
+		return ""
+	}
+
+	// 2. 准备输入通道 (带缓冲)
+	inputChan := make(chan string, len(lines))
+	for _, line := range lines {
+		inputChan <- line
+	}
+	close(inputChan)
+	opts.Input = inputChan
+
+	// 3. 准备输出通道
+	outputChan := make(chan string, 1)
+	opts.Output = outputChan
+
+	// 4. 运行 fzf
+	// 在启动 fzf 前，必须先关闭 readline 实例以释放终端控制权
+	if app.RLInstance != nil {
+		app.RLInstance.Close()
+	}
+
+	code, _ := fzf.Run(opts)
+
+	// fzf 运行结束后，重新初始化 readline 供主循环使用
+	app.RLInstance, _ = readline.NewEx(&readline.Config{
+		Prompt:          fmt.Sprintf("%s请选择功能 [0-18,100]: %s", utils.ColorCyan, utils.ColorNC),
+		InterruptPrompt: "^C",
+		EOFPrompt:       "exit",
+	})
+
+	// 5. 处理结果 (130 为取消/Esc/Ctrl-C)
+	if code == 130 {
+		return ""
+	}
+
+	// 6. 获取选中项
+	select {
+	case result := <-outputChan:
+		return result
+	case <-time.After(50 * time.Millisecond):
+	}
+
+	return ""
+}
+
+// parseFzfResult 是一个辅助函数，用于解析 FZF 返回的行（通常格式为 "ID: Text" 或 "Index. Name: Status"）
+func (app *App) parseFzfResult(result string) string {
+	if result == "" {
+		return ""
+	}
+
+	// 策略 1: 经典的 "ID: Text" 格式 (用于菜单)
+	if strings.Contains(result, ": ") {
+		parts := strings.Split(result, ":")
+		if len(parts) > 0 {
+			left := strings.TrimSpace(parts[0])
+			// 策略 2: "1. Name" 格式 (用于服务搜索)
+			dotIdx := strings.Index(left, ". ")
+			if dotIdx != -1 {
+				return strings.TrimSpace(left[dotIdx+2:])
+			}
+			return left
+		}
+	}
+
+	return strings.TrimSpace(result)
+}
+
+func (app *App) runSubprocessWithQuitKey(cmd *exec.Cmd) error {
+	cmd.Stdout = os.Stdout
+	cmd.Stderr = os.Stderr
+
+	// 为子进程创建一个单独的管道，用于在特殊情况下强制杀掉它
+	if err := cmd.Start(); err != nil {
+		return err
+	}
+
+	done := make(chan error, 1)
+	go func() {
+		done <- cmd.Wait()
+	}()
+
+	// 捕获系统中断，防止主程序退出
+	sigChan := make(chan os.Signal, 1)
+	signal.Notify(sigChan, os.Interrupt, syscall.SIGTERM)
+	defer signal.Stop(sigChan)
+
+	// 监听 'exit' 命令来退出
+	quitChan := make(chan bool, 1)
+	go func() {
+		scanner := bufio.NewScanner(os.Stdin)
+		for scanner.Scan() {
+			text := strings.TrimSpace(scanner.Text())
+			if text == "exit" {
+				quitChan <- true
+				return
+			}
+			// 如果只是按了回车或其他，不做任何操作，继续等待
+		}
+	}()
+
+	select {
+	case err := <-done:
+		if err != nil {
+			fmt.Printf("\n%s指令执行出错: %v%s\n", utils.ColorRed, err, utils.ColorNC)
+		}
+		fmt.Printf("\n%s--- 执行完毕，输入 'exit' 返回主菜单 ---%s\n", utils.ColorBlue, utils.ColorNC)
+		// 子进程虽然结束了，但我们要继续等待 quitChan 里的 'exit' 命令
+		goto WAIT_LOOP
+	case <-sigChan:
+		// 收到 Ctrl+C，打印提示但不退出
+		fmt.Printf("\n%s[提示] 请输入 'exit' 并回车以返回主菜单%s\n", utils.ColorYellow, utils.ColorNC)
+		goto WAIT_LOOP
+	case <-quitChan:
+		// 收到 exit 命令，如果子进程还在跑，就杀掉它
+		_ = cmd.Process.Signal(os.Interrupt)
+		fmt.Printf("\n%s正在返回主菜单...%s\n", utils.ColorBlue, utils.ColorNC)
+		select {
+		case <-done:
+		case <-time.After(500 * time.Millisecond):
+			_ = cmd.Process.Kill()
+		}
+		return nil
+	}
+
+WAIT_LOOP:
+	// 无论是因为子进程结束还是收到信号，都必须等到 quitChan 收到 'exit' 为止
+	for {
+		select {
+		case <-sigChan:
+			fmt.Printf("\n%s[提示] 必须输入 'exit' 才能退出当前界面%s\n", utils.ColorYellow, utils.ColorNC)
+		case <-quitChan:
+			return nil
+		case <-done:
+			// 这种情况下进程已经通过 done 退出了，不需要再处理，只需处理 quitChan
+		}
+	}
+}
+
+// runInteractiveSubprocess 专门用于需要完全控制 Stdin 的场景（如 docker exec -it）
+func (app *App) runInteractiveSubprocess(cmd *exec.Cmd) error {
+	cmd.Stdout = os.Stdout
+	cmd.Stderr = os.Stderr
+	cmd.Stdin = os.Stdin
+
+	// 捕获信号，但不做特殊处理，让它透传给子进程
+	sigChan := make(chan os.Signal, 1)
+	signal.Notify(sigChan, os.Interrupt, syscall.SIGTERM)
+	defer signal.Stop(sigChan)
+
+	if err := cmd.Start(); err != nil {
+		return err
+	}
+
+	done := make(chan error, 1)
+	go func() {
+		done <- cmd.Wait()
+	}()
+
+	select {
+	case err := <-done:
+		return err
+	case <-sigChan:
+		// 收到信号时，将信号发给子进程
+		_ = cmd.Process.Signal(os.Interrupt)
+		// 等待子进程退出
+		select {
+		case <-done:
+		case <-time.After(1 * time.Second):
+			_ = cmd.Process.Kill()
+		}
+		return nil
+	}
+}
+
+// ReadInput is a helper to read input using readline
+func (app *App) ReadInput(prompt string) string {
+	if app.RLInstance == nil {
+		var input string
+		fmt.Print(prompt)
+		_, _ = fmt.Scanln(&input)
+		return input
+	}
+
+	oldPrompt := app.RLInstance.Config.Prompt
+	app.RLInstance.SetPrompt(prompt)
+	defer app.RLInstance.SetPrompt(oldPrompt)
+
+	line, err := app.RLInstance.Readline()
+	if err != nil {
+		return ""
+	}
+	return strings.TrimSpace(line)
+}

pkg/app/gui.go

@@ -0,0 +1,104 @@
+package manager
+
+import (
+	"fmt"
+	"os/exec"
+	"strings"
+	"time"
+
+	"github.com/docker/docker/api/types/image"
+	"github.com/yaogh99123/dcli/pkg/commands"
+	"github.com/yaogh99123/dcli/pkg/utils"
+)
+
+// RunImageMenu handles image management CLI interactions
+func RunImageMenu(dockerCmd *commands.DockerCommand, readInput func(string) string, runInteractiveSubprocess func(*exec.Cmd) error) {
+	for {
+		images, err := dockerCmd.RefreshImages()
+		if err != nil {
+			fmt.Printf("%s错误: %v%s\n", utils.ColorRed, err, utils.ColorNC)
+			return
+		}
+
+		fmt.Printf("\033[H\033[2J") // 清屏
+		fmt.Printf("%s========================================%s\n", utils.ColorBlue, utils.ColorNC)
+		fmt.Printf("%s      Docker 镜像管理%s\n", utils.ColorBlue, utils.ColorNC)
+		fmt.Printf("%s========================================%s\n\n", utils.ColorBlue, utils.ColorNC)
+
+		// 打印表头
+		fmt.Printf("%-50s %-12s %-12s %-12s %-10s\n", "IMAGE", "ID", "DISK USAGE", "CONTENT SIZE", "EXTRA")
+		for i, img := range images {
+			name := img.Name
+			if len(name) > 48 {
+				name = name[:45] + "..."
+			}
+
+			extra := ""
+			if img.Image.Containers == -1 {
+				extra = "U" // Unused
+			}
+
+			fmt.Printf("%2d. %-46s %-12s %-12s %-12s %-10s\n",
+				i+1, name, img.ID[:12], img.GetDisplaySize(), img.GetDisplayContentSize(), extra)
+		}
+
+		fmt.Printf("\n%s镜像操作%s\n", utils.ColorGreen, utils.ColorNC)
+		fmt.Println("------------------------------")
+		fmt.Println("  1. 拉取镜像")
+		fmt.Println("  2. 删除指定镜像")
+		fmt.Println("  3. 删除所有镜像")
+		fmt.Println("------------------------------")
+		fmt.Println("  0. 返回")
+		fmt.Println("------------------------------")
+		fmt.Print("请选择: ")
+
+		input := readInput("")
+		input = strings.TrimSpace(input)
+
+		if input == "0" || input == "" {
+			break
+		}
+
+		switch input {
+		case "1":
+			fmt.Print("请输入要拉取的镜像名称 (如 nginx:latest): ")
+			name := readInput("")
+			name = strings.TrimSpace(name)
+			if name != "" {
+				fmt.Printf("%s正在拉取镜像 %s...%s\n", utils.ColorYellow, name, utils.ColorNC)
+				cmd := exec.Command("docker", "pull", name)
+				_ = runInteractiveSubprocess(cmd)
+			}
+		case "2":
+			fmt.Print("请输入要删除的镜像索引: ")
+			idxStr := readInput("")
+			var idx int
+			_, _ = fmt.Sscanf(strings.TrimSpace(idxStr), "%d", &idx)
+			if idx > 0 && idx <= len(images) {
+				img := images[idx-1]
+				fmt.Printf("%s确定要删除镜像 %s (ID: %s) 吗? (y/n): %s", utils.ColorYellow, img.Name, img.ID[:12], utils.ColorNC)
+				confirm := readInput("")
+				if strings.ToLower(strings.TrimSpace(confirm)) == "y" {
+					fmt.Printf("%s正在删除镜像...%s\n", utils.ColorYellow, utils.ColorNC)
+					if err := img.Remove(image.RemoveOptions{Force: true}); err != nil {
+						fmt.Printf("%s失败: %v%s\n", utils.ColorRed, err, utils.ColorNC)
+					} else {
+						fmt.Printf("%s成功%s\n", utils.ColorGreen, utils.ColorNC)
+					}
+					time.Sleep(1 * time.Second)
+				}
+			} else {
+				fmt.Printf("%s无效的索引%s\n", utils.ColorRed, utils.ColorNC)
+				time.Sleep(1 * time.Second)
+			}
+		case "3":
+			fmt.Printf("%s危险: 这将删除所有未被使用的镜像! 是否继续? (y/n): %s", utils.ColorRed, utils.ColorNC)
+			confirm := readInput("")
+			if strings.ToLower(strings.TrimSpace(confirm)) == "y" {
+				fmt.Printf("%s正在清理所有未使用镜像...%s\n", utils.ColorYellow, utils.ColorNC)
+				cmd := exec.Command("docker", "image", "prune", "-af")
+				_ = runInteractiveSubprocess(cmd)
+			}
+		}
+	}
+}