From ee58d9902c6a6e2882c73db7a3f7fa167dd03dea Mon Sep 17 00:00:00 2001
From: Elijah Perrault <elijah@starcoder.info>
Date: Fri, 11 Feb 2011 14:38:40 -0700
Subject: [PATCH] Support for more advanced modules that have their own
 folders.

---
 bin/auto | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/bin/auto b/bin/auto
index 7e343eb..d82abec 100755
--- a/bin/auto
+++ b/bin/auto
@@ -172,7 +172,7 @@ if (!-e "$Bin/../etc/auto.db") {
     system "touch $Bin/../etc/auto.db";
     system "chmod a+x $Bin/../etc/auto.db";
 }
-our $DB = DBI->connect("dbi:SQLite:dbname=$Bin/../etc/auto.db", q{}, q{});
+our $DB = DBI->connect("dbi:SQLite:dbname=$Bin/../etc/auto.db", q{}, q{}) or exit;
 
 # Parse privileges.
 our (%PRIVILEGES);
@@ -446,8 +446,13 @@ sub mod_load {
     if (-e "$Bin/../modules/$module.pm") {
         do "$Bin/../modules/$module.pm" and return 1;
     }
+    else {
+        if (-e "$Bin/../modules/$module/main.pm") {
+            do "$Bin/../modules/$module/main.pm" and return 1;
+        }
+    }
 
-    return 0;
+    return;
 }
 
 # vim: set ai sw=4 ts=4: