Added distinction between CTCP requests and CTCP replies

2 files changed, 24 insertions, 6 deletions

diff --git a/noncore/net/opieirc/ircmessage.cpp b/noncore/net/opieirc/ircmessage.cpp
index d19e9e6..1b7072e 100644
--- a/noncore/net/opieirc/ircmessage.cpp
+++ b/noncore/net/opieirc/ircmessage.cpp
@@ -31,66 +31,73 @@ IRCMessage::IRCMessage(QString line) {
         m_allParameters = line.right(line.length() - m_prefix.length() - m_command.length() - 3);
     } else {
         m_command = temp.upper();
         m_allParameters = line.right(line.length() - m_command.length() - 1);
     }
     
     /* Create a list of all parameters */
     while(!(stream.atEnd())) {
         stream >> temp;
         if (temp.startsWith(":")) {
             /* last parameter */
             m_trailing = line.right(line.length() - line.find(QChar(':'), 1) - 1);
             m_parameters << m_trailing;
             break;
         } else {
             m_parameters << temp;
         }
     }
 
     
     m_commandNumber = m_command.toInt(&m_isNumerical);
     /* Is this a CTCP command */
     if ((m_command == "PRIVMSG" || m_command == "NOTICE") && m_trailing.length()>0 && m_trailing.left(1) == QChar(1)) {
         m_ctcp = TRUE;
+        if (m_command == "PRIVMSG") {
+	    m_ctcpRequest = TRUE;
+        }
+        else {
+            m_ctcpRequest = FALSE;
+        }
+        
         /* Strip CTCP \001 characters */
         m_allParameters = m_allParameters.replace(QRegExp(QChar(1)), "");
         QTextIStream ctcpStream(&m_allParameters);
-        if (m_command == "PRIVMSG")
-            ctcpStream >> m_ctcpDestination;
+        ctcpStream >> m_ctcpDestination;
         ctcpStream >> temp;
         m_ctcpCommand = temp.upper().right(temp.length()-1);
         m_parameters.clear();
         int length = m_allParameters.length() - m_ctcpCommand.length() - 1;
-        if (m_command == "PRIVMSG")
-            length -= m_ctcpDestination.length() + 1;
+        length -= m_ctcpDestination.length() + 1;
         if (length <= 0) {
             m_allParameters = "";
-        } else {
+        } 
+        else {
             m_allParameters = m_allParameters.right(length);
             m_parameters << m_allParameters;
         }
-    } else {
+    } 
+    else {
         m_ctcp = FALSE;
     }
 
     
     odebug << "Parsed: " << line << oendl;
     odebug << "Prefix: " << m_prefix << oendl;
     odebug << "Command: " << m_command << oendl;
     odebug << "Allparameters: " << m_allParameters << oendl;
     for (unsigned int i=0; i<m_parameters.count(); i++) {
         odebug << "Parameter " << i << ":" << m_parameters[i] << oendl;
     }
     odebug << "CTCP Command: " << m_ctcpCommand << oendl;
     odebug << "CTCP Destination: " << m_ctcpDestination << oendl;
     odebug << "CTCP param  count is: " << m_parameters.count() << oendl;
     
 }
 
 QString IRCMessage::param(int param) {
     return m_parameters[param];
 }
 
 QStringList IRCMessage::params(const QString &paramstring) const {
     QStringList params, retvalue;
     params = QStringList::split(',', paramstring);
@@ -123,32 +130,40 @@ QString IRCMessage::prefix() {
 QString IRCMessage::command() {
     return m_command;
 }
 
 QString IRCMessage::ctcpCommand() {
     return m_ctcpCommand;
 }
 
 QString IRCMessage::ctcpDestination() {
     return m_ctcpDestination;
 }
 
 unsigned short IRCMessage::commandNumber() {
     return m_commandNumber;
 }
 
 bool IRCMessage::isNumerical() {
     return m_isNumerical;
 }
 
 bool IRCMessage::isCTCP() {
     return m_ctcp;
 }
 
+bool IRCMessage::isCTCPRequest() {
+    return m_ctcpRequest;
+}
+
+bool IRCMessage::isCTCPReply() {
+    return !m_ctcpRequest;
+}
+
 QString IRCMessage::trailing() {
     return m_trailing;
 }
 
 QString IRCMessage::allParameters() {
     return m_allParameters;
 }
 
diff --git a/noncore/net/opieirc/ircmessage.h b/noncore/net/opieirc/ircmessage.h
index 10ba450..40bb210 100644
--- a/noncore/net/opieirc/ircmessage.h
+++ b/noncore/net/opieirc/ircmessage.h
@@ -17,54 +17,57 @@
     Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
 */
 
 #ifndef __IRCMESSAGE_H
 #define __IRCMESSAGE_H
 
 class QString;
 class QStringList;
 
 /* IRCMessage objects are used to encapsulate information
    which the IRC server sent to us. */
 
 class IRCMessage {
 public:
     /* Parse an IRC message and create the IRCMessage object */
     IRCMessage(QString line);
 
     /* Return the IRC message prefix (usually sender etc) */
     QString prefix();
     /* Check if this IRCMessage's command is literal or numerical */
     bool isNumerical();
     /* CHeck if this IRCMessage is a CTCP message */
     bool isCTCP();
+    bool isCTCPRequest();
+    bool isCTCPReply();
     /* Return the IRC command (literal commands) */
     QString command();
     /* Return the CTCP command */
     QString ctcpCommand();
     /* Return the CTCP destination if applicable (channel/person) */
     QString ctcpDestination();
     /* Return the IRC command (numerical commands) */
     unsigned short commandNumber();
     /* Return the trailing parameter string */
     QString trailing();
     /* Return the complete parameter string */
     QString allParameters();
     /* Return one parameter */ 
     QString param(int param);
     /* Return some parameters */
     QStringList params(const QString &paramstring) const;
 protected:
     QString         m_prefix;
     QString         m_command;
     QString         m_ctcpCommand;
     QString         m_ctcpDestination;
     unsigned short  m_commandNumber;
     QString         m_allParameters;
     QString         m_trailing;
     QStringList     m_parameters;
     bool            m_isNumerical;
     bool            m_ctcp;
+    bool            m_ctcpRequest;
 };
 
 #endif