1 /**************************************************************
2 *
3 * Licensed to the Apache Software Foundation (ASF) under one
4 * or more contributor license agreements. See the NOTICE file
5 * distributed with this work for additional information
6 * regarding copyright ownership. The ASF licenses this file
7 * to you under the Apache License, Version 2.0 (the
8 * "License"); you may not use this file except in compliance
9 * with the License. You may obtain a copy of the License at
10 *
11 * http://www.apache.org/licenses/LICENSE-2.0
12 *
13 * Unless required by applicable law or agreed to in writing,
14 * software distributed under the License is distributed on an
15 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
16 * KIND, either express or implied. See the License for the
17 * specific language governing permissions and limitations
18 * under the License.
19 *
20 *************************************************************/
21
22
23
24 // MARKER(update_precomp.py): autogen include statement, do not remove
25 #include "precompiled_shell.hxx"
26 #include "internal/global.hxx"
27 #include "internal/columninfo.hxx"
28 #include "internal/fileextensions.hxx"
29 #include "internal/metainforeader.hxx"
30 #include "internal/utilities.hxx"
31 #include "internal/config.hxx"
32
33 #include <malloc.h>
34
35 //----------------------------
36 //
37 //----------------------------
38
39 namespace /* private */
40 {
41 SHCOLUMNINFO ColumnInfoTable[] =
42 {
43 {{PSGUID_SUMMARYINFORMATION, PIDSI_TITLE}, VT_BSTR, LVCFMT_LEFT, 30, SHCOLSTATE_TYPE_STR, L"Title", L"Title"},
44 {{PSGUID_SUMMARYINFORMATION, PIDSI_AUTHOR}, VT_BSTR, LVCFMT_LEFT, 30, SHCOLSTATE_TYPE_STR, L"Author", L"Author"},
45 {{PSGUID_SUMMARYINFORMATION, PIDSI_SUBJECT}, VT_BSTR, LVCFMT_LEFT, 30, SHCOLSTATE_TYPE_STR, L"Subject", L"Subject"},
46 {{PSGUID_SUMMARYINFORMATION, PIDSI_KEYWORDS}, VT_BSTR, LVCFMT_LEFT, 30, SHCOLSTATE_TYPE_STR, L"Keywords", L"Keywords"},
47 {{PSGUID_SUMMARYINFORMATION, PIDSI_COMMENTS}, VT_BSTR, LVCFMT_LEFT, 30, SHCOLSTATE_TYPE_STR, L"Comments", L"Comments"},
48 {{PSGUID_SUMMARYINFORMATION, PIDSI_PAGECOUNT},VT_BSTR, LVCFMT_LEFT, 30, SHCOLSTATE_TYPE_STR, L"Pagecount", L"Pagecount"}
49 };
50
51 size_t ColumnInfoTableSize = sizeof(ColumnInfoTable)/sizeof(ColumnInfoTable[0]);
52 }
53
54 //----------------------------
55 //
56 //----------------------------
57
CColumnInfo(long RefCnt)58 CColumnInfo::CColumnInfo(long RefCnt) :
59 m_RefCnt(RefCnt)
60 {
61 InterlockedIncrement(&g_DllRefCnt);
62 }
63
64 //----------------------------
65 //
66 //----------------------------
67
~CColumnInfo()68 CColumnInfo::~CColumnInfo()
69 {
70 InterlockedDecrement(&g_DllRefCnt);
71 }
72
73 //-----------------------------
74 // IUnknown methods
75 //-----------------------------
76
QueryInterface(REFIID riid,void __RPC_FAR * __RPC_FAR * ppvObject)77 HRESULT STDMETHODCALLTYPE CColumnInfo::QueryInterface(REFIID riid, void __RPC_FAR *__RPC_FAR *ppvObject)
78 {
79 *ppvObject = 0;
80
81 if (IID_IUnknown == riid || IID_IColumnProvider == riid)
82 {
83 IUnknown* pUnk = static_cast<IColumnProvider*>(this);
84 pUnk->AddRef();
85 *ppvObject = pUnk;
86 return S_OK;
87 }
88
89 return E_NOINTERFACE;
90 }
91
92 //----------------------------
93 //
94 //----------------------------
95
AddRef(void)96 ULONG STDMETHODCALLTYPE CColumnInfo::AddRef(void)
97 {
98 return InterlockedIncrement(&m_RefCnt);
99 }
100
101 //----------------------------
102 //
103 //----------------------------
104
Release(void)105 ULONG STDMETHODCALLTYPE CColumnInfo::Release( void)
106 {
107 long refcnt = InterlockedDecrement(&m_RefCnt);
108
109 if (0 == m_RefCnt)
110 delete this;
111
112 return refcnt;
113 }
114
115 //-----------------------------
116 // IColumnProvider
117 //-----------------------------
118
Initialize(LPCSHCOLUMNINIT)119 HRESULT STDMETHODCALLTYPE CColumnInfo::Initialize(LPCSHCOLUMNINIT /*psci*/)
120 {
121 return S_OK;
122 }
123
124 //-----------------------------
125 // Register all columns we support
126 //-----------------------------
127
GetColumnInfo(DWORD dwIndex,SHCOLUMNINFO * psci)128 HRESULT STDMETHODCALLTYPE CColumnInfo::GetColumnInfo(DWORD dwIndex, SHCOLUMNINFO *psci)
129 {
130 if (dwIndex >= ColumnInfoTableSize)
131 return S_FALSE;
132
133 // Return information on each column we support. Return S_FALSE
134 // to indicate that we have returned information on all our
135 // columns. GetColumnInfo will be called repeatedly until S_FALSE
136 // or an error is returned
137 psci->scid.fmtid = ColumnInfoTable[dwIndex].scid.fmtid;
138 psci->scid.pid = ColumnInfoTable[dwIndex].scid.pid;
139 ZeroMemory(psci->wszTitle, sizeof(psci->wszTitle));
140 wcsncpy(psci->wszTitle, ColumnInfoTable[dwIndex].wszTitle, (sizeof(psci->wszTitle) - 1));
141
142 //wcscpy(psci->wszTitle, ColumnInfoTable[dwIndex].wszTitle);
143
144 return S_OK;
145 }
146
147 //-----------------------------
148 //
149 //-----------------------------
150
GetItemData(LPCSHCOLUMNID pscid,LPCSHCOLUMNDATA pscd,VARIANT * pvarData)151 HRESULT STDMETHODCALLTYPE CColumnInfo::GetItemData(LPCSHCOLUMNID pscid, LPCSHCOLUMNDATA pscd, VARIANT *pvarData)
152 {
153 if (IsOOFileExtension(pscd->pwszExt))
154 {
155 try
156 {
157 std::wstring fname = getShortPathName( std::wstring( pscd->wszFile ) );
158
159 CMetaInfoReader meta_info_accessor(WStringToString(fname));
160
161 VariantClear(pvarData);
162
163 if (IsEqualGUID (pscid->fmtid, FMTID_SummaryInformation) && pscid->pid == PIDSI_TITLE)
164 {
165 pvarData->vt = VT_BSTR;
166 pvarData->bstrVal = SysAllocString(meta_info_accessor.getTagData( META_INFO_TITLE ).c_str());
167
168 return S_OK;
169 }
170 else if (IsEqualGUID (pscid->fmtid, FMTID_SummaryInformation) && pscid->pid == PIDSI_AUTHOR)
171 {
172 pvarData->vt = VT_BSTR;
173 pvarData->bstrVal = SysAllocString(meta_info_accessor.getTagData( META_INFO_AUTHOR).c_str());
174
175 return S_OK;
176 }
177 else if (IsEqualGUID (pscid->fmtid, FMTID_SummaryInformation) && pscid->pid == PIDSI_SUBJECT)
178 {
179 pvarData->vt = VT_BSTR;
180 pvarData->bstrVal = SysAllocString(meta_info_accessor.getTagData( META_INFO_SUBJECT).c_str());
181
182 return S_OK;
183 }
184 else if (IsEqualGUID (pscid->fmtid, FMTID_SummaryInformation) && pscid->pid == PIDSI_KEYWORDS)
185 {
186 pvarData->vt = VT_BSTR;
187 pvarData->bstrVal = SysAllocString(meta_info_accessor.getTagData( META_INFO_KEYWORDS).c_str());
188
189 return S_OK;
190 }
191 else if (IsEqualGUID (pscid->fmtid, FMTID_SummaryInformation) && pscid->pid == PIDSI_COMMENTS)
192 {
193 pvarData->vt = VT_BSTR;
194 pvarData->bstrVal = SysAllocString(meta_info_accessor.getTagData( META_INFO_DESCRIPTION).c_str());
195
196 return S_OK;
197 }
198 else if (IsEqualGUID (pscid->fmtid, FMTID_SummaryInformation) && pscid->pid == PIDSI_PAGECOUNT)
199 {
200 pvarData->vt = VT_BSTR;
201 pvarData->bstrVal = SysAllocString(meta_info_accessor.getTagAttribute( META_INFO_DOCUMENT_STATISTIC, META_INFO_PAGES).c_str());
202
203 return S_OK;
204 }
205 }
206 catch (const std::exception&)
207 {
208 return S_FALSE;
209 }
210 }
211
212 return S_FALSE;
213 }
214
215 //-----------------------------
216 //
217 //-----------------------------
218
IsOOFileExtension(wchar_t * Extension) const219 bool CColumnInfo::IsOOFileExtension(wchar_t* Extension) const
220 {
221 for (size_t i = 0; i < OOFileExtensionTableSize; i++)
222 {
223 if (0 == _wcsicmp(Extension, OOFileExtensionTable[i].ExtensionUnicode))
224 return true;
225 }
226
227 return false;
228 }
229