Generalize static site generation.

Assisted-By: devx/b3414b50-d642-461b-95a8-8f773c4d087b

Author: samuel-williams-shopify

bake/utopia/static.rb

@@ -3,20 +3,26 @@
 # Released under the MIT License.
 # Copyright, 2017-2025, by Samuel Williams.
 
-def generate(output_path: "static")
+# Generate a static copy of the application.
+# @parameter output_path [String] The output path for the generated site.
+# @parameter application_path [String] The application configuration path.
+# @parameter public_path [String] The public assets path.
+# @parameter force [Boolean] Remove the output directory before generating the site.
+def generate(output_path: "static", application_path: "config/application.rb", public_path: "public", force: true)
 	require "falcon/server"
-	require "async/io"
 	require "async/http/endpoint"
 	require "async/container"
+	require "fileutils"
 	require "utopia/application"
 
-	application_path = File.join(Dir.pwd, Utopia::Application::PATH)
+	application_path = File.expand_path(application_path, Dir.pwd)
+	public_path = File.expand_path(public_path, Dir.pwd)
 	container_class = Async::Container::Threaded
 	server_port = 9090
 
 	app = Utopia::Application.load(application_path)
 
-	container = container_class.run(count: 2) do
+	container = container_class.run(count: 1) do
 		Async do
 			server = Falcon::Server.new(
 				Falcon::Server.protocol_middleware(app),
@@ -29,17 +35,23 @@ def generate(output_path: "static")
 
 	output_path = File.expand_path(output_path, Dir.pwd)
 
-	# Delete any existing stuff:
-	FileUtils.rm_rf(output_path)
+	# Delete existing output when explicitly requested:
+	if force
+		FileUtils.rm_rf(output_path)
+	end
 
 	# Copy all public assets:
 	FileUtils::Verbose.mkpath(output_path)
-	Dir.glob(File.join(Dir.pwd, "public/*")) do |path|
+	Dir.glob(File.join(public_path, "*")) do |path|
 		FileUtils::Verbose.cp_r(path, output_path)
 	end
 
-	# Generate HTML pages:
-	system("wget", "--mirror", "--recursive", "--continue", "--convert-links", "--adjust-extension", "--no-host-directories", "--directory-prefix", output_path.to_s, "http://localhost:#{server_port}")
-	
-	container.stop
+	begin
+		# Generate HTML pages:
+		unless system("wget", "--mirror", "--recursive", "--continue", "--convert-links", "--adjust-extension", "--no-host-directories", "--directory-prefix", output_path.to_s, "http://localhost:#{server_port}")
+			raise "Static site generation failed!"
+		end
+	ensure
+		container.stop
+	end
 end