Update to support notes + new style2.0.0

6 files changed, 182 insertions, 0 deletions

diff --git a/Map/Logic/MapParser/Strategies/BlockerParserStrategy.swift b/Map/Logic/MapParser/Strategies/BlockerParserStrategy.swift
new file mode 100644
index 0000000..d9b6f22
--- /dev/null
+++ b/Map/Logic/MapParser/Strategies/BlockerParserStrategy.swift
@@ -0,0 +1,26 @@
+import Foundation
+
+struct BlockerParserStrategy: MapParserStrategy {
+  private let regex = MapParsingPatterns.blocker
+
+  func canHandle(line: String) -> Bool {
+    let range = NSRange(location: 0, length: line.utf16.count)
+    let matches = regex.matches(in: String(line), options: [], range: range)
+    return matches.count > 0 && matches[0].numberOfRanges == 3
+  }
+
+  func handle(index: Int, line: String, vertices: [String: Vertex]) -> (Any.Type, Any) {
+    let range = NSRange(location: 0, length: line.utf16.count)
+    let matches = regex.matches(in: String(line), options: [], range: range)
+
+    let match = matches[0]
+    let vertexA = String(line[Range(match.range(at: 2), in: line)!])
+
+    if let vertex = vertices[vertexA] {
+      let blocker = Blocker(id: index, position: vertex.position)
+      return (Blocker.self, blocker)
+    }
+
+    return (NSObject.self, NSObject())  // No matching object
+  }
+}
diff --git a/Map/Logic/MapParser/Strategies/EdgeParserStrategy.swift b/Map/Logic/MapParser/Strategies/EdgeParserStrategy.swift
new file mode 100644
index 0000000..0df1ff4
--- /dev/null
+++ b/Map/Logic/MapParser/Strategies/EdgeParserStrategy.swift
@@ -0,0 +1,32 @@
+import Foundation
+
+struct EdgeParserStrategy: MapParserStrategy {
+  private let regex = MapParsingPatterns.edge
+
+  func canHandle(line: String) -> Bool {
+    let range = NSRange(location: 0, length: line.utf16.count)
+    let matches = regex.matches(in: String(line), options: [], range: range)
+    return matches.count > 0 && matches[0].numberOfRanges == 4
+  }
+
+  func handle(index: Int, line: String, vertices: [String: Vertex]) -> (Any.Type, Any) {
+    let range = NSRange(location: 0, length: line.utf16.count)
+    let matches = regex.matches(in: String(line), options: [], range: range)
+
+    let match = matches[0]
+    let arrowhead = String(line[Range(match.range(at: 2), in: line)!]) == ">"
+    let vertexA = String(line[Range(match.range(at: 1), in: line)!])
+    let vertexB = String(line[Range(match.range(at: 3), in: line)!])
+
+    if let origin = vertices[vertexA] {
+      if let destination = vertices[vertexB] {
+        let edge = MapEdge(
+          id: index, origin: origin.position, destination: destination.position,
+          arrowhead: arrowhead)
+        return (MapEdge.self, edge)
+      }
+    }
+
+    return (NSObject.self, NSObject())  // No matching object
+  }
+}
diff --git a/Map/Logic/MapParser/Strategies/NoteParserStrategy.swift b/Map/Logic/MapParser/Strategies/NoteParserStrategy.swift
new file mode 100644
index 0000000..4dae4c7
--- /dev/null
+++ b/Map/Logic/MapParser/Strategies/NoteParserStrategy.swift
@@ -0,0 +1,31 @@
+import Foundation
+
+struct NoteParserStrategy: MapParserStrategy {
+  private let regex = MapParsingPatterns.note
+
+  func canHandle(line: String) -> Bool {
+    let range = NSRange(location: 0, length: line.utf16.count)
+    let matches = regex.matches(in: String(line), options: [], range: range)
+    return matches.count > 0 && matches[0].numberOfRanges == 5
+  }
+
+  func handle(index: Int, line: String, vertices: [String: Vertex]) -> (Any.Type, Any) {
+    let range = NSRange(location: 0, length: line.utf16.count)
+    let matches = regex.matches(in: String(line), options: [], range: range)
+
+    let match = matches[0]
+    let text = String(line[Range(match.range(at: 4), in: line)!])
+    let xString = String(line[Range(match.range(at: 2), in: line)!])
+    let yString = String(line[Range(match.range(at: 3), in: line)!])
+    let x = CGFloat(truncating: NumberFormatter().number(from: xString) ?? 0.0)
+    let y = CGFloat(truncating: NumberFormatter().number(from: yString) ?? 0.0)
+
+    let note = Note(
+      id: index,
+      position: CGPoint(x: x, y: y),
+      text: text
+    )
+
+    return (Note.self, note)
+  }
+}
diff --git a/Map/Logic/MapParser/Strategies/OpportunityParserStrategy.swift b/Map/Logic/MapParser/Strategies/OpportunityParserStrategy.swift
new file mode 100644
index 0000000..104fd30
--- /dev/null
+++ b/Map/Logic/MapParser/Strategies/OpportunityParserStrategy.swift
@@ -0,0 +1,33 @@
+import Foundation
+
+struct OpportunityParserStrategy: MapParserStrategy {
+  private let regex = MapParsingPatterns.opportunity
+
+  func canHandle(line: String) -> Bool {
+    let range = NSRange(location: 0, length: line.utf16.count)
+    let matches = regex.matches(in: String(line), options: [], range: range)
+    return matches.count > 0 && matches[0].numberOfRanges == 5
+  }
+
+  func handle(index: Int, line: String, vertices: [String: Vertex]) -> (Any.Type, Any) {
+    let range = NSRange(location: 0, length: line.utf16.count)
+    let matches = regex.matches(in: String(line), options: [], range: range)
+
+    let match = matches[0]
+    let multiplier = CGFloat(
+      String(line[Range(match.range(at: 3), in: line)!]) == "-" ? -1.0 : 1.0)
+    let vertex = String(line[Range(match.range(at: 2), in: line)!])
+    let opportunityString = String(line[Range(match.range(at: 4), in: line)!])
+    let opportunity = CGFloat(
+      truncating: NumberFormatter().number(from: opportunityString) ?? 0.0)
+
+    if let origin = vertices[vertex] {
+      let destination = CGPoint(
+        x: origin.position.x + opportunity * multiplier, y: origin.position.y)
+      let opportunity = Opportunity(id: index, origin: origin.position, destination: destination)
+      return (Opportunity.self, opportunity)
+    }
+
+    return (NSObject.self, NSObject())  // No matching object
+  }
+}
diff --git a/Map/Logic/MapParser/Strategies/StageParserStrategy.swift b/Map/Logic/MapParser/Strategies/StageParserStrategy.swift
new file mode 100644
index 0000000..42535fc
--- /dev/null
+++ b/Map/Logic/MapParser/Strategies/StageParserStrategy.swift
@@ -0,0 +1,24 @@
+import Foundation
+
+struct StageParserStrategy: MapParserStrategy {
+  private let regex = MapParsingPatterns.stage
+
+  func canHandle(line: String) -> Bool {
+    let range = NSRange(location: 0, length: line.utf16.count)
+    let matches = regex.matches(in: String(line), options: [], range: range)
+    return matches.count > 0 && matches[0].numberOfRanges == 3
+  }
+
+  func handle(index: Int, line: String, vertices: [String: Vertex]) -> (Any.Type, Any) {
+    let range = NSRange(location: 0, length: line.utf16.count)
+    let matches = regex.matches(in: String(line), options: [], range: range)
+
+    let match = matches[0]
+    let stage = String(line[Range(match.range(at: 1), in: line)!])
+    let dimensionsString = String(line[Range(match.range(at: 2), in: line)!])
+    let dimensions = CGFloat(truncating: NumberFormatter().number(from: dimensionsString) ?? 0.0)
+
+    let stageDimensions = StageDimensions(index: stage.count - 1, dimensions: dimensions)
+    return (StageDimensions.self, stageDimensions)
+  }
+}
diff --git a/Map/Logic/MapParser/Strategies/VertexParserStrategy.swift b/Map/Logic/MapParser/Strategies/VertexParserStrategy.swift
new file mode 100644
index 0000000..10c94a2
--- /dev/null
+++ b/Map/Logic/MapParser/Strategies/VertexParserStrategy.swift
@@ -0,0 +1,36 @@
+import Foundation
+
+struct VertexParserStrategy: MapParserStrategy {
+  private let regex = MapParsingPatterns.vertex
+
+  func canHandle(line: String) -> Bool {
+    let range = NSRange(location: 0, length: line.utf16.count)
+    let matches = regex.matches(in: String(line), options: [], range: range)
+    return matches.count > 0 && matches[0].numberOfRanges >= 4
+  }
+
+  func handle(index: Int, line: String, vertices: [String: Vertex]) -> (Any.Type, Any) {
+    let range = NSRange(location: 0, length: line.utf16.count)
+    let matches = regex.matches(in: String(line), options: [], range: range)
+
+    let match = matches[0]
+    let key = String(line[Range(match.range(at: 1), in: line)!])
+    let xString = String(line[Range(match.range(at: 2), in: line)!])
+    let yString = String(line[Range(match.range(at: 3), in: line)!])
+    let x = CGFloat(truncating: NumberFormatter().number(from: xString) ?? 0.0)
+    let y = CGFloat(truncating: NumberFormatter().number(from: yString) ?? 0.0)
+
+    var vertex = Vertex(
+      id: index,
+      label: key,
+      position: CGPoint(x: x, y: y)
+    )
+
+    if let range = Range(match.range(at: 4), in: line) {
+      let shapeString = String(line[range])
+      vertex.shape = VertexShape(rawValue: shapeString.lowercased()) ?? .circle
+    }
+
+    return (Vertex.self, vertex)
+  }
+}