#ifndef XML_CTRL_H
#define XML_CTRL_H
#define DEFAULT_XML_DECLARATION L"<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n"
#define DEFAULT_XML_DECLARATION_UTF8 "<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n"

#include <wx/wx.h>
#include <wx/stc/stc.h>
#include <string>
#include <set>
#include <map>

struct XmlCtrlProperties
{
    bool completion;
    bool fold;
    bool number;
    bool currentLine;
    bool whitespaceVisible;
    bool wrap;
    bool indentLines;
    bool protectHiddenElements;
    bool toggleLineBackground;
    bool insertCloseTag;
    bool deleteWholeTag;
    bool validateAsYouType;
    bool highlightSyntax;
    int zoom, colorScheme;
    wxString font;
};

enum VisibilityStates {
    SHOW_TAGS,
    HIDE_ATTRIBUTES,
    HIDE_TAGS
};

enum ColorSchemes {
    COLOR_SCHEME_DEFAULT,
    COLOR_SCHEME_DEFAULT_BACKGROUND,
    COLOR_SCHEME_REDUCED_GLARE,
    COLOR_SCHEME_REDUCED_GLARE_BACKGROUND,
    COLOR_SCHEME_NONE
};

enum TagTypes {
    TAG_TYPE_OPEN,
    TAG_TYPE_CLOSE,
    TAG_TYPE_EMPTY,
    TAG_TYPE_OTHER,
    TAG_TYPE_ERROR
};

enum BackgroundStates {
    BACKGROUND_STATE_NORMAL,
    BACKGROUND_STATE_LIGHT
};

enum XmlFileTypes {
    FILE_TYPE_XML,
    FILE_TYPE_DTD,
    FILE_TYPE_CSS,
    FILE_TYPE_PHP,
    FILE_TYPE_RNC,
    FILE_TYPE_BINARY
};

class XmlCtrl: public wxStyledTextCtrl
{
public:
    XmlCtrl (
        wxWindow *parent,
        XmlCtrlProperties propertiesParameter,
        bool *protectTagsParameter,
        int visibilityStateParameter = SHOW_TAGS,
        int typeParameter = FILE_TYPE_XML,
        wxWindowID id = wxID_ANY,
        //const std::string& buffer = DEFAULT_XML_DECLARATION_UTF8,
        const char *buffer = NULL,
        size_t bufferLen = 0,
        const std::string& catalogPath = "",
        const std::string& basePath = "",
        const std::string& auxPath = "",
        const wxPoint &position = wxDefaultPosition,
        const wxSize& size = wxDefaultSize,
        long style = 0 );
    ~XmlCtrl()
    {
        attributeMap.clear();
        elementMap.clear();
        entitySet.clear();
    }
    int getType();
    int getParentCloseAngleBracket ( int pos, int range = USHRT_MAX * 4 );
    void applyProperties (
        XmlCtrlProperties propertiesParameter,
        bool zoomOnly = false );
    void applyVisibilityState ( int state = SHOW_TAGS );
    void updatePromptMaps();
    void updatePromptMaps ( const char *buffer, size_t bufferLen );
    void adjustCursor();
    void adjustSelection();
    void foldAll();
    void unfoldAll();
    void toggleFold();
    bool insertChild ( const wxString& child );
    bool insertSibling ( const wxString& sibling, const wxString& parent );
    bool insertEntity ( const wxString& entity );
    bool getDtdFound();
    void setErrorIndicator ( int line, int column );
    void clearErrorIndicators ( int maxLine = 0 );
    wxString getParent();
    wxString getLastElementName ( int pos );
    std::set<wxString> getChildren ( const wxString& parent );
    std::set<std::string> getEntitySet();
    bool canInsertAt ( int pos );
    int getTagStartPos ( int pos );
    void toggleLineBackground();
    bool shallowValidate ( int maxLine = 0, bool segmentOnly = false );
    bool shallowValidate ( const char *buffer, size_t bufferLen,
                           int startLine = 0,
                           int maxLine = 0,
                           int columnOffset = 0,
                           bool segmentOnly = false );
    std::string myGetTextRaw(); // alternative to faulty stc implementation
    bool getValidationRequired();
    void setValidationRequired ( bool b );
private:
    int type;
    bool *protectTags;
    bool validationRequired, dtdFound;
    int visibilityState;
    int controlState;
    int currentMaxLine;
    int lineBackgroundState;
    wxColour baseBackground, alternateBackground;
    std::map<std::string, std::map<std::string, std::set<std::string> > >
        attributeMap;
    std::map<std::string, std::set<std::string> > requiredAttributeMap;
    std::map<std::string, std::set<std::string> > elementMap;
    std::set<std::string> entitySet;
    std::string catalogPath, basePath, auxPath;
    XmlCtrlProperties properties;
    wxString getLastAttributeName ( int pos );
    int getAttributeStartPos ( int pos );
    int getAttributeSectionEndPos ( int pos, int range = USHRT_MAX );
    int getTagType ( int pos );
    int getLexerStyleAt ( int pos );
    bool isCloseTag ( int pos );
    bool canMoveLeftAt ( int pos );
    bool canMoveRightAt ( int pos );
    wxString getOpenTag ( const wxString& element );
    void handleOpenAngleBracket ( wxKeyEvent& event );
    void handleCloseAngleBracket ( wxKeyEvent& event );
    void handleEquals ( wxKeyEvent& event );
    void handleSpace ( wxKeyEvent& event );
    void handleAmpersand ( wxKeyEvent& event );
    void handleForwardSlash ( wxKeyEvent& event );
    void handleBackspace ( wxKeyEvent& event );
    void handleDelete ( wxKeyEvent& event );
    void OnMarginClick ( wxStyledTextEvent& event );
    void OnChar ( wxKeyEvent& event );
    void OnIdle ( wxIdleEvent& event );
    void OnKeyPressed ( wxKeyEvent& event );
    void OnMouseLeftDown ( wxMouseEvent& event );
    void OnMouseLeftUp ( wxMouseEvent& event );
    void OnMouseRightUp ( wxMouseEvent& event );
    void OnMiddleDown ( wxMouseEvent& event );
    void insertNewLine();
    void adjustNoColumnWidth();
    void adjustPosRight();
    void adjustPosLeft();
    void setColorScheme ( int scheme );
    void expandFoldsToLevel ( int level, bool expand );
    void protectHeadLine();

    DECLARE_NO_COPY_CLASS ( XmlCtrl )
    DECLARE_EVENT_TABLE()
};

#endif