-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathmain.go
84 lines (75 loc) · 1.34 KB
/
main.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
package main
import (
"errors"
"flag"
"os"
"os/signal"
"time"
"go.bug.st/serial.v1/enumerator"
)
var (
ErrNotFound = errors.New("Port not found")
)
func check(vid string, pid string) (string, error) {
ports, err := enumerator.GetDetailedPortsList()
if err != nil {
return "", ErrNotFound
}
for _, port := range ports {
if port.IsUSB {
if port.VID == vid && port.PID == pid {
return port.Name, nil
}
}
}
return "", ErrNotFound
}
var (
vid = flag.String("vid", "", "USB vid to check for")
pid = flag.String("pid", "", "USB pid to check for")
)
func wait() (string, error) {
death := make(chan os.Signal, 1)
port, err := check(*vid, *pid)
if err == nil {
return port, nil
}
if err != ErrNotFound {
return "", err
}
signal.Notify(death, os.Interrupt, os.Kill)
ticker := time.NewTicker(time.Second * 5) // check every 5 minutes
for {
select {
case <-death:
os.Exit(0)
case <-ticker.C:
port, err := check(*vid, *pid)
if err != nil {
if err != ErrNotFound {
return "", err
}
continue
}
return port, nil
}
}
}
func main() {
flag.Parse()
port, err := wait()
if err != nil {
panic(err)
}
args := flag.Args()
command := args[0]
args = args[1:]
for i, arg := range args {
if arg == "{}" {
args[i] = port
}
}
if err := run(port, command, args); err != nil {
panic(err)
}
}