import os
import re
from datetime import datetime
from glob import glob

import lxml.html
from markdown import markdown

from sync.image import sync_image
from utils.fn import lmap, combine


def list_images(dir):
    image_types = ['*.jpg', '*.JPG', '*.png', '*.PNG']
    return combine(map(
        lambda type: glob(os.path.join(dir, type)),
        image_types
    ))


def create_date(date_str, date_formats=None):
    """
    2016
    2016.10
    2016.10.10
    -> datetime

    :param date_str:
    :return:
    """
    if not date_formats:
        date_formats = [
            '%Y',
            '%Y.%m',
            '%Y.%m.%d'
        ]

    for format in date_formats:
        try:
            return datetime.strptime(date_str, format)
        except:
            continue
    return None


# 2016 На крыльях бабочек
# 2016.10 На крыльях бабочек
# 2016.10.10 На крыльях бабочек
folder_name_pattern = re.compile('([\d.]+)(.*)')


def untouched(documents, store):
    return lmap(
        lambda i: i[0],
        filter(
            lambda i: (i[1] is None) or ('hash' not in i[1]) or (i[0]['hash'] != i[1]['hash']),
            map(
                lambda i: (i, store.read(i)),
                documents
            )
        ))


def synced_images_ids(images):
    ids = lmap(
        lambda image: image['_id'],
        sync_image(images)
    )
    return ids


def synced_image_id(image):
    return sync_image(image)['_id']


def create_date_and_title_from_folder_name(folder_name, date_formats=None):
    m = folder_name_pattern.findall(folder_name)
    if not m:
        return None, None

    date_str, title = m[0]
    title = os.path.splitext(title.strip())[0]
    date = create_date(date_str, date_formats)

    if not date:
        return None, title

    return date, title


def images_from_html(md):
    if not md:
        return []

    post_html = lxml.html.fromstring(md)

    images = []
    for img in post_html.cssselect('img'):
        src = img.get('src')
        images.append(src)
    return images


create_post_from_image_list = lambda images: markdown(
    '\n'.join(map(
        lambda i: '![]({img})'.format(img=i),
        images
    ))
)