| %{ |
| from gyb_syntax_support import * |
| # -*- mode: Swift -*- |
| # Ignore the following admonition it applies to the resulting .swift file only |
| def is_visitable(node): |
| return not node.is_base() and not node.collection_element |
| }% |
| //// Automatically Generated From SyntaxFactory.swift.gyb. |
| //// Do Not Edit Directly! |
| //===------------ SyntaxRewriter.swift - Syntax Rewriter class ------------===// |
| // |
| // This source file is part of the Swift.org open source project |
| // |
| // Copyright (c) 2014 - 2017 Apple Inc. and the Swift project authors |
| // Licensed under Apache License v2.0 with Runtime Library Exception |
| // |
| // See https://swift.org/LICENSE.txt for license information |
| // See https://swift.org/CONTRIBUTORS.txt for the list of Swift project authors |
| // |
| //===----------------------------------------------------------------------===// |
| // |
| // This file defines the SyntaxRewriter, a class that performs a standard walk |
| // and tree-rebuilding pattern. |
| // |
| // Subclassers of this class can override the walking behavior for any syntax |
| // node and transform nodes however they like. |
| // |
| //===----------------------------------------------------------------------===// |
| |
| open class SyntaxRewriter { |
| public init() {} |
| % for node in SYNTAX_NODES: |
| % if is_visitable(node): |
| open func visit(_ node: ${node.name}) -> ${node.base_type} { |
| % cast = ('as! ' + node.base_type) if node.base_type != 'Syntax' else '' |
| return visitChildren(node)${cast} |
| } |
| |
| % end |
| % end |
| |
| open func visit(_ token: TokenSyntax) -> Syntax { |
| return token |
| } |
| public func visit(_ node: Syntax) -> Syntax { |
| switch node.raw.kind { |
| case .token: return visit(node as! TokenSyntax) |
| % for node in SYNTAX_NODES: |
| % if is_visitable(node): |
| case .${node.swift_syntax_kind}: return visit(node as! ${node.name}) |
| % end |
| % end |
| default: return visitChildren(node) |
| } |
| } |
| |
| func visitChildren(_ node: Syntax) -> Syntax { |
| let newLayout = node.children.map { visit($0).raw } |
| return Syntax.fromRaw(node.raw.replacingLayout(newLayout)) |
| } |
| } |