odboy
6b6f8619df
- docker save/rmi：镜像名不再拼接 digest（{{.Digest}} 对多数镜像输出
sha256:…，导致 alpine:3.20 被存成 alpine:3.20@sha256:… 而查不到），
digest 单独保存；同时修正自定义 --format 输出中 IMAGE ID 表头未被过滤
- diagnose：目标不存在时给出相似名称候选，无相似时列出当前存在的对象
- net.pids：无进程占用时向 stderr 提示且 stdout 保持为空，新增 --pid-file
- net.ports：默认输出跨平台一致的端口表格，--raw 保留原生命令输出
- disk.usage：路径可省略（默认当前目录），新增 --depth N 控制层级
104 lines
3.2 KiB
Go
104 lines
3.2 KiB
Go
package executor
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"fmt"
|
|
"io"
|
|
|
|
"gitea.odboy.cn/odboy/kenaito-kopsc/internal/diagnose"
|
|
"gitea.odboy.cn/odboy/kenaito-kopsc/internal/platform"
|
|
)
|
|
|
|
// BuiltinRequest carries the caller's arguments and output streams to an
|
|
// in-process command implementation.
|
|
type BuiltinRequest struct {
|
|
Args []string
|
|
Stdout io.Writer
|
|
Stderr io.Writer
|
|
Info platform.Info
|
|
// Plan is the resolved variant and forwarded arguments. Builtins that
|
|
// drive an external runtime (for example docker save) read the runtime's
|
|
// program name and the passthrough arguments from it.
|
|
Plan Plan
|
|
}
|
|
|
|
// Builtin executes a command inside kopsc instead of spawning a child. It is
|
|
// used where no portable system utility exists (for example du on Windows).
|
|
type Builtin func(ctx context.Context, req BuiltinRequest) error
|
|
|
|
var builtins = map[string]Builtin{
|
|
"disk.usage": runDiskUsage,
|
|
"diagnose": runDiagnose,
|
|
"docker.df": runDockerDF,
|
|
"docker.exec": runDockerExec,
|
|
"docker.load": runDockerLoad,
|
|
"docker.logs": runDockerLogs,
|
|
"docker.restart": runDockerRestart,
|
|
"docker.rm": runDockerRM,
|
|
"docker.rm.preview": runDockerPreviewRM,
|
|
"docker.rmi": runDockerRMI,
|
|
"docker.rmi.preview": runDockerPreviewRMI,
|
|
"docker.save": runDockerSave,
|
|
"docker.stats": runDockerStats,
|
|
"docker.stop": runDockerStop,
|
|
"net.pids": runNetPIDs,
|
|
"net.ports": runNetPorts,
|
|
"proc.nohup": runNohup,
|
|
}
|
|
|
|
// runDiagnose bridges the registry builtin to the diagnose package.
|
|
func runDiagnose(ctx context.Context, req BuiltinRequest) error {
|
|
service := diagnose.New(req.Info, diagnose.ExecRunner{}, req.Stdout, req.Stderr)
|
|
return service.Run(ctx, req.Args)
|
|
}
|
|
|
|
// Preview runs a command's ConfirmBuiltin (if any) and writes the list of
|
|
// objects the command would act on. It is used for the confirmation prompt and
|
|
// for --dry-run, so neither path needs to touch the objects themselves.
|
|
//
|
|
// The list is built from the container/image references kopsc resolves, which
|
|
// means every target is validated first: an unknown target is reported here
|
|
// rather than after the user typed --yes. Commands without a ConfirmBuiltin
|
|
// print nothing.
|
|
func Preview(ctx context.Context, plan Plan, w io.Writer) error {
|
|
name := plan.PreviewBuiltin()
|
|
if name == "" {
|
|
return nil
|
|
}
|
|
fn, ok := LookupBuiltin(name)
|
|
if !ok {
|
|
return fmt.Errorf("未注册的确认动作 %q", name)
|
|
}
|
|
// Build the list in memory first: a target that fails to resolve must not
|
|
// leave a dangling "将影响:" heading behind.
|
|
var buf bytes.Buffer
|
|
if err := fn(ctx, BuiltinRequest{
|
|
Args: plan.ExtraArgs,
|
|
Stdout: &buf,
|
|
Stderr: io.Discard,
|
|
Info: plan.Result.Info,
|
|
Plan: plan,
|
|
}); err != nil {
|
|
return err
|
|
}
|
|
fmt.Fprintln(w, "将影响:")
|
|
_, err := w.Write(buf.Bytes())
|
|
return err
|
|
}
|
|
|
|
// LookupBuiltin returns the implementation registered under name.
|
|
func LookupBuiltin(name string) (Builtin, bool) {
|
|
fn, ok := builtins[name]
|
|
return fn, ok
|
|
}
|
|
|
|
// BuiltinNames lists the registered builtin implementations.
|
|
func BuiltinNames() []string {
|
|
names := make([]string, 0, len(builtins))
|
|
for name := range builtins {
|
|
names = append(names, name)
|
|
}
|
|
return names
|
|
}
|