Added command StringParser.

twitchio/ext/commands/stringparser.py

@@ -0,0 +1,69 @@
+# -*- coding: utf-8 -*-
+
+"""
+The MIT License (MIT)
+
+Copyright (c) 2017-2019 TwitchIO
+
+Permission is hereby granted, free of charge, to any person obtaining a
+copy of this software and associated documentation files (the "Software"),
+to deal in the Software without restriction, including without limitation
+the rights to use, copy, modify, merge, publish, distribute, sublicense,
+and/or sell copies of the Software, and to permit persons to whom the
+Software is furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in
+all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
+OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
+FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
+DEALINGS IN THE SOFTWARE.
+"""
+
+from collections import OrderedDict
+
+
+class StringParser:
+    def __init__(self):
+        self.count = 0
+        self.index = 0
+        self.eof = 0
+        self.start = 0
+        self.words = OrderedDict()
+        self.ignore = False
+
+    def process_string(self, msg: str):
+
+        while True:
+            try:
+                loc = msg[self.count]
+            except IndexError:
+                self.eof = self.count
+                word = msg[self.start:self.eof]
+                if not word:
+                    break
+                self.words[self.index] = msg[self.start:self.eof]
+                break
+
+            if loc.isspace() and not self.ignore:
+                self.words[self.index] = msg[self.start:self.count].replace(' ', '', 1)
+                self.index += 1
+                self.start = self.count + 1
+
+            elif loc == '"':
+                if not self.ignore:
+                    self.start = self.count + 1
+                    self.ignore = True
+                else:
+                    self.words[self.index] = msg[self.start:self.count]
+                    self.index += 1
+                    self.count += 1
+                    self.start = self.count
+                    self.ignore = False
+
+            self.count += 1
+        return self.words