import asyncio
from datetime import datetime
from asyncio_dvrip import DVRIPCam

from nvr_types import File as NvrFile
from nvr_types import list_local_files
from nvr_types import PRIMARY_STREAM, SECONDARY_STREAM
from nvr_types import H264

START = "2024-08-04 6:22:34"
END = "2024-08-04 23:23:09"
CHANNEL = 0

def date_today(begin = True):
    if begin:
        return datetime.now().strftime("%Y-%m-%d 00:00:00")
    else:
        return datetime.now().strftime("%Y-%m-%d 23:59:59")

class NVR:
    def __init__(self, client, loop) -> None:
        self.client:DVRIPCam = client
        self.loop = loop

    async def login(self):
        await self.client.login(self.loop)

    def logout(self):
        self.client.close()

    async def channels(self):
        return await self.client.get_command("ChannelTitle", 1048)

    async def files(self, channel, start = None, end = None, ftype = H264, stype = SECONDARY_STREAM, json = False):
        if not start:
            start = date_today()
        if not end:
            end = date_today(False)
        print("Search files", start, end)
        for raw_file in await list_local_files(self.client, startTime=start, endTime=end, filetype=ftype, channel=channel, streamType=stype):
            if json:
                yield NvrFile(raw_file, channel, stype).json
            else:
                yield NvrFile(raw_file, channel, stype)

    async def stream_file(self, file: NvrFile) -> bytes:
        len_data = await file.generate_first_bytes(self.client)
        print("len data =",len_data)
        if (len_data is None):
            yield b""
        else:
            async for chunk in file.get_file_stream(self.client, len_data):
                if (chunk == None):
                    print("end of file")
                    break
                yield chunk

    async def save_file(self, file:NvrFile, savePath = "out.unknown"):
        downloaded_bytes = 0
        with open(savePath, "wb") as f:
            async for byte in file.generate_bytes(self.client):
                f.write(byte)
                downloaded_bytes += len(byte)
                print("\r", downloaded_bytes, "/", file.size)