Changed the init to Setup() for better control over DB creation

gophish.go

@@ -37,10 +37,11 @@ import (
 
 func main() {
 	//Setup the global variables and settings
+	err := models.Setup()
 	defer models.DB.Close()
-	/*	if err != nil {
+	if err != nil {
 		fmt.Println(err)
-	}*/
+	}
 	fmt.Printf("Gophish server started at http://%s\n", config.Conf.URL)
 	http.Handle("/", controllers.Use(controllers.CreateRouter().ServeHTTP, middleware.GetContext))
 	http.ListenAndServe(config.Conf.URL, nil)

gophish_test.go

@@ -1,12 +1,13 @@
 package main
 
 import (
-	"github.com/jordan-wright/gophish/db"
 	"testing"
+
+	"github.com/jordan-wright/gophish/models"
 )
 
 func TestDBSetup(t *testing.T) {
-	err := db.Setup()
+	err := models.Setup()
 	if err != nil {
 		t.Fatalf("Failed creating database: %v", err)
 	}

models/models.go

@@ -19,7 +19,7 @@ var Logger = log.New(os.Stdout, "", log.Ldate|log.Ltime|log.Lshortfile)
 
 // Setup initializes the Conn object
 // It also populates the Gophish Config object
-func init() {
+func Setup() error {
 	DB, err := sql.Open("sqlite3", config.Conf.DBPath)
 	Conn = &gorp.DbMap{Db: DB, Dialect: gorp.SqliteDialect{}}
 	//If the file already exists, delete it and recreate it
@@ -50,7 +50,7 @@ func init() {
 		for _, stmt := range createTablesSQL {
 			_, err = DB.Exec(stmt)
 			if err != nil {
-				/*				return nil, err*/
+				return err
 			}
 		}
 		//Create the default user
@@ -64,7 +64,7 @@ func init() {
 			Logger.Println(err)
 		}
 	}
-	/*	return Conn, nil*/
+	return nil
 }
 
 // Flash is used to hold flash information for use in templates.