Function newAgentHeartbeatWakeCommand

internal/cli/authored_context.go:639–688

(deps commandDeps)

}

func newAgentHeartbeatWakeCommand(deps commandDeps) *cobra.Command {
	var (
		sessionID      string
		source         string
		dryRun         bool
		idempotencyKey string
	)
	cmd := &cobra.Command{
		Use:     "wake <agent>",
		Short:   "Request one manual advisory Heartbeat wake",
		Example: "  agh agent heartbeat wake coder --session sess_123 --dry-run --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, err := changedNonEmptyStringFlag(cmd, sessionSessionKey, sessionID)
			if err != nil {
				return err
			}
			wakeSource := contract.HeartbeatWakeSource(strings.TrimSpace(source))
			if wakeSource == "" {
				wakeSource = contract.HeartbeatWakeSourceManual
			}
			record, err := client.WakeAgentHeartbeat(cmd.Context(), args[0], AgentHeartbeatWakeRequest{
				WorkspaceID:    workspace,
				AgentName:      args[0],
				SessionID:      session,
				Source:         wakeSource,
				DryRun:         dryRun,
				IdempotencyKey: strings.TrimSpace(idempotencyKey),
			})
			if err != nil {
				return err
			}
			return writeCommandOutput(cmd, agentHeartbeatWakeBundle(record))
		},
	}
	addWorkspaceFlag(cmd)
	cmd.Flags().StringVar(&sessionID, sessionSessionKey, "", "Session id to wake")
	cmd.Flags().StringVar(&source, "source", string(contract.HeartbeatWakeSourceManual), "Wake source")
	cmd.Flags().BoolVar(&dryRun, "dry-run", false, "Evaluate wake gates without sending a prompt")
	cmd.Flags().StringVar(&idempotencyKey, "idempotency-key", "", "Optional idempotency key")
	return cmd
}

func newSessionSoulCommand(deps commandDeps) *cobra.Command {
	cmd := &cobra.Command{