summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
Diffstat (limited to 'Swiften')
-rw-r--r--Swiften/History/HistoryManager.h2
-rw-r--r--Swiften/History/HistoryMessage.h21
-rw-r--r--Swiften/History/SQLiteHistoryManager.cpp70
-rw-r--r--Swiften/History/SQLiteHistoryManager.h2
4 files changed, 58 insertions, 37 deletions
diff --git a/Swiften/History/HistoryManager.h b/Swiften/History/HistoryManager.h
index 151f8be..1c0d784 100644
--- a/Swiften/History/HistoryManager.h
+++ b/Swiften/History/HistoryManager.h
@@ -17,7 +17,7 @@ namespace Swift {
virtual void addMessage(const HistoryMessage& message) = 0;
- virtual std::vector<HistoryMessage> getMessages(const JID&) const = 0;
+ virtual std::vector<HistoryMessage> getMessages(const JID& selfJID, const JID& contactJID) const = 0;
virtual std::vector<JID> getAllContacts() const = 0;
};
}
diff --git a/Swiften/History/HistoryMessage.h b/Swiften/History/HistoryMessage.h
index 03e9206..0bcff99 100644
--- a/Swiften/History/HistoryMessage.h
+++ b/Swiften/History/HistoryMessage.h
@@ -14,14 +14,12 @@ namespace Swift {
public:
HistoryMessage(
const std::string& message,
- const JID& baseJID,
const JID& fromJID,
- const std::string& displayNick,
- const boost::posix_time::ptime time) :
+ const JID& toJID,
+ const boost::posix_time::ptime& time) :
message_(message),
- baseJID_(baseJID),
fromJID_(fromJID),
- displayNick_(displayNick),
+ toJID_(toJID),
time_(time) {
}
@@ -29,16 +27,12 @@ namespace Swift {
return message_;
}
- const JID& getBaseJID() const {
- return baseJID_;
- }
-
const JID& getFromJID() const {
return fromJID_;
}
- const std::string& getDisplayNick() const {
- return displayNick_;
+ const JID& getToJID() const {
+ return toJID_;
}
boost::posix_time::ptime getTime() const {
@@ -46,14 +40,13 @@ namespace Swift {
}
bool operator==(const HistoryMessage& o) const {
- return baseJID_ == o.baseJID_ && message_ == o.message_ && fromJID_ == o.fromJID_ && displayNick_ == o.displayNick_ && time_ == o.time_;
+ return message_ == o.message_ && fromJID_ == o.fromJID_ && toJID_ == o.toJID_ && time_ == o.time_;
}
private:
std::string message_;
- JID baseJID_;
JID fromJID_;
- std::string displayNick_;
+ JID toJID_;
boost::posix_time::ptime time_;
};
}
diff --git a/Swiften/History/SQLiteHistoryManager.cpp b/Swiften/History/SQLiteHistoryManager.cpp
index 5a5c832..9f02ca2 100644
--- a/Swiften/History/SQLiteHistoryManager.cpp
+++ b/Swiften/History/SQLiteHistoryManager.cpp
@@ -9,7 +9,6 @@
#include <sqlite3.h>
#include <Swiften/History/SQLiteHistoryManager.h>
-#include <sstream>
inline std::string getEscapedString(const std::string& s) {
std::string result(s);
@@ -31,7 +30,7 @@ SQLiteHistoryManager::SQLiteHistoryManager(const std::string& file) : db_(0) {
}
char* errorMessage;
- int result = sqlite3_exec(db_, "CREATE TABLE IF NOT EXISTS messages('baseID' INTEGER, 'from' INTEGER, 'nick' STRING, 'message' STRING, 'time' INTEGER)", 0, 0, &errorMessage);
+ int result = sqlite3_exec(db_, "CREATE TABLE IF NOT EXISTS messages('message' STRING, 'fromBare' INTEGER, 'fromResource' STRING, 'toBare' INTEGER, 'toResource' STRING, 'time' INTEGER)", 0, 0, &errorMessage);
if (result != SQLITE_OK) {
std::cerr << "SQL Error: " << errorMessage << std::endl;
sqlite3_free(errorMessage);
@@ -43,6 +42,7 @@ SQLiteHistoryManager::SQLiteHistoryManager(const std::string& file) : db_(0) {
sqlite3_free(errorMessage);
}
+ // FIXME: remove
result = sqlite3_exec(db_, "CREATE TABLE IF NOT EXISTS conversations('id' INTEGER UNIQUE NOT NULL, 'time' INTEGER)", 0, 0, &errorMessage);
if (result != SQLITE_OK) {
std::cerr << "SQL Error: " << errorMessage << std::endl;
@@ -55,14 +55,16 @@ SQLiteHistoryManager::~SQLiteHistoryManager() {
}
void SQLiteHistoryManager::addMessage(const HistoryMessage& message) {
- updateConversation(message);
+ // updateConversation(message);
int secondsSinceEpoch = (message.getTime() - boost::posix_time::ptime(boost::gregorian::date(1970, 1, 1))).total_seconds();
- std::string statement = std::string("INSERT INTO messages('baseID', 'from', 'nick', 'message', 'time') VALUES(") +
- boost::lexical_cast<std::string>(getIDForJID(message.getBaseJID())) + ", " +
- boost::lexical_cast<std::string>(getIDForJID(message.getFromJID())) + ", '" +
- message.getDisplayNick() + "', '" +
- getEscapedString(message.getMessage()) + "', " +
+
+ std::string statement = std::string("'message', 'fromBare', 'fromResource', 'toBare', 'toResource', 'time') VALUES(") +
+ "'" + getEscapedString(message.getMessage()) + "', " +
+ boost::lexical_cast<std::string>(getIDForJID(message.getFromJID().toBare())) + ", '" +
+ getEscapedString(message.getFromJID().getResource()) + "', " +
+ boost::lexical_cast<std::string>(getIDForJID(message.getFromJID().toBare())) + ", '" +
+ getEscapedString(message.getToJID().getResource()) + "', " +
boost::lexical_cast<std::string>(secondsSinceEpoch) + ")";
char* errorMessage;
int result = sqlite3_exec(db_, statement.c_str(), 0, 0, &errorMessage);
@@ -72,37 +74,60 @@ void SQLiteHistoryManager::addMessage(const HistoryMessage& message) {
}
}
-std::vector<HistoryMessage> SQLiteHistoryManager::getMessages(const JID& baseJID) const {
- std::vector<HistoryMessage> result;
+std::vector<HistoryMessage> SQLiteHistoryManager::getMessages(const JID& selfJID, const JID& contactJID) const {
sqlite3_stmt* selectStatement;
- boost::optional<int> baseID = getIDFromJID(baseJID);
- if (!baseID) {
+
+ boost::optional<int> selfID = getIDFromJID(selfJID.toBare());
+ boost::optional<int> contactID = getIDFromJID(contactJID.toBare());
+
+ if (!selfID || !contactID) {
+ // JIDs missing from the database
return std::vector<HistoryMessage>();
}
- std::stringstream selectQuery;
- selectQuery << "SELECT messages.'baseID', messages.'from', messages.'nick', messages.'message', messages.'time' FROM messages WHERE baseID=" << *baseID;
+ std::string selectQuery = "SELECT * FROM messages WHERE (fromBare=" +
+ boost::lexical_cast<std::string>(*selfID) + " AND fromBare=" +
+ boost::lexical_cast<std::string>(*contactID) + ") OR (fromBare=" +
+ boost::lexical_cast<std::string>(*contactID) + " AND fromBare=" +
+ boost::lexical_cast<std::string>(*selfID) + ")";
- int r = sqlite3_prepare(db_, selectQuery.str().c_str(), selectQuery.str().size(), &selectStatement, NULL);
+ int r = sqlite3_prepare(db_, selectQuery.c_str(), selectQuery.size(), &selectStatement, NULL);
if (r != SQLITE_OK) {
std::cout << "Error: " << sqlite3_errmsg(db_) << std::endl;
}
r = sqlite3_step(selectStatement);
+
+ // Retrieve result
+ std::vector<HistoryMessage> result;
while (r == SQLITE_ROW) {
- boost::optional<JID> baseJID(getJIDFromID(sqlite3_column_int(selectStatement, 0)));
- boost::optional<JID> from(getJIDFromID(sqlite3_column_int(selectStatement, 1)));
- std::string nick(reinterpret_cast<const char*>(sqlite3_column_text(selectStatement, 2)));
- std::string message(reinterpret_cast<const char*>(sqlite3_column_text(selectStatement, 3)));
- int secondsSinceEpoch(sqlite3_column_int(selectStatement, 4));
+ std::string message(reinterpret_cast<const char*>(sqlite3_column_text(selectStatement, 0)));
+
+ // fromJID
+ boost::optional<JID> fromJID(getJIDFromID(sqlite3_column_int(selectStatement, 1)));
+ std::string fromResource(reinterpret_cast<const char*>(sqlite3_column_text(selectStatement, 2)));
+ if (fromJID) {
+ fromJID = boost::optional<JID>(JID(fromJID->getNode(), fromJID->getDomain(), fromResource));
+ }
+
+ // toJID
+ boost::optional<JID> toJID(getJIDFromID(sqlite3_column_int(selectStatement, 3)));
+ std::string toResource(reinterpret_cast<const char*>(sqlite3_column_text(selectStatement, 4)));
+ if (toJID) {
+ toJID = boost::optional<JID>(JID(toJID->getNode(), toJID->getDomain(), toResource));
+ }
+
+ // timestamp
+ int secondsSinceEpoch(sqlite3_column_int(selectStatement, 5));
boost::posix_time::ptime time(boost::gregorian::date(1970, 1, 1), boost::posix_time::seconds(secondsSinceEpoch));
- result.push_back(HistoryMessage(message, (baseJID ? *baseJID : JID()), (from ? *from : JID()), nick, time));
+ result.push_back(HistoryMessage(message, (fromJID ? *fromJID : JID()), (toJID ? *toJID : JID()), time));
r = sqlite3_step(selectStatement);
}
if (r != SQLITE_DONE) {
std::cout << "Error: " << sqlite3_errmsg(db_) << std::endl;
}
sqlite3_finalize(selectStatement);
+
return result;
}
@@ -160,6 +185,8 @@ boost::optional<int> SQLiteHistoryManager::getIDFromJID(const JID& jid) const {
}
void SQLiteHistoryManager::updateConversation(const HistoryMessage& message) {
+
+ /*
int id = getIDForJID(message.getBaseJID());
int secondsSinceEpoch = (message.getTime() - boost::posix_time::ptime(boost::gregorian::date(1970, 1, 1))).total_seconds(); // this might fail in 2038
@@ -174,6 +201,7 @@ void SQLiteHistoryManager::updateConversation(const HistoryMessage& message) {
std::cerr << "SQL Error: " << errorMessage << std::endl;
sqlite3_free(errorMessage);
}
+ */
}
std::vector<JID> SQLiteHistoryManager::getAllContacts() const {
diff --git a/Swiften/History/SQLiteHistoryManager.h b/Swiften/History/SQLiteHistoryManager.h
index ff82f5e..9d9ec05 100644
--- a/Swiften/History/SQLiteHistoryManager.h
+++ b/Swiften/History/SQLiteHistoryManager.h
@@ -19,7 +19,7 @@ namespace Swift {
~SQLiteHistoryManager();
void addMessage(const HistoryMessage& message);
- std::vector<HistoryMessage> getMessages(const JID&) const;
+ std::vector<HistoryMessage> getMessages(const JID& selfJID, const JID& contactJID) const;
std::vector<JID> getAllContacts() const;
private: