add cleanup step

Author: felangel

bricks/dart_frog_prod_server/hooks/lib/dart_frog_prod_server_hooks.dart

@@ -7,3 +7,6 @@ export 'src/exit_overrides.dart';
 export 'src/get_internal_path_dependencies.dart';
 export 'src/get_pubspec_lock.dart';
 export 'src/uses_workspace_resolution.dart';
+
+/// A void callback function (e.g. `void Function()`).
+typedef VoidCallback = void Function();

bricks/dart_frog_prod_server/hooks/lib/src/copy_workspace_pubspec_lock.dart

@@ -1,12 +1,13 @@
 import 'dart:io';
+import 'package:dart_frog_prod_server_hooks/dart_frog_prod_server_hooks.dart';
 import 'package:mason/mason.dart';
 import 'package:path/path.dart' as path;
 import 'package:yaml/yaml.dart';
 
 /// Copies the pubspec.lock from the workspace root into the project directory
 /// in order to ensure the production build uses the exact same versions of all
 /// dependencies.
-void copyWorkspacePubspecLock(
+VoidCallback copyWorkspacePubspecLock(
   HookContext context, {
   required String projectDirectory,
   required void Function(int exitCode) exit,
@@ -16,17 +17,22 @@ void copyWorkspacePubspecLock(
     context.logger.err(
       'Unable to determine workspace root for $projectDirectory',
     );
-    return exit(1);
+    exit(1);
+    return () {};
   }
 
   final pubspecLockFile = File(path.join(workspaceRoot.path, 'pubspec.lock'));
-  if (!pubspecLockFile.existsSync()) return;
+  if (!pubspecLockFile.existsSync()) return () {};
 
   try {
     pubspecLockFile.copySync(path.join(projectDirectory, 'pubspec.lock'));
+    return () {
+      File(path.join(projectDirectory, 'pubspec.lock')).delete().ignore();
+    };
   } on Exception catch (error) {
     context.logger.err('$error');
-    return exit(1);
+    exit(1);
+    return () {};
   }
 }
 

bricks/dart_frog_prod_server/hooks/lib/src/disable_workspace_resolution.dart

@@ -1,12 +1,10 @@
 import 'dart:io';
+import 'package:dart_frog_prod_server_hooks/dart_frog_prod_server_hooks.dart';
 import 'package:mason/mason.dart';
 import 'package:path/path.dart' as path;
 import 'package:yaml/yaml.dart';
 import 'package:yaml_edit/yaml_edit.dart';
 
-/// A void callback function (e.g. `void Function()`).
-typedef VoidCallback = void Function();
-
 /// Opts out of dart workspaces until we can generate per package lockfiles.
 /// https://github.com/dart-lang/pub/issues/4594
 VoidCallback disableWorkspaceResolution(

bricks/dart_frog_prod_server/hooks/pre_gen.dart

@@ -30,6 +30,7 @@ Future<void> preGen(
   );
 
   VoidCallback? restoreWorkspaceResolution;
+  VoidCallback? revertPubspecLock;
 
   if (usesWorkspaces) {
     // Disable workspace resolution until we can generate per-package lockfiles.
@@ -41,7 +42,7 @@ Future<void> preGen(
     );
     // Copy the pubspec.lock from the workspace root to ensure the same versions
     // of dependencies are used in the production build.
-    copyWorkspacePubspecLock(
+    revertPubspecLock = copyWorkspacePubspecLock(
       context,
       projectDirectory: projectDirectory.path,
       exit: exit,
@@ -101,10 +102,6 @@ Future<void> preGen(
     onViolationEnd: () => exit(1),
   );
 
-  final customDockerFile = io.File(
-    path.join(projectDirectory.path, 'Dockerfile'),
-  );
-
   final internalPathDependencies = await getInternalPathDependencies(
     projectDirectory,
   );
@@ -115,6 +112,11 @@ Future<void> preGen(
     copyPath: copyPath,
   );
 
+  revertPubspecLock?.call();
+
+  final customDockerFile = io.File(
+    path.join(projectDirectory.path, 'Dockerfile'),
+  );
   final addDockerfile = !customDockerFile.existsSync();
 
   context.vars = {