import pathlib
from typing import List
from urllib.request import Request

from sanic import response

from wrolpi.common import get_media_directory
from wrolpi.errors import InvalidFile
from wrolpi.root_api import get_blueprint, json_response
from wrolpi.schema import validate_doc
from . import lib, schema

bp = get_blueprint('Files', '/api/files')


def paths_to_files(paths: List[pathlib.Path]):
    """Convert Paths to what the React UI expects."""
    media_directory = get_media_directory()
    new_files = []
    for path in paths:
        stat = path.stat()
        key = path.relative_to(media_directory)
        modified = stat.st_mtime
        if path.is_dir():
            key = f'{key}/'
            new_files.append(dict(
                key=key,
                modified=modified,
                url=key,
                name=path.name,
            ))
        else:
            # A File should know it's size.
            new_files.append(dict(
                key=key,
                modified=modified,
                size=stat.st_size,
                url=key,
                name=path.name,
            ))
    return new_files


@bp.post('/')
@validate_doc(
    'List files in a directory',
    consumes=schema.FilesRequest,
)
async def get_files(_: Request, data: dict):
    directories = data.get('directories') or []

    files = lib.list_files(directories)
    files = paths_to_files(files)
    return json_response({'files': files})


@bp.post('/delete')
@validate_doc(
    'Delete a single file.  Returns an error if WROL Mode is enabled.',
    consumes=schema.DeleteRequest,
)
async def delete_file(_: Request, data: dict):
    file = data.get('file')
    if not file:
        raise InvalidFile('file cannot be empty')
    lib.delete_file(file)
    return response.empty()