src/TortoiseShell/ItemIDList.cpp

   1 // TortoiseSVN - a Windows shell extension for easy version control
   2
   3 // Copyright (C) 2003-2006 - Stefan Kueng
   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 #include "StdAfx.h"
  20 #include "ShellExt.h"
  21 #include "ItemIDList.h"
  22
  23
  24 ItemIDList::ItemIDList(LPCITEMIDLIST item, LPCITEMIDLIST parent) :
  25           item_ (item)
  26         , parent_ (parent)
  27         , count_ (-1)
  28 {
  29 }
  30
  31 ItemIDList::~ItemIDList()
  32 {
  33 }
  34
  35 int ItemIDList::size() const
  36 {
  37         if (count_ == -1)
  38         {
  39                 count_ = 0;
  40                 if (item_)
  41                 {
  42                         LPCSHITEMID ptr = &item_->mkid;
  43                         while (ptr != 0 && ptr->cb != 0)
  44                         {
  45                                 ++count_;
  46                                 LPBYTE byte = (LPBYTE) ptr;
  47                                 byte += ptr->cb;
  48                                 ptr = (LPCSHITEMID) byte;
  49                         }
  50                 }
  51         }
  52         return count_;
  53 }
  54
  55 LPCSHITEMID ItemIDList::get(int index) const
  56 {
  57         int count = 0;
  58
  59         if (item_ == NULL)
  60                 return NULL;
  61         LPCSHITEMID ptr = &item_->mkid;
  62         if (ptr == NULL)
  63                 return NULL;
  64         while (ptr->cb != 0)
  65         {
  66                 if (count == index)
  67                         break;
  68
  69                 ++count;
  70                 LPBYTE byte = (LPBYTE) ptr;
  71                 byte += ptr->cb;
  72                 ptr = (LPCSHITEMID) byte;
  73         }
  74         return ptr;
  75
  76 }
  77 stdstring ItemIDList::toString()
  78 {
  79         IShellFolder *shellFolder = NULL;
  80         IShellFolder *parentFolder = NULL;
  81         STRRET name;
  82         TCHAR * szDisplayName = NULL;
  83         stdstring ret;
  84         HRESULT hr;
  85
  86         hr = ::SHGetDesktopFolder(&shellFolder);
  87         if (!SUCCEEDED(hr))
  88                 return ret;
  89         if (parent_)
  90         {
  91                 hr = shellFolder->BindToObject(parent_, 0, IID_IShellFolder, (void**) &parentFolder);
  92                 if (!SUCCEEDED(hr))
  93                         parentFolder = shellFolder;
  94         }
  95         else
  96         {
  97                 parentFolder = shellFolder;
  98         }
  99
 100         if ((parentFolder != 0)&&(item_ != 0))
 101         {
 102                 hr = parentFolder->GetDisplayNameOf(item_, SHGDN_NORMAL | SHGDN_FORPARSING, &name);
 103                 if (!SUCCEEDED(hr))
 104                 {
 105                         parentFolder->Release();
 106                         return ret;
 107                 }
 108                 hr = StrRetToStr (&name, item_, &szDisplayName);
 109                 if (!SUCCEEDED(hr))
 110                         return ret;
 111         }
 112         parentFolder->Release();
 113         if (szDisplayName == NULL)
 114         {
 115                 CoTaskMemFree(szDisplayName);
 116                 return ret;                     //to avoid a crash!
 117         }
 118         ret = szDisplayName;
 119         CoTaskMemFree(szDisplayName);
 120         return ret;
 121 }
 122
 123 LPCITEMIDLIST ItemIDList::operator& ()
 124 {
 125         return item_;
 126 }