summaryrefslogtreecommitdiff
path: root/vendor/github.com/playwright-community/playwright-go/writable_stream.go
blob: 7f144b41c95b640f46190adfff085cc14c0eca4f (plain)
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
package playwright

import (
	"encoding/base64"
	"io"
	"os"
)

type writableStream struct {
	channelOwner
}

func (s *writableStream) Copy(file string) error {
	f, err := os.OpenFile(file, os.O_RDONLY, 0)
	if err != nil {
		return err
	}
	defer f.Close()

	for {
		buf := make([]byte, defaultCopyBufSize)
		n, err := f.Read(buf)
		if err != nil && err != io.EOF {
			return err
		}
		if n == 0 {
			break
		}
		_, err = s.channel.Send("write", map[string]interface{}{
			"binary": base64.StdEncoding.EncodeToString(buf[:n]),
		})
		if err != nil {
			return err
		}
	}
	_, err = s.channel.Send("close")
	return err
}

func newWritableStream(parent *channelOwner, objectType string, guid string, initializer map[string]interface{}) *writableStream {
	stream := &writableStream{}
	stream.createChannelOwner(stream, parent, objectType, guid, initializer)
	return stream
}