Function newAgentHeartbeatStatusCommand

internal/cli/authored_context.go:593–637

(deps commandDeps)

}

func newAgentHeartbeatStatusCommand(deps commandDeps) *cobra.Command {
	var (
		sessionID     string
		includeHealth bool
		includeEvents bool
	)
	cmd := &cobra.Command{
		Use:     "status <agent>",
		Short:   "Read Heartbeat policy status and wake eligibility",
		Example: "  agh agent heartbeat status coder --session sess_123 --workspace checkout-api --json",
		Args:    exactOneNonBlankArg(),
		RunE: func(cmd *cobra.Command, args []string) error {
			client, err := clientFromDeps(deps)
			if err != nil {
				return err
			}
			workspace, err := commandWorkspaceFlag(cmd)
			if err != nil {
				return err
			}
			session := strings.TrimSpace(sessionID)
			record, err := client.GetAgentHeartbeatStatus(cmd.Context(), args[0], AgentHeartbeatStatusRequest{
				WorkspaceID:             workspace,
				AgentName:               args[0],
				SessionID:               session,
				IncludeSessionHealth:    includeHealth || session != "",
				IncludeRecentWakeEvents: includeEvents,
			})
			if err != nil {
				return err
			}
			return writeCommandOutput(cmd, agentHeartbeatStatusBundle(record))
		},
	}
	addWorkspaceFlag(cmd)
	cmd.Flags().StringVar(&sessionID, sessionSessionKey, "", "Session id for wake state and health")
	cmd.Flags().BoolVar(
		&includeHealth,
		"include-session-health",
		false,
		"Include session health when a session id is supplied",
	)
	cmd.Flags().BoolVar(&includeEvents, "include-wake-events", false, "Include recent wake audit rows")
	return cmd
}

func newAgentHeartbeatWakeCommand(deps commandDeps) *cobra.Command {
	var (