OSDN Git Service

fd680b0b3b02e7804e39b680fff742b534710f95
[tortoisegit/TortoiseGitJp.git] / src / Git / GitFolderStatus.cpp
1 // TortoiseSVN - a Windows shell extension for easy version control\r
2 \r
3 // Copyright (C) 2003-2008 - TortoiseSVN\r
4 \r
5 // This program is free software; you can redistribute it and/or\r
6 // modify it under the terms of the GNU General Public License\r
7 // as published by the Free Software Foundation; either version 2\r
8 // of the License, or (at your option) any later version.\r
9 \r
10 // This program is distributed in the hope that it will be useful,\r
11 // but WITHOUT ANY WARRANTY; without even the implied warranty of\r
12 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the\r
13 // GNU General Public License for more details.\r
14 \r
15 // You should have received a copy of the GNU General Public License\r
16 // along with this program; if not, write to the Free Software Foundation,\r
17 // 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.\r
18 \r
19 #include "stdafx.h"\r
20 #include "ShellExt.h"\r
21 #include "GitFolderStatus.h"\r
22 #include "UnicodeUtils.h"\r
23 #include "..\TGitCache\CacheInterface.h"\r
24 #include "Git.h"\r
25 //#include "GitGlobal.h"\r
26 #include "gitindex.h"\r
27 \r
28 extern ShellCache g_ShellCache;\r
29 \r
30 extern CGitIndexFileMap g_IndexFileMap;\r
31 \r
32 // get / auto-alloc a string "copy"\r
33 \r
34 const char* StringPool::GetString (const char* value)\r
35 {\r
36         // special case: NULL pointer\r
37         \r
38         if (value == NULL)\r
39         {\r
40                 return emptyString;\r
41         }\r
42 \r
43         // do we already have a string with the desired value?\r
44 \r
45         pool_type::const_iterator iter = pool.find (value);\r
46         if (iter != pool.end())\r
47         {\r
48                 // yes -> return it\r
49                 return *iter;\r
50         }\r
51         \r
52         // no -> add one        \r
53         \r
54         const char* newString =  _strdup (value);\r
55         if (newString)\r
56         {\r
57                 pool.insert (newString);\r
58         }\r
59         else\r
60                 return emptyString;\r
61         \r
62         // .. and return it\r
63         \r
64         return newString;\r
65 }\r
66 \r
67 // clear internal pool\r
68 \r
69 void StringPool::clear()\r
70 {\r
71         // delete all strings\r
72 \r
73         for (pool_type::iterator iter = pool.begin(), end = pool.end(); iter != end; ++iter)\r
74         {\r
75                 free((void*)*iter);\r
76         }\r
77                 \r
78         // remove pointers from pool\r
79                 \r
80         pool.clear();\r
81 }\r
82 \r
83 CTGitPath       GitFolderStatus::folderpath;\r
84 \r
85 \r
86 GitFolderStatus::GitFolderStatus(void)\r
87 {\r
88         m_TimeStamp = 0;\r
89         emptyString[0] = 0;\r
90         invalidstatus.author = emptyString;\r
91         invalidstatus.askedcounter = -1;\r
92         invalidstatus.status = git_wc_status_none;\r
93         invalidstatus.url = emptyString;\r
94 //      invalidstatus.rev = -1;\r
95         invalidstatus.owner = emptyString;\r
96         invalidstatus.needslock = false;\r
97         invalidstatus.tree_conflict = false;\r
98         m_nCounter = 0;\r
99         dirstatus = NULL;\r
100         sCacheKey.reserve(MAX_PATH);\r
101 \r
102         //rootpool = svn_pool_create (NULL);\r
103 \r
104         m_hInvalidationEvent = CreateEvent(NULL, FALSE, FALSE, _T("TortoiseGitCacheInvalidationEvent"));\r
105 }\r
106 \r
107 GitFolderStatus::~GitFolderStatus(void)\r
108 {\r
109         //svn_pool_destroy(rootpool);\r
110         CloseHandle(m_hInvalidationEvent);\r
111 }\r
112 \r
113 const FileStatusCacheEntry * GitFolderStatus::BuildCache(const CTGitPath& filepath, const CString& sProjectRoot, BOOL bIsFolder, BOOL bDirectFolder)\r
114 {\r
115 //      svn_client_ctx_t *                      localctx;\r
116 //      apr_hash_t *                            statushash;\r
117 //      apr_pool_t *                            pool;\r
118         //git_error_t *                         err = NULL; // If svn_client_status comes out through catch(...), err would else be unassigned\r
119         git_error_t err = 0;\r
120 \r
121         //dont' build the cache if an instance of TortoiseProc is running\r
122         //since this could interfere with svn commands running (concurrent\r
123         //access of the .git directory).\r
124         if (g_ShellCache.BlockStatus())\r
125         {\r
126                 HANDLE TGitMutex = ::CreateMutex(NULL, FALSE, _T("TortoiseGitProc.exe"));       \r
127                 if (TGitMutex != NULL)\r
128                 {\r
129                         if (::GetLastError() == ERROR_ALREADY_EXISTS)\r
130                         {\r
131                                 ::CloseHandle(TGitMutex);\r
132                                 return &invalidstatus;\r
133                         }\r
134                 }\r
135                 ::CloseHandle(TGitMutex);\r
136         }\r
137 \r
138 //      pool = svn_pool_create (rootpool);                              // create the memory pool\r
139 \r
140         ClearCache();\r
141 //      svn_error_clear(svn_client_create_context(&localctx, pool));\r
142         // set up the configuration\r
143         // Note: I know this is an 'expensive' call, but without this, ignores\r
144         // done in the global ignore pattern won't show up.\r
145         if (g_ShellCache.ShowIgnoredOverlay())\r
146 ;//             svn_error_clear(svn_config_get_config (&(localctx->config), g_pConfigDir, pool));\r
147 \r
148         // strings pools are unused, now -> we may clear them\r
149         \r
150         authors.clear();\r
151         urls.clear();\r
152         owners.clear();\r
153         \r
154         ATLTRACE2(_T("building cache for %s\n"), filepath);\r
155         if (bIsFolder)\r
156         {\r
157                 if (bDirectFolder)\r
158                 {\r
159                         // NOTE: see not in GetFullStatus about project inside another project, we should only get here when\r
160                         //       that occurs, and this is not correctly handled yet\r
161 \r
162                         // initialize record members\r
163 //                      dirstat.rev = -1;\r
164                         dirstat.status = git_wc_status_none;\r
165                         dirstat.author = authors.GetString(NULL);\r
166                         dirstat.url = urls.GetString(NULL);\r
167                         dirstat.owner = owners.GetString(NULL);\r
168                         dirstat.askedcounter = GITFOLDERSTATUS_CACHETIMES;\r
169                         dirstat.needslock = false;\r
170                         dirstat.tree_conflict = false;\r
171 \r
172                         dirstatus = NULL;\r
173 //                      statushash = apr_hash_make(pool);\r
174 //                      git_revnum_t youngest = GIT_INVALID_REVNUM;\r
175 //                      git_opt_revision_t rev;\r
176 //                      rev.kind = git_opt_revision_unspecified;\r
177                         try\r
178                         {\r
179                                 folderpath = filepath;\r
180 \r
181                                 /*err = svn_client_status4 (&youngest,\r
182                                         filepath.GetDirectory().GetSVNApiPath(pool),\r
183                                         &rev,\r
184                                         findfolderstatus,\r
185                                         this,\r
186                                         svn_depth_empty,//depth\r
187                                         TRUE,           //getall\r
188                                         FALSE,          //update\r
189                                         TRUE,           //noignore\r
190                                         FALSE,          //ignore externals\r
191                                         NULL,\r
192                                         localctx,\r
193                                         pool);*/\r
194                         }\r
195                         catch ( ... )\r
196                         {\r
197                                 dirstatus = NULL;\r
198                         }\r
199 \r
200 \r
201                         if (dirstatus)\r
202                         {\r
203 /*                              if (dirstatus->entry)\r
204                                 {\r
205                                         dirstat.author = authors.GetString (dirstatus->entry->cmt_author);\r
206                                         dirstat.url = authors.GetString (dirstatus->entry->url);\r
207                                         dirstat.rev = dirstatus->entry->cmt_rev;\r
208                                         dirstat.owner = owners.GetString(dirstatus->entry->lock_owner);\r
209                                 }*/\r
210                                 dirstat.status = GitStatus::GetMoreImportant(dirstatus->text_status, dirstatus->prop_status);\r
211 //                              dirstat.tree_conflict = dirstatus->tree_conflict != NULL;\r
212                         }\r
213                         m_cache[filepath.GetWinPath()] = dirstat;\r
214                         m_TimeStamp = GetTickCount();\r
215 //                      svn_error_clear(err);\r
216 //                      svn_pool_destroy (pool);                                //free allocated memory\r
217                         return &dirstat;\r
218                 }\r
219         } // if (bIsFolder) \r
220         \r
221         m_nCounter = 0;\r
222         \r
223         //Fill in the cache with\r
224         //all files inside the same folder as the asked file/folder is\r
225         //since subversion can do this in one step\r
226 //      localctx->auth_baton = NULL;\r
227 \r
228 //      statushash = apr_hash_make(pool);\r
229 //      git_revnum_t youngest = GIT_INVALID_REVNUM;\r
230 //      git_opt_revision_t rev;\r
231 //      rev.kind = git_opt_revision_unspecified;\r
232         try\r
233         {\r
234                 if (g_ShellCache.GetCacheType() == ShellCache::dll)\r
235                 {\r
236                         // gitindex.h based status\r
237 \r
238                         // extract the sub-path (relative to project root)\r
239                         CString sSubPath;\r
240                         CString s = filepath.GetWinPathString();\r
241                         if (s.GetLength() > sProjectRoot.GetLength())\r
242                         {\r
243                                         sSubPath = s.Right(s.GetLength() - sProjectRoot.GetLength() - 1/*otherwise it gets initial slash*/);\r
244                         }\r
245 \r
246                         git_wc_status_kind status;\r
247 \r
248                         err = g_IndexFileMap.GetFileStatus((CString&)sProjectRoot,sSubPath,&status,true,true,fillstatusmap_idx,this);\r
249                 }\r
250                 else\r
251                 {\r
252                         // extract the sub-path (relative to project root)\r
253 //MessageBox(NULL, filepath.GetDirectory().GetWinPathString(), sProjectRoot, MB_OK);\r
254                         LPCTSTR lpszSubPath = NULL;\r
255                         CString sSubPath;\r
256                         CString s = filepath.GetDirectory().GetWinPathString();\r
257                         if (s.GetLength() > sProjectRoot.GetLength())\r
258                         {\r
259                                 sSubPath = s.Right(s.GetLength() - sProjectRoot.GetLength() - 1/*otherwise it gets initial slash*/);\r
260                                 lpszSubPath = sSubPath;\r
261                         }\r
262 \r
263 //if (lpszSubPath) MessageBoxA(NULL, lpszSubPath, "BuildCache", MB_OK);\r
264 //MessageBoxA(NULL, CStringA(sProjectRoot), sSubPath, MB_OK);\r
265 //OutputDebugStringA("---");OutputDebugStringW(sProjectRoot);OutputDebugStringA(" = ");OutputDebugStringW(filepath.GetWinPathString());OutputDebugStringA(" - ");OutputDebugStringW(sSubPath);OutputDebugStringA("\r\n");\r
266                         err = !wgEnumFiles(sProjectRoot, lpszSubPath, WGEFF_NoRecurse|WGEFF_FullPath|WGEFF_DirStatusAll, &fillstatusmap, this);\r
267 \r
268                         /*err = svn_client_status4 (&youngest,\r
269                                 filepath.GetDirectory().GetSVNApiPath(pool),\r
270                                 &rev,\r
271                                 fillstatusmap,\r
272                                 this,\r
273                                 svn_depth_immediates,           //depth\r
274                                 TRUE,           //getall\r
275                                 FALSE,          //update\r
276                                 TRUE,           //noignore\r
277                                 FALSE,          //ignore externals\r
278                                 NULL,\r
279                                 localctx,\r
280                                 pool);*/\r
281                 }\r
282         }\r
283         catch ( ... )\r
284         {\r
285         }\r
286 \r
287         // Error present if function is not under version control\r
288         if (err != NULL)\r
289         {\r
290 //              svn_error_clear(err);\r
291 //              svn_pool_destroy (pool);                                //free allocated memory\r
292                 return &invalidstatus;  \r
293         }\r
294 \r
295 //      svn_error_clear(err);\r
296 //      svn_pool_destroy (pool);                                //free allocated memory\r
297         m_TimeStamp = GetTickCount();\r
298         const FileStatusCacheEntry * ret = NULL;\r
299         FileStatusMap::const_iterator iter;\r
300         if ((iter = m_cache.find(filepath.GetWinPath())) != m_cache.end())\r
301         {\r
302                 ret = &iter->second;\r
303                 m_mostRecentPath = filepath;\r
304                 m_mostRecentStatus = ret;\r
305         }\r
306         else\r
307         {\r
308                 // for SUBST'ed drives, Subversion doesn't return a path with a backslash\r
309                 // e.g. G:\ but only G: when fetching the status. So search for that\r
310                 // path too before giving up.\r
311                 // This is especially true when right-clicking directly on a SUBST'ed\r
312                 // drive to get the context menu\r
313                 if (_tcslen(filepath.GetWinPath())==3)\r
314                 {\r
315                         if ((iter = m_cache.find((LPCTSTR)filepath.GetWinPathString().Left(2))) != m_cache.end())\r
316                         {\r
317                                 ret = &iter->second;\r
318                                 m_mostRecentPath = filepath;\r
319                                 m_mostRecentStatus = ret;\r
320                         }\r
321                 }               \r
322         }\r
323         if (ret)\r
324                 return ret;\r
325         return &invalidstatus;\r
326 }\r
327 \r
328 DWORD GitFolderStatus::GetTimeoutValue()\r
329 {\r
330         DWORD timeout = GITFOLDERSTATUS_CACHETIMEOUT;\r
331         DWORD factor = m_cache.size()/200;\r
332         if (factor==0)\r
333                 factor = 1;\r
334         return factor*timeout;\r
335 }\r
336 \r
337 const FileStatusCacheEntry * GitFolderStatus::GetFullStatus(const CTGitPath& filepath, BOOL bIsFolder, BOOL bColumnProvider)\r
338 {\r
339         const FileStatusCacheEntry * ret = NULL;\r
340 \r
341         CString sProjectRoot;\r
342         BOOL bHasAdminDir = g_ShellCache.HasSVNAdminDir(filepath.GetWinPath(), bIsFolder, &sProjectRoot);\r
343         \r
344         //no overlay for unversioned folders\r
345         if ((!bColumnProvider)&&(!bHasAdminDir))\r
346                 return &invalidstatus;\r
347         //for the SVNStatus column, we have to check the cache to see\r
348         //if it's not just unversioned but ignored\r
349         ret = GetCachedItem(filepath);\r
350         if ((ret)&&(ret->status == git_wc_status_unversioned)&&(bIsFolder)&&(bHasAdminDir))\r
351         {\r
352                 // an 'unversioned' folder, but with an ADMIN dir --> nested layout!\r
353                 // NOTE: this could be a sub-project in git, or just some standalone project inside of another, either way a TODO\r
354                 ret = BuildCache(filepath, sProjectRoot, bIsFolder, TRUE);\r
355                 if (ret)\r
356                         return ret;\r
357                 else\r
358                         return &invalidstatus;\r
359         }\r
360         if (ret)\r
361                 return ret;\r
362 \r
363         //if it's not in the cache and has no admin dir, then we assume\r
364         //it's not ignored too\r
365         if ((bColumnProvider)&&(!bHasAdminDir))\r
366                 return &invalidstatus;\r
367         ret = BuildCache(filepath, sProjectRoot, bIsFolder);\r
368         if (ret)\r
369                 return ret;\r
370         else\r
371                 return &invalidstatus;\r
372 }\r
373 \r
374 const FileStatusCacheEntry * GitFolderStatus::GetCachedItem(const CTGitPath& filepath)\r
375 {\r
376         sCacheKey.assign(filepath.GetWinPath());\r
377         FileStatusMap::const_iterator iter;\r
378         const FileStatusCacheEntry *retVal;\r
379 \r
380         if(m_mostRecentPath.IsEquivalentTo(CTGitPath(sCacheKey.c_str())))\r
381         {\r
382                 // We've hit the same result as we were asked for last time\r
383                 ATLTRACE2(_T("fast cache hit for %s\n"), filepath);\r
384                 retVal = m_mostRecentStatus;\r
385         }\r
386         else if ((iter = m_cache.find(sCacheKey)) != m_cache.end())\r
387         {\r
388                 ATLTRACE2(_T("cache found for %s\n"), filepath);\r
389                 retVal = &iter->second;\r
390                 m_mostRecentStatus = retVal;\r
391                 m_mostRecentPath = CTGitPath(sCacheKey.c_str());\r
392         }\r
393         else\r
394         {\r
395                 retVal = NULL;\r
396         }\r
397 \r
398         if(retVal != NULL)\r
399         {\r
400                 // We found something in a cache - check that the cache is not timed-out or force-invalidated\r
401                 DWORD now = GetTickCount();\r
402 \r
403                 if ((now >= m_TimeStamp)&&((now - m_TimeStamp) > GetTimeoutValue()))\r
404                 {\r
405                         // Cache is timed-out\r
406                         ATLTRACE("Cache timed-out\n");\r
407                         ClearCache();\r
408                         retVal = NULL;\r
409                 }\r
410                 else if(WaitForSingleObject(m_hInvalidationEvent, 0) == WAIT_OBJECT_0)\r
411                 {\r
412                         // TortoiseProc has just done something which has invalidated the cache\r
413                         ATLTRACE("Cache invalidated\n");\r
414                         ClearCache();\r
415                         retVal = NULL;\r
416                 }\r
417                 return retVal;\r
418         }\r
419         return NULL;\r
420 }\r
421 \r
422 BOOL GitFolderStatus::fillstatusmap(const struct wgFile_s *pFile, void *pUserData)\r
423 {\r
424         GitFolderStatus *Stat = (GitFolderStatus*)pUserData;\r
425 \r
426         FileStatusMap &cache = Stat->m_cache;\r
427         FileStatusCacheEntry s;\r
428         s.needslock = false;\r
429         s.tree_conflict = false;\r
430 \r
431         s.author = Stat->authors.GetString(NULL);\r
432         s.url = Stat->urls.GetString(NULL);\r
433         if (pFile->sha1)\r
434                 s.rev = ConvertHashToRevnum(pFile->sha1);\r
435         s.owner = Stat->owners.GetString(NULL);\r
436 \r
437         s.status = git_wc_status_none;\r
438 \r
439         //s.status = GitStatus::GetMoreImportant(s.status, status->text_status);\r
440         //s.status = GitStatus::GetMoreImportant(s.status, status->prop_status);\r
441         s.status = GitStatusFromWingit(pFile->nStatus);\r
442 \r
443         // TODO ?: s.blaha = pFile->nStage\r
444 \r
445         //s.lock = status->repos_lock;\r
446         //s.tree_conflict = (status->tree_conflict != NULL);\r
447 \r
448         s.askedcounter = GITFOLDERSTATUS_CACHETIMES;\r
449         stdstring str;\r
450         if (pFile->sFileName)\r
451         {\r
452                 str = pFile->sFileName;//CUnicodeUtils::StdGetUnicode(pFile->sFileName);\r
453                 std::replace(str.begin(), str.end(), '/', '\\');\r
454 //MessageBox(NULL, str.c_str(), _T(""), MB_OK);\r
455         }\r
456         else\r
457                 str = _T(" ");\r
458         cache[str] = s;\r
459 \r
460         return FALSE;\r
461 }\r
462 \r
463 void GitFolderStatus::fillstatusmap_idx(CString &path,git_wc_status_kind status,void *pUserData)\r
464 {\r
465         GitFolderStatus *Stat = (GitFolderStatus*)pUserData;\r
466 \r
467         FileStatusMap &cache = Stat->m_cache;\r
468         FileStatusCacheEntry s;\r
469         s.needslock = false;\r
470         s.tree_conflict = false;\r
471 \r
472         s.author = Stat->authors.GetString(NULL);\r
473         s.url = Stat->urls.GetString(NULL);\r
474 //      s.rev = -1;\r
475         s.owner = Stat->owners.GetString(NULL);\r
476 \r
477         s.status = status;\r
478 \r
479         //s.status = GitStatus::GetMoreImportant(s.status, status->text_status);\r
480         //s.status = GitStatus::GetMoreImportant(s.status, status->prop_status);\r
481         //s.status = GitStatusFromWingit(pFile->nStatus);\r
482 \r
483         // TODO ?: s.blaha = pFile->nStage\r
484 \r
485         //s.lock = status->repos_lock;\r
486         //s.tree_conflict = (status->tree_conflict != NULL);\r
487 \r
488         s.askedcounter = GITFOLDERSTATUS_CACHETIMES;\r
489         //stdstring str;\r
490         //if (pFile->sFileName)\r
491         //{\r
492         //      str = CUnicodeUtils::StdGetUnicode(pFile->sFileName);\r
493         //      std::replace(str.begin(), str.end(), '/', '\\');\r
494 //MessageBox(NULL, str.c_str(), _T(""), MB_OK);\r
495         //}\r
496         //else\r
497         //      str = _T(" ");\r
498         if( path.Right(1) == _T("\\"))\r
499         {\r
500                 path=path.Left(path.GetLength()-1);\r
501         }\r
502         stdstring str;\r
503         str=path;\r
504         cache[str] = s;\r
505 \r
506         return;\r
507 }\r
508 \r
509 #if 0\r
510 git_error_t* GitFolderStatus::fillstatusmap(void * baton, const char * path, git_wc_status2_t * status, apr_pool_t * /*pool*/)\r
511 {\r
512         GitFolderStatus * Stat = (GitFolderStatus *)baton;\r
513         FileStatusMap * cache = &Stat->m_cache;\r
514         FileStatusCacheEntry s;\r
515         s.needslock = false;\r
516         s.tree_conflict = false;\r
517         if ((status)&&(status->entry))\r
518         {\r
519                 s.author = Stat->authors.GetString(status->entry->cmt_author);\r
520                 s.url = Stat->urls.GetString(status->entry->url);\r
521                 s.rev = status->entry->cmt_rev;\r
522                 s.owner = Stat->owners.GetString(status->entry->lock_owner);\r
523                 if (status->entry->present_props)\r
524                         s.needslock = strstr(status->entry->present_props, "svn:needs-lock") ? true : false;\r
525         }\r
526         else\r
527         {\r
528                 s.author = Stat->authors.GetString(NULL);\r
529                 s.url = Stat->urls.GetString(NULL);\r
530                 s.rev = -1;\r
531                 s.owner = Stat->owners.GetString(NULL);\r
532         }\r
533         s.status = git_wc_status_none;\r
534         if (status)\r
535         {\r
536                 s.status = GitStatus::GetMoreImportant(s.status, status->text_status);\r
537                 s.status = GitStatus::GetMoreImportant(s.status, status->prop_status);\r
538                 s.lock = status->repos_lock;\r
539                 s.tree_conflict = (status->tree_conflict != NULL);\r
540         }\r
541         s.askedcounter = GITFOLDERSTATUS_CACHETIMES;\r
542         stdstring str;\r
543         if (path)\r
544         {\r
545                 str = CUnicodeUtils::StdGetUnicode(path);\r
546                 std::replace(str.begin(), str.end(), '/', '\\');\r
547         }\r
548         else\r
549                 str = _T(" ");\r
550         (*cache)[str] = s;\r
551 \r
552         return GIT_NO_ERROR;\r
553 }\r
554 \r
555 git_error_t* GitFolderStatus::findfolderstatus(void * baton, const char * path, git_wc_status2_t * status, apr_pool_t * /*pool*/)\r
556 {\r
557         GitFolderStatus * Stat = (GitFolderStatus *)baton;\r
558         if ((Stat)&&(Stat->folderpath.IsEquivalentTo(CTGitPath(CString(path)))))\r
559         {\r
560                 Stat->dirstatus = status;\r
561         }\r
562 \r
563         return GIT_NO_ERROR;\r
564 }\r
565 #endif\r
566 \r
567 void GitFolderStatus::ClearCache()\r
568 {\r
569         m_cache.clear();\r
570         m_mostRecentStatus = NULL;\r
571         m_mostRecentPath.Reset();\r
572         // If we're about to rebuild the cache, there's no point hanging on to \r
573         // an event which tells us that it's invalid\r
574         ResetEvent(m_hInvalidationEvent);\r
575 }\r