src/Utils/PersonalDictionary.cpp

   1 // TortoiseGit - a Windows shell extension for easy version control
   2
   3 // Copyright (C) 2003-2006, 2008 - TortoiseSVN
   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 <fstream>
  22 #include "PersonalDictionary.h"
  23 #include "PathUtils.h"
  24
  25 CPersonalDictionary::CPersonalDictionary(LONG lLanguage /* = 0*/) :
  26         m_bLoaded(false)
  27 {
  28         m_lLanguage = lLanguage;
  29 }
  30
  31 CPersonalDictionary::~CPersonalDictionary()
  32 {
  33 }
  34
  35 bool CPersonalDictionary::Load()
  36 {
  37         CString sWord;
  38         TCHAR line[PDICT_MAX_WORD_LENGTH + 1];
  39
  40         if (m_bLoaded)
  41                 return true;
  42         TCHAR path[MAX_PATH];           //MAX_PATH ok here.
  43         _tcscpy_s (path, CPathUtils::GetAppDataDirectory());
  44
  45         if (m_lLanguage==0)
  46                 m_lLanguage = GetUserDefaultLCID();
  47
  48         TCHAR sLang[10];
  49         _stprintf_s(sLang, 10, _T("%ld"), m_lLanguage);
  50         _tcscat_s(path, MAX_PATH, sLang);
  51         _tcscat_s(path, MAX_PATH, _T(".dic"));
  52
  53         std::wifstream File;
  54         char filepath[MAX_PATH+1];
  55         SecureZeroMemory(filepath, sizeof(filepath));
  56         WideCharToMultiByte(CP_ACP, NULL, path, -1, filepath, MAX_PATH, NULL, NULL);
  57         File.open(filepath);
  58         if (!File.good())
  59         {
  60                 return false;
  61         }
  62         std::vector<std::wstring> entry;
  63         do
  64         {
  65                 File.getline(line, _countof(line));
  66                 sWord = line;
  67                 dict.insert(sWord);
  68         } while (File.gcount() > 0);
  69         File.close();
  70         m_bLoaded = true;
  71         return true;
  72 }
  73
  74 bool CPersonalDictionary::AddWord(const CString& sWord)
  75 {
  76         if (!m_bLoaded)
  77                 Load();
  78         if (sWord.GetLength() >= PDICT_MAX_WORD_LENGTH)
  79                 return false;
  80         dict.insert(sWord);
  81         return true;
  82 }
  83
  84 bool CPersonalDictionary::FindWord(const CString& sWord)
  85 {
  86         if (!m_bLoaded)
  87                 Load();
  88         // even if the load failed for some reason, we mark it as loaded
  89         // and just assume an empty personal dictionary
  90         m_bLoaded = true;
  91         std::set<CString>::iterator it;
  92         it = dict.find(sWord);
  93         return (it != dict.end());
  94 }
  95
  96 bool CPersonalDictionary::Save()
  97 {
  98         if (!m_bLoaded)
  99                 return false;
 100         TCHAR path[MAX_PATH];           //MAX_PATH ok here.
 101         _tcscpy_s (path, CPathUtils::GetAppDataDirectory());
 102
 103         if (m_lLanguage==0)
 104                 m_lLanguage = GetUserDefaultLCID();
 105
 106         TCHAR sLang[10];
 107         _stprintf_s(sLang, 10, _T("%ld"), m_lLanguage);
 108         _tcscat_s(path, MAX_PATH, sLang);
 109         _tcscat_s(path, MAX_PATH, _T(".dic"));
 110
 111         std::wofstream File;
 112         char filepath[MAX_PATH+1];
 113         SecureZeroMemory(filepath, sizeof(filepath));
 114         WideCharToMultiByte(CP_ACP, NULL, path, -1, filepath, MAX_PATH, NULL, NULL);
 115         File.open(filepath, std::ios_base::binary);
 116         for (std::set<CString>::iterator it = dict.begin(); it != dict.end(); ++it)
 117         {
 118                 File << (LPCTSTR)*it << _T("\n");
 119         }
 120         File.close();
 121         return true;
 122 }