From 2ee3a7282b279047697aa392901d516fd04bd23f Mon Sep 17 00:00:00 2001
From: Robert Habermeier <rphmeier@gmail.com>
Date: Fri, 17 Mar 2017 00:14:29 +0100
Subject: [PATCH] adjust to minor on_demand API changes in RPC

---
 rpc/src/v1/helpers/dispatch.rs |  5 ++++-
 rpc/src/v1/impls/light/eth.rs  | 23 +++++++++++++++--------
 2 files changed, 19 insertions(+), 9 deletions(-)

diff --git a/rpc/src/v1/helpers/dispatch.rs b/rpc/src/v1/helpers/dispatch.rs
index b11ada048..36d1b330f 100644
--- a/rpc/src/v1/helpers/dispatch.rs
+++ b/rpc/src/v1/helpers/dispatch.rs
@@ -268,7 +268,10 @@ impl LightDispatcher {
 		}));
 
 		match nonce_future {
-			Some(x) => x.map(|acc| acc.nonce).map_err(|_| errors::no_light_peers()).boxed(),
+			Some(x) =>
+				x.map(|acc| acc.map_or_else(Default::default, |acc| acc.nonce))
+					.map_err(|_| errors::no_light_peers())
+					.boxed(),
 			None =>  future::err(errors::network_disabled()).boxed()
 		}
 	}
diff --git a/rpc/src/v1/impls/light/eth.rs b/rpc/src/v1/impls/light/eth.rs
index f889faf00..a35d48fb6 100644
--- a/rpc/src/v1/impls/light/eth.rs
+++ b/rpc/src/v1/impls/light/eth.rs
@@ -105,15 +105,22 @@ impl EthClient {
 				match cht_root {
 					None => return future::ok(None).boxed(),
 					Some(root) => {
-						let req = request::HeaderByNumber::new(n, root)
+						let req = request::HeaderProof::new(n, root)
 							.expect("only fails for 0; client always stores genesis; client already queried; qed");
 
-						self.sync.with_context(|ctx|
-							self.on_demand.header_by_number(ctx, req)
-								.map(Some)
-								.map_err(err_premature_cancel)
-								.boxed()
-						)
+						let (sync, on_demand) = (self.sync.clone(), self.on_demand.clone());
+						self.sync.with_context(|ctx| {
+							let fut = self.on_demand.hash_by_number(ctx, req)
+								.map(request::HeaderByHash)
+								.map_err(err_premature_cancel);
+
+							fut.and_then(move |req| {
+								match sync.with_context(|ctx| on_demand.header_by_hash(ctx, req)) {
+									Some(fut) => fut.map_err(err_premature_cancel).boxed(),
+									None => future::err(errors::network_disabled()).boxed(),
+								}
+							}).map(Some).boxed()
+						})
 					}
 				}
 			}
@@ -149,7 +156,7 @@ impl EthClient {
 			sync.with_context(|ctx| on_demand.account(ctx, request::Account {
 				header: header,
 				address: address,
-			}).map(Some))
+			}))
 				.map(|x| x.map_err(err_premature_cancel).boxed())
 				.unwrap_or_else(|| future::err(errors::network_disabled()).boxed())
 		}).boxed()