cmd

var HelperCmd = &cobra.Command{
	Use:   "slurm",
	Short: "Helper for setting up and running in Slurm",
	Args:  cobra.ExactArgs(1),
}

var RestoreCmd = &cobra.Command{
	Use:    "slurm [args...]",
	Short:  "Restore a Slurm job",
	Args:   cobra.NoArgs,
	Hidden: true,
	RunE: func(cmd *cobra.Command, args []string) error {
		req, ok := cmd.Context().Value(keys.RESTORE_REQ_CONTEXT_KEY).(*daemon.RestoreReq)
		if !ok {
			return fmt.Errorf("invalid restore request in context")
		}

		jidStr, _ := cmd.Flags().GetString(slurm_flags.JidFlag.Full)
		jid, err := strconv.ParseUint(jidStr, 10, 32)
		if err != nil {
			return fmt.Errorf("invalid job id: %v", err)
		}

		req.Type = "slurm"
		req.Details = &daemon.Details{Slurm: &slurmpb.Slurm{
			JobID: uint32(jid),
		}}

		return nil
	},
}