Add callback command

cmd/callbacks/callbacks.go

@@ -0,0 +1,41 @@
+package callbacks
+
+import (
+	"github.com/resonatehq/resonate/pkg/client"
+	"github.com/spf13/cobra"
+)
+
+func NewCmd() *cobra.Command {
+	var (
+		c        = client.New()
+		server   string
+		username string
+		password string
+	)
+
+	cmd := &cobra.Command{
+		Use:     "callbacks",
+		Aliases: []string{"callback"},
+		Short:   "Resonate callbacks",
+		Run: func(cmd *cobra.Command, args []string) {
+			_ = cmd.Help()
+		},
+		PersistentPreRunE: func(cmd *cobra.Command, args []string) error {
+			if username != "" || password != "" {
+				c.SetBasicAuth(username, password)
+			}
+
+			return c.Setup(server)
+		},
+	}
+
+	// Add subcommands
+	cmd.AddCommand(CreateCallbackCmd(c))
+
+	// Flags
+	cmd.PersistentFlags().StringVarP(&server, "server", "", "http://localhost:8001", "resonate url")
+	cmd.PersistentFlags().StringVarP(&username, "username", "U", "", "basic auth username")
+	cmd.PersistentFlags().StringVarP(&password, "password", "P", "", "basic auth password")
+
+	return cmd
+}

cmd/callbacks/create.go

@@ -0,0 +1,95 @@
+package callbacks
+
+import (
+	"context"
+	"encoding/json"
+	"errors"
+	"time"
+
+	"github.com/resonatehq/resonate/pkg/client"
+	v1 "github.com/resonatehq/resonate/pkg/client/v1"
+	"github.com/spf13/cobra"
+)
+
+var createCallbacksExample = `
+# Create a callback
+resonate callback create foo --promise-id bar --root-promise-id baz --timeout 1h --recv default
+
+# Create a callback with url
+resonate callback create foo --promise-id bar --root-promise-id baz --timeout 1h --recv poll://default/6fa89b7e-4a56-40e8-ba4e-78864caa3278
+
+# Create a callback with object
+resonate callback create foo --promise-id bar --root-promise-id baz --timeout 1h --recv {"type": "poll", "data": {"group": "default", "id": "6fa89b7e-4a56-40e8-ba4e-78864caa3278"}}
+`
+
+func CreateCallbackCmd(c client.Client) *cobra.Command {
+	var (
+		promiseId     string
+		rootPromiseId string
+		timeout       time.Duration
+		recvStr       string
+	)
+	cmd := &cobra.Command{
+		Use:     "create <id>",
+		Short:   "Create callback",
+		Example: createCallbacksExample,
+		RunE: func(cmd *cobra.Command, args []string) error {
+			if len(args) != 1 {
+				return errors.New("must specify an id")
+			}
+
+			id := args[0]
+
+			var recv v1.Recv
+
+			if json.Valid([]byte(recvStr)) {
+				var recv0 v1.Recv0
+
+				if err := json.Unmarshal([]byte(recvStr), &recv0); err != nil {
+					return err
+				}
+				if err := recv.FromRecv0(recv0); err != nil {
+					return err
+				}
+			} else {
+				if err := recv.FromRecv1(recvStr); err != nil {
+					return err
+				}
+			}
+
+			body := v1.CreateCallbackJSONRequestBody{
+				Id:            id,
+				PromiseId:     promiseId,
+				RootPromiseId: rootPromiseId,
+				Timeout:       time.Now().Add(timeout).UnixMilli(),
+				Recv:          recv,
+			}
+
+			res, err := c.V1().CreateCallbackWithResponse(context.TODO(), nil, body)
+			if err != nil {
+				return err
+			}
+
+			if res.StatusCode() == 201 {
+				cmd.Printf("Created callback: %s\n", id)
+			} else if res.StatusCode() == 200 {
+				cmd.Printf("Created callback: %s (deduplicated)\n", id)
+			} else {
+				cmd.PrintErrln(res.Status(), string(res.Body))
+			}
+
+			return nil
+		},
+	}
+
+	cmd.Flags().StringVar(&promiseId, "promise-id", "", "promise id")
+	cmd.Flags().StringVar(&rootPromiseId, "root-promise-id", "", "root promise id")
+	cmd.Flags().DurationVar(&timeout, "timeout", 0, "task timeout")
+	cmd.Flags().StringVar(&recvStr, "recv", "default", "task receiver")
+
+	_ = cmd.MarkFlagRequired("promise-id")
+	_ = cmd.MarkFlagRequired("root-promise-id")
+	_ = cmd.MarkFlagRequired("timeout")
+
+	return cmd
+}

