Added create block RPC wrapper

static-data/official-plugins/rpc/longrpc.py

@@ -2,20 +2,16 @@
 
 """
 import threading
-import os
-import time
-import traceback
 import collections
 from typing import Union
 
 import ujson
 import jsonrpc
 
-from logger import log as logging
-
 
 rpc_results = collections.deque(maxlen=10000)
 
+
 def get_results(id) -> Union[str, None]:
     final = None
     for result in rpc_results:
@@ -29,10 +25,12 @@ def get_results(id) -> Union[str, None]:
 
 
 def _exec_rpc(rpc_json_str):
-    json_resp = jsonrpc.JSONRPCResponseManager.handle(rpc_json_str, jsonrpc.dispatcher)
+    json_resp = jsonrpc.JSONRPCResponseManager.handle(
+        rpc_json_str, jsonrpc.dispatcher)
     data = json_resp.data
     rpc_results.append(data)
 
+
 def threaded_rpc(rpc_json_str):
     threading.Thread(
         target=_exec_rpc,

static-data/official-plugins/rpc/main.py

@@ -57,6 +57,7 @@ import longrpc
 
 plugin_apis['rpc.add_module_to_api'] = add_module_to_api
 
+
 def _detect_cors_and_add_headers():
     cherrypy.response.headers['Access-Control-Allow-Headers'] = 'Content-Type'
     cherrypy.response.headers['Access-Control-Allow-Origin'] = '*'
@@ -65,6 +66,7 @@ def _detect_cors_and_add_headers():
         return True
     return False
 
+
 class OnionrRPC(object):
     @cherrypy.expose
     def threaded_rpc(self):

static-data/official-plugins/rpc/rpc/blocks/__init__.py

@@ -1,26 +1,39 @@
 from secrets import randbits
 import base64
-from base64 import b85decode
+from typing import Union
 
 from onionrblocks import Block
 import onionrblocks
 from jsonrpc import dispatcher
 
 from gossip.blockqueues import gossip_block_queues
-from blockdb import get_blocks_after_timestamp
+import blockdb
 from utils import multiproc
 
 
+@dispatcher.add_method
+def get_block(block_id: str) -> dict:
+    bl = blockdb.get_block(block_id)
+    
+
+
 @dispatcher.add_method
 def get_blocks(timestamp):
-    return [block.raw for block in get_blocks_after_timestamp(timestamp)]
+    blocks = []
+    for block in blockdb.get_blocks_after_timestamp(timestamp):
+        blocks.append({
+            'id': block.id,
+            'raw': base64.b64encode(block.raw).decode('utf-8')
+        })
+    return blocks
 
 
 @dispatcher.add_method
 def create_block(
         block_data: 'base64', block_type: str, ttl: int, metadata: dict):
-    # Wrapper for onionrblocks.create_block (take base64 to be compatible with RPC)
-    bl = multiproc.subprocess_compute(
+    # Wrapper for onionrblocks.create_block
+    # (take base64 to be compatible with RPC)
+    bl: Block = multiproc.subprocess_compute(
         onionrblocks.create_anonvdf_block,
         3600,
         base64.b64decode(block_data),
@@ -28,14 +41,35 @@ def create_block(
         ttl,
         **metadata
         )
+    try:
+        block_id = bl.id.decode('utf-8')
+    except AttributeError:
+        block_id = bl.id
+    bl_json = {
+        'id': block_id,
+        'raw': base64.b64encode(bl.raw).decode('utf-8')
+    }
+    return bl_json
 
-    return base64.b85encode(bl.raw).decode('utf-8')
 
-queue_to_use = randbits(1)
 @dispatcher.add_method
-def insert_block(block):
-    block = Block(
-        block['id'], b85decode(block['raw']), auto_verify=False)
+def create_and_insert_block(
+        block_data: 'base64',
+        block_type: str, ttl: int, metadata: dict) -> str:
+    bl = create_block(block_data, block_type, ttl, metadata)['id']
+    insert_block(bl)
+    return bl['id']
+
+
+# As per dandelion++ spec the edge should be the same.
+# We keep it the same for each daemon life time.
+queue_to_use = randbits(1)
+
+
+@dispatcher.add_method
+def insert_block(block: Union[dict, Block]):
+    if isinstance(block, dict):
+        block = Block(
+            block['id'], base64.b64decode(block['raw']), auto_verify=False)
     gossip_block_queues[queue_to_use].put_nowait(block)
     return "ok"
-