[Swift] CommandPlugin で クラス図を作る(swift-syntax/Hatch を使ったCommandPlugin のブラッシュアップ)

extension Hatch.Symbol {
    func mermaidString() -> String {
        guard let namedSymbol = self as? Hatch.InheritingSymbol else { return "unsupported type" }
        let mermaidName = namedSymbol.name.replacingOccurrences(of: ".", with: "_")
        return """
               class \(mermaidName) {
                \(self.swiftTypeName())>>
               }
               
               """
    }
}

extension Hatch.Symbol {
    func swiftTypeName() -> String {
        switch self {
        case is Hatch.Struct: return "struct"
        case is Hatch.Class: return "class"
        case is Hatch.Enum: return "enum"
        case is Hatch.Actor: return "actor"
        case is Hatch.ProtocolType: return "protocol"
        case is Hatch.Extension: return "extension"
        default:
            return "UnsupportedType"
        }
    }
}

classDiagram
%% Actor.swift
class Actor1 {
 >
}
%% Class.swift
class Class1 {
 >
}
%% CommandPluginExample.swift
class CommandPluginExample {
 >
}
class CommandPluginExample {
 >
}
%% CommandPluginExampleLib.swift
class CommandPluginExampleLib {
 >
}
class Hatch_Symbol {
 >
}
%% CommandPluginExampleLibParserTests.swift
class CommandPluginExampleLibParserTests {
 >
}
%% Enum.swift
class Enum1 {
 >
}
%% Package.swift
%% Protocol.swift
class Protocol1 {
 >
}
%% Struct.swift
class Struct1 {
 >
}
%% UnderstandingTests.swift
class UnderstandingTests {
 >
}
%% swiftymermaid.swift
class SwiftyMermaid {
 >
}