apps/streamctl
1
0
Fork 0
mirror of https://github.com/xaionaro-go/streamctl.git synced 2025-10-13 03:03:44 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Initial commit, pt. 73

Browse Source
This commit is contained in:
Dmitrii Okunev
2024-08-04 23:06:49 +01:00
parent 2a7d947b78
commit c00508fc72
2 changed files with 34 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

47
pkg/streamcontrol/stream_control.go
View File

@@ -43,16 +43,39 @@ type StreamProfile interface {
AbstractStreamProfile
}
func AssertStreamProfile[T StreamProfile](
ctx context.Context,
v AbstractStreamProfile,
) (*T, error) {
profile, ok := v.(T)
if ok {
return &profile, nil
}
profilePtr, ok := (any)(v).(*T)
if ok {
return profilePtr, nil
}
var zeroProfile T
return nil, ErrInvalidStreamProfileType{Expected: zeroProfile, Received: v}
}
func GetStreamProfile[T StreamProfile](
ctx context.Context,
v AbstractStreamProfile,
) (*T, error) {
var profile T
profilePtr, err := AssertStreamProfile[T](ctx, v)
if err == nil {
return profilePtr, nil
}
b, err := json.Marshal(v)
if err != nil {
return nil, fmt.Errorf("unable to serialize: %w: %#+v", err, v)
}
logger.Debugf(ctx, "JSON representation: <%s>", b)
var profile T
err = json.Unmarshal(b, &profile)
if err != nil {
return nil, fmt.Errorf("unable to deserialize: %w: <%s>", err, b)
@@ -192,24 +215,18 @@ func ToAbstract[T StreamProfile](c StreamController[T]) AbstractStreamController
return &abstractStreamController{
StreamController: c,
applyProfile: func(ctx context.Context, _profile AbstractStreamProfile, customArgs ...any) error {
if _profile == nil {
return nil
profile, err := AssertStreamProfile[T](ctx, _profile)
if err != nil {
return err
}
profile, ok := _profile.(T)
if !ok {
return ErrInvalidStreamProfileType{Expected: zeroProfile, Received: _profile}
}
return c.ApplyProfile(ctx, profile, customArgs...)
return c.ApplyProfile(ctx, *profile, customArgs...)
},
startStream: func(ctx context.Context, title string, description string, _profile AbstractStreamProfile, customArgs ...any) error {
if _profile == nil {
return nil
profile, err := AssertStreamProfile[T](ctx, _profile)
if err != nil {
return err
}
profile, ok := _profile.(T)
if !ok {
return ErrInvalidStreamProfileType{Expected: zeroProfile, Received: _profile}
}
return c.StartStream(ctx, title, description, profile, customArgs...)
return c.StartStream(ctx, title, description, *profile, customArgs...)
},
StreamProfileTypeValue: profileType,
}