#include <sourcemod>

#pragma semicolon 1

new Handle:g_hDB = INVALID_HANDLE;

public Plugin:myinfo =
{
    name = "Learning MySQL",
    author = "Headline",
    description = "Learning MySQL",
    version = "1.0"
};

public OnPluginStart()
{
	decl String:sError[128];
	g_hDB = SQL_Connect("storenumber", true, sError, sizeof(sError));
	SQL_TQuery(g_hDB, SQLCallback_Void, "CREATE TABLE IF NOT EXISTS `storenumber` ( `number` int(16) NOT NULL , PRIMARY KEY (`number`)) DEFAULT CHARSET=latin1 AUTO_INCREMENT=1", 1);
	if (g_hDB == INVALID_HANDLE)
	{
		PrintToServer("Could not connect to MySQL Server : %s", sError);
		CloseHandle(g_hDB);
	}
	else
	{
		PrintToServer("Connection Is Successful!");
	}
	RegAdminCmd("sm_storenumber", Command_StoreNumber, ADMFLAG_SLAY, "");
	RegAdminCmd("sm_getnumber", Command_GetNumber, ADMFLAG_SLAY, "");
}

public Action:Command_StoreNumber(client, args)
{
	decl String:sArg1[16];
	GetCmdArg(1, sArg1, sizeof(sArg1));
	new iNumber = StringToInt(sArg1);
	
	new String:sQuery[255];
	Format(sQuery, sizeof(sQuery), "INSERT INTO storenumber (number) VALUES ('%i')", iNumber);
	new Handle:hQuery = SQL_Query(g_hDB, sQuery);
	
	if(hQuery != INVALID_HANDLE)
	{
		ReplyToCommand(client, "Your number %i has been stored!", iNumber);
		LogToFile("logs/storenumber.txt", "Number %i was stored by %N", iNumber, client);
	}
	return Plugin_Handled;
}

public Action:Command_GetNumber(client, args)
{
	decl String:sQuery[255];
	new iNumber;
	Format(sQuery, sizeof(sQuery), "SELECT number FROM storenumber WHERE number=\"%i\"", iNumber);
	new Handle:hQuery = SQL_Query(g_hDB, sQuery);
	iNumber = SQL_FetchInt(hQuery, 1);
	if(hQuery != INVALID_HANDLE)
	{
		ReplyToCommand(client, "Your number is %i!", iNumber);
		LogToFile("logs/storenumber.txt", "Number %i was retrieved by %N", iNumber, client);
	}
	return Plugin_Handled;
}


public OnPluginEnd()
{
	CloseHandle(g_hDB);
}

public SQLCallback_Void(Handle:hOwner, Handle:hHndl, const String:sError[], any:iData)
{
	if(hHndl == INVALID_HANDLE)
	{
		SetFailState("Error (%i): %s", iData, sError);
	}
}