package reactive

import "context"

type Observer[T any] interface {
	OnNext(T)
	OnError(error)
	OnComplete()
}

type ObserverFn[T any] func(T)

func (f ObserverFn[T]) OnNext(v T)        { f(v) }
func (f ObserverFn[T]) OnError(err error) {}
func (f ObserverFn[T]) OnComplete()       {}

func Subscribe[T any](ctx context.Context, o Observable[T], observer Observer[T]) {
	notifs := o.Observe(ctx)
	if notifs == nil {
		observer.OnComplete()
		return
	}

	for {
		select {
		case <-ctx.Done():
			observer.OnComplete()
			return
		case n, ok := <-notifs:
			if !ok {
				observer.OnComplete()
				return
			}
			switch n.Kind() {
			case NextNotification:
				observer.OnNext(n.Value())
			case ErrorNotification:
				observer.OnError(n.Error())
			}
		}
	}
}