import os
import logging

import streamlit as st
from ..utils.form_documentation import generate_doc

from typing import Optional
from ..command_handler import CommandHandler


class DocumentationHandler(CommandHandler):
    def __init__(self, commands, successor: Optional["CommandHandler"] = None):
        super().__init__(successor)
        self.commands = commands

    def handle_command(self, command):
        if command.lower() in self.commands:
            self.execute_command()
        else:
            super().handle_command(command)

    def execute_command(self):
        current_dir = os.path.dirname(os.path.realpath(__file__))
        path_to_file = generate_doc(path=current_dir)

        if path_to_file is not None:
            with st.sidebar:
                st.download_button('Download doc', path_to_file)
                st.success("Done", icon="✅")
        else:
            logging.error("path_to_file is None")