#!/usr/bin/env python3
import asyncio
import logging
import os.path
import include_homekit

from argparse import ArgumentParser
from homekit.camera.esp32 import WebClient
from homekit.util import Addr
from apscheduler.schedulers.asyncio import AsyncIOScheduler
from datetime import datetime
from typing import Optional

logger = logging.getLogger(__name__)
cam: Optional[WebClient] = None


class ESP32Capture:
    def __init__(self, addr: Addr, interval: float, output_directory: str):
        self.logger = logging.getLogger(self.__class__.__name__)
        self.client = WebClient(addr)
        self.output_directory = output_directory
        self.interval = interval

        self.scheduler = AsyncIOScheduler()
        self.scheduler.add_job(self.capture, 'interval', seconds=arg.interval)
        self.scheduler.start()

    async def capture(self):
        self.logger.debug('capture: start')
        now = datetime.now()
        filename = os.path.join(
            self.output_directory,
            now.strftime('%Y-%m-%d-%H:%M:%S.%f.jpg')
        )
        if not await self.client.capture(filename):
            self.logger.error('failed to capture')
        self.logger.debug('capture: done')


if __name__ == '__main__':
    parser = ArgumentParser()
    parser.add_argument('--addr', type=str, required=True)
    parser.add_argument('--output-directory', type=str, required=True)
    parser.add_argument('--interval', type=float, default=0.5)
    parser.add_argument('--verbose', action='store_true')
    arg = parser.parse_args()

    if arg.verbose:
        logging.basicConfig(level=logging.DEBUG)

    loop = asyncio.get_event_loop()

    ESP32Capture(Addr.fromstring(arg.addr), arg.interval, arg.output_directory)
    try:
        loop.run_forever()
    except KeyboardInterrupt:
        pass