mirror of
https://github.com/xaionaro-go/streamctl.git
synced 2025-09-26 19:41:17 +08:00
120 lines
3.1 KiB
Go
120 lines
3.1 KiB
Go
package config
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
"image"
|
|
"time"
|
|
|
|
"github.com/xaionaro-go/obs-grpc-proxy/protobuf/go/obs_grpc"
|
|
"github.com/xaionaro-go/recoder"
|
|
"github.com/xaionaro-go/streamctl/pkg/streamtypes"
|
|
)
|
|
|
|
type DashboardSourceImageType string
|
|
|
|
const (
|
|
DashboardSourceImageTypeUndefined = DashboardSourceImageType("")
|
|
DashboardSourceImageTypeDummy = DashboardSourceImageType("dummy")
|
|
DashboardSourceImageTypeStreamScreenshot = DashboardSourceImageType("stream_screenshot")
|
|
DashboardSourceImageTypeOBSScreenshot = DashboardSourceImageType("obs_screenshot")
|
|
DashboardSourceImageTypeOBSVolume = DashboardSourceImageType("obs_volume")
|
|
)
|
|
|
|
func (mst DashboardSourceImageType) New() SourceImage {
|
|
switch mst {
|
|
case DashboardSourceImageTypeDummy:
|
|
return &DashboardSourceImageDummy{}
|
|
case DashboardSourceImageTypeStreamScreenshot:
|
|
return &DashboardSourceImageStreamScreenshot{}
|
|
case DashboardSourceImageTypeOBSScreenshot:
|
|
return &DashboardSourceImageOBSScreenshot{}
|
|
case DashboardSourceImageTypeOBSVolume:
|
|
return &DashboardSourceImageOBSVolume{}
|
|
default:
|
|
return nil
|
|
}
|
|
}
|
|
|
|
type ImageFormat string
|
|
|
|
const (
|
|
ImageFormatUndefined = ImageFormat("")
|
|
ImageFormatPNG = ImageFormat("png")
|
|
ImageFormatJPEG = ImageFormat("jpeg")
|
|
ImageFormatWebP = ImageFormat("webp")
|
|
)
|
|
|
|
type Duration time.Duration
|
|
|
|
func (d Duration) MarshalJSON() ([]byte, error) {
|
|
return json.Marshal(time.Duration(d).String())
|
|
}
|
|
|
|
func (d *Duration) UnmarshalJSON(b []byte) error {
|
|
var value any
|
|
if err := json.Unmarshal(b, &value); err != nil {
|
|
return fmt.Errorf("unable to un-JSON-ize '%s': %w", b, err)
|
|
}
|
|
|
|
switch value := value.(type) {
|
|
case float64:
|
|
*d = Duration(value)
|
|
return nil
|
|
case string:
|
|
duration, err := time.ParseDuration(value)
|
|
if err != nil {
|
|
return fmt.Errorf("unable to parse '%s' as duration: %w", value, err)
|
|
}
|
|
*d = Duration(duration)
|
|
return nil
|
|
default:
|
|
return fmt.Errorf("unexpected type: %T", value)
|
|
}
|
|
}
|
|
|
|
var _ SourceImage = (*DashboardSourceImageDummy)(nil)
|
|
|
|
type ImageDataProvider interface {
|
|
GetOBSServer(context.Context) (obs_grpc.OBSServer, error)
|
|
GetOBSState(context.Context) (*streamtypes.OBSState, error)
|
|
GetCurrentStreamFrame(context.Context, streamtypes.StreamID) ([]byte, recoder.VideoCodec, error)
|
|
}
|
|
|
|
type SourceImage interface {
|
|
GetImage(
|
|
ctx context.Context,
|
|
el DashboardElementConfig,
|
|
imageDataProvider ImageDataProvider,
|
|
) (image.Image, time.Time, error)
|
|
|
|
SourceType() DashboardSourceImageType
|
|
}
|
|
|
|
type GetImageBytes interface {
|
|
GetImageBytes(
|
|
ctx context.Context,
|
|
el DashboardElementConfig,
|
|
imageDataProvider ImageDataProvider,
|
|
) ([]byte, string, time.Time, error)
|
|
}
|
|
|
|
type serializableSourceImage struct {
|
|
Type DashboardSourceImageType `yaml:"type"`
|
|
Config map[string]any `yaml:"config,omitempty"`
|
|
}
|
|
|
|
func (s serializableSourceImage) Unwrap() SourceImage {
|
|
result := s.Type.New()
|
|
fromMap(s.Config, &result)
|
|
return result
|
|
}
|
|
|
|
func wrapSourceImageForYaml(source SourceImage) serializableSourceImage {
|
|
return serializableSourceImage{
|
|
Type: source.SourceType(),
|
|
Config: toMap(source),
|
|
}
|
|
}
|