wildthink · January 16, 2022 16:43
diff --git a/script.swift b/script.swift
 import SwiftSyntax
 import SwiftSemantics
 import Foundation

 let source = try! String(contentsOf: URL(fileURLWithPath: "/Applications/Xcode.app/Contents/Developer/Platforms/iPhoneOS.platform/Developer/SDKs/iPhoneOS13.2.sdk/System/Library/Frameworks/SwiftUI.framework/Modules/SwiftUI.swiftmodule/arm64e.swiftinterface"))

 var collector = DeclarationCollector()
 let tree = try SyntaxParser.parse(source: source)
 tree.walk(&collector)

 extension String {
    var simplified: String {
        return String(split(separator: ".").last!)
    }
 }

 var allExtensions: [String:[String]] = Dictionary(collector.extensions.map { e in
    (e.extendedType, e.inheritance)
 }, uniquingKeysWith: { $0 + $1 })

 var conformances: [String:Set<String>] = allExtensions.mapValues(Set.init)

 protocol Decl {
    var inheritance: [String] { get }
    var name: String { get }
    
 }

 func getExtensions(name: String) -> [String] {
    return allExtensions[name] ?? []
 }


 extension Decl {
    func process() {
        for prot in inheritance {
            conformances[prot, default: []].insert(name)
        }

    }
 }

 extension Structure: Decl {}
 extension Enumeration: Decl {}
 extension Class: Decl {}

 for s in collector.structures {
    s.process()
 }
 for s in collector.enumerations {
    s.process()
 }
 for s in collector.classes {
    s.process()
 }
 for p in collector.protocols {
    for p2 in p.inheritance {
        conformances[p2, default: []].insert(p.name)
    }
 }



 //extension Decl {
 //    var allInheritance: [String] {
 //        let result = inheritance + inheritance.flatMap(getExtensions) + getExtensions(name: name)
 //        return Array(Set(result))
 //    }
 //}

 func fix<A: Equatable>(_ value: A, _ apply: (A) -> A) -> A {
    var result = value
    while true {
        let newResult = apply(result)
        if newResult == result { return result }
        result = newResult
    }
 }

 extension Decl {
    func print() {
        Swift.print("###", name)
        let inherits = inheritance + getExtensions(name: name)
        let allInherits = fix(Set(inherits), { Set($0 + $0.flatMap { Set(getExtensions(name: $0)) } ) })
        Swift.print(allInherits.sorted().map { "  - \($0.simplified)" }.joined(separator: "\n"))
        Swift.print("\n")
    }
 }


 print("# Structs")
 for s in collector.structures.sorted(by: { $0.name < $1.name }) {
    s.print()
 }
 print("# Enums")
 for s in collector.enumerations.sorted(by: { $0.name < $1.name }) {
    s.print()
 }
 print("# Classes")
 for s in collector.classes.sorted(by: { $0.name < $1.name }) {
    s.print()
 }


 print("# Protocols")
 for (prot, conformances) in conformances.sorted(by: { $0.key < $1.key }) {
    print("###", prot)
    let confs = fix(Set(conformances), { Set($0 + $0.flatMap { Set(getExtensions(name: $0)) } ) })
    print(confs.sorted().map { "  - \($0)" }.joined(separator: "\n"))
    print("\n")
 }
diff --git a/swiftui.md b/swiftui.md
	import SwiftSyntax
	import SwiftSemantics
	import Foundation

	let source = try! String(contentsOf: URL(fileURLWithPath: "/Applications/Xcode.app/Contents/Developer/Platforms/iPhoneOS.platform/Developer/SDKs/iPhoneOS13.2.sdk/System/Library/Frameworks/SwiftUI.framework/Modules/SwiftUI.swiftmodule/arm64e.swiftinterface"))

	var collector = DeclarationCollector()
	let tree = try SyntaxParser.parse(source: source)
	tree.walk(&collector)

	extension String {
	var simplified: String {
	return String(split(separator: ".").last!)
	}
	}

	var allExtensions: [String:[String]] = Dictionary(collector.extensions.map { e in
	(e.extendedType, e.inheritance)
	}, uniquingKeysWith: { $0 + $1 })

	var conformances: [String:Set<String>] = allExtensions.mapValues(Set.init)

	protocol Decl {
	var inheritance: [String] { get }
	var name: String { get }

	}

	func getExtensions(name: String) -> [String] {
	return allExtensions[name] ?? []
	}


	extension Decl {
	func process() {
	for prot in inheritance {
	conformances[prot, default: []].insert(name)
	}

	}
	}

	extension Structure: Decl {}
	extension Enumeration: Decl {}
	extension Class: Decl {}

	for s in collector.structures {
	s.process()
	}
	for s in collector.enumerations {
	s.process()
	}
	for s in collector.classes {
	s.process()
	}
	for p in collector.protocols {
	for p2 in p.inheritance {
	conformances[p2, default: []].insert(p.name)
	}
	}



	//extension Decl {
	// var allInheritance: [String] {
	// let result = inheritance + inheritance.flatMap(getExtensions) + getExtensions(name: name)
	// return Array(Set(result))
	// }
	//}

	func fix<A: Equatable>(_ value: A, _ apply: (A) -> A) -> A {
	var result = value
	while true {
	let newResult = apply(result)
	if newResult == result { return result }
	result = newResult
	}
	}

	extension Decl {
	func print() {
	Swift.print("###", name)
	let inherits = inheritance + getExtensions(name: name)
	let allInherits = fix(Set(inherits), { Set($0 + $0.flatMap { Set(getExtensions(name: $0)) } ) })
	Swift.print(allInherits.sorted().map { " - \($0.simplified)" }.joined(separator: "\n"))
	Swift.print("\n")
	}
	}


	print("# Structs")
	for s in collector.structures.sorted(by: { $0.name < $1.name }) {
	s.print()
	}
	print("# Enums")
	for s in collector.enumerations.sorted(by: { $0.name < $1.name }) {
	s.print()
	}
	print("# Classes")
	for s in collector.classes.sorted(by: { $0.name < $1.name }) {
	s.print()
	}


	print("# Protocols")
	for (prot, conformances) in conformances.sorted(by: { $0.key < $1.key }) {
	print("###", prot)
	let confs = fix(Set(conformances), { Set($0 + $0.flatMap { Set(getExtensions(name: $0)) } ) })
	print(confs.sorted().map { " - \($0)" }.joined(separator: "\n"))
	print("\n")
	}