import datetime
import os
from telegram.ext import ConversationHandler, CommandHandler, MessageHandler, filters, CallbackQueryHandler
from telegram import InlineKeyboardButton, InlineKeyboardMarkup
DATE_CHOICE, DATE_ENTRY, CONTENT = range(3)

from .basehandler import BaseHandler

class JournalHandler(BaseHandler):
    def __init__(self, entry_string, models):
        self.models = models
        self.entry_string = entry_string
        self.handler = ConversationHandler(
            entry_points=[CommandHandler(entry_string, self.entry_point)],
            states={
                DATE_CHOICE: [
                    CallbackQueryHandler(self.date_choice, pattern="today|yesterday"),
                    CallbackQueryHandler(self.date_custom, pattern="custom"),
                    ],
                DATE_ENTRY: [
                    MessageHandler(filters.ALL, self.date_entry),
                    ],
                CONTENT: [
                    MessageHandler(filters.ALL, self.content_save),
                    ],
            },
            fallbacks=[],
        )

        self.current_model = None


    async def entry_point(self, update, context):
        await super().entry_point(update, context)
        if os.getenv("DOCKERIZED", "false") == "true" and os.getenv("CHAT_ID") != str(update.message.chat_id):
            await update.message.reply_text("You are not authorized to use this bot")
            return ConversationHandler.END


        options = [
            InlineKeyboardButton("Today", callback_data="today"),
            InlineKeyboardButton("Yesterday", callback_data="yesterday"),
            InlineKeyboardButton("Custom date", callback_data="custom"),
        ]
        keyboard = InlineKeyboardMarkup([options])
        await update.message.reply_text("Please choose an option for the entry:", reply_markup=keyboard)
        return DATE_CHOICE


    async def date_choice(self, update, context):
        query = update.callback_query
        await query.answer()
        if query.data == "today":
            date = datetime.datetime.now().date()
        elif query.data == "yesterday":
            date = datetime.datetime.now().date() - datetime.timedelta(days=1)
        else:
            raise ValueError("Invalid date choice")

        with self.models.db:
            self.current_model, new = self.models.JournalEntry.get_or_create(
                date = date
            )
        if new:
            await query.edit_message_text(
                text="Please enter the content for the entry"
            )
        else:
            await query.edit_message_text(text="An entry already exists for this date")
            return ConversationHandler.END
        
        return CONTENT


    async def date_custom(self, update, context):
        query = update.callback_query
        await query.answer()
        await query.edit_message_text(text="Please enter the date in the format DDMMYYYY")
        return DATE_ENTRY


    async def date_entry(self, update, context):
        date = update.message.text
        try:
            date = datetime.datetime.strptime(date, "%d%m%Y").date()
        except ValueError:
            await update.message.reply_text("Please enter the date in the format DDMMYYYY")
            return DATE_ENTRY
        
        with self.models.db:
            self.current_model, new = self.models.JournalEntry.get_or_create(
                date = date
            )
        if not new:
            await update.message.reply_text("An entry already exists for this date")
            return ConversationHandler.END
        else:
            await update.message.reply_text("Please enter the content for the entry")
            return CONTENT


    async def content_save(self, update, context):
        with self.models.db:
            self.current_model.author_id = update.message.from_user.id

            if update.message.text:
                self.current_model.text = update.message.text
            else:
                if update.message.photo:
                    file = await update.message.effective_attachment[-1].get_file()
                else:
                    file = await update.message.effective_attachment.get_file()
                
                file_bytes = await file.download_as_bytearray()
                file_path = file.file_path
                self.current_model.save_media(file_bytes, file_path)

                self.current_model.text = update.message.caption
            
            self.current_model.save()

        await update.message.reply_text(f"Saved entry for {self.current_model.date.strftime('%A, %-d. %B %Y')}")
        return ConversationHandler.END