import threading
import queue
import json


class NetworkReader(threading.Thread):
  def __init__(self,stream):
    threading.Thread.__init__(self,daemon=True)
    self.parent=threading.current_thread()
    
    self.stream=stream
    self.output=queue.Queue()
    
  def run(self):
    while True:
      if not self.parent.is_alive(): return
      self.output.put(self.readMsg(),timeout=2)
      
  def readMsg(self):
    data=self.stream.readline()
    if not data: pass # !! raise something
    jsonLength=int(data.split(b":")[1].strip()) # "json-length: length" -> length
    data=self.stream.readline()
    if not data: pass # !! raise something
    binLength=int(data.split(b":")[1].strip()) # "bin-length: length" -> length
    jsonData=self.stream.read(jsonLength)
    jsonData=json.loads(str(jsonData,encoding="utf-8"))
    binData=self.stream.read(binLength)
    
    return (jsonData,binData)
    

class NetworkWriter(threading.Thread):
  def __init__(self,stream):
    threading.Thread.__init__(self,daemon=True)
    self.parent=threading.current_thread()
    
    self.stream=stream
    self.input=queue.Queue()
    
  def run(self):
    while True:
      if not self.parent.is_alive(): return
      msg=self.input.get(timeout=2)
      if msg is None: return
      self.stream.write(self.writeMsg(msg))
      self.stream.flush()
      
  def writeMsg(self,msg):
    jsonData,binData=msg
    jsonData=bytes(json.dumps(jsonData)+"\n",encoding="utf-8")
    jsonLength=bytes("json-length: "+str(len(jsonData))+"\n",encoding="utf-8")
    binLength=bytes("bin-length: "+str(len(binData))+"\n",encoding="utf-8")
    return b"".join((jsonLength,binLength,jsonData,binData))