forked from pion/mediadevices
/
wrapper.go
106 lines (92 loc) · 1.94 KB
/
wrapper.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
package driver
import (
"github.com/erichsu/mediadevices/pkg/io/audio"
"github.com/erichsu/mediadevices/pkg/io/video"
"github.com/erichsu/mediadevices/pkg/prop"
"github.com/google/uuid"
)
func wrapAdapter(a Adapter, info Info) Driver {
generator, err := uuid.NewRandom()
if err != nil {
panic(err)
}
id := generator.String()
d := &adapterWrapper{
Adapter: a,
id: id,
info: info,
state: StateClosed,
}
switch v := a.(type) {
case VideoRecorder:
// Only expose Driver and VideoRecorder interfaces
d.VideoRecorder = v
r := &struct {
Driver
VideoRecorder
}{d, d}
return r
case AudioRecorder:
// Only expose Driver and AudioRecorder interfaces
d.AudioRecorder = v
return &struct {
Driver
AudioRecorder
}{d, d}
default:
panic("adapter has to be either VideoRecorder/AudioRecorder")
}
}
type adapterWrapper struct {
Adapter
VideoRecorder
AudioRecorder
id string
info Info
state State
}
func (w *adapterWrapper) ID() string {
return w.id
}
func (w *adapterWrapper) Info() Info {
return w.info
}
func (w *adapterWrapper) Status() State {
return w.state
}
func (w *adapterWrapper) Open() error {
return w.state.Update(StateOpened, w.Adapter.Open)
}
func (w *adapterWrapper) Close() error {
return w.state.Update(StateClosed, w.Adapter.Close)
}
func (w *adapterWrapper) Properties() []prop.Media {
if w.state == StateClosed {
return nil
}
p := w.Adapter.Properties()
for i := range p {
p[i].DeviceID = w.id
}
return p
}
func (w *adapterWrapper) VideoRecord(p prop.Media) (r video.Reader, err error) {
err = w.state.Update(StateRunning, func() error {
r, err = w.VideoRecorder.VideoRecord(p)
return err
})
if err != nil {
_ = w.Close()
}
return
}
func (w *adapterWrapper) AudioRecord(p prop.Media) (r audio.Reader, err error) {
err = w.state.Update(StateRunning, func() error {
r, err = w.AudioRecorder.AudioRecord(p)
return err
})
if err != nil {
_ = w.Close()
}
return
}