mirror of
https://github.com/legop3/MultiRoombaRover.git
synced 2026-09-16 01:21:20 -04:00
Compare commits
690
Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
b19bec7ee5 | ||
|
|
987ef29f30 | ||
|
|
ba4cf80f41 | ||
|
|
622cd2875c | ||
|
|
98ca2e0626 | ||
|
|
dbc5111c53 | ||
|
|
c30325afc0 | ||
|
|
12512b8856 | ||
|
|
bb3197ca67 | ||
|
|
0b1551cabf | ||
|
|
69198a5556 | ||
|
|
f3a5e7bf77 | ||
|
|
d2acd9f19a | ||
|
|
94c3939671 | ||
|
|
f7de6d52be | ||
|
|
fff07ffbe6 | ||
|
|
9c046cc3b1 | ||
|
|
c465c03e34 | ||
|
|
99a1c5418d | ||
|
|
2fc9becc88 | ||
|
|
fe94c38963 | ||
|
|
55c72e0d9b | ||
|
|
a237d32d58 | ||
|
|
213630cb68 | ||
|
|
eb85b85b8f | ||
|
|
0600b8552c | ||
|
|
662efb300b | ||
|
|
e2f94720ec | ||
|
|
bcfb2946cf | ||
|
|
02603aa664 | ||
|
|
c67c3dc761 | ||
|
|
26d7335c9a | ||
|
|
a99e6e5286 | ||
|
|
d93d78cb2b | ||
|
|
e01878dd4e | ||
|
|
598f353fe5 | ||
|
|
7249621ebe | ||
|
|
6009056edd | ||
|
|
fd7b858395 | ||
|
|
1fd0de5b1e | ||
|
|
720a010973 | ||
|
|
2564a451b5 | ||
|
|
8326898403 | ||
|
|
e0edbcfb28 | ||
|
|
4f60248bc6 | ||
|
|
629d979b67 | ||
|
|
59c19f1da9 | ||
|
|
b9a007bd48 | ||
|
|
850ac3ec15 | ||
|
|
87dd9a24a0 | ||
|
|
bdcb488988 | ||
|
|
9e6053d5c0 | ||
|
|
4d699e1c4c | ||
|
|
5e7581c82d | ||
|
|
f7e4269959 | ||
|
|
f3f1db1398 | ||
|
|
133c061ee9 | ||
|
|
6859a6c048 | ||
|
|
465bddb9dd | ||
|
|
c7a639a568 | ||
|
|
df952b27e0 | ||
|
|
36f28a9708 | ||
|
|
2975d394b6 | ||
|
|
befcf7ad5a | ||
|
|
27ae3bcd1e | ||
|
|
90fd0262c7 | ||
|
|
64a9dd3583 | ||
|
|
972942e497 | ||
|
|
e34d838d41 | ||
|
|
705b38ea7b | ||
|
|
f129449d5a | ||
|
|
d4ce6e9092 | ||
|
|
2252527d89 | ||
|
|
d184590bf2 | ||
|
|
5b9167ab2a | ||
|
|
655fed4326 | ||
|
|
6c01087738 | ||
|
|
d272a33dc2 | ||
|
|
8212a14ea8 | ||
|
|
9622129c73 | ||
|
|
37927d82af | ||
|
|
23fe03ca54 | ||
|
|
899e647a9d | ||
|
|
cab4dc716d | ||
|
|
5caf26bb2d | ||
|
|
e666b4c195 | ||
|
|
c2ca951fe8 | ||
|
|
399c077c7a | ||
|
|
b8a4a18455 | ||
|
|
3aedaa26c8 | ||
|
|
a6e7e683ba | ||
|
|
27c46b44b2 | ||
|
|
5fdcb233f8 | ||
|
|
b7a79b03c9 | ||
|
|
fb00467106 | ||
|
|
32f9a2befc | ||
|
|
e10e2f9a0b | ||
|
|
8b0928f428 | ||
|
|
f3e13f6ab7 | ||
|
|
4ab6bad1b8 | ||
|
|
b4754b1c54 | ||
|
|
9aad7ff775 | ||
|
|
c3a9ea7281 | ||
|
|
ba652005f5 | ||
|
|
a49a090026 | ||
|
|
c265121335 | ||
|
|
56b89b5dff | ||
|
|
00f97c8005 | ||
|
|
d8ce85be60 | ||
|
|
3b67e81451 | ||
|
|
870fdc63be | ||
|
|
5ae0b0ca79 | ||
|
|
778543dd5c | ||
|
|
29154c56e6 | ||
|
|
50d3977814 | ||
|
|
acebcd7307 | ||
|
|
55782dabf0 | ||
|
|
79c118bc88 | ||
|
|
173d6fffa7 | ||
|
|
cf072ef0e0 | ||
|
|
30ebd05a97 | ||
|
|
d3f2f15370 | ||
|
|
008cf96c7e | ||
|
|
981262a0ae | ||
|
|
b884f23b13 | ||
|
|
dc9f93ca60 | ||
|
|
4005ced6f9 | ||
|
|
f7dd4f7b89 | ||
|
|
4b7c26a7b4 | ||
|
|
7335cefffb | ||
|
|
aba01161c9 | ||
|
|
b4edf72c9a | ||
|
|
066f976cf4 | ||
|
|
2f726035c9 | ||
|
|
5b8fb30e71 | ||
|
|
eb4e1621c0 | ||
|
|
6d55073bfb | ||
|
|
242fe461d9 | ||
|
|
fcc71426ee | ||
|
|
4e30eb1d52 | ||
|
|
e0cadb493f | ||
|
|
bbb4014f61 | ||
|
|
3c1c75a2ec | ||
|
|
c4dda507e2 | ||
|
|
56ed528412 | ||
|
|
258f244194 | ||
|
|
ec0e1564d7 | ||
|
|
4f2907a105 | ||
|
|
5829cd370c | ||
|
|
5ef6b640c6 | ||
|
|
ff638d84d2 | ||
|
|
061d4d4689 | ||
|
|
3cc12b76b9 | ||
|
|
1241c867a5 | ||
|
|
8b3806bcd5 | ||
|
|
eeb4e65df7 | ||
|
|
b714bd0fed | ||
|
|
a989f2029a | ||
|
|
ca1f7ef54a | ||
|
|
87db66ac1f | ||
|
|
8da2196ee0 | ||
|
|
49d822d468 | ||
|
|
fd765ec1da | ||
|
|
086d3cdee2 | ||
|
|
83f6f7eac1 | ||
|
|
3df303bc09 | ||
|
|
94a779581b | ||
|
|
c87767d3ee | ||
|
|
6ca30adcaf | ||
|
|
588305184e | ||
|
|
293b73dc20 | ||
|
|
d4440a4f05 | ||
|
|
3349120c0d | ||
|
|
bd862bf3f2 | ||
|
|
1a6ca5ce73 | ||
|
|
9ea29f8e36 | ||
|
|
b9d7a62b0e | ||
|
|
6301058f87 | ||
|
|
8e1de78cc7 | ||
|
|
0b564c7ee7 | ||
|
|
195ecb7d61 | ||
|
|
108de883a0 | ||
|
|
01df853592 | ||
|
|
020ddfea07 | ||
|
|
b4a5633bd4 | ||
|
|
f579516d64 | ||
|
|
6f887b00f1 | ||
|
|
0ce6c05e05 | ||
|
|
fe9f1c13e2 | ||
|
|
e3874b6237 | ||
|
|
e0a64e828e | ||
|
|
a8fb23287c | ||
|
|
ed9bb3216c | ||
|
|
1b013ed91b | ||
|
|
8e01dcaabd | ||
|
|
0836afc0c0 | ||
|
|
8296278e1c | ||
|
|
6028e275da | ||
|
|
314b88b146 | ||
|
|
de4295caf1 | ||
|
|
c5e66b3ec5 | ||
|
|
aba5efcf40 | ||
|
|
e69fd18256 | ||
|
|
0be105e9dd | ||
|
|
a1fef27a59 | ||
|
|
77ec19a4d6 | ||
|
|
d4d6f0674d | ||
|
|
e334246cef | ||
|
|
81a8c35873 | ||
|
|
9c3dbfebcb | ||
|
|
cd9e82938b | ||
|
|
0fb6ea49b3 | ||
|
|
a3dae83bf7 | ||
|
|
1bd486a6c0 | ||
|
|
b965f7b561 | ||
|
|
45aef7339e | ||
|
|
fb7ecec17e | ||
|
|
7b29d741b0 | ||
|
|
67f4f014c4 | ||
|
|
c42d314cc9 | ||
|
|
102192c631 | ||
|
|
cfe9055cb7 | ||
|
|
501bc234ee | ||
|
|
c07e5971f5 | ||
|
|
bbdecc324a | ||
|
|
8b59d021d3 | ||
|
|
3221dbe877 | ||
|
|
57d4b5a415 | ||
|
|
6580de08d4 | ||
|
|
161929e280 | ||
|
|
e9fd5f55a8 | ||
|
|
ae1c92a0ce | ||
|
|
127eb446ed | ||
|
|
66824240d8 | ||
|
|
115ecad9a6 | ||
|
|
d857668109 | ||
|
|
b72c58452e | ||
|
|
2b1122dce9 | ||
|
|
9f0bb8bd24 | ||
|
|
edeac5e6ff | ||
|
|
e0312c3fbb | ||
|
|
e94a52d7fd | ||
|
|
ace39d5a70 | ||
|
|
9bc5d2d06d | ||
|
|
57575f7199 | ||
|
|
2ad71a51c7 | ||
|
|
9dd17929df | ||
|
|
543c2b452e | ||
|
|
a65e80c82a | ||
|
|
185a826387 | ||
|
|
40d6ca9bf0 | ||
|
|
c129f38e6b | ||
|
|
0da5d72d91 | ||
|
|
1c4b9b8c3d | ||
|
|
c94abeca07 | ||
|
|
f14aea2070 | ||
|
|
1d664cbe4c | ||
|
|
2b2b6cbb60 | ||
|
|
c7b16d6ddd | ||
|
|
d068f07c4b | ||
|
|
d9d8d44013 | ||
|
|
335fa4e21d | ||
|
|
29bb16ba1a | ||
|
|
4d57e679bf | ||
|
|
7a36d988ba | ||
|
|
56eba166f7 | ||
|
|
7f84a5eadf | ||
|
|
d8c3d8bb02 | ||
|
|
62ab8539c6 | ||
|
|
42533bf63c | ||
|
|
e6009c2d92 | ||
|
|
fb76ecd512 | ||
|
|
9be4a4e180 | ||
|
|
d7b42acced | ||
|
|
1003101976 | ||
|
|
c29d076f26 | ||
|
|
757684f383 | ||
|
|
4321e15323 | ||
|
|
7389a98f8b | ||
|
|
447b00c3cd | ||
|
|
305becec38 | ||
|
|
12ba219773 | ||
|
|
063afff53b | ||
|
|
0093c97873 | ||
|
|
a29753ef86 | ||
|
|
02b72cbce7 | ||
|
|
e6e6327165 | ||
|
|
a343c5d1e2 | ||
|
|
44d1920e2b | ||
|
|
5ad75aa465 | ||
|
|
b5e44100a2 | ||
|
|
f6a57b788a | ||
|
|
d00b2cf160 | ||
|
|
c7c6595942 | ||
|
|
d715c4fa54 | ||
|
|
1480fcc8c1 | ||
|
|
c22f04bdba | ||
|
|
96fdd62821 | ||
|
|
a7d42ba561 | ||
|
|
5645bfdee3 | ||
|
|
82b4e14194 | ||
|
|
85a42c9611 | ||
|
|
40a0409f7e | ||
|
|
4d02b6fcac | ||
|
|
b551f76c9c | ||
|
|
4134cd4bb1 | ||
|
|
32f0fe5f3d | ||
|
|
d8c8b7f727 | ||
|
|
620a01280a | ||
|
|
16bf8e880e | ||
|
|
7a95aabb8d | ||
|
|
aec83de635 | ||
|
|
30ada80b0b | ||
|
|
6def1d8cfc | ||
|
|
24f7066f6d | ||
|
|
bbf0da86f2 | ||
|
|
d70cf3d216 | ||
|
|
eb27e9e620 | ||
|
|
29facb35f3 | ||
|
|
65975124f4 | ||
|
|
d1dc3e470f | ||
|
|
96428938b9 | ||
|
|
ddfe68699b | ||
|
|
852af31e10 | ||
|
|
b70c47a402 | ||
|
|
65b3bbdfd8 | ||
|
|
5dede62b0e | ||
|
|
b8a8c412ca | ||
|
|
0cfac796d8 | ||
|
|
5debdbed94 | ||
|
|
b3d6830814 | ||
|
|
cfdaf0e11f | ||
|
|
2f969ff05d | ||
|
|
8be04781a5 | ||
|
|
e0259fe20c | ||
|
|
8453660872 | ||
|
|
b0d343109b | ||
|
|
ab8d327d18 | ||
|
|
1552bcca70 | ||
|
|
978e05e77f | ||
|
|
f40ab91c54 | ||
|
|
627e948c59 | ||
|
|
1daf5d7d03 | ||
|
|
9e089b6e4d | ||
|
|
52a5b9bab4 | ||
|
|
08988962c1 | ||
|
|
5f5591ca42 | ||
|
|
509dbd870a | ||
|
|
f0732a2a12 | ||
|
|
fb7b0be305 | ||
|
|
f9b95c078a | ||
|
|
afa2a8a7e5 | ||
|
|
5a1ff61bc0 | ||
|
|
c141fd151a | ||
|
|
97c7098e14 | ||
|
|
a73c6dfd6b | ||
|
|
1b1c9d1699 | ||
|
|
6d6bfaaa7f | ||
|
|
d57bd74132 | ||
|
|
365d059b0c | ||
|
|
d5c6b6157c | ||
|
|
7253e8af2e | ||
|
|
0067a50cb0 | ||
|
|
8f53faec56 | ||
|
|
a14062909b | ||
|
|
acec70a69c | ||
|
|
470d1f930d | ||
|
|
63f788c16e | ||
|
|
6eac64beab | ||
|
|
37d5841d1c | ||
|
|
1ec37a43f6 | ||
|
|
022249950e | ||
|
|
8cb42dee36 | ||
|
|
a588df0f0d | ||
|
|
99cf12c2b5 | ||
|
|
6d607e697d | ||
|
|
bc0aa86381 | ||
|
|
e2a4d2f3e5 | ||
|
|
9289fb3467 | ||
|
|
871e04a8ba | ||
|
|
63871ccd17 | ||
|
|
6c43bcbf8d | ||
|
|
959812300b | ||
|
|
23858b1b6a | ||
|
|
0e391ef26d | ||
|
|
4364c0eb07 | ||
|
|
22c4c7110d | ||
|
|
c7380e30fd | ||
|
|
c2968a7a33 | ||
|
|
3423718ac1 | ||
|
|
ad41a99d10 | ||
|
|
65160e8c14 | ||
|
|
d2e05f84b2 | ||
|
|
f7ce5cefd5 | ||
|
|
6b017f917e | ||
|
|
ba99d41972 | ||
|
|
126d8fbab9 | ||
|
|
0e14445685 | ||
|
|
4a3b91eeab | ||
|
|
76200dcd1f | ||
|
|
448c2e0239 | ||
|
|
a595d793d6 | ||
|
|
82f1b71c26 | ||
|
|
e6fec6d9bf | ||
|
|
7893814d03 | ||
|
|
c11ffc750d | ||
|
|
2f9349fb40 | ||
|
|
7bbb039da1 | ||
|
|
f80d2445b0 | ||
|
|
a4a3016c05 | ||
|
|
9f2b1e423f | ||
|
|
627ca21a56 | ||
|
|
1e788be509 | ||
|
|
00f31efc3d | ||
|
|
0e80bda371 | ||
|
|
03ef441f1b | ||
|
|
59021281e9 | ||
|
|
be1b345fbb | ||
|
|
9ba7f8fd7a | ||
|
|
c1c78585ba | ||
|
|
f5f183e7ce | ||
|
|
df175f7198 | ||
|
|
48ac39336c | ||
|
|
724df8c59e | ||
|
|
60a19494de | ||
|
|
6c4e2b4fdb | ||
|
|
74bbd19d85 | ||
|
|
81ada96450 | ||
|
|
0e44f1ac08 | ||
|
|
4ff6e8c6cd | ||
|
|
56dede2679 | ||
|
|
a9f2d62d30 | ||
|
|
988554dc5f | ||
|
|
86b4c9a18c | ||
|
|
fd4f96899a | ||
|
|
930f411549 | ||
|
|
3be9e17ecc | ||
|
|
d1d5ad91bd | ||
|
|
2d07a3ca4c | ||
|
|
ef8d1f021b | ||
|
|
d222f022aa | ||
|
|
84905544a3 | ||
|
|
751bf80338 | ||
|
|
4c3569c001 | ||
|
|
b582617535 | ||
|
|
7b017eb7f7 | ||
|
|
3b56427552 | ||
|
|
d95ba09e00 | ||
|
|
a26772055a | ||
|
|
042f653798 | ||
|
|
62075df80d | ||
|
|
f2a0670861 | ||
|
|
b0ca95807c | ||
|
|
2baecc90ff | ||
|
|
7dde0d0599 | ||
|
|
3d5df5938d | ||
|
|
746327553e | ||
|
|
fb748e78e8 | ||
|
|
23e8c84fa3 | ||
|
|
c52aa7d998 | ||
|
|
6473b076d5 | ||
|
|
e2a3d78bc0 | ||
|
|
e514b1b9e1 | ||
|
|
991645c957 | ||
|
|
ff094c2a2b | ||
|
|
efdef8ff1c | ||
|
|
aa048d7bdc | ||
|
|
ac22aa3f84 | ||
|
|
3b99ceecb2 | ||
|
|
3af8702d48 | ||
|
|
ac634a4ab2 | ||
|
|
da494043c8 | ||
|
|
220535e72a | ||
|
|
d816345560 | ||
|
|
e05f258fb3 | ||
|
|
a92294dba8 | ||
|
|
bb8540828e | ||
|
|
ea59a196bb | ||
|
|
bfdfa33b62 | ||
|
|
ff0bf30eac | ||
|
|
400054ae58 | ||
|
|
233dd993e2 | ||
|
|
d09c63ca7d | ||
|
|
a4cc514b9a | ||
|
|
d49ec34f0a | ||
|
|
598c0404a2 | ||
|
|
fc391a9d8b | ||
|
|
cf4e754aed | ||
|
|
1da2729251 | ||
|
|
ed233c1cb6 | ||
|
|
ed7804e0f3 | ||
|
|
a1bd8eea96 | ||
|
|
288309055a | ||
|
|
7515510880 | ||
|
|
1a38ed7b73 | ||
|
|
b4171eddc1 | ||
|
|
bc3d216915 | ||
|
|
4cbcdef9c2 | ||
|
|
978df8fb62 | ||
|
|
d812a1fc6c | ||
|
|
080c9c8fce | ||
|
|
fb5b166a37 | ||
|
|
457ad42565 | ||
|
|
8ca1acb172 | ||
|
|
b1bd1513b5 | ||
|
|
63d796dcd4 | ||
|
|
8e520a6368 | ||
|
|
e037096914 | ||
|
|
543337ac8b | ||
|
|
d60d765008 | ||
|
|
f32c2549d7 | ||
|
|
ff808476a2 | ||
|
|
cc389acbb9 | ||
|
|
5fe68d6027 | ||
|
|
3151419b92 | ||
|
|
22c5858ab6 | ||
|
|
519d2e2836 | ||
|
|
189eb4c309 | ||
|
|
481c0e1bcd | ||
|
|
6efa1b268f | ||
|
|
e7793c8ef2 | ||
|
|
98b61a7845 | ||
|
|
5c66bbd809 | ||
|
|
7390d5c1ac | ||
|
|
4c5dfa093c | ||
|
|
ffa4189763 | ||
|
|
d0ea99bc4c | ||
|
|
baf7e2ea44 | ||
|
|
10e78757b9 | ||
|
|
a038bf775a | ||
|
|
f8d01a810c | ||
|
|
9c45075cd1 | ||
|
|
12b5e18f0a | ||
|
|
dc6a4b3c3a | ||
|
|
2303fd3c7b | ||
|
|
3836bd3cc1 | ||
|
|
8c40c08b57 | ||
|
|
0c83df7933 | ||
|
|
c3ea5f02e2 | ||
|
|
9a6c54ce7c | ||
|
|
ae619b8faf | ||
|
|
88161468e8 | ||
|
|
a580563be5 | ||
|
|
95cfe4557c | ||
|
|
5ac94f7110 | ||
|
|
090fffe356 | ||
|
|
2a942ff55c | ||
|
|
9b552fb857 | ||
|
|
ab15ce7e17 | ||
|
|
4bd9e25fc4 | ||
|
|
3bd723202c | ||
|
|
a5d1cada5c | ||
|
|
9fa1656b6d | ||
|
|
fb04103593 | ||
|
|
922f0b1188 | ||
|
|
ce5cd909e1 | ||
|
|
d26587140d | ||
|
|
37dbb1c5d9 | ||
|
|
a6995e0789 | ||
|
|
e0f9f626b2 | ||
|
|
71b9723e7e | ||
|
|
51a8f7bdcf | ||
|
|
fcc99dd4b3 | ||
|
|
5c061d5103 | ||
|
|
ce8222bf76 | ||
|
|
ee46bcff9f | ||
|
|
2afab19ff5 | ||
|
|
9b80c2f4c6 | ||
|
|
750c6401d2 | ||
|
|
18022a2a91 | ||
|
|
e7767bacbf | ||
|
|
36bda59824 | ||
|
|
c35928b458 | ||
|
|
e8f03ab405 | ||
|
|
38c087b7d5 | ||
|
|
8c25114c50 | ||
|
|
57dc40dfd3 | ||
|
|
c55b5a7092 | ||
|
|
aa6808a400 | ||
|
|
f645390c92 | ||
|
|
e87938746c | ||
|
|
f27a5554f2 | ||
|
|
08011d7ea1 | ||
|
|
4acfe5cfaa | ||
|
|
938323d930 | ||
|
|
94a1ec1ce8 | ||
|
|
0caae506f4 | ||
|
|
d55726eb1f | ||
|
|
7a7a194fde | ||
|
|
edcb465630 | ||
|
|
392e2f661b | ||
|
|
6b52c84b86 | ||
|
|
e2a121c1da | ||
|
|
fc7644366b | ||
|
|
2e1186a482 | ||
|
|
6d4aacb903 | ||
|
|
e494b1027b | ||
|
|
c011a92267 | ||
|
|
d9d5f94ea0 | ||
|
|
a8090ce2fb | ||
|
|
e3a0161ba6 | ||
|
|
248fe3eef1 | ||
|
|
0de8fb9633 | ||
|
|
c9d5b4e0f3 | ||
|
|
eb2e2b15d8 | ||
|
|
50c56e7271 | ||
|
|
b242ecf5ac | ||
|
|
fc1e0268b2 | ||
|
|
d984951b20 | ||
|
|
49253a6207 | ||
|
|
b3f181ec45 | ||
|
|
85d0c9cfd3 | ||
|
|
d64cfe2d46 | ||
|
|
4d3bc09e2b | ||
|
|
093654ea89 | ||
|
|
577a277e7c | ||
|
|
d524c312cf | ||
|
|
6cfdde30ea | ||
|
|
835850e686 | ||
|
|
cd98e722b9 | ||
|
|
ed9c0966ef | ||
|
|
0166b6f9fb | ||
|
|
39b07ecfa0 | ||
|
|
0f9998ed16 | ||
|
|
29255fd80e | ||
|
|
930b48340d | ||
|
|
df0cf2a17b | ||
|
|
0d233677f6 | ||
|
|
6f6104792b | ||
|
|
62603643e8 | ||
|
|
879674238f | ||
|
|
73e58b213a | ||
|
|
bc88cafafb | ||
|
|
326ae16f6b | ||
|
|
2dc18da12d | ||
|
|
99f446679e | ||
|
|
f030376d35 | ||
|
|
2e92d928ae | ||
|
|
a35bdad805 | ||
|
|
ab3b49b13e | ||
|
|
d47ff3f60a | ||
|
|
ce99309b5d | ||
|
|
a90033e933 | ||
|
|
026ce2efac | ||
|
|
440325296c | ||
|
|
588c7aa571 | ||
|
|
c0dce561bd | ||
|
|
31e31bfe2d | ||
|
|
dc73336d25 | ||
|
|
5923e61d49 | ||
|
|
87ac9540fe | ||
|
|
1adaf6c5f6 | ||
|
|
7e7d644822 | ||
|
|
7195e257b8 | ||
|
|
fafc3e0312 | ||
|
|
2bc5baa156 | ||
|
|
a24e487c51 | ||
|
|
8c79c70520 | ||
|
|
d601e9aa82 | ||
|
|
4d5a918921 | ||
|
|
24e68e20cf | ||
|
|
73b1b44c35 | ||
|
|
b5ce6706d6 | ||
|
|
beea4453dd | ||
|
|
74a460a980 | ||
|
|
3bc2921f25 | ||
|
|
c89e9cdfd3 | ||
|
|
714929ef83 | ||
|
|
e472b0b2ac | ||
|
|
114a87ce60 | ||
|
|
f48ced67e4 | ||
|
|
7553d52c6a | ||
|
|
36f6329354 | ||
|
|
7476e2112b | ||
|
|
283298e5ff | ||
|
|
337435b128 | ||
|
|
50623236e5 | ||
|
|
c621f04d16 | ||
|
|
28cbd910dd | ||
|
|
3a91e0cff0 | ||
|
|
b6bc4674f4 | ||
|
|
f03a7a404a | ||
|
|
d8bdb5cbd1 | ||
|
|
f90083249f | ||
|
|
eb9538ec7b | ||
|
|
bfe642c3ed | ||
|
|
5b1a7f5271 | ||
|
|
b246c08217 | ||
|
|
a8e0d1416e | ||
|
|
6162678e02 |
+12
-3
@@ -3,7 +3,16 @@ create_2_Open_Interface_Spec.txt
|
|||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
logs
|
||||||
.node_modules
|
node_modules/
|
||||||
.pio
|
.pio
|
||||||
.vscode
|
.vscode/
|
||||||
|
config.h
|
||||||
|
robots.json
|
||||||
|
roverd-dummy
|
||||||
|
server/config.yaml
|
||||||
|
server/package-lock.json
|
||||||
|
package-lock.json
|
||||||
|
server/data/discord-guilds.json
|
||||||
|
server/data/community-goal.json
|
||||||
|
server/data
|
||||||
|
|||||||
@@ -1,14 +1,89 @@
|
|||||||
# Multi Roomba Rover
|
# Multi Roomba Rover
|
||||||
|
|
||||||
a remake of my RoombaRover project with a decentralized and embedded approach
|
a remake of my RoombaRover project with a decentralized and embedded approach
|
||||||
|
supports multiple roombas
|
||||||
|
|
||||||
on each roomba:
|
on each roomba:
|
||||||
- an esp32
|
- a raspberry pi zero 2 w
|
||||||
- a level shifter
|
- a raspberry pi camera
|
||||||
- DONT FORGET THE BRC PIN PULSE
|
- roomba's serial port hooked up to the built in serial port on the raspberry pi
|
||||||
- a power supply
|
|
||||||
- an openIPC camera
|
pi provisioning:
|
||||||
- USB wifi card
|
- enable serial port
|
||||||
- microphone
|
- disable wifi powersave
|
||||||
- speaker
|
- disable bluetooth
|
||||||
- MAYBE a master relay which can be turned off programatically to save the roomba from discharging. based on battery voltage plus urgent battery #?
|
|
||||||
|
## Repo layout
|
||||||
|
|
||||||
|
- `pi/roverd`: tiny Go daemon that bridges the Create 2 serial port, BRC pin, and the control server via WebSockets.
|
||||||
|
- `server`: Node.js process that terminates rover sockets, relays commands to/from the Socket.IO UI, and serves `public/`.
|
||||||
|
- `pi/bin` / `pi/systemd`: helper scripts + systemd units for the rover (roverd itself plus the SRT video publisher that streams camera feeds to the server).
|
||||||
|
- `docs/pi-deployment.md`: per-rover build + install instructions (cross-compiling on Fedora 43, deploying roverd + mediaMTX).
|
||||||
|
|
||||||
|
## Quick start
|
||||||
|
|
||||||
|
```bash
|
||||||
|
# build the Pi agent (armv7)
|
||||||
|
cd pi/roverd
|
||||||
|
mkdir -p ../../dist
|
||||||
|
make pi-build
|
||||||
|
|
||||||
|
# start the server + UI
|
||||||
|
cd ../../server
|
||||||
|
npm install
|
||||||
|
npm run start
|
||||||
|
```
|
||||||
|
|
||||||
|
Need a fake rover or multi-rover testing without hardware? Use the dummy build:
|
||||||
|
|
||||||
|
```bash
|
||||||
|
cd pi/roverd
|
||||||
|
mkdir -p ../../dist
|
||||||
|
make dummy
|
||||||
|
./../../dist/roverd-dummy -config ./roverd.sample.yaml
|
||||||
|
```
|
||||||
|
|
||||||
|
The dummy binary connects to the Node server, emits simulated sensor frames, and logs every command it receives, so you can spin up as many virtual rovers as you’d like on your dev machine.
|
||||||
|
|
||||||
|
## Admin config & authentication
|
||||||
|
|
||||||
|
Before running the Node server, copy `server/config.example.yaml` to `server/config.yaml` and customize the admin records (password hashes, Discord IDs, lockdown permission). Those credentials are used by the driver UI’s login panel—only admins can toggle locks/modes, and lockdown admins retain access when the system enters lockdown mode. The spectator page (future) can set `role:set` to `spectator`, and the server enforces all permissions server-side so client tweaks can’t grant extra control.
|
||||||
|
|
||||||
|
Deploy a rover by copying the repo + `dist/roverd` to the Pi and running the helper (it installs roverd plus the SRT video publisher service):
|
||||||
|
|
||||||
|
```bash
|
||||||
|
cd ~/MultiRoombaRover
|
||||||
|
sudo ./pi/install_roverd.sh
|
||||||
|
```
|
||||||
|
|
||||||
|
Then point each rover's `/etc/roverd.yaml` at `ws://<server>:8080/rover`, set `name` to the rover’s ID, and (optionally) override `media.publishUrl` if your control server isn’t `192.168.0.86`. The video publisher service (`video-publisher.service`) captures the Pi camera with `rpicam-vid`/`libcamera-vid`, pipes the raw H264 into the stock FFmpeg binary, and publishes via SRT to the server’s mediaMTX instance at `srt://<server>:9000?streamid=#!::r=<name>,m=publish`. The installer now just pulls `libcamera-apps` + `ffmpeg` from apt (no custom build). Use the “Restart Camera” button if you enable media management so roverd can bounce the publisher service remotely.
|
||||||
|
Heads-up: the BRC pulser now uses libgpiod; make sure the `roverd` service account is in the `gpio` group (or otherwise allowed to access `/dev/gpiochip*`) and set `brc.gpioChip` if your hardware exposes a different chip name.
|
||||||
|
|
||||||
|
## Fedora server deployment
|
||||||
|
|
||||||
|
Run the installer from inside the `server/` directory after cloning the repo onto your Fedora 43 Server box:
|
||||||
|
|
||||||
|
```bash
|
||||||
|
cd ~/MultiRoombaRover/server
|
||||||
|
sudo ./install_server.sh
|
||||||
|
```
|
||||||
|
|
||||||
|
The script must be executed via `sudo` from the user that owns the repo. It will:
|
||||||
|
|
||||||
|
- install Node.js/npm plus curl/tar
|
||||||
|
- run `npm install --production`
|
||||||
|
- copy `config.example.yaml` to `config.yaml` if needed (edit the file afterwards for admins + `media.whepBaseUrl`)
|
||||||
|
- download mediaMTX v1.15.3 and drop it into `/usr/local/bin`
|
||||||
|
- write `/etc/mediamtx/mediamtx.yml` from `server/mediamtx/mediamtx.yml` (SRT ingest on :9000, open ingest, viewer auth webhook at `/mediamtx/auth`)
|
||||||
|
- create + enable `mediamtx.service` and `multirover.service`, both running as your repo user and pointing at the clone directly
|
||||||
|
|
||||||
|
Publishing rovers lives on a trusted network, so the shipped config (tracked at `server/mediamtx/mediamtx.yml`) skips HTTP auth for SRT ingest and whitelists any path that matches `rover-*`. The installer overwrites `/etc/mediamtx/mediamtx.yml` every time you run it—if you need to tweak ports or add TURN servers, edit the template in the repo and rerun `install_server.sh` so every box stays in sync automatically.
|
||||||
|
|
||||||
|
Once finished, update `server/config.yaml` with your admin passwords and `media.whepBaseUrl` (set it to the URL you expose publicly, e.g. `https://rover.otter.land/video`). If your proxy can’t rewrite paths, create the `/video` location there and add a custom nginx snippet to rewrite `/video/<rover>/whep` to `/<>/whep` before forwarding to mediaMTX. Restart `multirover.service` whenever you edit the config. To pull updates later, just `git pull`, re-run `npm install --production` inside `server/`, and restart the service—no need to rerun the installer.
|
||||||
|
|
||||||
|
Room cameras now use JPEG snapshots (4 fps) instead of WHEP. Each entry in `roomCameras` must include a `url` pointing at the snapshot endpoint; the server polls and relays frames over socket.io with the same access rules as before.
|
||||||
|
|
||||||
|
### Video handshake + diagnostics
|
||||||
|
|
||||||
|
- Every `video:request` returns `{ url, token }`. The browser posts the SDP offer to `url` and includes `Authorization: Basic base64(token:token)`. mediaMTX forwards the username (`token`) to `/mediamtx/auth`, which checks the socket’s permissions (driver assignment, admin/spectator role, lockdown state) and either returns 200 or 401—no query parameters are involved anymore.
|
||||||
|
- To see what mediaMTX is ingesting from the Pis, run `npm run check:media` (or `node scripts/checkMedia.js`). It hits `/v3/paths/list` and prints each rover’s `ready` state and byte counters so you can instantly spot publish issues.
|
||||||
|
|||||||
Vendored
+22
@@ -0,0 +1,22 @@
|
|||||||
|
#configuration for roverd
|
||||||
|
name: dummy1
|
||||||
|
serverUrl: ws://127.0.0.1:8080/rover
|
||||||
|
serial:
|
||||||
|
device: /dev/ttyS0
|
||||||
|
baud: 115200
|
||||||
|
brc:
|
||||||
|
gpioPin: 25
|
||||||
|
pulseEvery: 1m
|
||||||
|
pulseWidth: 1s
|
||||||
|
battery:
|
||||||
|
full: 2068
|
||||||
|
warn: 1700
|
||||||
|
urgent: 1650
|
||||||
|
maxWheelSpeed: 350
|
||||||
|
media:
|
||||||
|
manage: false
|
||||||
|
service: mediamtx.service
|
||||||
|
healthUrl: http://127.0.0.1:9997/v3/paths/list
|
||||||
|
healthInterval: 30s
|
||||||
|
nightVision:
|
||||||
|
enabled: false
|
||||||
Vendored
+22
@@ -0,0 +1,22 @@
|
|||||||
|
#configuration for roverd
|
||||||
|
name: dummy2
|
||||||
|
serverUrl: ws://127.0.0.1:8080/rover
|
||||||
|
serial:
|
||||||
|
device: /dev/ttyS0
|
||||||
|
baud: 115200
|
||||||
|
brc:
|
||||||
|
gpioPin: 25
|
||||||
|
pulseEvery: 1m
|
||||||
|
pulseWidth: 1s
|
||||||
|
battery:
|
||||||
|
full: 2068
|
||||||
|
warn: 1700
|
||||||
|
urgent: 1650
|
||||||
|
maxWheelSpeed: 350
|
||||||
|
media:
|
||||||
|
manage: false
|
||||||
|
service: mediamtx.service
|
||||||
|
healthUrl: http://127.0.0.1:9997/v3/paths/list
|
||||||
|
healthInterval: 30s
|
||||||
|
nightVision:
|
||||||
|
enabled: false
|
||||||
Vendored
+22
@@ -0,0 +1,22 @@
|
|||||||
|
#configuration for roverd
|
||||||
|
name: dummy3
|
||||||
|
serverUrl: ws://127.0.0.1:8080/rover
|
||||||
|
serial:
|
||||||
|
device: /dev/ttyS0
|
||||||
|
baud: 115200
|
||||||
|
brc:
|
||||||
|
gpioPin: 25
|
||||||
|
pulseEvery: 1m
|
||||||
|
pulseWidth: 1s
|
||||||
|
battery:
|
||||||
|
full: 2068
|
||||||
|
warn: 1700
|
||||||
|
urgent: 1650
|
||||||
|
maxWheelSpeed: 350
|
||||||
|
media:
|
||||||
|
manage: false
|
||||||
|
service: mediamtx.service
|
||||||
|
healthUrl: http://127.0.0.1:9997/v3/paths/list
|
||||||
|
healthInterval: 30s
|
||||||
|
nightVision:
|
||||||
|
enabled: false
|
||||||
BIN
Binary file not shown.
BIN
Binary file not shown.
@@ -0,0 +1,112 @@
|
|||||||
|
# Pi Deployment Guide
|
||||||
|
|
||||||
|
## Tooling
|
||||||
|
|
||||||
|
Fedora 43:
|
||||||
|
|
||||||
|
```bash
|
||||||
|
sudo dnf install golang libgpiod
|
||||||
|
```
|
||||||
|
|
||||||
|
Cross-compiling roverd for Pi Zero 2 W (ARMv7):
|
||||||
|
|
||||||
|
```bash
|
||||||
|
cd pi/roverd
|
||||||
|
mkdir -p ../../dist
|
||||||
|
make pi-build
|
||||||
|
```
|
||||||
|
|
||||||
|
The binary is placed in `dist/roverd` (relative to the repo root).
|
||||||
|
|
||||||
|
### Dummy rover build (for local testing)
|
||||||
|
|
||||||
|
If you need extra “virtual” rovers on your laptop/CI box, build the dummy binary:
|
||||||
|
|
||||||
|
```bash
|
||||||
|
cd pi/roverd
|
||||||
|
mkdir -p ../../dist
|
||||||
|
make dummy
|
||||||
|
```
|
||||||
|
|
||||||
|
Run the resulting `dist/roverd-dummy` on any machine; it will connect to the server, stream fake Group 100 sensor data, and log drive commands so you can test multi-rover features without additional hardware.
|
||||||
|
|
||||||
|
## Automated installation (recommended)
|
||||||
|
|
||||||
|
Install the Raspberry Pi camera helpers (Bookworm ships them):
|
||||||
|
|
||||||
|
```bash
|
||||||
|
sudo apt update
|
||||||
|
sudo apt install libcamera-apps
|
||||||
|
```
|
||||||
|
|
||||||
|
Once the binary (and repo) are on the Pi, run the helper script from the repo root:
|
||||||
|
|
||||||
|
```bash
|
||||||
|
cd ~/MultiRoombaRover
|
||||||
|
sudo ./pi/install_roverd.sh
|
||||||
|
```
|
||||||
|
|
||||||
|
What the script does:
|
||||||
|
|
||||||
|
- creates the `roverd` service account (dialout/gpio/video groups) if missing and installs `/usr/local/bin/roverd`
|
||||||
|
- copies `pi/roverd/roverd.sample.yaml` to `/etc/roverd.yaml` if the file is absent (existing configs are left untouched)
|
||||||
|
- installs/enables the `roverd.service` systemd unit, restarting it automatically when a config already exists
|
||||||
|
- installs `/usr/local/bin/video-publisher`, drops `video-publisher.service`, and enables it so video is published automatically on boot
|
||||||
|
|
||||||
|
Flags:
|
||||||
|
|
||||||
|
| Flag | Purpose |
|
||||||
|
|------|---------|
|
||||||
|
| `-b PATH` | use a different roverd binary (defaults to `dist/roverd`) |
|
||||||
|
| `-c PATH` | seed `/etc/roverd.yaml` from another template |
|
||||||
|
|
||||||
|
If the script installs the sample config, it will remind you to edit `/etc/roverd.yaml` before manually restarting the service: set `name`, `serverUrl`, serial device, BRC pin, battery thresholds, and optionally override `media.publishUrl`. When left blank, roverd automatically publishes to `srt://<server-host>:9000?streamid=#!::r=<name>,m=publish…` (the host comes from `serverUrl`). If your reverse proxy adds prefixes (like `/video/<name>`), use its rewrite options so the rover keeps publishing to plain `<name>`.
|
||||||
|
|
||||||
|
Re-run `pi/install_roverd.sh` any time you pull updates—the script overwrites the roverd + video-publisher binaries and drops the latest systemd units so the only configuration you ever touch manually is `/etc/roverd.yaml`. `roverd` rewrites `/var/lib/roverd/video.env` on startup, so no other files need editing.
|
||||||
|
## Manual installation
|
||||||
|
|
||||||
|
1. Copy the binary and config:
|
||||||
|
```bash
|
||||||
|
sudo useradd -r -s /usr/sbin/nologin roverd || true
|
||||||
|
sudo install -o roverd -g roverd -m 0755 dist/roverd /usr/local/bin/roverd
|
||||||
|
sudo install -o roverd -g roverd -m 0640 pi/roverd/roverd.sample.yaml /etc/roverd.yaml
|
||||||
|
```
|
||||||
|
Adjust `/etc/roverd.yaml` for each rover: `name`, `serverUrl` (e.g. `ws://control-server:8080/rover`), serial port path, battery thresholds, GPIO pin for BRC, and (if needed) the media `publishUrl` override. Otherwise, roverd derives `srt://<server-host>:9000?streamid=#!::r=<name>,m=publish&latency=20&mode=caller&transtype=live&pkt_size=1316` based on the `serverUrl`; keep proxy-only path prefixes out of this URL so every rover keeps the same simple stream name.
|
||||||
|
|
||||||
|
2. Install the systemd unit:
|
||||||
|
```bash
|
||||||
|
sudo install -m 0644 pi/systemd/roverd.service /etc/systemd/system/roverd.service
|
||||||
|
sudo systemctl daemon-reload
|
||||||
|
sudo systemctl enable --now roverd.service
|
||||||
|
```
|
||||||
|
|
||||||
|
`roverd` requires access to `/dev/ttyAMA0` and `/dev/gpiochip*`; keeping it under its own user ensures the rest of the system stays isolated—just make sure the account belongs to the `dialout` and `gpio` groups so it can reach the UART and libgpiod.
|
||||||
|
The video publisher needs read access to the camera devices (`/dev/media*`, `/dev/video*`), so the install script adds the `roverd` service account to the `video` group; if you created the user manually, make sure it belongs to `video`.
|
||||||
|
**BRC note:** configure `brc.gpioPin` (and `brc.gpioChip` if you’re not using `gpiochip0`) and ensure the `roverd` user has permission to toggle that line—no root privileges are required anymore.
|
||||||
|
If you set `media.manage: true` in `/etc/roverd.yaml`, make sure the `roverd` service account can invoke `systemctl <action> <media.service>` (the installer wires `video-publisher.service` to run as `roverd`, so no sudo tweaks are required unless you rename it).
|
||||||
|
|
||||||
|
## Video publisher details
|
||||||
|
|
||||||
|
The `video-publisher.service` unit runs `/usr/local/bin/video-publisher`, piping `rpicam-vid`/`libcamera-vid` straight into the stock FFmpeg package:
|
||||||
|
|
||||||
|
```bash
|
||||||
|
rpicam-vid (or libcamera-vid) --inline --timeout 0 --width=WIDTH --height=HEIGHT \
|
||||||
|
--framerate=FPS --bitrate=BITRATE --codec h264 --profile baseline --output - \
|
||||||
|
| ffmpeg -hide_banner -loglevel warning -fflags nobuffer \
|
||||||
|
-f h264 -i pipe:0 -c:v copy -an -flush_packets 1 -f mpegts $PUBLISH_URL
|
||||||
|
```
|
||||||
|
|
||||||
|
`/var/lib/roverd/video.env` carries all tunables (`PUBLISH_URL`, resolution, FPS, bitrate). `roverd` rewrites the file whenever you restart it or hit the “Restart Camera” button so the publisher always inherits the correct rover ID + bitrate knobs. Because we now use the distro FFmpeg build with SRT enabled, the installer is much faster—no custom compile steps.
|
||||||
|
Use `sudo systemctl status video-publisher` to watch logs; the unit auto-restarts whenever the connection drops or FFmpeg exits with an error.
|
||||||
|
|
||||||
|
## Server + UI
|
||||||
|
|
||||||
|
From the repo root:
|
||||||
|
|
||||||
|
```bash
|
||||||
|
cd server
|
||||||
|
npm install
|
||||||
|
npm run start
|
||||||
|
```
|
||||||
|
|
||||||
|
This launches the HTTP server (serving the barebones UI) and the rover WebSocket endpoint at `ws://<server>:8080/rover`. The UI expects `roverd` instances to send `hello` frames so it can populate the rover list. Use the mode buttons to emit Start/Safe/Full/Passive/Dock commands (they send the raw OI opcode bytes), tap the sensor toggle to request Group 100 streaming, and use WASD for drive testing; the UI emits Drive Direct commands ~8 times per second, so the rover sees them immediately.
|
||||||
@@ -1,26 +0,0 @@
|
|||||||
# notes and ideas
|
|
||||||
|
|
||||||
## frontend
|
|
||||||
|
|
||||||
each frontend module will have a JSON state update over socket.io
|
|
||||||
each module will export a function to update it's state
|
|
||||||
|
|
||||||
use these functions to update all modules on connect, from a UI init state event.
|
|
||||||
this event will be sent all of the module data at once, in JSON
|
|
||||||
|
|
||||||
split each JSON element to the UI element functions
|
|
||||||
|
|
||||||
one single system for loading and saving user settings
|
|
||||||
|
|
||||||
|
|
||||||
## esp-server comms
|
|
||||||
|
|
||||||
the ESP will report to the server when it connects:
|
|
||||||
|
|
||||||
rover's name
|
|
||||||
enable / disable for each motor
|
|
||||||
camera IP address
|
|
||||||
battery info (for different battery behaviors):
|
|
||||||
- full number
|
|
||||||
- warn number
|
|
||||||
- urgent number
|
|
||||||
@@ -0,0 +1,26 @@
|
|||||||
|
# Use the Google Voice HAT soundcard as the primary device by name (card id is "sndrpigooglevoi")
|
||||||
|
options snd_rpi_googlevoicehat_soundcard index=0
|
||||||
|
|
||||||
|
# Software playback volume (adjust with: amixer -c0 sset 'SoftMaster' 70%)
|
||||||
|
pcm.softvol {
|
||||||
|
type softvol
|
||||||
|
slave.pcm "plughw:0,0"
|
||||||
|
control {
|
||||||
|
name "SoftMaster"
|
||||||
|
card 0
|
||||||
|
}
|
||||||
|
min_dB -51.0
|
||||||
|
max_dB 0.0
|
||||||
|
}
|
||||||
|
|
||||||
|
# Defaults: playback through softvol, capture raw on the HAT
|
||||||
|
pcm.!default {
|
||||||
|
type asym
|
||||||
|
playback.pcm "softvol"
|
||||||
|
capture.pcm "hw:0,0"
|
||||||
|
}
|
||||||
|
|
||||||
|
ctl.!default {
|
||||||
|
type hw
|
||||||
|
card 0
|
||||||
|
}
|
||||||
@@ -0,0 +1,66 @@
|
|||||||
|
#!/usr/bin/env bash
|
||||||
|
set -euo pipefail
|
||||||
|
set +H
|
||||||
|
|
||||||
|
ENV_FILE="${VIDEO_ENV_FILE:-/var/lib/roverd/video.env}"
|
||||||
|
|
||||||
|
if [[ ! -f "$ENV_FILE" ]]; then
|
||||||
|
echo "Environment file ${ENV_FILE} missing; cannot publish audio" >&2
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
# shellcheck disable=SC1090
|
||||||
|
source "$ENV_FILE"
|
||||||
|
AUDIO_ENABLE="${AUDIO_ENABLE:-0}"
|
||||||
|
if [[ "${AUDIO_ENABLE}" -ne 1 ]]; then
|
||||||
|
echo "Audio capture disabled; skipping audio-only publisher" >&2
|
||||||
|
exit 0
|
||||||
|
fi
|
||||||
|
: "${AUDIO_PUBLISH_URL:?AUDIO_PUBLISH_URL not set in ${ENV_FILE}}"
|
||||||
|
|
||||||
|
AUDIO_DEVICE="${AUDIO_DEVICE:-hw:0,0}"
|
||||||
|
AUDIO_RATE="${AUDIO_RATE:-48000}"
|
||||||
|
AUDIO_CHANNELS="${AUDIO_CHANNELS:-2}"
|
||||||
|
|
||||||
|
if [[ -n "${FFMPEG_BIN:-}" ]]; then
|
||||||
|
FFMPEG_BIN_PATH="$FFMPEG_BIN"
|
||||||
|
elif command -v ffmpeg >/dev/null 2>&1; then
|
||||||
|
FFMPEG_BIN_PATH="$(command -v ffmpeg)"
|
||||||
|
else
|
||||||
|
echo "ffmpeg not found; install it via apt install ffmpeg." >&2
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
run_pipeline() {
|
||||||
|
arecord -D "${AUDIO_DEVICE}" -f S32_LE -c "${AUDIO_CHANNELS}" -r "${AUDIO_RATE}" -B 65536 -F 2048 -q -t raw \
|
||||||
|
| "${FFMPEG_BIN_PATH}" \
|
||||||
|
-hide_banner \
|
||||||
|
-loglevel warning \
|
||||||
|
-fflags nobuffer \
|
||||||
|
-rtbufsize 0 \
|
||||||
|
-thread_queue_size 4096 \
|
||||||
|
-f s32le \
|
||||||
|
-ar "${AUDIO_RATE}" \
|
||||||
|
-ac "${AUDIO_CHANNELS}" \
|
||||||
|
-i pipe:0 \
|
||||||
|
-af "aresample=16000,pan=mono|c0=0.5*FL+0.5*FR,volume=25dB" \
|
||||||
|
-c:a libopus \
|
||||||
|
-b:a 24000 \
|
||||||
|
-ar:a 16000 \
|
||||||
|
-ac:a 1 \
|
||||||
|
-application lowdelay \
|
||||||
|
-frame_duration 20 \
|
||||||
|
-compression_level 0 \
|
||||||
|
-f mpegts \
|
||||||
|
"${AUDIO_PUBLISH_URL}"
|
||||||
|
}
|
||||||
|
|
||||||
|
trap 'kill 0 2>/dev/null' EXIT INT TERM
|
||||||
|
|
||||||
|
while true; do
|
||||||
|
if run_pipeline; then
|
||||||
|
exit 0
|
||||||
|
fi
|
||||||
|
echo "Audio-only publisher exited arecord=${PIPESTATUS[0]} ffmpeg=${PIPESTATUS[1]}, restarting in 2s..." >&2
|
||||||
|
sleep 2
|
||||||
|
done
|
||||||
@@ -0,0 +1,144 @@
|
|||||||
|
#!/usr/bin/env bash
|
||||||
|
set -euo pipefail
|
||||||
|
|
||||||
|
# Keep history expansion off so values containing "!" are safe.
|
||||||
|
set +H
|
||||||
|
|
||||||
|
ENV_FILE="${VIDEO_ENV_FILE:-/var/lib/roverd/video.env}"
|
||||||
|
|
||||||
|
if [[ ! -f "$ENV_FILE" ]]; then
|
||||||
|
echo "Environment file ${ENV_FILE} missing; cannot publish" >&2
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
# Load KEY=VALUE pairs from ENV_FILE WITHOUT evaluating shell metacharacters.
|
||||||
|
# This makes URLs containing characters like '&' and '#!' safe without requiring quoting.
|
||||||
|
load_env_file() {
|
||||||
|
local content=""
|
||||||
|
|
||||||
|
if [[ -r "$ENV_FILE" ]]; then
|
||||||
|
content="$(cat "$ENV_FILE")"
|
||||||
|
elif command -v sudo >/dev/null 2>&1; then
|
||||||
|
# Try to read via sudo without prompting (useful when the service runs as an unprivileged user)
|
||||||
|
content="$(sudo -n cat "$ENV_FILE" 2>/dev/null || true)"
|
||||||
|
fi
|
||||||
|
|
||||||
|
if [[ -z "$content" ]]; then
|
||||||
|
echo "Cannot read ${ENV_FILE} (permission denied). Run as a user that can read it, or allow sudo -n for cat." >&2
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
local line key val
|
||||||
|
while IFS= read -r line || [[ -n "$line" ]]; do
|
||||||
|
# Skip blank lines and full-line comments.
|
||||||
|
[[ "$line" =~ ^[[:space:]]*$ ]] && continue
|
||||||
|
[[ "$line" =~ ^[[:space:]]*# ]] && continue
|
||||||
|
|
||||||
|
# Support optional leading 'export '
|
||||||
|
if [[ "$line" =~ ^[[:space:]]*export[[:space:]]+([A-Za-z_][A-Za-z0-9_]*)=(.*)$ ]]; then
|
||||||
|
key="${BASH_REMATCH[1]}"
|
||||||
|
val="${BASH_REMATCH[2]}"
|
||||||
|
elif [[ "$line" =~ ^[[:space:]]*([A-Za-z_][A-Za-z0-9_]*)=(.*)$ ]]; then
|
||||||
|
key="${BASH_REMATCH[1]}"
|
||||||
|
val="${BASH_REMATCH[2]}"
|
||||||
|
else
|
||||||
|
# Ignore anything that isn't a simple assignment.
|
||||||
|
continue
|
||||||
|
fi
|
||||||
|
|
||||||
|
# Trim leading/trailing whitespace in value.
|
||||||
|
val="${val#${val%%[![:space:]]*}}"
|
||||||
|
val="${val%${val##*[![:space:]]}}"
|
||||||
|
|
||||||
|
# If value is wrapped in matching single or double quotes, unwrap.
|
||||||
|
if [[ "$val" =~ ^\".*\"$ ]]; then
|
||||||
|
val="${val:1:${#val}-2}"
|
||||||
|
elif [[ "$val" =~ ^\'.*\'$ ]]; then
|
||||||
|
val="${val:1:${#val}-2}"
|
||||||
|
fi
|
||||||
|
|
||||||
|
# Assign without evaluation.
|
||||||
|
printf -v "$key" '%s' "$val"
|
||||||
|
export "$key"
|
||||||
|
done <<< "$content"
|
||||||
|
}
|
||||||
|
|
||||||
|
load_env_file
|
||||||
|
: "${PUBLISH_URL:?PUBLISH_URL not set in ${ENV_FILE}}"
|
||||||
|
|
||||||
|
# Defaults tuned for OV5647: use 4:3 output and force the common 2x2 binned full-FOV mode.
|
||||||
|
VIDEO_WIDTH="640"
|
||||||
|
VIDEO_HEIGHT="480"
|
||||||
|
VIDEO_FPS="30"
|
||||||
|
VIDEO_BITRATE="${VIDEO_BITRATE:-3000000}"
|
||||||
|
VIDEO_SENSOR_MODE="${VIDEO_SENSOR_MODE:-1296:972}"
|
||||||
|
|
||||||
|
# Flip the camera 180deg (supported by rpicam-vid/libcamera-vid)
|
||||||
|
FLIP_ARGS=(--rotation 180)
|
||||||
|
|
||||||
|
MODE_ARGS=()
|
||||||
|
if [[ -n "${VIDEO_SENSOR_MODE}" ]]; then
|
||||||
|
MODE_ARGS=(--mode "${VIDEO_SENSOR_MODE}")
|
||||||
|
fi
|
||||||
|
|
||||||
|
if [[ -n "${LIBCAMERA_BIN:-}" ]]; then
|
||||||
|
LIBCAMERA_BIN_PATH="$LIBCAMERA_BIN"
|
||||||
|
elif command -v rpicam-vid >/dev/null 2>&1; then
|
||||||
|
LIBCAMERA_BIN_PATH="$(command -v rpicam-vid)"
|
||||||
|
elif command -v libcamera-vid >/dev/null 2>&1; then
|
||||||
|
LIBCAMERA_BIN_PATH="$(command -v libcamera-vid)"
|
||||||
|
else
|
||||||
|
echo "Neither rpicam-vid nor libcamera-vid found; install libcamera-apps." >&2
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
if [[ -n "${FFMPEG_BIN:-}" ]]; then
|
||||||
|
FFMPEG_BIN_PATH="$FFMPEG_BIN"
|
||||||
|
elif command -v ffmpeg >/dev/null 2>&1; then
|
||||||
|
FFMPEG_BIN_PATH="$(command -v ffmpeg)"
|
||||||
|
else
|
||||||
|
echo "ffmpeg not found; install it via apt install ffmpeg." >&2
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
run_pipeline() {
|
||||||
|
"${LIBCAMERA_BIN_PATH}" \
|
||||||
|
--inline \
|
||||||
|
--timeout 0 \
|
||||||
|
"${MODE_ARGS[@]}" \
|
||||||
|
--width "${VIDEO_WIDTH}" \
|
||||||
|
--height "${VIDEO_HEIGHT}" \
|
||||||
|
"${FLIP_ARGS[@]}" \
|
||||||
|
--framerate "${VIDEO_FPS}" \
|
||||||
|
--bitrate "${VIDEO_BITRATE}" \
|
||||||
|
--codec h264 \
|
||||||
|
--profile baseline \
|
||||||
|
--denoise auto \
|
||||||
|
--nopreview \
|
||||||
|
--metering centre \
|
||||||
|
--ev 0.1 \
|
||||||
|
--awb auto \
|
||||||
|
--saturation 0.6 \
|
||||||
|
--brightness 0 \
|
||||||
|
--output - \
|
||||||
|
| "${FFMPEG_BIN_PATH}" \
|
||||||
|
-hide_banner \
|
||||||
|
-loglevel warning \
|
||||||
|
-fflags nobuffer \
|
||||||
|
-use_wallclock_as_timestamps 1 \
|
||||||
|
-f h264 \
|
||||||
|
-i pipe:0 \
|
||||||
|
-c:v copy \
|
||||||
|
-an \
|
||||||
|
-flush_packets 1 \
|
||||||
|
-f mpegts \
|
||||||
|
"${PUBLISH_URL}"
|
||||||
|
}
|
||||||
|
|
||||||
|
while true; do
|
||||||
|
if run_pipeline; then
|
||||||
|
exit 0
|
||||||
|
fi
|
||||||
|
echo "Video publisher exited, restarting in 2s..." >&2
|
||||||
|
sleep 2
|
||||||
|
done
|
||||||
Executable
+256
@@ -0,0 +1,256 @@
|
|||||||
|
#!/usr/bin/env bash
|
||||||
|
#
|
||||||
|
# Installer for the roverd agent on Raspberry Pi
|
||||||
|
|
||||||
|
set -euo pipefail
|
||||||
|
|
||||||
|
BINARY_SRC="dist/roverd"
|
||||||
|
CONFIG_SRC="pi/roverd/roverd.sample.yaml"
|
||||||
|
|
||||||
|
usage() {
|
||||||
|
cat <<'USAGE'
|
||||||
|
Usage: sudo ./pi/install_roverd.sh [options]
|
||||||
|
|
||||||
|
Options:
|
||||||
|
-b, --binary <path> Path to the roverd binary (default: dist/roverd)
|
||||||
|
-c, --config <path> Source config to install if /etc/roverd.yaml is missing
|
||||||
|
(default: pi/roverd/roverd.sample.yaml)
|
||||||
|
-h, --help Show this help text
|
||||||
|
|
||||||
|
The script must run from the repository root and as root (sudo). It will:
|
||||||
|
* create system users/groups if needed
|
||||||
|
* install /usr/local/bin/roverd and /etc/roverd.yaml
|
||||||
|
* install /usr/local/bin/video-publisher and its systemd unit
|
||||||
|
* enable roverd.service and video-publisher.service
|
||||||
|
USAGE
|
||||||
|
}
|
||||||
|
|
||||||
|
while [[ $# -gt 0 ]]; do
|
||||||
|
case "$1" in
|
||||||
|
-b|--binary)
|
||||||
|
BINARY_SRC="${2:-}"
|
||||||
|
shift 2
|
||||||
|
;;
|
||||||
|
-c|--config)
|
||||||
|
CONFIG_SRC="${2:-}"
|
||||||
|
shift 2
|
||||||
|
;;
|
||||||
|
-h|--help)
|
||||||
|
usage
|
||||||
|
exit 0
|
||||||
|
;;
|
||||||
|
*)
|
||||||
|
echo "Unknown option: $1" >&2
|
||||||
|
usage
|
||||||
|
exit 1
|
||||||
|
;;
|
||||||
|
esac
|
||||||
|
done
|
||||||
|
|
||||||
|
if [[ "${EUID}" -ne 0 ]]; then
|
||||||
|
echo "Please run as root (sudo)" >&2
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
if [[ ! -f "$BINARY_SRC" ]]; then
|
||||||
|
echo "Binary not found at $BINARY_SRC" >&2
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
if [[ ! -f "$CONFIG_SRC" ]]; then
|
||||||
|
echo "Config source not found at $CONFIG_SRC" >&2
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
ensure_user() {
|
||||||
|
local user="$1"
|
||||||
|
local groups="${2:-}"
|
||||||
|
if ! id -u "$user" >/dev/null 2>&1; then
|
||||||
|
if [[ -n "$groups" ]]; then
|
||||||
|
useradd -r -s /usr/sbin/nologin -G "$groups" "$user"
|
||||||
|
else
|
||||||
|
useradd -r -s /usr/sbin/nologin "$user"
|
||||||
|
fi
|
||||||
|
elif [[ -n "$groups" ]]; then
|
||||||
|
usermod -a -G "$groups" "$user"
|
||||||
|
fi
|
||||||
|
}
|
||||||
|
|
||||||
|
log() {
|
||||||
|
echo "[$(date -u +%Y-%m-%dT%H:%M:%SZ)] $*"
|
||||||
|
}
|
||||||
|
|
||||||
|
if ! command -v rpicam-vid >/dev/null 2>&1 && ! command -v libcamera-vid >/dev/null 2>&1; then
|
||||||
|
log "WARNING: neither rpicam-vid nor libcamera-vid found in PATH; install libcamera-apps."
|
||||||
|
fi
|
||||||
|
|
||||||
|
install_video_deps() {
|
||||||
|
if command -v ffmpeg >/dev/null 2>&1 && (command -v rpicam-vid >/dev/null 2>&1 || command -v libcamera-vid >/dev/null 2>&1); then
|
||||||
|
log "Video dependencies already installed; skipping apt install"
|
||||||
|
return
|
||||||
|
fi
|
||||||
|
log "Installing video dependencies (libcamera-apps, ffmpeg)..."
|
||||||
|
apt-get update
|
||||||
|
apt-get install -y --no-install-recommends libcamera-apps ffmpeg
|
||||||
|
}
|
||||||
|
|
||||||
|
find_boot_config() {
|
||||||
|
if [[ -f /boot/firmware/config.txt ]]; then
|
||||||
|
printf "/boot/firmware/config.txt"
|
||||||
|
return 0
|
||||||
|
fi
|
||||||
|
if [[ -f /boot/config.txt ]]; then
|
||||||
|
printf "/boot/config.txt"
|
||||||
|
return 0
|
||||||
|
fi
|
||||||
|
return 1
|
||||||
|
}
|
||||||
|
|
||||||
|
ensure_pwm_overlay() {
|
||||||
|
local boot_config
|
||||||
|
if ! boot_config="$(find_boot_config)"; then
|
||||||
|
log "WARNING: unable to locate /boot config.txt; please ensure dtoverlay=pwm-2chan is added manually for servo support"
|
||||||
|
return
|
||||||
|
fi
|
||||||
|
if grep -Eq '^\s*dtoverlay=pwm(-2chan)?' "$boot_config"; then
|
||||||
|
log "PWM overlay already present in $boot_config"
|
||||||
|
return
|
||||||
|
fi
|
||||||
|
local backup="${boot_config}.roverd.$(date +%Y%m%d%H%M%S).bak"
|
||||||
|
cp "$boot_config" "$backup"
|
||||||
|
{
|
||||||
|
echo ""
|
||||||
|
echo "# Added by roverd installer to expose PWM hardware for camera servo control on GPIO12/13 (leaves GPIO18/19 free for I2S)"
|
||||||
|
echo "dtoverlay=pwm-2chan,pin=12,func=4,pin2=13,func2=4"
|
||||||
|
} >> "$boot_config"
|
||||||
|
log "Enabled dtoverlay=pwm-2chan on GPIO12/13 in $boot_config (backup at $backup). Reboot required for changes to apply."
|
||||||
|
}
|
||||||
|
|
||||||
|
ensure_user roverd "dialout,gpio,video,render,audio"
|
||||||
|
install -o roverd -g roverd -m 0755 "$BINARY_SRC" /usr/local/bin/roverd
|
||||||
|
log "Installed roverd binary"
|
||||||
|
|
||||||
|
CONFIG_DEST="/etc/roverd.yaml"
|
||||||
|
CONFIG_EXISTS=0
|
||||||
|
if [[ -f "$CONFIG_DEST" ]]; then
|
||||||
|
CONFIG_EXISTS=1
|
||||||
|
log "Existing $CONFIG_DEST found; leaving it in place"
|
||||||
|
else
|
||||||
|
install -D -o roverd -g roverd -m 0640 "$CONFIG_SRC" "$CONFIG_DEST"
|
||||||
|
log "Installed sample config to $CONFIG_DEST (edit before starting service)"
|
||||||
|
fi
|
||||||
|
|
||||||
|
install -m 0644 pi/systemd/roverd.service /etc/systemd/system/roverd.service
|
||||||
|
log "Installed roverd systemd unit"
|
||||||
|
|
||||||
|
install_video_deps
|
||||||
|
ensure_pwm_overlay
|
||||||
|
|
||||||
|
# Enable Google AIY v1 sound card, ALSA defaults, and TTS engines
|
||||||
|
install_audio_support() {
|
||||||
|
local boot_config
|
||||||
|
if ! boot_config="$(find_boot_config)"; then
|
||||||
|
log "WARNING: unable to locate /boot config.txt; please enable googlevoicehat-soundcard overlay manually"
|
||||||
|
else
|
||||||
|
# Ensure onboard audio is disabled (prevents card index flapping)
|
||||||
|
if grep -Eq '^\s*dtparam=audio=on\b' "$boot_config"; then
|
||||||
|
log "Disabling onboard audio (dtparam=audio=on -> off) in $boot_config"
|
||||||
|
sed -i 's/^\s*dtparam=audio=on\b/# roverd disabled onboard audio\ndtparam=audio=off/' "$boot_config"
|
||||||
|
fi
|
||||||
|
if ! grep -Eq '^\s*dtparam=audio=off\b' "$boot_config"; then
|
||||||
|
log "Adding dtparam=audio=off to $boot_config"
|
||||||
|
echo "dtparam=audio=off" >> "$boot_config"
|
||||||
|
fi
|
||||||
|
if ! grep -Eq '^\s*dtparam=i2s=on\b' "$boot_config"; then
|
||||||
|
log "Adding dtparam=i2s=on to $boot_config"
|
||||||
|
echo "dtparam=i2s=on" >> "$boot_config"
|
||||||
|
fi
|
||||||
|
if ! grep -Eq '^\s*dtoverlay=googlevoicehat-soundcard\b' "$boot_config"; then
|
||||||
|
local backup="${boot_config}.roverd.$(date +%Y%m%d%H%M%S).bak"
|
||||||
|
cp "$boot_config" "$backup"
|
||||||
|
{
|
||||||
|
echo ""
|
||||||
|
echo "# Added by roverd installer to enable Google AIY v1 sound card"
|
||||||
|
echo "dtoverlay=googlevoicehat-soundcard"
|
||||||
|
} >> "$boot_config"
|
||||||
|
log "Enabled googlevoicehat-soundcard overlay in $boot_config (backup at $backup). Reboot required."
|
||||||
|
else
|
||||||
|
log "googlevoicehat-soundcard overlay already present in $boot_config"
|
||||||
|
fi
|
||||||
|
fi
|
||||||
|
if [[ -f pi/asound.conf ]]; then
|
||||||
|
install -m 0644 pi/asound.conf /etc/asound.conf
|
||||||
|
log "Installed ALSA config to /etc/asound.conf"
|
||||||
|
alsa_reload_notice=1
|
||||||
|
else
|
||||||
|
log "WARNING: pi/asound.conf missing; skipping ALSA config install"
|
||||||
|
fi
|
||||||
|
|
||||||
|
if [[ "${alsa_reload_notice:-0}" -eq 1 ]]; then
|
||||||
|
log "ALSA config updated; reboot recommended for overlay + audio changes"
|
||||||
|
fi
|
||||||
|
|
||||||
|
log "Installing TTS/audio packages (flite, espeak)..."
|
||||||
|
# check for flite and espeak before installing, and then install them if either is missing
|
||||||
|
if command -v flite >/dev/null 2>&1 && command -v espeak >/dev/null 2>&1; then
|
||||||
|
log "TTS packages flite and espeak already installed; skipping apt install"
|
||||||
|
return
|
||||||
|
fi
|
||||||
|
|
||||||
|
apt-get update
|
||||||
|
apt-get install -y --no-install-recommends flite espeak
|
||||||
|
}
|
||||||
|
|
||||||
|
# Install video publisher assets
|
||||||
|
install -D -o root -g root -m 0755 pi/bin/video-publisher.sh /usr/local/bin/video-publisher
|
||||||
|
log "Installed video-publisher helper"
|
||||||
|
install -m 0644 pi/systemd/video-publisher.service /etc/systemd/system/video-publisher.service
|
||||||
|
log "Installed video-publisher systemd unit"
|
||||||
|
# Install audio-only publisher assets
|
||||||
|
install -D -o root -g root -m 0755 pi/bin/audio-only-publisher.sh /usr/local/bin/audio-only-publisher
|
||||||
|
install -m 0644 pi/systemd/audio-only-publisher.service /etc/systemd/system/audio-only-publisher.service
|
||||||
|
log "Installed audio-only publisher helper + systemd unit"
|
||||||
|
install -d -o roverd -g roverd /var/lib/roverd
|
||||||
|
cat > /var/lib/roverd/video.env <<'ENV'
|
||||||
|
# Managed by roverd; placeholder values will be overwritten at runtime.
|
||||||
|
PUBLISH_URL=srt://192.168.0.86:9000?streamid=#!::r=CHANGE_ME,m=publish&latency=10&mode=caller&transtype=live&pkt_size=1316
|
||||||
|
AUDIO_PUBLISH_URL=srt://192.168.0.86:9000?streamid=#!::r=CHANGE_ME-audio,m=publish&latency=10&mode=caller&transtype=live&pkt_size=1316
|
||||||
|
VIDEO_BITRATE=2000000
|
||||||
|
AUDIO_ENABLE=0
|
||||||
|
AUDIO_DEVICE=hw:0,0
|
||||||
|
AUDIO_RATE=48000
|
||||||
|
AUDIO_CHANNELS=2
|
||||||
|
ENV
|
||||||
|
chown roverd:roverd /var/lib/roverd/video.env
|
||||||
|
chmod 0640 /var/lib/roverd/video.env
|
||||||
|
# Create persistent audio FIFO for capture -> publisher
|
||||||
|
FIFO_PATH="/var/lib/roverd/audio.pcm"
|
||||||
|
if [[ -p "$FIFO_PATH" ]]; then
|
||||||
|
chown roverd:audio "$FIFO_PATH"
|
||||||
|
chmod 0660 "$FIFO_PATH"
|
||||||
|
else
|
||||||
|
rm -f "$FIFO_PATH"
|
||||||
|
mkfifo "$FIFO_PATH"
|
||||||
|
chown roverd:audio "$FIFO_PATH"
|
||||||
|
chmod 0660 "$FIFO_PATH"
|
||||||
|
fi
|
||||||
|
# Ensure ALSA config is in place for rovermic device
|
||||||
|
install -m 0644 pi/asound.conf /etc/asound.conf
|
||||||
|
log "Installed ALSA config (/etc/asound.conf)"
|
||||||
|
|
||||||
|
install_audio_support
|
||||||
|
|
||||||
|
systemctl daemon-reload
|
||||||
|
systemctl enable roverd.service
|
||||||
|
systemctl enable video-publisher.service
|
||||||
|
systemctl enable audio-only-publisher.service
|
||||||
|
if [[ $CONFIG_EXISTS -eq 1 ]]; then
|
||||||
|
systemctl restart roverd.service
|
||||||
|
systemctl restart video-publisher.service
|
||||||
|
systemctl restart audio-only-publisher.service
|
||||||
|
log "Restarted roverd + video/audio publishers"
|
||||||
|
else
|
||||||
|
log "Skipped auto-start because config is the sample; edit $CONFIG_DEST then run: sudo systemctl restart roverd video-publisher audio-only-publisher"
|
||||||
|
fi
|
||||||
|
|
||||||
|
log "Install complete"
|
||||||
@@ -0,0 +1,19 @@
|
|||||||
|
BIN_DIR ?= ../../dist
|
||||||
|
GOOS ?= linux
|
||||||
|
GOARCH ?= arm
|
||||||
|
GOARM ?= 6
|
||||||
|
|
||||||
|
.PHONY: build pi-build dummy clean
|
||||||
|
|
||||||
|
build:
|
||||||
|
go build -o $(BIN_DIR)/roverd ./cmd/roverd
|
||||||
|
|
||||||
|
pi-build:
|
||||||
|
GOOS=$(GOOS) GOARCH=$(GOARCH) GOARM=$(GOARM) go build -trimpath -ldflags="-s -w" -o $(BIN_DIR)/roverd ./cmd/roverd
|
||||||
|
GOOS=$(GOOS) GOARCH=$(GOARCH) GOARM=$(GOARM) go build -trimpath -ldflags="-s -w" -o $(BIN_DIR)/servoverifier ./cmd/servoverifier
|
||||||
|
|
||||||
|
dummy:
|
||||||
|
GOOS=linux GOARCH=amd64 go build -tags dummy -o $(BIN_DIR)/roverd-dummy ./cmd/roverd
|
||||||
|
|
||||||
|
clean:
|
||||||
|
rm -f $(BIN_DIR)/roverd $(BIN_DIR)/servoverifier
|
||||||
@@ -0,0 +1,109 @@
|
|||||||
|
package roverd
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"log"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
const (
|
||||||
|
autoChargeTimeout = 10 * time.Second
|
||||||
|
autoChargeCooldown = 0 * time.Minute
|
||||||
|
sourceHomeBase = 1 << 1
|
||||||
|
)
|
||||||
|
|
||||||
|
type AutoChargeController struct {
|
||||||
|
adapter *SerialAdapter
|
||||||
|
events chan<- RoverEvent
|
||||||
|
logger *log.Logger
|
||||||
|
timerStart time.Time
|
||||||
|
cooldownUntil time.Time
|
||||||
|
lastState byte
|
||||||
|
lastSources byte
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewAutoChargeController(adapter *SerialAdapter, events chan<- RoverEvent, logger *log.Logger) *AutoChargeController {
|
||||||
|
return &AutoChargeController{
|
||||||
|
adapter: adapter,
|
||||||
|
events: events,
|
||||||
|
logger: logger,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a *AutoChargeController) Run(ctx context.Context, samples <-chan SensorSample) {
|
||||||
|
for {
|
||||||
|
select {
|
||||||
|
case <-ctx.Done():
|
||||||
|
return
|
||||||
|
case sample := <-samples:
|
||||||
|
a.processSample(sample)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a *AutoChargeController) processSample(sample SensorSample) {
|
||||||
|
now := time.Now()
|
||||||
|
docked := sample.ChargeSources&sourceHomeBase != 0
|
||||||
|
charging := isCharging(sample.ChargingState)
|
||||||
|
|
||||||
|
if !docked || charging {
|
||||||
|
if !a.timerStart.IsZero() {
|
||||||
|
a.emitEvent("autoCharge.timerCleared", map[string]any{
|
||||||
|
"durationMs": time.Since(a.timerStart).Milliseconds(),
|
||||||
|
})
|
||||||
|
}
|
||||||
|
a.timerStart = time.Time{}
|
||||||
|
a.lastState = sample.ChargingState
|
||||||
|
a.lastSources = sample.ChargeSources
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// docked but not charging
|
||||||
|
if a.cooldownUntil.After(now) {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
if a.timerStart.IsZero() {
|
||||||
|
a.timerStart = now
|
||||||
|
a.emitEvent("autoCharge.timerStarted", map[string]any{
|
||||||
|
"chargingState": sample.ChargingState,
|
||||||
|
})
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
if now.Sub(a.timerStart) >= autoChargeTimeout {
|
||||||
|
if err := a.adapter.SeekDock(); err != nil {
|
||||||
|
a.emitEvent("autoCharge.seekDockError", map[string]any{"error": err.Error()})
|
||||||
|
} else {
|
||||||
|
a.emitEvent("autoCharge.seekDockIssued", map[string]any{
|
||||||
|
"waitingMs": autoChargeTimeout.Milliseconds(),
|
||||||
|
})
|
||||||
|
}
|
||||||
|
a.timerStart = time.Time{}
|
||||||
|
a.cooldownUntil = now.Add(autoChargeCooldown)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func isCharging(state byte) bool {
|
||||||
|
switch state {
|
||||||
|
case 1, 2, 3, 4:
|
||||||
|
return true
|
||||||
|
default:
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a *AutoChargeController) emitEvent(event string, data map[string]any) {
|
||||||
|
if a.events == nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
select {
|
||||||
|
case a.events <- RoverEvent{
|
||||||
|
Type: "event",
|
||||||
|
Event: event,
|
||||||
|
Ts: time.Now().UnixMilli(),
|
||||||
|
Data: data,
|
||||||
|
}:
|
||||||
|
default:
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,73 @@
|
|||||||
|
//go:build !dummy
|
||||||
|
|
||||||
|
package roverd
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"log"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
gpiocdev "github.com/warthog618/go-gpiocdev"
|
||||||
|
)
|
||||||
|
|
||||||
|
type BRCPulser struct {
|
||||||
|
cfg BRCConfig
|
||||||
|
logger *log.Logger
|
||||||
|
line *gpiocdev.Line
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewBRCPulser(cfg BRCConfig, logger *log.Logger) (*BRCPulser, error) {
|
||||||
|
chip := cfg.GPIOChip
|
||||||
|
if chip == "" {
|
||||||
|
chip = "gpiochip0"
|
||||||
|
}
|
||||||
|
|
||||||
|
line, err := gpiocdev.RequestLine(
|
||||||
|
chip,
|
||||||
|
cfg.GPIOPin,
|
||||||
|
gpiocdev.AsOutput(1),
|
||||||
|
gpiocdev.WithConsumer("roverd-brc"),
|
||||||
|
)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
return &BRCPulser{cfg: cfg, logger: logger, line: line}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (b *BRCPulser) Close() {
|
||||||
|
if b.line != nil {
|
||||||
|
_ = b.line.SetValue(1)
|
||||||
|
b.line.Close()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (b *BRCPulser) Start(ctx context.Context) {
|
||||||
|
go func() {
|
||||||
|
ticker := time.NewTicker(b.cfg.PulseEvery.Duration)
|
||||||
|
defer ticker.Stop()
|
||||||
|
|
||||||
|
for {
|
||||||
|
b.pulseOnce()
|
||||||
|
select {
|
||||||
|
case <-ctx.Done():
|
||||||
|
return
|
||||||
|
case <-ticker.C:
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (b *BRCPulser) pulseOnce() {
|
||||||
|
if b.line == nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if err := b.line.SetValue(0); err != nil {
|
||||||
|
b.logger.Printf("brc pulse low: %v", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
time.Sleep(b.cfg.PulseWidth.Duration)
|
||||||
|
if err := b.line.SetValue(1); err != nil {
|
||||||
|
b.logger.Printf("brc pulse high: %v", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,19 @@
|
|||||||
|
//go:build dummy
|
||||||
|
|
||||||
|
package roverd
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"log"
|
||||||
|
)
|
||||||
|
|
||||||
|
type BRCPulser struct{}
|
||||||
|
|
||||||
|
func NewBRCPulser(cfg BRCConfig, logger *log.Logger) (*BRCPulser, error) {
|
||||||
|
logger.Printf("[dummy] BRC configured on pin %d", cfg.GPIOPin)
|
||||||
|
return &BRCPulser{}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (b *BRCPulser) Close() {}
|
||||||
|
|
||||||
|
func (b *BRCPulser) Start(ctx context.Context) {}
|
||||||
@@ -0,0 +1,139 @@
|
|||||||
|
//go:build !dummy
|
||||||
|
|
||||||
|
package roverd
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"log"
|
||||||
|
"math"
|
||||||
|
"sync"
|
||||||
|
|
||||||
|
rpio "github.com/stianeikeland/go-rpio/v4"
|
||||||
|
)
|
||||||
|
|
||||||
|
type CameraServo struct {
|
||||||
|
cfg CameraServoConfig
|
||||||
|
logger *log.Logger
|
||||||
|
pin rpio.Pin
|
||||||
|
mu sync.Mutex
|
||||||
|
currentAngle float64
|
||||||
|
closed bool
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewCameraServo(cfg CameraServoConfig, logger *log.Logger) (*CameraServo, error) {
|
||||||
|
if !cfg.Enabled {
|
||||||
|
return nil, fmt.Errorf("camera servo disabled")
|
||||||
|
}
|
||||||
|
if err := rpio.Open(); err != nil {
|
||||||
|
return nil, fmt.Errorf("open gpio: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
pin := rpio.Pin(cfg.Pin)
|
||||||
|
pin.Mode(rpio.Pwm)
|
||||||
|
targetClock := cfg.FreqHz * cfg.CycleLen
|
||||||
|
pin.Freq(targetClock)
|
||||||
|
|
||||||
|
servo := &CameraServo{
|
||||||
|
cfg: cfg,
|
||||||
|
logger: logger,
|
||||||
|
pin: pin,
|
||||||
|
}
|
||||||
|
if err := servo.setAngleLocked(cfg.HomeAngle); err != nil {
|
||||||
|
rpio.Close()
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
logger.Printf("camera servo initialized on GPIO %d (%.1f..%.1f deg, %d..%d us, invert=%v)", cfg.Pin, cfg.MinAngle, cfg.MaxAngle, cfg.MinPulseUs, cfg.MaxPulseUs, cfg.Invert)
|
||||||
|
return servo, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *CameraServo) Close() {
|
||||||
|
s.mu.Lock()
|
||||||
|
defer s.mu.Unlock()
|
||||||
|
if s.closed {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
s.applyPulseLocked(s.angleToPulse(s.cfg.HomeAngle))
|
||||||
|
rpio.Close()
|
||||||
|
s.closed = true
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *CameraServo) SetAngle(angle float64) error {
|
||||||
|
s.mu.Lock()
|
||||||
|
defer s.mu.Unlock()
|
||||||
|
return s.setAngleLocked(angle)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *CameraServo) setAngleLocked(angle float64) error {
|
||||||
|
if s.closed {
|
||||||
|
return fmt.Errorf("servo closed")
|
||||||
|
}
|
||||||
|
clamped := clampFloat(angle, s.cfg.MinAngle, s.cfg.MaxAngle)
|
||||||
|
s.applyPulseLocked(s.angleToPulse(clamped))
|
||||||
|
s.currentAngle = clamped
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *CameraServo) Nudge(delta float64) error {
|
||||||
|
s.mu.Lock()
|
||||||
|
defer s.mu.Unlock()
|
||||||
|
if delta == 0 {
|
||||||
|
delta = s.cfg.NudgeDegrees
|
||||||
|
}
|
||||||
|
target := s.currentAngle + delta
|
||||||
|
return s.setAngleLocked(target)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *CameraServo) SetPulseWidth(micros int) error {
|
||||||
|
s.mu.Lock()
|
||||||
|
defer s.mu.Unlock()
|
||||||
|
if s.closed {
|
||||||
|
return fmt.Errorf("servo closed")
|
||||||
|
}
|
||||||
|
if !s.cfg.AllowRawPulse {
|
||||||
|
return fmt.Errorf("raw pulse commands disabled")
|
||||||
|
}
|
||||||
|
if micros <= 0 {
|
||||||
|
return fmt.Errorf("pulse width must be > 0")
|
||||||
|
}
|
||||||
|
s.applyPulseLocked(micros)
|
||||||
|
s.currentAngle = s.pulseToAngle(micros)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *CameraServo) CurrentAngle() float64 {
|
||||||
|
s.mu.Lock()
|
||||||
|
defer s.mu.Unlock()
|
||||||
|
return s.currentAngle
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *CameraServo) applyPulseLocked(micros int) {
|
||||||
|
micros = clampInt(micros, s.cfg.MinPulseUs, s.cfg.MaxPulseUs)
|
||||||
|
s.pin.DutyCycle(uint32(micros), uint32(s.cfg.CycleLen))
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *CameraServo) angleToPulse(angle float64) int {
|
||||||
|
totalRange := s.cfg.MaxAngle - s.cfg.MinAngle
|
||||||
|
if totalRange == 0 {
|
||||||
|
return s.cfg.MinPulseUs
|
||||||
|
}
|
||||||
|
norm := (angle - s.cfg.MinAngle) / totalRange
|
||||||
|
norm = math.Max(0, math.Min(1, norm))
|
||||||
|
if s.cfg.Invert {
|
||||||
|
norm = 1 - norm
|
||||||
|
}
|
||||||
|
pulseRange := s.cfg.MaxPulseUs - s.cfg.MinPulseUs
|
||||||
|
return s.cfg.MinPulseUs + int(math.Round(norm*float64(pulseRange)))
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *CameraServo) pulseToAngle(pulse int) float64 {
|
||||||
|
pulseRange := s.cfg.MaxPulseUs - s.cfg.MinPulseUs
|
||||||
|
if pulseRange == 0 {
|
||||||
|
return s.cfg.MinAngle
|
||||||
|
}
|
||||||
|
norm := float64(pulse-s.cfg.MinPulseUs) / float64(pulseRange)
|
||||||
|
norm = math.Max(0, math.Min(1, norm))
|
||||||
|
if s.cfg.Invert {
|
||||||
|
norm = 1 - norm
|
||||||
|
}
|
||||||
|
return s.cfg.MinAngle + norm*(s.cfg.MaxAngle-s.cfg.MinAngle)
|
||||||
|
}
|
||||||
@@ -0,0 +1,32 @@
|
|||||||
|
//go:build dummy
|
||||||
|
|
||||||
|
package roverd
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"log"
|
||||||
|
)
|
||||||
|
|
||||||
|
type CameraServo struct{}
|
||||||
|
|
||||||
|
func NewCameraServo(cfg CameraServoConfig, logger *log.Logger) (*CameraServo, error) {
|
||||||
|
return nil, fmt.Errorf("camera servo not supported in dummy build")
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *CameraServo) Close() {}
|
||||||
|
|
||||||
|
func (c *CameraServo) SetAngle(angle float64) error {
|
||||||
|
return fmt.Errorf("camera servo disabled")
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *CameraServo) Nudge(delta float64) error {
|
||||||
|
return fmt.Errorf("camera servo disabled")
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *CameraServo) SetPulseWidth(micros int) error {
|
||||||
|
return fmt.Errorf("camera servo disabled")
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *CameraServo) CurrentAngle() float64 {
|
||||||
|
return 0
|
||||||
|
}
|
||||||
@@ -0,0 +1,11 @@
|
|||||||
|
package roverd
|
||||||
|
|
||||||
|
func clampInt(value, min, max int) int {
|
||||||
|
if value < min {
|
||||||
|
return min
|
||||||
|
}
|
||||||
|
if value > max {
|
||||||
|
return max
|
||||||
|
}
|
||||||
|
return value
|
||||||
|
}
|
||||||
@@ -0,0 +1,102 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"flag"
|
||||||
|
"log"
|
||||||
|
"os"
|
||||||
|
"os/signal"
|
||||||
|
"syscall"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
roverd "multiroombarover/pi/roverd"
|
||||||
|
)
|
||||||
|
|
||||||
|
func main() {
|
||||||
|
var cfgPath string
|
||||||
|
flag.StringVar(&cfgPath, "config", "/etc/roverd.yaml", "path to roverd configuration file")
|
||||||
|
flag.Parse()
|
||||||
|
|
||||||
|
cfg, err := roverd.LoadConfig(cfgPath)
|
||||||
|
if err != nil {
|
||||||
|
log.Fatalf("load config: %v", err)
|
||||||
|
}
|
||||||
|
if err := roverd.UpdatePublisherEnv(cfg.Media, cfg.Audio); err != nil {
|
||||||
|
log.Fatalf("prepare media env: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
ctx, cancel := signal.NotifyContext(context.Background(), syscall.SIGINT, syscall.SIGTERM)
|
||||||
|
defer cancel()
|
||||||
|
|
||||||
|
logger := log.New(os.Stdout, "roverd: ", log.LstdFlags|log.Lmicroseconds|log.LUTC)
|
||||||
|
|
||||||
|
serialPort, err := roverd.OpenSerial(cfg.Serial)
|
||||||
|
if err != nil {
|
||||||
|
logger.Fatalf("open serial: %v", err)
|
||||||
|
}
|
||||||
|
defer serialPort.Close()
|
||||||
|
|
||||||
|
var pulser *roverd.BRCPulser
|
||||||
|
if cfg.BRC.Enabled() {
|
||||||
|
pulser, err = roverd.NewBRCPulser(cfg.BRC, logger)
|
||||||
|
if err != nil {
|
||||||
|
logger.Fatalf("init BRC pulser: %v", err)
|
||||||
|
}
|
||||||
|
defer pulser.Close()
|
||||||
|
pulser.Start(ctx)
|
||||||
|
}
|
||||||
|
|
||||||
|
sensorFrames := make(chan []byte, 8)
|
||||||
|
sensorSamples := make(chan roverd.SensorSample, 8)
|
||||||
|
eventStream := make(chan roverd.RoverEvent, 16)
|
||||||
|
|
||||||
|
streamer := roverd.NewSensorStreamer(serialPort, sensorFrames, sensorSamples, logger)
|
||||||
|
go streamer.Run(ctx)
|
||||||
|
|
||||||
|
adapter := roverd.NewSerialAdapter(serialPort, logger)
|
||||||
|
|
||||||
|
mediaSupervisor := roverd.NewMediaSupervisor(cfg.Media, cfg.Audio, logger)
|
||||||
|
if mediaSupervisor != nil {
|
||||||
|
mediaSupervisor.Start(ctx)
|
||||||
|
}
|
||||||
|
|
||||||
|
var cameraServo *roverd.CameraServo
|
||||||
|
if cfg.CameraServo.Enabled {
|
||||||
|
cameraServo, err = roverd.NewCameraServo(cfg.CameraServo, logger)
|
||||||
|
if err != nil {
|
||||||
|
logger.Fatalf("init camera servo: %v", err)
|
||||||
|
}
|
||||||
|
defer cameraServo.Close()
|
||||||
|
}
|
||||||
|
|
||||||
|
var nightVision *roverd.NightVisionLight
|
||||||
|
if cfg.NightVision.Enabled {
|
||||||
|
nightVision, err = roverd.NewNightVisionLight(cfg.NightVision, logger)
|
||||||
|
if err != nil {
|
||||||
|
logger.Fatalf("init night vision: %v", err)
|
||||||
|
}
|
||||||
|
defer nightVision.Close()
|
||||||
|
}
|
||||||
|
|
||||||
|
autoCharge := roverd.NewAutoChargeController(adapter, eventStream, logger)
|
||||||
|
go autoCharge.Run(ctx, sensorSamples)
|
||||||
|
|
||||||
|
client := roverd.NewWSClient(cfg, adapter, sensorFrames, eventStream, mediaSupervisor, cameraServo, nightVision, logger)
|
||||||
|
|
||||||
|
retryDelay := time.Second
|
||||||
|
for ctx.Err() == nil {
|
||||||
|
if err := client.Run(ctx); err != nil {
|
||||||
|
logger.Printf("websocket loop ended: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
select {
|
||||||
|
case <-ctx.Done():
|
||||||
|
return
|
||||||
|
case <-time.After(retryDelay):
|
||||||
|
}
|
||||||
|
|
||||||
|
if retryDelay < 30*time.Second {
|
||||||
|
retryDelay *= 2
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,95 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"flag"
|
||||||
|
"log"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
rpio "github.com/stianeikeland/go-rpio/v4"
|
||||||
|
)
|
||||||
|
|
||||||
|
func main() {
|
||||||
|
var (
|
||||||
|
pinNum = flag.Int("pin", 19, "BCM pin connected to the servo signal line")
|
||||||
|
freqHz = flag.Int("freq", 50, "Servo PWM frequency in Hz")
|
||||||
|
cycleLen = flag.Int("cycle", 20000, "PWM cycle length (counts per period)")
|
||||||
|
minPulse = flag.Int("min", 900, "Minimum pulse width in microseconds")
|
||||||
|
maxPulse = flag.Int("max", 2100, "Maximum pulse width in microseconds")
|
||||||
|
stepPulse = flag.Int("step", 100, "Pulse width increment in microseconds when sweeping")
|
||||||
|
sweeps = flag.Int("sweeps", 2, "How many full min→max→min sweeps to perform")
|
||||||
|
pause = flag.Duration("pause", 150*time.Millisecond, "Delay between pulse adjustments")
|
||||||
|
holdPulse = flag.Int("hold", 0, "Pulse width to hold before exiting (0 = midpoint of min/max)")
|
||||||
|
)
|
||||||
|
flag.Parse()
|
||||||
|
|
||||||
|
if *freqHz <= 0 || *cycleLen <= 0 {
|
||||||
|
log.Fatalf("invalid freq (%d) or cycle (%d)", *freqHz, *cycleLen)
|
||||||
|
}
|
||||||
|
if *minPulse <= 0 || *maxPulse <= 0 || *minPulse >= *maxPulse {
|
||||||
|
log.Fatalf("invalid min/max pulses (%d/%d)", *minPulse, *maxPulse)
|
||||||
|
}
|
||||||
|
if *stepPulse <= 0 {
|
||||||
|
log.Fatalf("step must be > 0 (got %d)", *stepPulse)
|
||||||
|
}
|
||||||
|
if *pause <= 0 {
|
||||||
|
log.Fatalf("pause must be > 0 (got %s)", pause)
|
||||||
|
}
|
||||||
|
if *sweeps < 0 {
|
||||||
|
log.Fatalf("sweeps must be >= 0 (got %d)", *sweeps)
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := rpio.Open(); err != nil {
|
||||||
|
log.Fatalf("open gpio: %v", err)
|
||||||
|
}
|
||||||
|
defer rpio.Close()
|
||||||
|
|
||||||
|
pin := rpio.Pin(*pinNum)
|
||||||
|
pin.Mode(rpio.Pwm)
|
||||||
|
|
||||||
|
targetClock := *freqHz * *cycleLen
|
||||||
|
pin.Freq(targetClock)
|
||||||
|
log.Printf("Configured PWM pin %d at %d Hz (clock=%d Hz, cycle=%d)", *pinNum, *freqHz, targetClock, *cycleLen)
|
||||||
|
|
||||||
|
setPulse := func(us int) {
|
||||||
|
clamped := clamp(us, *minPulse, *maxPulse)
|
||||||
|
pin.DutyCycle(uint32(clamped), uint32(*cycleLen))
|
||||||
|
log.Printf("pulse -> %dµs", clamped)
|
||||||
|
}
|
||||||
|
|
||||||
|
mid := (*minPulse + *maxPulse) / 2
|
||||||
|
setPulse(mid)
|
||||||
|
|
||||||
|
runSweep := func() {
|
||||||
|
for pulse := *minPulse; pulse <= *maxPulse; pulse += *stepPulse {
|
||||||
|
setPulse(pulse)
|
||||||
|
time.Sleep(*pause)
|
||||||
|
}
|
||||||
|
for pulse := *maxPulse - *stepPulse; pulse >= *minPulse; pulse -= *stepPulse {
|
||||||
|
setPulse(pulse)
|
||||||
|
time.Sleep(*pause)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
for i := 0; i < *sweeps; i++ {
|
||||||
|
log.Printf("Sweep %d/%d", i+1, *sweeps)
|
||||||
|
runSweep()
|
||||||
|
}
|
||||||
|
|
||||||
|
finalPulse := *holdPulse
|
||||||
|
if finalPulse <= 0 {
|
||||||
|
finalPulse = mid
|
||||||
|
}
|
||||||
|
setPulse(finalPulse)
|
||||||
|
log.Printf("Holding at %dµs", clamp(finalPulse, *minPulse, *maxPulse))
|
||||||
|
log.Print("Servo verification complete")
|
||||||
|
}
|
||||||
|
|
||||||
|
func clamp(value, min, max int) int {
|
||||||
|
if value < min {
|
||||||
|
return min
|
||||||
|
}
|
||||||
|
if value > max {
|
||||||
|
return max
|
||||||
|
}
|
||||||
|
return value
|
||||||
|
}
|
||||||
@@ -0,0 +1,87 @@
|
|||||||
|
package roverd
|
||||||
|
|
||||||
|
type helloMessage struct {
|
||||||
|
Type string `json:"type"`
|
||||||
|
Name string `json:"name"`
|
||||||
|
Battery BatteryConfig `json:"battery"`
|
||||||
|
MaxWheelSpeed int `json:"maxWheelSpeed"`
|
||||||
|
Media MediaConfig `json:"media"`
|
||||||
|
CameraServo CameraServoConfig `json:"cameraServo"`
|
||||||
|
Audio AudioConfig `json:"audio"`
|
||||||
|
NightVision NightVisionConfig `json:"nightVision"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type sensorMessage struct {
|
||||||
|
Type string `json:"type"`
|
||||||
|
Timestamp int64 `json:"ts"`
|
||||||
|
Data string `json:"data"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type inboundMessage struct {
|
||||||
|
Type string `json:"type"`
|
||||||
|
ID string `json:"id"`
|
||||||
|
DriveDirect *driveDirectPayload `json:"driveDirect,omitempty"`
|
||||||
|
MotorPWM *motorPWMPayload `json:"motorPwm,omitempty"`
|
||||||
|
Raw string `json:"raw,omitempty"`
|
||||||
|
SensorStream *sensorStreamPayload `json:"sensorStream,omitempty"`
|
||||||
|
Media *mediaCommand `json:"media,omitempty"`
|
||||||
|
Servo *servoPayload `json:"servo,omitempty"`
|
||||||
|
TTS *ttsPayload `json:"tts,omitempty"`
|
||||||
|
NightVision *nightVisionPayload `json:"nightVision,omitempty"`
|
||||||
|
Song *songPayload `json:"song,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type driveDirectPayload struct {
|
||||||
|
Left int `json:"left"`
|
||||||
|
Right int `json:"right"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type motorPWMPayload struct {
|
||||||
|
Main int `json:"main"`
|
||||||
|
Side int `json:"side"`
|
||||||
|
Vacuum int `json:"vacuum"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type sensorStreamPayload struct {
|
||||||
|
Enable bool `json:"enable"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type mediaCommand struct {
|
||||||
|
Action string `json:"action"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type servoPayload struct {
|
||||||
|
Angle *float64 `json:"angle,omitempty"`
|
||||||
|
Nudge *float64 `json:"nudge,omitempty"`
|
||||||
|
PulseUs *int `json:"pulseUs,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type ttsPayload struct {
|
||||||
|
Text string `json:"text"`
|
||||||
|
Engine string `json:"engine,omitempty"`
|
||||||
|
Voice string `json:"voice,omitempty"`
|
||||||
|
Pitch int `json:"pitch,omitempty"`
|
||||||
|
Speak bool `json:"speak,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type nightVisionPayload struct {
|
||||||
|
Action string `json:"action"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type songPayload struct {
|
||||||
|
Slot *int `json:"slot,omitempty"`
|
||||||
|
Notes []songNote `json:"notes"`
|
||||||
|
Loop bool `json:"loop,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type songNote struct {
|
||||||
|
Note int `json:"note"`
|
||||||
|
Duration int `json:"duration"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type ackMessage struct {
|
||||||
|
Type string `json:"type"`
|
||||||
|
ID string `json:"id"`
|
||||||
|
Status string `json:"status"`
|
||||||
|
Error string `json:"error,omitempty"`
|
||||||
|
}
|
||||||
@@ -0,0 +1,324 @@
|
|||||||
|
package roverd
|
||||||
|
|
||||||
|
import (
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
"net/url"
|
||||||
|
"os"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"gopkg.in/yaml.v3"
|
||||||
|
)
|
||||||
|
|
||||||
|
type SerialConfig struct {
|
||||||
|
Device string `yaml:"device"`
|
||||||
|
Baud int `yaml:"baud"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type Duration struct {
|
||||||
|
time.Duration
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *Duration) UnmarshalYAML(value *yaml.Node) error {
|
||||||
|
var raw string
|
||||||
|
if err := value.Decode(&raw); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
parsed, err := time.ParseDuration(raw)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
d.Duration = parsed
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d Duration) MarshalYAML() (interface{}, error) {
|
||||||
|
return d.Duration.String(), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
type BRCConfig struct {
|
||||||
|
GPIOPin int `yaml:"gpioPin"`
|
||||||
|
GPIOChip string `yaml:"gpioChip"`
|
||||||
|
PulseEvery Duration `yaml:"pulseEvery"`
|
||||||
|
PulseWidth Duration `yaml:"pulseWidth"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (b BRCConfig) Enabled() bool {
|
||||||
|
return b.GPIOPin >= 0
|
||||||
|
}
|
||||||
|
|
||||||
|
type BatteryConfig struct {
|
||||||
|
Full int `yaml:"full"`
|
||||||
|
Warn int `yaml:"warn"`
|
||||||
|
Urgent int `yaml:"urgent"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type AudioConfig struct {
|
||||||
|
CaptureEnabled bool `yaml:"captureEnabled" json:"captureEnabled"`
|
||||||
|
CaptureDevice string `yaml:"captureDevice" json:"captureDevice,omitempty"`
|
||||||
|
SampleRate int `yaml:"sampleRate" json:"sampleRate,omitempty"`
|
||||||
|
Channels int `yaml:"channels" json:"channels,omitempty"`
|
||||||
|
Bitrate int `yaml:"bitrate" json:"bitrate,omitempty"`
|
||||||
|
TTSEnabled bool `yaml:"ttsEnabled" json:"ttsEnabled"`
|
||||||
|
DefaultEngine string `yaml:"defaultEngine" json:"defaultEngine,omitempty"`
|
||||||
|
DefaultVoice string `yaml:"defaultVoice" json:"defaultVoice,omitempty"`
|
||||||
|
DefaultPitch int `yaml:"defaultPitch" json:"defaultPitch,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type MediaConfig struct {
|
||||||
|
PublishURL string `yaml:"publishUrl" json:"publishUrl,omitempty"`
|
||||||
|
AudioPublishURL string `yaml:"audioPublishUrl" json:"audioPublishUrl,omitempty"`
|
||||||
|
PublishPort int `yaml:"publishPort" json:"-"`
|
||||||
|
Manage bool `yaml:"manage"`
|
||||||
|
ManageAudio bool `yaml:"manageAudio"`
|
||||||
|
Service string `yaml:"service"`
|
||||||
|
AudioService string `yaml:"audioService"`
|
||||||
|
HealthURL string `yaml:"healthUrl"`
|
||||||
|
HealthInterval Duration `yaml:"healthInterval"`
|
||||||
|
VideoWidth int `yaml:"videoWidth" json:"-"`
|
||||||
|
VideoHeight int `yaml:"videoHeight" json:"-"`
|
||||||
|
VideoFPS int `yaml:"videoFps" json:"-"`
|
||||||
|
VideoBitrate int `yaml:"videoBitrate" json:"-"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type CameraServoConfig struct {
|
||||||
|
Enabled bool `yaml:"enabled" json:"enabled"`
|
||||||
|
Pin int `yaml:"pin" json:"pin"`
|
||||||
|
FreqHz int `yaml:"freqHz" json:"freqHz"`
|
||||||
|
CycleLen int `yaml:"cycleLen" json:"cycleLen"`
|
||||||
|
MinPulseUs int `yaml:"minPulseUs" json:"minPulseUs"`
|
||||||
|
MaxPulseUs int `yaml:"maxPulseUs" json:"maxPulseUs"`
|
||||||
|
MinAngle float64 `yaml:"minAngle" json:"minAngle"`
|
||||||
|
MaxAngle float64 `yaml:"maxAngle" json:"maxAngle"`
|
||||||
|
HomeAngle float64 `yaml:"homeAngle" json:"homeAngle"`
|
||||||
|
NudgeDegrees float64 `yaml:"nudgeDegrees" json:"nudgeDegrees"`
|
||||||
|
AllowRawPulse bool `yaml:"allowRawPulse" json:"allowRawPulse"`
|
||||||
|
Invert bool `yaml:"invert" json:"invert"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type NightVisionConfig struct {
|
||||||
|
Enabled bool `yaml:"enabled" json:"enabled"`
|
||||||
|
GPIOPin int `yaml:"gpioPin" json:"gpioPin"`
|
||||||
|
GPIOChip string `yaml:"gpioChip" json:"gpioChip"`
|
||||||
|
InitialOn bool `yaml:"initialOn" json:"initialOn"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type Config struct {
|
||||||
|
Name string `yaml:"name"`
|
||||||
|
ServerURL string `yaml:"serverUrl"`
|
||||||
|
Serial SerialConfig `yaml:"serial"`
|
||||||
|
BRC BRCConfig `yaml:"brc"`
|
||||||
|
Battery BatteryConfig `yaml:"battery"`
|
||||||
|
MaxWheelMMs int `yaml:"maxWheelSpeed"`
|
||||||
|
Media MediaConfig `yaml:"media"`
|
||||||
|
CameraServo CameraServoConfig `yaml:"cameraServo"`
|
||||||
|
Audio AudioConfig `yaml:"audio"`
|
||||||
|
NightVision NightVisionConfig `yaml:"nightVision" json:"nightVision"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func LoadConfig(path string) (*Config, error) {
|
||||||
|
data, err := os.ReadFile(path)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
cfg := Config{
|
||||||
|
MaxWheelMMs: 500,
|
||||||
|
BRC: BRCConfig{
|
||||||
|
GPIOPin: 4,
|
||||||
|
GPIOChip: "gpiochip0",
|
||||||
|
PulseEvery: Duration{
|
||||||
|
Duration: time.Minute,
|
||||||
|
},
|
||||||
|
PulseWidth: Duration{
|
||||||
|
Duration: time.Second,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
Media: MediaConfig{
|
||||||
|
PublishPort: 9000,
|
||||||
|
HealthInterval: Duration{Duration: 30 * time.Second},
|
||||||
|
VideoBitrate: 2000000,
|
||||||
|
},
|
||||||
|
CameraServo: CameraServoConfig{
|
||||||
|
Pin: 12,
|
||||||
|
FreqHz: 50,
|
||||||
|
CycleLen: 20000,
|
||||||
|
MinPulseUs: 900,
|
||||||
|
MaxPulseUs: 2100,
|
||||||
|
MinAngle: -15,
|
||||||
|
MaxAngle: 30,
|
||||||
|
HomeAngle: 0,
|
||||||
|
NudgeDegrees: 2,
|
||||||
|
},
|
||||||
|
Audio: AudioConfig{
|
||||||
|
CaptureEnabled: false,
|
||||||
|
CaptureDevice: "rovermic",
|
||||||
|
SampleRate: 48000,
|
||||||
|
Channels: 2,
|
||||||
|
Bitrate: 24000,
|
||||||
|
TTSEnabled: false,
|
||||||
|
DefaultEngine: "flite",
|
||||||
|
DefaultVoice: "rms",
|
||||||
|
DefaultPitch: 50,
|
||||||
|
},
|
||||||
|
NightVision: NightVisionConfig{
|
||||||
|
Enabled: true,
|
||||||
|
GPIOPin: 22,
|
||||||
|
GPIOChip: "gpiochip0",
|
||||||
|
InitialOn: true,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
if err := yaml.Unmarshal(data, &cfg); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if cfg.Name == "" {
|
||||||
|
return nil, errors.New("missing name")
|
||||||
|
}
|
||||||
|
if cfg.ServerURL == "" {
|
||||||
|
return nil, errors.New("missing serverUrl")
|
||||||
|
}
|
||||||
|
if cfg.Serial.Device == "" || cfg.Serial.Baud == 0 {
|
||||||
|
return nil, errors.New("serial device/baud required")
|
||||||
|
}
|
||||||
|
if cfg.Battery.Full == 0 {
|
||||||
|
return nil, errors.New("battery thresholds required")
|
||||||
|
}
|
||||||
|
if cfg.MaxWheelMMs <= 0 || cfg.MaxWheelMMs > 500 {
|
||||||
|
return nil, fmt.Errorf("maxWheelSpeed must be 1-500, got %d", cfg.MaxWheelMMs)
|
||||||
|
}
|
||||||
|
if cfg.BRC.GPIOChip == "" {
|
||||||
|
cfg.BRC.GPIOChip = "gpiochip0"
|
||||||
|
}
|
||||||
|
if cfg.Media.Manage && cfg.Media.Service == "" {
|
||||||
|
return nil, errors.New("media.manage requires media.service")
|
||||||
|
}
|
||||||
|
if cfg.Media.Manage && cfg.Media.HealthInterval.Duration <= 0 {
|
||||||
|
cfg.Media.HealthInterval = Duration{Duration: 30 * time.Second}
|
||||||
|
}
|
||||||
|
if cfg.Media.VideoBitrate <= 0 {
|
||||||
|
cfg.Media.VideoBitrate = 3000000
|
||||||
|
}
|
||||||
|
if cfg.Media.PublishPort <= 0 {
|
||||||
|
cfg.Media.PublishPort = 9000
|
||||||
|
}
|
||||||
|
if cfg.Media.PublishURL == "" {
|
||||||
|
derived, err := derivePublishURL(cfg.ServerURL, cfg.Name, cfg.Media.PublishPort)
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("derive publishUrl: %w", err)
|
||||||
|
}
|
||||||
|
cfg.Media.PublishURL = derived
|
||||||
|
}
|
||||||
|
if cfg.Media.AudioPublishURL == "" {
|
||||||
|
derived, err := derivePublishURL(cfg.ServerURL, cfg.Name+"-audio", cfg.Media.PublishPort)
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("derive audioPublishUrl: %w", err)
|
||||||
|
}
|
||||||
|
cfg.Media.AudioPublishURL = derived
|
||||||
|
}
|
||||||
|
if err := validateServoConfig(&cfg.CameraServo); err != nil {
|
||||||
|
return nil, fmt.Errorf("cameraServo: %w", err)
|
||||||
|
}
|
||||||
|
if err := validateNightVisionConfig(&cfg.NightVision); err != nil {
|
||||||
|
return nil, fmt.Errorf("nightVision: %w", err)
|
||||||
|
}
|
||||||
|
validateAudioConfig(&cfg.Audio)
|
||||||
|
return &cfg, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func validateServoConfig(cfg *CameraServoConfig) error {
|
||||||
|
if !cfg.Enabled {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
if cfg.Pin <= 0 {
|
||||||
|
return errors.New("pin must be > 0")
|
||||||
|
}
|
||||||
|
if cfg.FreqHz <= 0 {
|
||||||
|
return errors.New("freqHz must be > 0")
|
||||||
|
}
|
||||||
|
if cfg.CycleLen <= 0 {
|
||||||
|
return errors.New("cycleLen must be > 0")
|
||||||
|
}
|
||||||
|
if cfg.MinPulseUs <= 0 || cfg.MaxPulseUs <= 0 {
|
||||||
|
return errors.New("minPulseUs/maxPulseUs invalid")
|
||||||
|
}
|
||||||
|
if cfg.MinPulseUs == cfg.MaxPulseUs {
|
||||||
|
return errors.New("minPulseUs/maxPulseUs cannot be equal")
|
||||||
|
}
|
||||||
|
if cfg.MinPulseUs > cfg.MaxPulseUs {
|
||||||
|
cfg.MinPulseUs, cfg.MaxPulseUs = cfg.MaxPulseUs, cfg.MinPulseUs
|
||||||
|
cfg.Invert = !cfg.Invert
|
||||||
|
}
|
||||||
|
if cfg.MinAngle >= cfg.MaxAngle {
|
||||||
|
return errors.New("minAngle must be less than maxAngle")
|
||||||
|
}
|
||||||
|
cfg.HomeAngle = clampFloat(cfg.HomeAngle, cfg.MinAngle, cfg.MaxAngle)
|
||||||
|
if cfg.NudgeDegrees <= 0 {
|
||||||
|
cfg.NudgeDegrees = 2
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func clampFloat(value, min, max float64) float64 {
|
||||||
|
if value < min {
|
||||||
|
return min
|
||||||
|
}
|
||||||
|
if value > max {
|
||||||
|
return max
|
||||||
|
}
|
||||||
|
return value
|
||||||
|
}
|
||||||
|
|
||||||
|
func validateAudioConfig(cfg *AudioConfig) {
|
||||||
|
if cfg.CaptureEnabled && cfg.CaptureDevice == "" {
|
||||||
|
cfg.CaptureDevice = "hw:0,0"
|
||||||
|
}
|
||||||
|
if cfg.SampleRate <= 0 {
|
||||||
|
cfg.SampleRate = 48000
|
||||||
|
}
|
||||||
|
if cfg.Channels <= 0 {
|
||||||
|
cfg.Channels = 2
|
||||||
|
}
|
||||||
|
if cfg.Bitrate <= 0 {
|
||||||
|
cfg.Bitrate = 64000
|
||||||
|
}
|
||||||
|
if cfg.DefaultEngine == "" {
|
||||||
|
cfg.DefaultEngine = "flite"
|
||||||
|
}
|
||||||
|
if cfg.DefaultVoice == "" {
|
||||||
|
cfg.DefaultVoice = "rms"
|
||||||
|
}
|
||||||
|
if cfg.DefaultPitch <= 0 {
|
||||||
|
cfg.DefaultPitch = 50
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func validateNightVisionConfig(cfg *NightVisionConfig) error {
|
||||||
|
if !cfg.Enabled {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
if cfg.GPIOPin <= 0 {
|
||||||
|
return errors.New("gpioPin must be > 0")
|
||||||
|
}
|
||||||
|
if cfg.GPIOChip == "" {
|
||||||
|
cfg.GPIOChip = "gpiochip0"
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func derivePublishURL(serverURL, streamName string, port int) (string, error) {
|
||||||
|
if streamName == "" {
|
||||||
|
return "", errors.New("missing stream name for publishUrl")
|
||||||
|
}
|
||||||
|
parsed, err := url.Parse(serverURL)
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
host := parsed.Hostname()
|
||||||
|
if host == "" {
|
||||||
|
return "", errors.New("serverUrl missing host")
|
||||||
|
}
|
||||||
|
if port <= 0 {
|
||||||
|
port = 9000
|
||||||
|
}
|
||||||
|
escaped := url.PathEscape(streamName)
|
||||||
|
return fmt.Sprintf("srt://%s:%d?streamid=#!::r=%s,m=publish&latency=10&mode=caller&transtype=live&pkt_size=1316", host, port, escaped), nil
|
||||||
|
}
|
||||||
@@ -0,0 +1,8 @@
|
|||||||
|
package roverd
|
||||||
|
|
||||||
|
type RoverEvent struct {
|
||||||
|
Type string `json:"type"`
|
||||||
|
Event string `json:"event"`
|
||||||
|
Ts int64 `json:"ts"`
|
||||||
|
Data map[string]any `json:"data,omitempty"`
|
||||||
|
}
|
||||||
@@ -0,0 +1,13 @@
|
|||||||
|
module multiroombarover/pi/roverd
|
||||||
|
|
||||||
|
go 1.25.4
|
||||||
|
|
||||||
|
require (
|
||||||
|
github.com/stianeikeland/go-rpio/v4 v4.6.0
|
||||||
|
github.com/tarm/serial v0.0.0-20180830185346-98f6abe2eb07
|
||||||
|
github.com/warthog618/go-gpiocdev v0.9.1
|
||||||
|
gopkg.in/yaml.v3 v3.0.1
|
||||||
|
nhooyr.io/websocket v1.8.17
|
||||||
|
)
|
||||||
|
|
||||||
|
require golang.org/x/sys v0.38.0 // indirect
|
||||||
@@ -0,0 +1,24 @@
|
|||||||
|
github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
|
||||||
|
github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
||||||
|
github.com/pkg/errors v0.9.1 h1:FEBLx1zS214owpjy7qsBeixbURkuhQAwrK5UwLGTwt4=
|
||||||
|
github.com/pkg/errors v0.9.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=
|
||||||
|
github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
|
||||||
|
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
|
||||||
|
github.com/stianeikeland/go-rpio/v4 v4.6.0 h1:eAJgtw3jTtvn/CqwbC82ntcS+dtzUTgo5qlZKe677EY=
|
||||||
|
github.com/stianeikeland/go-rpio/v4 v4.6.0/go.mod h1:A3GvHxC1Om5zaId+HqB3HKqx4K/AqeckxB7qRjxMK7o=
|
||||||
|
github.com/stretchr/testify v1.9.0 h1:HtqpIVDClZ4nwg75+f6Lvsy/wHu+3BoSGCbBAcpTsTg=
|
||||||
|
github.com/stretchr/testify v1.9.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY=
|
||||||
|
github.com/tarm/serial v0.0.0-20180830185346-98f6abe2eb07 h1:UyzmZLoiDWMRywV4DUYb9Fbt8uiOSooupjTq10vpvnU=
|
||||||
|
github.com/tarm/serial v0.0.0-20180830185346-98f6abe2eb07/go.mod h1:kDXzergiv9cbyO7IOYJZWg1U88JhDg3PB6klq9Hg2pA=
|
||||||
|
github.com/warthog618/go-gpiocdev v0.9.1 h1:pwHPaqjJfhCipIQl78V+O3l9OKHivdRDdmgXYbmhuCI=
|
||||||
|
github.com/warthog618/go-gpiocdev v0.9.1/go.mod h1:dN3e3t/S2aSNC+hgigGE/dBW8jE1ONk9bDSEYfoPyl8=
|
||||||
|
github.com/warthog618/go-gpiosim v0.1.1 h1:MRAEv+T+itmw+3GeIGpQJBfanUVyg0l3JCTwHtwdre4=
|
||||||
|
github.com/warthog618/go-gpiosim v0.1.1/go.mod h1:YXsnB+I9jdCMY4YAlMSRrlts25ltjmuIsrnoUrBLdqU=
|
||||||
|
golang.org/x/sys v0.38.0 h1:3yZWxaJjBmCWXqhN1qh02AkOnCQ1poK6oF+a7xWL6Gc=
|
||||||
|
golang.org/x/sys v0.38.0/go.mod h1:OgkHotnGiDImocRcuBABYBEXf8A9a87e/uXjp9XT3ks=
|
||||||
|
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405 h1:yhCVgyC4o1eVCa2tZl7eS0r+SDo693bJlVdllGtEeKM=
|
||||||
|
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
|
||||||
|
gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
|
||||||
|
gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
||||||
|
nhooyr.io/websocket v1.8.17 h1:KEVeLJkUywCKVsnLIDlD/5gtayKp8VoCkksHCGGfT9Y=
|
||||||
|
nhooyr.io/websocket v1.8.17/go.mod h1:rN9OFWIUwuxg4fR5tELlYC04bXYowCP9GX47ivo2l+c=
|
||||||
@@ -0,0 +1,65 @@
|
|||||||
|
package roverd
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
"fmt"
|
||||||
|
"os"
|
||||||
|
"path/filepath"
|
||||||
|
)
|
||||||
|
|
||||||
|
const publisherEnvPath = "/var/lib/roverd/video.env"
|
||||||
|
|
||||||
|
func UpdatePublisherEnv(media MediaConfig, audio AudioConfig) error {
|
||||||
|
if media.PublishURL == "" {
|
||||||
|
return fmt.Errorf("media publishUrl missing")
|
||||||
|
}
|
||||||
|
if media.AudioPublishURL == "" && audio.CaptureEnabled {
|
||||||
|
return fmt.Errorf("audio publishUrl missing")
|
||||||
|
}
|
||||||
|
if media.VideoWidth < 0 || media.VideoHeight < 0 || media.VideoFPS < 0 || media.VideoBitrate <= 0 {
|
||||||
|
return fmt.Errorf("invalid media dimensions/bitrate")
|
||||||
|
}
|
||||||
|
if err := os.MkdirAll(filepath.Dir(publisherEnvPath), 0o755); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
var buf bytes.Buffer
|
||||||
|
fmt.Fprintf(&buf, "PUBLISH_URL=%s\n", media.PublishURL)
|
||||||
|
if audio.CaptureEnabled && media.AudioPublishURL != "" {
|
||||||
|
fmt.Fprintf(&buf, "AUDIO_PUBLISH_URL=%s\n", media.AudioPublishURL)
|
||||||
|
}
|
||||||
|
if media.VideoWidth > 0 {
|
||||||
|
fmt.Fprintf(&buf, "VIDEO_WIDTH=%d\n", media.VideoWidth)
|
||||||
|
}
|
||||||
|
if media.VideoHeight > 0 {
|
||||||
|
fmt.Fprintf(&buf, "VIDEO_HEIGHT=%d\n", media.VideoHeight)
|
||||||
|
}
|
||||||
|
if media.VideoFPS > 0 {
|
||||||
|
fmt.Fprintf(&buf, "VIDEO_FPS=%d\n", media.VideoFPS)
|
||||||
|
}
|
||||||
|
fmt.Fprintf(&buf, "VIDEO_BITRATE=%d\n", media.VideoBitrate)
|
||||||
|
audioDevice := audio.CaptureDevice
|
||||||
|
if audioDevice == "" || audioDevice == "rovermic" {
|
||||||
|
audioDevice = "hw:0,0"
|
||||||
|
}
|
||||||
|
if audio.SampleRate <= 0 {
|
||||||
|
audio.SampleRate = 48000
|
||||||
|
}
|
||||||
|
if audio.Channels <= 0 {
|
||||||
|
audio.Channels = 2
|
||||||
|
}
|
||||||
|
fmt.Fprintf(&buf, "AUDIO_ENABLE=%d\n", boolToInt(audio.CaptureEnabled))
|
||||||
|
fmt.Fprintf(&buf, "AUDIO_DEVICE=%s\n", audioDevice)
|
||||||
|
fmt.Fprintf(&buf, "AUDIO_RATE=%d\n", audio.SampleRate)
|
||||||
|
fmt.Fprintf(&buf, "AUDIO_CHANNELS=%d\n", audio.Channels)
|
||||||
|
if err := os.WriteFile(publisherEnvPath, buf.Bytes(), 0o640); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func boolToInt(v bool) int {
|
||||||
|
if v {
|
||||||
|
return 1
|
||||||
|
}
|
||||||
|
return 0
|
||||||
|
}
|
||||||
@@ -0,0 +1,142 @@
|
|||||||
|
package roverd
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
"io"
|
||||||
|
"log"
|
||||||
|
"net/http"
|
||||||
|
"os/exec"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
type MediaSupervisor struct {
|
||||||
|
cfg MediaConfig
|
||||||
|
audio AudioConfig
|
||||||
|
logger *log.Logger
|
||||||
|
client *http.Client
|
||||||
|
checkInterval time.Duration
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewMediaSupervisor(cfg MediaConfig, audio AudioConfig, logger *log.Logger) *MediaSupervisor {
|
||||||
|
if err := UpdatePublisherEnv(cfg, audio); err != nil {
|
||||||
|
logger.Printf("media supervisor: update env failed: %v", err)
|
||||||
|
}
|
||||||
|
if !cfg.Manage || cfg.Service == "" {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
interval := cfg.HealthInterval.Duration
|
||||||
|
if interval <= 0 {
|
||||||
|
interval = 30 * time.Second
|
||||||
|
}
|
||||||
|
var client *http.Client
|
||||||
|
if cfg.HealthURL != "" {
|
||||||
|
client = &http.Client{Timeout: 5 * time.Second}
|
||||||
|
}
|
||||||
|
return &MediaSupervisor{
|
||||||
|
cfg: cfg,
|
||||||
|
audio: audio,
|
||||||
|
logger: logger,
|
||||||
|
client: client,
|
||||||
|
checkInterval: interval,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *MediaSupervisor) Start(ctx context.Context) {
|
||||||
|
if m == nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if err := UpdatePublisherEnv(m.cfg, m.audio); err != nil {
|
||||||
|
m.logger.Printf("media supervisor: update env failed: %v", err)
|
||||||
|
}
|
||||||
|
if m.cfg.HealthURL == "" || m.client == nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
go func() {
|
||||||
|
ticker := time.NewTicker(m.checkInterval)
|
||||||
|
defer ticker.Stop()
|
||||||
|
|
||||||
|
if err := m.checkAndRepair(); err != nil {
|
||||||
|
m.logger.Printf("media supervisor: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
for {
|
||||||
|
select {
|
||||||
|
case <-ctx.Done():
|
||||||
|
return
|
||||||
|
case <-ticker.C:
|
||||||
|
if err := m.checkAndRepair(); err != nil {
|
||||||
|
m.logger.Printf("media supervisor: %v", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *MediaSupervisor) HandleAction(ctx context.Context, action string) error {
|
||||||
|
if m == nil {
|
||||||
|
return errors.New("media supervisor disabled")
|
||||||
|
}
|
||||||
|
if err := UpdatePublisherEnv(m.cfg, m.audio); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
switch action {
|
||||||
|
case "start", "stop", "restart", "reload", "status":
|
||||||
|
return m.runSystemctl(ctx, action)
|
||||||
|
default:
|
||||||
|
return fmt.Errorf("unknown media action: %s", action)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *MediaSupervisor) checkAndRepair() error {
|
||||||
|
ctx, cancel := context.WithTimeout(context.Background(), 7*time.Second)
|
||||||
|
defer cancel()
|
||||||
|
|
||||||
|
if m.checkHealth(ctx) {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
m.logger.Printf("media supervisor: health check failed, restarting %s", m.cfg.Service)
|
||||||
|
if err := m.runSystemctl(ctx, "restart"); err != nil {
|
||||||
|
return fmt.Errorf("restart mediamtx: %w", err)
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *MediaSupervisor) checkHealth(ctx context.Context) bool {
|
||||||
|
if m.client == nil || m.cfg.HealthURL == "" {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
req, err := http.NewRequestWithContext(ctx, http.MethodGet, m.cfg.HealthURL, nil)
|
||||||
|
if err != nil {
|
||||||
|
m.logger.Printf("media supervisor: health request: %v", err)
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
resp, err := m.client.Do(req)
|
||||||
|
if err != nil {
|
||||||
|
m.logger.Printf("media supervisor: health request failed: %v", err)
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
defer resp.Body.Close()
|
||||||
|
_, _ = io.Copy(io.Discard, resp.Body)
|
||||||
|
if resp.StatusCode >= 200 && resp.StatusCode < 300 {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
m.logger.Printf("media supervisor: unexpected health status %d", resp.StatusCode)
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *MediaSupervisor) runSystemctl(ctx context.Context, action string) error {
|
||||||
|
if m.cfg.Service == "" {
|
||||||
|
return errors.New("no media service configured")
|
||||||
|
}
|
||||||
|
runCtx, cancel := context.WithTimeout(ctx, 15*time.Second)
|
||||||
|
defer cancel()
|
||||||
|
|
||||||
|
cmd := exec.CommandContext(runCtx, "systemctl", action, m.cfg.Service)
|
||||||
|
output, err := cmd.CombinedOutput()
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("systemctl %s %s: %w (%s)", action, m.cfg.Service, err, string(output))
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
@@ -0,0 +1,97 @@
|
|||||||
|
//go:build !dummy
|
||||||
|
|
||||||
|
package roverd
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"log"
|
||||||
|
"strings"
|
||||||
|
"sync"
|
||||||
|
|
||||||
|
gpiocdev "github.com/warthog618/go-gpiocdev"
|
||||||
|
)
|
||||||
|
|
||||||
|
type NightVisionLight struct {
|
||||||
|
cfg NightVisionConfig
|
||||||
|
logger *log.Logger
|
||||||
|
line *gpiocdev.Line
|
||||||
|
mu sync.Mutex
|
||||||
|
on bool
|
||||||
|
closed bool
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewNightVisionLight(cfg NightVisionConfig, logger *log.Logger) (*NightVisionLight, error) {
|
||||||
|
if !cfg.Enabled {
|
||||||
|
return nil, fmt.Errorf("night vision disabled")
|
||||||
|
}
|
||||||
|
chip := cfg.GPIOChip
|
||||||
|
if chip == "" {
|
||||||
|
chip = "gpiochip0"
|
||||||
|
}
|
||||||
|
initial := 0
|
||||||
|
if cfg.InitialOn {
|
||||||
|
initial = 1
|
||||||
|
}
|
||||||
|
line, err := gpiocdev.RequestLine(
|
||||||
|
chip,
|
||||||
|
cfg.GPIOPin,
|
||||||
|
gpiocdev.AsOutput(initial),
|
||||||
|
gpiocdev.WithConsumer("roverd-nightvision"),
|
||||||
|
)
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("gpio request: %w", err)
|
||||||
|
}
|
||||||
|
nv := &NightVisionLight{
|
||||||
|
cfg: cfg,
|
||||||
|
logger: logger,
|
||||||
|
line: line,
|
||||||
|
on: cfg.InitialOn,
|
||||||
|
}
|
||||||
|
logger.Printf("night vision LED on GPIO %d (initial=%v)", cfg.GPIOPin, cfg.InitialOn)
|
||||||
|
return nv, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (n *NightVisionLight) Close() {
|
||||||
|
n.mu.Lock()
|
||||||
|
defer n.mu.Unlock()
|
||||||
|
if n.closed {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
_ = n.line.SetValue(boolToGPIO(n.on))
|
||||||
|
n.line.Close()
|
||||||
|
n.closed = true
|
||||||
|
}
|
||||||
|
|
||||||
|
func (n *NightVisionLight) HandleAction(action string) error {
|
||||||
|
n.mu.Lock()
|
||||||
|
defer n.mu.Unlock()
|
||||||
|
if n.closed {
|
||||||
|
return fmt.Errorf("night vision controller closed")
|
||||||
|
}
|
||||||
|
act := strings.ToLower(strings.TrimSpace(action))
|
||||||
|
switch act {
|
||||||
|
case "", "toggle":
|
||||||
|
return n.setLocked(!n.on)
|
||||||
|
case "on":
|
||||||
|
return n.setLocked(true)
|
||||||
|
case "off":
|
||||||
|
return n.setLocked(false)
|
||||||
|
default:
|
||||||
|
return fmt.Errorf("unknown action %q", action)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (n *NightVisionLight) setLocked(on bool) error {
|
||||||
|
if err := n.line.SetValue(boolToGPIO(on)); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
n.on = on
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func boolToGPIO(value bool) int {
|
||||||
|
if value {
|
||||||
|
return 1
|
||||||
|
}
|
||||||
|
return 0
|
||||||
|
}
|
||||||
@@ -0,0 +1,20 @@
|
|||||||
|
//go:build dummy
|
||||||
|
|
||||||
|
package roverd
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"log"
|
||||||
|
)
|
||||||
|
|
||||||
|
type NightVisionLight struct{}
|
||||||
|
|
||||||
|
func NewNightVisionLight(cfg NightVisionConfig, logger *log.Logger) (*NightVisionLight, error) {
|
||||||
|
return nil, fmt.Errorf("night vision not supported in dummy build")
|
||||||
|
}
|
||||||
|
|
||||||
|
func (n *NightVisionLight) Close() {}
|
||||||
|
|
||||||
|
func (n *NightVisionLight) HandleAction(action string) error {
|
||||||
|
return fmt.Errorf("night vision not supported in dummy build")
|
||||||
|
}
|
||||||
Executable
BIN
Binary file not shown.
@@ -0,0 +1,52 @@
|
|||||||
|
# Sample configuration for roverd
|
||||||
|
name: roomba-alpha
|
||||||
|
serverUrl: ws://control-server.local:8080/rover
|
||||||
|
serial:
|
||||||
|
device: /dev/ttyAMA0
|
||||||
|
baud: 115200
|
||||||
|
brc:
|
||||||
|
gpioPin: 4
|
||||||
|
gpioChip: gpiochip0
|
||||||
|
pulseEvery: 1m
|
||||||
|
pulseWidth: 1s
|
||||||
|
battery:
|
||||||
|
full: 2068
|
||||||
|
warn: 1700
|
||||||
|
urgent: 1650
|
||||||
|
maxWheelSpeed: 350
|
||||||
|
media:
|
||||||
|
publishUrl: srt://192.168.0.86:9000?streamid=#!::r=roomba-alpha,m=publish&latency=10&mode=caller&transtype=live&pkt_size=1316
|
||||||
|
publishPort: 9000
|
||||||
|
videoBitrate: 2000000
|
||||||
|
manage: true
|
||||||
|
service: video-publisher.service
|
||||||
|
healthUrl: ""
|
||||||
|
healthInterval: 30s
|
||||||
|
cameraServo:
|
||||||
|
enabled: false
|
||||||
|
pin: 12
|
||||||
|
freqHz: 50
|
||||||
|
cycleLen: 20000
|
||||||
|
minPulseUs: 900
|
||||||
|
maxPulseUs: 2100
|
||||||
|
invert: false
|
||||||
|
minAngle: -15
|
||||||
|
maxAngle: 30
|
||||||
|
homeAngle: 0
|
||||||
|
nudgeDegrees: 2
|
||||||
|
allowRawPulse: false
|
||||||
|
audio:
|
||||||
|
captureEnabled: false
|
||||||
|
captureDevice: hw:0,0
|
||||||
|
sampleRate: 48000
|
||||||
|
channels: 2
|
||||||
|
bitrate: 24000
|
||||||
|
ttsEnabled: false
|
||||||
|
defaultEngine: flite
|
||||||
|
defaultVoice: rms
|
||||||
|
defaultPitch: 50
|
||||||
|
nightVision:
|
||||||
|
enabled: true
|
||||||
|
gpioPin: 22
|
||||||
|
gpioChip: gpiochip0
|
||||||
|
initialOn: true
|
||||||
@@ -0,0 +1,33 @@
|
|||||||
|
# Sample configuration for roverd
|
||||||
|
name: roomba-alpha
|
||||||
|
serverUrl: ws://control-server.local:8080/rover
|
||||||
|
serial:
|
||||||
|
device: /dev/ttyAMA0
|
||||||
|
baud: 115200
|
||||||
|
brc:
|
||||||
|
gpioPin: 4
|
||||||
|
gpioChip: gpiochip0
|
||||||
|
pulseEvery: 1m
|
||||||
|
pulseWidth: 1s
|
||||||
|
battery:
|
||||||
|
full: 2068
|
||||||
|
warn: 1700
|
||||||
|
urgent: 1650
|
||||||
|
maxWheelSpeed: 350
|
||||||
|
media:
|
||||||
|
manage: false
|
||||||
|
service: mediamtx.service
|
||||||
|
healthUrl: http://127.0.0.1:9997/v3/paths/list
|
||||||
|
healthInterval: 30s
|
||||||
|
cameraServo:
|
||||||
|
enabled: false
|
||||||
|
pin: 19
|
||||||
|
freqHz: 50
|
||||||
|
cycleLen: 20000
|
||||||
|
minPulseUs: 900
|
||||||
|
maxPulseUs: 2100
|
||||||
|
minAngle: -15
|
||||||
|
maxAngle: 30
|
||||||
|
homeAngle: 0
|
||||||
|
nudgeDegrees: 2
|
||||||
|
allowRawPulse: false
|
||||||
@@ -0,0 +1,23 @@
|
|||||||
|
package roverd
|
||||||
|
|
||||||
|
var (
|
||||||
|
defaultStreamPackets = []byte{100, 21, 34}
|
||||||
|
packetSizes = map[byte]int{
|
||||||
|
100: 80,
|
||||||
|
21: 1,
|
||||||
|
34: 1,
|
||||||
|
}
|
||||||
|
expectedPayloadLength = func() int {
|
||||||
|
sum := 0
|
||||||
|
for _, id := range defaultStreamPackets {
|
||||||
|
sum += 1 + packetSizes[id]
|
||||||
|
}
|
||||||
|
return sum
|
||||||
|
}()
|
||||||
|
)
|
||||||
|
|
||||||
|
type SensorSample struct {
|
||||||
|
Timestamp int64
|
||||||
|
ChargingState byte
|
||||||
|
ChargeSources byte
|
||||||
|
}
|
||||||
@@ -0,0 +1,136 @@
|
|||||||
|
//go:build !dummy
|
||||||
|
|
||||||
|
package roverd
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bufio"
|
||||||
|
"context"
|
||||||
|
"encoding/hex"
|
||||||
|
"io"
|
||||||
|
"log"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
const (
|
||||||
|
sensorHeader = 19
|
||||||
|
sensorReadTimeout = 150 * time.Millisecond
|
||||||
|
sensorThrottleMinimum = 50 * time.Millisecond
|
||||||
|
)
|
||||||
|
|
||||||
|
type SensorStreamer struct {
|
||||||
|
r io.Reader
|
||||||
|
rawOut chan<- []byte
|
||||||
|
parsed chan<- SensorSample
|
||||||
|
logger *log.Logger
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewSensorStreamer(r io.Reader, rawOut chan<- []byte, parsed chan<- SensorSample, logger *log.Logger) *SensorStreamer {
|
||||||
|
return &SensorStreamer{r: r, rawOut: rawOut, parsed: parsed, logger: logger}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *SensorStreamer) Run(ctx context.Context) {
|
||||||
|
reader := bufio.NewReader(s.r)
|
||||||
|
var nextSend time.Time
|
||||||
|
for {
|
||||||
|
select {
|
||||||
|
case <-ctx.Done():
|
||||||
|
return
|
||||||
|
default:
|
||||||
|
}
|
||||||
|
|
||||||
|
header, err := reader.ReadByte()
|
||||||
|
if err != nil {
|
||||||
|
if ctx.Err() != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if header != sensorHeader {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
nBytes, err := reader.ReadByte()
|
||||||
|
if err != nil {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
frame := make([]byte, int(nBytes)+3)
|
||||||
|
frame[0] = sensorHeader
|
||||||
|
frame[1] = nBytes
|
||||||
|
if _, err := io.ReadFull(reader, frame[2:]); err != nil {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
if !validateChecksum(frame) {
|
||||||
|
s.logger.Printf("sensor checksum failed: %s", hex.EncodeToString(frame))
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
now := time.Now()
|
||||||
|
if !nextSend.IsZero() && now.Before(nextSend) {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
nextSend = now.Add(sensorThrottleMinimum)
|
||||||
|
|
||||||
|
select {
|
||||||
|
case s.rawOut <- frame:
|
||||||
|
default:
|
||||||
|
}
|
||||||
|
|
||||||
|
if s.parsed != nil {
|
||||||
|
if sample, ok := decodeSensorSample(frame); ok {
|
||||||
|
select {
|
||||||
|
case s.parsed <- sample:
|
||||||
|
default:
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func validateChecksum(buf []byte) bool {
|
||||||
|
var sum int
|
||||||
|
for _, b := range buf {
|
||||||
|
sum += int(b)
|
||||||
|
}
|
||||||
|
return byte(sum&0xFF) == 0
|
||||||
|
}
|
||||||
|
|
||||||
|
func decodeSensorSample(frame []byte) (SensorSample, bool) {
|
||||||
|
if len(frame) < 3 {
|
||||||
|
return SensorSample{}, false
|
||||||
|
}
|
||||||
|
nBytes := int(frame[1])
|
||||||
|
if nBytes+3 != len(frame) {
|
||||||
|
return SensorSample{}, false
|
||||||
|
}
|
||||||
|
payload := frame[2 : 2+nBytes]
|
||||||
|
if len(payload) != expectedPayloadLength {
|
||||||
|
return SensorSample{}, false
|
||||||
|
}
|
||||||
|
|
||||||
|
idx := 0
|
||||||
|
var sample SensorSample
|
||||||
|
var seen byte
|
||||||
|
for idx < len(payload) {
|
||||||
|
id := payload[idx]
|
||||||
|
idx++
|
||||||
|
size, ok := packetSizes[id]
|
||||||
|
if !ok {
|
||||||
|
return SensorSample{}, false
|
||||||
|
}
|
||||||
|
if idx+size > len(payload) {
|
||||||
|
return SensorSample{}, false
|
||||||
|
}
|
||||||
|
segment := payload[idx : idx+size]
|
||||||
|
switch id {
|
||||||
|
case 21:
|
||||||
|
sample.ChargingState = segment[0]
|
||||||
|
seen |= 1
|
||||||
|
case 34:
|
||||||
|
sample.ChargeSources = segment[0]
|
||||||
|
seen |= 2
|
||||||
|
}
|
||||||
|
idx += size
|
||||||
|
}
|
||||||
|
sample.Timestamp = time.Now().UnixMilli()
|
||||||
|
return sample, seen&3 == 3
|
||||||
|
}
|
||||||
@@ -0,0 +1,73 @@
|
|||||||
|
//go:build dummy
|
||||||
|
|
||||||
|
package roverd
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"log"
|
||||||
|
"math/rand"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
const sensorHeader = 19
|
||||||
|
|
||||||
|
type SensorStreamer struct {
|
||||||
|
rawOut chan<- []byte
|
||||||
|
parsed chan<- SensorSample
|
||||||
|
logger *log.Logger
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewSensorStreamer(_ interface{}, rawOut chan<- []byte, parsed chan<- SensorSample, logger *log.Logger) *SensorStreamer {
|
||||||
|
return &SensorStreamer{rawOut: rawOut, parsed: parsed, logger: logger}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *SensorStreamer) Run(ctx context.Context) {
|
||||||
|
ticker := time.NewTicker(200 * time.Millisecond)
|
||||||
|
defer ticker.Stop()
|
||||||
|
for {
|
||||||
|
select {
|
||||||
|
case <-ctx.Done():
|
||||||
|
return
|
||||||
|
case <-ticker.C:
|
||||||
|
frame := buildDummyFrame()
|
||||||
|
select {
|
||||||
|
case s.rawOut <- frame:
|
||||||
|
default:
|
||||||
|
}
|
||||||
|
sample := SensorSample{
|
||||||
|
Timestamp: time.Now().UnixMilli(),
|
||||||
|
ChargingState: 3, // trickle charging
|
||||||
|
ChargeSources: 0b10, // home base present
|
||||||
|
}
|
||||||
|
select {
|
||||||
|
case s.parsed <- sample:
|
||||||
|
default:
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func buildDummyFrame() []byte {
|
||||||
|
payload := make([]byte, 0, expectedPayloadLength)
|
||||||
|
payload = append(payload, 100)
|
||||||
|
group := make([]byte, packetSizes[100])
|
||||||
|
group[0] = byte(rand.Intn(16)) // bumps
|
||||||
|
payload = append(payload, group...)
|
||||||
|
payload = append(payload, 21, 3)
|
||||||
|
payload = append(payload, 34, 0b10)
|
||||||
|
|
||||||
|
buf := make([]byte, 0, len(payload)+3)
|
||||||
|
buf = append(buf, sensorHeader, byte(len(payload)))
|
||||||
|
buf = append(buf, payload...)
|
||||||
|
checksum := calcChecksum(buf)
|
||||||
|
buf = append(buf, checksum)
|
||||||
|
return buf
|
||||||
|
}
|
||||||
|
|
||||||
|
func calcChecksum(buf []byte) byte {
|
||||||
|
sum := 0
|
||||||
|
for _, b := range buf {
|
||||||
|
sum += int(b)
|
||||||
|
}
|
||||||
|
return byte((-sum) & 0xFF)
|
||||||
|
}
|
||||||
@@ -0,0 +1,124 @@
|
|||||||
|
//go:build !dummy
|
||||||
|
|
||||||
|
package roverd
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/base64"
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
"io"
|
||||||
|
"log"
|
||||||
|
"sync"
|
||||||
|
|
||||||
|
"github.com/tarm/serial"
|
||||||
|
)
|
||||||
|
|
||||||
|
type SerialAdapter struct {
|
||||||
|
port io.ReadWriteCloser
|
||||||
|
encoder *base64.Encoding
|
||||||
|
mu sync.Mutex
|
||||||
|
log *log.Logger
|
||||||
|
}
|
||||||
|
|
||||||
|
func OpenSerial(cfg SerialConfig) (*serial.Port, error) {
|
||||||
|
return serial.OpenPort(&serial.Config{
|
||||||
|
Name: cfg.Device,
|
||||||
|
Baud: cfg.Baud,
|
||||||
|
ReadTimeout: sensorReadTimeout,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewSerialAdapter(port io.ReadWriteCloser, logger *log.Logger) *SerialAdapter {
|
||||||
|
return &SerialAdapter{
|
||||||
|
port: port,
|
||||||
|
encoder: base64.StdEncoding,
|
||||||
|
log: logger,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *SerialAdapter) write(buf []byte) error {
|
||||||
|
s.mu.Lock()
|
||||||
|
defer s.mu.Unlock()
|
||||||
|
|
||||||
|
n, err := s.port.Write(buf)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if n != len(buf) {
|
||||||
|
return fmt.Errorf("short write %d/%d", n, len(buf))
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *SerialAdapter) DriveDirect(left, right int) error {
|
||||||
|
payload := []byte{
|
||||||
|
145,
|
||||||
|
byte((right >> 8) & 0xFF),
|
||||||
|
byte(right & 0xFF),
|
||||||
|
byte((left >> 8) & 0xFF),
|
||||||
|
byte(left & 0xFF),
|
||||||
|
}
|
||||||
|
return s.write(payload)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *SerialAdapter) MotorPWM(main, side, vacuum int) error {
|
||||||
|
payload := []byte{
|
||||||
|
144,
|
||||||
|
byte(main & 0xFF),
|
||||||
|
byte(side & 0xFF),
|
||||||
|
byte(vacuum & 0xFF),
|
||||||
|
}
|
||||||
|
return s.write(payload)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *SerialAdapter) StartSensorStream(packets []byte) error {
|
||||||
|
if len(packets) == 0 {
|
||||||
|
return errors.New("sensor stream requires packets")
|
||||||
|
}
|
||||||
|
payload := []byte{148, byte(len(packets))}
|
||||||
|
payload = append(payload, packets...)
|
||||||
|
return s.write(payload)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *SerialAdapter) PauseSensorStream(pause bool) error {
|
||||||
|
state := byte(1)
|
||||||
|
if pause {
|
||||||
|
state = 0
|
||||||
|
}
|
||||||
|
return s.write([]byte{150, state})
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *SerialAdapter) SendRaw(raw []byte) error {
|
||||||
|
return s.write(raw)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *SerialAdapter) StartOI() error {
|
||||||
|
return s.write([]byte{128})
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *SerialAdapter) SeekDock() error {
|
||||||
|
return s.write([]byte{143})
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *SerialAdapter) PlaySong(slot int, notes []songNote) error {
|
||||||
|
if len(notes) == 0 {
|
||||||
|
return fmt.Errorf("song requires at least one note")
|
||||||
|
}
|
||||||
|
if len(notes) > 16 {
|
||||||
|
return fmt.Errorf("song supports up to 16 notes, got %d", len(notes))
|
||||||
|
}
|
||||||
|
if slot < 0 || slot > 4 {
|
||||||
|
return fmt.Errorf("song slot must be 0-4")
|
||||||
|
}
|
||||||
|
|
||||||
|
payload := []byte{140, byte(slot), byte(len(notes))}
|
||||||
|
for _, n := range notes {
|
||||||
|
note := clampInt(n.Note, 31, 127)
|
||||||
|
duration := clampInt(n.Duration, 1, 255)
|
||||||
|
payload = append(payload, byte(note), byte(duration))
|
||||||
|
}
|
||||||
|
if err := s.write(payload); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
return s.write([]byte{141, byte(slot)})
|
||||||
|
}
|
||||||
@@ -0,0 +1,68 @@
|
|||||||
|
//go:build dummy
|
||||||
|
|
||||||
|
package roverd
|
||||||
|
|
||||||
|
import (
|
||||||
|
"errors"
|
||||||
|
"io"
|
||||||
|
"log"
|
||||||
|
)
|
||||||
|
|
||||||
|
type dummyPort struct{}
|
||||||
|
|
||||||
|
func (dummyPort) Read(p []byte) (int, error) { return 0, io.EOF }
|
||||||
|
func (dummyPort) Write(p []byte) (int, error) { return len(p), nil }
|
||||||
|
func (dummyPort) Close() error { return nil }
|
||||||
|
|
||||||
|
func OpenSerial(cfg SerialConfig) (io.ReadWriteCloser, error) {
|
||||||
|
return dummyPort{}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
type SerialAdapter struct {
|
||||||
|
log *log.Logger
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewSerialAdapter(_ io.ReadWriteCloser, logger *log.Logger) *SerialAdapter {
|
||||||
|
return &SerialAdapter{log: logger}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *SerialAdapter) DriveDirect(left, right int) error {
|
||||||
|
s.log.Printf("[dummy] drive L=%d R=%d", left, right)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *SerialAdapter) MotorPWM(main, side, vacuum int) error {
|
||||||
|
s.log.Printf("[dummy] motor main=%d side=%d vacuum=%d", main, side, vacuum)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *SerialAdapter) StartSensorStream(packets []byte) error {
|
||||||
|
if len(packets) == 0 {
|
||||||
|
return errors.New("sensor stream requires packets")
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *SerialAdapter) PauseSensorStream(pause bool) error {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *SerialAdapter) SendRaw(raw []byte) error {
|
||||||
|
s.log.Printf("[dummy] raw %v", raw)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *SerialAdapter) StartOI() error {
|
||||||
|
s.log.Printf("[dummy] start OI")
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *SerialAdapter) SeekDock() error {
|
||||||
|
s.log.Printf("[dummy] seek dock")
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *SerialAdapter) PlaySong(slot int, notes []songNote) error {
|
||||||
|
s.log.Printf("[dummy] play song slot=%d notes=%v", slot, notes)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
@@ -0,0 +1,75 @@
|
|||||||
|
package roverd
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"fmt"
|
||||||
|
"os/exec"
|
||||||
|
"strings"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
func (c *WSClient) handleTTSPayload(ctx context.Context, payload *ttsPayload) error {
|
||||||
|
if payload == nil {
|
||||||
|
return fmt.Errorf("tts payload required")
|
||||||
|
}
|
||||||
|
if !c.cfg.Audio.TTSEnabled {
|
||||||
|
return fmt.Errorf("tts disabled on rover")
|
||||||
|
}
|
||||||
|
if payload.Speak == false {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
text := strings.TrimSpace(payload.Text)
|
||||||
|
if text == "" {
|
||||||
|
return fmt.Errorf("tts text required")
|
||||||
|
}
|
||||||
|
if len([]rune(text)) > 512 {
|
||||||
|
text = string([]rune(text)[:512])
|
||||||
|
}
|
||||||
|
|
||||||
|
engine := strings.ToLower(strings.TrimSpace(payload.Engine))
|
||||||
|
if engine == "" {
|
||||||
|
engine = strings.ToLower(strings.TrimSpace(c.cfg.Audio.DefaultEngine))
|
||||||
|
}
|
||||||
|
if engine == "" {
|
||||||
|
engine = "flite"
|
||||||
|
}
|
||||||
|
|
||||||
|
voice := strings.TrimSpace(payload.Voice)
|
||||||
|
if voice == "" {
|
||||||
|
voice = strings.TrimSpace(c.cfg.Audio.DefaultVoice)
|
||||||
|
}
|
||||||
|
pitch := payload.Pitch
|
||||||
|
if pitch <= 0 {
|
||||||
|
pitch = c.cfg.Audio.DefaultPitch
|
||||||
|
}
|
||||||
|
pitch = clampInt(pitch, 0, 99)
|
||||||
|
|
||||||
|
runCtx, cancel := context.WithTimeout(ctx, 12*time.Second)
|
||||||
|
defer cancel()
|
||||||
|
|
||||||
|
var cmd *exec.Cmd
|
||||||
|
switch engine {
|
||||||
|
case "espeak", "e":
|
||||||
|
args := []string{}
|
||||||
|
if pitch > 0 {
|
||||||
|
args = append(args, "-p", fmt.Sprintf("%d", pitch))
|
||||||
|
}
|
||||||
|
args = append(args, text)
|
||||||
|
cmd = exec.CommandContext(runCtx, "espeak", args...)
|
||||||
|
case "flite", "f":
|
||||||
|
args := []string{}
|
||||||
|
if voice != "" {
|
||||||
|
args = append(args, "-voice", voice)
|
||||||
|
}
|
||||||
|
args = append(args, "-t", text)
|
||||||
|
cmd = exec.CommandContext(runCtx, "flite", args...)
|
||||||
|
default:
|
||||||
|
return fmt.Errorf("unsupported tts engine: %s", engine)
|
||||||
|
}
|
||||||
|
|
||||||
|
out, err := cmd.CombinedOutput()
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("tts exec failed: %w (%s)", err, string(out))
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
@@ -0,0 +1,440 @@
|
|||||||
|
package roverd
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"encoding/base64"
|
||||||
|
"encoding/json"
|
||||||
|
"fmt"
|
||||||
|
"log"
|
||||||
|
"sync"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"nhooyr.io/websocket"
|
||||||
|
)
|
||||||
|
|
||||||
|
type WSClient struct {
|
||||||
|
cfg *Config
|
||||||
|
adapter *SerialAdapter
|
||||||
|
sensorFrames <-chan []byte
|
||||||
|
events chan RoverEvent
|
||||||
|
media *MediaSupervisor
|
||||||
|
servo *CameraServo
|
||||||
|
nightVision *NightVisionLight
|
||||||
|
log *log.Logger
|
||||||
|
recoverMu sync.Mutex
|
||||||
|
recovering bool
|
||||||
|
ttsQueue chan *ttsPayload
|
||||||
|
connMu sync.Mutex
|
||||||
|
connected bool
|
||||||
|
disconnectT *time.Timer
|
||||||
|
seekIssued bool
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewWSClient(cfg *Config, adapter *SerialAdapter, frames <-chan []byte, events chan RoverEvent, media *MediaSupervisor, servo *CameraServo, nightVision *NightVisionLight, logger *log.Logger) *WSClient {
|
||||||
|
var ttsQueue chan *ttsPayload
|
||||||
|
if cfg.Audio.TTSEnabled {
|
||||||
|
ttsQueue = make(chan *ttsPayload, 2)
|
||||||
|
}
|
||||||
|
return &WSClient{
|
||||||
|
cfg: cfg,
|
||||||
|
adapter: adapter,
|
||||||
|
sensorFrames: frames,
|
||||||
|
events: events,
|
||||||
|
media: media,
|
||||||
|
servo: servo,
|
||||||
|
nightVision: nightVision,
|
||||||
|
log: logger,
|
||||||
|
ttsQueue: ttsQueue,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *WSClient) Run(ctx context.Context) error {
|
||||||
|
conn, _, err := websocket.Dial(ctx, c.cfg.ServerURL, nil)
|
||||||
|
if err != nil {
|
||||||
|
c.markDisconnected()
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
c.markConnected()
|
||||||
|
defer conn.Close(websocket.StatusInternalError, "closed")
|
||||||
|
defer c.markDisconnected()
|
||||||
|
|
||||||
|
if err := c.sendHello(ctx, conn); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if err := c.ensureSensorStream(); err != nil {
|
||||||
|
c.log.Printf("sensor stream init failed: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
errCh := make(chan error, 1)
|
||||||
|
c.startTTSWorker(ctx)
|
||||||
|
go func() {
|
||||||
|
errCh <- c.readLoop(ctx, conn)
|
||||||
|
}()
|
||||||
|
go c.forwardSensors(ctx, conn)
|
||||||
|
go c.forwardEvents(ctx, conn)
|
||||||
|
|
||||||
|
select {
|
||||||
|
case <-ctx.Done():
|
||||||
|
conn.Close(websocket.StatusNormalClosure, "context done")
|
||||||
|
return ctx.Err()
|
||||||
|
case err := <-errCh:
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *WSClient) sendHello(ctx context.Context, conn *websocket.Conn) error {
|
||||||
|
msg := helloMessage{
|
||||||
|
Type: "hello",
|
||||||
|
Name: c.cfg.Name,
|
||||||
|
Battery: c.cfg.Battery,
|
||||||
|
MaxWheelSpeed: c.cfg.MaxWheelMMs,
|
||||||
|
Media: c.cfg.Media,
|
||||||
|
CameraServo: c.cfg.CameraServo,
|
||||||
|
Audio: c.cfg.Audio,
|
||||||
|
NightVision: c.cfg.NightVision,
|
||||||
|
}
|
||||||
|
c.log.Printf("sending hello (camera servo enabled=%v pin=%d)", msg.CameraServo.Enabled, msg.CameraServo.Pin)
|
||||||
|
return writeJSON(ctx, conn, msg)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *WSClient) readLoop(ctx context.Context, conn *websocket.Conn) error {
|
||||||
|
for {
|
||||||
|
_, data, err := conn.Read(ctx)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
var msg inboundMessage
|
||||||
|
if err := json.Unmarshal(data, &msg); err != nil {
|
||||||
|
c.log.Printf("invalid command: %v", err)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if msg.ID == "" {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
status := "ok"
|
||||||
|
cmdErr := c.dispatch(ctx, &msg)
|
||||||
|
if cmdErr != nil {
|
||||||
|
status = "error"
|
||||||
|
}
|
||||||
|
ack := ackMessage{
|
||||||
|
Type: "ack",
|
||||||
|
ID: msg.ID,
|
||||||
|
Status: status,
|
||||||
|
}
|
||||||
|
if cmdErr != nil {
|
||||||
|
ack.Error = cmdErr.Error()
|
||||||
|
}
|
||||||
|
if err := writeJSON(ctx, conn, ack); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *WSClient) dispatch(ctx context.Context, msg *inboundMessage) error {
|
||||||
|
switch {
|
||||||
|
case msg.DriveDirect != nil:
|
||||||
|
left := clamp(msg.DriveDirect.Left, -c.cfg.MaxWheelMMs, c.cfg.MaxWheelMMs)
|
||||||
|
right := clamp(msg.DriveDirect.Right, -c.cfg.MaxWheelMMs, c.cfg.MaxWheelMMs)
|
||||||
|
return c.adapter.DriveDirect(left, right)
|
||||||
|
case msg.MotorPWM != nil:
|
||||||
|
main := clamp(msg.MotorPWM.Main, -127, 127)
|
||||||
|
side := clamp(msg.MotorPWM.Side, -127, 127)
|
||||||
|
vac := clamp(msg.MotorPWM.Vacuum, 0, 127)
|
||||||
|
return c.adapter.MotorPWM(main, side, vac)
|
||||||
|
case msg.SensorStream != nil:
|
||||||
|
if msg.SensorStream.Enable {
|
||||||
|
return c.adapter.StartSensorStream(defaultStreamPackets)
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
case msg.Raw != "" && len(msg.Raw) > 0:
|
||||||
|
buf, err := base64.StdEncoding.DecodeString(msg.Raw)
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("raw decode: %w", err)
|
||||||
|
}
|
||||||
|
if err := c.adapter.SendRaw(buf); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if len(buf) > 0 && isModeOpcode(buf[0]) {
|
||||||
|
return c.ensureSensorStream()
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
case msg.Media != nil:
|
||||||
|
if c.media == nil {
|
||||||
|
return fmt.Errorf("media supervisor disabled")
|
||||||
|
}
|
||||||
|
return c.media.HandleAction(ctx, msg.Media.Action)
|
||||||
|
case msg.Servo != nil:
|
||||||
|
if c.servo == nil {
|
||||||
|
return fmt.Errorf("camera servo disabled")
|
||||||
|
}
|
||||||
|
return c.handleServoCommand(msg.Servo)
|
||||||
|
case msg.TTS != nil:
|
||||||
|
return c.enqueueTTS(msg.TTS)
|
||||||
|
case msg.NightVision != nil:
|
||||||
|
if c.nightVision == nil {
|
||||||
|
return fmt.Errorf("night vision disabled")
|
||||||
|
}
|
||||||
|
return c.nightVision.HandleAction(msg.NightVision.Action)
|
||||||
|
case msg.Song != nil:
|
||||||
|
slot := 0
|
||||||
|
if msg.Song.Slot != nil {
|
||||||
|
slot = clampInt(*msg.Song.Slot, 0, 4)
|
||||||
|
}
|
||||||
|
return c.adapter.PlaySong(slot, msg.Song.Notes)
|
||||||
|
default:
|
||||||
|
return fmt.Errorf("unsupported command type: %s", msg.Type)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *WSClient) enqueueTTS(payload *ttsPayload) error {
|
||||||
|
if c.ttsQueue == nil {
|
||||||
|
return fmt.Errorf("tts disabled")
|
||||||
|
}
|
||||||
|
select {
|
||||||
|
case c.ttsQueue <- payload:
|
||||||
|
return nil
|
||||||
|
default:
|
||||||
|
return fmt.Errorf("tts busy")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *WSClient) startTTSWorker(ctx context.Context) {
|
||||||
|
if c.ttsQueue == nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
go func() {
|
||||||
|
for {
|
||||||
|
select {
|
||||||
|
case <-ctx.Done():
|
||||||
|
return
|
||||||
|
case payload := <-c.ttsQueue:
|
||||||
|
if payload == nil {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if err := c.handleTTSPayload(ctx, payload); err != nil {
|
||||||
|
c.log.Printf("tts failed: %v", err)
|
||||||
|
c.emitEvent("tts.error", map[string]any{"error": err.Error()})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *WSClient) handleServoCommand(payload *servoPayload) error {
|
||||||
|
switch {
|
||||||
|
case payload.Angle != nil:
|
||||||
|
return c.servo.SetAngle(*payload.Angle)
|
||||||
|
case payload.Nudge != nil:
|
||||||
|
return c.servo.Nudge(*payload.Nudge)
|
||||||
|
case payload.PulseUs != nil:
|
||||||
|
return c.servo.SetPulseWidth(*payload.PulseUs)
|
||||||
|
default:
|
||||||
|
return fmt.Errorf("servo command requires angle, nudge, or pulseUs")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *WSClient) forwardSensors(ctx context.Context, conn *websocket.Conn) {
|
||||||
|
const (
|
||||||
|
sensorSilenceTimeout = 5 * time.Second
|
||||||
|
sensorRecoveryCooldown = 3 * time.Second
|
||||||
|
sensorCommandPause = 50 * time.Millisecond
|
||||||
|
)
|
||||||
|
|
||||||
|
timer := time.NewTimer(sensorSilenceTimeout)
|
||||||
|
defer timer.Stop()
|
||||||
|
|
||||||
|
resetTimer := func() {
|
||||||
|
if !timer.Stop() {
|
||||||
|
select {
|
||||||
|
case <-timer.C:
|
||||||
|
default:
|
||||||
|
}
|
||||||
|
}
|
||||||
|
timer.Reset(sensorSilenceTimeout)
|
||||||
|
}
|
||||||
|
|
||||||
|
lastRecovery := time.Time{}
|
||||||
|
lastFrame := time.Now()
|
||||||
|
|
||||||
|
for {
|
||||||
|
select {
|
||||||
|
case <-ctx.Done():
|
||||||
|
return
|
||||||
|
case <-timer.C:
|
||||||
|
now := time.Now()
|
||||||
|
if !lastRecovery.IsZero() && now.Sub(lastRecovery) < sensorRecoveryCooldown {
|
||||||
|
resetTimer()
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
idleFor := now.Sub(lastFrame)
|
||||||
|
if idleFor < 0 {
|
||||||
|
idleFor = sensorSilenceTimeout
|
||||||
|
}
|
||||||
|
|
||||||
|
c.recoverSensorStream(idleFor, sensorCommandPause)
|
||||||
|
lastRecovery = now
|
||||||
|
resetTimer()
|
||||||
|
case frame := <-c.sensorFrames:
|
||||||
|
lastFrame = time.Now()
|
||||||
|
resetTimer()
|
||||||
|
msg := sensorMessage{
|
||||||
|
Type: "sensor",
|
||||||
|
Timestamp: time.Now().UnixMilli(),
|
||||||
|
Data: base64.StdEncoding.EncodeToString(frame),
|
||||||
|
}
|
||||||
|
if err := writeJSON(ctx, conn, msg); err != nil {
|
||||||
|
c.log.Printf("sensor send failed: %v", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *WSClient) forwardEvents(ctx context.Context, conn *websocket.Conn) {
|
||||||
|
if c.events == nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
for {
|
||||||
|
select {
|
||||||
|
case <-ctx.Done():
|
||||||
|
return
|
||||||
|
case evt := <-c.events:
|
||||||
|
if evt.Type == "" {
|
||||||
|
evt.Type = "event"
|
||||||
|
}
|
||||||
|
if err := writeJSON(ctx, conn, evt); err != nil {
|
||||||
|
c.log.Printf("event send failed: %v", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *WSClient) emitEvent(event string, data map[string]any) {
|
||||||
|
if c.events == nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
select {
|
||||||
|
case c.events <- RoverEvent{
|
||||||
|
Type: "event",
|
||||||
|
Event: event,
|
||||||
|
Ts: time.Now().UnixMilli(),
|
||||||
|
Data: data,
|
||||||
|
}:
|
||||||
|
default:
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func writeJSON(ctx context.Context, conn *websocket.Conn, v any) error {
|
||||||
|
data, err := json.Marshal(v)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
return conn.Write(ctx, websocket.MessageText, data)
|
||||||
|
}
|
||||||
|
|
||||||
|
func clamp(value, min, max int) int {
|
||||||
|
if value < min {
|
||||||
|
return min
|
||||||
|
}
|
||||||
|
if value > max {
|
||||||
|
return max
|
||||||
|
}
|
||||||
|
return value
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *WSClient) ensureSensorStream() error {
|
||||||
|
if err := c.adapter.StartSensorStream(defaultStreamPackets); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
const disconnectSeekDelay = time.Minute
|
||||||
|
|
||||||
|
func (c *WSClient) markConnected() {
|
||||||
|
c.connMu.Lock()
|
||||||
|
c.connected = true
|
||||||
|
c.seekIssued = false
|
||||||
|
if c.disconnectT != nil {
|
||||||
|
c.disconnectT.Stop()
|
||||||
|
c.disconnectT = nil
|
||||||
|
}
|
||||||
|
c.connMu.Unlock()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *WSClient) markDisconnected() {
|
||||||
|
c.connMu.Lock()
|
||||||
|
if c.connected {
|
||||||
|
c.connected = false
|
||||||
|
}
|
||||||
|
if c.disconnectT == nil {
|
||||||
|
c.disconnectT = time.AfterFunc(disconnectSeekDelay, c.handleDisconnectTimeout)
|
||||||
|
}
|
||||||
|
c.connMu.Unlock()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *WSClient) handleDisconnectTimeout() {
|
||||||
|
c.connMu.Lock()
|
||||||
|
if c.connected || c.seekIssued {
|
||||||
|
c.connMu.Unlock()
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.seekIssued = true
|
||||||
|
c.connMu.Unlock()
|
||||||
|
|
||||||
|
if err := c.adapter.SeekDock(); err != nil {
|
||||||
|
c.log.Printf("seek dock on disconnect failed: %v", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.log.Printf("seek dock issued after websocket disconnect")
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *WSClient) recoverSensorStream(idleFor time.Duration, cmdPause time.Duration) {
|
||||||
|
c.recoverMu.Lock()
|
||||||
|
if c.recovering {
|
||||||
|
c.recoverMu.Unlock()
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.recovering = true
|
||||||
|
c.recoverMu.Unlock()
|
||||||
|
|
||||||
|
defer func() {
|
||||||
|
c.recoverMu.Lock()
|
||||||
|
c.recovering = false
|
||||||
|
c.recoverMu.Unlock()
|
||||||
|
}()
|
||||||
|
|
||||||
|
c.emitEvent("sensorWatchdog.restart", map[string]any{
|
||||||
|
"idleMs": idleFor.Milliseconds(),
|
||||||
|
})
|
||||||
|
|
||||||
|
if err := c.adapter.StartOI(); err != nil {
|
||||||
|
c.log.Printf("watchdog start OI failed: %v", err)
|
||||||
|
c.emitEvent("sensorWatchdog.error", map[string]any{"error": err.Error()})
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if cmdPause > 0 {
|
||||||
|
time.Sleep(cmdPause)
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := c.adapter.StartSensorStream(defaultStreamPackets); err != nil {
|
||||||
|
c.log.Printf("watchdog start stream failed: %v", err)
|
||||||
|
c.emitEvent("sensorWatchdog.error", map[string]any{"error": err.Error()})
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
c.emitEvent("sensorWatchdog.ok", map[string]any{
|
||||||
|
"idleMs": idleFor.Milliseconds(),
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
func isModeOpcode(op byte) bool {
|
||||||
|
switch op {
|
||||||
|
case 128, 131, 132:
|
||||||
|
return true
|
||||||
|
default:
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,18 @@
|
|||||||
|
[Unit]
|
||||||
|
Description=Rover Audio Publisher (ALSA -> SRT)
|
||||||
|
After=network-online.target roverd.service
|
||||||
|
Wants=network-online.target
|
||||||
|
|
||||||
|
[Service]
|
||||||
|
Type=simple
|
||||||
|
User=roverd
|
||||||
|
Group=roverd
|
||||||
|
EnvironmentFile=/var/lib/roverd/video.env
|
||||||
|
ExecStart=/usr/local/bin/audio-only-publisher
|
||||||
|
KillMode=control-group
|
||||||
|
TimeoutStopSec=5
|
||||||
|
Restart=on-failure
|
||||||
|
RestartSec=2
|
||||||
|
|
||||||
|
[Install]
|
||||||
|
WantedBy=multi-user.target
|
||||||
@@ -0,0 +1,14 @@
|
|||||||
|
[Unit]
|
||||||
|
Description=Multi-Roomba rover control agent
|
||||||
|
After=network-online.target mediamtx.service
|
||||||
|
Wants=network-online.target
|
||||||
|
|
||||||
|
[Service]
|
||||||
|
Type=simple
|
||||||
|
ExecStart=/usr/local/bin/roverd -config /etc/roverd.yaml
|
||||||
|
Restart=on-failure
|
||||||
|
RestartSec=5
|
||||||
|
AmbientCapabilities=CAP_SYS_TTY_CONFIG CAP_SYS_RAWIO
|
||||||
|
|
||||||
|
[Install]
|
||||||
|
WantedBy=multi-user.target
|
||||||
@@ -0,0 +1,17 @@
|
|||||||
|
[Unit]
|
||||||
|
Description=Rover Video Publisher (libcamera -> SRT)
|
||||||
|
After=network-online.target roverd.service
|
||||||
|
Wants=network-online.target
|
||||||
|
|
||||||
|
[Service]
|
||||||
|
Type=simple
|
||||||
|
User=roverd
|
||||||
|
Group=roverd
|
||||||
|
WorkingDirectory=/var/lib/roverd
|
||||||
|
EnvironmentFile=/var/lib/roverd/video.env
|
||||||
|
ExecStart=/usr/local/bin/video-publisher
|
||||||
|
Restart=always
|
||||||
|
RestartSec=2
|
||||||
|
|
||||||
|
[Install]
|
||||||
|
WantedBy=multi-user.target
|
||||||
@@ -0,0 +1,27 @@
|
|||||||
|
# general idea
|
||||||
|
- the pi on each roomba will have a speaker and microphone
|
||||||
|
- alsa devices
|
||||||
|
- they are both the default audio device
|
||||||
|
- you will have to modify roverd, the pi install script, and the server to get this all to work
|
||||||
|
|
||||||
|
# adding stuff to installation script
|
||||||
|
- I am using the google voice AIY v1 kits for audio
|
||||||
|
- boot config stuff
|
||||||
|
- enable `dtoverlay=googlevoicehat-soundcard`
|
||||||
|
- disable `dtparam=audio=on`
|
||||||
|
- copy asound.conf in pi folder to /etc/asound.conf
|
||||||
|
|
||||||
|
# microphone
|
||||||
|
- add microphone to the SRT publish stream
|
||||||
|
|
||||||
|
# speaker
|
||||||
|
- user's chat messages will be TTS'ed through the speaker on the pi
|
||||||
|
- either flite with a way to choose the voice
|
||||||
|
- or espeak where you can choose the pitch
|
||||||
|
- on the web UI and in the chat API
|
||||||
|
- add new stuff to chat
|
||||||
|
- only shows up if you are on a rover
|
||||||
|
- only shows up if TTS is enabled on that rover
|
||||||
|
- people can choose between flite or espeak
|
||||||
|
- if they choose flite, they can choose the voice from the default flite voices (exclude awb and awb_time)
|
||||||
|
- if they choose espeak, they can choose the pitch. Have a dropdown with increments of 10 from 0 to 99
|
||||||
@@ -0,0 +1,33 @@
|
|||||||
|
# access control modes:
|
||||||
|
- open
|
||||||
|
- open to the public
|
||||||
|
- users are randomly assigned a rover to drive, with priority on the rover with the least amount of drivers
|
||||||
|
- if there are multiple people on one rover, those people will all be controlling that rover at the same time
|
||||||
|
- turns
|
||||||
|
- open to the public
|
||||||
|
- users are randomly assigned to a rover same as open mode
|
||||||
|
- if there are multiple people on one rover, they will each have one minute at a time to drive.
|
||||||
|
- the turn queue loops
|
||||||
|
- the rover will stop moving and stop all aux. motors if the turn switches to a different person
|
||||||
|
- admin
|
||||||
|
- admin authentication is required to access the driver page at all
|
||||||
|
- lockdown
|
||||||
|
- ONLY lockdown admins can access the driver page
|
||||||
|
- the future spectator page is DISABLED (not even a way to log into it)
|
||||||
|
|
||||||
|
# roles in the access control system:
|
||||||
|
- user
|
||||||
|
- default, for normal people who visit the site
|
||||||
|
- admin
|
||||||
|
- authentication needed
|
||||||
|
- can drive and view any rover when not on lockdown, no matter who is controlling it
|
||||||
|
- can switch modes (including switching to lockdown mode)
|
||||||
|
- lockdown admin
|
||||||
|
- authentication needed
|
||||||
|
- only works if said admin has lockdown: enabled in config
|
||||||
|
- can ALWAYS access and control EVERYTHING that there is to do on the site
|
||||||
|
|
||||||
|
# other things to keep in mind:
|
||||||
|
- in the future, there will be a Discord bot for community alerts and a few admin controls
|
||||||
|
- wherever the admin list is configured, there has to be a spot for their discord ID
|
||||||
|
- admin's discord IDs are linked to their admin name server-side
|
||||||
@@ -0,0 +1,27 @@
|
|||||||
|
# general idea
|
||||||
|
- uses the rover's reported battery full, warn, and urgent values
|
||||||
|
- will apply to all rovers individually
|
||||||
|
- uses the rover locking system
|
||||||
|
- completely server side
|
||||||
|
- always use the battery warn value as 0% battery and the full value as 100%
|
||||||
|
|
||||||
|
## first: the server-side server-wide event bus
|
||||||
|
- global server event bus
|
||||||
|
- used for realtime alerts between modules
|
||||||
|
- includes a way to tell where the event is coming from
|
||||||
|
|
||||||
|
## what will the battery manager do?
|
||||||
|
- watch each rover's battery charge number (reported in sensors)
|
||||||
|
- if the number reaches warn, fire an event on the event bus
|
||||||
|
- the UI will show a warning to users, independently based on sensor data
|
||||||
|
- once the rover is docked and charging, lock it.
|
||||||
|
- when the battery is fully charged, unlock it.
|
||||||
|
|
||||||
|
## while we're at it...
|
||||||
|
- add a lock reason to the rover locking system
|
||||||
|
- add the following to the rover roster in the UI session state:
|
||||||
|
- battery full #
|
||||||
|
- battery warn #
|
||||||
|
- battery urgent #
|
||||||
|
- in the rover roster UI component, add a display for battery percentage and lock reason.
|
||||||
|
- make the rover's background red when locked
|
||||||
@@ -0,0 +1,9 @@
|
|||||||
|
# general idea
|
||||||
|
- IR led on the front of each roomba
|
||||||
|
- connected to a pin on the pi
|
||||||
|
- made to be very directional
|
||||||
|
- each roomba has a different IR code
|
||||||
|
- use the omni reciever to get a shot
|
||||||
|
- keep score per user
|
||||||
|
- use the go alert system
|
||||||
|
- alert discord with a live score count
|
||||||
@@ -0,0 +1,35 @@
|
|||||||
|
# general idea
|
||||||
|
- discord bot run by the server
|
||||||
|
- use discord.js
|
||||||
|
- don't use discord's slash commands
|
||||||
|
- listen for commands the old fashioned way
|
||||||
|
- only admins with IDs set in the server config can use commands
|
||||||
|
- use nice looking embeds with colors for everything but messages
|
||||||
|
|
||||||
|
# features
|
||||||
|
|
||||||
|
## admin server management commands
|
||||||
|
- admins can lock and unlock rovers from discord
|
||||||
|
- admins can change the server access control mode
|
||||||
|
|
||||||
|
## announcements
|
||||||
|
- the bot has an announcements channel assigned in the server config. It will also have an announcement role ID to optionally ping. it will announce:
|
||||||
|
- when a rover is locked / unlocked (no ping)
|
||||||
|
- when the server access mode is changed (ping)
|
||||||
|
- the bot will have an admin ping role and alert channel. in here will be:
|
||||||
|
- rover activity (no pings)
|
||||||
|
- docking
|
||||||
|
- charging
|
||||||
|
- stopping charging
|
||||||
|
- undocking
|
||||||
|
- rover alerts (ping)
|
||||||
|
- rover comes online
|
||||||
|
- rover goes offline
|
||||||
|
- rover is at warn battery
|
||||||
|
- rover is at urgent battery
|
||||||
|
|
||||||
|
## chat bridge
|
||||||
|
- bridge between the server chat and discord chat
|
||||||
|
- use chat events that are on the bus already
|
||||||
|
|
||||||
|
|
||||||
@@ -0,0 +1,15 @@
|
|||||||
|
# general idea:
|
||||||
|
- control on / off switches and lights in home assistant from the web UI
|
||||||
|
- use this:
|
||||||
|
- https://www.npmjs.com/package/home-assistant-js-websocket/v/3.1.2
|
||||||
|
- remember to ignore updates that are of the same states, this library will give you a lot of those
|
||||||
|
- switches and lights are configured in the server's config file
|
||||||
|
- give each one a name (no description)
|
||||||
|
- auto detect a switch type or light type
|
||||||
|
- deliver list of lights to the UI through the session service
|
||||||
|
- show realtime on/off status of the switches / lights in the UI
|
||||||
|
- create a react component for the controls
|
||||||
|
- it will automatically create a control for each switch / light
|
||||||
|
|
||||||
|
## permissions:
|
||||||
|
- even if someone isnt assigned to a rover, they should be able to control the switches / lights
|
||||||
@@ -0,0 +1,13 @@
|
|||||||
|
# general idea
|
||||||
|
- The idea of this feature is to toggle night vision on the camera by turning on an LED from GPIO on the pi which will me mounted in front of the camera's light sensor
|
||||||
|
- with this, the camera will disable night vision when the LED is ON, and enable night vision when the LED is OFF.
|
||||||
|
- this will only really involve pi and webui programming. the server passes commands straight through.
|
||||||
|
- do not track the state of night vision, it is not needed.
|
||||||
|
|
||||||
|
## pi side
|
||||||
|
- new GPIO on/off control on GPIO 17, 27, or 22. any of these will work.
|
||||||
|
|
||||||
|
## web UI
|
||||||
|
- a new keyboard shortcut to toggle night vision
|
||||||
|
- a new button in the mobile UI to toggle night vision
|
||||||
|
- put it above the mobile horizontal servo slider
|
||||||
@@ -0,0 +1,6 @@
|
|||||||
|
## general idea
|
||||||
|
- new page
|
||||||
|
- designed for a very small screen
|
||||||
|
- not interactive at all
|
||||||
|
- a small summarized status of what's going on in the basement (where the rovers are)
|
||||||
|
- this small screen does not have space to show all info and all video for all rovers
|
||||||
@@ -0,0 +1,21 @@
|
|||||||
|
# general idea
|
||||||
|
- a help system that pops up onscreen
|
||||||
|
- changes based on control layout
|
||||||
|
- mobile / desktop
|
||||||
|
- on desktop especially, keybinds will dynamically update, based on what is actually assigned
|
||||||
|
|
||||||
|
# UI layout specifics:
|
||||||
|
- remake the current help component from scratch
|
||||||
|
- leave it in the place where it is, so people can always look at the help
|
||||||
|
- match the styling of the rest of the page
|
||||||
|
- make sure the help is easy to edit
|
||||||
|
- add a pop-up which displays underneath any other popup layers, which will display the help component in a large format
|
||||||
|
- make this pop-up show every time UNLESS the user checks "dont show again"
|
||||||
|
- use the settings system to store this in the browser
|
||||||
|
|
||||||
|
# what will the help explain?
|
||||||
|
- how to use the nickname and chat
|
||||||
|
- how to start driving the rover
|
||||||
|
- rover controls
|
||||||
|
- this is the biggest thing that will change in different layouts
|
||||||
|
- how it works technically (just a placeholder for now)
|
||||||
@@ -0,0 +1,36 @@
|
|||||||
|
# Video flow (SRT ingest, WHEP playback)
|
||||||
|
|
||||||
|
1. **Raspberry Pi Zero 2 W**
|
||||||
|
- `pi/bin/video-publisher.sh` captures the CSI camera with `rpicam-vid`/`libcamera-vid` using the onboard H.264 encoder (`--inline --profile baseline --bitrate …`).
|
||||||
|
- The Annex-B stream is piped into the stock FFmpeg package and pushed to the control server over SRT as MPEG‑TS: `ffmpeg -f h264 -i - -c copy -f mpegts "srt://<server>:9000?streamid=#!::r=rover-alpha,m=publish&latency=20&mode=caller&transtype=live&pkt_size=1316"`.
|
||||||
|
- Configuration lives in `/var/lib/roverd/video.env` (`PUBLISH_URL`, resolution, FPS, bitrate). `roverd` rewrites this file whenever the rover config changes, so onboarding a new rover is just flashing the SD card, setting its `name`, and plugging it into the trusted LAN—**no auth or per-rover server config is required on the Pi ⇄ server hop.**
|
||||||
|
|
||||||
|
2. **mediaMTX on the server**
|
||||||
|
- Single wildcard path handles every rover:
|
||||||
|
```yaml
|
||||||
|
paths:
|
||||||
|
"~^rover-(?P<id>[a-z0-9_-]+)$":
|
||||||
|
source: publisher
|
||||||
|
sourceOnDemand: no
|
||||||
|
sourceProtocol: srt
|
||||||
|
readBufferCount: 512
|
||||||
|
webrtcEnable: yes
|
||||||
|
webrtcMaxPlayoutDelay: 0
|
||||||
|
alwaysRemuxWhep: no
|
||||||
|
```
|
||||||
|
- Pis publish to `srt://<server>:9000` with the streamid above; mediaMTX auto-creates the path and fans it out over WHEP/WebRTC. If you expose playback under `/video/<id>` externally, let the reverse proxy rewrite that prefix back to `<id>` before forwarding to mediaMTX so the wildcard continues to match every rover.
|
||||||
|
- Only playback is gated: mediaMTX calls the Node server to validate JWTs on `/whep/rover-<id>`, so “locking” a stream is as simple as refusing to mint viewer tokens for that rover. Ingest stays unauthenticated because it lives on a secure LAN.
|
||||||
|
|
||||||
|
3. **Web clients**
|
||||||
|
- Tiny helper (React hook or vanilla class) that:
|
||||||
|
1. Requests a viewer token for rover `<id>`.
|
||||||
|
2. Issues a `POST` to `<mediamtx-host>/<rover-id>/whep` with `Authorization: Basic base64(token:token)` (token issued by the server when the client calls `video:request`).
|
||||||
|
3. Maintains auto-reconnect timers on ICE failure so dashboard widgets can come/go without reloading the page.
|
||||||
|
- Operator dashboard mounts one player tied to the assigned rover. The spectator view instantiates one player per tile, muting + pausing hidden elements to keep CPU usage sane even when every rover is shown simultaneously.
|
||||||
|
|
||||||
|
# Deployment checklist
|
||||||
|
|
||||||
|
- `pi/install_roverd.sh` installs the `video-publisher` helper, drops `/var/lib/roverd/video.env`, and pulls in `libcamera-apps` + `ffmpeg` from apt. No custom FFmpeg, no WHIP builds, no extra config—Pis become plug-and-play.
|
||||||
|
- `roverd` derives `media.publishUrl` automatically from `serverUrl`: `srt://<server>:9000?streamid=#!::r=<name>,m=publish&latency=20&mode=caller&transtype=live&pkt_size=1316`. The media supervisor rewrites `video.env` and manages `video-publisher.service` whenever you hit “Restart Camera” or change `/etc/roverd.yaml`.
|
||||||
|
- The server’s mediaMTX config switches to the wildcard block above, leaves SRT ingest open, and enforces JWTs only on WHEP viewers. Fan-out stays inside mediaMTX so every browser sees the same low-latency stream (≈250–350 ms glass-to-glass).
|
||||||
|
- Adding a rover = flash SD → set `/etc/roverd.yaml` (`name`, `serverUrl`, camera knobs if needed) → boot it. The server auto-discovers the new `rover-<id>` stream with zero manual edits.
|
||||||
@@ -0,0 +1,53 @@
|
|||||||
|
# nicknames:
|
||||||
|
- users will be able to have and set nicknames
|
||||||
|
- a user's nickname will store in the browser using persistence.js
|
||||||
|
- the user's nickname should probably just be in socket.data.nickname
|
||||||
|
- on connection, the web UI will tell the server "this is my nickname"
|
||||||
|
- completely enforced by the web UI
|
||||||
|
- replace any place in the web UI that shows a socket ID with the nickname
|
||||||
|
- create a small react component which can set your nickname and save it
|
||||||
|
|
||||||
|
# user list in session data:
|
||||||
|
- add a list of users to the session data for the web UI
|
||||||
|
- contains for each user:
|
||||||
|
- socket ID
|
||||||
|
- nickname
|
||||||
|
- the rover that they are driving
|
||||||
|
- their role (user, admin, lockdown, spectator)
|
||||||
|
|
||||||
|
# user list:
|
||||||
|
- a list of users
|
||||||
|
- new react component
|
||||||
|
- uses user list with nicknames from session data
|
||||||
|
- show their nickname and whether or not they are an admin
|
||||||
|
|
||||||
|
# chat:
|
||||||
|
- server side chat system:
|
||||||
|
- don't store message history
|
||||||
|
- emit an event to the event bus for each message
|
||||||
|
- eventually will be forwarded to discord through a bot
|
||||||
|
- listen to chat message events on the bus
|
||||||
|
- eventually the discord bot will also send messages to the server chat
|
||||||
|
- profanity filter
|
||||||
|
- spam filter
|
||||||
|
- repeated words
|
||||||
|
- keymashing
|
||||||
|
- etc
|
||||||
|
- somehow link chat messages to rovers
|
||||||
|
- in the future rovers will have TTS onboard, and will speak chat messages only from the person driving the rover
|
||||||
|
- chat on the UI
|
||||||
|
- new react component
|
||||||
|
- show messages as they come in:
|
||||||
|
- time (just like 19:23), sender nickname, rover they are driving, message
|
||||||
|
- pressing enter on the keyboard will pause rover control, and focus the chat box.
|
||||||
|
- pressing enter again after typing will send the message
|
||||||
|
|
||||||
|
|
||||||
|
# layout and styling for new UI elements:
|
||||||
|
- use index.css styles to match the new stuff to the current UI
|
||||||
|
- just edit the desktop page for now
|
||||||
|
- layout:
|
||||||
|
- in the left column, a new row in between video and logs
|
||||||
|
- 50/50 split between:
|
||||||
|
- user list and nickname entry
|
||||||
|
- chat history and chat box
|
||||||
@@ -0,0 +1,58 @@
|
|||||||
|
## multi roomba rover
|
||||||
|
|
||||||
|
a website where people can control multiple irobot create 2 robots in real time 100% responsively
|
||||||
|
with a raspberry pi zero 2 W on each roomba, along with a raspberry pi camera
|
||||||
|
a central nodejs control server will tell the raspberry pis what to do with the roomba
|
||||||
|
|
||||||
|
## the pi side (roomba side)
|
||||||
|
- pi's onboard UART is hooked up to the roomba's serial port
|
||||||
|
- another GPIO pin connected to the roomba's BRC pin
|
||||||
|
- pull it low for one second every minute to keep the roomba awake
|
||||||
|
- streams the rpi camera over webRTC with mediamtx
|
||||||
|
- streams roomba sensor group 100 to the server
|
||||||
|
- sensor streaming is required and important, but is allowed to falter sometimes
|
||||||
|
- listens for roomba commands from the server
|
||||||
|
- commands NEED to happen
|
||||||
|
- roomba control program needs to be simple, lightweight, and 100% responsive
|
||||||
|
|
||||||
|
## pi -> server communication
|
||||||
|
- stateless
|
||||||
|
- streaming based
|
||||||
|
- on connection, the pi will send the following info:
|
||||||
|
- rover's name
|
||||||
|
- motor enable / disable
|
||||||
|
- vacuum
|
||||||
|
- main brush
|
||||||
|
- side brush
|
||||||
|
- battery full number
|
||||||
|
- battery warning number
|
||||||
|
- battery urgent number
|
||||||
|
|
||||||
|
## nodejs server
|
||||||
|
- KISS
|
||||||
|
- decode the sensor data from each roomba
|
||||||
|
- can support multiple roombas connected from the ground up
|
||||||
|
- keep it simple, worry about getting the pi comms right.
|
||||||
|
- but the server DOES have to exist for testing
|
||||||
|
- IS the web server, hosts an entire static folder for the web UI
|
||||||
|
|
||||||
|
## server -> web UI communication
|
||||||
|
- socket.io
|
||||||
|
- don't do anything fancy with the socket.io setup
|
||||||
|
- it works fine out of the box, we will optimize it later
|
||||||
|
|
||||||
|
## the web UI
|
||||||
|
- KISS
|
||||||
|
- plain old html. no styling even. just bare minimum for testing
|
||||||
|
- what it needs to do:
|
||||||
|
- allow user to select the roomba from a list
|
||||||
|
- make the selected roomba drive with WASD
|
||||||
|
- have buttons to set the OI mode, and tell the roomba to dock
|
||||||
|
- show a plain list of the sensor data from the selected roomba
|
||||||
|
|
||||||
|
### general javascript programming guidelines (applies to the web UI too)
|
||||||
|
- everything ES6
|
||||||
|
- one entrypoint file in the web UI
|
||||||
|
- everything modular
|
||||||
|
- everything easy to read, understand, and work on
|
||||||
|
- comment where you think is best to describe whats going on
|
||||||
@@ -0,0 +1,13 @@
|
|||||||
|
# general idea
|
||||||
|
- servo connected to pin 19 on the pi
|
||||||
|
- it will allow the camera to look up and down
|
||||||
|
- use go-rpio
|
||||||
|
- it has a hardware PWM implementation
|
||||||
|
- mechanical design will have limits
|
||||||
|
- set the limits in roverd config
|
||||||
|
- control the servo from the web browser
|
||||||
|
- expose both a slider and buttons for "nudging" it up and down
|
||||||
|
|
||||||
|
## verification of functionality before full implementation
|
||||||
|
- make sure that the go program can properly use the GPIO correctly before full implementation
|
||||||
|
- modify the roverd installer to add the needed stuff to enable PWM access
|
||||||
@@ -0,0 +1,23 @@
|
|||||||
|
# general idea
|
||||||
|
- the pi's microphone needs to be streamed to users
|
||||||
|
- as part of the rover's video feed
|
||||||
|
- the pi model is a zero 2 W
|
||||||
|
- it CANNOT handle transcoding the audio onboard, I have tried.
|
||||||
|
- the server is super powerful and can handle transcoding
|
||||||
|
- it is NOT acceptable to add more than 100ms of video latency from glass to glass
|
||||||
|
- it is NOT acceptable to break video streaming for pis that don't have microphones
|
||||||
|
|
||||||
|
### OLD, BAD, but working audio configuration that worked for the microphone
|
||||||
|
```
|
||||||
|
-f alsa -guess_layout_max 0 -thread_queue_size 2048 -ac 1 -ar 48000 -i plughw:0,0 \
|
||||||
|
-map 1:a:0 \
|
||||||
|
-c:a libopus -b:a 24000 -compression_level 0 -application voip -frame_duration 60 \
|
||||||
|
-ac:a 1 -ar:a 48000 -af "pan=1c|c0=c0,volume=20dB"
|
||||||
|
```
|
||||||
|
- the raspberry pi CANNOT HANDLE AUDIO ENCODING onboard. this is just to show you how to use the mic.
|
||||||
|
|
||||||
|
|
||||||
|
sudo systemctl restart roverd
|
||||||
|
#sudo systemctl restart audio-capture
|
||||||
|
sudo systemctl restart video-publisher
|
||||||
|
sudo cat /var/lib/roverd/video.env
|
||||||
@@ -0,0 +1,27 @@
|
|||||||
|
## raspberry pi
|
||||||
|
<!-- - need to throttle sensor sending, maybe only send one in every 5 packets. -->
|
||||||
|
|
||||||
|
## 3d models for camera:
|
||||||
|
- https://www.thingiverse.com/thing:2873677
|
||||||
|
- https://www.printables.com/model/356894-raspberry-camera-module-with-automatic-ir-cut-swit
|
||||||
|
- https://www.thingiverse.com/thing:4514531
|
||||||
|
|
||||||
|
## todo
|
||||||
|
<!-- 1. battery manager -->
|
||||||
|
<!-- 2. pi-side sensor throttle (1/5th) -->
|
||||||
|
<!-- 3. convert rover roster to a reusable component, unbake it from telemetry and admin panels -->
|
||||||
|
<!-- 4. room cameras -->
|
||||||
|
4. fix gamepad input - more complex axis assignment? division, combined axes?
|
||||||
|
<!-- 5. home assistant controls -->
|
||||||
|
<!-- 6. nicknames -->
|
||||||
|
<!-- 7. discord bot -->
|
||||||
|
<!-- 8. turns mode display -->
|
||||||
|
<!-- 9. online user list -->
|
||||||
|
<!-- 10. chat -->
|
||||||
|
<!-- 9. discord invite button -->
|
||||||
|
<!-- 10. redo both mobile layouts -->
|
||||||
|
<!-- 11. redo spectator view (last) -->
|
||||||
|
<!-- 21. finally.. set the favicon and title -->
|
||||||
|
22. rover snapshots freezing and never coming back
|
||||||
|
23. rover snapshot -> video switching needs to be smoother, no black flash. connect and play before showing.
|
||||||
|
24. rover snapshots delayed (not just because of framerate)
|
||||||
@@ -0,0 +1,69 @@
|
|||||||
|
# general idea
|
||||||
|
- a modular control system
|
||||||
|
- every piece of the UI that uses rover controls will go through this, including onscreen click buttons
|
||||||
|
- allows for dynamic control labels based on saved settings
|
||||||
|
- allow a place to assign OI command macros
|
||||||
|
- like the one that the drive button uses now
|
||||||
|
- allows realtime responsive control of the rover that you are driving
|
||||||
|
|
||||||
|
## some sort of system for the site to save settings per browser
|
||||||
|
- needs to be extensible
|
||||||
|
- future things will use it
|
||||||
|
- expose functions like saveSettings and loadSettings
|
||||||
|
- other parts of the UI will need to save and load settings using this function
|
||||||
|
- use cookies
|
||||||
|
|
||||||
|
## keyboard controls:
|
||||||
|
- remappable by new component in settings tab
|
||||||
|
- key mappings will save
|
||||||
|
- controls for the keyboard
|
||||||
|
- driving
|
||||||
|
- WASD blended for tank steering
|
||||||
|
- hold backslash to move faster
|
||||||
|
- hold right shift to move slower
|
||||||
|
- aux motors
|
||||||
|
- main brush
|
||||||
|
- hold O to move it forward at speed 127
|
||||||
|
- hold L to move it backward at speed -127
|
||||||
|
- side brush
|
||||||
|
- hold P to move it forward at speed 127
|
||||||
|
- hold ; to move it backward at speed -70
|
||||||
|
- vacuum motor
|
||||||
|
- hold [ to move it at speed 127
|
||||||
|
- hold ' to move it at speed 50
|
||||||
|
- ALL AUX MOTORS
|
||||||
|
- hold . to move at full speed forward (127)
|
||||||
|
- camera movement
|
||||||
|
- hold I to look up
|
||||||
|
- hold K to look down
|
||||||
|
- dock and drive hotkeys
|
||||||
|
- G for drive (use same 3 part macro as the drive button)
|
||||||
|
- H for dock (seek dock command)
|
||||||
|
|
||||||
|
## Mobile controls
|
||||||
|
- 2 different layouts, already implemented just needs improved.
|
||||||
|
### mobile landscape
|
||||||
|
- a video game style layout
|
||||||
|
- on the left, buttons that you hold to operate the aux motors
|
||||||
|
- in the middle, is the rover video component.
|
||||||
|
- on the right, an area for a floating joystick and above is a unified control to see and change the rover's mode (drive or dock)
|
||||||
|
### mobile portrait
|
||||||
|
- designed to be driven vertically with 2 hands
|
||||||
|
- mostly fine already
|
||||||
|
- rover video at the top
|
||||||
|
- then below it, is a section with the aux motor buttons on the left, and the floating joystick are on the right.
|
||||||
|
|
||||||
|
|
||||||
|
## Gamepad controls:
|
||||||
|
- use some sort of react thing that makes it easy to use the web gamepad stuff
|
||||||
|
- left joystick for movement, right joystick for moving camera up / down
|
||||||
|
- both fully analog
|
||||||
|
- right trigger for the main brush motor
|
||||||
|
- fully analog
|
||||||
|
- press right bumper to switch it to reverse
|
||||||
|
- left trigger for the side brush motor
|
||||||
|
- fully analog
|
||||||
|
- press left bumper to switch it to reverse
|
||||||
|
- hold the right face button to run the vacuum motor
|
||||||
|
- hold the lower face button to run all aux motors forward
|
||||||
|
- left and right on the Dpad switch between drive and dock modes
|
||||||
@@ -0,0 +1,10 @@
|
|||||||
|
# general idea
|
||||||
|
- convert room cameras from webrtc h264 streams to 4fps jpegs sent over socket.io
|
||||||
|
|
||||||
|
## requirements
|
||||||
|
- make updated services for server and pi room cams
|
||||||
|
- no backwards compatability is needed
|
||||||
|
- room cameras are both 4:3, make sure they show as such in the UI
|
||||||
|
- remove anything related to the webRTC room cams
|
||||||
|
- send the jpegs efficiently (as binary)
|
||||||
|
- make sure the room cams are still authed as they are now
|
||||||
@@ -0,0 +1,14 @@
|
|||||||
|
# general idea
|
||||||
|
- add support for multiple room cameras through the server to the UI
|
||||||
|
|
||||||
|
## what needs to happen on the server:
|
||||||
|
- add support for a path of room cameras on mediamtx
|
||||||
|
- /room/<camera_name>
|
||||||
|
- CANNOT interfere with rover cameras (/<rover_name>)
|
||||||
|
- needs to use the same auth system as the rover camera
|
||||||
|
- room cameras will be streamed to the server over SRT
|
||||||
|
|
||||||
|
## what needs to happen in the web UI
|
||||||
|
- users should be able to see all room cameras, even if not assigned to a rover
|
||||||
|
- automatically add a room camera player for each room camera
|
||||||
|
- make a component that shows all room cameras
|
||||||
@@ -0,0 +1,51 @@
|
|||||||
|
# server code structure:
|
||||||
|
The server's internal structure will be modular, and by modular i mean completely modular,
|
||||||
|
the modules will import what they need from other modules, and export what other modules will need from them.
|
||||||
|
the entrypoint file will contain nothing but a long list of `require('')`s for all of the modules in the proper order
|
||||||
|
|
||||||
|
for example, A service which automatically assigns a newly connected user to a roomba that isn't in use.
|
||||||
|
this service would import the roomba list from whatever other service contains it, and import the global socket.io server instance. It will add its own io.on('connection') to the socket.io instance, which contains the logic for assigning users to roombas.
|
||||||
|
|
||||||
|
- modular code structure
|
||||||
|
- one folder for each of these categories
|
||||||
|
- globals
|
||||||
|
- GLOBALS ARE: "static" parts of the server that don't contain any interactive logic.
|
||||||
|
- where the express, websocket, and socket.io instances will be
|
||||||
|
- other global things
|
||||||
|
- services
|
||||||
|
- SERVICES ARE: parts of the program that are part of the interaction pipeline.
|
||||||
|
- contains things like the roomba manager
|
||||||
|
- will also in the future contain other things like a discord bot, home assistant integration, etc
|
||||||
|
- anything with a large amount of controlling logic should be in here
|
||||||
|
- helpers
|
||||||
|
- HELPERS ARE: parts of the program that other modules only pull helper funcions or classes from.
|
||||||
|
- if a function or class is dedicated to a service, it should NOT be in a helper.
|
||||||
|
- contains passive helpers
|
||||||
|
- things like the logger system
|
||||||
|
- no "service" logic in here, only things that are passively pulled out and used inside other modules
|
||||||
|
- one entrypoint file that contains NOTHING but `require('')`s.
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
# server <-> web client logistics
|
||||||
|
|
||||||
|
I want each connected roomba to have a list of drivers. if a socket is not in this list, they are not allowed to drive the roomba. I will be adding admins, authentication, a turns system, and automatic roomba assignment later so it is important that we start with this system in place.
|
||||||
|
a roomba's list of drivers will be completely managed by the server, users should not be able to change driver lists, even in a hacky way.
|
||||||
|
|
||||||
|
Each user will also have to see sensor data from the roomba that they are controlling. I want it to be done in this way:
|
||||||
|
- for each connected roomba, there is a socket.io room where all of the sensor data is streamed out to clients
|
||||||
|
- the client will be added to the room, where they can see all of the active roomba's sensor data
|
||||||
|
- ALSO for future use, it needs to work properly if a socket is subscribed to all rooms, in the future there will be a spectator page which can view all of the roombas at once.
|
||||||
|
|
||||||
|
I want there to be a ground up system where I can set the entire service to four different modes:
|
||||||
|
- open (anyone can drive, roombas are assigned randomly. if they are all full, two people will be controlling the same roomba)
|
||||||
|
- turns (anyone can drive, roombas are assigned randomly. if they are all full, each person gets one minute on their selected roomba)
|
||||||
|
- admin (only authenticated admins can log in, anyone can still view but no one is allowed to drive but admins)
|
||||||
|
- lockdown (only "lockdown" admins can view or drive. no one else can view, not even spectators).
|
||||||
|
only admins can change modes.
|
||||||
|
|
||||||
|
do NOT implement any authentication stuff yet, just add a stub service with places set up to put the auth. logic
|
||||||
|
|
||||||
|
# web client code structure:
|
||||||
|
- the same as the server's code structure
|
||||||
|
- all ES6
|
||||||
@@ -0,0 +1,44 @@
|
|||||||
|
# spectate system overhaul
|
||||||
|
- rip out and remake the entire page
|
||||||
|
- basically keep nothing from it
|
||||||
|
- right now the spectator page is a mess, and hasn't been updated in a while
|
||||||
|
- does not work properly, aside from the page being out of date
|
||||||
|
- the spectate page should use modules from the driver page
|
||||||
|
- there should be no such thing as a "spectate mode" on the driver page. If someone wants to spectate, they must go to the spectator page.
|
||||||
|
|
||||||
|
## rules of the spectator page
|
||||||
|
- spectators are not allowed to drive
|
||||||
|
- spectators are not allowed to chat
|
||||||
|
- spectators are not allowed to have nicknames
|
||||||
|
- spectators are not allowed to login as admin
|
||||||
|
- these are the permissions for spectators based on the sever's access control mode
|
||||||
|
- open
|
||||||
|
- spectators can spectate
|
||||||
|
- turns
|
||||||
|
- spectators can spectate
|
||||||
|
- admin
|
||||||
|
- spectators can spectate
|
||||||
|
- lockdown
|
||||||
|
- spectators can NOT spectate
|
||||||
|
|
||||||
|
|
||||||
|
## what should be on the spectator page?
|
||||||
|
- for each rover
|
||||||
|
- video component
|
||||||
|
- current driver
|
||||||
|
- sensor telemetry component
|
||||||
|
- logs
|
||||||
|
- room cameras
|
||||||
|
- online user list
|
||||||
|
|
||||||
|
## layout and styling of the spectator page
|
||||||
|
- use same styling style as the driver page
|
||||||
|
- feel free to add spectator role checks to hide the features that spectators can't use in certain components.
|
||||||
|
- mainly designed for a 4:3 monitor
|
||||||
|
- make use of the vertical space
|
||||||
|
- a row of columns, one for each rover at the top
|
||||||
|
- below the row of columns:
|
||||||
|
- 50/50 split
|
||||||
|
- user list
|
||||||
|
- chat
|
||||||
|
- room cameras
|
||||||
@@ -0,0 +1,67 @@
|
|||||||
|
# general idea:
|
||||||
|
- very compact UI
|
||||||
|
- utilitarian
|
||||||
|
- function before pretty
|
||||||
|
- tiny text, too.
|
||||||
|
- no large margins or padding. very little wasted space
|
||||||
|
- up to 1 tailwind unit unless otherwise needed
|
||||||
|
- black background, gray cards, white or gray text
|
||||||
|
- tailwind
|
||||||
|
- no title or bar at the top, vertical screen space is precious on all layouts
|
||||||
|
- alerts should be moved to a tiny toast popup that will show at the top center of the screen
|
||||||
|
|
||||||
|
# three different layouts for the driver page
|
||||||
|
- desktop
|
||||||
|
- rover video central, no title or anything above it
|
||||||
|
- overlayed on rover video (HUD like in a video game)
|
||||||
|
- build the HUD into the video tile
|
||||||
|
- because everywhere the video is, the HUD should be too
|
||||||
|
- shows a dot that blinks on each sensor frame
|
||||||
|
- shows wheel drops and bumpers
|
||||||
|
- shows a big "loud" warning overlay during an overcurrent
|
||||||
|
- simple, easy to understand battery bar UNDERNEATH the video:
|
||||||
|
- takes in battery full, warn, and urgent values
|
||||||
|
- treat full as 100% and warn as 0%
|
||||||
|
- make the bar flash red when its 0% or lower
|
||||||
|
- on left side of rover video:
|
||||||
|
- sensor data, visualized.
|
||||||
|
- battery charge out of capacity (1763/2068)
|
||||||
|
- charging status
|
||||||
|
- OI mode
|
||||||
|
- battery voltage (in volts not mv)
|
||||||
|
- battery current
|
||||||
|
- a teeny show of the raw data (because we get it on the client and it looks cool)
|
||||||
|
- on right side of rover video
|
||||||
|
- a simple, easy control panel with the following actions:
|
||||||
|
- a button that runs the start OI, dock, then full commands (in that order)
|
||||||
|
- also contains a little status of whether or not the rover is in "driving mode" (OI in full mode)
|
||||||
|
- a button that tells the rover to seek dock
|
||||||
|
- explains that you should be straight in front of, and about a foot from the dock for a successful attempt
|
||||||
|
- contains two status indicators:
|
||||||
|
- docked / not docked (homebase true)
|
||||||
|
- charging / not charging (when chargingstate isn't "not charging" the rover is charging)
|
||||||
|
- below the upper control and video row
|
||||||
|
- center:
|
||||||
|
- a place for the room camera (not implemented yet)
|
||||||
|
- left:
|
||||||
|
- admin login and controls
|
||||||
|
- right:
|
||||||
|
- logs
|
||||||
|
- mobile in portrait mode
|
||||||
|
- rover video at the top, with HUD
|
||||||
|
- just below:
|
||||||
|
- joystick with buttons for aux. motor controls
|
||||||
|
- have a button to run all aux motors forward at max speed
|
||||||
|
- also have buttons to run motors forward / backward individually
|
||||||
|
- probably use some premade react joystick
|
||||||
|
- scroll down more to see:
|
||||||
|
- the same simple mode control buttons from the desktop layout
|
||||||
|
- also needs to have the admin login and controls
|
||||||
|
- needs to have all the functionality as the desktop page
|
||||||
|
- mobile in landscape mode
|
||||||
|
- made for videogame-like control of the rover
|
||||||
|
- control buttons and aux motors on the left
|
||||||
|
- rover video in the middle
|
||||||
|
- joystick on the right
|
||||||
|
- again, scroll down to see more
|
||||||
|
- also has to do everything
|
||||||
@@ -0,0 +1,14 @@
|
|||||||
|
# general idea
|
||||||
|
- users can choose from preset, and create UI themes
|
||||||
|
- the default theme is the current in place theme
|
||||||
|
- themes will NOT change margins, borders, padding, spacing, etc.
|
||||||
|
- should be as simple as changing the global css stuff
|
||||||
|
- themes can change colors, fonts, border radius, borders, etc
|
||||||
|
|
||||||
|
## technical stuff
|
||||||
|
- save theme info in
|
||||||
|
|
||||||
|
## UI for themeing
|
||||||
|
- allow people to tune every theme element that there is to tune
|
||||||
|
- allow people to export themes as json files, and import them
|
||||||
|
-
|
||||||
@@ -0,0 +1,23 @@
|
|||||||
|
# general idea
|
||||||
|
- a bunch of small UI adjustments
|
||||||
|
- always keep styling consistent to the rest of the page
|
||||||
|
|
||||||
|
## everywhere
|
||||||
|
- add an audio stream status to the rover video panel, only show it if the rover has the separate audio stream
|
||||||
|
- make the room camera panels save their stack setting per instance of the panel in the web UI's settings storage system
|
||||||
|
- remove the status bar below room cameras
|
||||||
|
- put it in the corner of the feed, like the other video panels
|
||||||
|
- make the styling of the room camera panels match everything else
|
||||||
|
- when the user list switches to turns, always show the plain user list, but below the turns info
|
||||||
|
|
||||||
|
## on the spectator page
|
||||||
|
- rover video panels
|
||||||
|
- no more telemetry bar below each rover video
|
||||||
|
- ovelay the tiny telemetry summary on the left side of the video, with the rest of the HUD elements
|
||||||
|
- including the battery bar, make it vertical on the right side of the video
|
||||||
|
- no more title bar above each rover video
|
||||||
|
- the rover's name is already on the HUD
|
||||||
|
- add current driver to the HUD
|
||||||
|
- sidebar
|
||||||
|
- shrink the logs panel
|
||||||
|
- add an actual rover list to the top of the sidebar
|
||||||
@@ -1,9 +0,0 @@
|
|||||||
[env:esp32s3]
|
|
||||||
platform = espressif32
|
|
||||||
board = esp32-s3-devkitc-1
|
|
||||||
framework = arduino
|
|
||||||
monitor_speed = 115200
|
|
||||||
|
|
||||||
; Enable verbose FreeRTOS symbols in Arduino builds (optional but handy for demos)
|
|
||||||
build_flags =
|
|
||||||
-DCORE_DEBUG_LEVEL=3
|
|
||||||
@@ -0,0 +1,28 @@
|
|||||||
|
# Room camera snapshot service
|
||||||
|
|
||||||
|
Lightweight systemd service that serves a JPEG snapshot from any MJPEG-capable webcam (most USB webcams) at 4 fps. The server pulls `http://<host>:8080/snapshot.jpg` for room cams.
|
||||||
|
|
||||||
|
## Files
|
||||||
|
- `room-cam-snapshot.sh` – ffmpeg + simple HTTP server wrapper
|
||||||
|
- `room-cam.service` – systemd unit template
|
||||||
|
|
||||||
|
## Usage
|
||||||
|
1) Copy the service into place (adjust path/env as needed):
|
||||||
|
```bash
|
||||||
|
sudo cp roomcam-service/room-cam.service /etc/systemd/system/room-cam.service
|
||||||
|
sudo systemctl daemon-reload
|
||||||
|
sudo systemctl enable --now room-cam.service
|
||||||
|
```
|
||||||
|
2) Override defaults via `Environment=` in the unit or drop-ins:
|
||||||
|
- `DEVICE=/dev/video0`
|
||||||
|
- `RESOLUTION=640x480`
|
||||||
|
- `QUALITY=5` (ffmpeg MJPEG quality; lower is higher quality)
|
||||||
|
- `PORT=8080`
|
||||||
|
- `WORKDIR=/run/roomcam`
|
||||||
|
- `INPUT_FORMAT=mjpeg` (use `bayer_grbg8` for OV534/raw Bayer cams; aliases `GRBG`/`grbg` are accepted)
|
||||||
|
3) Point the server `roomCameras[].url` to `http://<host>:8080/snapshot.jpg`.
|
||||||
|
|
||||||
|
Notes:
|
||||||
|
- The camera runs at its native MJPEG frame rate; the server polls snapshots at ~4 fps, so no extra filtering is applied here.
|
||||||
|
|
||||||
|
To run outside systemd, just execute `./room-cam-snapshot.sh` with any overrides.
|
||||||
@@ -0,0 +1,159 @@
|
|||||||
|
#!/usr/bin/env bash
|
||||||
|
set -euo pipefail
|
||||||
|
|
||||||
|
# Configurable via environment
|
||||||
|
export DEVICE="${DEVICE:-/dev/video0}"
|
||||||
|
export RESOLUTION="${RESOLUTION:-640x480}"
|
||||||
|
export QUALITY="${QUALITY:-10}" # ffmpeg MJPEG quality (lower is better)
|
||||||
|
export PORT="${PORT:-8088}"
|
||||||
|
export WORKDIR="${WORKDIR:-/run/roomcam}"
|
||||||
|
# Optional: set INPUT_FORMAT=bayer_grbg8 to transcode raw Bayer cams (e.g., OV534) to JPEG.
|
||||||
|
export INPUT_FORMAT="${INPUT_FORMAT:-mjpeg}"
|
||||||
|
export MJPEG_FPS="${MJPEG_FPS:-15}"
|
||||||
|
export MJPEG_QUALITY="${MJPEG_QUALITY:-8}"
|
||||||
|
|
||||||
|
mkdir -p "${WORKDIR}"
|
||||||
|
SNAPSHOT_PATH="${WORKDIR}/snapshot.jpg"
|
||||||
|
rm -f "${SNAPSHOT_PATH}"
|
||||||
|
# push
|
||||||
|
cleanup() {
|
||||||
|
[[ -n "${HTTP_PID:-}" ]] && kill "${HTTP_PID}" 2>/dev/null || true
|
||||||
|
}
|
||||||
|
trap cleanup EXIT
|
||||||
|
trap 'exit 0' SIGTERM INT
|
||||||
|
|
||||||
|
cat > "${WORKDIR}/mjpeg_server.py" <<'PY'
|
||||||
|
import os
|
||||||
|
import threading
|
||||||
|
import time
|
||||||
|
import subprocess
|
||||||
|
from http.server import BaseHTTPRequestHandler, ThreadingHTTPServer
|
||||||
|
|
||||||
|
DEVICE = os.environ.get("DEVICE", "/dev/video0")
|
||||||
|
RESOLUTION = os.environ.get("RESOLUTION", "640x480")
|
||||||
|
INPUT_FORMAT = os.environ.get("INPUT_FORMAT", "mjpeg")
|
||||||
|
MJPEG_FPS = os.environ.get("MJPEG_FPS", "15")
|
||||||
|
MJPEG_QUALITY = os.environ.get("MJPEG_QUALITY", "8")
|
||||||
|
WORKDIR = os.environ.get("WORKDIR", "/run/roomcam")
|
||||||
|
SNAPSHOT_PATH = os.path.join(WORKDIR, "snapshot.jpg")
|
||||||
|
|
||||||
|
FFMPEG_INPUT_ARGS = [
|
||||||
|
"-f", "v4l2",
|
||||||
|
"-input_format", INPUT_FORMAT,
|
||||||
|
"-video_size", RESOLUTION,
|
||||||
|
"-i", DEVICE,
|
||||||
|
]
|
||||||
|
FFMPEG_FILTERS = []
|
||||||
|
if INPUT_FORMAT.startswith("bayer_"):
|
||||||
|
FFMPEG_FILTERS = ["-pix_fmt", "yuv420p"]
|
||||||
|
|
||||||
|
FRAME_LOCK = threading.Lock()
|
||||||
|
FRAME_EVENT = threading.Event()
|
||||||
|
LATEST_FRAME = b""
|
||||||
|
|
||||||
|
def spawn_mjpeg():
|
||||||
|
cmd = [
|
||||||
|
"/usr/bin/ffmpeg",
|
||||||
|
"-loglevel", "warning", "-nostats",
|
||||||
|
*FFMPEG_INPUT_ARGS,
|
||||||
|
*FFMPEG_FILTERS,
|
||||||
|
"-r", str(MJPEG_FPS),
|
||||||
|
"-q:v", str(MJPEG_QUALITY),
|
||||||
|
"-f", "mjpeg",
|
||||||
|
"-",
|
||||||
|
]
|
||||||
|
return subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=subprocess.DEVNULL)
|
||||||
|
|
||||||
|
def update_frame(frame_bytes):
|
||||||
|
global LATEST_FRAME
|
||||||
|
with FRAME_LOCK:
|
||||||
|
LATEST_FRAME = frame_bytes
|
||||||
|
FRAME_EVENT.set()
|
||||||
|
try:
|
||||||
|
with open(SNAPSHOT_PATH, "wb") as fh:
|
||||||
|
fh.write(frame_bytes)
|
||||||
|
except OSError:
|
||||||
|
pass
|
||||||
|
|
||||||
|
def frame_reader():
|
||||||
|
while True:
|
||||||
|
proc = spawn_mjpeg()
|
||||||
|
buffer = b""
|
||||||
|
try:
|
||||||
|
while True:
|
||||||
|
chunk = proc.stdout.read(8192)
|
||||||
|
if not chunk:
|
||||||
|
break
|
||||||
|
buffer += chunk
|
||||||
|
while True:
|
||||||
|
start = buffer.find(b"\xff\xd8")
|
||||||
|
end = buffer.find(b"\xff\xd9", start + 2)
|
||||||
|
if start == -1 or end == -1:
|
||||||
|
break
|
||||||
|
frame = buffer[start : end + 2]
|
||||||
|
buffer = buffer[end + 2 :]
|
||||||
|
update_frame(frame)
|
||||||
|
finally:
|
||||||
|
try:
|
||||||
|
proc.kill()
|
||||||
|
except Exception:
|
||||||
|
pass
|
||||||
|
time.sleep(1)
|
||||||
|
|
||||||
|
class Handler(BaseHTTPRequestHandler):
|
||||||
|
def do_GET(self):
|
||||||
|
if self.path == "/" or self.path == "/snapshot.jpg":
|
||||||
|
with FRAME_LOCK:
|
||||||
|
frame = LATEST_FRAME
|
||||||
|
if not frame:
|
||||||
|
self.send_error(404, "snapshot missing")
|
||||||
|
return
|
||||||
|
self.send_response(200)
|
||||||
|
self.send_header("Content-Type", "image/jpeg")
|
||||||
|
self.send_header("Content-Length", str(len(frame)))
|
||||||
|
self.end_headers()
|
||||||
|
self.wfile.write(frame)
|
||||||
|
return
|
||||||
|
|
||||||
|
if self.path == "/stream.mjpg":
|
||||||
|
self.send_response(200)
|
||||||
|
self.send_header("Content-Type", "multipart/x-mixed-replace; boundary=frame")
|
||||||
|
self.end_headers()
|
||||||
|
try:
|
||||||
|
while True:
|
||||||
|
FRAME_EVENT.wait(timeout=2)
|
||||||
|
FRAME_EVENT.clear()
|
||||||
|
with FRAME_LOCK:
|
||||||
|
frame = LATEST_FRAME
|
||||||
|
if not frame:
|
||||||
|
continue
|
||||||
|
header = (
|
||||||
|
b"--frame\r\n"
|
||||||
|
b"Content-Type: image/jpeg\r\n"
|
||||||
|
+ f"Content-Length: {len(frame)}\r\n\r\n".encode("ascii")
|
||||||
|
)
|
||||||
|
self.wfile.write(header)
|
||||||
|
self.wfile.write(frame)
|
||||||
|
self.wfile.write(b"\r\n")
|
||||||
|
except BrokenPipeError:
|
||||||
|
pass
|
||||||
|
return
|
||||||
|
|
||||||
|
self.send_error(404, "not found")
|
||||||
|
|
||||||
|
def log_message(self, format, *args):
|
||||||
|
return
|
||||||
|
|
||||||
|
def main():
|
||||||
|
threading.Thread(target=frame_reader, daemon=True).start()
|
||||||
|
addr = ("0.0.0.0", int(os.environ.get("PORT", "8088")))
|
||||||
|
ThreadingHTTPServer(addr, Handler).serve_forever()
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
main()
|
||||||
|
PY
|
||||||
|
|
||||||
|
/usr/bin/python3 -u "${WORKDIR}/mjpeg_server.py" &
|
||||||
|
HTTP_PID=$!
|
||||||
|
|
||||||
|
wait -n "${HTTP_PID}"
|
||||||
@@ -0,0 +1,21 @@
|
|||||||
|
[Unit]
|
||||||
|
Description=Room camera snapshot server (MJPEG webcam)
|
||||||
|
After=network-online.target
|
||||||
|
Wants=network-online.target
|
||||||
|
|
||||||
|
[Service]
|
||||||
|
Type=simple
|
||||||
|
Environment=DEVICE=/dev/video0
|
||||||
|
Environment=RESOLUTION=640x480
|
||||||
|
Environment=QUALITY=5
|
||||||
|
Environment=PORT=8088
|
||||||
|
Environment=WORKDIR=/run/roomcam
|
||||||
|
Environment=INPUT_FORMAT=mjpeg
|
||||||
|
ExecStart=/usr/bin/env bash /usr/local/bin/room-cam-snapshot.sh
|
||||||
|
Restart=always
|
||||||
|
RestartSec=2
|
||||||
|
User=root
|
||||||
|
Group=root
|
||||||
|
|
||||||
|
[Install]
|
||||||
|
WantedBy=multi-user.target
|
||||||
@@ -0,0 +1,49 @@
|
|||||||
|
admins:
|
||||||
|
- username: admin
|
||||||
|
password_hash: "$2b$10$ZW4Jy7ctIt7k9V1AogFky.v4wedLF92t4/ZlT9kWPlIiCmdQNzJ.C" # password: adminpass
|
||||||
|
discord_id: "1234567890"
|
||||||
|
lockdown: false
|
||||||
|
- username: lockdown
|
||||||
|
password_hash: "$2b$10$n0L0oe1ZQy7IgM.FvVAzb.aXz43uaZWFiT0wr.05uNoVIDLawmrCG" # password: lockdownpass
|
||||||
|
discord_id: "0987654321"
|
||||||
|
lockdown: true
|
||||||
|
timezone: "America/New_York"
|
||||||
|
media:
|
||||||
|
# Base address for mediaMTX (scheme + host + optional port/path). The UI will always request
|
||||||
|
# http://<base>/<roverId>/whep
|
||||||
|
# Example: http://192.168.0.86:8889/video
|
||||||
|
whepBaseUrl: "http://192.168.0.86:8889/video"
|
||||||
|
|
||||||
|
homeAssistant:
|
||||||
|
url: "http://homeassistant.local:8123"
|
||||||
|
token: "REPLACE_WITH_LONG_LIVED_TOKEN"
|
||||||
|
entities:
|
||||||
|
- id: "light.lab_main"
|
||||||
|
name: "Lab Lights"
|
||||||
|
- id: "switch.dock_power"
|
||||||
|
name: "Dock Power"
|
||||||
|
# type is optional; if omitted it is inferred from the entity id (light/switch)
|
||||||
|
roomCameras:
|
||||||
|
- id: "lobby"
|
||||||
|
name: "Lobby Camera"
|
||||||
|
description: "Wide shot of the staging area."
|
||||||
|
url: "http://192.168.0.50/snapshot.jpg"
|
||||||
|
streamUrl: "http://192.168.0.50/stream.mjpg"
|
||||||
|
- id: "workshop"
|
||||||
|
name: "Workshop Bench"
|
||||||
|
description: "Shows the workbench and charging docks."
|
||||||
|
url: "http://192.168.0.51/snapshot.jpg"
|
||||||
|
streamUrl: "http://192.168.0.51/stream.mjpg"
|
||||||
|
|
||||||
|
discord:
|
||||||
|
token: "DISCORD_BOT_TOKEN"
|
||||||
|
guildId: "123456789012345678" # optional; bot works in any guild it's invited to
|
||||||
|
siteUrl: "https://rover.example.com"
|
||||||
|
channels:
|
||||||
|
announcements: "123456789012345678"
|
||||||
|
adminAlerts: "123456789012345678"
|
||||||
|
# chat bridge is configured per guild via `rs bridge` commands
|
||||||
|
replay: "123456789012345678"
|
||||||
|
roles:
|
||||||
|
announcementPing: "123456789012345678"
|
||||||
|
adminPing: "123456789012345678"
|
||||||
@@ -0,0 +1,36 @@
|
|||||||
|
require('./src/globals/logger');
|
||||||
|
require('./src/globals/config');
|
||||||
|
require('./src/globals/http');
|
||||||
|
require('./src/globals/io');
|
||||||
|
require('./src/globals/ws');
|
||||||
|
|
||||||
|
require('./src/helpers/sensorDecoder');
|
||||||
|
|
||||||
|
require('./src/services/alertService');
|
||||||
|
require('./src/services/authService');
|
||||||
|
require('./src/services/eventBus');
|
||||||
|
require('./src/services/modeManager');
|
||||||
|
require('./src/services/lockdownGuard');
|
||||||
|
require('./src/services/roverManager');
|
||||||
|
require('./src/services/commandService');
|
||||||
|
require('./src/services/roverConnectionService');
|
||||||
|
require('./src/services/assignmentService');
|
||||||
|
require('./src/services/nicknameService');
|
||||||
|
require('./src/services/chatService');
|
||||||
|
require('./src/services/communityGoalService');
|
||||||
|
require('./src/services/videoSessions');
|
||||||
|
require('./src/services/videoAuthService');
|
||||||
|
require('./src/services/videoSocketService');
|
||||||
|
require('./src/services/roomCameraSocketService');
|
||||||
|
require('./src/services/roverSnapshotSocketService');
|
||||||
|
require('./src/services/embedHttpService');
|
||||||
|
require('./src/services/logStreamService');
|
||||||
|
require('./src/services/adminLogService');
|
||||||
|
require('./src/services/homeAssistantService');
|
||||||
|
require('./src/services/sessionService');
|
||||||
|
require('./src/services/moderationService');
|
||||||
|
require('./src/services/batteryManager');
|
||||||
|
require('./src/services/replaySocketService');
|
||||||
|
require('./src/services/replaySegmentManager');
|
||||||
|
require('./src/services/discordBotService');
|
||||||
|
require('./src/services/httpServer');
|
||||||
Executable
+139
@@ -0,0 +1,139 @@
|
|||||||
|
#!/usr/bin/env bash
|
||||||
|
set -euo pipefail
|
||||||
|
|
||||||
|
MEDIAMTX_VERSION="1.15.3"
|
||||||
|
MEDIAMTX_BASE_URL="https://github.com/bluenviron/mediamtx/releases/download/v${MEDIAMTX_VERSION}"
|
||||||
|
MEDIAMTX_BIN="/usr/local/bin/mediamtx"
|
||||||
|
MEDIAMTX_CONF_DIR="/etc/mediamtx"
|
||||||
|
MEDIAMTX_CONFIG="$MEDIAMTX_CONF_DIR/mediamtx.yml"
|
||||||
|
MEDIAMTX_SERVICE="/etc/systemd/system/mediamtx.service"
|
||||||
|
MULTIROVER_SERVICE="/etc/systemd/system/multirover.service"
|
||||||
|
SNAPSHOT_DIR="/var/lib/rover-snapshots"
|
||||||
|
REPLAY_SEGMENT_DIR="/var/lib/replay-segments"
|
||||||
|
|
||||||
|
if [[ $EUID -ne 0 ]]; then
|
||||||
|
echo "This installer must be run with sudo/root." >&2
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
if [[ -z "${SUDO_USER:-}" || "${SUDO_USER}" == "root" ]]; then
|
||||||
|
echo "Run this script via 'sudo' from the normal user that owns the repo." >&2
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
TARGET_USER="$SUDO_USER"
|
||||||
|
SCRIPT_DIR=$(cd "$(dirname "$0")" && pwd)
|
||||||
|
SERVER_DIR="$SCRIPT_DIR"
|
||||||
|
CONFIG_PATH="$SERVER_DIR/config.yaml"
|
||||||
|
MEDIAMTX_TEMPLATE="$SERVER_DIR/mediamtx/mediamtx.yml"
|
||||||
|
|
||||||
|
echo "[1/6] Installing dependencies..."
|
||||||
|
dnf install -y nodejs npm curl tar >/dev/null
|
||||||
|
NODE_BIN="$(command -v node)"
|
||||||
|
|
||||||
|
echo "[2/6] Installing Node production deps..."
|
||||||
|
runuser -u "$TARGET_USER" -- bash -c "cd '$SERVER_DIR' && npm install --production"
|
||||||
|
|
||||||
|
if [[ ! -f "$CONFIG_PATH" ]]; then
|
||||||
|
cp "$SERVER_DIR/config.example.yaml" "$CONFIG_PATH"
|
||||||
|
chown "$TARGET_USER":"$TARGET_USER" "$CONFIG_PATH"
|
||||||
|
echo "Copied config.example.yaml to config.yaml; edit it before exposing the service."
|
||||||
|
fi
|
||||||
|
|
||||||
|
tmpdir=$(mktemp -d)
|
||||||
|
trap 'rm -rf "$tmpdir"' EXIT
|
||||||
|
|
||||||
|
arch=$(uname -m)
|
||||||
|
case "$arch" in
|
||||||
|
x86_64|amd64)
|
||||||
|
mediamtx_pkg="mediamtx_v${MEDIAMTX_VERSION}_linux_amd64.tar.gz"
|
||||||
|
;;
|
||||||
|
aarch64)
|
||||||
|
mediamtx_pkg="mediamtx_v${MEDIAMTX_VERSION}_linux_arm64.tar.gz"
|
||||||
|
;;
|
||||||
|
armv7l)
|
||||||
|
mediamtx_pkg="mediamtx_v${MEDIAMTX_VERSION}_linux_armv7.tar.gz"
|
||||||
|
;;
|
||||||
|
*)
|
||||||
|
echo "Unsupported architecture: $arch" >&2
|
||||||
|
exit 1
|
||||||
|
;;
|
||||||
|
esac
|
||||||
|
|
||||||
|
echo "[3/6] Installing mediaMTX ${MEDIAMTX_VERSION}..."
|
||||||
|
curl -L "$MEDIAMTX_BASE_URL/$mediamtx_pkg" -o "$tmpdir/mediamtx.tgz"
|
||||||
|
tar -xzf "$tmpdir/mediamtx.tgz" -C "$tmpdir" mediamtx
|
||||||
|
install -m 0755 "$tmpdir/mediamtx" "$MEDIAMTX_BIN"
|
||||||
|
|
||||||
|
mkdir -p "$MEDIAMTX_CONF_DIR"
|
||||||
|
if [[ ! -f "$MEDIAMTX_TEMPLATE" ]]; then
|
||||||
|
echo "mediaMTX template missing at $MEDIAMTX_TEMPLATE" >&2
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
echo " Installing mediaMTX config -> $MEDIAMTX_CONFIG"
|
||||||
|
rm -f "$MEDIAMTX_CONFIG"
|
||||||
|
install -m 0644 "$MEDIAMTX_TEMPLATE" "$MEDIAMTX_CONFIG"
|
||||||
|
chown -R "$TARGET_USER":"$TARGET_USER" "$MEDIAMTX_CONF_DIR"
|
||||||
|
|
||||||
|
echo "[4/6] Writing systemd units..."
|
||||||
|
mkdir -p "$SNAPSHOT_DIR"
|
||||||
|
chown "$TARGET_USER":"$TARGET_USER" "$SNAPSHOT_DIR"
|
||||||
|
mkdir -p "$REPLAY_SEGMENT_DIR"
|
||||||
|
chown "$TARGET_USER":"$TARGET_USER" "$REPLAY_SEGMENT_DIR"
|
||||||
|
cat > "$MEDIAMTX_SERVICE" <<EOF
|
||||||
|
[Unit]
|
||||||
|
Description=mediaMTX WebRTC Server
|
||||||
|
After=network-online.target
|
||||||
|
Wants=network-online.target
|
||||||
|
|
||||||
|
[Service]
|
||||||
|
User=$TARGET_USER
|
||||||
|
Group=$TARGET_USER
|
||||||
|
WorkingDirectory=$MEDIAMTX_CONF_DIR
|
||||||
|
Environment=ROVER_SNAPSHOT_DIR=$SNAPSHOT_DIR
|
||||||
|
ExecStart=$MEDIAMTX_BIN $MEDIAMTX_CONFIG
|
||||||
|
Restart=on-failure
|
||||||
|
RestartSec=2
|
||||||
|
|
||||||
|
[Install]
|
||||||
|
WantedBy=multi-user.target
|
||||||
|
EOF
|
||||||
|
|
||||||
|
cat > "$MULTIROVER_SERVICE" <<EOF
|
||||||
|
[Unit]
|
||||||
|
Description=Multi-Roomba Rover control server
|
||||||
|
After=network-online.target mediamtx.service
|
||||||
|
Wants=network-online.target
|
||||||
|
|
||||||
|
[Service]
|
||||||
|
User=$TARGET_USER
|
||||||
|
Group=$TARGET_USER
|
||||||
|
WorkingDirectory=$SERVER_DIR
|
||||||
|
Environment=NODE_ENV=production
|
||||||
|
Environment=SERVER_CONFIG=$CONFIG_PATH
|
||||||
|
Environment=ROVER_SNAPSHOT_DIR=$SNAPSHOT_DIR
|
||||||
|
Environment=REPLAY_SEGMENT_DIR=$REPLAY_SEGMENT_DIR
|
||||||
|
ExecStart=$NODE_BIN $SERVER_DIR/index.js
|
||||||
|
Restart=on-failure
|
||||||
|
RestartSec=2
|
||||||
|
|
||||||
|
[Install]
|
||||||
|
WantedBy=multi-user.target
|
||||||
|
EOF
|
||||||
|
|
||||||
|
chmod 644 "$MEDIAMTX_SERVICE" "$MULTIROVER_SERVICE"
|
||||||
|
|
||||||
|
echo "[5/6] Enabling services..."
|
||||||
|
systemctl daemon-reload
|
||||||
|
systemctl enable --now mediamtx.service
|
||||||
|
systemctl enable --now multirover.service
|
||||||
|
systemctl restart mediamtx.service
|
||||||
|
systemctl restart multirover.service
|
||||||
|
|
||||||
|
echo "[6/6] Done."
|
||||||
|
echo
|
||||||
|
echo "Services installed:"
|
||||||
|
echo " mediamtx.service (WebRTC fan-out)"
|
||||||
|
echo " multirover.service (Node.js control server)"
|
||||||
|
echo
|
||||||
|
echo "Update $CONFIG_PATH to set admins, lockdown settings, and media parameters."
|
||||||
@@ -0,0 +1,44 @@
|
|||||||
|
# Managed by install_server.sh; edit server/mediamtx/mediamtx.yml and rerun the installer.
|
||||||
|
logLevel: info
|
||||||
|
|
||||||
|
api: yes
|
||||||
|
apiAddress: 0.0.0.0:9997
|
||||||
|
metrics: yes
|
||||||
|
metricsAddress: 0.0.0.0:9998
|
||||||
|
pprof: no
|
||||||
|
pprofAddress: 127.0.0.1:9999
|
||||||
|
|
||||||
|
rtsp: no
|
||||||
|
rtmp: no
|
||||||
|
hls: no
|
||||||
|
|
||||||
|
webrtc: yes
|
||||||
|
webrtcLocalUDPAddress: :8189
|
||||||
|
webrtcLocalTCPAddress: :8189
|
||||||
|
webrtcAdditionalHosts: ['rover.otter.land', '192.168.0.100']
|
||||||
|
webrtcICEServers2:
|
||||||
|
# Google public STUN (world-wide, very commonly used)
|
||||||
|
- url: stun:stun.l.google.com:19302
|
||||||
|
- url: stun:stun1.l.google.com:19302
|
||||||
|
- url: stun:stun2.l.google.com:19302
|
||||||
|
- url: stun:stun3.l.google.com:19302
|
||||||
|
- url: stun:stun4.l.google.com:19302
|
||||||
|
|
||||||
|
# Cloudflare STUN (anycast, global PoPs)
|
||||||
|
- url: stun:stun.cloudflare.com:3478
|
||||||
|
|
||||||
|
srt: yes
|
||||||
|
srtAddress: :9000
|
||||||
|
|
||||||
|
authMethod: http
|
||||||
|
authHTTPAddress: http://127.0.0.1:8080/mediamtx/auth
|
||||||
|
authHTTPExclude:
|
||||||
|
- action: publish
|
||||||
|
- action: api
|
||||||
|
- action: metrics
|
||||||
|
- action: pprof
|
||||||
|
|
||||||
|
paths:
|
||||||
|
all:
|
||||||
|
source: publisher
|
||||||
|
sourceOnDemand: no
|
||||||
@@ -0,0 +1,26 @@
|
|||||||
|
{
|
||||||
|
"name": "multiroombarover-server",
|
||||||
|
"version": "0.1.0",
|
||||||
|
"private": true,
|
||||||
|
"scripts": {
|
||||||
|
"start": "node index.js",
|
||||||
|
"dev": "nodemon index.js",
|
||||||
|
"check:media": "node scripts/checkMedia.js"
|
||||||
|
},
|
||||||
|
"dependencies": {
|
||||||
|
"bcrypt": "^6.0.0",
|
||||||
|
"discord.js": "^14.25.1",
|
||||||
|
"express": "^4.19.2",
|
||||||
|
"home-assistant-js-websocket": "^3.1.2",
|
||||||
|
"js-yaml": "^4.1.1",
|
||||||
|
"morgan": "^1.10.0",
|
||||||
|
"obscenity": "^0.4.6",
|
||||||
|
"sharp": "^0.33.5",
|
||||||
|
"socket.io": "^4.7.5",
|
||||||
|
"uuid": "^9.0.1",
|
||||||
|
"ws": "^8.18.0"
|
||||||
|
},
|
||||||
|
"devDependencies": {
|
||||||
|
"nodemon": "^3.1.0"
|
||||||
|
}
|
||||||
|
}
|
||||||
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
Binary file not shown.
Binary file not shown.
|
After Width: | Height: | Size: 15 KiB |
@@ -0,0 +1,20 @@
|
|||||||
|
<!doctype html>
|
||||||
|
<html lang="en">
|
||||||
|
<head>
|
||||||
|
<meta charset="UTF-8" />
|
||||||
|
<link rel="icon" type="image/png" href="/bitmap.png" />
|
||||||
|
<link rel="apple-touch-icon" href="/bitmap.png" />
|
||||||
|
<link rel="manifest" href="/manifest.json" />
|
||||||
|
<meta name="viewport" content="width=device-width, initial-scale=1.0" />
|
||||||
|
<meta name="theme-color" content="#020617" />
|
||||||
|
<meta name="apple-mobile-web-app-capable" content="yes" />
|
||||||
|
<meta name="apple-mobile-web-app-status-bar-style" content="black-translucent" />
|
||||||
|
<meta name="apple-mobile-web-app-title" content="Multi Roomba Rover" />
|
||||||
|
<title>Multi Roomba Rover</title>
|
||||||
|
<script type="module" crossorigin src="/assets/index-DECQ2TrX.js"></script>
|
||||||
|
<link rel="stylesheet" crossorigin href="/assets/index-f8xLbmgU.css">
|
||||||
|
</head>
|
||||||
|
<body>
|
||||||
|
<div id="root"></div>
|
||||||
|
</body>
|
||||||
|
</html>
|
||||||
@@ -0,0 +1,18 @@
|
|||||||
|
{
|
||||||
|
"name": "Multi Roomba Rover",
|
||||||
|
"short_name": "MRR",
|
||||||
|
"description": "Remote driving interface for the MultiRoomba Rover fleet.",
|
||||||
|
"start_url": "/",
|
||||||
|
"scope": "/",
|
||||||
|
"display": "standalone",
|
||||||
|
"background_color": "#000000",
|
||||||
|
"theme_color": "#020617",
|
||||||
|
"icons": [
|
||||||
|
{
|
||||||
|
"src": "/bitmap.png",
|
||||||
|
"sizes": "512x512",
|
||||||
|
"type": "image/png",
|
||||||
|
"purpose": "any"
|
||||||
|
}
|
||||||
|
]
|
||||||
|
}
|
||||||
@@ -0,0 +1 @@
|
|||||||
|
<svg xmlns="http://www.w3.org/2000/svg" xmlns:xlink="http://www.w3.org/1999/xlink" aria-hidden="true" role="img" class="iconify iconify--logos" width="31.88" height="32" preserveAspectRatio="xMidYMid meet" viewBox="0 0 256 257"><defs><linearGradient id="IconifyId1813088fe1fbc01fb466" x1="-.828%" x2="57.636%" y1="7.652%" y2="78.411%"><stop offset="0%" stop-color="#41D1FF"></stop><stop offset="100%" stop-color="#BD34FE"></stop></linearGradient><linearGradient id="IconifyId1813088fe1fbc01fb467" x1="43.376%" x2="50.316%" y1="2.242%" y2="89.03%"><stop offset="0%" stop-color="#FFEA83"></stop><stop offset="8.333%" stop-color="#FFDD35"></stop><stop offset="100%" stop-color="#FFA800"></stop></linearGradient></defs><path fill="url(#IconifyId1813088fe1fbc01fb466)" d="M255.153 37.938L134.897 252.976c-2.483 4.44-8.862 4.466-11.382.048L.875 37.958c-2.746-4.814 1.371-10.646 6.827-9.67l120.385 21.517a6.537 6.537 0 0 0 2.322-.004l117.867-21.483c5.438-.991 9.574 4.796 6.877 9.62Z"></path><path fill="url(#IconifyId1813088fe1fbc01fb467)" d="M185.432.063L96.44 17.501a3.268 3.268 0 0 0-2.634 3.014l-5.474 92.456a3.268 3.268 0 0 0 3.997 3.378l24.777-5.718c2.318-.535 4.413 1.507 3.936 3.838l-7.361 36.047c-.495 2.426 1.782 4.5 4.151 3.78l15.304-4.649c2.372-.72 4.652 1.36 4.15 3.788l-11.698 56.621c-.732 3.542 3.979 5.473 5.943 2.437l1.313-2.028l72.516-144.72c1.215-2.423-.88-5.186-3.54-4.672l-25.505 4.922c-2.396.462-4.435-1.77-3.759-4.114l16.646-57.705c.677-2.35-1.37-4.583-3.769-4.113Z"></path></svg>
|
||||||
|
After Width: | Height: | Size: 1.5 KiB |
Executable
+40
@@ -0,0 +1,40 @@
|
|||||||
|
#!/usr/bin/env node
|
||||||
|
const http = require('http');
|
||||||
|
|
||||||
|
const api = process.env.MEDIAMTX_API || 'http://127.0.0.1:9997';
|
||||||
|
|
||||||
|
function fetchJSON(path) {
|
||||||
|
return new Promise((resolve, reject) => {
|
||||||
|
const req = http.request(api + path, (res) => {
|
||||||
|
let data = '';
|
||||||
|
res.on('data', (chunk) => (data += chunk));
|
||||||
|
res.on('end', () => {
|
||||||
|
try {
|
||||||
|
resolve(JSON.parse(data));
|
||||||
|
} catch (err) {
|
||||||
|
reject(err);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
});
|
||||||
|
req.on('error', reject);
|
||||||
|
req.end();
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
async function main() {
|
||||||
|
const list = await fetchJSON('/v3/paths/list');
|
||||||
|
if (!list.items || !list.items.length) {
|
||||||
|
console.log('No active paths');
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
list.items.forEach((item) => {
|
||||||
|
console.log(
|
||||||
|
`${item.name.padEnd(12)} ready=${item.ready} tracks=${item.tracks.join(',') || 'none'} bytes=${item.bytesReceived}`
|
||||||
|
);
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
main().catch((err) => {
|
||||||
|
console.error('check-media failed:', err.message);
|
||||||
|
process.exit(1);
|
||||||
|
});
|
||||||
Executable
+27
@@ -0,0 +1,27 @@
|
|||||||
|
#!/usr/bin/env node
|
||||||
|
const bcrypt = require('bcrypt');
|
||||||
|
const readline = require('readline');
|
||||||
|
|
||||||
|
const passwordFromArg = process.argv[2];
|
||||||
|
|
||||||
|
async function hashPassword(password) {
|
||||||
|
try {
|
||||||
|
const hash = await bcrypt.hash(password, 10);
|
||||||
|
console.log(`Password: ${password}`);
|
||||||
|
console.log(`Hash: ${hash}`);
|
||||||
|
process.exit(0);
|
||||||
|
} catch (err) {
|
||||||
|
console.error('Error hashing password:', err.message);
|
||||||
|
process.exit(1);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (passwordFromArg) {
|
||||||
|
hashPassword(passwordFromArg);
|
||||||
|
} else {
|
||||||
|
const rl = readline.createInterface({ input: process.stdin, output: process.stdout });
|
||||||
|
rl.question('Password to hash: ', (answer) => {
|
||||||
|
rl.close();
|
||||||
|
hashPassword(answer);
|
||||||
|
});
|
||||||
|
}
|
||||||
@@ -0,0 +1,6 @@
|
|||||||
|
const path = require('path');
|
||||||
|
|
||||||
|
module.exports = {
|
||||||
|
port: process.env.PORT || 8080,
|
||||||
|
staticDir: path.join(__dirname, '..', '..', 'public'),
|
||||||
|
};
|
||||||
@@ -0,0 +1,29 @@
|
|||||||
|
const http = require('http');
|
||||||
|
const express = require('express');
|
||||||
|
const morgan = require('morgan');
|
||||||
|
const { v4: uuidv4 } = require('uuid');
|
||||||
|
const config = require('./config');
|
||||||
|
const { parseCookieHeader } = require('../helpers/cookieParser');
|
||||||
|
|
||||||
|
const VISITOR_COOKIE = 'roverd_visitor';
|
||||||
|
const VISITOR_COOKIE_MAX_AGE = 60 * 60 * 24 * 365; // 1 year
|
||||||
|
|
||||||
|
const app = express();
|
||||||
|
app.use(morgan('dev'));
|
||||||
|
app.use(express.json());
|
||||||
|
app.use((req, res, next) => {
|
||||||
|
const cookies = parseCookieHeader(req.headers?.cookie || '');
|
||||||
|
let token = cookies[VISITOR_COOKIE];
|
||||||
|
if (!token) {
|
||||||
|
token = uuidv4();
|
||||||
|
const cookie = `${VISITOR_COOKIE}=${token}; Path=/; HttpOnly; SameSite=Strict; Max-Age=${VISITOR_COOKIE_MAX_AGE}`;
|
||||||
|
res.setHeader('Set-Cookie', cookie);
|
||||||
|
}
|
||||||
|
req.visitorToken = token;
|
||||||
|
next();
|
||||||
|
});
|
||||||
|
app.use(express.static(config.staticDir, { index: false }));
|
||||||
|
|
||||||
|
const httpServer = http.createServer(app);
|
||||||
|
|
||||||
|
module.exports = { app, httpServer };
|
||||||
@@ -0,0 +1,15 @@
|
|||||||
|
const { Server: SocketIOServer } = require('socket.io');
|
||||||
|
const { httpServer } = require('./http');
|
||||||
|
|
||||||
|
const io = new SocketIOServer(httpServer, {
|
||||||
|
cors: { origin: '*' },
|
||||||
|
transports: ['websocket', 'polling'],
|
||||||
|
pingInterval: 5000,
|
||||||
|
pingTimeout: 7000,
|
||||||
|
});
|
||||||
|
|
||||||
|
// Allow more service listeners without warnings.
|
||||||
|
io.sockets.setMaxListeners(30);
|
||||||
|
io.of('/').setMaxListeners(30);
|
||||||
|
|
||||||
|
module.exports = io;
|
||||||
@@ -0,0 +1,56 @@
|
|||||||
|
const sinks = new Set();
|
||||||
|
|
||||||
|
function notifySinks(level, label, args) {
|
||||||
|
if (!sinks.size) return;
|
||||||
|
const entry = {
|
||||||
|
level,
|
||||||
|
label: label || null,
|
||||||
|
args,
|
||||||
|
message: args.map((value) => {
|
||||||
|
if (typeof value === 'string') return value;
|
||||||
|
try {
|
||||||
|
return JSON.stringify(value);
|
||||||
|
} catch (err) {
|
||||||
|
return String(value);
|
||||||
|
}
|
||||||
|
}).join(' '),
|
||||||
|
timestamp: new Date().toISOString(),
|
||||||
|
};
|
||||||
|
sinks.forEach((sink) => {
|
||||||
|
try {
|
||||||
|
sink(entry);
|
||||||
|
} catch (err) {
|
||||||
|
// avoid recursive logging
|
||||||
|
console.error(entry.timestamp, '[ERROR]', '[logger]', 'Log sink failed', err);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
function stamp(level, label, args) {
|
||||||
|
const fields = [new Date().toISOString(), `[${level}]`];
|
||||||
|
if (label) {
|
||||||
|
fields.push(`[${label}]`);
|
||||||
|
}
|
||||||
|
notifySinks(level, label, args);
|
||||||
|
return [...fields, ...args];
|
||||||
|
}
|
||||||
|
|
||||||
|
function baseLogger(label) {
|
||||||
|
return {
|
||||||
|
info: (...args) => console.log(...stamp('INFO', label, args)),
|
||||||
|
warn: (...args) => console.warn(...stamp('WARN', label, args)),
|
||||||
|
error: (...args) => console.error(...stamp('ERROR', label, args)),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
function registerSink(fn) {
|
||||||
|
if (typeof fn !== 'function') return () => {};
|
||||||
|
sinks.add(fn);
|
||||||
|
return () => sinks.delete(fn);
|
||||||
|
}
|
||||||
|
|
||||||
|
module.exports = {
|
||||||
|
...baseLogger(),
|
||||||
|
child: (label) => baseLogger(label),
|
||||||
|
registerSink,
|
||||||
|
};
|
||||||
@@ -0,0 +1,19 @@
|
|||||||
|
const { WebSocketServer } = require('ws');
|
||||||
|
const { httpServer } = require('./http');
|
||||||
|
const logger = require('./logger');
|
||||||
|
|
||||||
|
const roverWSS = new WebSocketServer({ noServer: true });
|
||||||
|
|
||||||
|
httpServer.on('upgrade', (req, socket, head) => {
|
||||||
|
if (req.url.startsWith('/rover')) {
|
||||||
|
roverWSS.handleUpgrade(req, socket, head, (ws) => {
|
||||||
|
roverWSS.emit('connection', ws, req);
|
||||||
|
});
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
roverWSS.on('connection', () => {
|
||||||
|
logger.info('Rover websocket connected');
|
||||||
|
});
|
||||||
|
|
||||||
|
module.exports = roverWSS;
|
||||||
@@ -0,0 +1,20 @@
|
|||||||
|
const fs = require('fs');
|
||||||
|
const path = require('path');
|
||||||
|
const yaml = require('js-yaml');
|
||||||
|
|
||||||
|
const CONFIG_PATH = process.env.SERVER_CONFIG || path.join(__dirname, '..', '..', 'config.yaml');
|
||||||
|
|
||||||
|
let cachedConfig;
|
||||||
|
|
||||||
|
function loadConfig() {
|
||||||
|
if (cachedConfig) {
|
||||||
|
return cachedConfig;
|
||||||
|
}
|
||||||
|
const file = fs.readFileSync(CONFIG_PATH, 'utf8');
|
||||||
|
cachedConfig = yaml.load(file);
|
||||||
|
return cachedConfig;
|
||||||
|
}
|
||||||
|
|
||||||
|
module.exports = {
|
||||||
|
loadConfig,
|
||||||
|
};
|
||||||
@@ -0,0 +1,17 @@
|
|||||||
|
function parseCookieHeader(header = '') {
|
||||||
|
if (!header || typeof header !== 'string') return {};
|
||||||
|
return header
|
||||||
|
.split(';')
|
||||||
|
.map((part) => part.trim())
|
||||||
|
.filter(Boolean)
|
||||||
|
.reduce((acc, part) => {
|
||||||
|
const [key, ...rest] = part.split('=');
|
||||||
|
if (!key) return acc;
|
||||||
|
acc[key] = rest.join('=');
|
||||||
|
return acc;
|
||||||
|
}, {});
|
||||||
|
}
|
||||||
|
|
||||||
|
module.exports = {
|
||||||
|
parseCookieHeader,
|
||||||
|
};
|
||||||
@@ -0,0 +1,45 @@
|
|||||||
|
function extractForwardedIp(value) {
|
||||||
|
if (typeof value === 'string' && value.trim()) {
|
||||||
|
return value.split(',')[0].trim();
|
||||||
|
}
|
||||||
|
if (Array.isArray(value) && value.length) {
|
||||||
|
return String(value[0]).trim();
|
||||||
|
}
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
|
||||||
|
function getSocketIp(socket) {
|
||||||
|
if (!socket) return null;
|
||||||
|
const headers = socket.handshake?.headers || {};
|
||||||
|
const forwarded = extractForwardedIp(headers['x-forwarded-for']);
|
||||||
|
if (forwarded) return forwarded;
|
||||||
|
const realIp = headers['x-real-ip'];
|
||||||
|
if (typeof realIp === 'string' && realIp.trim()) {
|
||||||
|
return realIp.trim();
|
||||||
|
}
|
||||||
|
return (
|
||||||
|
socket.handshake?.address ||
|
||||||
|
socket.conn?.remoteAddress ||
|
||||||
|
socket.request?.connection?.remoteAddress ||
|
||||||
|
null
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
function getRequestIp(req, override) {
|
||||||
|
const fromOverride = extractForwardedIp(override);
|
||||||
|
if (fromOverride) return fromOverride;
|
||||||
|
if (!req) return null;
|
||||||
|
const headers = req.headers || {};
|
||||||
|
const forwarded = extractForwardedIp(headers['x-forwarded-for']);
|
||||||
|
if (forwarded) return forwarded;
|
||||||
|
const realIp = headers['x-real-ip'];
|
||||||
|
if (typeof realIp === 'string' && realIp.trim()) {
|
||||||
|
return realIp.trim();
|
||||||
|
}
|
||||||
|
return req.ip || req.connection?.remoteAddress || null;
|
||||||
|
}
|
||||||
|
|
||||||
|
module.exports = {
|
||||||
|
getSocketIp,
|
||||||
|
getRequestIp,
|
||||||
|
};
|
||||||
@@ -0,0 +1,223 @@
|
|||||||
|
const HEADER = 0x13;
|
||||||
|
const CHARGING_STATE = {
|
||||||
|
0: 'not charging',
|
||||||
|
1: 'reconditioning charging',
|
||||||
|
2: 'full charging',
|
||||||
|
3: 'trickle charging',
|
||||||
|
4: 'waiting',
|
||||||
|
5: 'charging fault',
|
||||||
|
};
|
||||||
|
|
||||||
|
const OI_MODES = {
|
||||||
|
1: 'off',
|
||||||
|
2: 'passive',
|
||||||
|
3: 'safe',
|
||||||
|
4: 'full',
|
||||||
|
};
|
||||||
|
|
||||||
|
const GROUP100_LAYOUT = [
|
||||||
|
{ id: 7, key: 'bumpsAndWheelDrops', bytes: 1, parser: parseBumps },
|
||||||
|
{ id: 8, key: 'wall', bytes: 1, parser: parseBool },
|
||||||
|
{ id: 9, key: 'cliffLeft', bytes: 1, parser: parseBool },
|
||||||
|
{ id: 10, key: 'cliffFrontLeft', bytes: 1, parser: parseBool },
|
||||||
|
{ id: 11, key: 'cliffFrontRight', bytes: 1, parser: parseBool },
|
||||||
|
{ id: 12, key: 'cliffRight', bytes: 1, parser: parseBool },
|
||||||
|
{ id: 13, key: 'virtualWall', bytes: 1, parser: parseBool },
|
||||||
|
{ id: 14, key: 'wheelOvercurrents', bytes: 1, parser: parseWheelCurrents },
|
||||||
|
{ id: 15, key: 'dirtDetect', bytes: 1, parser: parseUInt },
|
||||||
|
{ id: 16, key: 'dirtDetectLeft', bytes: 1, parser: parseUInt },
|
||||||
|
{ id: 17, key: 'infraredCharacterOmni', bytes: 1, parser: parseUInt },
|
||||||
|
{ id: 18, key: 'buttons', bytes: 1, parser: parseButtons },
|
||||||
|
{ id: 19, key: 'distanceMm', bytes: 2, parser: parseInt },
|
||||||
|
{ id: 20, key: 'angleDeg', bytes: 2, parser: parseInt },
|
||||||
|
{ id: 21, key: 'chargingState', bytes: 1, parser: parseChargingState },
|
||||||
|
{ id: 22, key: 'voltageMv', bytes: 2, parser: parseUInt },
|
||||||
|
{ id: 23, key: 'currentMa', bytes: 2, parser: parseInt },
|
||||||
|
{ id: 24, key: 'batteryTemperatureC', bytes: 1, parser: parseInt },
|
||||||
|
{ id: 25, key: 'batteryChargeMah', bytes: 2, parser: parseUInt },
|
||||||
|
{ id: 26, key: 'batteryCapacityMah', bytes: 2, parser: parseUInt },
|
||||||
|
{ id: 27, key: 'wallSignal', bytes: 2, parser: parseUInt },
|
||||||
|
{ id: 28, key: 'cliffLeftSignal', bytes: 2, parser: parseUInt },
|
||||||
|
{ id: 29, key: 'cliffFrontLeftSignal', bytes: 2, parser: parseUInt },
|
||||||
|
{ id: 30, key: 'cliffFrontRightSignal', bytes: 2, parser: parseUInt },
|
||||||
|
{ id: 31, key: 'cliffRightSignal', bytes: 2, parser: parseUInt },
|
||||||
|
{ id: 32, key: 'chargingSourcesAvailable', bytes: 1, parser: parseChargeSources },
|
||||||
|
{ id: 33, key: 'chargingSourcesReserved', bytes: 2, parser: parseUInt },
|
||||||
|
{ id: 34, key: 'chargingSources', bytes: 1, parser: parseChargeSources },
|
||||||
|
{ id: 35, key: 'oiMode', bytes: 1, parser: parseOiMode },
|
||||||
|
{ id: 36, key: 'songNumber', bytes: 1, parser: parseUInt },
|
||||||
|
{ id: 37, key: 'songPlaying', bytes: 1, parser: parseBool },
|
||||||
|
{ id: 38, key: 'streamPacketCount', bytes: 1, parser: parseUInt },
|
||||||
|
{ id: 39, key: 'requestedVelocity', bytes: 2, parser: parseInt },
|
||||||
|
{ id: 40, key: 'requestedRadius', bytes: 2, parser: parseInt },
|
||||||
|
{ id: 41, key: 'requestedRightVelocity', bytes: 2, parser: parseInt },
|
||||||
|
{ id: 42, key: 'requestedLeftVelocity', bytes: 2, parser: parseInt },
|
||||||
|
{ id: 43, key: 'encoderCountsLeft', bytes: 2, parser: parseUInt },
|
||||||
|
{ id: 44, key: 'encoderCountsRight', bytes: 2, parser: parseUInt },
|
||||||
|
{ id: 45, key: 'lightBumper', bytes: 1, parser: parseLightBumper },
|
||||||
|
{ id: 46, key: 'lightBumpLeftSignal', bytes: 2, parser: parseUInt },
|
||||||
|
{ id: 47, key: 'lightBumpFrontLeftSignal', bytes: 2, parser: parseUInt },
|
||||||
|
{ id: 48, key: 'lightBumpCenterLeftSignal', bytes: 2, parser: parseUInt },
|
||||||
|
{ id: 49, key: 'lightBumpCenterRightSignal', bytes: 2, parser: parseUInt },
|
||||||
|
{ id: 50, key: 'lightBumpFrontRightSignal', bytes: 2, parser: parseUInt },
|
||||||
|
{ id: 51, key: 'lightBumpRightSignal', bytes: 2, parser: parseUInt },
|
||||||
|
{ id: 52, key: 'infraredCharacterLeft', bytes: 1, parser: parseUInt },
|
||||||
|
{ id: 53, key: 'infraredCharacterRight', bytes: 1, parser: parseUInt },
|
||||||
|
{ id: 54, key: 'wheelLeftCurrentMa', bytes: 2, parser: parseInt },
|
||||||
|
{ id: 55, key: 'wheelRightCurrentMa', bytes: 2, parser: parseInt },
|
||||||
|
{ id: 56, key: 'mainBrushCurrentMa', bytes: 2, parser: parseInt },
|
||||||
|
{ id: 57, key: 'sideBrushCurrentMa', bytes: 2, parser: parseInt },
|
||||||
|
{ id: 58, key: 'stasis', bytes: 1, parser: parseBool },
|
||||||
|
];
|
||||||
|
|
||||||
|
const GROUP100_TOTAL = GROUP100_LAYOUT.reduce((sum, spec) => sum + spec.bytes, 0);
|
||||||
|
|
||||||
|
const TOP_LEVEL_PACKETS = {
|
||||||
|
100: GROUP100_TOTAL,
|
||||||
|
21: 1,
|
||||||
|
34: 1,
|
||||||
|
};
|
||||||
|
|
||||||
|
function parseSensorFrame(base64Data) {
|
||||||
|
if (!base64Data) return null;
|
||||||
|
const buf = Buffer.from(base64Data, 'base64');
|
||||||
|
if (buf.length < 4 || buf[0] !== HEADER) {
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
const nBytes = buf[1];
|
||||||
|
if (buf.length < nBytes + 3) {
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
const payload = buf.slice(2, 2 + nBytes);
|
||||||
|
const checksum = buf[2 + nBytes];
|
||||||
|
if (!validateChecksum(buf.slice(0, 2 + nBytes + 1), checksum)) {
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
const decoded = {};
|
||||||
|
let offset = 0;
|
||||||
|
while (offset < payload.length) {
|
||||||
|
const packetId = payload[offset++];
|
||||||
|
const size = TOP_LEVEL_PACKETS[packetId];
|
||||||
|
if (!size || offset + size > payload.length) {
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
const segment = payload.slice(offset, offset + size);
|
||||||
|
offset += size;
|
||||||
|
if (packetId === 100) {
|
||||||
|
Object.assign(decoded, decodeGroup100(segment));
|
||||||
|
} else if (packetId === 21 && decoded.chargingState == null) {
|
||||||
|
decoded.chargingState = parseChargingState(segment);
|
||||||
|
} else if (packetId === 34 && decoded.chargingSources == null) {
|
||||||
|
decoded.chargingSources = parseChargeSources(segment);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return decoded;
|
||||||
|
}
|
||||||
|
|
||||||
|
function decodeGroup100(buf) {
|
||||||
|
if (buf.length !== GROUP100_TOTAL) {
|
||||||
|
return {};
|
||||||
|
}
|
||||||
|
const values = {};
|
||||||
|
let offset = 0;
|
||||||
|
for (const spec of GROUP100_LAYOUT) {
|
||||||
|
const slice = buf.slice(offset, offset + spec.bytes);
|
||||||
|
offset += spec.bytes;
|
||||||
|
try {
|
||||||
|
values[spec.key] = spec.parser ? spec.parser(slice) : parseUInt(slice);
|
||||||
|
} catch (err) {
|
||||||
|
values[spec.key] = null;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return values;
|
||||||
|
}
|
||||||
|
|
||||||
|
function parseBool(buf) {
|
||||||
|
return Boolean(buf[0]);
|
||||||
|
}
|
||||||
|
|
||||||
|
function parseUInt(buf) {
|
||||||
|
return buf.readUIntBE(0, buf.length);
|
||||||
|
}
|
||||||
|
|
||||||
|
function parseInt(buf) {
|
||||||
|
return buf.readIntBE(0, buf.length);
|
||||||
|
}
|
||||||
|
|
||||||
|
function parseBumps(buf) {
|
||||||
|
const value = buf[0];
|
||||||
|
return {
|
||||||
|
bumpRight: Boolean(value & 0x01),
|
||||||
|
bumpLeft: Boolean(value & 0x02),
|
||||||
|
wheelDropRight: Boolean(value & 0x04),
|
||||||
|
wheelDropLeft: Boolean(value & 0x08),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
function parseWheelCurrents(buf) {
|
||||||
|
const value = buf[0];
|
||||||
|
return {
|
||||||
|
sideBrush: Boolean(value & 0x01),
|
||||||
|
mainBrush: Boolean(value & 0x04),
|
||||||
|
rightWheel: Boolean(value & 0x08),
|
||||||
|
leftWheel: Boolean(value & 0x10),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
const BUTTON_LABELS = ['clean', 'spot', 'dock', 'minute', 'hour', 'day', 'schedule', 'clock'];
|
||||||
|
function parseButtons(buf) {
|
||||||
|
const v = buf[0];
|
||||||
|
const result = {};
|
||||||
|
BUTTON_LABELS.forEach((label, idx) => {
|
||||||
|
result[label] = Boolean(v & (1 << idx));
|
||||||
|
});
|
||||||
|
return result;
|
||||||
|
}
|
||||||
|
|
||||||
|
function parseChargingState(buf) {
|
||||||
|
const code = buf[0];
|
||||||
|
return {
|
||||||
|
code,
|
||||||
|
label: CHARGING_STATE[code] || 'unknown',
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
function parseChargeSources(buf) {
|
||||||
|
const value = buf[0];
|
||||||
|
return {
|
||||||
|
internalCharger: Boolean(value & 0x01),
|
||||||
|
homeBase: Boolean(value & 0x02),
|
||||||
|
raw: value,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
function parseOiMode(buf) {
|
||||||
|
const code = buf[0];
|
||||||
|
return {
|
||||||
|
code,
|
||||||
|
label: OI_MODES[code] || 'unknown',
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
const LIGHT_BUMPER_LABELS = ['left', 'frontLeft', 'centerLeft', 'centerRight', 'frontRight', 'right'];
|
||||||
|
function parseLightBumper(buf) {
|
||||||
|
const value = buf[0];
|
||||||
|
const obj = {};
|
||||||
|
LIGHT_BUMPER_LABELS.forEach((label, idx) => {
|
||||||
|
obj[label] = Boolean(value & (1 << idx));
|
||||||
|
});
|
||||||
|
return obj;
|
||||||
|
}
|
||||||
|
|
||||||
|
function validateChecksum(frame, checksum) {
|
||||||
|
let sum = 0;
|
||||||
|
for (const byte of frame) {
|
||||||
|
sum = (sum + byte) & 0xff;
|
||||||
|
}
|
||||||
|
return (sum & 0xff) === 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
module.exports = {
|
||||||
|
parseSensorFrame,
|
||||||
|
CHARGING_STATE,
|
||||||
|
};
|
||||||
@@ -0,0 +1,84 @@
|
|||||||
|
const { v4: uuidv4 } = require('uuid');
|
||||||
|
const io = require('../globals/io');
|
||||||
|
const { getRole, roleEvents } = require('./roleService');
|
||||||
|
const { getSocketIp } = require('../helpers/ipResolver');
|
||||||
|
|
||||||
|
const ADMIN_ROLES = new Set(['admin', 'lockdown', 'lockdown-admin']);
|
||||||
|
const MAX_HISTORY = 200;
|
||||||
|
const history = [];
|
||||||
|
|
||||||
|
function isAdminRole(role) {
|
||||||
|
return ADMIN_ROLES.has(role);
|
||||||
|
}
|
||||||
|
|
||||||
|
function isAdminSocket(socket) {
|
||||||
|
return isAdminRole(getRole(socket));
|
||||||
|
}
|
||||||
|
|
||||||
|
function pushEntry(entry) {
|
||||||
|
history.push(entry);
|
||||||
|
if (history.length > MAX_HISTORY) {
|
||||||
|
history.shift();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
function emitToAdmins(event, payload) {
|
||||||
|
io.sockets.sockets.forEach((socket) => {
|
||||||
|
if (!isAdminSocket(socket)) return;
|
||||||
|
socket.emit(event, payload);
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
function logAdminEvent({ label, message, ip, meta = null, socketId = null }) {
|
||||||
|
const entry = {
|
||||||
|
id: uuidv4(),
|
||||||
|
ts: Date.now(),
|
||||||
|
label: label || null,
|
||||||
|
message: message || '',
|
||||||
|
ip: ip || null,
|
||||||
|
meta: meta || null,
|
||||||
|
socketId: socketId || null,
|
||||||
|
};
|
||||||
|
pushEntry(entry);
|
||||||
|
emitToAdmins('adminlog:entry', entry);
|
||||||
|
}
|
||||||
|
|
||||||
|
function hydrateSocket(socket) {
|
||||||
|
if (!socket || !isAdminSocket(socket)) return;
|
||||||
|
socket.emit('adminlog:init', history);
|
||||||
|
}
|
||||||
|
|
||||||
|
io.on('connection', (socket) => {
|
||||||
|
hydrateSocket(socket);
|
||||||
|
const ip = getSocketIp(socket);
|
||||||
|
if (ip) {
|
||||||
|
logAdminEvent({
|
||||||
|
label: 'socket',
|
||||||
|
message: 'Socket connected',
|
||||||
|
ip,
|
||||||
|
meta: { role: getRole(socket) },
|
||||||
|
socketId: socket.id,
|
||||||
|
});
|
||||||
|
}
|
||||||
|
socket.on('disconnect', () => {
|
||||||
|
const disconnectIp = getSocketIp(socket);
|
||||||
|
if (!disconnectIp) return;
|
||||||
|
logAdminEvent({
|
||||||
|
label: 'socket',
|
||||||
|
message: 'Socket disconnected',
|
||||||
|
ip: disconnectIp,
|
||||||
|
meta: { role: getRole(socket) },
|
||||||
|
socketId: socket.id,
|
||||||
|
});
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
roleEvents.on('change', ({ socket, role }) => {
|
||||||
|
if (!socket) return;
|
||||||
|
if (!isAdminRole(role)) return;
|
||||||
|
hydrateSocket(socket);
|
||||||
|
});
|
||||||
|
|
||||||
|
module.exports = {
|
||||||
|
logAdminEvent,
|
||||||
|
};
|
||||||
@@ -0,0 +1,16 @@
|
|||||||
|
const io = require('../globals/io');
|
||||||
|
|
||||||
|
function sendAlert({ color, title, message, ts = Date.now() }) {
|
||||||
|
const payload = {
|
||||||
|
color: color || '#2196f3',
|
||||||
|
title,
|
||||||
|
message,
|
||||||
|
ts,
|
||||||
|
};
|
||||||
|
io.emit('alert', payload);
|
||||||
|
io.emit('alert:new', { id: `${ts}-${Math.random().toString(36).slice(2)}`, ...payload });
|
||||||
|
}
|
||||||
|
|
||||||
|
module.exports = {
|
||||||
|
sendAlert,
|
||||||
|
};
|
||||||
@@ -0,0 +1,224 @@
|
|||||||
|
const EventEmitter = require('events');
|
||||||
|
const io = require('../globals/io');
|
||||||
|
const logger = require('../globals/logger').child('assignment');
|
||||||
|
const { MODES, getMode, modeEvents } = require('./modeManager');
|
||||||
|
const { roleEvents, getRole, isAdmin } = require('./roleService');
|
||||||
|
const roverManager = require('./roverManager');
|
||||||
|
|
||||||
|
const socketRefs = new Map(); // socketId -> socket
|
||||||
|
const assignments = new Map(); // socketId -> roverId
|
||||||
|
const waiting = new Set(); // socketIds waiting for placement
|
||||||
|
const assignmentEvents = new EventEmitter();
|
||||||
|
|
||||||
|
io.on('connection', (socket) => {
|
||||||
|
socketRefs.set(socket.id, socket);
|
||||||
|
socket.on('disconnect', () => {
|
||||||
|
socketRefs.delete(socket.id);
|
||||||
|
unassignSocket(socket);
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
roleEvents.on('change', ({ socket, role }) => {
|
||||||
|
if (!socket || !socket.id) return;
|
||||||
|
if (role === 'user') {
|
||||||
|
assignSocket(socket);
|
||||||
|
} else {
|
||||||
|
unassignSocket(socket);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
modeEvents.on('change', (mode) => {
|
||||||
|
if (mode === MODES.ADMIN || mode === MODES.LOCKDOWN) {
|
||||||
|
// release non-admin drivers
|
||||||
|
for (const [socketId, roverId] of assignments.entries()) {
|
||||||
|
const socket = socketRefs.get(socketId);
|
||||||
|
if (socket && !isAdmin(socket)) {
|
||||||
|
releaseAssignment(socket, roverId);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
reassignWaiting();
|
||||||
|
});
|
||||||
|
|
||||||
|
roverManager.managerEvents.on('lock', ({ roverId, locked }) => {
|
||||||
|
if (locked) {
|
||||||
|
reassignFromRover(roverId);
|
||||||
|
} else {
|
||||||
|
reassignWaiting();
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
roverManager.managerEvents.on('rover', ({ action }) => {
|
||||||
|
if (action === 'removed' || action === 'upsert') {
|
||||||
|
reassignWaiting();
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
roverManager.managerEvents.on('switch', ({ socketId, roverId }) => {
|
||||||
|
if (!socketId || !roverId) return;
|
||||||
|
const socket = socketRefs.get(socketId);
|
||||||
|
if (!socket) return;
|
||||||
|
assignments.set(socketId, roverId);
|
||||||
|
waiting.delete(socketId);
|
||||||
|
assignmentEvents.emit('update', socketId);
|
||||||
|
});
|
||||||
|
|
||||||
|
function assignSocket(socket) {
|
||||||
|
if (!socket || isAdmin(socket) || getRole(socket) !== 'user') {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
// avoid double assignment
|
||||||
|
if (assignments.has(socket.id)) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
const target = pickRover();
|
||||||
|
if (!target) {
|
||||||
|
waiting.add(socket.id);
|
||||||
|
logger.info('No rover available, user waiting', socket.id);
|
||||||
|
assignmentEvents.emit('update', socket.id);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
try {
|
||||||
|
roverManager.requestControl(target.id, socket, { allowUser: true });
|
||||||
|
assignments.set(socket.id, target.id);
|
||||||
|
waiting.delete(socket.id);
|
||||||
|
logger.info('Assigned user to rover', socket.id, target.id);
|
||||||
|
assignmentEvents.emit('update', socket.id);
|
||||||
|
} catch (err) {
|
||||||
|
logger.warn('Failed to assign user', err.message);
|
||||||
|
waiting.add(socket.id);
|
||||||
|
assignmentEvents.emit('update', socket.id);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
function unassignSocket(socket) {
|
||||||
|
if (!socket) return;
|
||||||
|
waiting.delete(socket.id);
|
||||||
|
const roverId = assignments.get(socket.id);
|
||||||
|
if (roverId) {
|
||||||
|
roverManager.releaseControl(roverId, socket);
|
||||||
|
assignments.delete(socket.id);
|
||||||
|
logger.info('Unassigned socket from rover', socket.id, roverId);
|
||||||
|
assignmentEvents.emit('update', socket.id);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
function reassignFromRover(roverId) {
|
||||||
|
for (const [socketId, rid] of assignments.entries()) {
|
||||||
|
if (rid !== roverId) continue;
|
||||||
|
const socket = socketRefs.get(socketId);
|
||||||
|
if (!socket) {
|
||||||
|
assignments.delete(socketId);
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
roverManager.releaseControl(rid, socket);
|
||||||
|
assignments.delete(socketId);
|
||||||
|
assignSocket(socket);
|
||||||
|
assignmentEvents.emit('update', socketId);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
function reassignWaiting() {
|
||||||
|
for (const socketId of Array.from(waiting)) {
|
||||||
|
const socket = socketRefs.get(socketId);
|
||||||
|
if (socket) {
|
||||||
|
assignSocket(socket);
|
||||||
|
} else {
|
||||||
|
waiting.delete(socketId);
|
||||||
|
}
|
||||||
|
assignmentEvents.emit('update', socketId);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
function releaseAssignment(socket, roverId) {
|
||||||
|
roverManager.releaseControl(roverId, socket);
|
||||||
|
assignments.delete(socket.id);
|
||||||
|
waiting.add(socket.id);
|
||||||
|
logger.info('Released assignment back to queue', socket.id, roverId);
|
||||||
|
assignmentEvents.emit('update', socket.id);
|
||||||
|
}
|
||||||
|
|
||||||
|
function forceRelease(roverId, socketId) {
|
||||||
|
const socket = socketRefs.get(socketId) || io.sockets.sockets.get(socketId);
|
||||||
|
if (assignments.get(socketId) === roverId) {
|
||||||
|
assignments.delete(socketId);
|
||||||
|
}
|
||||||
|
waiting.delete(socketId);
|
||||||
|
if (socket) {
|
||||||
|
roverManager.releaseControl(roverId, socket);
|
||||||
|
logger.info('Force released socket from rover', socketId, roverId);
|
||||||
|
} else {
|
||||||
|
logger.warn('Force release: socket not found', socketId, roverId);
|
||||||
|
}
|
||||||
|
assignmentEvents.emit('update', socketId);
|
||||||
|
}
|
||||||
|
|
||||||
|
function pickRover() {
|
||||||
|
const mode = getMode();
|
||||||
|
if (mode === MODES.ADMIN || mode === MODES.LOCKDOWN) {
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
const candidates = Array.from(roverManager.rovers.values()).filter((rover) => {
|
||||||
|
if (!rover || rover.locked) return false;
|
||||||
|
return true;
|
||||||
|
});
|
||||||
|
if (candidates.length === 0) {
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
const dockedRank = (rover) => {
|
||||||
|
if (!rover) return 0;
|
||||||
|
if (rover.docked === true) return -1;
|
||||||
|
if (rover.docked === false) return 1;
|
||||||
|
const sensors = rover.lastSensor?.decoded || rover.lastSensor?.sensors || null;
|
||||||
|
const docked = sensors?.chargingSources?.homeBase;
|
||||||
|
if (docked === true) return -1;
|
||||||
|
if (docked === false) return 1;
|
||||||
|
return 0;
|
||||||
|
};
|
||||||
|
const idleRank = (rover) => (rover?.drivers?.size === 0 ? 1 : 0);
|
||||||
|
candidates.sort((a, b) => {
|
||||||
|
const aEmpty = idleRank(a);
|
||||||
|
const bEmpty = idleRank(b);
|
||||||
|
if (aEmpty !== bEmpty) return bEmpty - aEmpty;
|
||||||
|
const aDockRank = dockedRank(a);
|
||||||
|
const bDockRank = dockedRank(b);
|
||||||
|
if (aEmpty === 1 && aDockRank !== bDockRank) {
|
||||||
|
return bDockRank - aDockRank;
|
||||||
|
}
|
||||||
|
if (a.drivers.size !== b.drivers.size) {
|
||||||
|
return a.drivers.size - b.drivers.size;
|
||||||
|
}
|
||||||
|
return bDockRank - aDockRank;
|
||||||
|
});
|
||||||
|
return candidates[0];
|
||||||
|
}
|
||||||
|
|
||||||
|
function describeAssignment(socketId) {
|
||||||
|
const assignedRoverId = assignments.get(socketId) || null;
|
||||||
|
const adminRoverId = assignedRoverId ? null : roverManager.getPrimaryRoverForSocket(socketId);
|
||||||
|
const waitingIndex = waiting.has(socketId) ? Array.from(waiting).indexOf(socketId) : -1;
|
||||||
|
const roverId = assignedRoverId || adminRoverId || null;
|
||||||
|
const waitingStatus = waiting.has(socketId);
|
||||||
|
return {
|
||||||
|
roverId,
|
||||||
|
status: assignedRoverId ? 'assigned' : adminRoverId ? 'admin' : waitingStatus ? 'waiting' : null,
|
||||||
|
queuePosition: waitingIndex >= 0 ? waitingIndex + 1 : null,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
module.exports = {
|
||||||
|
assignmentEvents,
|
||||||
|
describeAssignment,
|
||||||
|
forceRelease,
|
||||||
|
getAssignedRover: (socketId) => assignments.get(socketId) || null,
|
||||||
|
moveAssignment: (socket, roverId, { releasePrevious = true } = {}) => {
|
||||||
|
if (!socket || !roverId) return;
|
||||||
|
const previous = assignments.get(socket.id);
|
||||||
|
if (previous && previous !== roverId && releasePrevious) {
|
||||||
|
roverManager.releaseControl(previous, socket);
|
||||||
|
}
|
||||||
|
assignments.set(socket.id, roverId);
|
||||||
|
waiting.delete(socket.id);
|
||||||
|
assignmentEvents.emit('update', socket.id);
|
||||||
|
},
|
||||||
|
};
|
||||||
@@ -0,0 +1,76 @@
|
|||||||
|
const bcrypt = require('bcrypt');
|
||||||
|
const io = require('../globals/io');
|
||||||
|
const logger = require('../globals/logger').child('authService');
|
||||||
|
const { loadConfig } = require('../helpers/configLoader');
|
||||||
|
const { clearLockdownTimer } = require('./lockdownGuard');
|
||||||
|
const { setRole } = require('./roleService');
|
||||||
|
|
||||||
|
const config = loadConfig();
|
||||||
|
const admins = config.admins || [];
|
||||||
|
|
||||||
|
function findAdmin(username) {
|
||||||
|
return admins.find((admin) => admin.username === username);
|
||||||
|
}
|
||||||
|
|
||||||
|
async function authenticate(username, password) {
|
||||||
|
const admin = findAdmin(username);
|
||||||
|
if (!admin) {
|
||||||
|
throw new Error('Invalid credentials');
|
||||||
|
}
|
||||||
|
const ok = await bcrypt.compare(password, admin.password_hash);
|
||||||
|
if (!ok) {
|
||||||
|
throw new Error('Invalid credentials');
|
||||||
|
}
|
||||||
|
return admin;
|
||||||
|
}
|
||||||
|
|
||||||
|
function isAdmin(socket) {
|
||||||
|
return socket?.data?.role === 'admin' || socket?.data?.role === 'lockdown';
|
||||||
|
}
|
||||||
|
|
||||||
|
function isLockdownAdmin(socket) {
|
||||||
|
return socket?.data?.role === 'lockdown';
|
||||||
|
}
|
||||||
|
|
||||||
|
io.on('connection', (socket) => {
|
||||||
|
const requestedRole = socket.handshake?.query?.role;
|
||||||
|
const initialRole = requestedRole === 'spectator' ? 'spectator' : 'user';
|
||||||
|
setRole(socket, initialRole);
|
||||||
|
logger.info('Socket connected with role', socket.id, initialRole);
|
||||||
|
socket.emit('auth:role', { role: initialRole });
|
||||||
|
socket.on('auth:login', async ({ username, password }, cb = () => {}) => {
|
||||||
|
try {
|
||||||
|
const admin = await authenticate(username, password);
|
||||||
|
const role = admin.lockdown ? 'lockdown' : 'admin';
|
||||||
|
socket.data.user = { username: admin.username, discordId: admin.discord_id };
|
||||||
|
setRole(socket, role);
|
||||||
|
socket.emit('auth:role', { role });
|
||||||
|
clearLockdownTimer(socket);
|
||||||
|
logger.info('Login success', socket.id, role);
|
||||||
|
cb({ success: true, role: socket.data.role });
|
||||||
|
} catch (err) {
|
||||||
|
logger.warn('Login failed', socket.id, err.message);
|
||||||
|
cb({ success: false, error: err.message });
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
function handleRoleChange({ role } = {}, cb = () => {}) {
|
||||||
|
if (role === 'spectator' || role === 'user') {
|
||||||
|
setRole(socket, role);
|
||||||
|
socket.emit('auth:role', { role });
|
||||||
|
logger.info('Role changed via client request', socket.id, role);
|
||||||
|
cb({ success: true, role });
|
||||||
|
} else {
|
||||||
|
cb({ error: 'Invalid role' });
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
socket.on('role:set', handleRoleChange);
|
||||||
|
socket.on('session:setRole', handleRoleChange);
|
||||||
|
});
|
||||||
|
|
||||||
|
module.exports = {
|
||||||
|
isAdmin,
|
||||||
|
isLockdownAdmin,
|
||||||
|
authenticate,
|
||||||
|
};
|
||||||
@@ -0,0 +1,188 @@
|
|||||||
|
const logger = require('../globals/logger').child('batteryManager');
|
||||||
|
const { publishEvent } = require('./eventBus');
|
||||||
|
const { managerEvents, lockRover, rovers } = require('./roverManager');
|
||||||
|
|
||||||
|
const STATES = {
|
||||||
|
NORMAL: 'normal',
|
||||||
|
WARN: 'warn',
|
||||||
|
URGENT: 'urgent',
|
||||||
|
DOCKED: 'docked',
|
||||||
|
CHARGING: 'charging',
|
||||||
|
FULL: 'full',
|
||||||
|
LOCKED: 'locked',
|
||||||
|
};
|
||||||
|
|
||||||
|
const WAITING_UNLOCK_MS = 5 * 60 * 1000;
|
||||||
|
|
||||||
|
const roverState = new Map(); // roverId -> state snapshot
|
||||||
|
|
||||||
|
function getState(roverId) {
|
||||||
|
if (!roverState.has(roverId)) {
|
||||||
|
roverState.set(roverId, {
|
||||||
|
lastPercent: null,
|
||||||
|
warned: false,
|
||||||
|
urgent: false,
|
||||||
|
onDock: false,
|
||||||
|
dockedCharging: false,
|
||||||
|
charging: false,
|
||||||
|
batteryLocked: false,
|
||||||
|
waitingSince: null,
|
||||||
|
});
|
||||||
|
}
|
||||||
|
return roverState.get(roverId);
|
||||||
|
}
|
||||||
|
|
||||||
|
function shouldLock(record) {
|
||||||
|
return record.lockReason == null || record.lockReason === 'battery';
|
||||||
|
}
|
||||||
|
|
||||||
|
function handleSensorEvent({ roverId, sensors, batteryState }) {
|
||||||
|
if (!roverId) return;
|
||||||
|
const record = rovers.get(roverId);
|
||||||
|
if (!record) return;
|
||||||
|
const state = getState(roverId);
|
||||||
|
const config = record.meta?.battery || {};
|
||||||
|
const lockable = shouldLock(record);
|
||||||
|
|
||||||
|
if (batteryState?.percent != null) {
|
||||||
|
state.lastPercent = batteryState.percent;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!state.warned && batteryState?.warnActive) {
|
||||||
|
state.warned = true;
|
||||||
|
publishEvent({
|
||||||
|
source: 'batteryManager',
|
||||||
|
type: 'battery.warn',
|
||||||
|
payload: { roverId, batteryState },
|
||||||
|
});
|
||||||
|
}
|
||||||
|
if (!state.urgent && batteryState?.urgentActive) {
|
||||||
|
state.urgent = true;
|
||||||
|
publishEvent({
|
||||||
|
source: 'batteryManager',
|
||||||
|
type: 'battery.urgent',
|
||||||
|
payload: { roverId, batteryState },
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
const chargingState = sensors?.chargingState?.code;
|
||||||
|
const chargingSources = sensors?.chargingSources;
|
||||||
|
const onDock = Boolean(chargingSources?.homeBase);
|
||||||
|
if (onDock && !state.onDock) {
|
||||||
|
state.onDock = true;
|
||||||
|
publishEvent({
|
||||||
|
source: 'batteryManager',
|
||||||
|
type: 'battery.docked',
|
||||||
|
payload: { roverId, batteryState },
|
||||||
|
});
|
||||||
|
} else if (!onDock && state.onDock) {
|
||||||
|
state.onDock = false;
|
||||||
|
publishEvent({
|
||||||
|
source: 'batteryManager',
|
||||||
|
type: 'battery.undocked',
|
||||||
|
payload: { roverId, batteryState },
|
||||||
|
});
|
||||||
|
}
|
||||||
|
const chargingLabel = sensors?.chargingState?.label?.toLowerCase();
|
||||||
|
const chargingByLabel =
|
||||||
|
chargingLabel === 'waiting' || chargingLabel === 'full charging' || chargingLabel === 'trickle charging';
|
||||||
|
const chargingByCode = chargingState === 2 || chargingState === 3 || chargingState === 4;
|
||||||
|
const isCharging = chargingByLabel || chargingByCode;
|
||||||
|
if (isCharging && !state.charging) {
|
||||||
|
state.charging = true;
|
||||||
|
publishEvent({
|
||||||
|
source: 'batteryManager',
|
||||||
|
type: 'battery.charging.start',
|
||||||
|
payload: { roverId, batteryState },
|
||||||
|
});
|
||||||
|
} else if (!isCharging && state.charging) {
|
||||||
|
state.charging = false;
|
||||||
|
publishEvent({
|
||||||
|
source: 'batteryManager',
|
||||||
|
type: 'battery.charging.stop',
|
||||||
|
payload: { roverId, batteryState },
|
||||||
|
});
|
||||||
|
}
|
||||||
|
const dockedCharging = onDock && isCharging;
|
||||||
|
if (dockedCharging && !state.dockedCharging) {
|
||||||
|
state.dockedCharging = true;
|
||||||
|
} else if (!dockedCharging && state.dockedCharging) {
|
||||||
|
state.dockedCharging = false;
|
||||||
|
}
|
||||||
|
|
||||||
|
const waitingState = chargingState === 4;
|
||||||
|
if (waitingState && state.waitingSince == null) {
|
||||||
|
state.waitingSince = Date.now();
|
||||||
|
} else if (!waitingState && state.waitingSince != null) {
|
||||||
|
state.waitingSince = null;
|
||||||
|
}
|
||||||
|
const waitingLongEnough =
|
||||||
|
waitingState && state.waitingSince != null && Date.now() - state.waitingSince >= WAITING_UNLOCK_MS;
|
||||||
|
|
||||||
|
const warnThreshold =
|
||||||
|
typeof config?.Warn === 'number' ? config.Warn : batteryState?.warn ?? null;
|
||||||
|
const fullThreshold =
|
||||||
|
typeof config?.Full === 'number' ? config.Full : batteryState?.full ?? null;
|
||||||
|
const chargeMah = batteryState?.charge ?? null;
|
||||||
|
let waitingPercent = batteryState?.percent ?? null;
|
||||||
|
if (chargeMah != null && warnThreshold != null && fullThreshold != null && fullThreshold > warnThreshold) {
|
||||||
|
waitingPercent = (chargeMah - warnThreshold) / (fullThreshold - warnThreshold);
|
||||||
|
waitingPercent = Math.max(0, Math.min(1, waitingPercent));
|
||||||
|
}
|
||||||
|
|
||||||
|
const isFull =
|
||||||
|
(batteryState?.charge != null && fullThreshold != null && batteryState.charge >= fullThreshold) ||
|
||||||
|
(batteryState?.percent != null && batteryState.percent >= 0.99);
|
||||||
|
|
||||||
|
const needsCharge = state.warned;
|
||||||
|
|
||||||
|
if (dockedCharging && lockable && needsCharge && !state.batteryLocked && !isFull && !waitingLongEnough) {
|
||||||
|
logger.info('Auto-locking rover for charging', { roverId });
|
||||||
|
lockRover(roverId, true, { reason: 'battery' });
|
||||||
|
state.batteryLocked = true;
|
||||||
|
publishEvent({
|
||||||
|
source: 'batteryManager',
|
||||||
|
type: 'battery.locked',
|
||||||
|
payload: { roverId, batteryState },
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
const waitingHalfOrMore = waitingPercent != null && waitingPercent >= 0.01;
|
||||||
|
// const waitingHalfOrMore = true;
|
||||||
|
const shouldUnlock = state.batteryLocked && waitingLongEnough && waitingHalfOrMore;
|
||||||
|
|
||||||
|
if (shouldUnlock) {
|
||||||
|
logger.info('Unlocking rover after charge', {
|
||||||
|
roverId,
|
||||||
|
isFull,
|
||||||
|
docked: dockedCharging,
|
||||||
|
waitingState,
|
||||||
|
waitingPercent,
|
||||||
|
waitedMs: state.waitingSince ? Date.now() - state.waitingSince : null,
|
||||||
|
});
|
||||||
|
lockRover(roverId, false, { reason: 'battery' });
|
||||||
|
state.batteryLocked = false;
|
||||||
|
state.warned = false;
|
||||||
|
state.urgent = false;
|
||||||
|
state.waitingSince = null;
|
||||||
|
publishEvent({
|
||||||
|
source: 'batteryManager',
|
||||||
|
type: 'battery.unlocked',
|
||||||
|
payload: { roverId, batteryState },
|
||||||
|
});
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
function handleLockEvent({ roverId, locked, reason }) {
|
||||||
|
const state = getState(roverId);
|
||||||
|
if (!state) return;
|
||||||
|
if (!locked) {
|
||||||
|
state.batteryLocked = false;
|
||||||
|
state.waitingSince = null;
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
state.batteryLocked = reason === 'battery';
|
||||||
|
}
|
||||||
|
|
||||||
|
managerEvents.on('sensor', handleSensorEvent);
|
||||||
|
managerEvents.on('lock', handleLockEvent);
|
||||||
@@ -0,0 +1,391 @@
|
|||||||
|
const { v4: uuidv4 } = require('uuid');
|
||||||
|
const { DataSet, RegExpMatcher, englishDataset, englishRecommendedTransformers } = require('obscenity');
|
||||||
|
const io = require('../globals/io');
|
||||||
|
const logger = require('../globals/logger').child('chatService');
|
||||||
|
const { publishEvent, subscribe } = require('./eventBus');
|
||||||
|
const { getRole } = require('./roleService');
|
||||||
|
const { describeAssignment } = require('./assignmentService');
|
||||||
|
const roverManager = require('./roverManager');
|
||||||
|
const { getNickname } = require('./nicknameService');
|
||||||
|
const { issueCommand } = require('./commandService');
|
||||||
|
const { isBannedSocket } = require('./moderationService');
|
||||||
|
|
||||||
|
const RATE_LIMIT_WINDOW_MS = 8000;
|
||||||
|
const RATE_LIMIT_MAX = 5;
|
||||||
|
const rateBuckets = new Map(); // socketId -> [timestamps]
|
||||||
|
|
||||||
|
const MAX_HISTORY = 100;
|
||||||
|
const history = [];
|
||||||
|
|
||||||
|
// Words in this list are removed from the profanity dataset entirely.
|
||||||
|
const PROFANITY_ALLOWLIST = ['fuck', 'ass', 'shit'];
|
||||||
|
const normalizedProfanityAllowlist = new Set(PROFANITY_ALLOWLIST
|
||||||
|
.filter((term) => typeof term === 'string')
|
||||||
|
.map((term) => term.trim().toLowerCase())
|
||||||
|
.filter(Boolean));
|
||||||
|
const profanityDataset = new DataSet()
|
||||||
|
.addAll(englishDataset)
|
||||||
|
.removePhrasesIf((phrase) => normalizedProfanityAllowlist.has(phrase.metadata?.originalWord))
|
||||||
|
.build();
|
||||||
|
const profanityMatcher = new RegExpMatcher({
|
||||||
|
...profanityDataset,
|
||||||
|
...englishRecommendedTransformers,
|
||||||
|
whitelistedTerms: profanityDataset.whitelistedTerms,
|
||||||
|
});
|
||||||
|
const DUPLICATE_WINDOW_MS = 15000;
|
||||||
|
const lastMessageBySocket = new Map(); // socketId -> { text, ts }
|
||||||
|
const typingBySocket = new Map(); // socketId -> boolean
|
||||||
|
const TYPING_START_NOTE = 72;
|
||||||
|
const TYPING_SEND_NOTE = 79;
|
||||||
|
const TYPING_NOTE_DURATION = 8;
|
||||||
|
|
||||||
|
function withinRateLimit(socketId) {
|
||||||
|
const now = Date.now();
|
||||||
|
const entries = rateBuckets.get(socketId) || [];
|
||||||
|
const next = entries.filter((ts) => now - ts <= RATE_LIMIT_WINDOW_MS);
|
||||||
|
next.push(now);
|
||||||
|
rateBuckets.set(socketId, next);
|
||||||
|
return next.length <= RATE_LIMIT_MAX;
|
||||||
|
}
|
||||||
|
|
||||||
|
function hasProfanity(text) {
|
||||||
|
if (typeof text !== 'string' || !text) return false;
|
||||||
|
return profanityMatcher.hasMatch(text);
|
||||||
|
}
|
||||||
|
|
||||||
|
function isDuplicate(socketId, text) {
|
||||||
|
const prev = lastMessageBySocket.get(socketId);
|
||||||
|
const now = Date.now();
|
||||||
|
if (!prev) {
|
||||||
|
lastMessageBySocket.set(socketId, { text, ts: now });
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
lastMessageBySocket.set(socketId, { text, ts: now });
|
||||||
|
return prev.text === text && now - prev.ts <= DUPLICATE_WINDOW_MS;
|
||||||
|
}
|
||||||
|
|
||||||
|
function isKeymash(text) {
|
||||||
|
if (!text) return false;
|
||||||
|
if (/(.)\1{6,}/.test(text)) return true; // same char 7+
|
||||||
|
if (/^[asdfghjkl;'\-=\[\]\\]{6,}$/i.test(text)) return true;
|
||||||
|
if (/^[qwertyuiop]{6,}$/i.test(text)) return true;
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
function resolveRoverId(socketId) {
|
||||||
|
const primary = roverManager.getPrimaryRoverForSocket(socketId);
|
||||||
|
if (primary) return primary;
|
||||||
|
const assignment = describeAssignment(socketId);
|
||||||
|
return assignment?.roverId || null;
|
||||||
|
}
|
||||||
|
|
||||||
|
function normalizeUserText(raw) {
|
||||||
|
if (typeof raw !== 'string') return '';
|
||||||
|
return raw.replace(/\\n/g, '\n');
|
||||||
|
}
|
||||||
|
|
||||||
|
function buildMessage(socket, text, meta = {}) {
|
||||||
|
const roverId = meta.roverId || resolveRoverId(socket?.id);
|
||||||
|
return {
|
||||||
|
id: uuidv4(),
|
||||||
|
ts: Date.now(),
|
||||||
|
socketId: socket?.id || null,
|
||||||
|
nickname: meta.nickname || getNickname(socket) || null,
|
||||||
|
role: meta.role || getRole(socket),
|
||||||
|
roverId,
|
||||||
|
fromDiscord: Boolean(meta.fromDiscord),
|
||||||
|
discordGuildId: meta.discordGuildId || null,
|
||||||
|
discordGuildName: meta.discordGuildName || null,
|
||||||
|
discordGuildIconUrl: meta.discordGuildIconUrl || null,
|
||||||
|
discordChannelId: meta.discordChannelId || null,
|
||||||
|
discordUserId: meta.discordUserId || null,
|
||||||
|
discordUserName: meta.discordUserName || null,
|
||||||
|
discordUserAvatarUrl: meta.discordUserAvatarUrl || null,
|
||||||
|
text,
|
||||||
|
tts: meta.tts || null,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
function buildTypingPayload(socket, meta = {}) {
|
||||||
|
const roverId = meta.roverId || resolveRoverId(socket?.id);
|
||||||
|
const socketId = socket?.id || null;
|
||||||
|
const fromDiscord = Boolean(meta.fromDiscord);
|
||||||
|
let typingId = meta.typingId || null;
|
||||||
|
if (!typingId) {
|
||||||
|
if (fromDiscord) {
|
||||||
|
if (meta.discordUserId) {
|
||||||
|
typingId = `discord:${meta.discordUserId}`;
|
||||||
|
} else if (meta.discordUserName) {
|
||||||
|
typingId = `discord:${meta.discordUserName}`;
|
||||||
|
} else if (meta.nickname) {
|
||||||
|
typingId = `discord:${meta.nickname}`;
|
||||||
|
} else {
|
||||||
|
typingId = 'discord:unknown';
|
||||||
|
}
|
||||||
|
} else if (socketId) {
|
||||||
|
typingId = `socket:${socketId}`;
|
||||||
|
} else if (meta.nickname) {
|
||||||
|
typingId = `socket:${meta.nickname}`;
|
||||||
|
} else {
|
||||||
|
typingId = 'socket:unknown';
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return {
|
||||||
|
id: uuidv4(),
|
||||||
|
ts: Date.now(),
|
||||||
|
typingId,
|
||||||
|
isTyping: Boolean(meta.isTyping),
|
||||||
|
socketId,
|
||||||
|
nickname: meta.nickname || getNickname(socket) || null,
|
||||||
|
role: meta.role || getRole(socket),
|
||||||
|
roverId,
|
||||||
|
fromDiscord,
|
||||||
|
discordGuildId: meta.discordGuildId || null,
|
||||||
|
discordGuildName: meta.discordGuildName || null,
|
||||||
|
discordGuildIconUrl: meta.discordGuildIconUrl || null,
|
||||||
|
discordChannelId: meta.discordChannelId || null,
|
||||||
|
discordUserId: meta.discordUserId || null,
|
||||||
|
discordUserName: meta.discordUserName || null,
|
||||||
|
discordUserAvatarUrl: meta.discordUserAvatarUrl || null,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
function pushHistory(message) {
|
||||||
|
history.push(message);
|
||||||
|
if (history.length > MAX_HISTORY) {
|
||||||
|
history.shift();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
function broadcastMessage(message) {
|
||||||
|
pushHistory(message);
|
||||||
|
publishEvent({ source: 'chat', type: 'chat:message', payload: message });
|
||||||
|
}
|
||||||
|
|
||||||
|
function broadcastTyping(payload) {
|
||||||
|
publishEvent({ source: 'chat', type: 'chat:typing', payload });
|
||||||
|
}
|
||||||
|
|
||||||
|
function playTypingNote(roverId, note, socketId) {
|
||||||
|
if (!roverId) return;
|
||||||
|
try {
|
||||||
|
issueCommand(roverId, {
|
||||||
|
type: 'song',
|
||||||
|
song: {
|
||||||
|
notes: [{ note, duration: TYPING_NOTE_DURATION }],
|
||||||
|
},
|
||||||
|
});
|
||||||
|
const log = typeof logger.debug === 'function' ? logger.debug.bind(logger) : logger.info.bind(logger);
|
||||||
|
log('Typing tone sent', { roverId, note, socketId });
|
||||||
|
} catch (err) {
|
||||||
|
const log = typeof logger.debug === 'function' ? logger.debug.bind(logger) : logger.info.bind(logger);
|
||||||
|
log('Typing tone failed', { roverId, note, socketId, error: err.message });
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
function normalizeTtsOptions(raw = {}) {
|
||||||
|
if (!raw || typeof raw !== 'object') return null;
|
||||||
|
const speak = raw.speak !== false;
|
||||||
|
if (!speak) return null;
|
||||||
|
const engine = typeof raw.engine === 'string' && raw.engine.toLowerCase() === 'espeak' ? 'espeak' : 'flite';
|
||||||
|
const voice = typeof raw.voice === 'string' ? raw.voice.trim() : undefined;
|
||||||
|
let pitch = Number.isFinite(raw.pitch) ? Math.round(raw.pitch) : undefined;
|
||||||
|
if (typeof pitch === 'number') {
|
||||||
|
pitch = Math.max(0, Math.min(99, pitch));
|
||||||
|
}
|
||||||
|
return { speak, engine, voice, pitch };
|
||||||
|
}
|
||||||
|
|
||||||
|
function handleIncoming({ text, tts } = {}, socket, cb = () => {}) {
|
||||||
|
const role = getRole(socket);
|
||||||
|
// if (role === 'spectator') {
|
||||||
|
// cb({ error: 'Spectators cannot chat' });
|
||||||
|
// return;
|
||||||
|
// }
|
||||||
|
const normalized = normalizeUserText(text);
|
||||||
|
const clean = normalized.trim();
|
||||||
|
if (!clean) {
|
||||||
|
cb({ error: 'Message required' });
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
if (!withinRateLimit(socket.id)) {
|
||||||
|
cb({ error: 'Slow down' });
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
if (clean.length > 150) {
|
||||||
|
cb({ error: 'Message too long' });
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
if (hasProfanity(clean)) {
|
||||||
|
cb({ error: 'Message blocked' });
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
// if (isDuplicate(socket.id, clean)) {
|
||||||
|
// cb({ error: 'Duplicate message' });
|
||||||
|
// return;
|
||||||
|
// }
|
||||||
|
// if (isKeymash(clean)) {
|
||||||
|
// cb({ error: 'Message looks like spam' });
|
||||||
|
// return;
|
||||||
|
// }
|
||||||
|
const roverId = resolveRoverId(socket?.id);
|
||||||
|
const ttsOptions = normalizeTtsOptions(tts);
|
||||||
|
const message = buildMessage(socket, clean, { fromDiscord: false, roverId, tts: ttsOptions });
|
||||||
|
logger.info('Chat message', { socket: socket.id, roverId: message.roverId });
|
||||||
|
playTypingNote(roverId, TYPING_SEND_NOTE, socket?.id);
|
||||||
|
broadcastMessage(message);
|
||||||
|
maybeSpeak(socket, message, ttsOptions);
|
||||||
|
cb({ success: true });
|
||||||
|
}
|
||||||
|
|
||||||
|
function maybeSpeak(socket, message, ttsOptions) {
|
||||||
|
if (!ttsOptions || !message?.roverId) return;
|
||||||
|
const record = roverManager.rovers.get(message.roverId);
|
||||||
|
const audio = record?.meta?.audio || {};
|
||||||
|
const ttsEnabled = Boolean(audio.ttsEnabled);
|
||||||
|
if (!ttsEnabled) return;
|
||||||
|
if (!roverManager.canDrive(message.roverId, socket)) return;
|
||||||
|
try {
|
||||||
|
issueCommand(message.roverId, {
|
||||||
|
type: 'tts',
|
||||||
|
tts: {
|
||||||
|
text: message.text,
|
||||||
|
engine: ttsOptions.engine,
|
||||||
|
voice: ttsOptions.voice,
|
||||||
|
pitch: ttsOptions.pitch,
|
||||||
|
speak: true,
|
||||||
|
},
|
||||||
|
});
|
||||||
|
logger.info('TTS sent', { roverId: message.roverId, engine: ttsOptions.engine, socket: socket.id });
|
||||||
|
} catch (err) {
|
||||||
|
logger.warn('TTS send failed', { roverId: message.roverId, error: err.message });
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
function sendExternalMessage({
|
||||||
|
text,
|
||||||
|
nickname = 'Discord',
|
||||||
|
role = 'admin',
|
||||||
|
roverId = null,
|
||||||
|
discordGuildId = null,
|
||||||
|
discordGuildName = null,
|
||||||
|
discordGuildIconUrl = null,
|
||||||
|
discordChannelId = null,
|
||||||
|
discordUserId = null,
|
||||||
|
discordUserName = null,
|
||||||
|
discordUserAvatarUrl = null,
|
||||||
|
}) {
|
||||||
|
const normalized = normalizeUserText(text);
|
||||||
|
const clean = normalized.trim();
|
||||||
|
if (!clean || clean.length > 256) {
|
||||||
|
throw new Error('Message invalid');
|
||||||
|
}
|
||||||
|
if (hasProfanity(clean)) {
|
||||||
|
throw new Error('Message blocked');
|
||||||
|
}
|
||||||
|
if (isKeymash(clean)) {
|
||||||
|
throw new Error('Message looks like spam');
|
||||||
|
}
|
||||||
|
const message = buildMessage(null, clean, {
|
||||||
|
nickname,
|
||||||
|
role,
|
||||||
|
roverId,
|
||||||
|
fromDiscord: true,
|
||||||
|
discordGuildId,
|
||||||
|
discordGuildName,
|
||||||
|
discordGuildIconUrl,
|
||||||
|
discordChannelId,
|
||||||
|
discordUserId,
|
||||||
|
discordUserName,
|
||||||
|
discordUserAvatarUrl,
|
||||||
|
});
|
||||||
|
logger.info('External chat message', { roverId, nickname });
|
||||||
|
broadcastMessage(message);
|
||||||
|
return message;
|
||||||
|
}
|
||||||
|
|
||||||
|
function sendExternalTyping({
|
||||||
|
nickname = 'Discord',
|
||||||
|
role = 'user',
|
||||||
|
roverId = null,
|
||||||
|
discordGuildId = null,
|
||||||
|
discordGuildName = null,
|
||||||
|
discordGuildIconUrl = null,
|
||||||
|
discordChannelId = null,
|
||||||
|
discordUserId = null,
|
||||||
|
discordUserName = null,
|
||||||
|
discordUserAvatarUrl = null,
|
||||||
|
isTyping = true,
|
||||||
|
}) {
|
||||||
|
const payload = buildTypingPayload(null, {
|
||||||
|
nickname,
|
||||||
|
role,
|
||||||
|
roverId,
|
||||||
|
fromDiscord: true,
|
||||||
|
discordGuildId,
|
||||||
|
discordGuildName,
|
||||||
|
discordGuildIconUrl,
|
||||||
|
discordChannelId,
|
||||||
|
discordUserId,
|
||||||
|
discordUserName,
|
||||||
|
discordUserAvatarUrl,
|
||||||
|
isTyping,
|
||||||
|
});
|
||||||
|
broadcastTyping(payload);
|
||||||
|
return payload;
|
||||||
|
}
|
||||||
|
|
||||||
|
io.on('connection', (socket) => {
|
||||||
|
if (!isBannedSocket(socket)) {
|
||||||
|
socket.emit('chat:init', history);
|
||||||
|
}
|
||||||
|
socket.on('chat:send', (payload = {}, cb = () => {}) => handleIncoming(payload, socket, cb));
|
||||||
|
socket.on('chat:typing', (payload = {}) => {
|
||||||
|
const isTyping = Boolean(payload?.isTyping);
|
||||||
|
const wasTyping = typingBySocket.get(socket.id);
|
||||||
|
if (isTyping) {
|
||||||
|
typingBySocket.set(socket.id, true);
|
||||||
|
if (!wasTyping) {
|
||||||
|
const roverId = resolveRoverId(socket?.id);
|
||||||
|
playTypingNote(roverId, TYPING_START_NOTE, socket?.id);
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
typingBySocket.delete(socket.id);
|
||||||
|
}
|
||||||
|
const roverId = resolveRoverId(socket?.id);
|
||||||
|
const typingPayload = buildTypingPayload(socket, { roverId, fromDiscord: false, isTyping });
|
||||||
|
broadcastTyping(typingPayload);
|
||||||
|
});
|
||||||
|
socket.on('disconnect', () => {
|
||||||
|
if (!typingBySocket.has(socket.id)) return;
|
||||||
|
typingBySocket.delete(socket.id);
|
||||||
|
const roverId = resolveRoverId(socket?.id);
|
||||||
|
const typingPayload = buildTypingPayload(socket, { roverId, fromDiscord: false, isTyping: false });
|
||||||
|
broadcastTyping(typingPayload);
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
subscribe('chat:message', ({ payload }) => {
|
||||||
|
if (!payload) return;
|
||||||
|
io.sockets.sockets.forEach((socket) => {
|
||||||
|
if (!isBannedSocket(socket)) {
|
||||||
|
socket.emit('chat:message', payload);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
subscribe('chat:typing', ({ payload }) => {
|
||||||
|
if (!payload) return;
|
||||||
|
io.sockets.sockets.forEach((socket) => {
|
||||||
|
if (!isBannedSocket(socket)) {
|
||||||
|
socket.emit('chat:typing', payload);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
module.exports = {
|
||||||
|
handleIncoming,
|
||||||
|
sendExternalMessage,
|
||||||
|
sendExternalTyping,
|
||||||
|
buildTypingPayload,
|
||||||
|
};
|
||||||
Some files were not shown because too many files have changed in this diff Show More
Reference in New Issue
Block a user