diff --git a/CHANGELOG.md b/CHANGELOG.md index 5f009cb..b1ae738 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -12,6 +12,8 @@ [Ilya Puchka](https://github.com/yonaskolb) [#178](https://github.com/stencilproject/Stencil/pull/178) +- Added support for dynamic filter using `filter` filter + ### Bug Fixes - Fixed using quote as a filter parameter diff --git a/Sources/Expression.swift b/Sources/Expression.swift index c7199fc..bb4ef67 100644 --- a/Sources/Expression.swift +++ b/Sources/Expression.swift @@ -1,4 +1,4 @@ -protocol Expression: CustomStringConvertible { +public protocol Expression: CustomStringConvertible { func evaluate(context: Context) throws -> Bool } diff --git a/Sources/Extension.swift b/Sources/Extension.swift index 33a9925..0bb826d 100644 --- a/Sources/Extension.swift +++ b/Sources/Extension.swift @@ -26,6 +26,10 @@ open class Extension { /// Registers a template filter with the given name public func registerFilter(_ name: String, filter: @escaping (Any?, [Any?]) throws -> Any?) { + filters[name] = .arguments({ value, args, _ in try filter(value, args) }) + } + + public func registerFilter(_ name: String, filter: @escaping (Any?, [Any?], Context) throws -> Any?) { filters[name] = .arguments(filter) } } @@ -59,28 +63,28 @@ class DefaultExtension: Extension { registerFilter("join", filter: joinFilter) registerFilter("split", filter: splitFilter) registerFilter("indent", filter: indentFilter) + registerFilter("filter", filter: filterFilter) } } protocol FilterType { - func invoke(value: Any?, arguments: [Any?]) throws -> Any? + func invoke(value: Any?, arguments: [Any?], context: Context) throws -> Any? } enum Filter: FilterType { case simple(((Any?) throws -> Any?)) - case arguments(((Any?, [Any?]) throws -> Any?)) + case arguments(((Any?, [Any?], Context) throws -> Any?)) - func invoke(value: Any?, arguments: [Any?]) throws -> Any? { + func invoke(value: Any?, arguments: [Any?], context: Context) 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) + return try filter(value, arguments, context) } } } diff --git a/Sources/Filters.swift b/Sources/Filters.swift index fece6eb..4d664f3 100644 --- a/Sources/Filters.swift +++ b/Sources/Filters.swift @@ -39,7 +39,7 @@ func defaultFilter(value: Any?, arguments: [Any?]) -> Any? { func joinFilter(value: Any?, arguments: [Any?]) throws -> Any? { guard arguments.count < 2 else { - throw TemplateSyntaxError("'join' filter takes a single argument") + throw TemplateSyntaxError("'join' filter takes at most one argument") } let separator = stringify(arguments.first ?? "") @@ -55,7 +55,7 @@ func joinFilter(value: Any?, arguments: [Any?]) throws -> Any? { func splitFilter(value: Any?, arguments: [Any?]) throws -> Any? { guard arguments.count < 2 else { - throw TemplateSyntaxError("'split' filter takes a single argument") + throw TemplateSyntaxError("'split' filter takes at most one argument") } let separator = stringify(arguments.first ?? " ") @@ -111,3 +111,16 @@ func indent(_ content: String, indentation: String, indentFirst: Bool) -> String return result.joined(separator: "\n") } +func filterFilter(value: Any?, arguments: [Any?], context: Context) throws -> Any? { + guard let value = value else { return nil } + guard arguments.count == 1 else { + throw TemplateSyntaxError("'filter' filter takes one argument") + } + + let attribute = stringify(arguments[0]) + + let expr = try context.environment.compileFilter("$0|\(attribute)") + return try context.push(dictionary: ["$0": value]) { + try expr.resolve(context) + } +} diff --git a/Sources/ForTag.swift b/Sources/ForTag.swift index 4d7eff2..2450d5f 100644 --- a/Sources/ForTag.swift +++ b/Sources/ForTag.swift @@ -42,7 +42,7 @@ class ForNode : NodeType { let resolvable = try parser.compileResolvable(components[3]) let `where` = hasToken("where", at: 4) - ? try parseExpression(components: Array(components.suffix(from: 5)), tokenParser: parser) + ? try parser.compileExpression(components: Array(components.suffix(from: 5))) : nil return ForNode(resolvable: resolvable, loopVariables: loopVariables, nodes: forNodes, emptyNodes:emptyNodes, where: `where`) diff --git a/Sources/IfTag.swift b/Sources/IfTag.swift index a857d3e..e8fddad 100644 --- a/Sources/IfTag.swift +++ b/Sources/IfTag.swift @@ -100,7 +100,7 @@ final class IfExpressionParser { let tokens: [IfToken] var position: Int = 0 - init(components: [String], tokenParser: TokenParser) throws { + init(components: [String], environment: Environment) throws { self.tokens = try components.map { component in if let op = findOperator(name: component) { switch op { @@ -111,7 +111,7 @@ final class IfExpressionParser { } } - return .variable(try tokenParser.compileResolvable(component)) + return .variable(try environment.compileResolvable(component)) } } @@ -155,12 +155,6 @@ final class IfExpressionParser { } -func parseExpression(components: [String], tokenParser: TokenParser) throws -> Expression { - let parser = try IfExpressionParser(components: components, tokenParser: tokenParser) - return try parser.parse() -} - - /// Represents an if condition and the associated nodes when the condition /// evaluates final class IfCondition { @@ -187,7 +181,7 @@ class IfNode : NodeType { var components = token.components() components.removeFirst() - let expression = try parseExpression(components: components, tokenParser: parser) + let expression = try parser.compileExpression(components: components) let nodes = try parser.parse(until(["endif", "elif", "else"])) var conditions: [IfCondition] = [ IfCondition(expression: expression, nodes: nodes) @@ -197,7 +191,7 @@ class IfNode : NodeType { while let current = token, current.contents.hasPrefix("elif") { var components = current.components() components.removeFirst() - let expression = try parseExpression(components: components, tokenParser: parser) + let expression = try parser.compileExpression(components: components) let nodes = try parser.parse(until(["endif", "elif", "else"])) token = parser.nextToken() @@ -236,7 +230,7 @@ class IfNode : NodeType { _ = parser.nextToken() } - let expression = try parseExpression(components: components, tokenParser: parser) + let expression = try parser.compileExpression(components: components) return IfNode(conditions: [ IfCondition(expression: expression, nodes: trueNodes), IfCondition(expression: nil, nodes: falseNodes), diff --git a/Sources/Parser.swift b/Sources/Parser.swift index 81a44e1..7bd7ad7 100644 --- a/Sources/Parser.swift +++ b/Sources/Parser.swift @@ -48,7 +48,7 @@ public class TokenParser { } if let tag = token.components().first { - let parser = try findTag(name: tag) + let parser = try environment.findTag(name: tag) nodes.append(try parser(self, token)) } case .comment: @@ -71,8 +71,24 @@ public class TokenParser { tokens.insert(token, at: 0) } + public func compileFilter(_ token: String) throws -> Resolvable { + return try environment.compileFilter(token) + } + + public func compileExpression(components: [String]) throws -> Expression { + return try environment.compileExpression(components: components) + } + + public func compileResolvable(_ token: String) throws -> Resolvable { + return try environment.compileResolvable(token) + } + +} + +extension Environment { + func findTag(name: String) throws -> Extension.TagParser { - for ext in environment.extensions { + for ext in extensions { if let filter = ext.tags[name] { return filter } @@ -82,7 +98,7 @@ public class TokenParser { } func findFilter(_ name: String) throws -> FilterType { - for ext in environment.extensions { + for ext in extensions { if let filter = ext.filters[name] { return filter } @@ -97,7 +113,7 @@ public class TokenParser { } private func suggestedFilters(for name: String) -> [String] { - let allFilters = environment.extensions.flatMap({ $0.filters.keys }) + let allFilters = extensions.flatMap({ $0.filters.keys }) let filtersWithDistance = allFilters .map({ (filterName: $0, distance: $0.levenshteinDistance(name)) }) @@ -111,11 +127,15 @@ public class TokenParser { } public func compileFilter(_ token: String) throws -> Resolvable { - return try FilterExpression(token: token, parser: self) + return try FilterExpression(token: token, environment: self) + } + + public func compileExpression(components: [String]) throws -> Expression { + return try IfExpressionParser(components: components, environment: self).parse() } public func compileResolvable(_ token: String) throws -> Resolvable { - return try RangeVariable(token, parser: self) + return try RangeVariable(token, environment: self) ?? compileFilter(token) } diff --git a/Sources/Variable.swift b/Sources/Variable.swift index b357021..ac0816e 100644 --- a/Sources/Variable.swift +++ b/Sources/Variable.swift @@ -8,8 +8,8 @@ class FilterExpression : Resolvable { let filters: [(FilterType, [Variable])] let variable: Variable - init(token: String, parser: TokenParser) throws { - let bits = token.characters.split(separator: "|").map({ String($0).trim(character: " ") }) + init(token: String, environment: Environment) throws { + let bits = token.smartSplit(separator: "|").map({ String($0).trim(character: " ") }) if bits.isEmpty { filters = [] variable = Variable("") @@ -22,7 +22,7 @@ class FilterExpression : Resolvable { do { filters = try filterBits.map { let (name, arguments) = parseFilterComponents(token: $0) - let filter = try parser.findFilter(name) + let filter = try environment.findFilter(name) return (filter, arguments) } } catch { @@ -36,7 +36,7 @@ class FilterExpression : Resolvable { return try filters.reduce(result) { x, y in let arguments = try y.1.map { try $0.resolve(context) } - return try y.0.invoke(value: x, arguments: arguments) + return try y.0.invoke(value: x, arguments: arguments, context: context) } } } @@ -138,14 +138,14 @@ public struct RangeVariable: Resolvable { public let from: Resolvable public let to: Resolvable - public init?(_ token: String, parser: TokenParser) throws { + public init?(_ token: String, environment: Environment) throws { let components = token.components(separatedBy: "...") guard components.count == 2 else { return nil } - self.from = try parser.compileFilter(components[0]) - self.to = try parser.compileFilter(components[1]) + self.from = try environment.compileFilter(components[0]) + self.to = try environment.compileFilter(components[1]) } public func resolve(_ context: Context) throws -> Any? { diff --git a/Tests/StencilTests/ExpressionSpec.swift b/Tests/StencilTests/ExpressionSpec.swift index c41575f..b214a78 100644 --- a/Tests/StencilTests/ExpressionSpec.swift +++ b/Tests/StencilTests/ExpressionSpec.swift @@ -105,19 +105,19 @@ func testExpressions() { $0.describe("expression parsing") { $0.it("can parse a variable expression") { - let expression = try parseExpression(components: ["value"], tokenParser: parser) + let expression = try parser.compileExpression(components: ["value"]) try expect(expression.evaluate(context: Context())).to.beFalse() try expect(expression.evaluate(context: Context(dictionary: ["value": true]))).to.beTrue() } $0.it("can parse a not expression") { - let expression = try parseExpression(components: ["not", "value"], tokenParser: parser) + let expression = try parser.compileExpression(components: ["not", "value"]) try expect(expression.evaluate(context: Context())).to.beTrue() try expect(expression.evaluate(context: Context(dictionary: ["value": true]))).to.beFalse() } $0.describe("and expression") { - let expression = try! parseExpression(components: ["lhs", "and", "rhs"], tokenParser: parser) + let expression = try! parser.compileExpression(components: ["lhs", "and", "rhs"]) $0.it("evaluates to false with lhs false") { try expect(expression.evaluate(context: Context(dictionary: ["lhs": false, "rhs": true]))).to.beFalse() @@ -137,7 +137,7 @@ func testExpressions() { } $0.describe("or expression") { - let expression = try! parseExpression(components: ["lhs", "or", "rhs"], tokenParser: parser) + let expression = try! parser.compileExpression(components: ["lhs", "or", "rhs"]) $0.it("evaluates to true with lhs true") { try expect(expression.evaluate(context: Context(dictionary: ["lhs": true, "rhs": false]))).to.beTrue() @@ -157,7 +157,7 @@ func testExpressions() { } $0.describe("equality expression") { - let expression = try! parseExpression(components: ["lhs", "==", "rhs"], tokenParser: parser) + let expression = try! parser.compileExpression(components: ["lhs", "==", "rhs"]) $0.it("evaluates to true with equal lhs/rhs") { try expect(expression.evaluate(context: Context(dictionary: ["lhs": "a", "rhs": "a"]))).to.beTrue() @@ -193,7 +193,7 @@ func testExpressions() { } $0.describe("inequality expression") { - let expression = try! parseExpression(components: ["lhs", "!=", "rhs"], tokenParser: parser) + let expression = try! parser.compileExpression(components: ["lhs", "!=", "rhs"]) $0.it("evaluates to true with inequal lhs/rhs") { try expect(expression.evaluate(context: Context(dictionary: ["lhs": "a", "rhs": "b"]))).to.beTrue() @@ -205,7 +205,7 @@ func testExpressions() { } $0.describe("more than expression") { - let expression = try! parseExpression(components: ["lhs", ">", "rhs"], tokenParser: parser) + let expression = try! parser.compileExpression(components: ["lhs", ">", "rhs"]) $0.it("evaluates to true with lhs > rhs") { try expect(expression.evaluate(context: Context(dictionary: ["lhs": 5.0, "rhs": 4]))).to.beTrue() @@ -217,7 +217,7 @@ func testExpressions() { } $0.describe("more than equal expression") { - let expression = try! parseExpression(components: ["lhs", ">=", "rhs"], tokenParser: parser) + let expression = try! parser.compileExpression(components: ["lhs", ">=", "rhs"]) $0.it("evaluates to true with lhs == rhs") { try expect(expression.evaluate(context: Context(dictionary: ["lhs": 5.0, "rhs": 5]))).to.beTrue() @@ -229,7 +229,7 @@ func testExpressions() { } $0.describe("less than expression") { - let expression = try! parseExpression(components: ["lhs", "<", "rhs"], tokenParser: parser) + let expression = try! parser.compileExpression(components: ["lhs", "<", "rhs"]) $0.it("evaluates to true with lhs < rhs") { try expect(expression.evaluate(context: Context(dictionary: ["lhs": 4, "rhs": 4.5]))).to.beTrue() @@ -241,7 +241,7 @@ func testExpressions() { } $0.describe("less than equal expression") { - let expression = try! parseExpression(components: ["lhs", "<=", "rhs"], tokenParser: parser) + let expression = try! parser.compileExpression(components: ["lhs", "<=", "rhs"]) $0.it("evaluates to true with lhs == rhs") { try expect(expression.evaluate(context: Context(dictionary: ["lhs": 5.0, "rhs": 5]))).to.beTrue() @@ -253,7 +253,7 @@ func testExpressions() { } $0.describe("multiple expression") { - let expression = try! parseExpression(components: ["one", "or", "two", "and", "not", "three"], tokenParser: parser) + let expression = try! parser.compileExpression(components: ["one", "or", "two", "and", "not", "three"]) $0.it("evaluates to true with one") { try expect(expression.evaluate(context: Context(dictionary: ["one": true]))).to.beTrue() @@ -281,7 +281,7 @@ func testExpressions() { } $0.describe("in expression") { - let expression = try! parseExpression(components: ["lhs", "in", "rhs"], tokenParser: parser) + let expression = try! parser.compileExpression(components: ["lhs", "in", "rhs"]) $0.it("evaluates to true when rhs contains lhs") { try expect(expression.evaluate(context: Context(dictionary: ["lhs": 1, "rhs": [1, 2, 3]]))).to.beTrue() diff --git a/Tests/StencilTests/FilterSpec.swift b/Tests/StencilTests/FilterSpec.swift index f8de1b8..8b3aba7 100644 --- a/Tests/StencilTests/FilterSpec.swift +++ b/Tests/StencilTests/FilterSpec.swift @@ -3,14 +3,19 @@ import Spectre func testFilter() { + + func environmentWithFilter(_ name: String, closure: @escaping (Any?) throws -> Any?) -> Environment { + let filterExtension = Extension() + filterExtension.registerFilter(name, filter: closure) + return Environment(extensions: [filterExtension]) + } + describe("template filters") { let context: [String: Any] = ["name": "Kyle"] $0.it("allows you to register a custom filter") { let template = Template(templateString: "{{ name|repeat }}") - - let repeatExtension = Extension() - repeatExtension.registerFilter("repeat") { (value: Any?) in + let env = environmentWithFilter("repeat") { (value: Any?) in if let value = value as? String { return "\(value) \(value)" } @@ -18,7 +23,7 @@ func testFilter() { return nil } - let result = try template.render(Context(dictionary: context, environment: Environment(extensions: [repeatExtension]))) + let result = try template.render(Context(dictionary: context, environment: env)) try expect(result) == "Kyle Kyle" } @@ -39,19 +44,19 @@ func testFilter() { } $0.it("allows you to register a custom filter which accepts several arguments") { - let template = Template(templateString: "{{ name|repeat:'value\"1\"',\"value'2'\",'(key, value)' }}") - - let repeatExtension = Extension() - repeatExtension.registerFilter("repeat") { value, arguments in - if !arguments.isEmpty { - return "\(value!) \(value!) with args 0: \(arguments[0]!), 1: \(arguments[1]!), 2: \(arguments[2]!)" - } - - return nil + let template = Template(templateString: "{{ name|repeat:'value\"1\"',\"value'2'\",'(key, value)' }}") + + let repeatExtension = Extension() + repeatExtension.registerFilter("repeat") { value, arguments in + if !arguments.isEmpty { + return "\(value!) \(value!) with args 0: \(arguments[0]!), 1: \(arguments[1]!), 2: \(arguments[2]!)" } - - let result = try template.render(Context(dictionary: context, environment: Environment(extensions: [repeatExtension]))) - try expect(result) == "Kyle Kyle with args 0: value\"1\", 1: value'2', 2: (key, value)" + + return nil + } + + let result = try template.render(Context(dictionary: context, environment: Environment(extensions: [repeatExtension]))) + try expect(result) == "Kyle Kyle with args 0: value\"1\", 1: value'2', 2: (key, value)" } $0.it("allows you to register a custom which throws") { @@ -67,13 +72,11 @@ func testFilter() { $0.it("allows you to override a default filter") { let template = Template(templateString: "{{ name|join }}") - - let repeatExtension = Extension() - repeatExtension.registerFilter("join") { (value: Any?) in + let env = environmentWithFilter("join") { (value: Any?) in return "joined" } - let result = try template.render(Context(dictionary: context, environment: Environment(extensions: [repeatExtension]))) + let result = try template.render(Context(dictionary: context, environment: env)) try expect(result) == "joined" } @@ -212,7 +215,6 @@ func testFilter() { } } - describe("filter suggestion") { $0.it("made for unknown filter") { @@ -244,7 +246,6 @@ func testFilter() { } - describe("indent filter") { $0.it("indents content") { let template = Template(templateString: "{{ value|indent:2 }}") @@ -270,4 +271,27 @@ func testFilter() { try expect(result) == "One\n\n\n Two\n\n" } } + + describe("dynamic filter") { + + $0.it("can apply dynamic filter") { + let template = Template(templateString: "{{ name|filter:somefilter }}") + let result = try template.render(Context(dictionary: ["name": "Jhon", "somefilter": "uppercase"])) + try expect(result) == "JHON" + } + + $0.it("can apply dynamic filter on array") { + let template = Template(templateString: "{{ values|filter:joinfilter }}") + let result = try template.render(Context(dictionary: ["values": [1, 2, 3], "joinfilter": "join:\", \""])) + try expect(result) == "1, 2, 3" + } + + $0.it("throws on unknown dynamic filter") { + let template = Template(templateString: "{{ values|filter:unknown }}") + let context = Context(dictionary: ["values": [1, 2, 3], "unknown": "absurd"]) + try expect(try template.render(context)).toThrow() + } + + } + } diff --git a/Tests/StencilTests/ForNodeSpec.swift b/Tests/StencilTests/ForNodeSpec.swift index 4bb3fca..03b40fc 100644 --- a/Tests/StencilTests/ForNodeSpec.swift +++ b/Tests/StencilTests/ForNodeSpec.swift @@ -98,7 +98,8 @@ func testForNode() { $0.it("renders the given nodes while filtering items using where expression") { let nodes: [NodeType] = [VariableNode(variable: "item"), VariableNode(variable: "forloop.counter")] - let `where` = try parseExpression(components: ["item", ">", "1"], tokenParser: TokenParser(tokens: [], environment: Environment())) + let parser = TokenParser(tokens: [], environment: Environment()) + let `where` = try parser.compileExpression(components: ["item", ">", "1"]) let node = ForNode(resolvable: Variable("items"), loopVariables: ["item"], nodes: nodes, emptyNodes: [], where: `where`) try expect(try node.render(context)) == "2132" } @@ -106,7 +107,8 @@ func testForNode() { $0.it("renders the given empty nodes when all items filtered out with where expression") { let nodes: [NodeType] = [VariableNode(variable: "item")] let emptyNodes: [NodeType] = [TextNode(text: "empty")] - let `where` = try parseExpression(components: ["item", "==", "0"], tokenParser: TokenParser(tokens: [], environment: Environment())) + let parser = TokenParser(tokens: [], environment: Environment()) + let `where` = try parser.compileExpression(components: ["item", "==", "0"]) let node = ForNode(resolvable: Variable("emptyItems"), loopVariables: ["item"], nodes: nodes, emptyNodes: emptyNodes, where: `where`) try expect(try node.render(context)) == "empty" } diff --git a/Tests/StencilTests/VariableSpec.swift b/Tests/StencilTests/VariableSpec.swift index 3ca28cb..3e1b634 100644 --- a/Tests/StencilTests/VariableSpec.swift +++ b/Tests/StencilTests/VariableSpec.swift @@ -200,7 +200,7 @@ func testVariable() { }() func makeVariable(_ token: String) throws -> RangeVariable? { - return try RangeVariable(token, parser: TokenParser(tokens: [], environment: context.environment)) + return try RangeVariable(token, environment: context.environment) } $0.it("can resolve closed range as array") {