From 906f8b447feca24360bb4a9a67a5cf18ba67d6ef Mon Sep 17 00:00:00 2001
From: Roman Langolf <rolang@pm.me>
Date: Mon, 12 Jan 2026 13:57:17 +0700
Subject: [PATCH] add missing docker cleanup on failures

---
 PgCodeGen.scala | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/PgCodeGen.scala b/PgCodeGen.scala
index 0c7b824..a05c92f 100644
--- a/PgCodeGen.scala
+++ b/PgCodeGen.scala
@@ -787,7 +787,9 @@ object PgCodeGen {
                   Future.failed(Exception(s"Cannot find any .sql files in ${sourceDir.toPath()}"))
                 else Future.unit
               _ = println("Generating Postgres models")
-              db <- initGeneratorDatabase(useConnection)
+              db <- initGeneratorDatabase(useConnection).transformWith:
+                case Success(db)  => Future.successful(db)
+                case Failure(err) => cleanup.flatMap(_ => Future.failed(err))
               codegen = PgCodeGen(
                 pkgName = pkgName,
                 sourceFiles = sourceFiles,