From b2dea8d09b0f4186341cb5eb1496a80df3ee4f1f Mon Sep 17 00:00:00 2001
From: Harrison Deng <yunyangdeng@gmail.com>
Date: Thu, 20 Jan 2022 17:11:44 -0600
Subject: [PATCH] Now detects pom.xml in root directory.

---
 gen.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/gen.py b/gen.py
index 045008f..459e018 100644
--- a/gen.py
+++ b/gen.py
@@ -163,11 +163,12 @@ def find_pomxml() -> str:
 
 def find_pomxmldir() -> str:
     directories = os.listdir()
-    
+    if os.path.exists("./pom.xml"):
+        return "./"
+
     for directory in directories:
         if os.path.isdir(directory) and "pom.xml" in os.listdir(directory):
             return directory
-    return ""
 
 def __printhelp():
     print("usage: {0} <setup | project | server | reset>".format(sys.argv[0]))