+ while (query.next()) {
+ CoreIdentity identity(IdentityId(query.value(0).toInt()));
+
+ identity.setIdentityName(query.value(1).toString());
+ identity.setRealName(query.value(2).toString());
+ identity.setAwayNick(query.value(3).toString());
+ identity.setAwayNickEnabled(!!query.value(4).toInt());
+ identity.setAwayReason(query.value(5).toString());
+ identity.setAwayReasonEnabled(!!query.value(6).toInt());
+ identity.setAutoAwayEnabled(!!query.value(7).toInt());
+ identity.setAutoAwayTime(query.value(8).toInt());
+ identity.setAutoAwayReason(query.value(9).toString());
+ identity.setAutoAwayReasonEnabled(!!query.value(10).toInt());
+ identity.setDetachAwayEnabled(!!query.value(11).toInt());
+ identity.setDetachAwayReason(query.value(12).toString());
+ identity.setDetachAwayReasonEnabled(!!query.value(13).toInt());
+ identity.setIdent(query.value(14).toString());
+ identity.setKickReason(query.value(15).toString());
+ identity.setPartReason(query.value(16).toString());
+ identity.setQuitReason(query.value(17).toString());
+#ifdef HAVE_SSL
+ identity.setSslCert(query.value(18).toByteArray());
+ identity.setSslKey(query.value(19).toByteArray());
+#endif
+
+ nickQuery.bindValue(":identityid", identity.id().toInt());
+ QList<QString> nicks;
+ safeExec(nickQuery);
+ watchQuery(nickQuery);
+ while (nickQuery.next()) {
+ nicks << nickQuery.value(0).toString();
+ }
+ identity.setNicks(nicks);
+ identities << identity;
+ }
+ db.commit();
+ return identities;
+}
+
+
+NetworkId PostgreSqlStorage::createNetwork(UserId user, const NetworkInfo &info)
+{
+ NetworkId networkId;
+
+ QSqlDatabase db = logDb();
+ if (!beginTransaction(db)) {
+ qWarning() << "PostgreSqlStorage::createNetwork(): failed to begin transaction!";
+ qWarning() << " -" << qPrintable(db.lastError().text());
+ return false;
+ }
+
+ QSqlQuery query(db);
+ query.prepare(queryString("insert_network"));
+ query.bindValue(":userid", user.toInt());
+ bindNetworkInfo(query, info);
+ safeExec(query);
+ if (!watchQuery(query)) {
+ db.rollback();
+ return NetworkId();
+ }
+
+ query.first();
+ networkId = query.value(0).toInt();
+
+ if (!networkId.isValid()) {
+ db.rollback();
+ return NetworkId();
+ }
+
+ QSqlQuery insertServersQuery(db);
+ insertServersQuery.prepare(queryString("insert_server"));
+ foreach(Network::Server server, info.serverList) {
+ insertServersQuery.bindValue(":userid", user.toInt());
+ insertServersQuery.bindValue(":networkid", networkId.toInt());
+ bindServerInfo(insertServersQuery, server);
+ safeExec(insertServersQuery);
+ if (!watchQuery(insertServersQuery)) {
+ db.rollback();
+ return NetworkId();
+ }
+ }
+
+ if (!db.commit()) {
+ qWarning() << "PostgreSqlStorage::createNetwork(): committing data failed!";
+ qWarning() << " -" << qPrintable(db.lastError().text());
+ return NetworkId();
+ }
+ return networkId;
+}
+
+
+void PostgreSqlStorage::bindNetworkInfo(QSqlQuery &query, const NetworkInfo &info)
+{
+ query.bindValue(":networkname", info.networkName);
+ query.bindValue(":identityid", info.identity.isValid() ? info.identity.toInt() : QVariant());
+ query.bindValue(":encodingcodec", QString(info.codecForEncoding));
+ query.bindValue(":decodingcodec", QString(info.codecForDecoding));
+ query.bindValue(":servercodec", QString(info.codecForServer));
+ query.bindValue(":userandomserver", info.useRandomServer);
+ query.bindValue(":perform", info.perform.join("\n"));
+ query.bindValue(":useautoidentify", info.useAutoIdentify);
+ query.bindValue(":autoidentifyservice", info.autoIdentifyService);
+ query.bindValue(":autoidentifypassword", info.autoIdentifyPassword);
+ query.bindValue(":usesasl", info.useSasl);
+ query.bindValue(":saslaccount", info.saslAccount);
+ query.bindValue(":saslpassword", info.saslPassword);
+ query.bindValue(":useautoreconnect", info.useAutoReconnect);
+ query.bindValue(":autoreconnectinterval", info.autoReconnectInterval);
+ query.bindValue(":autoreconnectretries", info.autoReconnectRetries);
+ query.bindValue(":unlimitedconnectretries", info.unlimitedReconnectRetries);
+ query.bindValue(":rejoinchannels", info.rejoinChannels);
+ if (info.networkId.isValid())
+ query.bindValue(":networkid", info.networkId.toInt());
+}
+
+
+void PostgreSqlStorage::bindServerInfo(QSqlQuery &query, const Network::Server &server)
+{
+ query.bindValue(":hostname", server.host);
+ query.bindValue(":port", server.port);
+ query.bindValue(":password", server.password);
+ query.bindValue(":ssl", server.useSsl);
+ query.bindValue(":sslversion", server.sslVersion);
+ query.bindValue(":useproxy", server.useProxy);
+ query.bindValue(":proxytype", server.proxyType);
+ query.bindValue(":proxyhost", server.proxyHost);
+ query.bindValue(":proxyport", server.proxyPort);
+ query.bindValue(":proxyuser", server.proxyUser);
+ query.bindValue(":proxypass", server.proxyPass);
+}
+
+
+bool PostgreSqlStorage::updateNetwork(UserId user, const NetworkInfo &info)
+{
+ QSqlDatabase db = logDb();
+ if (!beginTransaction(db)) {
+ qWarning() << "PostgreSqlStorage::updateNetwork(): failed to begin transaction!";
+ qWarning() << " -" << qPrintable(db.lastError().text());
+ return false;
+ }
+
+ QSqlQuery updateQuery(db);
+ updateQuery.prepare(queryString("update_network"));
+ updateQuery.bindValue(":userid", user.toInt());
+ bindNetworkInfo(updateQuery, info);
+ safeExec(updateQuery);
+ if (!watchQuery(updateQuery)) {
+ db.rollback();
+ return false;
+ }
+ if (updateQuery.numRowsAffected() != 1) {
+ // seems this is not our network...
+ db.rollback();
+ return false;
+ }
+
+ QSqlQuery dropServersQuery(db);
+ dropServersQuery.prepare("DELETE FROM ircserver WHERE networkid = :networkid");
+ dropServersQuery.bindValue(":networkid", info.networkId.toInt());
+ safeExec(dropServersQuery);
+ if (!watchQuery(dropServersQuery)) {
+ db.rollback();
+ return false;
+ }
+
+ QSqlQuery insertServersQuery(db);
+ insertServersQuery.prepare(queryString("insert_server"));
+ foreach(Network::Server server, info.serverList) {
+ insertServersQuery.bindValue(":userid", user.toInt());
+ insertServersQuery.bindValue(":networkid", info.networkId.toInt());
+ bindServerInfo(insertServersQuery, server);
+ safeExec(insertServersQuery);
+ if (!watchQuery(insertServersQuery)) {
+ db.rollback();
+ return false;
+ }
+ }
+
+ if (!db.commit()) {
+ qWarning() << "PostgreSqlStorage::updateNetwork(): committing data failed!";
+ qWarning() << " -" << qPrintable(db.lastError().text());
+ return false;
+ }
+ return true;
+}
+
+
+bool PostgreSqlStorage::removeNetwork(UserId user, const NetworkId &networkId)
+{
+ QSqlDatabase db = logDb();
+ if (!beginTransaction(db)) {
+ qWarning() << "PostgreSqlStorage::removeNetwork(): cannot start transaction!";
+ qWarning() << " -" << qPrintable(db.lastError().text());
+ return false;
+ }
+
+ QSqlQuery query(db);
+ query.prepare(queryString("delete_network"));
+ query.bindValue(":userid", user.toInt());
+ query.bindValue(":networkid", networkId.toInt());
+ safeExec(query);
+ if (!watchQuery(query)) {
+ db.rollback();
+ return false;
+ }
+
+ db.commit();
+ return true;
+}