From 3b9a5c4a4a8cf1646fe1c1caa184c91d1300657d Mon Sep 17 00:00:00 2001
From: Blankeos <carloantonioct@gmail.com>
Date: Fri, 18 Jul 2025 20:55:07 +0800
Subject: [PATCH] fix: added postgresql:// to be inferred as postgres db type.

---
 src/main.rs | 11 ++++++-----
 1 file changed, 6 insertions(+), 5 deletions(-)

diff --git a/src/main.rs b/src/main.rs
index 7a8aa53..53d63ac 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -88,11 +88,12 @@ pub enum DatabaseType {
 }
 
 async fn generate_rust_from_database(args: &Cli) -> DbSetsFsWriter {
-    let database_type = if args.db_url.starts_with("postgres://") {
-        DatabaseType::Postgres
-    } else {
-        DatabaseType::MySql
-    };
+    let database_type =
+        if args.db_url.starts_with("postgres://") || args.db_url.starts_with("postgresql://") {
+            DatabaseType::Postgres
+        } else {
+            DatabaseType::MySql
+        };
 
     if args.mode == Mode::Dbset && database_type == DatabaseType::MySql {
         panic!("DbSet not currently supported for MySql")