127 lines
3.4 KiB
C
127 lines
3.4 KiB
C
|
// Copyright 2019 Roman Perepelitsa.
|
||
|
|
//
|
||
|
|
// This file is part of GitStatus.
|
||
|
|
//
|
||
|
|
// GitStatus 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 3 of the License, or
|
||
|
|
// (at your option) any later version.
|
||
|
|
//
|
||
|
|
// GitStatus 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 GitStatus. If not, see <https://www.gnu.org/licenses/>.
|
||
|
|
|
||
|
|
#ifndef ROMKATV_GITSTATUS_REPO_H_
|
||
|
|
#define ROMKATV_GITSTATUS_REPO_H_
|
||
|
|
|
||
|
|
#include <stddef.h>
|
||
|
|
#include <sys/stat.h>
|
||
|
|
#include <sys/types.h>
|
||
|
|
#include <unistd.h>
|
||
|
|
|
||
|
|
#include <git2.h>
|
||
|
|
|
||
|
|
#include <algorithm>
|
||
|
|
#include <atomic>
|
||
|
|
#include <condition_variable>
|
||
|
|
#include <cstddef>
|
||
|
|
#include <cstring>
|
||
|
|
#include <functional>
|
||
|
|
#include <future>
|
||
|
|
#include <memory>
|
||
|
|
#include <mutex>
|
||
|
|
#include <string>
|
||
|
|
#include <utility>
|
||
|
|
#include <vector>
|
||
|
|
|
||
|
|
#include "check.h"
|
||
|
|
#include "index.h"
|
||
|
|
#include "options.h"
|
||
|
|
#include "string_cmp.h"
|
||
|
|
#include "tag_db.h"
|
||
|
|
#include "time.h"
|
||
|
|
|
||
|
|
namespace gitstatus {
|
||
|
|
|
||
|
|
struct IndexStats {
|
||
|
|
size_t index_size = 0;
|
||
|
|
size_t num_staged = 0;
|
||
|
|
size_t num_unstaged = 0;
|
||
|
|
size_t num_conflicted = 0;
|
||
|
|
size_t num_untracked = 0;
|
||
|
|
size_t num_staged_new = 0;
|
||
|
|
size_t num_staged_deleted = 0;
|
||
|
|
size_t num_unstaged_deleted = 0;
|
||
|
|
size_t num_skip_worktree = 0;
|
||
|
|
size_t num_assume_unchanged = 0;
|
||
|
|
};
|
||
|
|
|
||
|
|
class Repo {
|
||
|
|
public:
|
||
|
|
explicit Repo(git_repository* repo, Limits lim);
|
||
|
|
Repo(Repo&& other) = delete;
|
||
|
|
~Repo();
|
||
|
|
|
||
|
|
git_repository* repo() const { return repo_; }
|
||
|
|
|
||
|
|
// Head can be null, in which case has_staged will be false.
|
||
|
|
IndexStats GetIndexStats(const git_oid* head, git_config* cfg);
|
||
|
|
|
||
|
|
// Returns the last tag in lexicographical order whose target is equal to the given, or an
|
||
|
|
// empty string. Target can be null, in which case the tag is empty.
|
||
|
|
std::future<std::string> GetTagName(const git_oid* target);
|
||
|
|
|
||
|
|
private:
|
||
|
|
struct Shard {
|
||
|
|
bool Contains(Str<> str, StringView path) const;
|
||
|
|
std::string start_s;
|
||
|
|
std::string end_s;
|
||
|
|
size_t start_i;
|
||
|
|
size_t end_i;
|
||
|
|
};
|
||
|
|
|
||
|
|
void UpdateShards();
|
||
|
|
|
||
|
|
int OnDelta(const char* type, const git_diff_delta& d, std::atomic<size_t>& c1, size_t m1,
|
||
|
|
const std::atomic<size_t>& c2, size_t m2);
|
||
|
|
|
||
|
|
void StartStagedScan(const git_oid* head);
|
||
|
|
void StartDirtyScan(const std::vector<const char*>& paths);
|
||
|
|
|
||
|
|
void DecInflight();
|
||
|
|
void RunAsync(std::function<void()> f);
|
||
|
|
void Wait();
|
||
|
|
|
||
|
|
Limits lim_;
|
||
|
|
git_repository* const repo_;
|
||
|
|
git_index* git_index_ = nullptr;
|
||
|
|
std::vector<Shard> shards_;
|
||
|
|
git_oid head_ = {};
|
||
|
|
TagDb tag_db_;
|
||
|
|
|
||
|
|
std::unique_ptr<Index> index_;
|
||
|
|
|
||
|
|
std::mutex mutex_;
|
||
|
|
std::condition_variable cv_;
|
||
|
|
std::atomic<size_t> inflight_{0};
|
||
|
|
std::atomic<bool> error_{false};
|
||
|
|
std::atomic<size_t> staged_{0};
|
||
|
|
std::atomic<size_t> unstaged_{0};
|
||
|
|
std::atomic<size_t> conflicted_{0};
|
||
|
|
std::atomic<size_t> untracked_{0};
|
||
|
|
std::atomic<size_t> staged_new_{0};
|
||
|
|
std::atomic<size_t> staged_deleted_{0};
|
||
|
|
std::atomic<size_t> unstaged_deleted_{0};
|
||
|
|
std::atomic<size_t> skip_worktree_{0};
|
||
|
|
std::atomic<size_t> assume_unchanged_{0};
|
||
|
|
std::atomic<Tribool> untracked_cache_{Tribool::kUnknown};
|
||
|
|
};
|
||
|
|
|
||
|
|
} // namespace gitstatus
|
||
|
|
|
||
|
|
#endif // ROMKATV_GITSTATUS_REPO_H_
|