GitModel.h

Go to the documentation of this file.

// This may look like C code, but it's really -*- C++ -*-
/*
 * Copyright (C) 2008 Emweb bvba, Kessel-Lo, Belgium.
 *
 * See the LICENSE file for terms of use.
 */
#ifndef GIT_MODEL_H_
#define GIT_MODEL_H_

#include <Wt/WAbstractItemModel.h>

#include "Git.h"

using namespace Wt;


class GitModel : public Wt::WAbstractItemModel
{
public:
  static const ItemDataRole ContentsRole;
  static const ItemDataRole FilePathRole;

  GitModel();

  void setRepositoryPath(const std::string& repositoryPath);

  void loadRevision(const std::string& revName);

  virtual WModelIndex parent(const WModelIndex& index) const;

  virtual int columnCount(const WModelIndex& parent = WModelIndex())
    const;

  virtual int rowCount(const WModelIndex& parent = WModelIndex()) const;

  virtual WModelIndex
  index(int row, int column, const WModelIndex& parent = WModelIndex())
    const;

  virtual cpp17::any
  data(const WModelIndex& index, ItemDataRole role = ItemDataRole::Display) const;
  
  virtual cpp17::any
  headerData(int section, Orientation orientation = Orientation::Horizontal,
             ItemDataRole role = ItemDataRole::Display) const;

  using WAbstractItemModel::data;

private:
  Git git_;

  struct ChildIndex {
    int parentId;
    int index;

    ChildIndex(int aParent, int anIndex)
      : parentId(aParent), index(anIndex) { }

    bool operator< (const ChildIndex& other) const {
      if (parentId < other.parentId)
        return true;
      else if (parentId > other.parentId)
        return false;
      else return index < other.index;
    }
  };

  class Tree {
  public:
    Tree(int parentId, int index, const Git::ObjectId& object,
         int rowCount)
      : index_(parentId, index),
        treeObject_(object),
        rowCount_(rowCount)
    { }

    int parentId() const { return index_.parentId; }

    int index() const { return index_.index; }

    const Git::ObjectId& treeObject() const { return treeObject_; }

    int rowCount() const { return rowCount_; }

  private:
    ChildIndex    index_;
    Git::ObjectId treeObject_;
    int           rowCount_;
  };

  typedef std::map<ChildIndex, int> ChildPointerMap;

  mutable std::vector<Tree> treeData_;

  mutable ChildPointerMap   childPointer_;

  int getTreeId(int parentId, int childIndex) const;

  Git::Object getObject(const WModelIndex& index) const;
};

#endif // GIT_MODEL_H_