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
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
|
//
// This file is part of dummy-discovery.
//
// Copyright 2021 ARDUINO SA (http://www.arduino.cc/)
//
// This software is released under the GNU General Public License version 3,
// which covers the main part of arduino-cli.
// The terms of this license can be found at:
// https://www.gnu.org/licenses/gpl-3.0.en.html
//
// You can be released from the requirements of the above licenses by purchasing
// a commercial license. Buying such a license is mandatory if you want to modify or
// otherwise use the software for commercial activities involving the Arduino
// software without disclosing the source code of your own applications. To purchase
// a commercial license, send an email to license@arduino.cc.
//
package main
import (
"errors"
"fmt"
"os"
"time"
"github.com/arduino/go-properties-orderedmap"
discovery "github.com/arduino/pluggable-discovery-protocol-handler/v2"
"github.com/arduino/pluggable-discovery-protocol-handler/v2/dummy-discovery/args"
)
// dummyDiscovery is an example implementation of a Discovery.
// It simulates a real implementation of a Discovery by generating
// connected ports deterministically, it can also be used for testing
// purposes.
type dummyDiscovery struct {
startSyncCount int
closeChan chan<- bool
}
func main() {
args.Parse()
dummy := &dummyDiscovery{}
server := discovery.NewServer(dummy)
if err := server.Run(os.Stdin, os.Stdout); err != nil {
os.Exit(1)
}
}
// Hello does nothing.
// In a real implementation it could setup background processes
// or other kind of resources necessary to discover Ports.
func (d *dummyDiscovery) Hello(userAgent string, protocol int) error {
return nil
}
// Quit does nothing.
// In a real implementation it can be used to tear down resources
// used to discovery Ports.
func (d *dummyDiscovery) Quit() {}
// Stop is used to stop the goroutine started by StartSync
// used to discover ports.
func (d *dummyDiscovery) Stop() error {
if d.closeChan != nil {
d.closeChan <- true
close(d.closeChan)
d.closeChan = nil
}
return nil
}
// StartSync starts the goroutine that generates fake Ports.
func (d *dummyDiscovery) StartSync(eventCB discovery.EventCallback, errorCB discovery.ErrorCallback) error {
d.startSyncCount++
if d.startSyncCount%5 == 0 {
return errors.New("could not start_sync every 5 times")
}
c := make(chan bool)
d.closeChan = c
// Run synchronous event emitter
go func() {
var closeChan <-chan bool = c
// Output initial port state
eventCB("add", createDummyPort())
eventCB("add", createDummyPort())
// Start sending events
count := 0
for count < 2 {
count++
select {
case <-closeChan:
return
case <-time.After(2 * time.Second):
}
port := createDummyPort()
eventCB("add", port)
select {
case <-closeChan:
return
case <-time.After(2 * time.Second):
}
eventCB("remove", &discovery.Port{
Address: port.Address,
Protocol: port.Protocol,
})
}
errorCB("unrecoverable error, cannot send more events")
<-closeChan
}()
return nil
}
var dummyCounter = 0
// createDummyPort creates a Port with fake data
func createDummyPort() *discovery.Port {
dummyCounter++
mac := fmt.Sprintf("%d", dummyCounter*384782)
return &discovery.Port{
Address: fmt.Sprintf("%d", dummyCounter),
AddressLabel: "Dummy upload port",
Protocol: "dummy",
ProtocolLabel: "Dummy protocol",
HardwareID: mac,
Properties: properties.NewFromHashmap(map[string]string{
"vid": "0x2341",
"pid": "0x0041",
"mac": mac,
}),
}
}
|
