add deleteFriend(), modifyFriendDisplayName(), modifyPassword() and

modifyUserProfile()

Author: rayfung

db.cpp

@@ -118,3 +118,56 @@ bool DB::makeFriend(quint32 uid, quint32 friendID, QString displayName)
     q.addBindValue(displayName);
     return q.exec();
 }
+
+bool DB::removeFriend(quint32 uid, quint32 friendID)
+{
+    prepare();
+
+    QSqlQuery q;
+
+    q.prepare("delete from tbl_friend where user_id = ? and friend_id = ?");
+    q.addBindValue(uid);
+    q.addBindValue(friendID);
+    return q.exec();
+}
+
+bool DB::modifyFriendDisplayName(quint32 uid, quint32 friendID, QString displayName)
+{
+    prepare();
+
+    QSqlQuery q;
+
+    q.prepare("update tbl_friend set display_name = ? "
+              "where user_id = ? and friend_id = ?");
+    q.addBindValue(displayName);
+    q.addBindValue(uid);
+    q.addBindValue(friendID);
+    return q.exec();
+}
+
+bool DB::modifyPassword(quint32 uid, QString password)
+{
+    prepare();
+
+    QSqlQuery q;
+
+    q.prepare("update tbl_user set password = ? where id = ?");
+    q.addBindValue(password);
+    q.addBindValue(uid);
+    return q.exec();
+}
+
+bool DB::modifyUserProfile(User user)
+{
+    prepare();
+
+    QSqlQuery q;
+
+    q.prepare("update tbl_user set nickname = ?, gender = ?, address = ? "
+              "where id = ?");
+    q.addBindValue(user.nickname);
+    q.addBindValue(user.gender);
+    q.addBindValue(user.address);
+    q.addBindValue(user.id);
+    return q.exec();
+}

db.h

@@ -25,6 +25,10 @@ class DB
     bool getFriendList(quint32 id, QList<User> &userList, QList<QString> &displayNameList);
     QString getFriendDisplayName(quint32 uid, quint32 friendID);
     bool makeFriend(quint32 uid, quint32 friendID, QString displayName);
+    bool removeFriend(quint32 uid, quint32 friendID);
+    bool modifyFriendDisplayName(quint32 uid, quint32 friendID, QString displayName);
+    bool modifyPassword(quint32 uid, QString password);
+    bool modifyUserProfile(User user);
 
 private:
     void prepare();

server.cpp

@@ -142,11 +142,23 @@ bool Server::process(Connection &conn, QDataStream &in, QDataStream &out)
     case 0x04:
         return getUserProfile(conn, in, out);
 
+    case 0x05:
+        return deleteFriend(conn, in, out);
+
     case 0x06:
         return addFriend(conn, in, out);
 
     case 0x07:
         return getFriendIPAndPort(conn, in, out);
+
+    case 0xE0:
+        return modifyFriendDisplayName(conn, in, out);
+
+    case 0xE1:
+        return modifyPassword(conn, in, out);
+
+    case 0xE2:
+        return modifyUserProfile(conn, in, out);
     }
     return true;
 }
@@ -292,6 +304,45 @@ bool Server::getFriendIPAndPort(Connection &conn, QDataStream &in, QDataStream &
     return true;
 }
 
+bool Server::deleteFriend(Connection &conn, QDataStream &in, QDataStream &out)
+{
+    quint32 uid;
+
+    in >> uid;
+    out << db.removeFriend(conn.account, uid);
+    return true;
+}
+
+bool Server::modifyFriendDisplayName(Connection &conn, QDataStream &in, QDataStream &out)
+{
+    quint32 uid;
+    QString displayName;
+
+    in >> uid >> displayName;
+    out << db.modifyFriendDisplayName(conn.account, uid, displayName);
+    return true;
+}
+
+bool Server::modifyPassword(Connection &conn, QDataStream &in, QDataStream &out)
+{
+    QString password;
+
+    in >> password;
+    password = QCryptographicHash::hash(password.toUtf8(), QCryptographicHash::Sha1).toHex();
+    out << db.modifyPassword(conn.account, password);
+    return true;
+}
+
+bool Server::modifyUserProfile(Connection &conn, QDataStream &in, QDataStream &out)
+{
+    User user;
+
+    in >> user.nickname >> user.gender >> user.address;
+    user.id = conn.account;
+    out << db.modifyUserProfile(user);
+    return true;
+}
+
 void Server::clientDisconnected()
 {
     QTcpSocket *socket;

server.h

@@ -49,6 +49,10 @@ class Server : public QObject
     bool getUserProfile(Connection &conn, QDataStream &in, QDataStream &out);
     bool addFriend(Connection &conn, QDataStream &in, QDataStream &out);
     bool getFriendIPAndPort(Connection &conn, QDataStream &in, QDataStream &out);
+    bool deleteFriend(Connection &conn, QDataStream &in, QDataStream &out);
+    bool modifyFriendDisplayName(Connection &conn, QDataStream &in, QDataStream &out);
+    bool modifyPassword(Connection &conn, QDataStream &in, QDataStream &out);
+    bool modifyUserProfile(Connection &conn, QDataStream &in, QDataStream &out);
 
 private:
     Config config;