forked from MontFerret/ferret
-
Notifications
You must be signed in to change notification settings - Fork 0
/
browser.go
67 lines (49 loc) · 1.07 KB
/
browser.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
package browser
import (
"github.com/pkg/errors"
"os"
"os/exec"
"runtime"
)
type Browser struct {
cmd *exec.Cmd
flags Flags
}
func (b *Browser) Flags() Flags {
return b.flags
}
func (b *Browser) DebuggingAddress() string {
if !b.Flags().Has("remote-debugging-address") {
b.Flags().Set("remote-debugging-address", "0.0.0.0")
}
value, _ := b.Flags().Get("remote-debugging-address")
return value.(string)
}
func (b *Browser) DebuggingPort() int {
if !b.Flags().Has("remote-debugging-port") {
b.Flags().Set("remote-debugging-port", 9222)
}
value, _ := b.Flags().Get("remote-debugging-port")
return value.(int)
}
func (b *Browser) Close() error {
var err error
if runtime.GOOS != "windows" {
err = b.cmd.Process.Signal(os.Interrupt)
} else {
err = b.cmd.Process.Kill()
}
_, err = b.cmd.Process.Wait()
if err != nil {
return errors.Wrap(err, "error waiting for process exit, result unknown")
}
tmpDir, err := b.flags.GetString("user-data-dir")
if err != nil {
return nil
}
os.RemoveAll(tmpDir)
if err != nil {
return err
}
return nil
}