#include "modlistview.h"
#include <QUrl>
#include <QMimeData>
#include <QProxyStyle>


class ModListViewStyle: public QProxyStyle {
public:
  ModListViewStyle(QStyle *style, int indentation);

  void drawPrimitive (PrimitiveElement element, const QStyleOption *option,
                      QPainter *painter, const QWidget *widget = 0) const;
private:
  int m_Indentation;
};

ModListViewStyle::ModListViewStyle(QStyle *style, int indentation)
  : QProxyStyle(style), m_Indentation(indentation)
{
}

void ModListViewStyle::drawPrimitive(PrimitiveElement element, const QStyleOption *option,
                                     QPainter *painter, const QWidget *widget) const
{
  if (element == QStyle::PE_IndicatorItemViewItemDrop && !option->rect.isNull()) {
    QStyleOption opt(*option);
    opt.rect.setLeft(m_Indentation);
    if (widget) {
      opt.rect.setRight(widget->width() - 5); // 5 is an arbitrary value that seems to work ok
    }
    QProxyStyle::drawPrimitive(element, &opt, painter, widget);
  } else {
    QProxyStyle::drawPrimitive(element, option, painter, widget);
  }
}

ModListView::ModListView(QWidget *parent)
  : QTreeView(parent)
  , m_Scrollbar(new ViewMarkingScrollBar(this->model(), this))
{
//  setStyle(new ModListViewStyle(style(), indentation()));
  setVerticalScrollBar(m_Scrollbar);
}

void ModListView::dragEnterEvent(QDragEnterEvent *event)
{
  emit dropModeUpdate(event->mimeData()->hasUrls());

  QTreeView::dragEnterEvent(event);
}

void ModListView::setModel(QAbstractItemModel *model)
{
  QTreeView::setModel(model);
  setVerticalScrollBar(new ViewMarkingScrollBar(model, this));
}