JUCE/extras/Introjucer/Source/Code Editor/jucer_SourceCodeEditor.h

/*
  ==============================================================================

   This file is part of the JUCE library - "Jules' Utility Class Extensions"
   Copyright 2004-11 by Raw Material Software Ltd.

  ------------------------------------------------------------------------------

   JUCE can be redistributed and/or modified under the terms of the GNU General
   Public License (Version 2), as published by the Free Software Foundation.
   A copy of the license is included in the JUCE distribution, or can be found
   online at www.gnu.org/licenses.

   JUCE is distributed in the hope that it will be useful, but WITHOUT ANY
   WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
   A PARTICULAR PURPOSE.  See the GNU General Public License for more details.

  ------------------------------------------------------------------------------

   To release a closed-source product which uses JUCE, commercial licenses are
   available: visit www.rawmaterialsoftware.com/juce for more information.

  ==============================================================================
*/

#ifndef __JUCER_SOURCECODEEDITOR_JUCEHEADER__
#define __JUCER_SOURCECODEEDITOR_JUCEHEADER__

#include "../Project/jucer_Project.h"
#include "../Application/jucer_DocumentEditorComponent.h"

//==============================================================================
class SourceCodeEditor  : public DocumentEditorComponent,
                          private ValueTree::Listener
{
public:
    SourceCodeEditor (OpenDocumentManager::Document* document, CodeDocument& codeDocument);
    SourceCodeEditor (OpenDocumentManager::Document* document);
    ~SourceCodeEditor();

    void createEditor (CodeDocument& codeDocument);
    void setEditor (CodeEditorComponent*);

private:
    ScopedPointer<CodeEditorComponent> editor;

    void resized();

    void valueTreePropertyChanged (ValueTree&, const Identifier&);
    void valueTreeChildAdded (ValueTree&, ValueTree&);
    void valueTreeChildRemoved (ValueTree&, ValueTree&);
    void valueTreeChildOrderChanged (ValueTree&);
    void valueTreeParentChanged (ValueTree&);
    void valueTreeRedirected (ValueTree&);

    void updateColourScheme();

    JUCE_DECLARE_NON_COPYABLE_WITH_LEAK_DETECTOR (SourceCodeEditor);
};


//==============================================================================
class CppCodeEditorComponent  : public CodeEditorComponent
{
public:
    CppCodeEditorComponent (CodeDocument& codeDocument)
        : CodeEditorComponent (codeDocument, getCppTokeniser())
    {
    }

    void handleReturnKey()
    {
        CodeEditorComponent::handleReturnKey();

        const CodeDocument::Position pos (getCaretPos());

        if (pos.getLineNumber() > 0 && pos.getLineText().trim().isEmpty())
        {
            String indent (getIndentForCurrentBlock (pos));
            const String previousLine (pos.movedByLines (-1).getLineText());
            const String trimmedPreviousLine (previousLine.trim());
            const String leadingWhitespace (getLeadingWhitespace (previousLine));

            insertTextAtCaret (leadingWhitespace);

            if (trimmedPreviousLine.endsWithChar ('{')
                 || ((trimmedPreviousLine.startsWith ("if ")
                      || trimmedPreviousLine.startsWith ("for ")
                      || trimmedPreviousLine.startsWith ("while "))
                      && trimmedPreviousLine.endsWithChar (')')))
                insertTabAtCaret();
        }
    }

    void insertTextAtCaret (const String& newText)
    {
        if (getHighlightedRegion().isEmpty())
        {
            const CodeDocument::Position pos (getCaretPos());

            if ((newText == "{" || newText == "}") && pos.getLineNumber() > 0)
            {
                moveCaretToStartOfLine (true);
                CodeEditorComponent::insertTextAtCaret (getIndentForCurrentBlock (pos));

                if (newText == "{")
                    insertTabAtCaret();
            }
        }

        CodeEditorComponent::insertTextAtCaret (newText);
    }

private:
    static CPlusPlusCodeTokeniser* getCppTokeniser()
    {
        static CPlusPlusCodeTokeniser cppTokeniser;
        return &cppTokeniser;
    }

    static String getLeadingWhitespace (String line)
    {
        line = line.removeCharacters ("\r\n");
        const String::CharPointerType endOfLeadingWS (line.getCharPointer().findEndOfWhitespace());
        return String (line.getCharPointer(), endOfLeadingWS);
    }

    static String getIndentForCurrentBlock (CodeDocument::Position pos)
    {
        int braceCount = 0;

        while (pos.getLineNumber() > 0)
        {
            pos = pos.movedByLines (-1);

            const String line (pos.getLineText());
            const String trimmedLine (line.trimStart());

            StringArray tokens;
            tokens.addTokens (trimmedLine, true);

            for (int i = tokens.size(); --i >= 0;)
            {
                if (tokens[i] == "}")
                    ++braceCount;

                if (tokens[i] == "{")
                    if (--braceCount < 0)
                        return getLeadingWhitespace (line);
            }
        }

        return String::empty;
    }
};


#endif   // __JUCER_SOURCECODEEDITOR_JUCEHEADER__