1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
|
import PackagePlugin
import Foundation
@main
struct Plugin: CommandPlugin {
func performCommand(context: PluginContext, arguments: [String]) async throws {
var argumentExtractor = ArgumentExtractor(arguments)
let targets = argumentExtractor.extractOption(named: "target")
let sdk = argumentExtractor.extractOption(named: "sdk").last
let parameters = PackageManager.BuildParameters()
for target in targets {
try extractFromTarget(target: target, sdk: sdk, parameters: parameters, context: context)
}
}
func extractFromTarget(
target: String,
sdk: String?,
parameters: PackageManager.BuildParameters,
context: PluginContext
) throws {
let buildResult = try packageManager.build(.target(target), parameters: parameters)
guard buildResult.succeeded else {
throw PluginError(description: "Failed to build \(target): \(buildResult.logText)")
}
// TODO: Add proper API to PackagePlugin to get data directory
let dataPath = context.pluginWorkDirectory // output
.removingLastComponent() // WITExtractorPlugin
.removingLastComponent() // plugins
.removingLastComponent() // .build (by default)
let buildPath = dataPath.appending([parameters.configuration.rawValue])
let llbuildManifest = dataPath.appending([parameters.configuration.rawValue + ".yaml"])
guard let swiftcExecutable = ProcessInfo.processInfo.environment["WIT_EXTRACTOR_SWIFTC_PATH"]
?? inferSwiftcExecutablePath(llbuildManifest: llbuildManifest) else {
throw PluginError(description: "Cloudn't infer `swiftc` command path from build directory. Please specify WIT_EXTRACTOR_SWIFTC_PATH")
}
let digesterExecutable = Path(swiftcExecutable).removingLastComponent().appending(["swift-api-digester"])
let witOutputPath = context.pluginWorkDirectory.appending([target + ".wit"])
let swiftOutputPath = context.pluginWorkDirectory.appending([target + "_WITOverlay.swift"])
let tool = try context.tool(named: "WITTool")
var arguments = [
"extract-wit",
"--swift-api-digester", digesterExecutable.string,
"--module-name", target,
"--package-name", context.package.displayName,
"--wit-output-path", witOutputPath.string,
"--swift-output-path", swiftOutputPath.string,
"-I", buildPath.appending(["Modules"]).string,
]
#if compiler(<6.0)
// Swift 5.10 and earlier emit module files under the per-configuration build directory
// instead of the Modules directory.
arguments += [
"-I", buildPath.string,
]
#endif
if let sdk {
arguments += ["-sdk", sdk]
}
let process = try Process.run(URL(fileURLWithPath: tool.path.string), arguments: arguments)
process.waitUntilExit()
guard process.terminationStatus == 0 else {
throw PluginError(
description: "Failed to run \(([tool.path.string] + arguments).joined(separator: " "))"
)
}
print("""
{
"witOutputPath": "\(witOutputPath)",
"swiftOutputPath": "\(swiftOutputPath)"
}
""")
}
func inferSwiftcExecutablePath(llbuildManifest: Path) -> String? {
// FIXME: This is completely not the right way but there is no right way for now...
guard let contents = try? String(contentsOfFile: llbuildManifest.string, encoding: .utf8) else {
return nil
}
for line in contents.split(separator: "\n") {
do {
let prefix = " executable: \""
if line.hasPrefix(prefix), line.hasSuffix("/swiftc\"") {
let pathStart = line.index(line.startIndex, offsetBy: prefix.count)
let pathEnd = line.index(before: line.endIndex)
let executablePath = line[pathStart..<pathEnd]
return String(executablePath)
}
}
do {
// Swift 6.0 no longer uses llbuild's built-in swift tool. Instead,
// it uses the generic shell tool with full arguments.
// https://github.com/swiftlang/swift-package-manager/pull/6585
let prefix = " args: "
if line.hasPrefix(prefix) {
let argsString = line[line.index(line.startIndex, offsetBy: prefix.count)...]
guard let args = try? JSONDecoder().decode([String].self, from: Data(argsString.utf8)),
let swiftc = args.first(where: { $0.hasSuffix("/swiftc") }) else {
continue
}
return swiftc
}
}
}
return nil
}
}
struct PluginError: Error, CustomStringConvertible {
let description: String
}
|
