/* Uprising - Windows Shell Extension for Subversion
 * Copyright (C) 2001-2002  Jay Freeman (saurik)
*/

/*
 *        Redistribution and use in source and binary
 * forms, with or without modification, are permitted
 * provided that the following conditions are met:
 * 
 * 1. Redistributions of source code must retain the
 *    above copyright notice, this list of conditions
 *    and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the
 *    above copyright notice, this list of conditions
 *    and the following disclaimer in the documentation
 *    and/or other materials provided with the
 *    distribution.
 * 3. The name of the author may not be used to endorse
 *    or promote products derived from this software
 *    without specific prior written permission.
 * 
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS''
 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING,
 * BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
 * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR BE
 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
 * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
 * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR
 * TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
 * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
 * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*/

#ifndef UPRISING_ICONOVERLAY_H
#define UPRISING_ICONOVERLAY_H

#if _MSC_VER > 1000
#pragma once
#endif // _MSC_VER > 1000

#include "Status.h"

namespace Diapexis {
	extern HMODULE Module;
}

namespace Uprising {

template <svn_wc_status_kind kind>
DIAPEXIS_CLASS IconOverlay :
	public IShellIconOverlayIdentifier
{
	DIAPEXIS_QUERY_BEGIN(IconOverlay)
	DIAPEXIS_QUERY_MAP(Diapexis::Unknown)
	DIAPEXIS_QUERY_MAP_UUID(IShellIconOverlayIdentifier)
	DIAPEXIS_QUERY_END()

  public:
	STDMETHOD(IsMemberOf)(wcstr_t path, DWORD dwAttrib) {
		apr_pool_t *pool;
		apr_pool_create(&pool, NULL);
		svn_error_init_pool(pool);

		char evilbuf[MAX_PATH];
		_snprintf(evilbuf, MAX_PATH, "%S", path);
		svn_stringbuf_t *pathbuf = svn_stringbuf_create(evilbuf, pool);
		svn_path_internal_style(pathbuf);

		svn_wc_status_t *status = NULL;
		svn_wc_status(&status, pathbuf, pool);

		HRESULT value;
		if (status == NULL) value = S_FALSE;
		else {
			value = (status->text_status == kind) ? S_OK : S_FALSE;
		}

		apr_pool_destroy(pool);
		return value;
	}

    STDMETHOD(GetOverlayInfo)(wstr_t file, int cchMax, int *index, DWORD *flags) {
		/*::GetModuleFileNameW(Diapexis::Module, file, cchMax);
		*index = 2;
		*flags = ISIOI_ICONINDEX;*/

		/*char module[MAX_PATH];
		::GetModuleFileName(Diapexis::Module, module, MAX_PATH);*/

		int ikind = kind;
		_snwprintf(file, cchMax, L"D:\\Code\\ninetjer\\private\\uprising\\res\\%s.ico", Status[kind]);
		*flags = ISIOI_ICONFILE;
		return S_OK;
	}

    STDMETHOD(GetPriority)(int *priority) {
		*priority = 50;
		return S_OK;
	}
};

}

#endif//UPRISING_ICONOVERLAY_H