src/TortoiseProc/SubmoduleUpdateDlg.cpp

   1 // TortoiseGit - a Windows shell extension for easy version control
   2
   3 // Copyright (C) 2012 Sven Strickroth, <email@cs-ware.de>
   4
   5 // This program is free software; you can redistribute it and/or
   6 // modify it under the terms of the GNU General Public License
   7 // as published by the Free Software Foundation; either version 2
   8 // of the License, or (at your option) any later version.
   9
  10 // This program is distributed in the hope that it will be useful,
  11 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  12 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13 // GNU General Public License for more details.
  14
  15 // You should have received a copy of the GNU General Public License
  16 // along with this program; if not, write to the Free Software Foundation,
  17 // 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  18 //
  19
  20 #include "stdafx.h"
  21 #include "TortoiseProc.h"
  22 #include "SubmoduleUpdateDlg.h"
  23 #include "AppUtils.h"
  24 #include "UnicodeUtils.h"
  25
  26 IMPLEMENT_DYNAMIC(CSubmoduleUpdateDlg, CStandAloneDialog)
  27
  28 CSubmoduleUpdateDlg::CSubmoduleUpdateDlg(CWnd* pParent /*=NULL*/)
  29         : CStandAloneDialog(CSubmoduleUpdateDlg::IDD, pParent)
  30         , m_bInit(true)
  31         , m_bRecursive(FALSE)
  32         , m_bForce(FALSE)
  33         , m_bNoFetch(FALSE)
  34         , m_bMerge(FALSE)
  35         , m_bRebase(FALSE)
  36 {
  37 }
  38
  39 CSubmoduleUpdateDlg::~CSubmoduleUpdateDlg()
  40 {
  41 }
  42
  43 void CSubmoduleUpdateDlg::DoDataExchange(CDataExchange* pDX)
  44 {
  45         CStandAloneDialog::DoDataExchange(pDX);
  46         DDX_Control(pDX, IDC_LIST_PATH, m_PathListBox);
  47         DDX_Check(pDX, IDC_CHECK_SUBMODULE_INIT, m_bInit);
  48         DDX_Check(pDX, IDC_CHECK_SUBMODULE_RECURSIVE, m_bRecursive);
  49         DDX_Check(pDX, IDC_FORCE, m_bForce);
  50         DDX_Check(pDX, IDC_CHECK_SUBMODULE_NOFETCH, m_bNoFetch);
  51         DDX_Check(pDX, IDC_CHECK_SUBMODULE_MERGE, m_bMerge);
  52         DDX_Check(pDX, IDC_CHECK_SUBMODULE_REBASE, m_bRebase);
  53 }
  54
  55
  56 BEGIN_MESSAGE_MAP(CSubmoduleUpdateDlg, CStandAloneDialog)
  57         ON_BN_CLICKED(IDOK, &CSubmoduleUpdateDlg::OnBnClickedOk)
  58         ON_BN_CLICKED(IDHELP, &CSubmoduleUpdateDlg::OnBnClickedHelp)
  59         ON_LBN_SELCHANGE(IDC_LIST_PATH, &CSubmoduleUpdateDlg::OnLbnSelchangeListPath)
  60 END_MESSAGE_MAP()
  61
  62 static int SubmoduleCallback(git_submodule *sm, const char * /*name*/, void *payload)
  63 {
  64         STRING_VECTOR *list = *(STRING_VECTOR **)payload;
  65         STRING_VECTOR *prefixList = *((STRING_VECTOR **)payload + 1);
  66         CString path = CUnicodeUtils::GetUnicode(git_submodule_path(sm));
  67         if (prefixList->empty())
  68         {
  69                 list->push_back(path);
  70         }
  71         else
  72         {
  73                 for (size_t i = 0; i < prefixList->size(); ++i)
  74                 {
  75                         CString prefix = prefixList->at(i) + _T("/");
  76                         if (path.Left(prefix.GetLength()) == prefix)
  77                                 list->push_back(path);
  78                 }
  79         }
  80         return 0;
  81 }
  82
  83 static void GetSubmodulePathList(STRING_VECTOR &list, STRING_VECTOR &prefixList)
  84 {
  85         git_repository *repo;
  86         if (git_repository_open(&repo, CUnicodeUtils::GetUTF8(g_Git.m_CurrentDir)))
  87         {
  88                 MessageBox(NULL, CGit::GetLibGit2LastErr(_T("Could not open repository.")), _T("TortoiseGit"), MB_ICONERROR);
  89                 return;
  90         }
  91
  92         STRING_VECTOR *listParams[] = { &list, &prefixList };
  93         if (git_submodule_foreach(repo, SubmoduleCallback, &listParams))
  94         {
  95                 MessageBox(NULL, CGit::GetLibGit2LastErr(_T("Could not get submodule list.")), _T("TortoiseGit"), MB_ICONERROR);
  96                 return;
  97         }
  98
  99         git_repository_free(repo);
 100         std::sort(list.begin(), list.end());
 101 }
 102
 103 BOOL CSubmoduleUpdateDlg::OnInitDialog()
 104 {
 105         CStandAloneDialog::OnInitDialog();
 106         CAppUtils::MarkWindowAsUnpinnable(m_hWnd);
 107
 108         CString sWindowTitle;
 109         GetWindowText(sWindowTitle);
 110         CAppUtils::SetWindowTitle(m_hWnd, g_Git.m_CurrentDir, sWindowTitle);
 111
 112         AdjustControlSize(IDC_CHECK_SUBMODULE_INIT);
 113         AdjustControlSize(IDC_CHECK_SUBMODULE_RECURSIVE);
 114         AdjustControlSize(IDC_CHECK_SUBMODULE_NOFETCH);
 115         AdjustControlSize(IDC_CHECK_SUBMODULE_MERGE);
 116         AdjustControlSize(IDC_CHECK_SUBMODULE_REBASE);
 117
 118         CString WorkingDir = g_Git.m_CurrentDir;
 119         WorkingDir.Replace(_T(':'), _T('_'));
 120
 121         m_regPath = CRegString(CString(_T("Software\\TortoiseGit\\History\\SubmoduleUpdatePath\\") + WorkingDir));
 122         CString path = m_regPath;
 123         STRING_VECTOR list;
 124         GetSubmodulePathList(list, m_PathFilterList);
 125         STRING_VECTOR selected;
 126         if (m_PathList.empty())
 127         {
 128                 int pos = 0;
 129                 while (pos >= 0)
 130                 {
 131                         CString part = path.Tokenize(_T("|"), pos);
 132                         if (!part.IsEmpty())
 133                                 selected.push_back(part);
 134                 }
 135         }
 136         else
 137         {
 138                 for (size_t i = 0; i < m_PathList.size(); ++i)
 139                         selected.push_back(m_PathList[i]);
 140         }
 141
 142         for (size_t i = 0; i < list.size(); ++i)
 143         {
 144                 m_PathListBox.AddString(list[i]);
 145                 if (selected.size() == 0)
 146                         m_PathListBox.SetSel((int)i);
 147                 else
 148                 {
 149                         for (int j = 0; j < selected.size(); ++j)
 150                         {
 151                                 if (selected[j] == list[i])
 152                                         m_PathListBox.SetSel((int)i);
 153                         }
 154                 }
 155         }
 156
 157         UpdateData(FALSE);
 158
 159         return TRUE;
 160 }
 161
 162 void CSubmoduleUpdateDlg::OnBnClickedOk()
 163 {
 164         CStandAloneDialog::UpdateData(TRUE);
 165         m_PathList.clear();
 166         CString selected;
 167         for (int i = 0; i < m_PathListBox.GetCount(); ++i)
 168         {
 169                 if (m_PathListBox.GetSel(i))
 170                 {
 171                         if (!selected.IsEmpty())
 172                                 selected.Append(_T("|"));
 173                         CString text;
 174                         m_PathListBox.GetText(i, text);
 175                         m_PathList.push_back(text);
 176                         selected.Append(text);
 177                 }
 178         }
 179         m_regPath = selected;
 180
 181         CStandAloneDialog::OnOK();
 182 }
 183
 184 void CSubmoduleUpdateDlg::OnBnClickedHelp()
 185 {
 186         OnHelp();
 187 }
 188
 189 void CSubmoduleUpdateDlg::OnLbnSelchangeListPath()
 190 {
 191         GetDlgItem(IDOK)->EnableWindow(m_PathListBox.GetSelCount() > 0 ? TRUE : FALSE);
 192 }