import gc
import network
import socket

from st3m.application import Application, ApplicationContext

from umqtt.robust import MQTTClient

WIFI_SSID = b'...'
WIFI_PSK = b'...'

MQTT_BROKER_IP = '...'
MQTT_CLIENT_ID = b'flow3r-{}'
MQTT_BASE_TOPIC = b'flow3r-{}/event'


class Flow3rOpenhabMqtt(Application):

    def __init__(self, app_ctx: ApplicationContext) -> None:
        super().__init__(app_ctx)
        self.state = 0
        self.nic = network.WLAN(network.STA_IF)
        client_id = MQTT_CLIENT_ID.format(self.nic.config('mac').hex())
        self.base_topic = MQTT_BASE_TOPIC.format(self.nic.config('mac').hex())
        self.mqtt = MQTTClient(client_id, MQTT_BROKER_IP)
        self.before = [0]*10

    def draw(self, ctx: Context) -> None:
        if self.state == 0:
            return
        ctx.line_width = 18
        ctx.line_join = ctx.BEVEL
        ctx.rgb(0, 0, 0).rectangle(-120, -120, 240, 240).fill()
        ctx.rgb(0.3, 0.3, 0.3).arc(0, 0, 90, 0, 2.39, 0).arc(0, 0, 90, 2.79, 6.29, 0).stroke()
        ctx.rgb(0.9, 0.3, 0.1).move_to(-96, 53).line_to(-1, -43).line_to(71, 29).stroke()

    def do_connect(self):
        self.nic.active(True)
        if not self.nic.isconnected():
            print('connecting to network...')
            self.nic.connect(WIFI_SSID, WIFI_PSK)
            while not self.nic.isconnected():
                pass
        print('network config:', self.nic.ifconfig())

    def think(self, ins: InputState, delta_ms: int) -> None:
        if self.nic.status() not in [network.STAT_CONNECTING, network.STAT_GOT_IP]:
            self.do_connect()
        if self.nic.status() == network.STAT_GOT_IP and self.state == 0:
            gc.collect()
            self.mqtt.connect()
            self.state = 1
        if self.state == 1:
            for i in range(5):
                if ins.captouch.petals[i*2].pads.cw or ins.captouch.petals[i*2].pads.ccw:
                    if self.before[i*2] == 0:
                        self.mqtt.publish(self.base_topic, b'{{"petal":"{}","event":"press_tip"}}'.format(i))
                    self.before[i*2] = 1
                else:
                    self.before[i*2] = 0
                if ins.captouch.petals[i*2].pads.base:
                    if self.before[i*2+1] == 0:
                        self.mqtt.publish(self.base_topic, b'{{"petal":"{}","event":"press_base"}}'.format(i))
                    self.before[i*2+1] = 1
                else:
                    self.before[i*2+1] = 0


if __name__ == "__main__":
    import st3m.run
    st3m.run.run_view(Flow3rOpenhabMqtt(ApplicationContext()))