Nightly arg

Author: AndrewBarba

Plugins/VercelPackager/VercelOutput.swift

@@ -149,6 +149,10 @@ extension VercelOutput {
         arguments.contains("--prod")
     }
 
+    public var nightly: Bool {
+        arguments.contains("--nightly")
+    }
+
     public var functionMemory: String {
         argument("memory") ?? "512"
     }
@@ -558,8 +562,9 @@ extension VercelOutput {
 
     private func buildDockerProduct(_ product: Product) async throws -> Path {
         let dockerToolPath = try context.tool(named: "docker").path
-        let baseImage =
-            "swift:\(context.package.toolsVersion.major).\(context.package.toolsVersion.minor)-amazonlinux2"
+        let baseImage = nightly
+            ? "swiftlang/swift:nightly-\(context.package.toolsVersion.major).\(context.package.toolsVersion.minor)-amazonlinux2"
+            : "swift:\(context.package.toolsVersion.major).\(context.package.toolsVersion.minor)-amazonlinux2"
 
         // build the product
         try Shell.execute(