rainmeter-studio/Plugins/PluginFolderInfo/FolderInfoPlugin.cpp

159 lines
3.8 KiB
C++
Raw Normal View History

/*
Copyright (C) 2010 Elestel
This program is free software; you can redistribute it and/or
modify it under the terms of the GNU General Public License
as published by the Free Software Foundation; either version 2
of the License, or (at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program; if not, write to the Free Software
Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
*/
#include <windows.h>
#include <vector>
#include <algorithm>
#include "FolderInfo.h"
#include "../API/RainmeterAPI.h"
2011-02-03 18:09:24 +00:00
#include "../../Library/DisableThreadLibraryCalls.h" // contains DllMain entry point
enum MeasureType
{
MEASURE_FILECOUNT,
MEASURE_FOLDERCOUNT,
MEASURE_FOLDERSIZE
};
struct MeasureData
{
LPCWSTR section;
CFolderInfo* folder;
MeasureType type;
MeasureData(LPCWSTR section) :
section(section),
folder(),
type(MEASURE_FILECOUNT)
{
}
};
std::vector<MeasureData*> g_Measures;
PLUGIN_EXPORT void Initialize(void** data, void* rm)
{
MeasureData* measure = new MeasureData(RmGetMeasureName(rm));
*data = measure;
g_Measures.push_back(measure);
}
PLUGIN_EXPORT void Reload(void* data, void* rm, double* maxValue)
{
MeasureData* measure = (MeasureData*)data;
LPCWSTR str = RmReadString(rm, L"Folder", L"", FALSE);
if (*str == L'[')
{
CFolderInfo* oldFolder = measure->folder;
measure->folder = NULL;
int len = wcslen(str);
for (auto iter = g_Measures.cbegin(); iter != g_Measures.cend(); ++iter)
{
if (wcsncmp(&str[1], (*iter)->section, len - 2) == 0)
{
measure->folder = (*iter)->folder;
measure->folder->AddInstance();
}
}
if (oldFolder)
{
oldFolder->RemoveInstance();
}
}
else if (*str)
{
LPCWSTR path = RmPathToAbsolute(rm, str);
if (!measure->folder || wcscmp(measure->folder->GetPath(), path) != 0)
{
if (measure->folder)
{
measure->folder->RemoveInstance();
}
2011-03-29 19:21:57 +00:00
measure->folder = new CFolderInfo(path);
str = RmReadString(rm, L"RegExpFilter", L"");
if (*str)
{
measure->folder->SetRegExpFilter(str);
}
measure->folder->IncludeSubFolders((bool)RmReadInt(rm, L"IncludeSubFolders", 0));
measure->folder->IncludeHiddenFiles((bool)RmReadInt(rm, L"IncludeHiddenFiles", 0));
measure->folder->IncludeSystemFiles((bool)RmReadInt(rm, L"IncludeSystemFiles", 0));
}
}
str = RmReadString(rm, L"InfoType", L"");
if (_wcsicmp(str, L"FolderSize") == 0 || _wcsicmp(str, L"FolderSizeStr") == 0)
{
measure->type = MEASURE_FOLDERSIZE;
}
else if (_wcsicmp(str, L"FolderCount") == 0 || _wcsicmp(str, L"FolderCountStr") == 0)
{
measure->type = MEASURE_FOLDERCOUNT;
}
else if (_wcsicmp(str, L"FileCount") == 0 || _wcsicmp(str, L"FileCountStr") == 0)
{
measure->type = MEASURE_FILECOUNT;
}
}
PLUGIN_EXPORT double Update(void* data)
{
MeasureData* measure = (MeasureData*)data;
if (!measure->folder)
{
return 0.0;
}
measure->folder->Update();
switch (measure->type)
{
case MEASURE_FOLDERSIZE:
return measure->folder->GetSize();
case MEASURE_FILECOUNT:
return measure->folder->GetFileCount();
case MEASURE_FOLDERCOUNT:
return measure->folder->GetFolderCount();
}
return 0.0;
}
PLUGIN_EXPORT void Finalize(void* data)
{
MeasureData* measure = (MeasureData*)data;
if (measure->folder)
{
measure->folder->RemoveInstance();
}
delete measure;
std::vector<MeasureData*>::iterator iter = std::find(g_Measures.begin(), g_Measures.end(), measure);
g_Measures.erase(iter);
}