taskstore

type Authenticator func(context.Context) (string, error)

type InMemory struct {
	// contains filtered or unexported fields
}

type InMemoryStoreConfig struct {
	Authenticator Authenticator
	TimeProvider  func() time.Time
}

type Store interface {
	// Create creates a new task. It should return [ErrTaskAlreadyExists] if a task with the provided ID already exists.
	Create(ctx context.Context, task *a2a.Task) (TaskVersion, error)

	// Update updates the stored task. It should return [a2a.ErrTaskNotFound] if a task with the provided ID doesn't exist.
	Update(ctx context.Context, update *UpdateRequest) (TaskVersion, error)

	// Get retrieves a task by ID. If a Task doesn't exist the method should return [a2a.ErrTaskNotFound].
	Get(ctx context.Context, taskID a2a.TaskID) (*StoredTask, error)

	// List retrieves a list of tasks based on the provided request.
	List(ctx context.Context, req *a2a.ListTasksRequest) (*a2a.ListTasksResponse, error)
}

type StoredTask struct {
	// Task is the stored data.
	Task *a2a.Task
	// Version is the task store version used for tracking task modifications.
	Version TaskVersion
}

type TaskVersion int64

type UpdateRequest struct {
	// Task represents the desired state of the task in the store.
	Task *a2a.Task
	// Event is the event that triggered the update. It can be a user message which is added to task history.
	Event a2a.Event
	// PrevTask is the previous state of the task in the store. It is passed for detecting concurrent udpates.
	PrevTask *a2a.Task
	// PrevVersion is the version of the task before the update. It is passed for detecting concurrent udpates.
	// If the provided version does not match the latest task version the update request MUST be rejected with [ErrConcurrentModification].
	PrevVersion TaskVersion
}