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         CStatusCacheEntry GetStatusForMember(const CTGitPath& path, bool bRecursive, bool bFetch = true);
  44
  45 private:
  46         CStatusCacheEntry GetCacheStatusForMember(const CTGitPath& path);
  47
  48         // If path is not emtpy, means fetch special file status.
  49         int EnumFiles(const CTGitPath& path, CString sProjectRoot, const CString& sSubPath, bool isSelf);
  50         CStatusCacheEntry GetOwnStatus(bool bRecursive);
  51         bool IsOwnStatusValid() const;
  52 public:
  53         void Invalidate();
  54         void RefreshStatus(bool bRecursive);
  55 private:
  56         void RefreshMostImportant(bool bUpdateShell = true);
  57         BOOL SaveToDisk(FILE * pFile);
  58         BOOL LoadFromDisk(FILE * pFile);
  59 public:
  60         /// Get the current full status of this folder
  61         git_wc_status_kind GetCurrentFullStatus() const {return m_currentFullStatus;}
  62 private:
  63
  64         CStatusCacheEntry GetStatusFromCache(const CTGitPath &path, bool bRecursive);
  65         CStatusCacheEntry GetStatusFromGit(const CTGitPath &path, const CString& sProjectRoot, bool isSelf);
  66
  67         static BOOL GetStatusCallback(const CString& path, git_wc_status_kind status, bool isDir, __int64 lastwritetime, void* /*pUserData*/, bool assumeValid, bool skipWorktree);
  68         void AddEntry(const CTGitPath& path, const git_wc_status2_t* pGitStatus, __int64 lastwritetime);
  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         void SetChildStatus(const CString& childDir, git_wc_status_kind childStatus);
  80         void KeepChildStatus(const CString& childDir);
  81
  82 private:
  83         CComAutoCriticalSection m_critSec;
  84
  85         // The cache of files and directories within this directory
  86         typedef std::map<CString, CStatusCacheEntry> CacheEntryMap;
  87         CacheEntryMap m_entryCache;
  88         CacheEntryMap m_entryCache_tmp; // used for updating m_entryCache and removing "removed" entries
  89
  90         /// A vector if iterators to child directories - used to put-together recursive status
  91         typedef std::map<CString, git_wc_status_kind>  ChildDirStatus;
  92         ChildDirStatus m_childDirectories;
  93         ChildDirStatus m_childDirectories_tmp; // used for updating m_childDirectories and removing "removed" entries
  94
  95         // The path of the directory with this object looks after
  96         CTGitPath       m_directoryPath;
  97
  98         // The status of THIS directory (not a composite of children or members)
  99         CStatusCacheEntry m_ownStatus;
 100
 101         // Our current fully recursive status
 102         git_wc_status_kind  m_currentFullStatus;
 103
 104         // The most important status from all our file entries
 105         git_wc_status_kind m_mostImportantFileStatus;
 106
 107         bool m_bRecursive;              // used in the status callback
 108         friend class CGitStatusCache;
 109 };
 110