#!/usr/bin/env python
from deps.samplebase import SampleBase
from rgbmatrix import graphics

from flask import Flask, request, jsonify
from waitress import serve
import time, threading


api = Flask(__name__)

@api.route('/text', methods=['POST'])
def display_text():
	# prepare response data
	response = { 'success': True }
	try:
		global matrix

		# receive client data
		data = request.get_json()
		# try unpacking text and x,y coordinates 
		text = data.get('text')
		x_coord = data.get('x')
		y_coord = data.get('y')

		# call matrix method with data
		matrix.text(x_coord, y_coord, text)
	except Exception as e:
		# error handling
		print(e)
		response['success'] = False

	# respond to client
	return jsonify(response)

@api.route('/pixel', methods=['POST'])
def set_pixel():
	# prepare response data
	response = { 'success': True }
	try:
		global matrix

		# receive client data
		data = request.get_json()
		# try unpacking x,y coordinates 
		x_coord = data.get('x')
		y_coord = data.get('y')

		# call matrix method with data
		matrix.pixel(x_coord, y_coord)
	except Exception as e:
		# error handling
		print(e)
		response['success'] = False

	# respond to client
	return jsonify(response)

@api.route('/circle', methods=['POST'])
def draw_circle():
	# prepare response data
	response = { 'success': True }
	try:
		global matrix

		# receive client data
		data = request.get_json()
		# try unpacking radius and x,y coordinates 
		radius = data.get('r')
		x_coord = data.get('x')
		y_coord = data.get('y')

		# call matrix method with data
		matrix.circle(x_coord, y_coord, radius)
	except Exception as e:
		# error handling
		print(e)
		response['success'] = False

	# respond to client
	return jsonify(response)

@api.route('/rectangle', methods=['POST'])
def draw_rectangle():
	# prepare response data
	response = { 'success': True }
	try:
		global matrix

		# receive client data
		data = request.get_json()
		# try unpacking rectangle width and height
		width = data.get('w')
		height = data.get('h')
		# try unpacking x,y coordinates 
		x_coord = data.get('x')
		y_coord = data.get('y')

		# call matrix method with data
		matrix.rectangle(x_coord, y_coord, width, height)
	except Exception as e:
		# error handling
		print(e)
		response['success'] = False

	# respond to client
	return jsonify(response)

@api.route('/color', methods=['POST'])
def set_color():
	# prepare response data
	response = { 'success': True }
	try:
		global matrix

		# receive client data
		data = request.get_json()
		# try unpacking red, green and blue color values 
		red = data.get('r')
		green = data.get('g')
		blue = data.get('b')

		# call matrix method with data
		matrix.set_color(red, green, blue)
	except Exception as e:
		# error handling
		print(e)
		response['success'] = False

	# respond to client
	return jsonify(response)

@api.route('/clear', methods=['GET'])
def clear_canvas():
	# prepare response data
	response = { 'success': True }
	try:
		global matrix

		# receive client data
		matrix.clear()
	except Exception as e:
		# error handling
		print(e)
		response['success'] = False

	return jsonify(response)

@api.route('/update', methods=['GET'])
def update_matrix():
	response = { 'success': True }
	try:
		global matrix
		matrix.update()
	except Exception as e:
		print(e)
		response['success'] = False

	# respond to client
	return jsonify(response)


# derived class to access rgb-led-matrix
class Matrix(SampleBase):
	def __init__(self, *args, **kwargs):
		super(Matrix, self).__init__(*args, **kwargs)

	# run display program
	def run(self):
		# initialize font
		self.font = graphics.Font()
		self.font.LoadFont("deps/fonts/9x18B.bdf")
		# initialize color
		self.color = graphics.Color(255, 255, 255)
		
		# initialize canvas
		self.canvas = self.matrix.CreateFrameCanvas()
		self.canvas.Clear()

		# wait in idle
		while True:
			time.sleep(.05)

	# clear canvas
	def clear(self):
		self.canvas.Clear()

	# swap canvas with buffer
	def update(self):
		self.canvas = self.matrix.SwapOnVSync(self.canvas)
		self.clear()

	# set current color
	def set_color(self, r, g, b):
		self.color = graphics.Color(r, g, b)

	# set specified pixel to current color
	def pixel(self, x, y):
		self.canvas.SetPixel(x, y, self.color.red, self.color.green, self.color.blue)

	# display text at position with current color
	def text(self, x, y, t):
		graphics.DrawText(self.canvas, self.font, x*9, (y+1)*18, self.color, t)

	# display circle at position with radius in current color
	def circle(self, x, y, r):
		graphics.DrawCircle(self.canvas, self.font, x, y, r, self.color)

	# display rectangle at position with dimensions in current color
	def rectangle(self, x, y, w, h):
		# loop through each point in dimensions
		for i in range(w):
			for j in range(h):
				# set next pixel in rectangle
				self.canvas.SetPixel(x+i, y+j, self.color.red, self.color.green, self.color.blue)


# initialize global matrix variable
matrix = None
if __name__ == "__main__":
	# set matrix to instance
	matrix = Matrix()
	# run rest api in thread, serving with 'waitress'
	threading.Thread(target=lambda: serve(api, host="0.0.0.0", port=8080)).start()

	# help menu on call
	if not matrix.process():
		matrix.print_help()