fix: add helper to register operation without calling untyped

Author: graham-chainlink

operations/operation.go

@@ -18,14 +18,14 @@ type Bundle struct {
 	reporter   Reporter
 	// internal use only, for storing the hash of the report to avoid repeat sha256 computation.
 	reportHashCache   *sync.Map
-	OperationRegistry OperationRegistry
+	OperationRegistry *OperationRegistry
 }
 
 // BundleOption is a functional option for configuring a Bundle
 type BundleOption func(*Bundle)
 
 // WithOperationRegistry sets a custom OperationRegistry for the Bundle
-func WithOperationRegistry(registry OperationRegistry) BundleOption {
+func WithOperationRegistry(registry *OperationRegistry) BundleOption {
 	return func(b *Bundle) {
 		b.OperationRegistry = registry
 	}

operations/registry.go

@@ -8,8 +8,8 @@ type OperationRegistry struct {
 }
 
 // NewOperationRegistry creates a new OperationRegistry with the provided untyped operations.
-func NewOperationRegistry(ops ...*Operation[any, any, any]) OperationRegistry {
-	return OperationRegistry{
+func NewOperationRegistry(ops ...*Operation[any, any, any]) *OperationRegistry {
+	return &OperationRegistry{
 		ops: ops,
 	}
 }
@@ -26,3 +26,12 @@ func (s OperationRegistry) Retrieve(def Definition) (*Operation[any, any, any],
 
 	return nil, errors.New("operation not found in registry")
 }
+
+// RegisterOperation registers new operations in the registry.
+// To register operations with different input, output, and dependency types,
+// call RegisterOperation multiple times with different type parameters.
+func RegisterOperation[D, I, O any](r *OperationRegistry, op ...*Operation[D, I, O]) {
+	for _, o := range op {
+		r.ops = append(r.ops, o.AsUntyped())
+	}
+}

operations/registry_test.go

@@ -36,8 +36,11 @@ func ExampleOperationRegistry() {
 			return input, nil
 		},
 	)
-	// Create registry with untyped operations
-	registry := NewOperationRegistry(stringOp.AsUntyped(), intOp.AsUntyped())
+	// Create registry with untyped operations by providing optional initial operation
+	registry := NewOperationRegistry(stringOp.AsUntyped())
+
+	// An alternative way to register additional operations without calling AsUntyped()
+	RegisterOperation(registry, intOp)
 
 	// Create execution environment
 	b := NewBundle(context.Background, logger.Nop(), NewMemoryReporter(), WithOperationRegistry(registry))