from dotenv import load_dotenv, find_dotenv
import os
import zerorpc
import gevent
import time

load_dotenv(find_dotenv())

class CortexRPC(object):
  def __init__(self, fn):
    super().__init__() 
    self._listener = fn
    self.connect()

  def connect(self):
    print("Connecting to RPC on port {}".format(os.getenv('RPC_PORT')))
    self.rpc = zerorpc.Server(self)
    self.rpc.bind("tcp://127.0.0.1:" + os.getenv('RPC_PORT'))
    gevent.spawn(self.connected)
    self.rpc.run()

  def connected(self):
    print("RPC running on port {}".format(os.getenv('RPC_PORT')))
    print("Connecting to Relay on port {}".format(os.getenv('RELAY_PORT')))
    self.relay = zerorpc.Client()
    self.relay.connect("tcp://127.0.0.1:" + os.getenv('RELAY_PORT'))
    val = None
    try:
      val = self.relay.connected("rpc")
      print("RPC connected...".format(val))
    except zerorpc.TimeoutExpired:
      print("RPC timeout :(")
    # return val
    print("defer sending a file...")
    gevent.sleep(1)
    return self.get_last_frame()

  def get_last_frame(self):
    self.send_frame("test.png", '01234567' * 8 * 8 * 8 * 8 * 2)
    return "ok"

  def send_param(self, key, value):
    print("aloha. {}: {}".format(key, value))
    if self._listener is not None:
      self._listener(key, value)
    else:
      print("No listener provided to CortexRPC")
    return "ok"

  def get_params(self):
    print("Get params call received!")
    return "lmao ok"

  def send_frame(self, fn, frame):
    if self.relay is not None:
      print("Sending a frame...")
      self.relay.send_frame(fn, frame)
    else:
      print("Can't send frame, relay not ready...")

if __name__ == '__main__':
  def log_param(key, value):
    print("Log param. {}: {}".format(key, value))

  # this will block
  rpc = CortexRPC(log_param)