rainmeter-studio/Plugins/PluginFolderInfo/FolderInfoPlugin.cpp

/*
  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"
#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();
			}

			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);
}