X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Ffreenet%2Fidentityinserter.cpp;h=f246d66402989d8bb6c75e1bba1fa51bc1786b30;hb=853f67b0b7b8121d572cff34d40f7b28cac8f65e;hp=4fac82f0c2d4c96583d6cd6d52da3dc781c5da73;hpb=9048d8e482c91960265f29c2b5b3112f2a52f3d8;p=fms.git

diff --git a/src/freenet/identityinserter.cpp b/src/freenet/identityinserter.cpp
index 4fac82f..f246d66 100644
--- a/src/freenet/identityinserter.cpp
+++ b/src/freenet/identityinserter.cpp
@@ -80,16 +80,27 @@ const bool IdentityInserter::HandleMessage(FCPMessage &message)
 
 		if(message.GetName()=="PutSuccessful")
 		{
-			m_db->Execute("UPDATE tblLocalIdentity SET InsertingIdentity='false', LastInsertedIdentity='"+now.Format("%Y-%m-%d %H:%M:%S")+"' WHERE LocalIdentityID="+idparts[1]+";");
+			// a little hack here - if we just inserted index yesterday and it is now the next day - we would have inserted todays date not yesterdays as LastInsertedIdentity.
+			// If this is the case, we will skip updating LastInsertedIdentity so that we can insert this identity again for today
+			DateTime lastdate;
+			lastdate.Set(idparts[4]);
+			if(lastdate.GetDay()==now.GetDay())
+			{
+				m_db->Execute("UPDATE tblLocalIdentity SET InsertingIdentity='false', LastInsertedIdentity='"+now.Format("%Y-%m-%d %H:%M:%S")+"' WHERE LocalIdentityID="+idparts[1]+";");
+			}
+			else
+			{
+				m_db->Execute("UPDATE tblLocalIdentity SET InsertingIdentity='false' WHERE LocalIdentityID="+idparts[1]+";");
+			}
 			m_db->Execute("INSERT INTO tblLocalIdentityInserts(LocalIdentityID,Day,InsertIndex) VALUES("+idparts[1]+",'"+idparts[4]+"',"+idparts[2]+");");
-			m_log->WriteLog(LogFile::LOGLEVEL_DEBUG,__FUNCTION__" inserted Identity xml");
+			m_log->WriteLog(LogFile::LOGLEVEL_DEBUG,"IdentityInserter::HandleMessage inserted Identity xml");
 			return true;
 		}
 
 		if(message.GetName()=="PutFailed")
 		{
 			m_db->Execute("UPDATE tblLocalIdentity SET InsertingIdentity='false' WHERE LocalIdentityID="+idparts[1]+";");
-			m_log->WriteLog(LogFile::LOGLEVEL_DEBUG,__FUNCTION__" failure inserting identity xml.  Code="+message["Code"]+" Description="+message["CodeDescription"]);
+			m_log->WriteLog(LogFile::LOGLEVEL_DEBUG,"IdentityInserter::HandleMessage failure inserting Identity xml.  Code="+message["Code"]+" Description="+message["CodeDescription"]);
 			
 			// if code 9 (collision), then insert index into inserted table
 			if(message["Code"]=="9")
@@ -139,7 +150,7 @@ void IdentityInserter::StartInsert(const long localidentityid)
 	StringFunctions::Convert(localidentityid,idstring);
 	date.SetToGMTime();
 
-	SQLite3DB::Recordset rs=m_db->Query("SELECT Name,PrivateKey,SingleUse,PublishTrustList,PublishBoardList FROM tblLocalIdentity WHERE LocalIdentityID="+idstring+";");
+	SQLite3DB::Recordset rs=m_db->Query("SELECT Name,PrivateKey,SingleUse,PublishTrustList,PublishBoardList,PublishFreesite,FreesiteEdition FROM tblLocalIdentity WHERE LocalIdentityID="+idstring+";");
 
 	if(rs.Empty()==false)
 	{
@@ -155,6 +166,8 @@ void IdentityInserter::StartInsert(const long localidentityid)
 		std::string singleuse="false";
 		std::string publishtrustlist="false";
 		std::string publishboardlist="false";
+		std::string freesiteedition="";
+		int edition=-1;
 
 		now.SetToGMTime();
 
@@ -172,7 +185,7 @@ void IdentityInserter::StartInsert(const long localidentityid)
 		}
 		StringFunctions::Convert(index,indexstr);
 
-		Option::instance()->Get("MessageBase",messagebase);
+		Option::Instance()->Get("MessageBase",messagebase);
 
 		if(rs.GetField(0))
 		{
@@ -202,6 +215,16 @@ void IdentityInserter::StartInsert(const long localidentityid)
 		}
 		publishboardlist=="true" ? idxml.SetPublishBoardList(true) : idxml.SetPublishBoardList(false);
 
+		if(rs.GetField(5) && rs.GetField(6))
+		{
+			if(std::string(rs.GetField(5))=="true")
+			{
+				freesiteedition=rs.GetField(6);
+				StringFunctions::Convert(freesiteedition,edition);
+				idxml.SetFreesiteEdition(edition);
+			}
+		}
+
 		data=idxml.GetXML();
 		StringFunctions::Convert(data.size(),datasizestr);