/*****************************************************************************

* Copyright (c) 2011 Sven Brauch <svenbrauch@googlemail.com> *

* *

* This program is free software; you can redistribute it and/or *

* modify it under the terms of the GNU General Public License as *

* published by the Free Software Foundation; either version 2 of *

* the License, or (at your option) any later version. *

* *

* This program 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. *

* *

* You should have received a copy of the GNU General Public License *

* along with this program. If not, see <http://www.gnu.org/licenses/>. *

*****************************************************************************

*/

#include "implementfunction.h"

#include <language/codecompletion/codecompletionmodel.h>

#include <language/duchain/duchainutils.h>

#include <interfaces/icore.h>

#include <interfaces/isession.h>

#include <shell/partcontroller.h>

#include <KTextEditor/View>

#include <KTextEditor/Document>

#include <KTextEditor/Editor>

#include <QIcon>

using namespace KDevelop;

using namespace KTextEditor;

namespace Python {

ImplementFunctionCompletionItem::ImplementFunctionCompletionItem(const QString& name, const QStringList& arguments, const QString& previousIndent)

: m_arguments(arguments), m_name(name), m_previousIndent(previousIndent)

{

}

void ImplementFunctionCompletionItem::execute(KTextEditor::View* view, const KTextEditor::Range& word)

{

auto document = view->document();

const QString finalText = m_name + "(" + m_arguments.join(", ") + "):";

document->replaceText(word, finalText);

// 4 spaces is indentation for python. everyone does it like this. you must, too.

// TODO use kate settings

document->insertLine(word.start().line() + 1, m_previousIndent + " ");

if ( View* view = static_cast<KDevelop::PartController*>(ICore::self()->partController())->activeView() ) {

view->setCursorPosition(Cursor(word.end().line() + 1, m_previousIndent.length() + 4));

}

}

QVariant ImplementFunctionCompletionItem::data(const QModelIndex& index, int role, const KDevelop::CodeCompletionModel* model) const

{

switch ( role ) {

case KDevelop::CodeCompletionModel::MatchQuality: {

return QVariant(m_name.startsWith("__") ? 0 : 10);

}

case KDevelop::CodeCompletionModel::BestMatchesCount: {

return QVariant(5);

}

case Qt::DisplayRole:

switch ( index.column() ) {

case KDevelop::CodeCompletionModel::Name:

return m_name + "(" + m_arguments.join(", ") + ")";

case KDevelop::CodeCompletionModel::Postfix:

return "";

case KDevelop::CodeCompletionModel::Prefix:

return "Override method";

default:

return "";

}

case Qt::DecorationRole:

if( index.column() == KDevelop::CodeCompletionModel::Icon ) {

KDevelop::CodeCompletionModel::CompletionProperties p(KDevelop::CodeCompletionModel::Function);

return DUChainUtils::iconForProperties(p);

}

default: return CompletionTreeItem::data(index, role, model);

}

}

} // namespace Python