SyntaxVisitor.swift

override func visit(_ node: ClassDeclSyntax) -> SyntaxVisitorContinueKind {
  // Do something with a class node
  return .visitChildren
}

override func visit(_ node: ExtensionDeclSyntax) -> SyntaxVisitorContinueKind {
  // Do something totally different with an extension node
  return .visitChildren
}