SocketStringReader.swift 2.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869
  1. //
  2. // SocketStringReader.swift
  3. // Socket.IO-Client-Swift
  4. //
  5. // Created by Lukas Schmidt on 07.09.15.
  6. //
  7. // Permission is hereby granted, free of charge, to any person obtaining a copy
  8. // of this software and associated documentation files (the "Software"), to deal
  9. // in the Software without restriction, including without limitation the rights
  10. // to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  11. // copies of the Software, and to permit persons to whom the Software is
  12. // furnished to do so, subject to the following conditions:
  13. //
  14. // The above copyright notice and this permission notice shall be included in
  15. // all copies or substantial portions of the Software.
  16. //
  17. // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  18. // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  19. // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  20. // AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  21. // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  22. // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
  23. // THE SOFTWARE.
  24. struct SocketStringReader {
  25. let message: String
  26. var currentIndex: String.Index
  27. var hasNext: Bool {
  28. return currentIndex != message.endIndex
  29. }
  30. var currentCharacter: String {
  31. return String(message[currentIndex])
  32. }
  33. init(message: String) {
  34. self.message = message
  35. currentIndex = message.startIndex
  36. }
  37. mutating func advanceIndexBy(n: Int) {
  38. currentIndex = currentIndex.advancedBy(n)
  39. }
  40. mutating func read(readLength: Int) -> String {
  41. let readString = message[currentIndex..<currentIndex.advancedBy(readLength)]
  42. advanceIndexBy(readLength)
  43. return readString
  44. }
  45. mutating func readUntilStringOccurence(string: String) -> String {
  46. let substring = message[currentIndex..<message.endIndex]
  47. guard let foundRange = substring.rangeOfString(string) else {
  48. currentIndex = message.endIndex
  49. return substring
  50. }
  51. advanceIndexBy(message.startIndex.distanceTo(foundRange.startIndex) + 1)
  52. return substring.substringToIndex(foundRange.startIndex)
  53. }
  54. mutating func readUntilEnd() -> String {
  55. return read(currentIndex.distanceTo(message.endIndex))
  56. }
  57. }