From a78abe8b90f3d73056957e1f2c31d109d53df8cd Mon Sep 17 00:00:00 2001
From: Neil Booth <kyuupichan@gmail.com>
Date: Fri, 11 Jan 2019 20:18:16 +0800
Subject: [PATCH] Remove residual electrum-header cruft

Fixes #684
---
 electrumx/server/session.py | 14 ++++----------
 1 file changed, 4 insertions(+), 10 deletions(-)

diff --git a/electrumx/server/session.py b/electrumx/server/session.py
index a319cc9..1168034 100644
--- a/electrumx/server/session.py
+++ b/electrumx/server/session.py
@@ -342,7 +342,7 @@ class SessionManager(object):
         # Paranoia: a reorg could race and leave db_height lower
         height = min(height, self.db.db_height)
         raw = await self.raw_header(height)
-        self.hsub_results = (electrum, {'hex': raw.hex(), 'height': height})
+        self.hsub_results = {'hex': raw.hex(), 'height': height}
         self.notified_height = height
 
     # --- LocalRPC command handlers
@@ -697,7 +697,6 @@ class ElectrumX(SessionBase):
     def __init__(self, *args, **kwargs):
         super().__init__(*args, **kwargs)
         self.subscribe_headers = False
-        self.subscribe_headers_raw = False
         self.connection.max_response_size = self.env.max_send
         self.max_subs = self.env.max_session_subs
         self.hashX_subs = {}
@@ -780,17 +779,12 @@ class ElectrumX(SessionBase):
 
     async def subscribe_headers_result(self):
         '''The result of a header subscription or notification.'''
-        return self.session_mgr.hsub_results[self.subscribe_headers_raw]
-
-    async def _headers_subscribe(self, raw):
-        '''Subscribe to get headers of new blocks.'''
-        self.subscribe_headers_raw = assert_boolean(raw)
-        self.subscribe_headers = True
-        return await self.subscribe_headers_result()
+        return self.session_mgr.hsub_results
 
     async def headers_subscribe(self):
         '''Subscribe to get raw headers of new blocks.'''
-        return await self._headers_subscribe(True)
+        self.subscribe_headers = True
+        return await self.subscribe_headers_result()
 
     async def add_peer(self, features):
         '''Add a peer (but only if the peer resolves to the source).'''