47 lines
1.5 KiB
Rust
47 lines
1.5 KiB
Rust
|
use bevy::{app::{App, Plugin, Startup, Update}, ecs::{component::Component, query::With, system::{Commands, Query}}, tasks::TaskPool};
|
||
|
|
use caretta_sync_core::{cache::entity::{CachedPeerEntity, CachedPeerModel}, global::{CONFIG, DATABASE_CONNECTIONS}};
|
||
|
|
use caretta_sync_core::{
|
||
|
|
proto::*,
|
||
|
|
};
|
||
|
|
use sea_orm::EntityTrait;
|
||
|
|
|
||
|
|
#[derive(Component)]
|
||
|
|
pub struct Peer;
|
||
|
|
|
||
|
|
#[derive(Component)]
|
||
|
|
pub struct PeerId(String);
|
||
|
|
|
||
|
|
#[derive(Component)]
|
||
|
|
pub struct PeerAddress(String);
|
||
|
|
|
||
|
|
#[tokio::main]
|
||
|
|
async fn add_cached_peers(mut commands: Commands) {
|
||
|
|
let config = CONFIG.get_unchecked();
|
||
|
|
let path = String::from("unix://") + config.rpc.socket_path.as_os_str().to_str().expect("Invalid string");
|
||
|
|
let mut client = caretta_sync_core::proto::cached_peer_service_client::CachedPeerServiceClient::connect(path).await.expect("Unix socket should be accessible");
|
||
|
|
let request = tonic::Request::new(CachedPeerListRequest {});
|
||
|
|
let response = client.list(request).await.expect("Faild to request/response");
|
||
|
|
let peers = response.into_inner().peers;
|
||
|
|
for model in peers.into_iter() {
|
||
|
|
commands.spawn((Peer, PeerId(model.peer_id.to_string())));
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
fn print_peer(query: Query<&PeerId, With<Peer>>) {
|
||
|
|
for peer_id in &query {
|
||
|
|
println!("Hello {}!", peer_id.0);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
fn hello_world() {
|
||
|
|
println!("hello world!");
|
||
|
|
}
|
||
|
|
|
||
|
|
pub struct PeerPlugin;
|
||
|
|
|
||
|
|
impl Plugin for PeerPlugin {
|
||
|
|
fn build(&self, app: &mut App) {
|
||
|
|
app.add_systems(Startup, add_cached_peers);
|
||
|
|
app.add_systems(Update, (hello_world, print_peer));
|
||
|
|
}
|
||
|
|
}
|