/**

  * vim: set ts=4 :

  * =============================================================================

  * SourceMod Updater Extension

  * Copyright (C) 2004-2009 AlliedModders LLC.  All rights reserved.

  * =============================================================================

  *

  * This program is free software; you can redistribute it and/or modify it under

  * the terms of the GNU General Public License, version 3.0, as published by the

  * Free Software Foundation.

  * 

  * 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, see <http://www.gnu.org/licenses/>.

  *

  * As a special exception, AlliedModders LLC gives you permission to link the

  * code of this program (as well as its derivative works) to "Half-Life 2," the

  * "Source Engine," the "SourcePawn JIT," and any Game MODs that run on software

  * by the Valve Corporation.  You must obey the GNU General Public License in

  * all respects for all other code used.  Additionally, AlliedModders LLC grants

  * this exception to all derivative works.  AlliedModders LLC defines further

  * exceptions, found in LICENSE.txt (as of this writing, version JULY-31-2007),

  * or <http://www.sourcemod.net/license.php>.

  *

  * Version: $Id$

  */

 #include <stdio.h>

 #include <stdarg.h>

 #include <string.h>

 #include <stdlib.h>

 #include "extension.h"

 #include "Updater.h"

 #include <sh_list.h>

 #include <sh_string.h>

 #define DEFAULT_UPDATE_URL			"http://www.sourcemod.net/update/"

 using namespace SourceHook;

 SmUpdater g_Updater;		/**< Global singleton for extension's main interface */

 SMEXT_LINK(&g_Updater);

 IWebternet *webternet;

 static List<String *> update_errors;

 static IThreadHandle *update_thread;

 static String update_url;

 bool SmUpdater::SDK_OnLoad(char *error, size_t maxlength, bool late)

 {

 	sharesys->AddDependency(myself, "webternet.ext", true, true);

 	SM_GET_IFACE(WEBTERNET, webternet);

 	ThreadParams params;

 	params.flags = Thread_Default;

 	params.prio = ThreadPrio_Low;

 	update_thread = threader->MakeThread(this, &params);

 	if (update_thread == NULL)

 	{

 		smutils->Format(error, maxlength, "Could not create thread");

 		return false;

 	}

 	const char *url = smutils->GetCoreConfigValue("AutoUpdateURL");

 	if (url == NULL)

 	{

 		url = DEFAULT_UPDATE_URL;

 	}

 	update_url.assign(url);

 	return true;

 }

 void SmUpdater::SDK_OnUnload()

 {

 	/* An interface drop might have killed this thread.  

 	 * But if the handle is still there, we have to wait.

 	 */

 	if (update_thread != NULL)

 	{

 		update_thread->WaitForThread();

 		update_thread->DestroyThis();

 	}

 	/* Clear message tables */

 	List<String *>::iterator iter;

 	while (iter != update_errors.end())

 	{

 		iter = update_errors.erase(iter);

 	}

 }

 bool SmUpdater::QueryInterfaceDrop(SourceMod::SMInterface *pInterface)

 {

 	if (pInterface == webternet)

 	{

 		return false;

 	}

 	return true;

 }

 void SmUpdater::NotifyInterfaceDrop(SMInterface *pInterface)

 {

 	if (pInterface == webternet)

 	{

 		/* Can't be in the thread if we're losing this extension. */

 		update_thread->WaitForThread();

 		update_thread->DestroyThis();

 		update_thread = NULL;

 	}

 }

 static void PumpUpdate(void *data)

 {

 	String *str;

 	bool new_files = false;

 	List<String *>::iterator iter;

 	char path[PLATFORM_MAX_PATH];

 	UpdatePart *temp;

 	UpdatePart *part = (UpdatePart*)data;

 	while (part != NULL)

 	{

 		if (strstr(part->file, "..") != NULL)

 		{

 			/* Naughty naughty */

 			AddUpdateError("Detected invalid path escape (..): %s", part->file);

 			goto skip_create;

 		}

 		if (part->data == NULL)

 		{

 			smutils->BuildPath(Path_SM, path, sizeof(path), "gamedata/%s", part->file);

 			if (libsys->IsPathDirectory(path))

 			{

 				continue;

 			}

 			if (!libsys->CreateFolder(path))

 			{

 				AddUpdateError("Could not create folder: %s", path);

 			}

 			else

 			{

 				smutils->LogMessage(myself, "Created folder \"%s\" from updater", path);

 			}

 		}

 		else

 		{

 			smutils->BuildPath(Path_SM, path, sizeof(path), "gamedata/%s", part->file);

 			FILE *fp = fopen(path, "wb");

 			if (fp == NULL)

 			{

 				AddUpdateError("Could not open %s for writing", path);

 				return;

 			}

 			fwrite(part->data, 1, part->length, fp);

 			fclose(fp);

 			smutils->LogMessage(myself,

 				"Successfully updated gamedata file \"%s\"",

 				part->file);

 			new_files = true;

 		}

 skip_create:

 		temp = part->next;

 		free(part->data);

 		free(part->file);

 		delete part;

 		part = temp;

 	}

 	if (update_errors.size())

 	{

 		smutils->LogError(myself, "--- BEGIN ERRORS FROM AUTOMATIC UPDATER ---");

 		for (iter = update_errors.begin();

 			 iter != update_errors.end();

 			 iter++)

 		{

 			str = (*iter);

 			smutils->LogError(myself, "%s", str->c_str());

 		}

 		smutils->LogError(myself, "--- END ERRORS FROM AUTOMATIC UPDATER ---");

 	}

 	if (new_files)

 	{

 		const char *force_restart = smutils->GetCoreConfigValue("ForceRestartAfterUpdate");

 		if (force_restart == NULL || strcasecmp(force_restart, "yes") != 0)

 		{

 			smutils->LogMessage(myself,

 				"SourceMod has been updated, please reload it or restart your server.");

 		}

 		else

 		{

 			char buffer[255];

 			smutils->Format(buffer,

 				sizeof(buffer),

 				"meta unload %d\n",

 				smutils->GetPluginId());

 			gamehelpers->ServerCommand(buffer);

 			smutils->Format(buffer,

 				sizeof(buffer),

 				"changelevel \"%s\"\n",

 				gamehelpers->GetCurrentMap());

 			gamehelpers->ServerCommand(buffer);

 			gamehelpers->ServerCommand("echo SourceMod has been restarted from an automatic update.\n");

 		}

 	}

 }

 void SmUpdater::RunThread(IThreadHandle *pHandle)

 {

 	UpdateReader ur;

 	ur.PerformUpdate(update_url.c_str());

 	smutils->AddFrameAction(PumpUpdate, ur.DetachParts());

 }

 void SmUpdater::OnTerminate(IThreadHandle *pHandle, bool cancel)

 {

 }

 void AddUpdateError(const char *fmt, ...)

 {

 	va_list ap;

 	char buffer[2048];

 	va_start(ap, fmt);

 	smutils->FormatArgs(buffer, sizeof(buffer), fmt, ap);

 	va_end(ap);

 	update_errors.push_back(new String(buffer));

 }