summaryrefslogtreecommitdiff
path: root/src
diff options
context:
space:
mode:
Diffstat (limited to 'src')
-rw-r--r--src/irc/irc_message.cpp39
-rw-r--r--src/irc/irc_message.hpp4
2 files changed, 21 insertions, 22 deletions
diff --git a/src/irc/irc_message.cpp b/src/irc/irc_message.cpp
index 14fdb0e..62fe9a7 100644
--- a/src/irc/irc_message.cpp
+++ b/src/irc/irc_message.cpp
@@ -1,33 +1,30 @@
#include <irc/irc_message.hpp>
#include <iostream>
-IrcMessage::IrcMessage(std::string&& line)
+IrcMessage::IrcMessage(std::stringstream ss)
{
- std::string::size_type pos;
-
- // optional prefix
- if (line[0] == ':')
+ if (ss.peek() == ':')
{
- pos = line.find(' ');
- this->prefix = line.substr(1, pos - 1);
- line = line.substr(pos + 1, std::string::npos);
+ ss.ignore();
+ ss >> this->prefix;
}
- // command
- pos = line.find(' ');
- this->command = line.substr(0, pos);
- line = line.substr(pos + 1, std::string::npos);
- // arguments
- do
+ ss >> this->command;
+ while (ss >> std::ws)
{
- if (line[0] == ':')
+ std::string arg;
+ if (ss.peek() == ':')
+ {
+ ss.ignore();
+ std::getline(ss, arg);
+ }
+ else
{
- this->arguments.emplace_back(line.substr(1, std::string::npos));
- break ;
+ ss >> arg;
+ if (arg.empty())
+ break;
}
- pos = line.find(' ');
- this->arguments.emplace_back(line.substr(0, pos));
- line = line.substr(pos + 1, std::string::npos);
- } while (pos != std::string::npos);
+ this->arguments.push_back(std::move(arg));
+ }
}
IrcMessage::IrcMessage(std::string&& prefix,
diff --git a/src/irc/irc_message.hpp b/src/irc/irc_message.hpp
index 269a12a..5475fd2 100644
--- a/src/irc/irc_message.hpp
+++ b/src/irc/irc_message.hpp
@@ -4,11 +4,13 @@
#include <vector>
#include <string>
#include <ostream>
+#include <sstream>
class IrcMessage
{
public:
- IrcMessage(std::string&& line);
+ IrcMessage(std::stringstream ss);
+ IrcMessage(std::string str): IrcMessage{std::stringstream{str}} {}
IrcMessage(std::string&& prefix, std::string&& command, std::vector<std::string>&& args);
IrcMessage(std::string&& command, std::vector<std::string>&& args);
~IrcMessage() = default;