feat(filters): Allow filters with arguments

Sources/Namespace.swift

@@ -1,3 +1,26 @@
+public protocol FilterType {
+  func invoke(value: Any?, arguments: [Any?]) throws -> Any?
+}
+
+enum Filter: FilterType {
+  case simple(((Any?) throws -> Any?))
+  case arguments(((Any?, [Any?]) throws -> Any?))
+
+  func invoke(value: Any?, arguments: [Any?]) throws -> Any? {
+    switch self {
+    case let .simple(filter):
+      if !arguments.isEmpty {
+        throw TemplateSyntaxError("cannot invoke filter with an argument")
+      }
+
+      return try filter(value)
+    case let .arguments(filter):
+      return try filter(value, arguments)
+    }
+  }
+}
+
+
 open class Namespace {
   public typealias TagParser = (TokenParser, Token) throws -> NodeType
 
@@ -40,7 +63,12 @@ open class Namespace {
   }
 
   /// Registers a template filter with the given name
-  open func registerFilter(_ name: String, filter: @escaping Filter) {
-    filters[name] = filter
+  open func registerFilter(_ name: String, filter: @escaping (Any?) throws -> Any?) {
+    filters[name] = .simple(filter)
+  }
+
+  /// Registers a template filter with the given name
+  open func registerFilter(_ name: String, filter: @escaping (Any?, [Any?]) throws -> Any?) {
+    filters[name] = .arguments(filter)
   }
 }

Sources/Parser.swift

@@ -12,7 +12,6 @@ public func until(_ tags: [String]) -> ((TokenParser, Token) -> Bool) {
   }
 }
 
-public typealias Filter = (Any?) throws -> Any?
 
 /// A class for parsing an array of tokens and converts them into a collection of Node's
 open class TokenParser {
@@ -77,7 +76,7 @@ open class TokenParser {
     tokens.insert(token, at: 0)
   }
 
-  open func findFilter(_ name: String) throws -> Filter {
+  open func findFilter(_ name: String) throws -> FilterType {
     if let filter = namespace.filters[name] {
       return filter
     }

Sources/Variable.swift

@@ -2,7 +2,7 @@ import Foundation
 
 
 class FilterExpression : Resolvable {
-  let filters: [Filter]
+  let filters: [(FilterType, [Variable])]
   let variable: Variable
 
   init(token: String, parser: TokenParser) throws {
@@ -17,7 +17,11 @@ class FilterExpression : Resolvable {
     let filterBits = bits[bits.indices.suffix(from: 1)]
 
     do {
-      filters = try filterBits.map { try parser.findFilter($0) }
+      filters = try filterBits.map {
+        let (name, arguments) = parseFilterComponents(token: $0)
+        let filter = try parser.findFilter(name)
+        return (filter, arguments)
+      }
     } catch {
       filters = []
       throw error
@@ -28,7 +32,8 @@ class FilterExpression : Resolvable {
     let result = try variable.resolve(context)
 
     return try filters.reduce(result) { x, y in
-      return try y(x)
+      let arguments = try y.1.map { try $0.resolve(context) }
+      return try y.0.invoke(value: x, arguments: arguments)
     }
   }
 }
@@ -135,3 +140,10 @@ extension Dictionary : Normalizable {
     return dictionary
   }
 }
+
+func parseFilterComponents(token: String) -> (String, [Variable]) {
+  var components = token.characters.split(separator: ":").map(String.init)
+  let name = components.removeFirst()
+  let variables = components.joined(separator: ":").characters.split(separator: ",").map { Variable(String($0)) }
+  return (name, variables)
+}