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
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
#[macro_use]
mod events;
mod allocator;
mod api;
mod boss;
mod code;
mod input;
mod key;
mod lister;
mod mailbox;
mod nameplate;
mod order;
mod receive;
mod rendezvous;
mod send;
mod server_messages;
mod terminator;
#[cfg(test)]
mod test;
mod transfer;
mod util;
mod wordlist;
use rustc_serialize::hex::ToHex;
use std::collections::VecDeque;
pub use self::events::{AppID, Code};
use self::events::{Event, Events, MySide, Nameplate};
use self::util::random_bytes;
use log::trace;
pub use self::api::{
APIAction, APIEvent, Action, IOAction, IOEvent, InputHelperError, Mood,
TimerHandle, WSHandle,
};
pub use self::transfer::{
direct_type, error_message, file_ack, message, message_ack,
offer_directory, offer_file, relay_type, transit, Abilities, AnswerType,
DirectType, Hints, OfferType, PeerMessage, RelayType, TransitType,
};
pub struct WormholeCore {
allocator: allocator::AllocatorMachine,
boss: boss::BossMachine,
code: code::CodeMachine,
input: input::InputMachine,
key: key::KeyMachine,
lister: lister::ListerMachine,
mailbox: mailbox::MailboxMachine,
nameplate: nameplate::NameplateMachine,
order: order::OrderMachine,
receive: receive::ReceiveMachine,
rendezvous: rendezvous::RendezvousMachine,
send: send::SendMachine,
terminator: terminator::TerminatorMachine,
}
fn generate_side() -> String {
let mut bytes: [u8; 5] = [0; 5];
random_bytes(&mut bytes);
bytes.to_hex()
}
impl WormholeCore {
pub fn new<T>(appid: T, relay_url: &str) -> WormholeCore
where
T: Into<AppID>,
{
let appid: AppID = appid.into();
let side = MySide(generate_side());
WormholeCore {
allocator: allocator::AllocatorMachine::new(),
boss: boss::BossMachine::new(),
code: code::CodeMachine::new(),
input: input::InputMachine::new(),
key: key::KeyMachine::new(&appid.clone(), &side),
lister: lister::ListerMachine::new(),
mailbox: mailbox::MailboxMachine::new(&side),
nameplate: nameplate::NameplateMachine::new(),
order: order::OrderMachine::new(),
receive: receive::ReceiveMachine::new(),
rendezvous: rendezvous::RendezvousMachine::new(
&appid.clone(),
relay_url,
&side,
5.0,
),
send: send::SendMachine::new(&side),
terminator: terminator::TerminatorMachine::new(),
}
}
pub fn start(&mut self) -> Vec<Action> {
self.do_api(APIEvent::Start)
}
pub fn do_api(&mut self, event: APIEvent) -> Vec<Action> {
trace!("api: {:?}", event);
let events = self.boss.process_api(event);
self._execute(events)
}
pub fn do_io(&mut self, event: IOEvent) -> Vec<Action> {
trace!("io: {:?}", event);
let events = self.rendezvous.process_io(event);
self._execute(events)
}
pub fn derive_key(&mut self, _purpose: &str, _length: u8) -> Vec<u8> {
Vec::new()
}
pub fn input_helper_get_nameplate_completions(
&mut self,
prefix: &str,
) -> Result<Vec<String>, InputHelperError> {
self.input.get_nameplate_completions(prefix)
}
pub fn input_helper_get_word_completions(
&mut self,
prefix: &str,
) -> Result<Vec<String>, InputHelperError> {
self.input.get_word_completions(prefix)
}
pub fn input_helper_committed_nameplate(&self) -> Option<&Nameplate> {
self.input.committed_nameplate()
}
fn _execute(&mut self, events: Events) -> Vec<Action> {
let mut action_queue: Vec<Action> = Vec::new(); let mut event_queue: VecDeque<Event> = VecDeque::new();
event_queue.append(&mut VecDeque::from(events.events));
while let Some(e) = event_queue.pop_front() {
trace!("event: {:?}", e);
use self::events::Event::*; let actions: Events = match e {
API(a) => {
action_queue.push(Action::API(a));
events![]
}
IO(a) => {
action_queue.push(Action::IO(a));
events![]
}
Allocator(e) => self.allocator.process(e),
Boss(e) => self.boss.process(e),
Code(e) => self.code.process(e),
Input(e) => self.input.process(e),
Key(e) => self.key.process(e),
Lister(e) => self.lister.process(e),
Mailbox(e) => self.mailbox.process(e),
Nameplate(e) => self.nameplate.process(e),
Order(e) => self.order.process(e),
Receive(e) => self.receive.process(e),
Rendezvous(e) => self.rendezvous.process(e),
Send(e) => self.send.process(e),
Terminator(e) => self.terminator.process(e),
};
for a in actions.events {
trace!(" out: {:?}", a);
event_queue.push_back(a);
}
}
action_queue
}
}