#!/usr/bin/python
# -*- coding:utf-8 -*-
import sys
import os
picdir = os.path.join(os.path.dirname(os.path.dirname(os.path.realpath(__file__))), 'pic')
libdir = os.path.join(os.path.dirname(os.path.dirname(os.path.realpath(__file__))), 'lib')
if os.path.exists(libdir):
    sys.path.append(libdir)

import logging
from waveshare_epd import epd2in13b_V3
import time
from PIL import Image,ImageDraw,ImageFont
import traceback

# Import MQTT client.
from paho.mqtt import client as mqtt_client


logging.basicConfig(level=logging.DEBUG)

# MQTT credentials
#mqtt_username  = 'homeassistant' # MQTT client username
#mqtt_password = '3355' # MQTT client password
client_id = 'solarmonitor'  # unique client_id
mqtt_broker = "openwb-2"  # broker address, usually your HASS IP address
port = 1883

#set topics
topic_house_battery_soc = "openWB/bat/get/soc"
topic_house_battery_power = "openWB/bat/get/power"

topic_power_import_export = "openWB/counter/7/get/power"
topic_pv_power = "openWB/pv/8/get/power"

topic_house_power = "openWB/graph/lastlivevaluesJson" #house-power

topic_car_soc = "openWB/vehicle/4/get/soc"
topic_charge_point_power = "openWB/internal_chargepoint/0/get/powers"


def connect_mqtt():
    def on_connect(client, userdata, flags, rc):
        if rc == 0:
            logging.info("Connected to MQTT Broker!")
        else:
            logging.info("Failed to connect, return code %d\n", rc)

    client = mqtt_client.Client(client_id)
    # client.username_pw_set(username, password)
    client.on_connect = on_connect
    client.connect(mqtt_broker, port)
    return client


def subscribe(client: mqtt_client, topic):
    def on_message(client, userdata, msg):
        print(f"Received `{msg.payload.decode()}` from `{msg.topic}` topic")

    client.subscribe(topic)
    client.on_message = on_message


def run():
    try:
        client = connect_mqtt()
        house_battery_soc = subscribe(client, topic_house_battery_soc)
        house_battery_power = subscribe(client, topic_house_battery_power)
        power_import_export = subscribe(client, topic_power_import_export)
        pv_power = subscribe(client, topic_pv_power)
        house_power = subscribe(client, topic_house_power)
        car_soc = subscribe(client, topic_car_soc)
        charge_point_power = subscribe(client, topic_charge_point_power)

        logging.info("epd2in13b_V3 Demo")
        
        epd = epd2in13b_V3.EPD()
        #logging.info("init and Clear")
        #epd.init()
        #epd.Clear()
        
        # Drawing on the image
        logging.info("Drawing")    
        font16 = ImageFont.truetype(os.path.join(picdir, 'Font.ttc'), 16)
        font18 = ImageFont.truetype(os.path.join(picdir, 'Font.ttc'), 18)
        
        # Drawing on the Horizontal image
        logging.info("1.Drawing on the Horizontal image...") 
        HBlackimage = Image.new('1', (epd.height, epd.width), 255)  # 298*126
        HRYimage = Image.new('1', (epd.height, epd.width), 255)  # 298*126  ryimage: red or yellow image  
        drawblack = ImageDraw.Draw(HBlackimage)
        drawry = ImageDraw.Draw(HRYimage)
        drawblack.text((2, 0), 'PV Production', font = font16, fill = 0)
        drawblack.text((2, 16), 'Hausverbrauch', font = font16, fill = 0)
        drawblack.text((2, 32), 'Strom Import', font = font16, fill = 0)
        drawblack.text((2, 48), 'Batterie', font = font16, fill = 0)
        drawblack.text((2, 64), 'Batterie Entladen', font = font16, fill = 0)
        drawblack.text((2, 80), 'Auto Laden', font = font16, fill = 0)

        drawblack.text((100, 80), str(charge_point_power), font = font16, fill = 0)
        epd.display(epd.getbuffer(HBlackimage), epd.getbuffer(HRYimage))

        #logging.info("Clear...")
        #epd.init()
        #epd.Clear()
        
        logging.info("Goto Sleep...")
        epd.sleep()

        client.loop_forever()
    
    except IOError as e:
        logging.info(e)
    
    except KeyboardInterrupt:    
        logging.info("ctrl + c:")
        epd2in13b_V3.epdconfig.module_exit()
        exit()



if __name__ == '__main__':
    run()