comparing 7112ae1e8f3393dbebcda3c2dff3f981cd84112e and main on jonmsterling.com/swift-pterodactyl

+1 -1

.gitignore

···

       2
       2
        
       #

     

       3
       3
        
       # SPDX-License-Identifier: MPL-2.0

     

       4
       4
        
       

     

       5
       5
       +
       .jj/*

     

       5
       6
        
       .DS_Store

     

       6
       7
        
       /.build

     

       7
       8
        
       /Packages

     
···

       10
       11
        
       .swiftpm/configuration/registries.json

     

       11
       12
        
       .swiftpm/xcode/package.xcworkspace/contents.xcworkspacedata

     

       12
       13
        
       .netrc

     

       13
       13
       -
       Package.resolved

+159

Package.resolved

···

       1
       1
       +
       {

     

       2
       2
       +
         "originHash" : "ed464b5a4d4d87a0158db3738aa249b07410a79ce65ef2fa9d16a0b168de3b32",

     

       3
       3
       +
         "pins" : [

     

       4
       4
       +
           {

     

       5
       5
       +
             "identity" : "jsonrpc",

     

       6
       6
       +
             "kind" : "remoteSourceControl",

     

       7
       7
       +
             "location" : "https://github.com/ChimeHQ/JSONRPC",

     

       8
       8
       +
             "state" : {

     

       9
       9
       +
               "revision" : "29987f721374f30e686af40ccffd0b13b14dde1f",

     

       10
       10
       +
               "version" : "0.9.2"

     

       11
       11
       +
             }

     

       12
       12
       +
           },

     

       13
       13
       +
           {

     

       14
       14
       +
             "identity" : "swift-algorithms",

     

       15
       15
       +
             "kind" : "remoteSourceControl",

     

       16
       16
       +
             "location" : "https://github.com/apple/swift-algorithms",

     

       17
       17
       +
             "state" : {

     

       18
       18
       +
               "revision" : "87e50f483c54e6efd60e885f7f5aa946cee68023",

     

       19
       19
       +
               "version" : "1.2.1"

     

       20
       20
       +
             }

     

       21
       21
       +
           },

     

       22
       22
       +
           {

     

       23
       23
       +
             "identity" : "swift-async-algorithms",

     

       24
       24
       +
             "kind" : "remoteSourceControl",

     

       25
       25
       +
             "location" : "https://github.com/apple/swift-async-algorithms.git",

     

       26
       26
       +
             "state" : {

     

       27
       27
       +
               "revision" : "6c050d5ef8e1aa6342528460db614e9770d7f804",

     

       28
       28
       +
               "version" : "1.1.1"

     

       29
       29
       +
             }

     

       30
       30
       +
           },

     

       31
       31
       +
           {

     

       32
       32
       +
             "identity" : "swift-atomics",

     

       33
       33
       +
             "kind" : "remoteSourceControl",

     

       34
       34
       +
             "location" : "https://github.com/apple/swift-atomics.git",

     

       35
       35
       +
             "state" : {

     

       36
       36
       +
               "revision" : "b601256eab081c0f92f059e12818ac1d4f178ff7",

     

       37
       37
       +
               "version" : "1.3.0"

     

       38
       38
       +
             }

     

       39
       39
       +
           },

     

       40
       40
       +
           {

     

       41
       41
       +
             "identity" : "swift-collections",

     

       42
       42
       +
             "kind" : "remoteSourceControl",

     

       43
       43
       +
             "location" : "https://github.com/apple/swift-collections.git",

     

       44
       44
       +
             "state" : {

     

       45
       45
       +
               "revision" : "7b847a3b7008b2dc2f47ca3110d8c782fb2e5c7e",

     

       46
       46
       +
               "version" : "1.3.0"

     

       47
       47
       +
             }

     

       48
       48
       +
           },

     

       49
       49
       +
           {

     

       50
       50
       +
             "identity" : "swift-distributed-tracing",

     

       51
       51
       +
             "kind" : "remoteSourceControl",

     

       52
       52
       +
             "location" : "https://github.com/apple/swift-distributed-tracing",

     

       53
       53
       +
             "state" : {

     

       54
       54
       +
               "revision" : "baa932c1336f7894145cbaafcd34ce2dd0b77c97",

     

       55
       55
       +
               "version" : "1.3.1"

     

       56
       56
       +
             }

     

       57
       57
       +
           },

     

       58
       58
       +
           {

     

       59
       59
       +
             "identity" : "swift-languageserver",

     

       60
       60
       +
             "kind" : "remoteSourceControl",

     

       61
       61
       +
             "location" : "https://github.com/jonsterling/swift-LanguageServer",

     

       62
       62
       +
             "state" : {

     

       63
       63
       +
               "branch" : "main",

     

       64
       64
       +
               "revision" : "8837131a4b8707a466d5a0fe1d122a566c73f3e1"

     

       65
       65
       +
             }

     

       66
       66
       +
           },

     

       67
       67
       +
           {

     

       68
       68
       +
             "identity" : "swift-languageserverprotocol",

     

       69
       69
       +
             "kind" : "remoteSourceControl",

     

       70
       70
       +
             "location" : "https://github.com/jonsterling/swift-LanguageServerProtocol",

     

       71
       71
       +
             "state" : {

     

       72
       72
       +
               "branch" : "main",

     

       73
       73
       +
               "revision" : "4dfba8054ae50840798242639afffa3411c93699"

     

       74
       74
       +
             }

     

       75
       75
       +
           },

     

       76
       76
       +
           {

     

       77
       77
       +
             "identity" : "swift-llbuild2",

     

       78
       78
       +
             "kind" : "remoteSourceControl",

     

       79
       79
       +
             "location" : "https://github.com/apple/swift-llbuild2.git",

     

       80
       80
       +
             "state" : {

     

       81
       81
       +
               "branch" : "main",

     

       82
       82
       +
               "revision" : "fa72f6a074360c6eb707ca53455874f4c0c4852f"

     

       83
       83
       +
             }

     

       84
       84
       +
           },

     

       85
       85
       +
           {

     

       86
       86
       +
             "identity" : "swift-log",

     

       87
       87
       +
             "kind" : "remoteSourceControl",

     

       88
       88
       +
             "location" : "https://github.com/apple/swift-log.git",

     

       89
       89
       +
             "state" : {

     

       90
       90
       +
               "revision" : "bc386b95f2a16ccd0150a8235e7c69eab2b866ca",

     

       91
       91
       +
               "version" : "1.8.0"

     

       92
       92
       +
             }

     

       93
       93
       +
           },

     

       94
       94
       +
           {

     

       95
       95
       +
             "identity" : "swift-nio",

     

       96
       96
       +
             "kind" : "remoteSourceControl",

     

       97
       97
       +
             "location" : "https://github.com/apple/swift-nio.git",

     

       98
       98
       +
             "state" : {

     

       99
       99
       +
               "revision" : "a1605a3303a28e14d822dec8aaa53da8a9490461",

     

       100
       100
       +
               "version" : "2.92.0"

     

       101
       101
       +
             }

     

       102
       102
       +
           },

     

       103
       103
       +
           {

     

       104
       104
       +
             "identity" : "swift-numerics",

     

       105
       105
       +
             "kind" : "remoteSourceControl",

     

       106
       106
       +
             "location" : "https://github.com/apple/swift-numerics.git",

     

       107
       107
       +
             "state" : {

     

       108
       108
       +
               "revision" : "0c0290ff6b24942dadb83a929ffaaa1481df04a2",

     

       109
       109
       +
               "version" : "1.1.1"

     

       110
       110
       +
             }

     

       111
       111
       +
           },

     

       112
       112
       +
           {

     

       113
       113
       +
             "identity" : "swift-protobuf",

     

       114
       114
       +
             "kind" : "remoteSourceControl",

     

       115
       115
       +
             "location" : "https://github.com/apple/swift-protobuf.git",

     

       116
       116
       +
             "state" : {

     

       117
       117
       +
               "revision" : "c169a5744230951031770e27e475ff6eefe51f9d",

     

       118
       118
       +
               "version" : "1.33.3"

     

       119
       119
       +
             }

     

       120
       120
       +
           },

     

       121
       121
       +
           {

     

       122
       122
       +
             "identity" : "swift-service-context",

     

       123
       123
       +
             "kind" : "remoteSourceControl",

     

       124
       124
       +
             "location" : "https://github.com/apple/swift-service-context.git",

     

       125
       125
       +
             "state" : {

     

       126
       126
       +
               "revision" : "1983448fefc717a2bc2ebde5490fe99873c5b8a6",

     

       127
       127
       +
               "version" : "1.2.1"

     

       128
       128
       +
             }

     

       129
       129
       +
           },

     

       130
       130
       +
           {

     

       131
       131
       +
             "identity" : "swift-system",

     

       132
       132
       +
             "kind" : "remoteSourceControl",

     

       133
       133
       +
             "location" : "https://github.com/apple/swift-system.git",

     

       134
       134
       +
             "state" : {

     

       135
       135
       +
               "revision" : "395a77f0aa927f0ff73941d7ac35f2b46d47c9db",

     

       136
       136
       +
               "version" : "1.6.3"

     

       137
       137
       +
             }

     

       138
       138
       +
           },

     

       139
       139
       +
           {

     

       140
       140
       +
             "identity" : "swift-tools-support-async",

     

       141
       141
       +
             "kind" : "remoteSourceControl",

     

       142
       142
       +
             "location" : "https://github.com/apple/swift-tools-support-async.git",

     

       143
       143
       +
             "state" : {

     

       144
       144
       +
               "revision" : "8cd215778056b7baab3cd9e57ba03559c02f480a",

     

       145
       145
       +
               "version" : "0.17.0"

     

       146
       146
       +
             }

     

       147
       147
       +
           },

     

       148
       148
       +
           {

     

       149
       149
       +
             "identity" : "swift-tools-support-core",

     

       150
       150
       +
             "kind" : "remoteSourceControl",

     

       151
       151
       +
             "location" : "https://github.com/apple/swift-tools-support-core.git",

     

       152
       152
       +
             "state" : {

     

       153
       153
       +
               "revision" : "e8fbc8b05a155f311b862178d92d043afb216fe3",

     

       154
       154
       +
               "version" : "0.7.3"

     

       155
       155
       +
             }

     

       156
       156
       +
           }

     

       157
       157
       +
         ],

     

       158
       158
       +
         "version" : 3

     

       159
       159
       +
       }

+35 -2

Package.swift

···

       23
       23
        
       		.library(

     

       24
       24
        
       			name: "PterodactylBuild",

     

       25
       25
        
       			targets: ["PterodactylBuild"]

     

       26
       26
       +
       		),

     

       27
       27
       +
       		.executable(

     

       28
       28
       +
       			name: "PterodactylLanguageServer",

     

       29
       29
       +
       			targets: ["PterodactylLanguageServer"]

     

       26
       30
        
       		)

     

       27
       31
        
       	],

     

       28
       32
        
       	dependencies: [

     

       29
       29
       -
       		.package(url: "https://github.com/apple/swift-llbuild2.git", branch: "main")

     

       33
       33
       +
       		.package(url: "https://github.com/jonsterling/swift-LanguageServer", branch: "main"),

     

       34
       34
       +
       		.package(url: "https://github.com/jonsterling/swift-LanguageServerProtocol", branch: "main"),

     

       35
       35
       +
       		.package(url: "https://github.com/apple/swift-llbuild2.git", branch: "main"),

     

       36
       36
       +
       		.package(url: "https://github.com/apple/swift-algorithms", from: "1.2.0"),

     

       37
       37
       +
       		.package(url: "https://github.com/apple/swift-log", from: "1.6.0")

     

       30
       38
        
       	],

     

       31
       39
        
       	targets: [

     

       32
       40
        
       		// Targets are the basic building blocks of a package, defining a module or a test suite.

     
···

       35
       43
        
       			name: "PterodactylKernel",

     

       36
       44
        
       		),

     

       37
       45
        
       		.target(

     

       38
       38
       -
       			name: "PterodactylSyntax"

     

       46
       46
       +
       			name: "PterodactylSyntax",

     

       47
       47
       +
       			dependencies: [

     

       48
       48
       +
       				.product(name: "Algorithms", package: "swift-algorithms"),

     

       49
       49
       +
       				.product(name: "LanguageServerProtocol", package: "swift-LanguageServerProtocol")

     

       50
       50
       +
       			]

     

       39
       51
        
       		),

     

       40
       52
        
       		.target(

     

       41
       53
        
       			name: "PterodactylBuild",

     

       42
       54
        
       			dependencies: [

     

       55
       55
       +
       				"PterodactylSyntax",

     

       43
       56
        
       				.product(name: "llbuild2fx", package: "swift-llbuild2")

     

       44
       57
        
       			]

     

       45
       58
        
       		),

     

       59
       59
       +
       		.executableTarget(

     

       60
       60
       +
       			name: "PterodactylLanguageServer",

     

       61
       61
       +
       			dependencies: [

     

       62
       62
       +
       				.product(name: "LanguageServer", package: "swift-LanguageServer"),

     

       63
       63
       +
       				.product(name: "llbuild2fx", package: "swift-llbuild2"),

     

       64
       64
       +
       				.product(name: "Logging", package: "swift-log"),

     

       65
       65
       +
       				"PterodactylBuild",

     

       66
       66
       +
       				"PterodactylSyntax"

     

       67
       67
       +
       			]

     

       68
       68
       +
       		),

     

       46
       69
        
       		.testTarget(

     

       47
       70
        
       			name: "PterodactylBuildTests",

     

       48
       71
        
       			dependencies: [

     

       49
       72
        
       				"PterodactylBuild",

     

       73
       73
       +
       				"PterodactylSyntax",

     

       74
       74
       +
       				.product(name: "llbuild2fx", package: "swift-llbuild2")

     

       75
       75
       +
       			]

     

       76
       76
       +
       		),

     

       77
       77
       +
       		.testTarget(

     

       78
       78
       +
       			name: "PterodactylLanguageServerTests",

     

       79
       79
       +
       			dependencies: [

     

       80
       80
       +
       				"PterodactylBuild",

     

       81
       81
       +
       				"PterodactylSyntax",

     

       82
       82
       +
       				"PterodactylLanguageServer",

     

       50
       83
        
       				.product(name: "llbuild2fx", package: "swift-llbuild2")

     

       51
       84
        
       			]

     

       52
       85
        
       		),

+9 -10

Sources/PterodactylBuild/BuildContext.swift

···

       2
       2
        
       //

     

       3
       3
        
       // SPDX-License-Identifier: MPL-2.0

     

       4
       4
        
       

     

       5
       5
       -
       import Foundation

     

       6
       5
        
       import TSCBasic

     

       7
       6
        
       import llbuild2fx

     

       8
       7
        
       

     
···

       31
       30
        
       }

     

       32
       31
        
       

     

       33
       32
        
       extension BuildKey {

     

       34
       34
       -
       	func computeValue(_ fi: FXFunctionInterface<Self>, _ ctx: Context) async throws -> ValueType {

     

       33
       33
       +
       	public func computeValue(_ fi: FXFunctionInterface<Self>, _ ctx: Context) async throws -> ValueType {

     

       35
       34
        
       		try await computeValue(BuildContext(functionInterface: fi, context: ctx))

     

       36
       35
        
       	}

     

       37
       36
        
       }

     

       38
       37
        
       

     

       39
       38
        
       extension LLBCASFileTree {

     

       40
       40
       -
       	static func load<X: FXKey>(id: LLBDataID, in ctx: BuildContext<X>) async throws -> LLBCASFileTree {

     

       41
       41
       -
           try await load(id: id, from: ctx.db, ctx.context).get()

     

       39
       39
       +
       	public static func load<X: FXKey>(id: LLBDataID, in ctx: BuildContext<X>) async throws -> LLBCASFileTree {

     

       40
       40
       +
       		try await load(id: id, from: ctx.db, ctx.context).get()

     

       42
       41
        
       	}

     

       43
       42
        
       

     

       44
       44
       -
         func remove<X: FXKey>(path: AbsolutePath, in ctx: BuildContext<X>) async throws -> LLBCASFileTree {

     

       45
       45
       -
           try await remove(path: path, in: ctx.db, ctx.context).get()

     

       46
       46
       -
         }

     

       43
       43
       +
       	public func remove<X: FXKey>(path: AbsolutePath, in ctx: BuildContext<X>) async throws -> LLBCASFileTree {

     

       44
       44
       +
       		try await remove(path: path, in: ctx.db, ctx.context).get()

     

       45
       45
       +
       	}

     

       47
       46
        
       

     

       48
       48
       -
         func traverse<X: FXKey>(root: AbsolutePath, in ctx: BuildContext<X>, _ callback: (AbsolutePath, LLBDataID, LLBDirectoryEntry) async throws -> Void) async throws {

     

       49
       49
       -
           try await traverse(root: root, in: ctx.db, ctx.context, callback)

     

       50
       50
       -
         }

     

       47
       47
       +
       	public func traverse<X: FXKey>(root: AbsolutePath, in ctx: BuildContext<X>, _ callback: (AbsolutePath, LLBDataID, LLBDirectoryEntry) async throws -> Void) async throws {

     

       48
       48
       +
       		try await traverse(root: root, in: ctx.db, ctx.context, callback)

     

       49
       49
       +
       	}

     

       51
       50
        
       }

+15

Sources/PterodactylBuild/FXValue+Conformances.swift

···

       1
       1
       +
       // SPDX-FileCopyrightText: 2025 The Project Pterodactyl Developers

     

       2
       2
       +
       //

     

       3
       3
       +
       // SPDX-License-Identifier: MPL-2.0

     

       4
       4
       +
       

     

       5
       5
       +
       import llbuild2fx

     

       6
       6
       +
       import PterodactylSyntax

     

       7
       7
       +
       

     

       8
       8
       +
       extension SyntaxTree: FXValue {}

     

       9
       9
       +
       extension Token: FXValue {}

     

       10
       10
       +
       extension Graph: FXValue where Vertex: Codable {}

     

       11
       11
       +
       extension LineMap: FXValue {}

     

       12
       12
       +
       

     

       13
       13
       +
       extension String: @retroactive FXValue {}

     

       14
       14
       +
       extension Set: @retroactive FXValue where Element: Codable {}

     

       15
       15
       +
       extension LLBDataID: @retroactive FXValue {}

-34

Sources/PterodactylBuild/Keys/AnalyseImports.swift

···

       1
       1
       -
       // SPDX-FileCopyrightText: 2025 The Project Pterodactyl Developers

     

       2
       2
       -
       //

     

       3
       3
       -
       // SPDX-License-Identifier: MPL-2.0

     

       4
       4
       -
       

     

       5
       5
       -
       import Foundation

     

       6
       6
       -
       import TSCBasic

     

       7
       7
       -
       import llbuild2fx

     

       8
       8
       -
       

     

       9
       9
       -
       extension Keys {

     

       10
       10
       -
       	struct AnalyseImports: BuildKey {

     

       11
       11
       -
       		typealias ValueType = [UnitName]

     

       12
       12
       -
       		let blobId: LLBDataID

     

       13
       13
       -
       

     

       14
       14
       -
       		func computeValue(_ ctx: BuildContext<Self>) async throws -> [UnitName] {

     

       15
       15
       -
       			let contents = try await ctx.load(blobId)

     

       16
       16
       -
       			let code = try await String(decoding: Data(ctx.read(blob: contents.blob!)), as: UTF8.self)

     

       17
       17
       -
       

     

       18
       18
       -
       			var results: [UnitName] = []

     

       19
       19
       -
       			let lines = code.split(separator: "\n", omittingEmptySubsequences: false)

     

       20
       20
       -
       

     

       21
       21
       -
       			for line in lines {

     

       22
       22
       -
       				let trimmed = line.trimmingCharacters(in: .whitespaces)

     

       23
       23
       -
       				guard trimmed.hasPrefix("import ") else { continue }

     

       24
       24
       -
       				let parts = trimmed.split(separator: " ", maxSplits: 1, omittingEmptySubsequences: true)

     

       25
       25
       -
       				if parts.count == 2 {

     

       26
       26
       -
       					let name = parts[1].trimmingCharacters(in: .whitespaces)

     

       27
       27
       -
       					results.append(UnitName(name: name))

     

       28
       28
       -
       				}

     

       29
       29
       -
       			}

     

       30
       30
       -
       

     

       31
       31
       -
       			return results

     

       32
       32
       -
       		}

     

       33
       33
       -
       	}

     

       34
       34
       -
       }

+24

Sources/PterodactylBuild/Keys/Blob/GetLineMap.swift

···

       1
       1
       +
       // SPDX-FileCopyrightText: 2025 The Project Pterodactyl Developers

     

       2
       2
       +
       //

     

       3
       3
       +
       // SPDX-License-Identifier: MPL-2.0

     

       4
       4
       +
       

     

       5
       5
       +
       import PterodactylSyntax

     

       6
       6
       +
       import llbuild2fx

     

       7
       7
       +
       

     

       8
       8
       +
       extension Keys.Blob {

     

       9
       9
       +
       	public struct GetLineMap: BuildKey {

     

       10
       10
       +
       		public let blobId: LLBDataID

     

       11
       11
       +
       		public init(blobId: LLBDataID) {

     

       12
       12
       +
       			self.blobId = blobId

     

       13
       13
       +
       		}

     

       14
       14
       +
       		

     

       15
       15
       +
       		public typealias ValueType = PterodactylSyntax.LineMap

     

       16
       16
       +
       

     

       17
       17
       +
       		public static let versionDependencies: [any FXVersioning.Type] = [ReadContents.self]

     

       18
       18
       +
       

     

       19
       19
       +
       		public func computeValue(_ ctx: BuildContext<Self>) async throws -> ValueType {

     

       20
       20
       +
       			let code = try await ctx.request(ReadContents(blobId: blobId))

     

       21
       21
       +
       			return PterodactylSyntax.LineMap(source: code)

     

       22
       22
       +
       		}

     

       23
       23
       +
       	}

     

       24
       24
       +
       }

+33

Sources/PterodactylBuild/Keys/Blob/ParseDocument.swift

···

       1
       1
       +
       // SPDX-FileCopyrightText: 2025 The Project Pterodactyl Developers

     

       2
       2
       +
       //

     

       3
       3
       +
       // SPDX-License-Identifier: MPL-2.0

     

       4
       4
       +
       

     

       5
       5
       +
       import PterodactylSyntax

     

       6
       6
       +
       import llbuild2fx

     

       7
       7
       +
       import Logging

     

       8
       8
       +
       

     

       9
       9
       +
       extension Keys.Blob {

     

       10
       10
       +
       	public struct ParseDocument: BuildKey {

     

       11
       11
       +
       		public let blobId: LLBDataID

     

       12
       12
       +
       		public init(blobId: LLBDataID) {

     

       13
       13
       +
       			self.blobId = blobId

     

       14
       14
       +
       		}

     

       15
       15
       +
       

     

       16
       16
       +
       		public struct ValueType: Codable, FXValue {

     

       17
       17
       +
       			public let tree: PterodactylSyntax.SyntaxTree

     

       18
       18
       +
       			public let diagnostics: [Diagnostic]

     

       19
       19
       +
       		}

     

       20
       20
       +
       

     

       21
       21
       +
       		public static let versionDependencies: [any FXVersioning.Type] = [ReadContents.self, Tokenise.self]

     

       22
       22
       +
       

     

       23
       23
       +
       		public func computeValue(_ ctx: BuildContext<Self>) async throws -> ValueType {

     

       24
       24
       +
       			let code = try await ctx.request(ReadContents(blobId: blobId))

     

       25
       25
       +
       			let tokens = try await ctx.request(Tokenise(blobId: blobId))

     

       26
       26
       +
       			var parser = Parser(source: code, tokens: tokens)

     

       27
       27
       +
       			var recoveryHandle = parser.recoveryStack.push([.lineComment, .blockComment(terminated: true), .blockComment(terminated: false)])

     

       28
       28
       +
       			defer { recoveryHandle.close() }

     

       29
       29
       +
       			parser.document()

     

       30
       30
       +
       			return ValueType(tree: parser.tree, diagnostics: parser.diagnostics)

     

       31
       31
       +
       		}

     

       32
       32
       +
       	}

     

       33
       33
       +
       }

+30

Sources/PterodactylBuild/Keys/Blob/ParseImports.swift

···

       1
       1
       +
       // SPDX-FileCopyrightText: 2025 The Project Pterodactyl Developers

     

       2
       2
       +
       //

     

       3
       3
       +
       // SPDX-License-Identifier: MPL-2.0

     

       4
       4
       +
       

     

       5
       5
       +
       import llbuild2fx

     

       6
       6
       +
       import PterodactylSyntax

     

       7
       7
       +
       

     

       8
       8
       +
       extension Keys.Blob {

     

       9
       9
       +
       	struct ParseImports: BuildKey {

     

       10
       10
       +
       		let blobId: LLBDataID

     

       11
       11
       +
       		

     

       12
       12
       +
       		typealias ValueType = [UnitName]

     

       13
       13
       +
       		

     

       14
       14
       +
       		static let versionDependencies: [any FXVersioning.Type] = [ReadContents.self]

     

       15
       15
       +
       

     

       16
       16
       +
       		func computeValue(_ ctx: BuildContext<Self>) async throws -> [UnitName] {

     

       17
       17
       +
       			let code = try await ctx.request(ReadContents(blobId: blobId))

     

       18
       18
       +
       			var importParser = ImportParser(input: code)

     

       19
       19
       +
       			importParser.parseHeader()

     

       20
       20
       +
       			

     

       21
       21
       +
       			return importParser.imports.map { name in

     

       22
       22
       +
       				UnitName(basename: name)

     

       23
       23
       +
       			}

     

       24
       24
       +
       		}

     

       25
       25
       +
       	}

     

       26
       26
       +
       }

     

       27
       27
       +
       

     

       28
       28
       +
       

     

       29
       29
       +
       

     

       30
       30
       +

Sources/PterodactylBuild/Keys/Blob/README.md

···

       1
       1
       +
       <!--

     

       2
       2
       +
       SPDX-FileCopyrightText: 2025 The Project Pterodactyl Developers

     

       3
       3
       +
       

     

       4
       4
       +
       SPDX-License-Identifier: MPL-2.0

     

       5
       5
       +
       -->

     

       6
       6
       +
       

     

       7
       7
       +
       Certain operations do not require knowledge of the entire source tree, only a specific blob inside the source tree. These include import analysis, tokenisation, line maps, parsing, etc.

+25

Sources/PterodactylBuild/Keys/Blob/ReadContents.swift

···

       1
       1
       +
       // SPDX-FileCopyrightText: 2025 The Project Pterodactyl Developers

     

       2
       2
       +
       //

     

       3
       3
       +
       // SPDX-License-Identifier: MPL-2.0

     

       4
       4
       +
       

     

       5
       5
       +
       import Foundation

     

       6
       6
       +
       import llbuild2fx

     

       7
       7
       +
       

     

       8
       8
       +
       extension Keys.Blob {

     

       9
       9
       +
       	public struct ReadContents: BuildKey {

     

       10
       10
       +
       		let blobId: LLBDataID

     

       11
       11
       +
       		

     

       12
       12
       +
       		public init(blobId: LLBDataID) {

     

       13
       13
       +
       			self.blobId = blobId

     

       14
       14
       +
       		}

     

       15
       15
       +
       

     

       16
       16
       +
       		public typealias ValueType = String

     

       17
       17
       +
       

     

       18
       18
       +
       		public static let versionDependencies: [any FXVersioning.Type] = []

     

       19
       19
       +
       

     

       20
       20
       +
       		public func computeValue(_ ctx: BuildContext<Self>) async throws -> String {

     

       21
       21
       +
       			let contents = try await ctx.load(blobId)

     

       22
       22
       +
       			return try await String(decoding: Data(ctx.read(blob: contents.blob!)), as: UTF8.self)

     

       23
       23
       +
       		}

     

       24
       24
       +
       	}

     

       25
       25
       +
       }

+22

Sources/PterodactylBuild/Keys/Blob/Tokenise.swift

···

       1
       1
       +
       // SPDX-FileCopyrightText: 2025 The Project Pterodactyl Developers

     

       2
       2
       +
       //

     

       3
       3
       +
       // SPDX-License-Identifier: MPL-2.0

     

       4
       4
       +
       

     

       5
       5
       +
       import llbuild2fx

     

       6
       6
       +
       import PterodactylSyntax

     

       7
       7
       +
       

     

       8
       8
       +
       extension Keys.Blob {

     

       9
       9
       +
       	struct Tokenise: BuildKey {

     

       10
       10
       +
       		let blobId: LLBDataID

     

       11
       11
       +
       		

     

       12
       12
       +
       		typealias ValueType = [Token]

     

       13
       13
       +
       

     

       14
       14
       +
       		static let versionDependencies: [any FXVersioning.Type] = [ReadContents.self]

     

       15
       15
       +
       

     

       16
       16
       +
       		func computeValue(_ ctx: BuildContext<Self>) async throws -> ValueType {

     

       17
       17
       +
       			let code = try await ctx.request(ReadContents(blobId: blobId))

     

       18
       18
       +
       			var lexer = PterodactylSyntax.Lexer(input: code)

     

       19
       19
       +
       			return lexer.tokenize()

     

       20
       20
       +
       		}

     

       21
       21
       +
       	}

     

       22
       22
       +
       }

-33

Sources/PterodactylBuild/Keys/DependencyGraphOfSourceTree.swift

···

       1
       1
       -
       // SPDX-FileCopyrightText: 2025 The Project Pterodactyl Developers

     

       2
       2
       -
       //

     

       3
       3
       -
       // SPDX-License-Identifier: MPL-2.0

     

       4
       4
       -
       

     

       5
       5
       -
       import Foundation

     

       6
       6
       -
       import TSCBasic

     

       7
       7
       -
       import llbuild2fx

     

       8
       8
       -
       

     

       9
       9
       -
       extension Graph: FXValue where Vertex: Codable {}

     

       10
       10
       -
       

     

       11
       11
       -
       extension Keys {

     

       12
       12
       -
       	struct DependencyGraphOfSourceTree: BuildKey {

     

       13
       13
       -
       		typealias ValueType = Graph<UnitName>

     

       14
       14
       -
       

     

       15
       15
       -
       		let sourceTreeId: LLBDataID

     

       16
       16
       -
       		static let versionDependencies: [any FXVersioning.Type] = [Keys.UnitMapOfSourceTree.self, Keys.AnalyseImports.self]

     

       17
       17
       -
       

     

       18
       18
       -
       		func computeValue(_ ctx: BuildContext<Self>) async throws -> Graph<UnitName> {

     

       19
       19
       -
       			let unitMap = try await ctx.request(Keys.UnitMapOfSourceTree(sourceTreeId: sourceTreeId))

     

       20
       20
       -
       			var edges: [UnitName: Set<UnitName>] = [:]

     

       21
       21
       -
       

     

       22
       22
       -
       			for (unitName, unitInfo) in unitMap.units {

     

       23
       23
       -
       				if edges[unitName] == nil { edges[unitName] = [] }

     

       24
       24
       -
       				let imports = try await ctx.request(Keys.AnalyseImports(blobId: unitInfo.blobId))

     

       25
       25
       -
       				for importedUnitName in imports {

     

       26
       26
       -
       					edges[unitName]!.insert(importedUnitName)

     

       27
       27
       -
       				}

     

       28
       28
       -
       			}

     

       29
       29
       -
       

     

       30
       30
       -
       			return Graph(edges: edges)

     

       31
       31
       -
       		}

     

       32
       32
       -
       	}

     

       33
       33
       -
       }

-7

Sources/PterodactylBuild/Keys/Keys.swift

···

       1
       1
       -
       // SPDX-FileCopyrightText: 2025 The Project Pterodactyl Developers

     

       2
       2
       -
       //

     

       3
       3
       -
       // SPDX-License-Identifier: MPL-2.0

     

       4
       4
       -
       

     

       5
       5
       -
       import Foundation

     

       6
       6
       -
       

     

       7
       7
       -
       enum Keys {}

-35

Sources/PterodactylBuild/Keys/NarrowSourceTree.swift

···

       1
       1
       -
       // SPDX-FileCopyrightText: 2025 The Project Pterodactyl Developers

     

       2
       2
       -
       //

     

       3
       3
       -
       // SPDX-License-Identifier: MPL-2.0

     

       4
       4
       -
       

     

       5
       5
       -
       import Foundation

     

       6
       6
       -
       import TSCBasic

     

       7
       7
       -
       import llbuild2fx

     

       8
       8
       -
       

     

       9
       9
       -
       extension Keys {

     

       10
       10
       -
       	/// Narrows a source tree to just the transitive dependencies of a given unit

     

       11
       11
       -
       	struct NarrowSourceTree: BuildKey {

     

       12
       12
       -
       		struct ValueType: Codable, FXValue {

     

       13
       13
       -
       			let sourceTreeId: LLBDataID

     

       14
       14
       -
       		}

     

       15
       15
       -
       

     

       16
       16
       -
       		let sourceTreeId: LLBDataID

     

       17
       17
       -
       		let unitName: UnitName

     

       18
       18
       -
       

     

       19
       19
       -
       		static let versionDependencies: [any FXVersioning.Type] = [TransitiveDependencies.self, UnitMapOfSourceTree.self]

     

       20
       20
       -
       

     

       21
       21
       -
       		func computeValue(_ ctx: BuildContext<Self>) async throws -> ValueType {

     

       22
       22
       -
       			let dependencies = try await ctx.request(TransitiveDependencies(sourceTreeId: sourceTreeId, unitName: unitName)).dependencies

     

       23
       23
       -
       			let unitMap = try await ctx.request(UnitMapOfSourceTree(sourceTreeId: sourceTreeId))

     

       24
       24
       -
       

     

       25
       25
       -
       			var sourceTree = try await LLBCASFileTree.load(id: sourceTreeId, in: ctx)

     

       26
       26
       -
       			for (unitName, unitInfo) in unitMap.units {

     

       27
       27
       -
       				if !dependencies.contains(unitName) {

     

       28
       28
       -
       					sourceTree = try await sourceTree.remove(path: unitInfo.path, in: ctx)

     

       29
       29
       -
       				}

     

       30
       30
       -
       			}

     

       31
       31
       -
       

     

       32
       32
       -
       			return ValueType(sourceTreeId: sourceTree.id)

     

       33
       33
       -
       		}

     

       34
       34
       -
       	}

     

       35
       35
       -
       }

-32

Sources/PterodactylBuild/Keys/SourceCode.swift

···

       1
       1
       -
       // SPDX-FileCopyrightText: 2025 The Project Pterodactyl Developers

     

       2
       2
       -
       //

     

       3
       3
       -
       // SPDX-License-Identifier: MPL-2.0

     

       4
       4
       -
       

     

       5
       5
       -
       import Foundation

     

       6
       6
       -
       import TSCBasic

     

       7
       7
       -
       import llbuild2fx

     

       8
       8
       -
       

     

       9
       9
       -
       extension Keys {

     

       10
       10
       -
       	struct SourceCode: BuildKey {

     

       11
       11
       -
       		struct ValueType: Codable, FXValue {

     

       12
       12
       -
       			let code: String

     

       13
       13
       -
       		}

     

       14
       14
       -
       

     

       15
       15
       -
       		enum SourceCodeError: Error {

     

       16
       16
       -
       			case unitNotFound

     

       17
       17
       -
       		}

     

       18
       18
       -
       

     

       19
       19
       -
       		let sourceTreeId: LLBDataID

     

       20
       20
       -
       		let unitName: UnitName

     

       21
       21
       -
       

     

       22
       22
       -
       		static let versionDependencies: [any FXVersioning.Type] = [UnitMapOfSourceTree.self]

     

       23
       23
       -
       

     

       24
       24
       -
       		func computeValue(_ ctx: BuildContext<Self>) async throws -> ValueType {

     

       25
       25
       -
       			let unitMap = try await ctx.request(UnitMapOfSourceTree(sourceTreeId: sourceTreeId))

     

       26
       26
       -
       			guard let unitInfo = unitMap.units[unitName] else { throw SourceCodeError.unitNotFound }

     

       27
       27
       -
       			let contents = try await ctx.load(unitInfo.blobId)

     

       28
       28
       -
       			let code = try await String(decoding: Data(ctx.read(blob: contents.blob!)), as: UTF8.self)

     

       29
       29
       -
       			return ValueType(code: code)

     

       30
       30
       -
       		}

     

       31
       31
       -
       	}

     

       32
       32
       -
       }

+21

Sources/PterodactylBuild/Keys/SourceTree/GetDependencies.swift

···

       1
       1
       +
       // SPDX-FileCopyrightText: 2025 The Project Pterodactyl Developers

     

       2
       2
       +
       //

     

       3
       3
       +
       // SPDX-License-Identifier: MPL-2.0

     

       4
       4
       +
       

     

       5
       5
       +
       import llbuild2fx

     

       6
       6
       +
       

     

       7
       7
       +
       extension Keys.SourceTree {

     

       8
       8
       +
       	struct GetDependencies: BuildKey {

     

       9
       9
       +
       		let sourceTreeId: LLBDataID

     

       10
       10
       +
       		let unitName: UnitName

     

       11
       11
       +
       

     

       12
       12
       +
       		typealias ValueType = Set<UnitName>

     

       13
       13
       +
       		

     

       14
       14
       +
       		static let versionDependencies: [any FXVersioning.Type] = [Keys.SourceTree.GetDependencyGraph.self]

     

       15
       15
       +
       

     

       16
       16
       +
       		func computeValue(_ ctx: BuildContext<Self>) async throws -> ValueType {

     

       17
       17
       +
       			let graph = try await ctx.request(Keys.SourceTree.GetDependencyGraph(sourceTreeId: sourceTreeId))

     

       18
       18
       +
       			return graph.verticesReachableFrom(unitName)

     

       19
       19
       +
       		}

     

       20
       20
       +
       	}

     

       21
       21
       +
       }

+30

Sources/PterodactylBuild/Keys/SourceTree/GetDependencyGraph.swift

···

       1
       1
       +
       // SPDX-FileCopyrightText: 2025 The Project Pterodactyl Developers

     

       2
       2
       +
       //

     

       3
       3
       +
       // SPDX-License-Identifier: MPL-2.0

     

       4
       4
       +
       

     

       5
       5
       +
       import llbuild2fx

     

       6
       6
       +
       

     

       7
       7
       +
       extension Keys.SourceTree {

     

       8
       8
       +
       	struct GetDependencyGraph: BuildKey {

     

       9
       9
       +
       		let sourceTreeId: LLBDataID

     

       10
       10
       +
       		

     

       11
       11
       +
       		typealias ValueType = Graph<UnitName>

     

       12
       12
       +
       		

     

       13
       13
       +
       		static let versionDependencies: [any FXVersioning.Type] = [Keys.SourceTree.GetUnitMap.self, Keys.Blob.ParseImports.self]

     

       14
       14
       +
       

     

       15
       15
       +
       		func computeValue(_ ctx: BuildContext<Self>) async throws -> Graph<UnitName> {

     

       16
       16
       +
       			let unitMap = try await ctx.request(Keys.SourceTree.GetUnitMap(sourceTreeId: sourceTreeId))

     

       17
       17
       +
       			var edges: [UnitName: Set<UnitName>] = [:]

     

       18
       18
       +
       

     

       19
       19
       +
       			for (unitName, unitInfo) in unitMap.units {

     

       20
       20
       +
       				if edges[unitName] == nil { edges[unitName] = [] }

     

       21
       21
       +
       				let imports = try await ctx.request(Keys.Blob.ParseImports(blobId: unitInfo.blobId))

     

       22
       22
       +
       				for importedUnitName in imports {

     

       23
       23
       +
       					edges[unitName]!.insert(importedUnitName)

     

       24
       24
       +
       				}

     

       25
       25
       +
       			}

     

       26
       26
       +
       

     

       27
       27
       +
       			return Graph(edges: edges)

     

       28
       28
       +
       		}

     

       29
       29
       +
       	}

     

       30
       30
       +
       }

+24

Sources/PterodactylBuild/Keys/SourceTree/GetUnitMap.swift

···

       1
       1
       +
       // SPDX-FileCopyrightText: 2025 The Project Pterodactyl Developers

     

       2
       2
       +
       //

     

       3
       3
       +
       // SPDX-License-Identifier: MPL-2.0

     

       4
       4
       +
       

     

       5
       5
       +
       import llbuild2fx

     

       6
       6
       +
       

     

       7
       7
       +
       extension Keys.SourceTree {

     

       8
       8
       +
       	struct GetUnitMap: BuildKey {

     

       9
       9
       +
       		let sourceTreeId: LLBDataID

     

       10
       10
       +
       

     

       11
       11
       +
       		typealias ValueType = UnitMap

     

       12
       12
       +
       

     

       13
       13
       +
       		func computeValue(_ ctx: BuildContext<Self>) async throws -> UnitMap {

     

       14
       14
       +
       			let sourceTree = try await LLBCASFileTree.load(id: sourceTreeId, in: ctx)

     

       15
       15
       +
       			var units: [UnitName: UnitInfo] = [:]

     

       16
       16
       +
       			try await sourceTree.traverse(root: .root, in: ctx) { path, blobID, directoryEntry in

     

       17
       17
       +
       				let unitName = UnitName.fromPath(path)

     

       18
       18
       +
       				units[unitName] = UnitInfo(path: path, blobId: blobID)

     

       19
       19
       +
       			}

     

       20
       20
       +
       

     

       21
       21
       +
       			return UnitMap(units: units)

     

       22
       22
       +
       		}

     

       23
       23
       +
       	}

     

       24
       24
       +
       }

+31

Sources/PterodactylBuild/Keys/SourceTree/NarrowToUnit.swift

···

       1
       1
       +
       // SPDX-FileCopyrightText: 2025 The Project Pterodactyl Developers

     

       2
       2
       +
       //

     

       3
       3
       +
       // SPDX-License-Identifier: MPL-2.0

     

       4
       4
       +
       

     

       5
       5
       +
       import llbuild2fx

     

       6
       6
       +
       

     

       7
       7
       +
       extension Keys.SourceTree {

     

       8
       8
       +
       	/// Narrows a source tree to just the transitive dependencies of a given unit

     

       9
       9
       +
       	struct NarrowToUnit: BuildKey {

     

       10
       10
       +
       		let sourceTreeId: LLBDataID

     

       11
       11
       +
       		let unitName: UnitName

     

       12
       12
       +
       

     

       13
       13
       +
       		typealias ValueType = LLBDataID

     

       14
       14
       +
       

     

       15
       15
       +
       		static let versionDependencies: [any FXVersioning.Type] = [GetDependencies.self, GetUnitMap.self]

     

       16
       16
       +
       

     

       17
       17
       +
       		func computeValue(_ ctx: BuildContext<Self>) async throws -> ValueType {

     

       18
       18
       +
       			let dependencies = try await ctx.request(GetDependencies(sourceTreeId: sourceTreeId, unitName: unitName))

     

       19
       19
       +
       			let unitMap = try await ctx.request(GetUnitMap(sourceTreeId: sourceTreeId))

     

       20
       20
       +
       

     

       21
       21
       +
       			var sourceTree = try await LLBCASFileTree.load(id: sourceTreeId, in: ctx)

     

       22
       22
       +
       			for (unitName, unitInfo) in unitMap.units {

     

       23
       23
       +
       				if !dependencies.contains(unitName) {

     

       24
       24
       +
       					sourceTree = try await sourceTree.remove(path: unitInfo.path, in: ctx)

     

       25
       25
       +
       				}

     

       26
       26
       +
       			}

     

       27
       27
       +
       

     

       28
       28
       +
       			return sourceTree.id

     

       29
       29
       +
       		}

     

       30
       30
       +
       	}

     

       31
       31
       +
       }

-25

Sources/PterodactylBuild/Keys/TransitiveDependencies.swift

···

       1
       1
       -
       // SPDX-FileCopyrightText: 2025 The Project Pterodactyl Developers

     

       2
       2
       -
       //

     

       3
       3
       -
       // SPDX-License-Identifier: MPL-2.0

     

       4
       4
       -
       

     

       5
       5
       -
       import Foundation

     

       6
       6
       -
       import TSCBasic

     

       7
       7
       -
       import llbuild2fx

     

       8
       8
       -
       

     

       9
       9
       -
       extension Keys {

     

       10
       10
       -
       	struct TransitiveDependencies: BuildKey {

     

       11
       11
       -
       		struct ValueType: Codable, FXValue {

     

       12
       12
       -
       			var dependencies: Set<UnitName>

     

       13
       13
       -
       		}

     

       14
       14
       -
       

     

       15
       15
       -
       		let sourceTreeId: LLBDataID

     

       16
       16
       -
       		let unitName: UnitName

     

       17
       17
       -
       

     

       18
       18
       -
       		static let versionDependencies: [any FXVersioning.Type] = [Keys.DependencyGraphOfSourceTree.self]

     

       19
       19
       -
       

     

       20
       20
       -
       		func computeValue(_ ctx: BuildContext<Self>) async throws -> ValueType {

     

       21
       21
       -
       			let graph = try await ctx.request(Keys.DependencyGraphOfSourceTree(sourceTreeId: sourceTreeId))

     

       22
       22
       -
       			return ValueType(dependencies: graph.verticesReachableFrom(unitName))

     

       23
       23
       -
       		}

     

       24
       24
       -
       	}

     

       25
       25
       -
       }

-25

Sources/PterodactylBuild/Keys/UnitMapOfSourceTree.swift

···

       1
       1
       -
       // SPDX-FileCopyrightText: 2025 The Project Pterodactyl Developers

     

       2
       2
       -
       //

     

       3
       3
       -
       // SPDX-License-Identifier: MPL-2.0

     

       4
       4
       -
       

     

       5
       5
       -
       import Foundation

     

       6
       6
       -
       import TSCBasic

     

       7
       7
       -
       import llbuild2fx

     

       8
       8
       -
       

     

       9
       9
       -
       extension Keys {

     

       10
       10
       -
       	struct UnitMapOfSourceTree: BuildKey {

     

       11
       11
       -
       		typealias ValueType = UnitMap

     

       12
       12
       -
       		let sourceTreeId: LLBDataID

     

       13
       13
       -
       

     

       14
       14
       -
       		func computeValue(_ ctx: BuildContext<Self>) async throws -> UnitMap {

     

       15
       15
       -
       			let sourceTree = try await LLBCASFileTree.load(id: sourceTreeId, in: ctx)

     

       16
       16
       -
       			var units: [UnitName: UnitInfo] = [:]

     

       17
       17
       -
       			try await sourceTree.traverse(root: .root, in: ctx) { path, blobID, directoryEntry in

     

       18
       18
       -
               let unitName = UnitName.fromPath(path)

     

       19
       19
       -
       				units[unitName] = UnitInfo(path: path, blobId: blobID)

     

       20
       20
       -
       			}

     

       21
       21
       -
       

     

       22
       22
       -
       			return UnitMap(units: units)

     

       23
       23
       -
       		}

     

       24
       24
       -
       	}

     

       25
       25
       -
       }

Sources/PterodactylBuild/Keys.swift

···

       1
       1
       +
       // SPDX-FileCopyrightText: 2025 The Project Pterodactyl Developers

     

       2
       2
       +
       //

     

       3
       3
       +
       // SPDX-License-Identifier: MPL-2.0

     

       4
       4
       +
       

     

       5
       5
       +
       

     

       6
       6
       +
       public enum Keys {

     

       7
       7
       +
       	public enum Blob {}

     

       8
       8
       +
       	public enum SourceTree {}

     

       9
       9
       +
       }

-1

Sources/PterodactylBuild/LLBCASFileTree+Traversal.swift

···

       2
       2
        
       //

     

       3
       3
        
       // SPDX-License-Identifier: MPL-2.0

     

       4
       4
        
       

     

       5
       5
       -
       import Foundation

     

       6
       5
        
       import TSCBasic

     

       7
       6
        
       import llbuild2fx

     

       8
       7

-1

Sources/PterodactylBuild/Types/Graph.swift

···

       2
       2
        
       //

     

       3
       3
        
       // SPDX-License-Identifier: MPL-2.0

     

       4
       4
        
       

     

       5
       5
       -
       import Foundation

     

       6
       5
        
       

     

       7
       6
        
       struct Graph<Vertex: Hashable> {

     

       8
       7
        
       	var edges: [Vertex: Set<Vertex>]

-1

Sources/PterodactylBuild/Types/UnitInfo.swift

···

       2
       2
        
       //

     

       3
       3
        
       // SPDX-License-Identifier: MPL-2.0

     

       4
       4
        
       

     

       5
       5
       -
       import Foundation

     

       6
       5
        
       import TSCBasic

     

       7
       6
        
       import llbuild2fx

     

       8
       7

-1

Sources/PterodactylBuild/Types/UnitMap.swift

···

       2
       2
        
       //

     

       3
       3
        
       // SPDX-License-Identifier: MPL-2.0

     

       4
       4
        
       

     

       5
       5
       -
       import Foundation

     

       6
       5
        
       import llbuild2fx

     

       7
       6
        
       

     

       8
       7
        
       struct UnitMap: Codable, FXValue {

+5 -6

Sources/PterodactylBuild/Types/UnitName.swift

···

       2
       2
        
       //

     

       3
       3
        
       // SPDX-License-Identifier: MPL-2.0

     

       4
       4
        
       

     

       5
       5
       -
       import Foundation

     

       5
       5
       +
       import TSCBasic

     

       6
       6
        
       import llbuild2fx

     

       7
       7
       -
       import TSCBasic

     

       8
       7
        
       

     

       9
       8
        
       struct UnitName: Codable, Equatable, Hashable {

     

       10
       10
       -
       	var name: String

     

       9
       9
       +
       	var basename: String

     

       11
       10
        
       }

     

       12
       11
        
       

     

       13
       12
        
       extension UnitName: FXValue {}

     

       14
       13
        
       

     

       15
       14
        
       extension UnitName {

     

       16
       16
       -
         static func fromPath(_ path: AbsolutePath) -> Self {

     

       17
       17
       -
           Self(name: path.basenameWithoutExt)

     

       18
       18
       -
         }

     

       15
       15
       +
       	static func fromPath(_ path: AbsolutePath) -> Self {

     

       16
       16
       +
       		Self(basename: path.basenameWithoutExt)

     

       17
       17
       +
       	}

     

       19
       18
        
       }

-1

Sources/PterodactylKernel/Control/AsyncThunk.swift

···

       2
       2
        
       //

     

       3
       3
        
       // SPDX-License-Identifier: MPL-2.0

     

       4
       4
        
       

     

       5
       5
       -
       import Foundation

     

       6
       5
        
       

     

       7
       6
        
       public actor AsyncThunk<Value> {

     

       8
       7
        
       	private var storage: Value?

-1

Sources/PterodactylKernel/Core Types/FieldDict.swift

···

       2
       2
        
       //

     

       3
       3
        
       // SPDX-License-Identifier: MPL-2.0

     

       4
       4
        
       

     

       5
       5
       -
       import Foundation

     

       6
       5
        
       

     

       7
       6
        
       // TODO: This might be rewritten. The goal is to provide a dictionary that I can easily construct and traverse in order. Maybe ordered dictionaries from swift-collections?

     

       8
       7
        
       public struct FieldDict<Value> {

-1

Sources/PterodactylKernel/Core Types/Size.swift

···

       2
       2
        
       //

     

       3
       3
        
       // SPDX-License-Identifier: MPL-2.0

     

       4
       4
        
       

     

       5
       5
       -
       import Foundation

     

       6
       5
        
       

     

       7
       6
        
       /// Pterodactyl has two sizes of small things: ``little(universe:)`` and ``big``. Beyond these, we have things that are not types but rather “type schemes”.

     

       8
       7
        
       public enum Size<Universe> {

-1

Sources/PterodactylKernel/Local Analysis/Equality.swift

···

       2
       2
        
       //

     

       3
       3
        
       // SPDX-License-Identifier: MPL-2.0

     

       4
       4
        
       

     

       5
       5
       -
       import Foundation

     

       6
       5
        
       

     

       7
       6
        
       extension LocalAnalysis {

     

       8
       7
        
       	enum EqualityError: Error {

-1

Sources/PterodactylKernel/Local Analysis/LocalAnalysis.swift

···

       2
       2
        
       //

     

       3
       3
        
       // SPDX-License-Identifier: MPL-2.0

     

       4
       4
        
       

     

       5
       5
       -
       import Foundation

     

       6
       5
        
       

     

       7
       6
        
       struct LocalAnalysis {

     

       8
       7
        
       	let depth: Int

-1

Sources/PterodactylKernel/Local Analysis/Quotation.swift

···

       2
       2
        
       //

     

       3
       3
        
       // SPDX-License-Identifier: MPL-2.0

     

       4
       4
        
       

     

       5
       5
       -
       import Foundation

     

       6
       5
        
       

     

       7
       6
        
       extension LocalAnalysis {

     

       8
       7
        
       	func quote(type: Value.Type_) -> Term.Type_ {

-1

Sources/PterodactylKernel/Smallness.swift

···

       2
       2
        
       //

     

       3
       3
        
       // SPDX-License-Identifier: MPL-2.0

     

       4
       4
        
       

     

       5
       5
       -
       import Foundation

     

       6
       5
        
       

     

       7
       6
        
       extension LocalAnalysis {

     

       8
       7
        
       	struct SizeError: Error { }

+19

Sources/PterodactylLanguageServer/AbsolutePath+URI.swift

···

       1
       1
       +
       // SPDX-FileCopyrightText: 2025 The Project Pterodactyl Developers

     

       2
       2
       +
       //

     

       3
       3
       +
       // SPDX-License-Identifier: MPL-2.0

     

       4
       4
       +
       

     

       5
       5
       +
       import Foundation

     

       6
       6
       +
       import TSCBasic

     

       7
       7
       +
       import LanguageServerProtocol

     

       8
       8
       +
       

     

       9
       9
       +
       extension AbsolutePath {

     

       10
       10
       +
       	public static func fromDocumentUri(_ uri: DocumentUri) throws -> Self {

     

       11
       11
       +
       		guard let url = URL(string: uri) else {

     

       12
       12
       +
       			throw URLError(.badURL)

     

       13
       13
       +
       		}

     

       14
       14
       +
       		guard url.isFileURL else {

     

       15
       15
       +
       			throw URLError(.unsupportedURL)

     

       16
       16
       +
       		}

     

       17
       17
       +
       		return try Self(validating: url.path)

     

       18
       18
       +
       	}

     

       19
       19
       +
       }

+16

Sources/PterodactylLanguageServer/Archive/LLBDeclFileTree+Singleton.swift

···

       1
       1
       +
       // SPDX-FileCopyrightText: 2025 The Project Pterodactyl Developers

     

       2
       2
       +
       //

     

       3
       3
       +
       // SPDX-License-Identifier: MPL-2.0

     

       4
       4
       +
       

     

       5
       5
       +
       import TSCBasic

     

       6
       6
       +
       import llbuild2fx

     

       7
       7
       +
       

     

       8
       8
       +
       extension LLBDeclFileTree {

     

       9
       9
       +
       	static func file(absolutePath: AbsolutePath, contents: String) -> Self {

     

       10
       10
       +
       		let components = absolutePath.components

     

       11
       11
       +
       		let leaf = LLBDeclFileTree.file(contents)

     

       12
       12
       +
       		return components.dropFirst().reversed().reduce(leaf) { tail, name in

     

       13
       13
       +
       			.directory(files: [name: tail])

     

       14
       14
       +
       		}

     

       15
       15
       +
       	}

     

       16
       16
       +
       }

+12

Sources/PterodactylLanguageServer/Event Handlers/ErrorHandler.swift

···

       1
       1
       +
       // SPDX-FileCopyrightText: 2025 The Project Pterodactyl Developers

     

       2
       2
       +
       //

     

       3
       3
       +
       // SPDX-License-Identifier: MPL-2.0

     

       4
       4
       +
       

     

       5
       5
       +
       import LanguageServer

     

       6
       6
       +
       import Logging

     

       7
       7
       +
       

     

       8
       8
       +
       struct PterodactylServerErrorHandler: LanguageServer.ErrorHandler {

     

       9
       9
       +
       	func internalError(_ error: any Error) async {

     

       10
       10
       +
       		Logger.shared.error("Received error: \(error)")

     

       11
       11
       +
       	}

     

       12
       12
       +
       }

+56

Sources/PterodactylLanguageServer/Event Handlers/NotificationHandler.swift

···

       1
       1
       +
       // SPDX-FileCopyrightText: 2025 The Project Pterodactyl Developers

     

       2
       2
       +
       //

     

       3
       3
       +
       // SPDX-License-Identifier: MPL-2.0

     

       4
       4
       +
       

     

       5
       5
       +
       import Foundation

     

       6
       6
       +
       import LanguageServer

     

       7
       7
       +
       import LanguageServerProtocol

     

       8
       8
       +
       import Logging

     

       9
       9
       +
       import PterodactylBuild

     

       10
       10
       +
       import llbuild2fx

     

       11
       11
       +
       

     

       12
       12
       +
       struct PterodactylServerNotificationHandler: LanguageServer.NotificationHandler {

     

       13
       13
       +
       	private let actor: PterodactylServerActor

     

       14
       14
       +
       	init(actor: PterodactylServerActor) {

     

       15
       15
       +
       		self.actor = actor

     

       16
       16
       +
       	}

     

       17
       17
       +
       	

     

       18
       18
       +
       	func internalError(_ error: any Error) async {

     

       19
       19
       +
       		Logger.shared.error("Notification handler received error: \(error)")

     

       20
       20
       +
       	}

     

       21
       21
       +
       	

     

       22
       22
       +
       	func textDocumentDidChange(_ params: DidChangeTextDocumentParams) async {

     

       23
       23
       +
       		guard let text = params.contentChanges.first?.text else { return }

     

       24
       24
       +
       		do {

     

       25
       25
       +
       			let blobId: LLBDataID = try await actor.storeBlob(text: text, uri: params.textDocument.uri, version: params.textDocument.version)

     

       26
       26
       +
       			try await actor.publishLiveDiagnostics(blobId: blobId, uri: params.textDocument.uri, version: params.textDocument.version)

     

       27
       27
       +
       		} catch {}

     

       28
       28
       +
       	}

     

       29
       29
       +
       

     

       30
       30
       +
       	func textDocumentDidOpen(_ params: DidOpenTextDocumentParams) async {

     

       31
       31
       +
       		let text = params.textDocument.text

     

       32
       32
       +
       		do {

     

       33
       33
       +
       			let blobId: LLBDataID = try await actor.storeBlob(text: text, uri: params.textDocument.uri, version: params.textDocument.version)

     

       34
       34
       +
       			try await actor.publishLiveDiagnostics(blobId: blobId, uri: params.textDocument.uri, version: params.textDocument.version)

     

       35
       35
       +
       		} catch {}

     

       36
       36
       +
       	}

     

       37
       37
       +
       

     

       38
       38
       +
       	func exit() async {

     

       39
       39
       +
       		Logger.shared.info("Received exit notification")

     

       40
       40
       +
       		Foundation.exit(0)

     

       41
       41
       +
       	}

     

       42
       42
       +
       

     

       43
       43
       +
       	func initialized(_ params: InitializedParams) async {}

     

       44
       44
       +
       	func textDocumentDidClose(_ params: DidCloseTextDocumentParams) async {}

     

       45
       45
       +
       	func textDocumentWillSave(_ params: WillSaveTextDocumentParams) async {}

     

       46
       46
       +
       	func textDocumentDidSave(_ params: DidSaveTextDocumentParams) async {}

     

       47
       47
       +
       	func protocolCancelRequest(_ params: CancelParams) async {}

     

       48
       48
       +
       	func protocolSetTrace(_ params: SetTraceParams) async {}

     

       49
       49
       +
       	func workspaceDidChangeWatchedFiles(_ params: DidChangeWatchedFilesParams) async {}

     

       50
       50
       +
       	func windowWorkDoneProgressCancel(_ params: WorkDoneProgressCancelParams) async {}

     

       51
       51
       +
       	func workspaceDidChangeWorkspaceFolders(_ params: DidChangeWorkspaceFoldersParams) async {}

     

       52
       52
       +
       	func workspaceDidChangeConfiguration(_ params: DidChangeConfigurationParams) async {}

     

       53
       53
       +
       	func workspaceDidCreateFiles(_ params: CreateFilesParams) async {}

     

       54
       54
       +
       	func workspaceDidRenameFiles(_ params: RenameFilesParams) async {}

     

       55
       55
       +
       	func workspaceDidDeleteFiles(_ params: DeleteFilesParams) async {}

     

       56
       56
       +
       }

+153

Sources/PterodactylLanguageServer/Event Handlers/RequestHandler.swift

···

       1
       1
       +
       // SPDX-FileCopyrightText: 2025 The Project Pterodactyl Developers

     

       2
       2
       +
       //

     

       3
       3
       +
       // SPDX-License-Identifier: MPL-2.0

     

       4
       4
       +
       

     

       5
       5
       +
       import JSONRPC

     

       6
       6
       +
       import LanguageServer

     

       7
       7
       +
       import LanguageServerProtocol

     

       8
       8
       +
       import Logging

     

       9
       9
       +
       

     

       10
       10
       +
       struct PterodactylServerRequestHandler: LanguageServer.RequestHandler {

     

       11
       11
       +
       	private let actor: PterodactylServerActor

     

       12
       12
       +
       	init(actor: PterodactylServerActor) {

     

       13
       13
       +
       		self.actor = actor

     

       14
       14
       +
       	}

     

       15
       15
       +
       

     

       16
       16
       +
       	func internalError(_ error: any Error) async {

     

       17
       17
       +
       		Logger.shared.error("Request handler received error: \(error)")

     

       18
       18
       +
       	}

     

       19
       19
       +
       

     

       20
       20
       +
       	// MARK: - Configurations

     

       21
       21
       +
       	var textDocumentSyncOptions: TextDocumentSyncOptions {

     

       22
       22
       +
       		TextDocumentSyncOptions(

     

       23
       23
       +
       			openClose: true,

     

       24
       24
       +
       			change: .full,

     

       25
       25
       +
       			save: .optionA(false)

     

       26
       26
       +
       		)

     

       27
       27
       +
       	}

     

       28
       28
       +
       

     

       29
       29
       +
       	var completionOptions: CompletionOptions {

     

       30
       30
       +
       		CompletionOptions(

     

       31
       31
       +
       			workDoneProgress: false,

     

       32
       32
       +
       			triggerCharacters: [],  // TODO

     

       33
       33
       +
       			allCommitCharacters: nil,

     

       34
       34
       +
       			resolveProvider: false,

     

       35
       35
       +
       			completionItem: nil

     

       36
       36
       +
       		)

     

       37
       37
       +
       	}

     

       38
       38
       +
       

     

       39
       39
       +
       	var semanticTokensLegend: SemanticTokensLegend {

     

       40
       40
       +
       		SemanticTokensLegend(

     

       41
       41
       +
       			tokenTypes: SemanticTokenTypes.allStrings,

     

       42
       42
       +
       			tokenModifiers: SemanticTokenModifiers.allStrings

     

       43
       43
       +
       		)

     

       44
       44
       +
       	}

     

       45
       45
       +
       

     

       46
       46
       +
       	var semanticTokensOptions: SemanticTokensOptions {

     

       47
       47
       +
       		SemanticTokensOptions(

     

       48
       48
       +
       			legend: semanticTokensLegend,

     

       49
       49
       +
       			range: .optionA(true)

     

       50
       50
       +
       		)

     

       51
       51
       +
       	}

     

       52
       52
       +
       

     

       53
       53
       +
       	var serverCapabilities: ServerCapabilities {

     

       54
       54
       +
       		var serverCapabilities = ServerCapabilities()

     

       55
       55
       +
       		serverCapabilities.textDocumentSync = .optionA(textDocumentSyncOptions)

     

       56
       56
       +
       		serverCapabilities.completionProvider = completionOptions

     

       57
       57
       +
       		serverCapabilities.hoverProvider = .optionA(false)

     

       58
       58
       +
       		serverCapabilities.semanticTokensProvider = .optionA(semanticTokensOptions)

     

       59
       59
       +
       		serverCapabilities.foldingRangeProvider = .optionA(true)

     

       60
       60
       +
       		serverCapabilities.inlayHintProvider = .optionA(true)

     

       61
       61
       +
       		return serverCapabilities

     

       62
       62
       +
       	}

     

       63
       63
       +
       

     

       64
       64
       +
       	var serverInfo: ServerInfo {

     

       65
       65
       +
       		ServerInfo(

     

       66
       66
       +
       			name: "PterodactylLanguageServer",

     

       67
       67
       +
       			version: nil

     

       68
       68
       +
       		)

     

       69
       69
       +
       	}

     

       70
       70
       +
       

     

       71
       71
       +
       	// MARK: - Responses

     

       72
       72
       +
       

     

       73
       73
       +
       	func initialize(id: JSONId, params: InitializeParams) async -> Response<InitializationResponse> {

     

       74
       74
       +
       		Logger.shared.debug("Received initialize request")

     

       75
       75
       +
       		let response = InitializationResponse(

     

       76
       76
       +
       			capabilities: serverCapabilities,

     

       77
       77
       +
       			serverInfo: serverInfo

     

       78
       78
       +
       		)

     

       79
       79
       +
       		return .success(response)

     

       80
       80
       +
       	}

     

       81
       81
       +
       

     

       82
       82
       +
       	func semanticTokensFull(id: JSONId, params: SemanticTokensParams) async -> Response<SemanticTokensResponse> {

     

       83
       83
       +
       		guard let tokens = await actor.semanticTokens(uri: params.textDocument.uri, in: nil) else { return .success(nil) }

     

       84
       84
       +
       		return .success(SemanticTokens(resultId: nil, tokens: tokens))

     

       85
       85
       +
       	}

     

       86
       86
       +
       

     

       87
       87
       +
       	func semanticTokensRange(id: JSONId, params: SemanticTokensRangeParams) async -> Response<SemanticTokensResponse> {

     

       88
       88
       +
       		guard let tokens = await actor.semanticTokens(uri: params.textDocument.uri, in: params.range) else { return .success(nil) }

     

       89
       89
       +
       		return .success(SemanticTokens(resultId: nil, tokens: tokens))

     

       90
       90
       +
       	}

     

       91
       91
       +
       

     

       92
       92
       +
       	func foldingRange(id: JSONId, params: FoldingRangeParams) async -> Response<FoldingRangeResponse> {

     

       93
       93
       +
       		guard let cursor = await actor.cursor(uri: params.textDocument.uri)?.cursor else { return .success(nil) }

     

       94
       94
       +
       		return .success(FoldingRangeResponse(cursor.foldingRanges))

     

       95
       95
       +
       	}

     

       96
       96
       +
       

     

       97
       97
       +
       	func inlayHint(id: JSONId, params: InlayHintParams) async -> Response<InlayHintResponse> {

     

       98
       98
       +
       		guard let cursor = await actor.cursor(uri: params.textDocument.uri)?.cursor else { return .success(nil) }

     

       99
       99
       +
       		let lineMap = cursor.lineMap

     

       100
       100
       +
       

     

       101
       101
       +
       		var hints: [(Range<Int>, String)] = []

     

       102
       102
       +
       

     

       103
       103
       +
       		let range = lineMap.range(from: params.range)

     

       104
       104
       +
       		cursor.collectHints(sink: &hints, in: range)

     

       105
       105
       +
       

     

       106
       106
       +
       		let lspHints = hints.map { (range, hint) in

     

       107
       107
       +
       			let position = lineMap.lspPosition(at: range.endIndex)

     

       108
       108
       +
       			return InlayHint(position: position, label: .optionA(hint))

     

       109
       109
       +
       		}

     

       110
       110
       +
       

     

       111
       111
       +
       		return .success(InlayHintResponse(lspHints))

     

       112
       112
       +
       	}

     

       113
       113
       +
       

     

       114
       114
       +
       	func shutdown(id: JSONId) async {}

     

       115
       115
       +
       	func workspaceInlayHintRefresh(id: JSONId) async {}

     

       116
       116
       +
       

     

       117
       117
       +
       	func inlayHintResolve(id: JSONId, params: InlayHint) async -> Response<InlayHintResponse> { return .success(nil) }

     

       118
       118
       +
       	func typeHierarchySupertypes(id: JSONRPC.JSONId, params: TypeHierarchySupertypesParams) async -> Response<TypeHierarchySupertypesResponse> { .success(nil) }

     

       119
       119
       +
       	func typeHierarchySubtypes(id: JSONId, params: TypeHierarchySubtypesParams) async -> Response<TypeHierarchySubtypesResponse> { .success(nil) }

     

       120
       120
       +
       	func workspaceExecuteCommand(id: JSONId, params: ExecuteCommandParams) async -> Response<LSPAny?> { .success(nil) }

     

       121
       121
       +
       	func workspaceWillCreateFiles(id: JSONId, params: CreateFilesParams) async -> Response<WorkspaceEdit?> { .success(nil) }

     

       122
       122
       +
       	func workspaceWillRenameFiles(id: JSONId, params: RenameFilesParams) async -> Response<WorkspaceEdit?> { .success(nil) }

     

       123
       123
       +
       	func workspaceWillDeleteFiles(id: JSONId, params: DeleteFilesParams) async -> Response<WorkspaceEdit?> { .success(nil) }

     

       124
       124
       +
       	func workspaceSymbol(id: JSONId, params: WorkspaceSymbolParams) async -> Response<WorkspaceSymbolResponse> { .success(nil) }

     

       125
       125
       +
       	func textDocumentWillSaveWaitUntil(id: JSONId, params: WillSaveTextDocumentParams) async -> Response<[TextEdit]?> { .success(nil) }

     

       126
       126
       +
       	func completion(id: JSONId, params: CompletionParams) async -> Response<CompletionResponse> { .success(nil) }

     

       127
       127
       +
       	func hover(id: JSONId, params: TextDocumentPositionParams) async -> Response<HoverResponse> { .success(nil) }

     

       128
       128
       +
       	func signatureHelp(id: JSONId, params: TextDocumentPositionParams) async -> Response<SignatureHelpResponse> { .success(nil) }

     

       129
       129
       +
       	func declaration(id: JSONId, params: TextDocumentPositionParams) async -> Response<DeclarationResponse> { .success(nil) }

     

       130
       130
       +
       	func definition(id: JSONId, params: TextDocumentPositionParams) async -> Response<DefinitionResponse> { .success(nil) }

     

       131
       131
       +
       	func typeDefinition(id: JSONId, params: TextDocumentPositionParams) async -> Response<TypeDefinitionResponse> { .success(nil) }

     

       132
       132
       +
       	func implementation(id: JSONId, params: TextDocumentPositionParams) async -> Response<ImplementationResponse> { .success(nil) }

     

       133
       133
       +
       	func documentHighlight(id: JSONId, params: DocumentHighlightParams) async -> Response<DocumentHighlightResponse> { .success(nil) }

     

       134
       134
       +
       	func documentSymbol(id: JSONId, params: DocumentSymbolParams) async -> Response<DocumentSymbolResponse> { .success(nil) }

     

       135
       135
       +
       	func codeAction(id: JSONId, params: CodeActionParams) async -> Response<CodeActionResponse> { .success(nil) }

     

       136
       136
       +
       	func codeLens(id: JSONId, params: CodeLensParams) async -> Response<CodeLensResponse> { .success(nil) }

     

       137
       137
       +
       	func selectionRange(id: JSONId, params: SelectionRangeParams) async -> Response<SelectionRangeResponse> { .success(nil) }

     

       138
       138
       +
       	func linkedEditingRange(id: JSONId, params: LinkedEditingRangeParams) async -> Response<LinkedEditingRangeResponse> { .success(nil) }

     

       139
       139
       +
       	func prepareCallHierarchy(id: JSONId, params: CallHierarchyPrepareParams) async -> Response<CallHierarchyPrepareResponse> { .success(nil) }

     

       140
       140
       +
       	func prepareRename(id: JSONId, params: PrepareRenameParams) async -> Response<PrepareRenameResponse> { .success(nil) }

     

       141
       141
       +
       	func prepareTypeHeirarchy(id: JSONId, params: TypeHierarchyPrepareParams) async -> Response<PrepareTypeHeirarchyResponse> { .success(nil) }

     

       142
       142
       +
       	func rename(id: JSONId, params: RenameParams) async -> Response<RenameResponse> { .success(nil) }

     

       143
       143
       +
       	func documentLink(id: JSONId, params: DocumentLinkParams) async -> Response<DocumentLinkResponse> { .success(nil) }

     

       144
       144
       +
       	func formatting(id: JSONId, params: DocumentFormattingParams) async -> Response<FormattingResult> { .success(nil) }

     

       145
       145
       +
       	func rangeFormatting(id: JSONId, params: DocumentRangeFormattingParams) async -> Response<FormattingResult> { .success(nil) }

     

       146
       146
       +
       	func onTypeFormatting(id: JSONId, params: DocumentOnTypeFormattingParams) async -> Response<FormattingResult> { .success(nil) }

     

       147
       147
       +
       	func references(id: JSONId, params: ReferenceParams) async -> Response<ReferenceResponse> { .success(nil) }

     

       148
       148
       +
       	func moniker(id: JSONId, params: MonikerParams) async -> Response<MonikerResponse> { .success(nil) }

     

       149
       149
       +
       	func semanticTokensFullDelta(id: JSONId, params: SemanticTokensDeltaParams) async -> Response<SemanticTokensDeltaResponse> { .success(nil) }

     

       150
       150
       +
       	func callHierarchyIncomingCalls(id: JSONId, params: CallHierarchyIncomingCallsParams) async -> Response<CallHierarchyIncomingCallsResponse> { .success(nil) }

     

       151
       151
       +
       	func callHierarchyOutgoingCalls(id: JSONId, params: CallHierarchyOutgoingCallsParams) async -> Response<CallHierarchyOutgoingCallsResponse> { .success(nil) }

     

       152
       152
       +
       	func custom(id: JSONId, method: String, params: LSPAny) async -> Response<LSPAny> { .success(nil) }

     

       153
       153
       +
       }

+63

Sources/PterodactylLanguageServer/Logger.swift

···

       1
       1
       +
       // SPDX-FileCopyrightText: 2025 The Project Pterodactyl Developers

     

       2
       2
       +
       //

     

       3
       3
       +
       // SPDX-License-Identifier: MPL-2.0

     

       4
       4
       +
       

     

       5
       5
       +
       import Foundation

     

       6
       6
       +
       import Logging

     

       7
       7
       +
       

     

       8
       8
       +
       struct FileLogHandler: LogHandler {

     

       9
       9
       +
       	private let label: String

     

       10
       10
       +
       	private let fileHandle: FileHandle

     

       11
       11
       +
       	var logLevel: Logger.Level = .info

     

       12
       12
       +
       	var metadata: Logger.Metadata = [:]

     

       13
       13
       +
       

     

       14
       14
       +
       	init(label: String, fileURL: URL) {

     

       15
       15
       +
       		self.label = label

     

       16
       16
       +
       

     

       17
       17
       +
       		// Ensure file exists

     

       18
       18
       +
       		if !FileManager.default.fileExists(atPath: fileURL.path) {

     

       19
       19
       +
       			FileManager.default.createFile(atPath: fileURL.path, contents: nil)

     

       20
       20
       +
       		}

     

       21
       21
       +
       

     

       22
       22
       +
       		// Open file for updating (read/write)

     

       23
       23
       +
       		self.fileHandle = try! FileHandle(forUpdating: fileURL)

     

       24
       24
       +
       		self.fileHandle.seekToEndOfFile()

     

       25
       25
       +
       	}

     

       26
       26
       +
       

     

       27
       27
       +
       	subscript(metadataKey key: String) -> Logger.Metadata.Value? {

     

       28
       28
       +
       		get { metadata[key] }

     

       29
       29
       +
       		set { metadata[key] = newValue }

     

       30
       30
       +
       	}

     

       31
       31
       +
       

     

       32
       32
       +
       	func log(

     

       33
       33
       +
       		level: Logger.Level,

     

       34
       34
       +
       		message: Logger.Message,

     

       35
       35
       +
       		metadata: Logger.Metadata?,

     

       36
       36
       +
       		source: String,

     

       37
       37
       +
       		file: String,

     

       38
       38
       +
       		function: String,

     

       39
       39
       +
       		line: UInt

     

       40
       40
       +
       	) {

     

       41
       41
       +
       		var fullMetadata = self.metadata

     

       42
       42
       +
       		metadata?.forEach { fullMetadata[$0] = $1 }

     

       43
       43
       +
       

     

       44
       44
       +
       		let line = "[\(level)] \(message)\n"

     

       45
       45
       +
       		if let data = line.data(using: .utf8) {

     

       46
       46
       +
       			fileHandle.write(data)

     

       47
       47
       +
       		}

     

       48
       48
       +
       	}

     

       49
       49
       +
       }

     

       50
       50
       +
       

     

       51
       51
       +
       extension Logger {

     

       52
       52
       +
       	static let shared: Self = {

     

       53
       53
       +
       		let logFile = URL(fileURLWithPath: "/tmp/pterodactyl-language-server.log")

     

       54
       54
       +
       

     

       55
       55
       +
       		LoggingSystem.bootstrap { label in

     

       56
       56
       +
       			FileLogHandler(label: label, fileURL: logFile)

     

       57
       57
       +
       		}

     

       58
       58
       +
       

     

       59
       59
       +
       		var logger = Logger(label: "org.pterodactyl.language-server")

     

       60
       60
       +
       		logger.logLevel = .debug

     

       61
       61
       +
       		return logger

     

       62
       62
       +
       	}()

     

       63
       63
       +
       }

+32

Sources/PterodactylLanguageServer/PterodactylServer.swift

···

       1
       1
       +
       // SPDX-FileCopyrightText: 2025 The Project Pterodactyl Developers

     

       2
       2
       +
       //

     

       3
       3
       +
       // SPDX-License-Identifier: MPL-2.0

     

       4
       4
       +
       

     

       5
       5
       +
       import JSONRPC

     

       6
       6
       +
       import LanguageServer

     

       7
       7
       +
       import LanguageServerProtocol

     

       8
       8
       +
       import Logging

     

       9
       9
       +
       

     

       10
       10
       +
       @main

     

       11
       11
       +
       struct PterodactylServer {

     

       12
       12
       +
       	private let connection: JSONRPCClientConnection

     

       13
       13
       +
       	private let dispatcher: LanguageServer.EventDispatcher

     

       14
       14
       +
       

     

       15
       15
       +
       	init() {

     

       16
       16
       +
       		let channel = DataChannel.stdio()

     

       17
       17
       +
       		connection = JSONRPCClientConnection(channel)

     

       18
       18
       +
       		let languageServerActor = PterodactylServerActor(connection: connection)

     

       19
       19
       +
       		dispatcher = LanguageServer.EventDispatcher(

     

       20
       20
       +
       			connection: connection,

     

       21
       21
       +
       			requestHandler: PterodactylServerRequestHandler(actor: languageServerActor),

     

       22
       22
       +
       			notificationHandler: PterodactylServerNotificationHandler(actor: languageServerActor),

     

       23
       23
       +
       			errorHandler: PterodactylServerErrorHandler()

     

       24
       24
       +
       		)

     

       25
       25
       +
       	}

     

       26
       26
       +
       

     

       27
       27
       +
       	static func main() async throws {

     

       28
       28
       +
       		Logger.shared.debug("Starting!")

     

       29
       29
       +
       		let server = Self()

     

       30
       30
       +
       		await server.dispatcher.run()

     

       31
       31
       +
       	}

     

       32
       32
       +
       }

+69

Sources/PterodactylLanguageServer/PterodactylServerActor.swift

···

       1
       1
       +
       // SPDX-FileCopyrightText: 2025 The Project Pterodactyl Developers

     

       2
       2
       +
       //

     

       3
       3
       +
       // SPDX-License-Identifier: MPL-2.0

     

       4
       4
       +
       

     

       5
       5
       +
       import JSONRPC

     

       6
       6
       +
       import LanguageServer

     

       7
       7
       +
       import LanguageServerProtocol

     

       8
       8
       +
       import Logging

     

       9
       9
       +
       import PterodactylBuild

     

       10
       10
       +
       import PterodactylSyntax

     

       11
       11
       +
       import TSCBasic

     

       12
       12
       +
       import llbuild2fx

     

       13
       13
       +
       

     

       14
       14
       +
       actor PterodactylServerActor {

     

       15
       15
       +
       	private let connection: JSONRPCClientConnection

     

       16
       16
       +
       	private let buildEngine: FXEngine

     

       17
       17
       +
       	private let casContext: TSCUtility.Context

     

       18
       18
       +
       	private let casClient: LLBCASFSClient

     

       19
       19
       +
       	

     

       20
       20
       +
       	private var storedBlobs: [DocumentUri: (blobId: LLBDataID, version: Int?)] = [:]

     

       21
       21
       +
       

     

       22
       22
       +
       	init(connection: JSONRPCClientConnection) {

     

       23
       23
       +
       		self.connection = connection

     

       24
       24
       +
       

     

       25
       25
       +
       		let group = LLBMakeDefaultDispatchGroup()

     

       26
       26
       +
       		let db = LLBInMemoryCASDatabase(group: group)

     

       27
       27
       +
       		let functionCache = FXInMemoryFunctionCache(group: group)

     

       28
       28
       +
       		let executor = FXLocalExecutor()

     

       29
       29
       +
       		self.buildEngine = FXEngine(group: group, db: db, functionCache: functionCache, executor: executor)

     

       30
       30
       +
       		self.casClient = LLBCASFSClient(db)

     

       31
       31
       +
       		self.casContext = Context()

     

       32
       32
       +
       	}

     

       33
       33
       +
       	

     

       34
       34
       +
       	func storeBlob(text: String, uri: DocumentUri, version: Int?) async throws -> LLBDataID {

     

       35
       35
       +
       		if let stored = storedBlobs[uri], let version, let storedVersion = stored.version, storedVersion >= version {

     

       36
       36
       +
       			return stored.blobId

     

       37
       37
       +
       		}

     

       38
       38
       +
       

     

       39
       39
       +
       		let blobId: LLBDataID = try await casClient.store(LLBByteBuffer(string: text), casContext).get()

     

       40
       40
       +
       		storedBlobs[uri] = (blobId: blobId, version: version)

     

       41
       41
       +
       		return blobId

     

       42
       42
       +
       	}

     

       43
       43
       +
       

     

       44
       44
       +
       	func publishLiveDiagnostics(blobId: LLBDataID, uri: DocumentUri, version: Int?) async throws {

     

       45
       45
       +
       		let lineMap = try await buildEngine.build(key: Keys.Blob.GetLineMap(blobId: blobId), casContext).get()

     

       46
       46
       +
       		let parseResult = try await buildEngine.build(key: Keys.Blob.ParseDocument(blobId: blobId), casContext).get()

     

       47
       47
       +
       		let diagnostics = parseResult.diagnostics.map { $0.lspDiagnostic(lineMap: lineMap) }

     

       48
       48
       +
       		let publishParams = PublishDiagnosticsParams(uri: uri, version: version, diagnostics: diagnostics)

     

       49
       49
       +
       		try await connection.sendNotification(.textDocumentPublishDiagnostics(publishParams))

     

       50
       50
       +
       	}

     

       51
       51
       +
       

     

       52
       52
       +
       	func cursor(uri: DocumentUri) async -> SyntaxCursor.Options? {

     

       53
       53
       +
       		guard let storedBlob = storedBlobs[uri] else { return nil }

     

       54
       54
       +
       		do {

     

       55
       55
       +
       			let lineMap = try await buildEngine.build(key: Keys.Blob.GetLineMap(blobId: storedBlob.blobId), casContext).get()

     

       56
       56
       +
       			let parseResult = try await buildEngine.build(key: Keys.Blob.ParseDocument(blobId: storedBlob.blobId), casContext).get()

     

       57
       57
       +
       			return SyntaxCursor.Options(lineMap: lineMap, node: .tree(parseResult.tree), utf16Offset: 0)

     

       58
       58
       +
       		} catch {

     

       59
       59
       +
       			return nil

     

       60
       60
       +
       		}

     

       61
       61
       +
       	}

     

       62
       62
       +
       

     

       63
       63
       +
       	func semanticTokens(uri: DocumentUri, in lspRange: LSPRange?) async -> [SemanticToken]? {

     

       64
       64
       +
       		guard let cursor = await cursor(uri: uri)?.cursor else { return nil }

     

       65
       65
       +
       		let lineMap = cursor.lineMap

     

       66
       66
       +
       		let range = lspRange.map { lineMap.range(from: $0 ) }

     

       67
       67
       +
       		return cursor.semanticTokens(in: range)

     

       68
       68
       +
       	}

     

       69
       69
       +
       }

-25

Sources/PterodactylSyntax/Cursor.swift

···

       1
       1
       -
       // SPDX-FileCopyrightText: 2025 The Project Pterodactyl Developers

     

       2
       2
       -
       //

     

       3
       3
       -
       // SPDX-License-Identifier: MPL-2.0

     

       4
       4
       -
       

     

       5
       5
       -
       import Foundation

     

       6
       6
       -
       

     

       7
       7
       -
       public struct Cursor: Sendable {

     

       8
       8
       -
         let node: SyntaxTree.Child

     

       9
       9
       -
       	let utf16Offset: Int

     

       10
       10
       -
       	let children: [Cursor]

     

       11
       11
       -
       	

     

       12
       12
       -
         init(node: SyntaxTree.Child, utf16Offset: Int) {

     

       13
       13
       -
       		self.node = node

     

       14
       14
       -
       		self.utf16Offset = utf16Offset

     

       15
       15
       -
       		

     

       16
       16
       -
       		var children: [Cursor] = []

     

       17
       17
       -
       		var utf16Offset = utf16Offset

     

       18
       18
       -
       		for childNode in node.children {

     

       19
       19
       -
       			children.append(Self(node: childNode, utf16Offset: utf16Offset))

     

       20
       20
       -
       			utf16Offset += childNode.utf16Length

     

       21
       21
       -
       		}

     

       22
       22
       -
       		

     

       23
       23
       -
       		self.children = children

     

       24
       24
       -
       	}

     

       25
       25
       -
       }

+37

Sources/PterodactylSyntax/Diagnostic.swift

···

       1
       1
       +
       // SPDX-FileCopyrightText: 2025 The Project Pterodactyl Developers

     

       2
       2
       +
       //

     

       3
       3
       +
       // SPDX-License-Identifier: MPL-2.0

     

       4
       4
       +
       

     

       5
       5
       +
       import LanguageServerProtocol

     

       6
       6
       +
       

     

       7
       7
       +
       public struct Diagnostic: Equatable, Codable, Sendable {

     

       8
       8
       +
       	public typealias Severity = LanguageServerProtocol.DiagnosticSeverity

     

       9
       9
       +
       

     

       10
       10
       +
       	public let message: String

     

       11
       11
       +
       	public let severity: Severity

     

       12
       12
       +
       	public let absoluteUtf16Range: Range<Int>

     

       13
       13
       +
       

     

       14
       14
       +
       	init(message: String, severity: Severity, absoluteRange: Range<Int>) {

     

       15
       15
       +
       		self.message = message

     

       16
       16
       +
       		self.severity = severity

     

       17
       17
       +
       		self.absoluteUtf16Range = absoluteRange

     

       18
       18
       +
       	}

     

       19
       19
       +
       

     

       20
       20
       +
       	init(message: String, absoluteRange: Range<Int>) {

     

       21
       21
       +
       		self.init(message: message, severity: Severity.error, absoluteRange: absoluteRange)

     

       22
       22
       +
       	}

     

       23
       23
       +
       

     

       24
       24
       +
       	func lspRange(lineMap: LineMap) -> LanguageServerProtocol.LSPRange {

     

       25
       25
       +
       		let start = lineMap.lspPosition(at: absoluteUtf16Range.lowerBound)

     

       26
       26
       +
       		let end = lineMap.lspPosition(at: absoluteUtf16Range.upperBound)

     

       27
       27
       +
       		return LSPRange(start: start, end: end)

     

       28
       28
       +
       	}

     

       29
       29
       +
       

     

       30
       30
       +
       	public func lspDiagnostic(lineMap: LineMap) -> LanguageServerProtocol.Diagnostic {

     

       31
       31
       +
       		LanguageServerProtocol.Diagnostic(

     

       32
       32
       +
       			range: lspRange(lineMap: lineMap),

     

       33
       33
       +
       			severity: severity,

     

       34
       34
       +
       			message: message

     

       35
       35
       +
       		)

     

       36
       36
       +
       	}

     

       37
       37
       +
       }

+61

Sources/PterodactylSyntax/FoldingRanges.swift

···

       1
       1
       +
       // SPDX-FileCopyrightText: 2025 The Project Pterodactyl Developers

     

       2
       2
       +
       //

     

       3
       3
       +
       // SPDX-License-Identifier: MPL-2.0

     

       4
       4
       +
       

     

       5
       5
       +
       import LanguageServerProtocol

     

       6
       6
       +
       

     

       7
       7
       +
       private enum ArraySymmetry {

     

       8
       8
       +
       	case identity

     

       9
       9
       +
       	case reverse

     

       10
       10
       +
       }

     

       11
       11
       +
       

     

       12
       12
       +
       fileprivate extension Array {

     

       13
       13
       +
       	func apply(symmetry: ArraySymmetry) -> any Collection<Element> {

     

       14
       14
       +
       		switch symmetry {

     

       15
       15
       +
       		case .identity: self

     

       16
       16
       +
       		case .reverse: reversed()

     

       17
       17
       +
       		}

     

       18
       18
       +
       	}

     

       19
       19
       +
       }

     

       20
       20
       +
       

     

       21
       21
       +
       extension SyntaxCursor {

     

       22
       22
       +
       	private func firstVisibleNode(under symmetry: ArraySymmetry) -> SyntaxCursor? {

     

       23
       23
       +
       		switch node {

     

       24
       24
       +
       		case .token(let token, _):

     

       25
       25
       +
       			return token.kind.isVisible ? self : nil

     

       26
       26
       +
       		case .tree:

     

       27
       27
       +
       			for child in children.apply(symmetry: symmetry) {

     

       28
       28
       +
       				if let visibleChild = child.firstVisibleNode(under: symmetry) { return visibleChild }

     

       29
       29
       +
       			}

     

       30
       30
       +
       

     

       31
       31
       +
       			return nil

     

       32
       32
       +
       		}

     

       33
       33
       +
       	}

     

       34
       34
       +
       

     

       35
       35
       +
       	private var visibleUtf16Range: Range<Int>? {

     

       36
       36
       +
       		guard

     

       37
       37
       +
       			let firstNode = firstVisibleNode(under: .identity),

     

       38
       38
       +
       			let lastNode = firstVisibleNode(under: .reverse)

     

       39
       39
       +
       		else { return nil }

     

       40
       40
       +
       		return firstNode.utf16Range.lowerBound..<lastNode.utf16Range.upperBound

     

       41
       41
       +
       	}

     

       42
       42
       +
       

     

       43
       43
       +
       	private func collectFoldingRanges(_ sink: inout [FoldingRange]) {

     

       44
       44
       +
       		if let foldingRangeKind = node.tree?.metadata?.delimitedFoldingRangeKind, let visibleUtf16Range {

     

       45
       45
       +
       			let startLocation = lineMap.lspPosition(at: visibleUtf16Range.lowerBound)

     

       46
       46
       +
       			let endLocation = lineMap.lspPosition(at: visibleUtf16Range.upperBound)

     

       47
       47
       +
       			let foldingRange = FoldingRange(startLine: startLocation.line, endLine: endLocation.line, kind: foldingRangeKind)

     

       48
       48
       +
       			sink.append(foldingRange)

     

       49
       49
       +
       		}

     

       50
       50
       +
       

     

       51
       51
       +
       		for child in children {

     

       52
       52
       +
       			child.collectFoldingRanges(&sink)

     

       53
       53
       +
       		}

     

       54
       54
       +
       	}

     

       55
       55
       +
       

     

       56
       56
       +
       	public var foldingRanges: [FoldingRange] {

     

       57
       57
       +
       		var sink: [FoldingRange] = []

     

       58
       58
       +
       		collectFoldingRanges(&sink)

     

       59
       59
       +
       		return sink

     

       60
       60
       +
       	}

     

       61
       61
       +
       }

+110

Sources/PterodactylSyntax/Grammar.swift

···

       1
       1
       +
       // SPDX-FileCopyrightText: 2025 The Project Pterodactyl Developers

     

       2
       2
       +
       //

     

       3
       3
       +
       // SPDX-License-Identifier: MPL-2.0

     

       4
       4
       +
       

     

       5
       5
       +
       

     

       6
       6
       +
       extension SyntaxTreeKind {

     

       7
       7
       +
       	static let hint = SyntaxTreeKind(name: "hint")

     

       8
       8
       +
       }

     

       9
       9
       +
       

     

       10
       10
       +
       /// This is an abstraction of grammatical productions.

     

       11
       11
       +
       public protocol Grammar: Sendable {

     

       12
       12
       +
       	/// The kinds of tree that the production produces.

     

       13
       13
       +
       	static var kinds: [SyntaxTreeKind] { get }

     

       14
       14
       +
       }

     

       15
       15
       +
       

     

       16
       16
       +
       public enum Document: Grammar {

     

       17
       17
       +
       	public static let kind = SyntaxTreeKind(name: "document")

     

       18
       18
       +
       	public static let kinds = [kind]

     

       19
       19
       +
       }

     

       20
       20
       +
       

     

       21
       21
       +
       enum ImportName: Grammar {

     

       22
       22
       +
       	static let kind = SyntaxTreeKind(name: "import.name")

     

       23
       23
       +
       	static let kinds = [kind]

     

       24
       24
       +
       }

     

       25
       25
       +
       

     

       26
       26
       +
       enum Import: Grammar {

     

       27
       27
       +
       	static let kind = SyntaxTreeKind(name: "import")

     

       28
       28
       +
       	static let kinds = [kind]

     

       29
       29
       +
       }

     

       30
       30
       +
       

     

       31
       31
       +
       enum Theory: Grammar {

     

       32
       32
       +
       	static let kind = SyntaxTreeKind(name: "theory")

     

       33
       33
       +
       	static let kinds = [kind]

     

       34
       34
       +
       }

     

       35
       35
       +
       

     

       36
       36
       +
       enum TheoryName: Grammar {

     

       37
       37
       +
       	static let kind = SyntaxTreeKind(name: "theory.name")

     

       38
       38
       +
       	static let kinds = [kind]

     

       39
       39
       +
       }

     

       40
       40
       +
       

     

       41
       41
       +
       enum Declaration: Grammar {

     

       42
       42
       +
       	enum Kinds {

     

       43
       43
       +
       		static let claim = SyntaxTreeKind(name: "decl.claim")

     

       44
       44
       +
       		static let refine = SyntaxTreeKind(name: "decl.refine")

     

       45
       45
       +
       		static let define = SyntaxTreeKind(name: "decl.define")

     

       46
       46
       +
       	}

     

       47
       47
       +
       

     

       48
       48
       +
       	static let kinds = [Kinds.claim, Kinds.refine, Kinds.define]

     

       49
       49
       +
       }

     

       50
       50
       +
       

     

       51
       51
       +
       extension Declaration {

     

       52
       52
       +
       	enum Lhs: Grammar {

     

       53
       53
       +
       		static let kind = SyntaxTreeKind(name: "declaration.lhs")

     

       54
       54
       +
       		static let kinds = [kind]

     

       55
       55
       +
       	}

     

       56
       56
       +
       

     

       57
       57
       +
       	enum Rhs: Grammar {

     

       58
       58
       +
       		static let kind = SyntaxTreeKind(name: "declaration.rhs")

     

       59
       59
       +
       		static let kinds = [kind]

     

       60
       60
       +
       	}

     

       61
       61
       +
       }

     

       62
       62
       +
       

     

       63
       63
       +
       enum TypedBinder: Grammar {

     

       64
       64
       +
       	static let kind = SyntaxTreeKind(name: "typed-binder")

     

       65
       65
       +
       	static let kinds = [kind]

     

       66
       66
       +
       }

     

       67
       67
       +
       

     

       68
       68
       +
       enum DependentFunctionType: Grammar {

     

       69
       69
       +
       	static let kind = SyntaxTreeKind(name: "pi-type")

     

       70
       70
       +
       	static let kinds = [kind]

     

       71
       71
       +
       }

     

       72
       72
       +
       

     

       73
       73
       +
       enum NondependentFunctionType: Grammar {

     

       74
       74
       +
       	static let kind = SyntaxTreeKind(name: "fun-type")

     

       75
       75
       +
       	static let kinds = [kind]

     

       76
       76
       +
       }

     

       77
       77
       +
       

     

       78
       78
       +
       enum Application: Grammar {

     

       79
       79
       +
       	static let kind = SyntaxTreeKind(name: "app")

     

       80
       80
       +
       	static let kinds = [kind]

     

       81
       81
       +
       }

     

       82
       82
       +
       

     

       83
       83
       +
       extension SyntaxView<Document> {

     

       84
       84
       +
       	var imports: [SyntaxView<Import>] { matchingSubviews() }

     

       85
       85
       +
       	var theories: [SyntaxView<Theory>] { matchingSubviews() }

     

       86
       86
       +
       }

     

       87
       87
       +
       

     

       88
       88
       +
       extension SyntaxView<Declaration> {

     

       89
       89
       +
       	var lhs: SyntaxView<Declaration.Lhs>? { matchingSubview() }

     

       90
       90
       +
       	var rhs: SyntaxView<Declaration.Rhs>? { matchingSubview() }

     

       91
       91
       +
       }

     

       92
       92
       +
       

     

       93
       93
       +
       extension SyntaxView<TheoryName> {

     

       94
       94
       +
       	var text: String { cursor.node.text }

     

       95
       95
       +
       }

     

       96
       96
       +
       

     

       97
       97
       +
       extension SyntaxView<Theory> {

     

       98
       98
       +
       	var name: SyntaxView<TheoryName>? { matchingSubview() }

     

       99
       99
       +
       	var decls: [SyntaxView<Declaration>] { matchingSubviews() }

     

       100
       100
       +
       }

     

       101
       101
       +
       

     

       102
       102
       +
       extension SyntaxView<ImportName> {

     

       103
       103
       +
       	var text: String { cursor.node.text }

     

       104
       104
       +
       }

     

       105
       105
       +
       

     

       106
       106
       +
       extension SyntaxView<Import> {

     

       107
       107
       +
       	var name: SyntaxView<ImportName>? {

     

       108
       108
       +
       		matchingSubview()

     

       109
       109
       +
       	}

     

       110
       110
       +
       }

+40

Sources/PterodactylSyntax/ImportParser.swift

···

       1
       1
       +
       // SPDX-FileCopyrightText: 2025 The Project Pterodactyl Developers

     

       2
       2
       +
       //

     

       3
       3
       +
       // SPDX-License-Identifier: MPL-2.0

     

       4
       4
       +
       

     

       5
       5
       +
       

     

       6
       6
       +
       public struct ImportParser {

     

       7
       7
       +
       	private var lexer: PterodactylSyntax.Lexer

     

       8
       8
       +
       	public private(set) var imports: [String] = []

     

       9
       9
       +
       

     

       10
       10
       +
       	public init(input: String) {

     

       11
       11
       +
       		self.lexer = PterodactylSyntax.Lexer(input: input)

     

       12
       12
       +
       	}

     

       13
       13
       +
       

     

       14
       14
       +
       	public mutating func parseHeader() {

     

       15
       15
       +
       		while true {

     

       16
       16
       +
       			let token = nextSignificantToken()

     

       17
       17
       +
       			switch token.kind {

     

       18
       18
       +
       			case .keyword(.import): parseImportStatement()

     

       19
       19
       +
       			default: return

     

       20
       20
       +
       			}

     

       21
       21
       +
       		}

     

       22
       22
       +
       	}

     

       23
       23
       +
       

     

       24
       24
       +
       	/// Returns the next non-whitespace token.

     

       25
       25
       +
       	private mutating func nextSignificantToken() -> Token {

     

       26
       26
       +
       		var token = lexer.scan()

     

       27
       27
       +
       		while token.kind.isTrivia == true {

     

       28
       28
       +
       			token = lexer.scan()

     

       29
       29
       +
       		}

     

       30
       30
       +
       

     

       31
       31
       +
       		return Token(kind: token.kind, text: token.text)

     

       32
       32
       +
       	}

     

       33
       33
       +
       

     

       34
       34
       +
       	/// Parses a single `import xyz` line.

     

       35
       35
       +
       	private mutating func parseImportStatement() {

     

       36
       36
       +
       		let next = nextSignificantToken()

     

       37
       37
       +
       		guard next.kind == .identifier else { return }

     

       38
       38
       +
       		imports.append(next.text)

     

       39
       39
       +
       	}

     

       40
       40
       +
       }

+263

Sources/PterodactylSyntax/Lexer.swift

···

       1
       1
       +
       // SPDX-FileCopyrightText: 2025 The Project Pterodactyl Developers

     

       2
       2
       +
       //

     

       3
       3
       +
       // SPDX-License-Identifier: MPL-2.0

     

       4
       4
       +
       

     

       5
       5
       +
       

     

       6
       6
       +
       // This is based on a tutorial by Amélia Liao: https://amelia.how/posts/parsing-layout.html

     

       7
       7
       +
       public struct Lexer {

     

       8
       8
       +
       	enum Layout {

     

       9
       9
       +
       		case explicit

     

       10
       10
       +
       		case layoutColumn(Int)

     

       11
       11
       +
       	}

     

       12
       12
       +
       

     

       13
       13
       +
       	enum Mode {

     

       14
       14
       +
       		case normal

     

       15
       15
       +
       		case layout

     

       16
       16
       +
       		case emptyLayout

     

       17
       17
       +
       		case newline

     

       18
       18
       +
       	}

     

       19
       19
       +
       

     

       20
       20
       +
       	struct ModeStack {

     

       21
       21
       +
       		private var stack: [Mode] = []

     

       22
       22
       +
       

     

       23
       23
       +
       		mutating func push(mode: Mode) {

     

       24
       24
       +
       			stack.append(mode)

     

       25
       25
       +
       		}

     

       26
       26
       +
       

     

       27
       27
       +
       		mutating func drop() {

     

       28
       28
       +
       			_ = stack.popLast()

     

       29
       29
       +
       		}

     

       30
       30
       +
       

     

       31
       31
       +
       		var current: Mode {

     

       32
       32
       +
       			stack.last ?? .normal

     

       33
       33
       +
       		}

     

       34
       34
       +
       	}

     

       35
       35
       +
       

     

       36
       36
       +
       	private let input: String

     

       37
       37
       +
       	private var index: String.Index

     

       38
       38
       +
       	private var layoutStack: [Layout]

     

       39
       39
       +
       	private var modeStack: ModeStack

     

       40
       40
       +
       	private var inputColumn: Int

     

       41
       41
       +
       

     

       42
       42
       +
       	public init(input: String) {

     

       43
       43
       +
       		self.input = input

     

       44
       44
       +
       		self.index = input.startIndex

     

       45
       45
       +
       		self.modeStack = ModeStack()

     

       46
       46
       +
       		self.layoutStack = []

     

       47
       47
       +
       		self.inputColumn = 0

     

       48
       48
       +
       	}

     

       49
       49
       +
       

     

       50
       50
       +
       	private var isAtEnd: Bool { index >= input.endIndex }

     

       51
       51
       +
       

     

       52
       52
       +
       	private var peek: Character? {

     

       53
       53
       +
       		guard index < input.endIndex else { return nil }

     

       54
       54
       +
       		return input[index]

     

       55
       55
       +
       	}

     

       56
       56
       +
       

     

       57
       57
       +
       	private func lookahead() -> Character? {

     

       58
       58
       +
       		guard index < input.endIndex else { return nil }

     

       59
       59
       +
       		let next = input.index(after: index)

     

       60
       60
       +
       		guard next < input.endIndex else { return nil }

     

       61
       61
       +
       		return input[next]

     

       62
       62
       +
       	}

     

       63
       63
       +
       

     

       64
       64
       +
       	private mutating func advance() -> Character {

     

       65
       65
       +
       		let c = input[index]

     

       66
       66
       +
       		if c.isNewline {

     

       67
       67
       +
       			inputColumn = 0

     

       68
       68
       +
       		} else {

     

       69
       69
       +
       			inputColumn += 1

     

       70
       70
       +
       		}

     

       71
       71
       +
       		index = input.index(after: index)

     

       72
       72
       +
       		return c

     

       73
       73
       +
       	}

     

       74
       74
       +
       

     

       75
       75
       +
       	private mutating func consume(while predicate: (Character) -> Bool) {

     

       76
       76
       +
       		while let c = peek, predicate(c) {

     

       77
       77
       +
       			_ = advance()

     

       78
       78
       +
       		}

     

       79
       79
       +
       	}

     

       80
       80
       +
       

     

       81
       81
       +
       	func text(from start: String.Index) -> String {

     

       82
       82
       +
       		let range = start..<index

     

       83
       83
       +
       		return String(input[range])

     

       84
       84
       +
       	}

     

       85
       85
       +
       

     

       86
       86
       +
       	var isAtNewline: Bool {

     

       87
       87
       +
       		peek?.isNewline ?? false

     

       88
       88
       +
       	}

     

       89
       89
       +
       

     

       90
       90
       +
       	var isAtLineComment: Bool {

     

       91
       91
       +
       		guard let c = peek else { return false }

     

       92
       92
       +
       		return c == "/" && lookahead() == "/"

     

       93
       93
       +
       	}

     

       94
       94
       +
       

     

       95
       95
       +
       	var isAtBlockComment: Bool {

     

       96
       96
       +
       		guard let c = peek else { return false }

     

       97
       97
       +
       		return c == "/" && lookahead() == "*"

     

       98
       98
       +
       	}

     

       99
       99
       +
       

     

       100
       100
       +
       	mutating func scanNormal() -> Token {

     

       101
       101
       +
       		guard let c = peek else { return eof() }

     

       102
       102
       +
       		let start = index

     

       103
       103
       +
       		if c.isWhitespace && !c.isNewline {

     

       104
       104
       +
       			consume { $0.isWhitespace && !$0.isNewline }

     

       105
       105
       +
       			return Token(kind: .whitespace, text: text(from: start))

     

       106
       106
       +
       		} else if c == "<" && lookahead() == "=" {

     

       107
       107
       +
       			_ = advance()

     

       108
       108
       +
       			_ = advance()

     

       109
       109
       +
       			return Token(kind: .punctuation(.doubleLeftArrow), text: text(from: start))

     

       110
       110
       +
       		} else if c == "-" && lookahead() == ">" {

     

       111
       111
       +
       			_ = advance()

     

       112
       112
       +
       			_ = advance()

     

       113
       113
       +
       			return Token(kind: .punctuation(.rightArrow), text: text(from: start))

     

       114
       114
       +
       		} else if c == "=" && lookahead() == ">" {

     

       115
       115
       +
       			_ = advance()

     

       116
       116
       +
       			_ = advance()

     

       117
       117
       +
       			return Token(kind: .punctuation(.doubleRightArrow), text: text(from: start))

     

       118
       118
       +
       		} else if let punct = Punctuation(rawValue: String(c)) {

     

       119
       119
       +
       			_ = advance()

     

       120
       120
       +
       			return Token(kind: .punctuation(punct), text: String(c))

     

       121
       121
       +
       		} else if c.isLetter || c == "_" {

     

       122
       122
       +
       			_ = advance()

     

       123
       123
       +
       			consume { $0.isLetter || $0.isNumber || $0 == "_" }

     

       124
       124
       +
       			let text = text(from: start)

     

       125
       125
       +
       			if let keyword = Keyword(rawValue: text) {

     

       126
       126
       +
       				if keyword.isBlockHerald { modeStack.push(mode: .layout) }

     

       127
       127
       +
       				return Token(kind: .keyword(keyword), text: text)

     

       128
       128
       +
       			}

     

       129
       129
       +
       			return Token(kind: .identifier, text: text)

     

       130
       130
       +
       		} else if isAtNewline || isAtLineComment || isAtBlockComment {

     

       131
       131
       +
       			modeStack.push(mode: .newline)

     

       132
       132
       +
       			return scan()

     

       133
       133
       +
       		} else {

     

       134
       134
       +
       			let c = advance()

     

       135
       135
       +
       			return Token(kind: .error, text: String(c))

     

       136
       136
       +
       		}

     

       137
       137
       +
       	}

     

       138
       138
       +
       

     

       139
       139
       +
       	mutating func scanLayout() -> Token {

     

       140
       140
       +
       		if let token = scanNewlineOrComment() { return token }

     

       141
       141
       +
       		return startLayout()

     

       142
       142
       +
       	}

     

       143
       143
       +
       

     

       144
       144
       +
       	mutating func scanLineComment() -> Token {

     

       145
       145
       +
       		let start = index

     

       146
       146
       +
       		_ = advance()  // consume '/'

     

       147
       147
       +
       		_ = advance()  // consume '/'

     

       148
       148
       +
       		consume { $0 != "\n" }

     

       149
       149
       +
       		return Token(kind: .lineComment, text: text(from: start))

     

       150
       150
       +
       	}

     

       151
       151
       +
       

     

       152
       152
       +
       	mutating func scanBlockComment() -> Token {

     

       153
       153
       +
       		let start = index

     

       154
       154
       +
       		_ = advance()  // consume '/'

     

       155
       155
       +
       		_ = advance()  // consume '*'

     

       156
       156
       +
       		var terminated = false

     

       157
       157
       +
       

     

       158
       158
       +
       		while let ch = peek {

     

       159
       159
       +
       			if ch == "*" && lookahead() == "/" {

     

       160
       160
       +
       				_ = advance()  // consume '*'

     

       161
       161
       +
       				_ = advance()  // consume '/'

     

       162
       162
       +
       				terminated = true

     

       163
       163
       +
       				break

     

       164
       164
       +
       			}

     

       165
       165
       +
       			_ = advance()

     

       166
       166
       +
       		}

     

       167
       167
       +
       

     

       168
       168
       +
       		return Token(kind: .blockComment(terminated: terminated), text: text(from: start))

     

       169
       169
       +
       	}

     

       170
       170
       +
       

     

       171
       171
       +
       	mutating func scanNewlineOrComment() -> Token? {

     

       172
       172
       +
       		guard let c = peek else { return eof() }

     

       173
       173
       +
       		if isAtNewline {

     

       174
       174
       +
       			_ = advance()

     

       175
       175
       +
       			return Token(kind: .newline, text: String(c))

     

       176
       176
       +
       		} else if isAtLineComment {

     

       177
       177
       +
       			return scanLineComment()

     

       178
       178
       +
       		} else if isAtBlockComment {

     

       179
       179
       +
       			return scanBlockComment()

     

       180
       180
       +
       		} else {

     

       181
       181
       +
       			return nil

     

       182
       182
       +
       		}

     

       183
       183
       +
       	}

     

       184
       184
       +
       

     

       185
       185
       +
       	mutating func scan() -> Token {

     

       186
       186
       +
       		switch modeStack.current {

     

       187
       187
       +
       		case .normal: return scanNormal()

     

       188
       188
       +
       		case .layout: return scanLayout()

     

       189
       189
       +
       		case .emptyLayout: return emptyLayout()

     

       190
       190
       +
       		case .newline: return scanNewlineOrComment() ?? offsideRule()

     

       191
       191
       +
       		}

     

       192
       192
       +
       	}

     

       193
       193
       +
       

     

       194
       194
       +
       	mutating func eof() -> Token {

     

       195
       195
       +
       		if layoutStack.last == nil {

     

       196
       196
       +
       			modeStack.drop()

     

       197
       197
       +
       			return Token(kind: .eof, text: "")

     

       198
       198
       +
       		} else {

     

       199
       199
       +
       			_ = layoutStack.popLast()

     

       200
       200
       +
       			return Token(kind: .blockEnd, text: "")

     

       201
       201
       +
       		}

     

       202
       202
       +
       	}

     

       203
       203
       +
       

     

       204
       204
       +
       	mutating func startLayout() -> Token {

     

       205
       205
       +
       		modeStack.drop()

     

       206
       206
       +
       

     

       207
       207
       +
       		let column = inputColumn

     

       208
       208
       +
       		let reference = layoutStack.last

     

       209
       209
       +
       		let layout = Layout.layoutColumn(column)

     

       210
       210
       +
       

     

       211
       211
       +
       		if let reference, layout <= reference {

     

       212
       212
       +
       			modeStack.push(mode: .emptyLayout)

     

       213
       213
       +
       		} else {

     

       214
       214
       +
       			layoutStack.append(layout)

     

       215
       215
       +
       		}

     

       216
       216
       +
       

     

       217
       217
       +
       		return Token(kind: .blockBegin, text: "")

     

       218
       218
       +
       	}

     

       219
       219
       +
       

     

       220
       220
       +
       	mutating func emptyLayout() -> Token {

     

       221
       221
       +
       		modeStack.drop()

     

       222
       222
       +
       		modeStack.push(mode: .newline)

     

       223
       223
       +
       		return Token(kind: .blockEnd, text: "")

     

       224
       224
       +
       	}

     

       225
       225
       +
       

     

       226
       226
       +
       	mutating func offsideRule() -> Token {

     

       227
       227
       +
       		let context = layoutStack.last

     

       228
       228
       +
       		let column = inputColumn

     

       229
       229
       +
       

     

       230
       230
       +
       		switch context {

     

       231
       231
       +
       		case .layoutColumn(let otherColumn):

     

       232
       232
       +
       			if column == otherColumn {

     

       233
       233
       +
       				modeStack.drop()

     

       234
       234
       +
       				return Token(kind: .blockSep, text: "")

     

       235
       235
       +
       			} else if column > otherColumn {

     

       236
       236
       +
       				modeStack.drop()

     

       237
       237
       +
       				return scan()

     

       238
       238
       +
       			} else {

     

       239
       239
       +
       				_ = layoutStack.popLast()

     

       240
       240
       +
       				return Token(kind: .blockEnd, text: "")

     

       241
       241
       +
       			}

     

       242
       242
       +
       		default:

     

       243
       243
       +
       			modeStack.drop()

     

       244
       244
       +
       			return scan()

     

       245
       245
       +
       		}

     

       246
       246
       +
       	}

     

       247
       247
       +
       

     

       248
       248
       +
       	public mutating func tokenize() -> [Token] {

     

       249
       249
       +
       		var tokens: [Token] = []

     

       250
       250
       +
       

     

       251
       251
       +
       		while true {

     

       252
       252
       +
       			let token = scan()

     

       253
       253
       +
       

     

       254
       254
       +
       			tokens.append(token)

     

       255
       255
       +
       			if token.kind == .eof {

     

       256
       256
       +
       				return tokens

     

       257
       257
       +
       			}

     

       258
       258
       +
       		}

     

       259
       259
       +
       	}

     

       260
       260
       +
       }

     

       261
       261
       +
       

     

       262
       262
       +
       extension Lexer.Layout: Comparable {

     

       263
       263
       +
       }

+41

Sources/PterodactylSyntax/LineMap.swift

···

       1
       1
       +
       // SPDX-FileCopyrightText: 2025 The Project Pterodactyl Developers

     

       2
       2
       +
       //

     

       3
       3
       +
       // SPDX-License-Identifier: MPL-2.0

     

       4
       4
       +
       

     

       5
       5
       +
       import Algorithms

     

       6
       6
       +
       import LanguageServerProtocol

     

       7
       7
       +
       

     

       8
       8
       +
       public struct LineMap: Codable, Sendable {

     

       9
       9
       +
       	private let utf16LineOffsets: [Int]

     

       10
       10
       +
       

     

       11
       11
       +
       	public init(source: String) {

     

       12
       12
       +
       		var offsets: [Int] = [0]

     

       13
       13
       +
       		for idx in source.indices {

     

       14
       14
       +
       			if source[idx].isNewline {

     

       15
       15
       +
       				let next = source.index(after: idx)

     

       16
       16
       +
       				let utf16Offset = next.utf16Offset(in: source)

     

       17
       17
       +
       				offsets.append(utf16Offset)

     

       18
       18
       +
       			}

     

       19
       19
       +
       		}

     

       20
       20
       +
       

     

       21
       21
       +
       		self.utf16LineOffsets = offsets

     

       22
       22
       +
       	}

     

       23
       23
       +
       

     

       24
       24
       +
       	public func lspPosition(at utf16Offset: Int) -> LanguageServerProtocol.Position {

     

       25
       25
       +
       		let partitioningIndex = utf16LineOffsets.partitioningIndex { $0 > utf16Offset }

     

       26
       26
       +
       		let lineIndex = partitioningIndex == 0 ? 0 : partitioningIndex - 1

     

       27
       27
       +
       		let lineStart = utf16LineOffsets[lineIndex]

     

       28
       28
       +
       		let lineNumber = lineIndex

     

       29
       29
       +
       		let columnNumber = utf16Offset - lineStart

     

       30
       30
       +
       		return Position(line: lineNumber, character: columnNumber)

     

       31
       31
       +
       	}

     

       32
       32
       +
       	

     

       33
       33
       +
       	public func range(from lspRange: LSPRange) -> Range<Int> {

     

       34
       34
       +
       		offset(at: lspRange.start)..<offset(at: lspRange.end)

     

       35
       35
       +
       	}

     

       36
       36
       +
       

     

       37
       37
       +
       	public func offset(at position: LanguageServerProtocol.Position) -> Int {

     

       38
       38
       +
       		let lineOffset = utf16LineOffsets[position.line]

     

       39
       39
       +
       		return lineOffset + position.character

     

       40
       40
       +
       	}

     

       41
       41
       +
       }

+269

Sources/PterodactylSyntax/ParseState.swift

···

       1
       1
       +
       // SPDX-FileCopyrightText: 2025 The Project Pterodactyl Developers

     

       2
       2
       +
       //

     

       3
       3
       +
       // SPDX-License-Identifier: MPL-2.0

     

       4
       4
       +
       

     

       5
       5
       +
       

     

       6
       6
       +
       public enum ControlFlow<Value> {

     

       7
       7
       +
       	case commit(Value)

     

       8
       8
       +
       	case abort

     

       9
       9
       +
       }

     

       10
       10
       +
       

     

       11
       11
       +
       extension ControlFlow: Equatable where Value: Equatable {}

     

       12
       12
       +
       

     

       13
       13
       +
       extension ControlFlow where Value == Void {

     

       14
       14
       +
       	static var commit: Self {

     

       15
       15
       +
       		.commit(())

     

       16
       16
       +
       	}

     

       17
       17
       +
       

     

       18
       18
       +
       	var isCommit: Bool {

     

       19
       19
       +
       		switch self {

     

       20
       20
       +
       		case .commit: true

     

       21
       21
       +
       		default: false

     

       22
       22
       +
       		}

     

       23
       23
       +
       	}

     

       24
       24
       +
       }

     

       25
       25
       +
       

     

       26
       26
       +
       

     

       27
       27
       +
       public class RecoveryStack {

     

       28
       28
       +
       	var stack: [Set<TokenKind>] = []

     

       29
       29
       +
       

     

       30
       30
       +
       	var current: Set<TokenKind> {

     

       31
       31
       +
       		stack.last ?? []

     

       32
       32
       +
       	}

     

       33
       33
       +
       

     

       34
       34
       +
       	public struct Handle: ~Copyable {

     

       35
       35
       +
       		let parent: RecoveryStack

     

       36
       36
       +
       		private var closed = false

     

       37
       37
       +
       

     

       38
       38
       +
       		init(parent: RecoveryStack, set: Set<TokenKind>) {

     

       39
       39
       +
       			self.parent = parent

     

       40
       40
       +
       			parent.stack.append(parent.current.union(set))

     

       41
       41
       +
       		}

     

       42
       42
       +
       

     

       43
       43
       +
       		mutating public func close() {

     

       44
       44
       +
       			precondition(!closed)

     

       45
       45
       +
       			_ = parent.stack.popLast()

     

       46
       46
       +
       			closed = true

     

       47
       47
       +
       		}

     

       48
       48
       +
       

     

       49
       49
       +
       		deinit {

     

       50
       50
       +
       			precondition(closed)

     

       51
       51
       +
       		}

     

       52
       52
       +
       	}

     

       53
       53
       +
       

     

       54
       54
       +
       	public func push(_ set: Set<TokenKind>) -> Handle {

     

       55
       55
       +
       		return Handle(parent: self, set: set)

     

       56
       56
       +
       	}

     

       57
       57
       +
       }

     

       58
       58
       +
       

     

       59
       59
       +
       public struct Parser {

     

       60
       60
       +
       	let source: String

     

       61
       61
       +
       	let tokens: [Token]

     

       62
       62
       +
       

     

       63
       63
       +
       	public init(source: String, tokens: [Token]) {

     

       64
       64
       +
       		self.source = source

     

       65
       65
       +
       		self.tokens = tokens

     

       66
       66
       +
       	}

     

       67
       67
       +
       

     

       68
       68
       +
       	public private(set) var diagnostics: [Diagnostic] = []

     

       69
       69
       +
       

     

       70
       70
       +
       	private let builder = SyntaxTreeBuilder()

     

       71
       71
       +
       	public var tree: SyntaxTree { builder.tree }

     

       72
       72
       +
       

     

       73
       73
       +
       	public let recoveryStack = RecoveryStack()

     

       74
       74
       +
       

     

       75
       75
       +
       	private var inError: Bool = false

     

       76
       76
       +
       	public private(set) var position: Int = 0

     

       77
       77
       +
       	private var absoluteUtf16Offset: Int = 0

     

       78
       78
       +
       

     

       79
       79
       +
       	public var absoluteRangeAtCursor: Range<Int> {

     

       80
       80
       +
       		return absoluteUtf16Offset..<absoluteUtf16Offset

     

       81
       81
       +
       	}

     

       82
       82
       +
       

     

       83
       83
       +
       	public var absoluteRangeOfCurrentToken: Range<Int> {

     

       84
       84
       +
       		return absoluteUtf16Offset..<absoluteUtf16Offset + currentToken.utf16Length

     

       85
       85
       +
       	}

     

       86
       86
       +
       

     

       87
       87
       +
       	public var isEndOfFile: Bool {

     

       88
       88
       +
       		position == tokens.count

     

       89
       89
       +
       	}

     

       90
       90
       +
       

     

       91
       91
       +
       	public var currentToken: Token {

     

       92
       92
       +
       		if tokens.indices.contains(position) {

     

       93
       93
       +
       			return tokens[position]

     

       94
       94
       +
       		} else {

     

       95
       95
       +
       			return Token(

     

       96
       96
       +
       				kind: .eof,

     

       97
       97
       +
       				text: "",

     

       98
       98
       +
       			)

     

       99
       99
       +
       		}

     

       100
       100
       +
       	}

     

       101
       101
       +
       

     

       102
       102
       +
       	public var nextSignificantToken: Token? {

     

       103
       103
       +
       		guard tokens.indices.contains(position) else { return nil }

     

       104
       104
       +
       		for index in position..<tokens.endIndex {

     

       105
       105
       +
       			let token = tokens[index]

     

       106
       106
       +
       			if !token.kind.isTrivia { return token }

     

       107
       107
       +
       		}

     

       108
       108
       +
       

     

       109
       109
       +
       		return nil

     

       110
       110
       +
       	}

     

       111
       111
       +
       

     

       112
       112
       +
       	public func isRoughlyAt(kind: TokenKind) -> Bool {

     

       113
       113
       +
       		nextSignificantToken?.kind == kind

     

       114
       114
       +
       	}

     

       115
       115
       +
       

     

       116
       116
       +
       	public func isAt(kind: TokenKind) -> Bool {

     

       117
       117
       +
       		currentToken.kind == kind

     

       118
       118
       +
       	}

     

       119
       119
       +
       

     

       120
       120
       +
       	public func isAt(kindSatisfying predicate: (TokenKind) -> Bool) -> Bool {

     

       121
       121
       +
       		return predicate(currentToken.kind)

     

       122
       122
       +
       	}

     

       123
       123
       +
       

     

       124
       124
       +
       

     

       125
       125
       +
       	mutating public func advance(metadata: TokenMetadata?) {

     

       126
       126
       +
       		precondition(!isEndOfFile)

     

       127
       127
       +
       		builder.advance(token: currentToken, metadata: metadata)

     

       128
       128
       +
       		absoluteUtf16Offset += currentToken.utf16Length

     

       129
       129
       +
       		position += 1

     

       130
       130
       +
       	}

     

       131
       131
       +
       

     

       132
       132
       +
       	public mutating func advance(error: String, metadata: TokenMetadata? = nil) {

     

       133
       133
       +
       		var handle = builder.open()

     

       134
       134
       +
       		defer { handle.close() }

     

       135
       135
       +
       		let diagnostic = Diagnostic(

     

       136
       136
       +
       			message: error,

     

       137
       137
       +
       			absoluteRange: absoluteRangeOfCurrentToken

     

       138
       138
       +
       		)

     

       139
       139
       +
       

     

       140
       140
       +
       		diagnostics.append(diagnostic)

     

       141
       141
       +
       		advance(metadata: metadata)

     

       142
       142
       +
       	}

     

       143
       143
       +
       

     

       144
       144
       +
       	public mutating func eat(kind: TokenKind, metadata: TokenMetadata?) -> Bool {

     

       145
       145
       +
       		guard !isEndOfFile && isAt(kindSatisfying: { $0 == kind }) else { return false }

     

       146
       146
       +
       		advance(metadata: metadata)

     

       147
       147
       +
       		return true

     

       148
       148
       +
       	}

     

       149
       149
       +
       

     

       150
       150
       +
       

     

       151
       151
       +
       	private mutating func ate(kind: TokenKind, metadata: TokenMetadata?) -> Bool {

     

       152
       152
       +
       		guard eat(kind: kind, metadata: metadata) else { return false }

     

       153
       153
       +
       		inError = false

     

       154
       154
       +
       		return true

     

       155
       155
       +
       	}

     

       156
       156
       +
       

     

       157
       157
       +
       	mutating func recoverUntil(_ anchors: Set<TokenKind>, expected: [TokenKind], error: String? = nil) {

     

       158
       158
       +
       		var discardTokens: [Token] = []

     

       159
       159
       +
       		let startOffset = absoluteUtf16Offset

     

       160
       160
       +
       

     

       161
       161
       +
       		while !self.isAt(kindSatisfying: { anchors.contains($0) }) {

     

       162
       162
       +
       			if isEndOfFile { break }

     

       163
       163
       +
       			let token = currentToken

     

       164
       164
       +
       			advance(metadata: nil)

     

       165
       165
       +
       			discardTokens.append(token)

     

       166
       166
       +
       		}

     

       167
       167
       +
       

     

       168
       168
       +
       		var endOffset = startOffset

     

       169
       169
       +
       

     

       170
       170
       +
       		let message = error ?? "Expected one of \(expected.map(String.init(describing:)).sorted(using: .localized)) but got \(discardTokens)"

     

       171
       171
       +
       

     

       172
       172
       +
       		if discardTokens.isEmpty {

     

       173
       173
       +
       			if !inError {

     

       174
       174
       +
       				inError = true

     

       175
       175
       +
       				diagnostics.append(Diagnostic(message: message, absoluteRange: absoluteRangeAtCursor))

     

       176
       176
       +
       			}

     

       177
       177
       +
       			return

     

       178
       178
       +
       		} else {

     

       179
       179
       +
       			do {

     

       180
       180
       +
       				var handle = builder.open()

     

       181
       181
       +
       				defer { handle.close() }

     

       182
       182
       +
       

     

       183
       183
       +
       				for discardToken in discardTokens {

     

       184
       184
       +
       					endOffset += discardToken.utf16Length

     

       185
       185
       +
       				}

     

       186
       186
       +
       			}

     

       187
       187
       +
       

     

       188
       188
       +
       			if !inError {

     

       189
       189
       +
       				inError = true

     

       190
       190
       +
       				diagnostics.append(Diagnostic(message: message, absoluteRange: startOffset..<endOffset))

     

       191
       191
       +
       			}

     

       192
       192
       +
       		}

     

       193
       193
       +
       	}

     

       194
       194
       +
       

     

       195
       195
       +
       	mutating func token(kind: TokenKind, metadata: TokenMetadata? = nil, error: String? = nil) {

     

       196
       196
       +
       		var anchors = recoveryStack.current

     

       197
       197
       +
       		if ate(kind: kind, metadata: metadata) { return }

     

       198
       198
       +
       		anchors.insert(kind)

     

       199
       199
       +
       		recoverUntil(anchors, expected: [kind], error: error)

     

       200
       200
       +
       		let _ = ate(kind: kind, metadata: metadata)

     

       201
       201
       +
       	}

     

       202
       202
       +
       

     

       203
       203
       +
       	mutating func token<Result>(choices: [TokenKind: (TokenMetadata?, Result)]) -> Result? {

     

       204
       204
       +
       		var anchors = recoveryStack.current

     

       205
       205
       +
       

     

       206
       206
       +
       		var result: Result? = nil

     

       207
       207
       +
       		for choice in choices {

     

       208
       208
       +
       			if ate(kind: choice.key, metadata: choice.value.0) {

     

       209
       209
       +
       				result = choice.value.1

     

       210
       210
       +
       				break

     

       211
       211
       +
       			}

     

       212
       212
       +
       		}

     

       213
       213
       +
       

     

       214
       214
       +
       		if let result { return result }

     

       215
       215
       +
       

     

       216
       216
       +
       		for kind in choices.keys {

     

       217
       217
       +
       			anchors.insert(kind)

     

       218
       218
       +
       		}

     

       219
       219
       +
       

     

       220
       220
       +
       		recoverUntil(anchors, expected: Array(choices.keys))

     

       221
       221
       +
       		for choice in choices {

     

       222
       222
       +
       			if ate(kind: choice.key, metadata: choice.value.0) {

     

       223
       223
       +
       				result = choice.value.1

     

       224
       224
       +
       				break

     

       225
       225
       +
       			}

     

       226
       226
       +
       		}

     

       227
       227
       +
       

     

       228
       228
       +
       		return result

     

       229
       229
       +
       	}

     

       230
       230
       +
       

     

       231
       231
       +
       	mutating func tryToken(kind: TokenKind, metadata: TokenMetadata? = nil) -> ControlFlow<Void> {

     

       232
       232
       +
       		guard eat(kind: kind, metadata: metadata) else { return .abort }

     

       233
       233
       +
       		return .commit

     

       234
       234
       +
       	}

     

       235
       235
       +
       

     

       236
       236
       +
       	func structure(kind: SyntaxTreeKind? = nil, metadata: SyntaxTreeMetadata? = nil) -> SyntaxTreeBuilder.SubtreeHandle {

     

       237
       237
       +
       		return builder.open(kind: kind, metadata: metadata)

     

       238
       238
       +
       	}

     

       239
       239
       +
       }

     

       240
       240
       +
       

     

       241
       241
       +
       extension Parser {

     

       242
       242
       +
       	mutating func eatTrivium() -> Bool {

     

       243
       243
       +
       		switch currentToken.kind {

     

       244
       244
       +
       		case .whitespace, .newline:

     

       245
       245
       +
       			advance(metadata: nil)

     

       246
       246
       +
       			return true

     

       247
       247
       +
       		case .blockComment(let terminated):

     

       248
       248
       +
       			let metadata = TokenMetadata(

     

       249
       249
       +
       				semanticTokenType: .comment,

     

       250
       250
       +
       				delimitedFoldingRangeKind: .comment

     

       251
       251
       +
       			)

     

       252
       252
       +
       			if terminated {

     

       253
       253
       +
       				advance(metadata: metadata)

     

       254
       254
       +
       			} else {

     

       255
       255
       +
       				advance(error: "Block comment was not terminated")

     

       256
       256
       +
       			}

     

       257
       257
       +
       			return true

     

       258
       258
       +
       		case .lineComment:

     

       259
       259
       +
       			advance(metadata: TokenMetadata(semanticTokenType: .comment))

     

       260
       260
       +
       			return true

     

       261
       261
       +
       		default:

     

       262
       262
       +
       			return false

     

       263
       263
       +
       		}

     

       264
       264
       +
       	}

     

       265
       265
       +
       

     

       266
       266
       +
       	mutating func eatTrivia() {

     

       267
       267
       +
       		while !isEndOfFile && eatTrivium() {}

     

       268
       268
       +
       	}

     

       269
       269
       +
       }

+287

Sources/PterodactylSyntax/Parser.swift

···

       1
       1
       +
       // SPDX-FileCopyrightText: 2025 The Project Pterodactyl Developers

     

       2
       2
       +
       //

     

       3
       3
       +
       // SPDX-License-Identifier: MPL-2.0

     

       4
       4
       +
       

     

       5
       5
       +
       

     

       6
       6
       +
       extension Parser {

     

       7
       7
       +
       	/// This will consume at least one ``TokenKind.blockSep``, possibly also consuming some block items containing only trivia.

     

       8
       8
       +
       	mutating func tryBlockSep() -> ControlFlow<Void> {

     

       9
       9
       +
       		var status: ControlFlow<Void> = .abort

     

       10
       10
       +
       		while case .commit = tryToken(kind: .blockSep) {

     

       11
       11
       +
       			status = .commit

     

       12
       12
       +
       			eatTrivia()

     

       13
       13
       +
       		}

     

       14
       14
       +
       		return status

     

       15
       15
       +
       	}

     

       16
       16
       +
       

     

       17
       17
       +
       	/// This will consume ``TokenKind.blockBegin`` followed by any number of trivial block items.

     

       18
       18
       +
       	mutating func tryBeginBlock() -> ControlFlow<Void> {

     

       19
       19
       +
       		guard isAt(kind: .blockBegin) else { return .abort }

     

       20
       20
       +
       		token(kind: .blockBegin)

     

       21
       21
       +
       		eatTrivia()

     

       22
       22
       +
       		_ = tryBlockSep()

     

       23
       23
       +
       		return .commit

     

       24
       24
       +
       	}

     

       25
       25
       +
       

     

       26
       26
       +
       	mutating func endBlock() {

     

       27
       27
       +
       		token(kind: .blockEnd)

     

       28
       28
       +
       	}

     

       29
       29
       +
       

     

       30
       30
       +
       	mutating func declLhs() {

     

       31
       31
       +
       		var handle = structure(kind: Declaration.Lhs.kind)

     

       32
       32
       +
       		defer { handle.close() }

     

       33
       33
       +
       		token(kind: .identifier, metadata: TokenMetadata(semanticTokenType: .function))

     

       34
       34
       +
       	}

     

       35
       35
       +
       

     

       36
       36
       +
       	struct Scanner {

     

       37
       37
       +
       		let tokens: [Token]

     

       38
       38
       +
       		var position: Int

     

       39
       39
       +
       

     

       40
       40
       +
       		var currentToken: Token { tokens[position] }

     

       41
       41
       +
       

     

       42
       42
       +
       		mutating func eat(kindSatisfying predicate: (TokenKind) -> Bool) -> Bool {

     

       43
       43
       +
       			guard position < tokens.count, predicate(currentToken.kind) else { return false }

     

       44
       44
       +
       			position += 1

     

       45
       45
       +
       			return true

     

       46
       46
       +
       		}

     

       47
       47
       +
       

     

       48
       48
       +
       		mutating func eat(kind: TokenKind) -> Bool {

     

       49
       49
       +
       			eat(kindSatisfying: { $0 == kind })

     

       50
       50
       +
       		}

     

       51
       51
       +
       

     

       52
       52
       +
       		mutating func eatTrivia() {

     

       53
       53
       +
       			while position < tokens.count, currentToken.kind.isTrivia { position += 1 }

     

       54
       54
       +
       		}

     

       55
       55
       +
       	}

     

       56
       56
       +
       

     

       57
       57
       +
       

     

       58
       58
       +
       	var isAtTypedBinder: Bool {

     

       59
       59
       +
       		var scanner = Scanner(tokens: tokens, position: self.position)

     

       60
       60
       +
       		guard scanner.eat(kind: .punctuation(.lparen)) else { return false }

     

       61
       61
       +
       		scanner.eatTrivia()

     

       62
       62
       +
       		guard scanner.eat(kind: .identifier) else { return false }

     

       63
       63
       +
       		scanner.eatTrivia()

     

       64
       64
       +
       		guard scanner.eat(kind: .punctuation(.colon)) else { return false }

     

       65
       65
       +
       		return true

     

       66
       66
       +
       	}

     

       67
       67
       +
       

     

       68
       68
       +
       	mutating func typedBinder() {

     

       69
       69
       +
       		var handle = structure(kind: TypedBinder.kind)

     

       70
       70
       +
       		defer { handle.close() }

     

       71
       71
       +
       

     

       72
       72
       +
       		token(kind: .punctuation(.lparen))

     

       73
       73
       +
       		eatTrivia()

     

       74
       74
       +
       		token(kind: .identifier)

     

       75
       75
       +
       		eatTrivia()

     

       76
       76
       +
       		token(kind: .punctuation(.colon))

     

       77
       77
       +
       		eatTrivia()

     

       78
       78
       +
       		expression()

     

       79
       79
       +
       		eatTrivia()

     

       80
       80
       +
       		token(kind: .punctuation(.rparen))

     

       81
       81
       +
       	}

     

       82
       82
       +
       

     

       83
       83
       +
       	mutating func dependentFunctionType() -> ControlFlow<Void> {

     

       84
       84
       +
       		guard isAtTypedBinder else { return .abort }

     

       85
       85
       +
       

     

       86
       86
       +
       		var handle = structure(kind: DependentFunctionType.kind)

     

       87
       87
       +
       		defer { handle.close() }

     

       88
       88
       +
       

     

       89
       89
       +
       		while isAtTypedBinder {

     

       90
       90
       +
       			typedBinder()

     

       91
       91
       +
       			eatTrivia()

     

       92
       92
       +
       		}

     

       93
       93
       +
       

     

       94
       94
       +
       		token(kind: .punctuation(.rightArrow), metadata: TokenMetadata(semanticTokenType: .keyword))

     

       95
       95
       +
       		eatTrivia()

     

       96
       96
       +
       		expression()

     

       97
       97
       +
       

     

       98
       98
       +
       		return .commit

     

       99
       99
       +
       	}

     

       100
       100
       +
       

     

       101
       101
       +
       	mutating func identifier() -> ControlFlow<Void> {

     

       102
       102
       +
       		guard isAt(kind: .identifier) else { return .abort }

     

       103
       103
       +
       		token(kind: .identifier)

     

       104
       104
       +
       		return .commit

     

       105
       105
       +
       	}

     

       106
       106
       +
       

     

       107
       107
       +
       	mutating func parenthesisedExpression() -> ControlFlow<Void> {

     

       108
       108
       +
       		guard isAt(kind: .punctuation(.lparen)) else { return .abort }

     

       109
       109
       +
       		token(kind: .punctuation(.lparen))

     

       110
       110
       +
       		eatTrivia()

     

       111
       111
       +
       		expression()

     

       112
       112
       +
       		eatTrivia()

     

       113
       113
       +
       		token(kind: .punctuation(.rparen))

     

       114
       114
       +
       		return .commit

     

       115
       115
       +
       	}

     

       116
       116
       +
       

     

       117
       117
       +
       	mutating func applicationExpression() {

     

       118
       118
       +
       		var handle = structure(kind: Application.kind)

     

       119
       119
       +
       		defer { handle.close() }

     

       120
       120
       +
       

     

       121
       121
       +
       		atomicExpression()

     

       122
       122
       +
       		eatTrivia()

     

       123
       123
       +
       

     

       124
       124
       +
       		var isApplication = false

     

       125
       125
       +
       		while case .commit = tryAtomicExpression() {

     

       126
       126
       +
       			eatTrivia()

     

       127
       127
       +
       			isApplication = true

     

       128
       128
       +
       		}

     

       129
       129
       +
       

     

       130
       130
       +
       		if !isApplication { handle.cancel() }

     

       131
       131
       +
       	}

     

       132
       132
       +
       

     

       133
       133
       +
       	mutating func tryAtomicExpression() -> ControlFlow<Void> {

     

       134
       134
       +
       		guard case .abort = identifier() else { return .commit }

     

       135
       135
       +
       		guard case .abort = parenthesisedExpression() else { return .commit }

     

       136
       136
       +
       		return .abort

     

       137
       137
       +
       	}

     

       138
       138
       +
       

     

       139
       139
       +
       	mutating func atomicExpression() {

     

       140
       140
       +
       		guard case .commit = tryAtomicExpression() else {

     

       141
       141
       +
       			return advance(error: "Expected atomic expression")

     

       142
       142
       +
       		}

     

       143
       143
       +
       	}

     

       144
       144
       +
       

     

       145
       145
       +
       	mutating func expression() {

     

       146
       146
       +
       		if case .commit = dependentFunctionType() { return }

     

       147
       147
       +
       

     

       148
       148
       +
       		var handle = structure()

     

       149
       149
       +
       		defer { handle.close() }

     

       150
       150
       +
       

     

       151
       151
       +
       		applicationExpression()

     

       152
       152
       +
       

     

       153
       153
       +
       		var scanner = Scanner(tokens: tokens, position: position)

     

       154
       154
       +
       		scanner.eatTrivia()

     

       155
       155
       +
       		if scanner.eat(kind: .punctuation(.rightArrow)) {

     

       156
       156
       +
       			handle.kind = NondependentFunctionType.kind

     

       157
       157
       +
       			eatTrivia()

     

       158
       158
       +
       			token(kind: .punctuation(.rightArrow), metadata: TokenMetadata(semanticTokenType: .keyword))

     

       159
       159
       +
       			eatTrivia()

     

       160
       160
       +
       			expression()

     

       161
       161
       +
       		} else {

     

       162
       162
       +
       			handle.cancel()

     

       163
       163
       +
       		}

     

       164
       164
       +
       	}

     

       165
       165
       +
       

     

       166
       166
       +
       	mutating func declRhs(declKind: SyntaxTreeKind) {

     

       167
       167
       +
       		var handle = structure(kind: Declaration.Rhs.kind)

     

       168
       168
       +
       		defer { handle.close() }

     

       169
       169
       +
       

     

       170
       170
       +
       		switch declKind {

     

       171
       171
       +
       		case Declaration.Kinds.claim:

     

       172
       172
       +
       			expression()

     

       173
       173
       +
       		case Declaration.Kinds.define:

     

       174
       174
       +
       			token(kind: .identifier, metadata: TokenMetadata(semanticTokenType: .variable))  // TODO: expr

     

       175
       175
       +
       		case Declaration.Kinds.refine:

     

       176
       176
       +
       			token(kind: .identifier, metadata: TokenMetadata(semanticTokenType: .keyword))  // TODO

     

       177
       177
       +
       		default: return

     

       178
       178
       +
       		}

     

       179
       179
       +
       	}

     

       180
       180
       +
       

     

       181
       181
       +
       	mutating func tryDecl() -> ControlFlow<Void> {

     

       182
       182
       +
       		guard isAt(kind: .identifier) else { return .abort }

     

       183
       183
       +
       

     

       184
       184
       +
       		var handle = structure()

     

       185
       185
       +
       		defer { handle.close() }

     

       186
       186
       +
       

     

       187
       187
       +
       		declLhs()

     

       188
       188
       +
       

     

       189
       189
       +
       		eatTrivia()

     

       190
       190
       +
       

     

       191
       191
       +
       		let choices: [TokenKind: (TokenMetadata?, SyntaxTreeKind)] = [

     

       192
       192
       +
       			.punctuation(.doubleRightArrow): Declaration.Kinds.define,

     

       193
       193
       +
       			.punctuation(.doubleLeftArrow): Declaration.Kinds.refine,

     

       194
       194
       +
       			.punctuation(.colon): Declaration.Kinds.claim

     

       195
       195
       +
       		].mapValues { (TokenMetadata(semanticTokenType: .keyword), $0) }

     

       196
       196
       +
       

     

       197
       197
       +
       		handle.kind = token(choices: choices) ?? .error

     

       198
       198
       +
       

     

       199
       199
       +
       		eatTrivia()

     

       200
       200
       +
       

     

       201
       201
       +
       		declRhs(declKind: handle.kind)

     

       202
       202
       +
       

     

       203
       203
       +
       		return .commit

     

       204
       204
       +
       	}

     

       205
       205
       +
       

     

       206
       206
       +
       	mutating func keyword(_ keyword: Keyword) {

     

       207
       207
       +
       		token(kind: .keyword(keyword), metadata: TokenMetadata(semanticTokenType: .keyword))

     

       208
       208
       +
       	}

     

       209
       209
       +
       

     

       210
       210
       +
       	mutating func theoryName() {

     

       211
       211
       +
       		var handle = structure(kind: TheoryName.kind)

     

       212
       212
       +
       		defer { handle.close() }

     

       213
       213
       +
       		token(kind: .identifier, metadata: TokenMetadata(semanticTokenType: .interface))

     

       214
       214
       +
       	}

     

       215
       215
       +
       

     

       216
       216
       +
       	mutating func theoryBlock() {

     

       217
       217
       +
       		var handle = recoveryStack.push([.blockSep, .blockEnd])

     

       218
       218
       +
       		defer { handle.close() }

     

       219
       219
       +
       

     

       220
       220
       +
       		guard case .commit = tryBeginBlock() else { return }

     

       221
       221
       +
       		eatTrivia()

     

       222
       222
       +
       

     

       223
       223
       +
       		guard case .commit = tryDecl() else { return }

     

       224
       224
       +
       		eatTrivia()

     

       225
       225
       +
       

     

       226
       226
       +
       		while true {

     

       227
       227
       +
       			guard case .commit = tryBlockSep() else { break }

     

       228
       228
       +
       			_ = tryDecl()

     

       229
       229
       +
       			eatTrivia()

     

       230
       230
       +
       		}

     

       231
       231
       +
       

     

       232
       232
       +
       		eatTrivia()

     

       233
       233
       +
       		endBlock()

     

       234
       234
       +
       	}

     

       235
       235
       +
       

     

       236
       236
       +
       	mutating func theory() -> ControlFlow<Void> {

     

       237
       237
       +
       		guard isAt(kind: .keyword(.theory)) else { return .abort }

     

       238
       238
       +
       

     

       239
       239
       +
       		var handle = structure(kind: Theory.kind, metadata: SyntaxTreeMetadata(delimitedFoldingRangeKind: .region))

     

       240
       240
       +
       		defer { handle.close() }

     

       241
       241
       +
       

     

       242
       242
       +
       		eatTrivia()

     

       243
       243
       +
       		keyword(.theory)

     

       244
       244
       +
       		eatTrivia()

     

       245
       245
       +
       		theoryName()

     

       246
       246
       +
       		eatTrivia()

     

       247
       247
       +
       		keyword(.where)

     

       248
       248
       +
       		eatTrivia()

     

       249
       249
       +
       		theoryBlock()

     

       250
       250
       +
       

     

       251
       251
       +
       		return .commit

     

       252
       252
       +
       	}

     

       253
       253
       +
       

     

       254
       254
       +
       	mutating func importName() {

     

       255
       255
       +
       		var handle = structure(kind: TheoryName.kind)

     

       256
       256
       +
       		defer { handle.close() }

     

       257
       257
       +
       		token(kind: .identifier)

     

       258
       258
       +
       	}

     

       259
       259
       +
       

     

       260
       260
       +
       	mutating func importDecl() -> ControlFlow<Void> {

     

       261
       261
       +
       		guard isAt(kind: .keyword(.import)) else { return .abort }

     

       262
       262
       +
       

     

       263
       263
       +
       		var handle = structure(kind: Import.kind)

     

       264
       264
       +
       		defer { handle.close() }

     

       265
       265
       +
       

     

       266
       266
       +
       		keyword(.import)

     

       267
       267
       +
       		eatTrivia()

     

       268
       268
       +
       		importName()

     

       269
       269
       +
       

     

       270
       270
       +
       		return .commit

     

       271
       271
       +
       	}

     

       272
       272
       +
       

     

       273
       273
       +
       	mutating public func document() {

     

       274
       274
       +
       		var handle = structure(kind: Document.kind)

     

       275
       275
       +
       		defer { handle.close() }

     

       276
       276
       +
       

     

       277
       277
       +
       		while true {

     

       278
       278
       +
       			guard case .commit = importDecl() else { break }

     

       279
       279
       +
       		}

     

       280
       280
       +
       

     

       281
       281
       +
       		while true {

     

       282
       282
       +
       			guard case .commit = theory() else { break }

     

       283
       283
       +
       		}

     

       284
       284
       +
       

     

       285
       285
       +
       		token(kind: .eof)

     

       286
       286
       +
       	}

     

       287
       287
       +
       }

+97

Sources/PterodactylSyntax/SemanticToken.swift

···

       1
       1
       +
       // SPDX-FileCopyrightText: 2025 The Project Pterodactyl Developers

     

       2
       2
       +
       //

     

       3
       3
       +
       // SPDX-License-Identifier: MPL-2.0

     

       4
       4
       +
       

     

       5
       5
       +
       import LanguageServerProtocol

     

       6
       6
       +
       

     

       7
       7
       +
       extension SemanticTokenTypes {

     

       8
       8
       +
       	public var index: Int {

     

       9
       9
       +
       		Self.allCases.firstIndex(of: self)!

     

       10
       10
       +
       	}

     

       11
       11
       +
       }

     

       12
       12
       +
       

     

       13
       13
       +
       extension SemanticTokenModifiers {

     

       14
       14
       +
       	private static let modifierBitPositions: [SemanticTokenModifiers: Int] = {

     

       15
       15
       +
       		var dict: [SemanticTokenModifiers: Int] = [:]

     

       16
       16
       +
       		for (i, modifier) in SemanticTokenModifiers.allCases.enumerated() {

     

       17
       17
       +
       			dict[modifier] = i

     

       18
       18
       +
       		}

     

       19
       19
       +
       		return dict

     

       20
       20
       +
       	}()

     

       21
       21
       +
       

     

       22
       22
       +
       	static func encodeBitset(_ modifiers: Set<SemanticTokenModifiers>) -> UInt32 {

     

       23
       23
       +
       		var bitset: UInt32 = 0

     

       24
       24
       +
       		for modifier in modifiers {

     

       25
       25
       +
       			if let bit = modifierBitPositions[modifier] {

     

       26
       26
       +
       				bitset |= (1 << bit)

     

       27
       27
       +
       			}

     

       28
       28
       +
       		}

     

       29
       29
       +
       		return bitset

     

       30
       30
       +
       	}

     

       31
       31
       +
       }

     

       32
       32
       +
       

     

       33
       33
       +
       

     

       34
       34
       +
       struct SingleLineRange {

     

       35
       35
       +
       	let line: Int

     

       36
       36
       +
       	let char: Int

     

       37
       37
       +
       	let length: Int

     

       38
       38
       +
       }

     

       39
       39
       +
       

     

       40
       40
       +
       

     

       41
       41
       +
       extension TokenMetadata {

     

       42
       42
       +
       	func semanticToken(range: SingleLineRange) -> SemanticToken? {

     

       43
       43
       +
       		guard range.length > 0 else { return nil }

     

       44
       44
       +
       		return SemanticToken(

     

       45
       45
       +
       			line: UInt32(range.line),

     

       46
       46
       +
       			char: UInt32(range.char),

     

       47
       47
       +
       			length: UInt32(range.length),

     

       48
       48
       +
       			type: UInt32(semanticTokenType.index),

     

       49
       49
       +
       			modifiers: SemanticTokenModifiers.encodeBitset(semanticTokenModifiers)

     

       50
       50
       +
       		)

     

       51
       51
       +
       	}

     

       52
       52
       +
       }

     

       53
       53
       +
       

     

       54
       54
       +
       extension SyntaxCursor {

     

       55
       55
       +
       	var singleLineRanges: [SingleLineRange] {

     

       56
       56
       +
       		var result: [SingleLineRange] = []

     

       57
       57
       +
       		var location = lineMap.lspPosition(at: utf16Offset)

     

       58
       58
       +
       

     

       59
       59
       +
       		for line in node.text.split(omittingEmptySubsequences: false, whereSeparator: \.isNewline) {

     

       60
       60
       +
       			let length = line.utf16.count

     

       61
       61
       +
       			result.append(SingleLineRange(line: location.line, char: location.character, length: length))

     

       62
       62
       +
       			location = Position(line: location.line + 1, character: 0)

     

       63
       63
       +
       		}

     

       64
       64
       +
       

     

       65
       65
       +
       		return result

     

       66
       66
       +
       	}

     

       67
       67
       +
       }

     

       68
       68
       +
       

     

       69
       69
       +
       extension SyntaxCursor {

     

       70
       70
       +
       	public func collectSemanticTokens(_ sink: inout [SemanticToken], in range: Range<Int>? = nil) {

     

       71
       71
       +
       		if let range {

     

       72
       72
       +
       			guard range.overlaps(utf16Range) else { return }

     

       73
       73
       +
       		}

     

       74
       74
       +
       		

     

       75
       75
       +
       		if let (_, metadata) = node.token, let metadata {

     

       76
       76
       +
       			for lineRange in singleLineRanges {

     

       77
       77
       +
       				if let semanticToken = metadata.semanticToken(range: lineRange) {

     

       78
       78
       +
       					sink.append(semanticToken)

     

       79
       79
       +
       				}

     

       80
       80
       +
       			}

     

       81
       81
       +
       		}

     

       82
       82
       +
       		

     

       83
       83
       +
       		for child in children {

     

       84
       84
       +
       			child.collectSemanticTokens(&sink, in: range)

     

       85
       85
       +
       		}

     

       86
       86
       +
       	}

     

       87
       87
       +
       	

     

       88
       88
       +
       	public func semanticTokens(in range: Range<Int>? = nil) -> [SemanticToken] {

     

       89
       89
       +
       		var tokens: [SemanticToken] = []

     

       90
       90
       +
       		collectSemanticTokens(&tokens, in: range)

     

       91
       91
       +
       		return tokens

     

       92
       92
       +
       	}

     

       93
       93
       +
       	

     

       94
       94
       +
       	public var semanticTokens: [SemanticToken] {

     

       95
       95
       +
       		semanticTokens()

     

       96
       96
       +
       	}

     

       97
       97
       +
       }

+98

Sources/PterodactylSyntax/SyntaxCursor.swift

···

       1
       1
       +
       // SPDX-FileCopyrightText: 2025 The Project Pterodactyl Developers

     

       2
       2
       +
       //

     

       3
       3
       +
       // SPDX-License-Identifier: MPL-2.0

     

       4
       4
       +
       

     

       5
       5
       +
       import LanguageServerProtocol

     

       6
       6
       +
       

     

       7
       7
       +
       /// This is a “red tree” in the sense of Roslyn. In essence it instruments syntax trees with non-relative location information.

     

       8
       8
       +
       public final class SyntaxCursor: CustomStringConvertible {

     

       9
       9
       +
       	public let lineMap: LineMap

     

       10
       10
       +
       	public let node: SyntaxTree.Child

     

       11
       11
       +
       	public let utf16Offset: Int

     

       12
       12
       +
       

     

       13
       13
       +
       	public private(set) lazy var children: [SyntaxCursor] = {

     

       14
       14
       +
       		var children: [SyntaxCursor] = []

     

       15
       15
       +
       		var utf16Offset = utf16Offset

     

       16
       16
       +
       		for childNode in node.children {

     

       17
       17
       +
       			children.append(Self(lineMap: lineMap, node: childNode, utf16Offset: utf16Offset))

     

       18
       18
       +
       			utf16Offset += childNode.utf16Length

     

       19
       19
       +
       		}

     

       20
       20
       +
       

     

       21
       21
       +
       		return children

     

       22
       22
       +
       	}()

     

       23
       23
       +
       

     

       24
       24
       +
       	public var utf16Range: Range<Int> {

     

       25
       25
       +
       		utf16Offset..<utf16Offset + node.utf16Length

     

       26
       26
       +
       	}

     

       27
       27
       +
       	

     

       28
       28
       +
       	/// A sendable type capturing the initialisation parameters of a ``SyntaxCursor``.

     

       29
       29
       +
       	public struct Options: Sendable {

     

       30
       30
       +
       		let lineMap: LineMap

     

       31
       31
       +
       		let node: SyntaxTree.Child

     

       32
       32
       +
       		let utf16Offset: Int

     

       33
       33
       +
       		public init(lineMap: LineMap, node: SyntaxTree.Child, utf16Offset: Int) {

     

       34
       34
       +
       			self.lineMap = lineMap

     

       35
       35
       +
       			self.node = node

     

       36
       36
       +
       			self.utf16Offset = utf16Offset

     

       37
       37
       +
       		}

     

       38
       38
       +
       		

     

       39
       39
       +
       		public var cursor: SyntaxCursor {

     

       40
       40
       +
       			SyntaxCursor(lineMap: lineMap, node: node, utf16Offset: utf16Offset)

     

       41
       41
       +
       		}

     

       42
       42
       +
       	}

     

       43
       43
       +
       

     

       44
       44
       +
       	public init(lineMap: LineMap, node: SyntaxTree.Child, utf16Offset: Int) {

     

       45
       45
       +
       		self.lineMap = lineMap

     

       46
       46
       +
       		self.node = node

     

       47
       47
       +
       		self.utf16Offset = utf16Offset

     

       48
       48
       +
       	}

     

       49
       49
       +
       	

     

       50
       50
       +
       	private func description(indent: String) -> String {

     

       51
       51
       +
       		var output = ""

     

       52
       52
       +
       		switch node {

     

       53
       53
       +
       		case .token(let token, _):

     

       54
       54
       +
       			output += "\(indent)- \(token.kind)\n"

     

       55
       55
       +
       		case .tree(let tree):

     

       56
       56
       +
       			output += "\(indent)+ \(tree.kind)\n"

     

       57
       57
       +
       		}

     

       58
       58
       +
       		for child in children {

     

       59
       59
       +
       			output += child.description(indent: indent + "\t")

     

       60
       60
       +
       		}

     

       61
       61
       +
       		return output

     

       62
       62
       +
       	}

     

       63
       63
       +
       	

     

       64
       64
       +
       	public var description: String {

     

       65
       65
       +
       		description(indent: "")

     

       66
       66
       +
       	}

     

       67
       67
       +
       	

     

       68
       68
       +
       	public func collectHints(sink: inout [(Range<Int>, String)], in range: Range<Int>) {

     

       69
       69
       +
       		// TODO: see if this is really working

     

       70
       70
       +
       		guard range.overlaps(utf16Range) else { return }

     

       71
       71
       +
       		

     

       72
       72
       +
       		if let tree = node.tree, let hint = tree.metadata?.hint {

     

       73
       73
       +
       			sink.append((utf16Range, hint))

     

       74
       74
       +
       		}

     

       75
       75
       +
       		

     

       76
       76
       +
       		for child in children {

     

       77
       77
       +
       			child.collectHints(sink: &sink, in: range)

     

       78
       78
       +
       		}

     

       79
       79
       +
       	}

     

       80
       80
       +
       }

     

       81
       81
       +
       

     

       82
       82
       +
       extension SyntaxCursor {

     

       83
       83
       +
       	public func firstChild<T>(mapping: (SyntaxCursor) -> T?) -> T? {

     

       84
       84
       +
       		for child in children {

     

       85
       85
       +
       			if let result = mapping(child) {

     

       86
       86
       +
       				return result

     

       87
       87
       +
       			} else {

     

       88
       88
       +
       				continue

     

       89
       89
       +
       			}

     

       90
       90
       +
       		}

     

       91
       91
       +
       		return nil

     

       92
       92
       +
       	}

     

       93
       93
       +
       

     

       94
       94
       +
       	public func children<T>(mapping: (SyntaxCursor) -> T?) -> [T] {

     

       95
       95
       +
       		children.compactMap(mapping)

     

       96
       96
       +
       	}

     

       97
       97
       +
       }

     

       98
       98
       +

+36 -9

Sources/PterodactylSyntax/SyntaxTree.swift

···

       2
       2
        
       //

     

       3
       3
        
       // SPDX-License-Identifier: MPL-2.0

     

       4
       4
        
       

     

       5
       5
       -
       import Foundation

     

       6
       5
        
       

     

       6
       6
       +
       /// This is a “green tree” in the sense of Roslyn.

     

       7
       7
        
       public struct SyntaxTree: Codable, Sendable {

     

       8
       8
        
       	public let kind: SyntaxTreeKind

     

       9
       9
        
       	public let metadata: SyntaxTreeMetadata?

     

       10
       10
        
       	public let children: [Child]

     

       11
       11
        
       	public let utf16Length: Int

     

       12
       12
       -
         

     

       13
       13
       -
         public enum Child: Codable, Sendable {

     

       14
       14
       -
           case token(Token, metadata: TokenMetadata?)

     

       15
       15
       -
           case tree(SyntaxTree)

     

       16
       16
       -
         }

     

       12
       12
       +
       

     

       13
       13
       +
       	public enum Child: Codable, Sendable {

     

       14
       14
       +
       		case token(Token, metadata: TokenMetadata?)

     

       15
       15
       +
       		case tree(SyntaxTree)

     

       16
       16
       +
       	}

     

       17
       17
        
       

     

       18
       18
       -
         public init(kind: SyntaxTreeKind, metadata: SyntaxTreeMetadata?, children: [SyntaxTree.Child]) {

     

       18
       18
       +
       	public init(kind: SyntaxTreeKind, metadata: SyntaxTreeMetadata? = nil, children: [SyntaxTree.Child]) {

     

       19
       19
        
       		self.kind = kind

     

       20
       20
        
       		self.metadata = metadata

     

       21
       21
        
       		self.children = children

     
···

       26
       26
        
       }

     

       27
       27
        
       

     

       28
       28
        
       extension SyntaxTree {

     

       29
       29
       +
       	/// A mutable version of ``SyntaxTree`` that does not keep track of textual length, for use when constructing trees.

     

       30
       30
       +
       	public struct MutableTree {

     

       31
       31
       +
       		public var kind: SyntaxTreeKind

     

       32
       32
       +
       		public var metadata: SyntaxTreeMetadata?

     

       33
       33
       +
       		public var children: [Child]

     

       34
       34
       +
       		

     

       35
       35
       +
       		var tree: SyntaxTree {

     

       36
       36
       +
       			SyntaxTree(kind: kind, metadata: metadata, children: children)

     

       37
       37
       +
       		}

     

       38
       38
       +
       	}

     

       39
       39
       +
       }

     

       40
       40
       +
       

     

       41
       41
       +
       extension SyntaxTree {

     

       29
       42
        
       	public var text: String {

     

       30
       43
        
       		children.map(\.text).joined()

     

       31
       44
        
       	}

     

       32
       45
        
       }

     

       33
       46
        
       

     

       34
       47
        
       extension SyntaxTree.Child {

     

       35
       35
       -
       	var text: String {

     

       48
       48
       +
       	public var text: String {

     

       36
       49
        
       		switch self {

     

       37
       50
        
       		case let .token(tok, _): tok.text

     

       38
       51
        
       		case let .tree(tree): tree.text

     

       39
       52
        
       		}

     

       40
       53
        
       	}

     

       54
       54
       +
       	

     

       55
       55
       +
       	public var tree: SyntaxTree? {

     

       56
       56
       +
       		switch self {

     

       57
       57
       +
       		case let .tree(tree): tree

     

       58
       58
       +
       		default: nil

     

       59
       59
       +
       		}

     

       60
       60
       +
       	}

     

       41
       61
        
       

     

       62
       62
       +
       	var token: (Token, TokenMetadata?)? {

     

       63
       63
       +
       		switch self {

     

       64
       64
       +
       		case let .token(token, metadata): (token, metadata)

     

       65
       65
       +
       		default: nil

     

       66
       66
       +
       		}

     

       67
       67
       +
       	}

     

       68
       68
       +
       	

     

       42
       69
        
       	var utf16Length: Int {

     

       43
       70
        
       		switch self {

     

       44
       71
        
       		case let .token(token, _): token.utf16Length

     

       45
       72
        
       		case let .tree(tree): tree.utf16Length

     

       46
       73
        
       		}

     

       47
       74
        
       	}

     

       48
       48
       -
       

     

       75
       75
       +
       	

     

       49
       76
        
       	var children: [Self] {

     

       50
       77
        
       		switch self {

     

       51
       78
        
       		case .token: []

+96

Sources/PterodactylSyntax/SyntaxTreeBuilder.swift

···

       1
       1
       +
       // SPDX-FileCopyrightText: 2025 The Project Pterodactyl Developers

     

       2
       2
       +
       //

     

       3
       3
       +
       // SPDX-License-Identifier: MPL-2.0

     

       4
       4
       +
       

     

       5
       5
       +
       

     

       6
       6
       +
       public class SyntaxTreeBuilder {

     

       7
       7
       +
       	private enum Event: Equatable {

     

       8
       8
       +
       		case open

     

       9
       9
       +
       		case close(kind: SyntaxTreeKind, metadata: SyntaxTreeMetadata?, cancelled: Bool)

     

       10
       10
       +
       		case advance(token: Token, metadata: TokenMetadata?)

     

       11
       11
       +
       	}

     

       12
       12
       +
       

     

       13
       13
       +
       	private var events: [Event] = []

     

       14
       14
       +
       

     

       15
       15
       +
       	public init() {}

     

       16
       16
       +
       

     

       17
       17
       +
       	public struct SubtreeHandle: ~Copyable {

     

       18
       18
       +
       		let parent: SyntaxTreeBuilder

     

       19
       19
       +
       		var kind: SyntaxTreeKind

     

       20
       20
       +
       		var metadata: SyntaxTreeMetadata?

     

       21
       21
       +
       		private var closed = false

     

       22
       22
       +
       		var cancelled = false

     

       23
       23
       +
       

     

       24
       24
       +
       		init(parent: SyntaxTreeBuilder, kind: SyntaxTreeKind? = nil, metadata: SyntaxTreeMetadata? = nil) {

     

       25
       25
       +
       			self.parent = parent

     

       26
       26
       +
       			self.kind = kind ?? .error

     

       27
       27
       +
       			self.metadata = metadata

     

       28
       28
       +
       

     

       29
       29
       +
       			parent.events.append(.open)

     

       30
       30
       +
       		}

     

       31
       31
       +
       		

     

       32
       32
       +
       		mutating func cancel() {

     

       33
       33
       +
       			cancelled = true

     

       34
       34
       +
       		}

     

       35
       35
       +
       

     

       36
       36
       +
       		mutating func close() {

     

       37
       37
       +
       			precondition(!closed)

     

       38
       38
       +
       			parent.events.append(.close(kind: kind, metadata: metadata, cancelled: cancelled))

     

       39
       39
       +
       			closed = true

     

       40
       40
       +
       		}

     

       41
       41
       +
       

     

       42
       42
       +
       		deinit {

     

       43
       43
       +
       			precondition(closed)

     

       44
       44
       +
       		}

     

       45
       45
       +
       	}

     

       46
       46
       +
       

     

       47
       47
       +
       	public func open(kind: SyntaxTreeKind? = nil, metadata: SyntaxTreeMetadata? = nil) -> SubtreeHandle {

     

       48
       48
       +
       		SubtreeHandle(parent: self, kind: kind, metadata: metadata)

     

       49
       49
       +
       	}

     

       50
       50
       +
       

     

       51
       51
       +
       	public func advance(token: Token, metadata: TokenMetadata?) {

     

       52
       52
       +
       		events.append(.advance(token: token, metadata: metadata))

     

       53
       53
       +
       	}

     

       54
       54
       +
       

     

       55
       55
       +
       	public var tree: SyntaxTree {

     

       56
       56
       +
       		var stack: [SyntaxTree.MutableTree] = []

     

       57
       57
       +
       

     

       58
       58
       +
       		for event in events {

     

       59
       59
       +
       			switch event {

     

       60
       60
       +
       			case .open:

     

       61
       61
       +
       				stack.append(SyntaxTree.MutableTree(kind: .error, metadata: nil, children: []))

     

       62
       62
       +
       

     

       63
       63
       +
       			case .close(let kind, let metadata, let cancelled):

     

       64
       64
       +
       				guard var node = stack.popLast() else { fatalError("Unbalanced tree") }

     

       65
       65
       +
       				node.kind = kind

     

       66
       66
       +
       				node.metadata = metadata

     

       67
       67
       +
       				guard let parentIndex = stack.indices.last else { return node.tree }

     

       68
       68
       +
       				if cancelled {

     

       69
       69
       +
       					stack[parentIndex].children.append(contentsOf: node.children)

     

       70
       70
       +
       				} else {

     

       71
       71
       +
       					stack[parentIndex].children.append(.tree(node.tree))

     

       72
       72
       +
       				}

     

       73
       73
       +
       

     

       74
       74
       +
       			case .advance(let token, let metadata):

     

       75
       75
       +
       				guard let parentIndex = stack.indices.last else {

     

       76
       76
       +
       					fatalError("Attempted to insert token at root of parse tree")

     

       77
       77
       +
       				}

     

       78
       78
       +
       				stack[parentIndex]

     

       79
       79
       +
       					.children

     

       80
       80
       +
       					.append(.token(token, metadata: metadata))

     

       81
       81
       +
       			}

     

       82
       82
       +
       		}

     

       83
       83
       +
       

     

       84
       84
       +
       		fatalError("Unbalanced syntax tree")

     

       85
       85
       +
       	}

     

       86
       86
       +
       }

     

       87
       87
       +
       

     

       88
       88
       +
       

     

       89
       89
       +
       extension Array {

     

       90
       90
       +
       	fileprivate mutating func modifyLast(_ modifier: (inout Element) -> Void) {

     

       91
       91
       +
       		if var last = popLast() {

     

       92
       92
       +
       			modifier(&last)

     

       93
       93
       +
       			append(last)

     

       94
       94
       +
       		}

     

       95
       95
       +
       	}

     

       96
       96
       +
       }

+21

Sources/PterodactylSyntax/SyntaxView.swift

···

       1
       1
       +
       // SPDX-FileCopyrightText: 2025 The Project Pterodactyl Developers

     

       2
       2
       +
       //

     

       3
       3
       +
       // SPDX-License-Identifier: MPL-2.0

     

       4
       4
       +
       

     

       5
       5
       +
       

     

       6
       6
       +
       /// An abstract syntax view around a ``SyntaxCursor``. This is to be populated by extensions targetting specific `G`.

     

       7
       7
       +
       struct SyntaxView<G: Grammar> {

     

       8
       8
       +
       	let cursor: SyntaxCursor

     

       9
       9
       +
       	init?(_ cursor: SyntaxCursor) {

     

       10
       10
       +
       		guard let kind = cursor.node.tree?.kind, G.kinds.contains(kind) else { return nil }

     

       11
       11
       +
       		self.cursor = cursor

     

       12
       12
       +
       	}

     

       13
       13
       +
       	

     

       14
       14
       +
       	func matchingSubview<X: Grammar>() -> SyntaxView<X>? {

     

       15
       15
       +
       		return cursor.firstChild(mapping: SyntaxView<X>.init)

     

       16
       16
       +
       	}

     

       17
       17
       +
       	

     

       18
       18
       +
       	func matchingSubviews<X: Grammar>() -> [SyntaxView<X>] {

     

       19
       19
       +
       		return cursor.children(mapping: SyntaxView<X>.init)

     

       20
       20
       +
       	}

     

       21
       21
       +
       }

+2 -3

Sources/PterodactylSyntax/Token.swift

···

       2
       2
        
       //

     

       3
       3
        
       // SPDX-License-Identifier: MPL-2.0

     

       4
       4
        
       

     

       5
       5
       -
       import Foundation

     

       6
       5
        
       

     

       7
       7
       -
       public struct Token: Codable {

     

       6
       6
       +
       public struct Token: Codable, Equatable {

     

       8
       7
        
       	public let kind: TokenKind

     

       9
       8
        
       	public let text: String

     

       10
       9
        
       	public let utf16Length: Int

     

       11
       10
        
       	

     

       12
       12
       -
       	init(kind: TokenKind, text: String) {

     

       11
       11
       +
       	public init(kind: TokenKind, text: String) {

     

       13
       12
        
       		self.kind = kind

     

       14
       13
        
       		self.text = text

     

       15
       14
        
       		self.utf16Length = text.utf16.count

+101 -7

Sources/PterodactylSyntax/Types.swift

···

       2
       2
        
       //

     

       3
       3
        
       // SPDX-License-Identifier: MPL-2.0

     

       4
       4
        
       

     

       5
       5
       -
       import Foundation

     

       5
       5
       +
       import LanguageServerProtocol

     

       6
       6
        
       

     

       7
       7
       -
       public enum TokenKind: Codable, Equatable, Sendable {

     

       8
       8
       -
         case eof

     

       7
       7
       +
       public enum Keyword: String, Codable, Sendable, CaseIterable {

     

       8
       8
       +
       	case theory = "theory"

     

       9
       9
       +
       	case `where` = "where"

     

       10
       10
       +
       	case `import` = "import"

     

       9
       11
        
       }

     

       10
       12
        
       

     

       11
       11
       -
       public enum SyntaxTreeKind: Codable, Equatable, Sendable {

     

       12
       12
       -
         case error

     

       13
       13
       +
       extension Keyword: CustomStringConvertible {

     

       14
       14
       +
       	public var description: String { rawValue }

     

       13
       15
        
       }

     

       14
       16
        
       

     

       15
       15
       -
       public struct TokenMetadata: Codable, Equatable, Sendable {

     

       17
       17
       +
       public enum Punctuation: String, CaseIterable, Codable, Equatable, Sendable {

     

       18
       18
       +
       	case lparen = "("

     

       19
       19
       +
       	case rparen = ")"

     

       20
       20
       +
       	case lbrace = "{"

     

       21
       21
       +
       	case rbrace = "}"

     

       22
       22
       +
       	case comma = ","

     

       23
       23
       +
       	case dot = "."

     

       24
       24
       +
       	case colon = ":"

     

       25
       25
       +
       	case doubleLeftArrow = "<="

     

       26
       26
       +
       	case doubleRightArrow = "=>"

     

       27
       27
       +
       	case rightArrow = "->"

     

       28
       28
       +
       	case equal = "="

     

       29
       29
       +
       }

     

       30
       30
       +
       

     

       31
       31
       +
       extension Punctuation: CustomStringConvertible {

     

       32
       32
       +
       	public var description: String { rawValue }

     

       33
       33
       +
       }

     

       34
       34
       +
       

     

       35
       35
       +
       public enum TokenKind: Codable, Equatable, Sendable, Hashable {

     

       36
       36
       +
       	case eof

     

       37
       37
       +
       	case keyword(Keyword)

     

       38
       38
       +
       	case punctuation(Punctuation)

     

       39
       39
       +
       	case error

     

       40
       40
       +
       	case identifier

     

       41
       41
       +
       	case newline

     

       42
       42
       +
       	case whitespace

     

       43
       43
       +
       	case blockBegin

     

       44
       44
       +
       	case blockEnd

     

       45
       45
       +
       	case blockSep

     

       46
       46
       +
       	case lineComment

     

       47
       47
       +
       	case blockComment(terminated: Bool)

     

       48
       48
       +
       }

     

       49
       49
       +
       

     

       50
       50
       +
       extension Keyword {

     

       51
       51
       +
       	public var isBlockHerald: Bool {

     

       52
       52
       +
       		switch self {

     

       53
       53
       +
       		case .where: true

     

       54
       54
       +
       		default: false

     

       55
       55
       +
       		}

     

       56
       56
       +
       	}

     

       57
       57
       +
       }

     

       58
       58
       +
       

     

       59
       59
       +
       extension TokenKind {

     

       60
       60
       +
       	public var isTrivia: Bool {

     

       61
       61
       +
       		switch self {

     

       62
       62
       +
       		case .whitespace, .newline, .lineComment, .blockComment: true

     

       63
       63
       +
       		default: false

     

       64
       64
       +
       		}

     

       65
       65
       +
       	}

     

       66
       66
       +
       

     

       67
       67
       +
       	public var isVisible: Bool {

     

       68
       68
       +
       		switch self {

     

       69
       69
       +
       		case .whitespace, .blockBegin, .blockSep, .blockEnd, .newline: false

     

       70
       70
       +
       		default: true

     

       71
       71
       +
       		}

     

       72
       72
       +
       	}

     

       73
       73
       +
       	

     

       74
       74
       +
       	public var canDetermineLayoutColumn: Bool {

     

       75
       75
       +
       		switch self {

     

       76
       76
       +
       		case .whitespace, .eof: false

     

       77
       77
       +
       		default: true

     

       78
       78
       +
       		}

     

       79
       79
       +
       	}

     

       80
       80
       +
       

     

       81
       81
       +
       	public var isBlockHerald: Bool {

     

       82
       82
       +
       		switch self {

     

       83
       83
       +
       		case .keyword(let kwd): kwd.isBlockHerald

     

       84
       84
       +
       		default: false

     

       85
       85
       +
       		}

     

       86
       86
       +
       	}

     

       87
       87
       +
       

     

       88
       88
       +
       }

     

       89
       89
       +
       

     

       90
       90
       +
       public final class SyntaxTreeKind: Codable, Equatable, Sendable, CustomStringConvertible {

     

       91
       91
       +
       	public static let error: SyntaxTreeKind = .init(name: "error")

     

       92
       92
       +
       

     

       93
       93
       +
       	public static func == (lhs: SyntaxTreeKind, rhs: SyntaxTreeKind) -> Bool {

     

       94
       94
       +
       		lhs === rhs

     

       95
       95
       +
       	}

     

       96
       96
       +
       

     

       97
       97
       +
       	let name: String

     

       98
       98
       +
       	public var description: String { name }

     

       99
       99
       +
       

     

       100
       100
       +
       	required init(name: String) {

     

       101
       101
       +
       		self.name = name

     

       102
       102
       +
       	}

     

       103
       103
       +
       }

     

       104
       104
       +
       

     

       105
       105
       +
       public struct TokenMetadata: Equatable, Codable, Sendable {

     

       106
       106
       +
       	public var semanticTokenType: SemanticTokenTypes

     

       107
       107
       +
       	public var semanticTokenModifiers: Set<SemanticTokenModifiers> = []

     

       108
       108
       +
       	public var delimitedFoldingRangeKind: FoldingRangeKind? = nil

     

       16
       109
        
       }

     

       17
       110
        
       

     

       18
       111
        
       public struct SyntaxTreeMetadata: Codable, Equatable, Sendable {

     

       19
       19
       -
         

     

       112
       112
       +
       	public var delimitedFoldingRangeKind: FoldingRangeKind? = nil

     

       113
       113
       +
       	public var hint: String? = nil

     

       20
       114
        
       }

-24

Sources/PterodactylSyntax/Utf16Position.swift

···

       1
       1
       -
       // SPDX-FileCopyrightText: 2025 The Project Pterodactyl Developers

     

       2
       2
       -
       //

     

       3
       3
       -
       // SPDX-License-Identifier: MPL-2.0

     

       4
       4
       -
       

     

       5
       5
       -
       import Foundation

     

       6
       6
       -
       

     

       7
       7
       -
       public struct Utf16Position: Equatable, Comparable, Sendable {

     

       8
       8
       -
       	public var absoluteOffset: Int

     

       9
       9
       -
       	public var line: Int

     

       10
       10
       -
       	public var column : Int

     

       11
       11
       -
       	public init(absoluteOffset: Int, line: Int, column: Int) {

     

       12
       12
       -
       		self.absoluteOffset = absoluteOffset

     

       13
       13
       -
       		self.line = line

     

       14
       14
       -
       		self.column = column

     

       15
       15
       -
       	}

     

       16
       16
       -
       	

     

       17
       17
       -
       	public static func < (lhs: Utf16Position, rhs: Utf16Position) -> Bool {

     

       18
       18
       -
       		lhs.absoluteOffset < rhs.absoluteOffset

     

       19
       19
       -
       	}

     

       20
       20
       -
       }

     

       21
       21
       -
       

     

       22
       22
       -
       public extension Utf16Position {

     

       23
       23
       -
       	static var zero : Self { Self(absoluteOffset: 0, line: 0, column: 0) }

     

       24
       24
       -
       }

+45 -5

Tests/PterodactylBuildTests/Test.swift

···

       5
       5
        
       import Testing

     

       6
       6
        
       

     

       7
       7
        
       @testable import PterodactylBuild

     

       8
       8
       +
       @testable import PterodactylSyntax

     

       8
       9
        
       @testable import llbuild2fx

     

       9
       10
        
       

     

       10
       11
        
       struct BuildTests {

     

       11
       12
        
       	@Test

     

       13
       13
       +
       	func testBlockLayout() async throws {

     

       14
       14
       +
       		let code = """

     

       15
       15
       +
       			theory Foo where

     

       16
       16
       +
       				 foo : bar

     

       17
       17
       +
       

     

       18
       18
       +
       				 baz : sdf

     

       19
       19
       +
       			"""

     

       20
       20
       +
       

     

       21
       21
       +
       		var lexer = PterodactylSyntax.Lexer(input: code)

     

       22
       22
       +
       		let tokens = lexer.tokenize()

     

       23
       23
       +
       

     

       24
       24
       +
       		#expect(

     

       25
       25
       +
       			tokens.map(\.kind) == [

     

       26
       26
       +
       				.keyword(.theory), .whitespace, .identifier, .whitespace, .keyword(.where), .newline, .blockBegin, .whitespace, .identifier, .whitespace,

     

       27
       27
       +
       				.punctuation(.colon),

     

       28
       28
       +
       				.whitespace, .identifier, .newline, .newline, .blockSep, .whitespace, .identifier, .whitespace, .punctuation(.colon), .whitespace, .identifier, .blockEnd, .eof

     

       29
       29
       +
       			])

     

       30
       30
       +
       	}

     

       31
       31
       +
       

     

       32
       32
       +
       	@Test

     

       33
       33
       +
       	func testParse() async throws {

     

       34
       34
       +
       		let code = """

     

       35
       35
       +
       			theory assdf where

     

       36
       36
       +
       				asdf : (x : y asdf) (u : Vasdf) -> asaaaadf

     

       37
       37
       +
       				asdf => asdf

     

       38
       38
       +
       			"""

     

       39
       39
       +
       

     

       40
       40
       +
       		var lexer = PterodactylSyntax.Lexer(input: code)

     

       41
       41
       +
       		let tokens = lexer.tokenize()

     

       42
       42
       +
       		var context = Parser(source: code, tokens: tokens)

     

       43
       43
       +
       		context.document()

     

       44
       44
       +
       		

     

       45
       45
       +
       		let tree = context.tree

     

       46
       46
       +
       //		let cursor = SyntaxCursor(lineMap: LineMap(source: code), node: .tree(tree), utf16Offset: 0)

     

       47
       47
       +
       		#expect(tree.kind == Document.kind)

     

       48
       48
       +
       		#expect(context.diagnostics.isEmpty)

     

       49
       49
       +
       	}

     

       50
       50
       +
       

     

       51
       51
       +
       	@Test

     

       12
       52
        
       	func testImports() async throws {

     

       13
       53
        
       		let group = LLBMakeDefaultDispatchGroup()

     

       14
       54
        
       		let db = LLBInMemoryCASDatabase(group: group)

     
···

       29
       69
        
       		)

     

       30
       70
        
       

     

       31
       71
        
       		let treeID: LLBDataID = try await client.store(declTree, ctx).get()

     

       32
       32
       -
       		let dependencyGraph = try await engine.build(key: Keys.DependencyGraphOfSourceTree(sourceTreeId: treeID), ctx).get()

     

       33
       33
       -
       		let foo = UnitName(name: "foo")

     

       34
       34
       -
       		let bar = UnitName(name: "bar")

     

       35
       35
       -
       		let baz = UnitName(name: "baz")

     

       72
       72
       +
       		let dependencyGraph = try await engine.build(key: Keys.SourceTree.GetDependencyGraph(sourceTreeId: treeID), ctx).get()

     

       73
       73
       +
       		let foo = UnitName(basename: "foo")

     

       74
       74
       +
       		let bar = UnitName(basename: "bar")

     

       75
       75
       +
       		let baz = UnitName(basename: "baz")

     

       36
       76
        
       

     

       37
       77
        
       		#expect(

     

       38
       78
        
       			dependencyGraph.edges == [

     
···

       42
       82
        
       			]

     

       43
       83
        
       		)

     

       44
       84
        
       

     

       45
       45
       -
       		let dependenciesOfBaz = try await engine.build(key: Keys.TransitiveDependencies(sourceTreeId: treeID, unitName: baz), ctx).get().dependencies

     

       85
       85
       +
       		let dependenciesOfBaz = try await engine.build(key: Keys.SourceTree.GetDependencies(sourceTreeId: treeID, unitName: baz), ctx).get()

     

       46
       86
        
       		#expect(dependenciesOfBaz == [foo, bar])

     

       47
       87
        
       		return

     

       48
       88
        
       	}

+29

Tests/PterodactylLanguageServerTests/Test.swift

···

       1
       1
       +
       // SPDX-FileCopyrightText: 2025 The Project Pterodactyl Developers

     

       2
       2
       +
       //

     

       3
       3
       +
       // SPDX-License-Identifier: MPL-2.0

     

       4
       4
       +
       

     

       5
       5
       +
       import Foundation

     

       6
       6
       +
       import Testing

     

       7
       7
       +
       

     

       8
       8
       +
       @testable import PterodactylBuild

     

       9
       9
       +
       @testable import PterodactylLanguageServer

     

       10
       10
       +
       @testable import TSCBasic

     

       11
       11
       +
       @testable import llbuild2fx

     

       12
       12
       +
       

     

       13
       13
       +
       struct LanguageServerTests {

     

       14
       14
       +
       	@Test

     

       15
       15
       +
       	func testSingletonFileTree() throws {

     

       16
       16
       +
       		let path = try AbsolutePath(validating: "/foo/bar/file.txt")

     

       17
       17
       +
       		let foo = LLBDeclFileTree.file(absolutePath: path, contents: "foobar")

     

       18
       18
       +
       		let expected: LLBDeclFileTree =

     

       19
       19
       +
       			.dir([

     

       20
       20
       +
       				"foo": .dir([

     

       21
       21
       +
       					"bar": .dir([

     

       22
       22
       +
       						"file.txt": .file("foobar")

     

       23
       23
       +
       					])

     

       24
       24
       +
       				])

     

       25
       25
       +
       			])

     

       26
       26
       +
       

     

       27
       27
       +
       		#expect(foo.debugDescription == expected.debugDescription)

     

       28
       28
       +
       	}

     

       29
       29
       +
       }

license.sh

···

       1
       1
       +
       #!/bin/bash

     

       2
       2
       +
       

     

       3
       3
       +
       reuse annotate --recursive --license MPL-2.0 --copyright "The Project Pterodactyl Developers" Tests Sources

     

       4
       4
       +

test.ptero

···

       1
       1
       +
       theory Group where

     

       2
       2
       +
       	carrier : Set

     

       3
       3
       +
       	multiply : carrier -> carrier -> carrier

     

       4
       4
       +
       	empty : carrier

Compare changes