cmd/root.go

@@ -5,6 +5,7 @@ import (
 	"os"
 	"strings"
 
+	"github.com/resonatehq/resonate/cmd/callbacks"
 	"github.com/resonatehq/resonate/cmd/dst"
 	"github.com/resonatehq/resonate/cmd/promises"
 	"github.com/resonatehq/resonate/cmd/quickstart"
@@ -39,6 +40,7 @@ func init() {
 	rootCmd.AddCommand(serve.ServeCmd())
 	rootCmd.AddCommand(quickstart.NewCmd())
 	rootCmd.AddCommand(tasks.NewCmd())
+	rootCmd.AddCommand(callbacks.NewCmd())
 
 	// Set default output
 	rootCmd.SetOut(os.Stdout)

cmd/tasks/claim.go

@@ -12,7 +12,7 @@
 
 // Example command usage for claiming a task
 var claimTasksExample = `
-# Claim a task 
+# Claim a task
 resonate tasks claim foo --counter 1 --process-id bar --ttl 1m`
 
 // ClaimTaskCmd returns a cobra command for claiming a task.
@@ -74,9 +74,9 @@
 	}
 
 	// Define command flags
-	cmd.Flags().IntVarP(&counter, "counter", "c", 0, "The task counter")
-	cmd.Flags().StringVarP(&processId, "process-id", "p", "", "Unique process ID that identifies the claimer")
-	cmd.Flags().DurationVarP(&ttl, "ttl", "t", 0, "Time to live in milliseconds")
+	cmd.Flags().IntVarP(&counter, "counter", "c", 0, "task counter")
+	cmd.Flags().StringVarP(&processId, "process-id", "p", "", "unique id that identifies the claimer")
+	cmd.Flags().DurationVarP(&ttl, "ttl", "t", 0, "task time to live")
 
 	// Mark flags as required
 	_ = cmd.MarkFlagRequired("id")

cmd/tasks/complete.go

@@ -11,7 +11,7 @@
 
 // Example command usage for completing a task
 var completeTasksExample = `
-# Complete a task 
+# Complete a task
 resonate tasks complete foo --counter 1`
 
 // CompleteTaskCmd returns a cobra command for completing a task.
@@ -63,7 +63,7 @@
 	}
 
 	// Define command flags
-	cmd.Flags().IntVarP(&counter, "counter", "c", 0, "The task counter")
+	cmd.Flags().IntVarP(&counter, "counter", "c", 0, "task counter")
 
 	// Mark flags as required
 	_ = cmd.MarkFlagRequired("id")

cmd/tasks/heartbeat.go

@@ -10,7 +10,7 @@
 
 // Example command usage for sending a heartbeat to a task
 var heartbeatTasksExample = `
-# Heartbeat a task 
+# Heartbeat a task
 resonate tasks heartbeat --process-id foo`
 
 // HeartbeatTaskCmd returns a cobra command for sending a heartbeat to a task.
@@ -22,7 +22,7 @@
 	// Define the cobra command
 	cmd := &cobra.Command{
 		Use:     "heartbeat",
-		Short:   "Send a heartbeat to a task",
+		Short:   "Heartbeat tasks",
 		Example: heartbeatTasksExample,
 		RunE: func(cmd *cobra.Command, args []string) error {
 			// Create parameters for the request
@@ -42,7 +42,7 @@
 
 			// Handle the response based on the status code
 			if res.StatusCode() == 200 {
-				cmd.Printf("Heartbeat tasks for process-id: %s\n", processId)
+				cmd.Printf("Tasks heartbeated: %d\n", *res.JSON200.TasksAffected)
 			} else {
 				cmd.PrintErrln(res.Status(), string(res.Body))
 				return nil
@@ -53,7 +53,7 @@
 	}
 
 	// Define command flags
-	cmd.Flags().StringVarP(&processId, "process-id", "p", "", "Unique process ID to heartbeat tasks")
+	cmd.Flags().StringVarP(&processId, "process-id", "p", "", "unique id that identifies the claimer")
 
 	// Mark flags as required
 	_ = cmd.MarkFlagRequired("process-id")