package tools

import (
	"bytes"
	"context"
	"os/exec"

	"go.uber.org/zap"
)

// CommandExecutor 用于执行命令并捕获输出
type CommandExecutor struct {
	Command string
	Args    []string
	Stdout  bytes.Buffer
	Stderr  bytes.Buffer
}

// NewCommandExecutor 创建一个新的 CommandExecutor 实例
func NewCommandExecutor() *CommandExecutor {
	return &CommandExecutor{}
}

// Run 执行命令并等待其完成
func (ce *CommandExecutor) Run(ctx context.Context, command string, args ...string) error {
	ce.Command = command
	ce.Args = args
	WithTraceID(ctx.Value(TraceIDKey{}).(string)).Info("args", zap.Any("args", args))
	cmd := exec.Command(ce.Command, ce.Args...)
	cmd.Stdout = &ce.Stdout
	cmd.Stderr = &ce.Stderr
	WithTraceID(ctx.Value(TraceIDKey{}).(string)).Info("Command:", zap.String("command", cmd.String()))
	if err := cmd.Start(); err != nil {
		WithTraceID(ctx.Value(TraceIDKey{}).(string)).Error("Error starting command:", zap.Error(err))
		return err
	}

	if err := cmd.Wait(); err != nil {
		WithTraceID(ctx.Value(TraceIDKey{}).(string)).Error("Command finished with error:", zap.Error(err))
		return err
	}

	return nil
}

// Output 获取标准输出
func (ce *CommandExecutor) Output() string {
	return ce.Stdout.String()
}

// StderrOutput 获取标准错误输出
func (ce *CommandExecutor) StderrOutput() string {
	return ce.Stderr.String()
}