# SPDX-FileCopyrightText: Copyright (c) 2022 Jeff Epler for Adafruit Industries
#
# SPDX-License-Identifier: MIT

"""
Show the live camera image on the viewfinder, then upload to adafruit IO when
the 'BOOT' button is pressed.
"""

import binascii
import os
import ssl
import struct

from adafruit_io.adafruit_io import IO_MQTT
import adafruit_minimqtt.adafruit_minimqtt as MQTT
import board
import espcamera
import keypad
import socketpool
import wifi

shutter_button = keypad.Keys((board.BOOT,), value_when_pressed=False)

aio_username = os.getenv("AIO_USERNAME")
aio_key = os.getenv("AIO_KEY")

image_feed = "image"

pool = socketpool.SocketPool(wifi.radio)

print("Connecting to Adafruit IO")
mqtt_client = MQTT.MQTT(
    broker="io.adafruit.com",
    username=aio_username,
    password=aio_key,
    socket_pool=pool,
    ssl_context=ssl.create_default_context(),
)
mqtt_client.connect()
io = IO_MQTT(mqtt_client)


print("Initializing camera")
cam = espcamera.Camera(
    data_pins=board.CAMERA_DATA,
    external_clock_pin=board.CAMERA_XCLK,
    pixel_clock_pin=board.CAMERA_PCLK,
    vsync_pin=board.CAMERA_VSYNC,
    href_pin=board.CAMERA_HREF,
    pixel_format=espcamera.PixelFormat.RGB565,
    frame_size=espcamera.FrameSize.R240X240,
    i2c=board.I2C(),
    external_clock_frequency=20_000_000,
    framebuffer_count=2,
)
cam.vflip = True
cam.hmirror = True

board.DISPLAY.auto_refresh = False
display_bus = board.DISPLAY.bus

print(cam.width, cam.height)

ow = (board.DISPLAY.width - cam.width) // 2
oh = (board.DISPLAY.height - cam.height) // 2
display_bus.send(42, struct.pack(">hh", ow, cam.width + ow - 1))
display_bus.send(43, struct.pack(">hh", oh, cam.height + ow - 1))

while True:
    frame = cam.take(1)
    display_bus.send(44, frame)
    if (ev := shutter_button.events.get()) and ev.pressed:
        cam.reconfigure(
            pixel_format=espcamera.PixelFormat.JPEG,
            frame_size=espcamera.FrameSize.SVGA,
        )
        frame = cam.take(1)
        if isinstance(frame, memoryview):
            jpeg = frame
            print(f"Captured {len(jpeg)} bytes of jpeg data")

            # b2a_base64() appends a trailing newline, which IO does not like
            encoded_data = binascii.b2a_base64(jpeg).strip()
            print(f"Expanded to {len(encoded_data)} for IO upload")

            io.publish("image", encoded_data)
        cam.reconfigure(
            pixel_format=espcamera.PixelFormat.RGB565,
            frame_size=espcamera.FrameSize.R240X240,
        )
    print(end=".")