X-Git-Url: https://git.netwichtig.de/gitweb/?a=blobdiff_plain;f=src%2Fmodules%2Fextra%2Fm_sql.h;h=ce9ce7743df244008ba06a7efd36ae6b4002aac3;hb=3a7023f2c595d14778b3f1f7e53d3914698dd500;hp=0620dedf7becf72a7dbe30e84799edce77f311f2;hpb=2eafeefbb1d2f96a5c59874a93d22cd6721ecde4;p=user%2Fhenk%2Fcode%2Finspircd.git

diff --git a/src/modules/extra/m_sql.h b/src/modules/extra/m_sql.h
index 0620dedf7..ce9ce7743 100644
--- a/src/modules/extra/m_sql.h
+++ b/src/modules/extra/m_sql.h
@@ -1,6 +1,8 @@
 #ifndef __M_SQL_H__
 #define __M_SQL_H__
 
+using namespace std;
+
 #include <string>
 #include <vector>
 
@@ -9,18 +11,27 @@
 #define SQL_ROW    3
 #define SQL_ERROR  4
 #define SQL_END    5
+#define SQL_DONE   6
+#define SQL_OK     7
 
 // SQLRequest is inherited from a basic Request object
 // so that we can neatly pass information around the
 // system.
 
-class SQLRequest
+class SQLRequest : public classbase
 {
  protected:
 	long conn_id;
 	int request_type;
 	std::string thisquery;
  public:
+	SQLRequest(int qt, long cid, std::string query)
+	{
+		this->SetQueryType(qt);
+		this->SetConnID(cid);
+		this->SetQuery(query);
+	}
+
 	void SetConnID(long id)
 	{
 		conn_id = id;
@@ -54,14 +65,15 @@ class SQLRequest
 
 // Upon completion, an SQLRequest returns an SQLResponse.
 
-class SQLResult
+class SQLResult : public classbase
 {
  protected:
 	int resptype;
-	unsigned long count;
+	long count;
 	std::string error;
 	std::map<std::string,std::string> row;
  public:
+
 	void SetRow(std::map<std::string,std::string> r)
 	{
 		row = r;
@@ -94,15 +106,163 @@ class SQLResult
 		return error;
 	}
 
-	void SetCount(unsigned long c)
+	void SetCount(long c)
 	{
 		count = c;
 	}
 
-	unsigned long GetCount()
+	/* This will return a negative value of the SQL server is down */
+	long GetCount()
 	{
 		return count;
 	}
 };
 
+class SQLQuery : public classbase
+{
+ private:
+	SQLRequest* rowrequest;
+	SQLRequest* query;
+	SQLResult* result;
+	SQLResult* rowresult;
+	Request* rowquery;
+	unsigned long dbid;
+	Module* parent;
+	Module* SQLModule;
+	Server* Srv;
+	std::string error;
+
+	bool MakeQueryGoNow(std::string qry)
+	{
+		// Insert Lack of More Original Name here.
+		Request queryrequest((char*)query, parent, SQLModule);
+		result = (SQLResult*)queryrequest.Send();
+		if (result->GetType() != SQL_ERROR)
+		{
+			// Query Is fine.. Prepare to get first row...
+			rowrequest = new SQLRequest(SQL_ROW,dbid,"");
+			rowquery = new Request((char*)rowrequest, parent, SQLModule);
+			return true;
+		}
+		// Query Failed. - Coder Fucked up! (Probably me too :/)
+		Srv->Log(DEBUG, " ============= SQL Error, Query And Error Follow. ============= ");
+		Srv->Log(DEBUG, "Query: "+ qry);
+		Srv->Log(DEBUG, "Error: "+ result->GetError());
+		Srv->Log(DEBUG, " ============================================================== ");
+		error = result->GetError();
+		// Destroy Variables that were set..
+		delete query;
+		query = NULL;
+		result = NULL;
+		return false;
+	}
+
+ public:
+
+	SQLQuery(Server* S) : Srv(S)
+	{
+	}
+
+	SQLQuery(Module* a, unsigned long b, Server* S) : dbid(b), parent(a), Srv(S)
+	{
+		// Make a few useful variables..
+		SQLModule = Srv->FindModule("m_sql.so");
+	}
+
+	~SQLQuery()
+	{
+	}
+
+	bool Query(std::string qry)
+	{
+		query = new SQLRequest(SQL_RESULT, dbid, qry);
+		return MakeQueryGoNow(qry);
+	}
+
+	bool QueryCount(std::string qry)
+	{
+		query = new SQLRequest(SQL_COUNT, dbid, qry);
+		return MakeQueryGoNow(qry);
+	}
+
+	bool GetRow()
+	{
+		rowresult = (SQLResult*)rowquery->Send();
+		if (rowresult->GetType() == SQL_ROW)
+		{
+			// We have got a row.. thats all for now.
+			return true;
+		}
+		// No Row, Error, or end. KILL CALLER! *BANG*
+		return false;
+	}
+
+	std::string GetField(std::string fname)
+	{
+		return rowresult->GetField(fname);
+	}
+
+	long GetCount()
+	{
+		rowresult = (SQLResult*)rowquery->Send();
+		if (rowresult->GetType() == SQL_COUNT)
+		{
+			return rowresult->GetCount();
+		}
+		else
+		{
+			return 0;
+		}
+	}
+
+	const std::string &GetError()
+	{
+		return error;
+	}
+
+	void SQLDone()
+	{
+		// Tell m_sql we are finished..
+		query->SetQueryType(SQL_DONE);
+		query->SetConnID(dbid);
+		Request donerequest((char*)query, parent, SQLModule);
+		donerequest.Send();
+
+		// Do Some Clearing up.
+		delete query;
+		delete rowrequest;
+		// Null the variables, so they can be re-used without confusion..
+		result = NULL;
+		query = NULL;
+		rowrequest = NULL;
+		rowresult = NULL;
+	}
+
+	static std::string Sanitise(const std::string& crap)
+	{
+		std::string temp = "";
+		for (unsigned int q = 0; q < crap.length(); q++)
+ 		{
+			if (crap[q] == '\'')
+			{
+				temp += "\\'";
+			}
+			else if (crap[q] == '"')
+			{
+				temp += "\\\"";
+			}
+			else if (crap[q] == '\\')
+			{
+				temp += "\\\\";
+			}
+			else
+			{
+				temp += crap[q];
+			}
+		}
+		return temp;
+	}
+};
+
+
 #endif