import datetime
from typing import Any
import json
from pathlib import Path

import frontmatter
from decimal import Decimal
from logging import getLogger
logger = getLogger(__name__)

StatisticKey = str

class ObsidianVault:

    def __init__(self, vault_path : Path, read_only: bool = 'silent'):
        self.vault_path  = vault_path

        assert (self.vault_path / '.obsidian').exists(), 'Not an Obsidian Vault'

        with open(self.vault_path  / '.obsidian'  / 'daily-notes.json') as f:
            daily_notes_config = json.load(f)
        self.daily_folder = daily_notes_config['folder']
        self.path_format = daily_notes_config['format']
        self.template_file_path = daily_notes_config['template']
        self.read_only = read_only

    def get_statistic(self, date: datetime.date, statistic_key: StatisticKey) -> Any | None:
        try:
            with open(self._date_file_path(date)) as f:
                data = frontmatter.load(f)
        except FileNotFoundError:
            return None

        return data.metadata.get(statistic_key)

    def add_statistic(self, date: datetime.date, statistic_key: StatisticKey, amount: Any) -> bool:
        if self.read_only == 'silent':
            logger.info('Real only ObsidianVault ignoring add_statistic(%s, "%s", ?)', date, statistic_key)
            return False

        self._create_date_if_not_present(date)

        with open(self._date_file_path(date)) as f:
            data = frontmatter.load(f)

        if isinstance(amount, Decimal):
            amount = float(amount)

        if data.metadata.get(statistic_key) == amount:
            return False

        data.metadata[statistic_key] = amount

        with open(self._date_file_path(date), 'wb') as f:
            frontmatter.dump(data, f)

        return True

    def add_event(self, date: datetime.date, verb: str, subject: str) -> None:
        if self.read_only == 'silent':
            logger.info('Real only ObsidianVault ignoring add_event(%s, "%s", ?)', date, verb)
            return

        self._create_date_if_not_present(date)
        # TODO

    def _create_date_if_not_present(self, date: datetime.date):
        date_file = self._date_file_path(date)
        if date_file.exists():
            return
        logger.info('File "%s" doesn\'t exist, creating...', date)
        with open(self._daily_template_path()) as f:
            template_text = f.read()
        with open(date_file, 'w') as f:
            f.write(template_text)

    def _date_file_path(self, date: datetime.date):
        path = self.path_format.replace('YYYY', str(date.year)).replace('MM', '{:02d}'.format(date.month)).replace('DD', '{:02d}'.format(date.day))
        return (self.vault_path / self.daily_folder / path).with_suffix('.md')

    def _daily_template_path(self):
        return (self.vault_path / self.template_file_path).with_suffix('.md')