Add functionality to swift-script

Author: stevapple

Sources/Commands/SwiftScriptTool.swift

@@ -75,6 +75,7 @@ public struct SwiftScriptTool: ParsableCommand {
             Clean.self,
             Reset.self,
             Resolve.self,
+            List.self,
         ],
         helpNames: [.short, .long, .customLong("help", withSingleDash: true)])
 
@@ -138,7 +139,10 @@ extension SwiftScriptTool {
 
         @OptionGroup()
         var options: ScriptToolOptions
-        
+
+        @Option(name: .shortAndLong, help: "Save the prebuilt script binary", transform: AbsolutePath.init)
+        var output: AbsolutePath?
+
         func run(_ swiftTool: SwiftTool, as productName: String, at cacheDirPath: AbsolutePath) throws {
             swiftTool.redirectStdoutToStderr()
             swiftTool.diagnostics.emit(note: "Using cache: \(cacheDirPath.basename)")
@@ -148,6 +152,15 @@ extension SwiftScriptTool {
                 if options.shouldBuild {
                     try buildSystem.build(subset: .product(productName))
                 }
+
+                if let output = output {
+                    let executablePath = try swiftTool.buildParameters().buildPath.appending(component: productName)
+                    guard !localFileSystem.isDirectory(output) else {
+                        throw ScriptError.isDirectory(output.pathString)
+                    }
+                    try? localFileSystem.removeFileTree(output)
+                    try localFileSystem.copy(from: executablePath, to: output)
+                }
             } catch let error as ScriptError {
                 swiftTool.diagnostics.emit(error)
                 throw ExitCode.failure
@@ -238,6 +251,35 @@ extension SwiftScriptTool {
     }
 }
 
+extension SwiftScriptTool {
+    struct List: ParsableCommand {
+        static let configuration = CommandConfiguration(
+            abstract: "List script caches")
+
+        @OptionGroup(_hiddenFromHelp: true)
+        var swiftOptions: SwiftToolOptions
+
+        func run() throws {
+            let cacheDir = try localFileSystem.getOrCreateSwiftScriptCacheDirectory()
+            let scripts = try localFileSystem.getDirectoryContents(cacheDir)
+            let resolved = try scripts.map { script -> (String, AbsolutePath) in
+                let sourceDir = cacheDir.appending(components: script, "Sources")
+                guard localFileSystem.isDirectory(sourceDir),
+                      let name = try localFileSystem.getDirectoryContents(sourceDir).first,
+                      case let realpath = resolveSymlinks(sourceDir.appending(components: name, "main.swift")) else {
+                          throw Diagnostics.fatalError
+                      }
+                return (script, realpath)
+            }
+            print("\(scripts.count) script\(scripts.count > 1 ? "s" : "") cached at \(cacheDir)")
+            guard let maxLength = resolved.map(\.0.count).max() else { return }
+            resolved.forEach { (name, path) in
+                print(name + String(repeating: " ", count: maxLength - name.count + 2) + path.pathString)
+            }
+        }
+    }
+}
+
 extension SwiftScriptTool {
     struct ResolveOptions: ParsableArguments {
         @Option(help: "The version to resolve at", transform: { Version(string: $0) })