openethereum/src/service.rs

use util::*;
use sync::*;
use spec::Spec;
use error::*;
use std::env;
use client::Client;

/// Client service setup. Creates and registers client and network services with the IO subsystem.
pub struct ClientService {
	net_service: NetworkService<SyncMessage>,
	client: Arc<RwLock<Client>>,
}

impl ClientService {
	/// Start the service in a separate thread.
	pub fn start(spec: Spec, init_nodes: &Vec<String>) -> Result<ClientService, Error> {
		let mut net_service = try!(NetworkService::start(init_nodes));
		info!("Starting {}", net_service.host_info());
		info!("Configured for {} using {} engine", spec.name, spec.engine_name);
		let mut dir = env::home_dir().unwrap();
		dir.push(".parity");
		dir.push(H64::from(spec.genesis_header().hash()).hex());
		let client = Arc::new(RwLock::new(try!(Client::new(spec, &dir, net_service.io().channel()))));
		EthSync::register(&mut net_service, client.clone());
		let client_io = Box::new(ClientIoHandler {
			client: client.clone()
		});
		try!(net_service.io().register_handler(client_io));

		Ok(ClientService {
			net_service: net_service,
			client: client,
		})
	}

	/// Get the network service.
	pub fn add_node(&mut self, _enode: &str) {
		unimplemented!();
	}

	/// TODO [arkpar] Please document me
	pub fn io(&mut self) -> &mut IoService<NetSyncMessage> {
		self.net_service.io()
	}

	/// TODO [arkpar] Please document me
	pub fn client(&self) -> Arc<RwLock<Client>> {
		self.client.clone()
	}
}

/// IO interface for the Client handler
struct ClientIoHandler {
	client: Arc<RwLock<Client>>
}

impl IoHandler<NetSyncMessage> for ClientIoHandler {
	fn initialize<'s>(&'s mut self, _io: &mut IoContext<'s, NetSyncMessage>) {
	}

	fn message<'s>(&'s mut self, _io: &mut IoContext<'s, NetSyncMessage>, net_message: &'s mut NetSyncMessage) {
		match net_message {
			&mut UserMessage(ref mut message) =>  {
				match message {
					&mut SyncMessage::BlockVerified => {
						self.client.write().unwrap().import_verified_blocks();
					},
					_ => {}, // ignore other messages
				}

			}
			_ => {}, // ignore other messages
		}

	}
}