olcrtc/internal/transport/videochannel/engine_session.go

package videochannel

import (
	"context"
	"fmt"

	"github.com/openlibrecommunity/olcrtc/internal/engine"
	"github.com/pion/webrtc/v4"
)

// engineVideoSession adapts engine.Session + engine.VideoTrackCapable to the
// videoSession interface the videochannel transport consumes. The wrapper
// drops the *webrtc.DataChannel argument from the engine reconnect callback
// (videochannel does not use data channels) and exposes the video-track
// helpers under shorter names.
type engineVideoSession struct {
	session engine.Session
	vt      engine.VideoTrackCapable
}

func (v *engineVideoSession) Connect(ctx context.Context) error {
	if err := v.session.Connect(ctx); err != nil {
		return fmt.Errorf("connect: %w", err)
	}
	return nil
}

func (v *engineVideoSession) Close() error {
	if err := v.session.Close(); err != nil {
		return fmt.Errorf("close: %w", err)
	}
	return nil
}

func (v *engineVideoSession) SetReconnectCallback(cb func()) {
	v.session.SetReconnectCallback(func(*webrtc.DataChannel) {
		if cb != nil {
			cb()
		}
	})
}

func (v *engineVideoSession) SetShouldReconnect(fn func() bool) { v.session.SetShouldReconnect(fn) }
func (v *engineVideoSession) SetEndedCallback(cb func(string))  { v.session.SetEndedCallback(cb) }
func (v *engineVideoSession) WatchConnection(ctx context.Context) {
	v.session.WatchConnection(ctx)
}
func (v *engineVideoSession) CanSend() bool { return v.session.CanSend() }

func (v *engineVideoSession) Reconnect(reason string) { v.session.Reconnect(reason) }

func (v *engineVideoSession) AddTrack(track webrtc.TrackLocal) error {
	if err := v.vt.AddVideoTrack(track); err != nil {
		return fmt.Errorf("add track: %w", err)
	}
	return nil
}

func (v *engineVideoSession) SetTrackHandler(cb func(*webrtc.TrackRemote, *webrtc.RTPReceiver)) {
	v.vt.SetVideoTrackHandler(cb)
}