# (C) Copyright 2004-2022 Enthought, Inc., Austin, TX
# All rights reserved.
#
# This software is provided without warranty under the terms of the BSD
# license included in LICENSE.txt and may be redistributed only under
# the conditions described in the aforementioned license. The license
# is also available online at http://www.enthought.com/licenses/BSD.txt
#
# Thanks for using Enthought open source!

""" Defines various directory editors for the wxPython user interface toolkit.
"""

from os.path import isdir

import wx

from pyface.api import DirectoryDialog

from .file_editor import (
    SimpleEditor as SimpleFileEditor,
    CustomEditor as CustomFileEditor,
    PopupFile,
)

# -------------------------------------------------------------------------
#  'SimpleEditor' class:
# -------------------------------------------------------------------------


class SimpleEditor(SimpleFileEditor):
    """Simple style of editor for directories, which displays a text field
    and a **Browse** button that opens a directory-selection dialog box.
    """

    def _create_file_dialog(self):
        """Creates the correct type of file dialog."""
        dlg = DirectoryDialog(
            parent=self.get_control_widget(),
            default_path=self._file_name.text(),
        )
        return dlg

    def _create_file_popup(self):
        """Creates the correct type of file popup."""
        return PopupDirectory(
            control=self.control,
            file_name=self.str_value,
            filter=self.factory.filter,
            height=300,
        )


class CustomEditor(CustomFileEditor):
    """Custom style of editor for directories, which displays a tree view of
    the file system.
    """

    def get_style(self):
        """Returns the basic style to use for the control."""
        return wx.DIRCTRL_DIR_ONLY | wx.DIRCTRL_EDIT_LABELS

    def update_object(self, event):
        """Handles the user changing the contents of the edit control."""
        if self.control is not None:
            path = self.control.GetPath()
            if isdir(path):
                self.value = path


class PopupDirectory(PopupFile):
    def get_style(self):
        """Returns the basic style to use for the popup."""
        return wx.DIRCTRL_DIR_ONLY | wx.DIRCTRL_EDIT_LABELS

    def is_valid(self, path):
        """Returns whether or not the path is valid."""
        return isdir(path)
