src/TortoiseProc/RepositoryBrowser.h

   1 // TortoiseGit - a Windows shell extension for easy version control
   2
   3 // Copyright (C) 2009-2012 - TortoiseGit
   4 // Copyright (C) 2012 - Sven Strickroth <email@cs-ware.de>
   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 "git2.h"
  23 #include <map>
  24 #include "StandAloneDlg.h"
  25
  26 #define REPOBROWSER_CTRL_MIN_WIDTH      20
  27
  28 class CShadowFilesTree;
  29
  30 typedef std::map<CString, CShadowFilesTree> TShadowFilesTreeMap;
  31
  32 class CShadowFilesTree
  33 {
  34 public:
  35         CShadowFilesTree()
  36         : m_hTree(NULL)
  37         , m_pParent(NULL)
  38         , m_bFolder(false)
  39         , m_iSize(0)
  40         {}
  41
  42         CString                         m_sName;
  43         size_t                          m_iSize;
  44         bool                            m_bFolder;
  45
  46         HTREEITEM                       m_hTree;
  47
  48         TShadowFilesTreeMap     m_ShadowTree;
  49         CShadowFilesTree*       m_pParent;
  50
  51         CString GetFullName() const
  52         {
  53                 if (m_pParent == NULL)
  54                         return m_sName;
  55
  56                 CString parentPath = m_pParent->GetFullName();
  57                 if (parentPath.IsEmpty())
  58                         return m_sName;
  59                 else
  60                         return m_pParent->GetFullName() + _T("/") + m_sName;
  61         }
  62 };
  63 typedef std::vector<CShadowFilesTree *> TShadowFilesTreeList;
  64
  65 class CRepositoryBrowser : public CResizableStandAloneDialog
  66 {
  67         DECLARE_DYNAMIC(CRepositoryBrowser)
  68
  69 public:
  70         CRepositoryBrowser(CString rev, CWnd* pParent = NULL);  // standard constructor
  71         virtual ~CRepositoryBrowser();
  72
  73         // Dialog Data
  74         enum { IDD = IDD_REPOSITORY_BROWSER };
  75
  76         static bool s_bSortLogical;
  77
  78         enum eCmd
  79         {
  80                 eCmd_Open = WM_APP,
  81                 eCmd_OpenWith,
  82                 eCmd_OpenWithAlternativeEditor,
  83                 eCmd_ViewLog,
  84                 eCmd_Blame,
  85                 eCmd_CompareWC,
  86                 eCmd_Revert,
  87                 eCmd_SaveAs,
  88                 eCmd_CopyPath,
  89         };
  90
  91         enum eCol
  92         {
  93                 eCol_Name,
  94                 eCol_Extension,
  95                 eCol_FileSize,
  96         };
  97
  98         enum eOpenType
  99         {
 100                 ALTERNATIVEEDITOR,
 101                 OPEN,
 102                 OPEN_WITH,
 103         };
 104
 105         enum eSelectionType
 106         {
 107                 ONLY_FILES,
 108                 ONLY_FOLDERS,
 109                 MIXED,
 110         };
 111
 112 private:
 113         virtual void DoDataExchange(CDataExchange* pDX);        // DDX/DDV support
 114
 115         DECLARE_MESSAGE_MAP()
 116
 117         afx_msg void                    OnOK();
 118         afx_msg void                    OnCancel();
 119         virtual BOOL                    OnInitDialog();
 120
 121         CTreeCtrl                               m_RepoTree;
 122         CListCtrl                               m_RepoList;
 123
 124         afx_msg void                    OnLvnColumnclickRepoList(NMHDR *pNMHDR, LRESULT *pResult);
 125         int                                             m_currSortCol;
 126         bool                                    m_currSortDesc;
 127
 128         CShadowFilesTree                m_TreeRoot;
 129         int                                             ReadTreeRecursive(git_repository &repo, git_tree * tree, CShadowFilesTree * treeroot);
 130         int                                             ReadTree(CShadowFilesTree * treeroot);
 131         int                                             m_nIconFolder;
 132         int                                             m_nOpenIconFolder;
 133
 134         bool                                    m_bHasWC;
 135
 136         void                                    Refresh();
 137         CString                                 m_sRevision;
 138         void                                    FillListCtrlForTreeNode(HTREEITEM treeNode);
 139         void                                    FillListCtrlForShadowTree(CShadowFilesTree* pTree);
 140         afx_msg void                    OnTvnSelchangedRepoTree(NMHDR *pNMHDR, LRESULT *pResult);
 141
 142         afx_msg void                    OnNMDblclk_RepoList(NMHDR *pNMHDR, LRESULT *pResult);
 143         afx_msg void                    OnContextMenu(CWnd* pWndFrom, CPoint point);
 144         void                                    OnContextMenu_RepoList(CPoint point);
 145         void                                    OnContextMenu_RepoTree(CPoint point);
 146         void                                    ShowContextMenu(CPoint point, TShadowFilesTreeList &selectedLeafs, eSelectionType selType);
 147
 148         void                                    FileSaveAs(const CString path);
 149         void                                    OpenFile(const CString path, eOpenType mode);
 150         bool                                    RevertItemToVersion(const CString &path);
 151
 152         afx_msg void                    OnBnClickedButtonRevision();
 153
 154         virtual BOOL                    PreTranslateMessage(MSG* pMsg);
 155
 156         /// resizes the control so that the divider is at position 'point'
 157         void                                    HandleDividerMove(CPoint point, bool bDraw);
 158         bool                                    bDragMode;
 159         void                                    SaveDividerPosition();
 160         int                                             oldy, oldx;
 161         /// draws the bar when the tree and list control are resized
 162         void                                    DrawXorBar(CDC * pDC, int x1, int y1, int width, int height);
 163         afx_msg BOOL                    OnSetCursor(CWnd* pWnd, UINT nHitTest, UINT message);
 164         afx_msg void                    OnMouseMove(UINT nFlags, CPoint point);
 165         afx_msg void                    OnCaptureChanged(CWnd *pWnd);
 166         afx_msg void                    OnLButtonDown(UINT nFlags, CPoint point);
 167         afx_msg void                    OnLButtonUp(UINT nFlags, CPoint point);
 168 };