feat: add TryOr() TryOrElse() func to Option type

Author: initdc

option/option.go

@@ -245,6 +245,22 @@ func (o Option[T]) OrElse(f func() Option[T]) Option[T] {
 	return f()
 }
 
+func TryOr[T, U, E any](o Option[T], f func(T) U, e E) result.Result[U, E] {
+	if o.some {
+		return result.Ok[U, E](f(o.value))
+	}
+	return result.Err[U, E](e)
+}
+
+func TryOrElse[T, U, E any](o Option[T], f func(T) U, closure func() result.Result[U, E]) result.Result[U, E] {
+	if o.some {
+		return result.Ok[U, E](f(o.value))
+	}
+	closure()
+	// unreachable
+	return closure()
+}
+
 func (o Option[T]) Xor(optb Option[T]) Option[T] {
 	if o.some && !optb.some {
 		return o

option/option_test.go

@@ -2,10 +2,11 @@ package option
 
 import (
 	"fmt"
-	"github.com/initdc/types/result"
-	"github.com/stretchr/testify/assert"
 	"strconv"
 	"testing"
+
+	"github.com/initdc/types/result"
+	"github.com/stretchr/testify/assert"
 )
 
 func TestSome(t *testing.T) {
@@ -267,6 +268,25 @@ func TestOrElse(t *testing.T) {
 	assert.Equal(t, None[int]().OrElse(nobody), None[int]())
 }
 
+func TestTryOr(t *testing.T) {
+	f := func(x int) float32 { return float32(x) * 2 }
+	x := Some(2)
+	y := None[int]()
+
+	assert.Equal(t, TryOr(x, f, "bad"), result.Ok[float32, string](4.0))
+	assert.Equal(t, TryOr(y, f, "bad"), result.Err[float32, string]("bad"))
+}
+
+func TestTryOrElse(t *testing.T) {
+	f := func(x int) float32 { return float32(x) * 2 }
+	closure := func() result.Result[float32, string] { return result.Err[float32, string]("bad") }
+	x := Some(2)
+	y := None[int]()
+
+	assert.Equal(t, TryOrElse(x, f, closure), result.Ok[float32, string](4.0))
+	assert.Equal(t, TryOrElse(y, f, closure), result.Err[float32, string]("bad"))
+}
+
 func TestXor(t *testing.T) {
 	x := Some(2)
 	y := None[int]()