src/TGitCache/CachedDirectory.h

   1 // TortoiseGit - a Windows shell extension for easy version control
   2
   3 // External Cache Copyright (C) 2005 - 2006, 2008, 2014 - TortoiseSVN
   4 // Copyright (C) 2008-2012, 2014, 2016-2017 - TortoiseGit
   5
   6 // This program is free software; you can redistribute it and/or
   7 // modify it under the terms of the GNU General Public License
   8 // as published by the Free Software Foundation; either version 2
   9 // of the License, or (at your option) any later version.
  10
  11 // This program is distributed in the hope that it will be useful,
  12 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14 // GNU General Public License for more details.
  15
  16 // You should have received a copy of the GNU General Public License
  17 // along with this program; if not, write to the Free Software Foundation,
  18 // 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  19 //
  20 #pragma once
  21
  22 #include "StatusCacheEntry.h"
  23 #include "TGitPath.h"
  24
  25 /**
  26  * \ingroup TGitCache
  27  * Holds the status for a folder and all files and folders directly inside
  28  * that folder.
  29  */
  30 #define GIT_CACHE_VERSION 2
  31
  32 class CCachedDirectory
  33 {
  34 public:
  35         typedef std::map<CTGitPath, CCachedDirectory *> CachedDirMap;
  36         typedef CachedDirMap::iterator ItDir;
  37
  38 public:
  39
  40         CCachedDirectory();
  41         CCachedDirectory(const CTGitPath& directoryPath);
  42         ~CCachedDirectory(void);
  43 private:
  44         CStatusCacheEntry GetStatusForMember(const CTGitPath& path, bool bRecursive, bool bFetch = true);
  45         CStatusCacheEntry GetCacheStatusForMember(const CTGitPath& path);
  46
  47         // If path is not emtpy, means fetch special file status.
  48         int EnumFiles(const CTGitPath& path, CString sProjectRoot, const CString& sSubPath, bool isSelf);
  49         CStatusCacheEntry GetOwnStatus(bool bRecursive);
  50         bool IsOwnStatusValid() const;
  51 public:
  52         void Invalidate();
  53         void RefreshStatus(bool bRecursive);
  54 private:
  55         void RefreshMostImportant();
  56         BOOL SaveToDisk(FILE * pFile);
  57         BOOL LoadFromDisk(FILE * pFile);
  58 public:
  59         /// Get the current full status of this folder
  60         git_wc_status_kind GetCurrentFullStatus() const {return m_currentFullStatus;}
  61 private:
  62
  63         CStatusCacheEntry GetStatusFromCache(const CTGitPath &path, bool bRecursive);
  64         CStatusCacheEntry GetStatusFromGit(const CTGitPath &path, const CString& sProjectRoot, bool isSelf);
  65
  66 //      static git_error_t* GetStatusCallback(void *baton, const char *path, git_wc_status2_t *status);
  67         static BOOL GetStatusCallback(const CString & path, git_wc_status_kind status, bool isDir, void * /*pUserData*/, bool assumeValid, bool skipWorktree);
  68         void AddEntry(const CTGitPath& path, const git_wc_status2_t* pGitStatus, DWORD validuntil = 0);
  69         CString GetCacheKey(const CTGitPath& path);
  70         CString GetFullPathString(const CString& cacheKey);
  71         CStatusCacheEntry LookForItemInCache(const CTGitPath& path, bool &bFound);
  72         void UpdateChildDirectoryStatus(const CTGitPath& childDir, git_wc_status_kind childStatus);
  73
  74         // Calculate the complete, composite status from ourselves, our files, and our descendants
  75         git_wc_status_kind CalculateRecursiveStatus();
  76
  77         // Update our composite status and deal with things if it's changed
  78         void UpdateCurrentStatus();
  79
  80
  81 private:
  82         CComAutoCriticalSection m_critSec;
  83
  84         // The cache of files and directories within this directory
  85         typedef std::map<CString, CStatusCacheEntry> CacheEntryMap;
  86         CacheEntryMap m_entryCache;
  87         CacheEntryMap m_entryCache_tmp; // used for updating m_entryCache and removing "removed" entries
  88
  89         /// A vector if iterators to child directories - used to put-together recursive status
  90         typedef std::map<CString, git_wc_status_kind>  ChildDirStatus;
  91         ChildDirStatus m_childDirectories;
  92
  93         // The path of the directory with this object looks after
  94         CTGitPath       m_directoryPath;
  95
  96         // The status of THIS directory (not a composite of children or members)
  97         CStatusCacheEntry m_ownStatus;
  98
  99         // Our current fully recursive status
 100         git_wc_status_kind  m_currentFullStatus;
 101
 102         // The most important status from all our file entries
 103         git_wc_status_kind m_mostImportantFileStatus;
 104
 105         bool m_bRecursive;              // used in the status callback
 106         friend class CGitStatusCache;
 107 };
 108