#!/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 waveshare-epaper
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
topic0 = {
    "topic" : "openWB/bat/get/soc",
    "line" : "0",
    "value" : ""
}
topic1 = {
    "topic" : "openWB/bat/get/power",
    "line" : "1",
    "value" : ""
}
topic2 = {
    "topic" : "openWB/counter/7/get/power",
    "line" : "2",
    "value" : ""
}
topic3 = {
    "topic" : "openWB/pv/8/get/power",
    "line" : "3",
    "value" : ""
}
topic4 = {
    "topic" : "openWB/graph/lastlivevaluesJson", #house-power
    "line" : "4",
    "value" : ""
}
topic5 = {
    "topic" : "openWB/vehicle/4/get/soc",
    "line" : "5",
    "value" : ""
}
topic6 = {
    "topic" : "openWB/internal_chargepoint/0/get/powers",
    "line" : "6",
    "value" : ""
}

topics = {0:topic0,1:topic1,2:topic2,3:topic3,4:topic4,5:topic5,6:topic6}


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.loop_start()
    client.subscribe(topic)
    client.on_message = on_message
    client.loop_stop()

def monitor(topics):
    
    logging.info("epd2in13b_V3 Demo")
    logging.info(libdir)
    logging.info(picdir)
    
    epd = epd2in13b_V3.EPD()
    #epd = epaper.epaper('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)

    for topic_id, topic_info in topics.items():
        print("\nTopic ID:", topic_id)
    
        for key in topic_info:
            print(key + ':', topic_info[key])
            drawblack.text((160, int(topic_info["line"])*16), topic_info["value"], font = font16, fill = 0)
            topic_info["value"] = subscribe(client, topic_info["topic"])

    #drawblack.text((160, 16), str(house_power), font = font16, fill = 0)
    #drawblack.text((160, 32), str(power_import_export), font = font16, fill = 0)
    #drawblack.text((160, 48), str(house_battery_soc), font = font16, fill = 0)
    #drawblack.text((160, 64), str(house_battery_power), font = font16, fill = 0)
    #drawblack.text((160, 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()

    #except IOError as e:
    #    logging.info(e)

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


def run():
    client = connect_mqtt()

    for topic_id, topic_info in topics.items():
        print("\nTopic ID:", topic_id)
    
        for key in topic_info:
            print(key + ':', topic_info[key])
            topic_info["value"] = subscribe(client, topic_info["topic"])

    try:
        monitor(topics)

    except IOError as e:
        logging.info(e)

    except KeyboardInterrupt:    
        logging.info("ctrl + c:")
        epd2in13b_V3.epdconfig.module_exit()
        exit()


if __name__ == '__main__':
    run()