GitObjectManager.h

Go to the documentation of this file.

#ifndef GITOBJECTMANAGER_H
#define GITOBJECTMANAGER_H
 
#include "GitRunner.h"
#include <QIODevice>
 
namespace std {
class mutex;
}
 
class GitPackIdxV2;
struct GitPackIdxItem;
struct GitPackObject;
 
class GitObjectManager {
    friend class GitObjectCache;
private:
    struct Private;
    Private *m;
 
    static QByteArray applyDelta(QByteArray const *base, QByteArray const *delta);
    static bool loadPackedObject(std::shared_ptr<GitPackIdxV2> const &idx, QIODevice *packfile, GitPackIdxItem const *item, GitPackObject *out);
    bool extractObjectFromPackFile(std::shared_ptr<GitPackIdxV2> const &idx, GitPackIdxItem const *item, GitPackObject *out);
    bool extractObjectFromPackFile(GitRunner g, const GitHash &id, QByteArray *out, GitObject::Type *type, std::mutex *mutex);
    void loadIndexes(GitRunner g, std::mutex *mutex);
    std::string findObjectPath(GitRunner g, const GitHash &id);
    bool loadObject(GitRunner g, const GitHash &id, QByteArray *out, GitObject::Type *type);
    void init();
public:
    GitObjectManager(std::mutex *mutex);
    GitObjectManager(GitObjectManager const &other);
    GitObjectManager(GitObjectManager &&other) = delete;
    ~GitObjectManager();
    GitObjectManager &operator = (GitObjectManager const &other);
    void setup();
    bool catFile(GitRunner g, const GitHash &id, QByteArray *out, GitObject::Type *type);
    void clearIndexes();
    
    static std::vector<std::string> findObject(const std::string &id, const QString &repo_local_dir);
};
 
class GitObjectCache {
public:
    struct Item {
        GitHash id;
        QByteArray ba;
        GitObject::Type type;
    };
private:
    std::mutex *mutex_ = nullptr;
    GitObjectManager object_manager_;
    using ItemPtr = std::shared_ptr<Item>;
    std::vector<ItemPtr> items_;
    std::map<std::string, GitHash> rev_parse_map_;
    size_t size() const;
public:
    GitObjectCache(std::mutex *mutex = nullptr)
        : mutex_(mutex)
        , object_manager_(mutex)
    {}
    void clear();
    GitHash rev_parse(GitRunner g, const std::string &name);
    GitObject catFile(GitRunner g, const GitHash &id);
 
    GitHash const &item_id(int i) const
    {
        return items_[i]->id;
    }
};
 
class GitCommit {
public:
    std::string tree_id;
    std::vector<std::string> parents;
 
    static bool parseCommit(GitRunner g, GitObjectCache *objcache, GitHash const &id, GitCommit *out);
};
 
struct GitTreeItem {
    enum Type {
        UNKNOWN,
        TREE,
        BLOB,
        COMMIT,
    };
    Type type = UNKNOWN;
    std::string name;
    std::string id;
    std::string mode;
 
    std::string to_string_() const;
};
 
using GitTreeItemList = QList<GitTreeItem>;
 
class GitCommitTree {
private:
    GitObjectCache *object_cache_;
    GitTreeItemList root_item_list_;
 
    std::map<QString, GitTreeItem> blob_map;
    std::map<QString, QString> tree_id_map;
 
    QString lookup_(GitRunner g, QString const &file, GitTreeItem *out);
public:
    GitCommitTree(GitObjectCache *objcache);
 
    QString lookup(GitRunner g, QString const &file);
    bool lookup(GitRunner g, QString const &file, GitTreeItem *out);
 
    void parseTree(GitRunner g, QString const &tree_id);
    QString parseCommit(GitRunner g, const GitHash &commit_id);
 
    GitTreeItemList const *treelist() const
    {
        return &root_item_list_;
    }
};
 
GitTreeItemList parseGitTreeObject(QByteArray const &ba, std::string const &path_prefix);
bool parseGitTreeObject(GitRunner g, GitObjectCache *objcache, std::string const &commit_id, std::string const &path_prefix, GitTreeItemList *out);
 
#endif // GITOBJECTMANAGER_H