2c805ed001
Before this commit, we held onto *cobra.Command objects, but that is not actually supported: after the first Execute(), commands like updateCmd are stuck on the first-ever provided ctx. Instead, turn command initialization into functions. I only noticed this when trying to do two 'gok update' from within the same test, where the fake build timestamp is injected via the context (the timestamp was always the same).
72 lines
1.5 KiB
Go
72 lines
1.5 KiB
Go
package gok
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"io"
|
|
"os"
|
|
|
|
"github.com/gokrazy/gokapi"
|
|
"github.com/gokrazy/gokapi/ondeviceapi"
|
|
"github.com/gokrazy/internal/config"
|
|
"github.com/gokrazy/internal/instanceflag"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
// psCmd is gok ps.
|
|
func psCmd() *cobra.Command {
|
|
cmd := &cobra.Command{
|
|
GroupID: "runtime",
|
|
Use: "ps",
|
|
Short: "list processes of a running gokrazy instance",
|
|
Long: `gok ps
|
|
|
|
Examples:
|
|
% gok -i scan2drive ps
|
|
`,
|
|
RunE: func(cmd *cobra.Command, args []string) error {
|
|
return psImpl.run(cmd.Context(), args, cmd.OutOrStdout(), cmd.OutOrStderr())
|
|
},
|
|
}
|
|
instanceflag.RegisterPflags(cmd.Flags())
|
|
return cmd
|
|
}
|
|
|
|
type psImplConfig struct {
|
|
}
|
|
|
|
var psImpl psImplConfig
|
|
|
|
func (r *psImplConfig) run(ctx context.Context, args []string, stdout, stderr io.Writer) error {
|
|
cfg, err := config.ApplyInstanceFlag()
|
|
if err != nil {
|
|
if os.IsNotExist(err) {
|
|
// best-effort compatibility for old setups
|
|
cfg = config.NewStruct(instanceflag.Instance())
|
|
} else {
|
|
return err
|
|
}
|
|
}
|
|
|
|
acfg, err := gokapi.ConnectRemotely(cfg)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
cl := ondeviceapi.NewAPIClient(acfg)
|
|
index, resp, err := cl.SuperviseApi.Index(ctx)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
_ = resp
|
|
fmt.Printf("Host: %s\n", cfg.Hostname)
|
|
fmt.Printf("Model: %s\n", index.Model)
|
|
fmt.Printf("Build: %s\n", index.BuildTimestamp)
|
|
fmt.Printf("Kernel: %s\n", index.Kernel)
|
|
fmt.Printf("Services (%d):\n", len(index.Services))
|
|
for _, svc := range index.Services {
|
|
fmt.Printf(" %s\n", svc.Path)
|
|
}
|
|
|
|
return nil
|
|
}
|