Manually init classes list

library/src/main/java/com/orm/SchemaGenerator.java

@@ -44,8 +44,8 @@ public static SchemaGenerator getInstance() {
         return new SchemaGenerator();
     }
 
-    public void createDatabase(SQLiteDatabase sqLiteDatabase) {
-        List<Class> domainClasses = getDomainClasses();
+    public void createDatabase(SQLiteDatabase sqLiteDatabase, List<Class> modelsClasses) {
+        List<Class> domainClasses = modelsClasses == null ? getDomainClasses() : modelsClasses;
         for (Class domain : domainClasses) {
             createTable(domain, sqLiteDatabase);
             afterTableCreated(domain,sqLiteDatabase);
@@ -59,8 +59,8 @@ public void afterTableCreated(Class<?> table, SQLiteDatabase sqLiteDatabase) {
 
     }
 
-    public void doUpgrade(SQLiteDatabase sqLiteDatabase, int oldVersion, int newVersion) {
-        List<Class> domainClasses = getDomainClasses();
+    public void doUpgrade(SQLiteDatabase sqLiteDatabase, int oldVersion, int newVersion, List<Class> modelsClasses) {
+        List<Class> domainClasses = modelsClasses == null ? getDomainClasses() : modelsClasses;
         String sql = "select count(*) from sqlite_master where type='table' and name='%s';";
 
         for (Class domain : domainClasses) {

library/src/main/java/com/orm/SugarApp.java

@@ -2,6 +2,8 @@
 
 import android.app.Application;
 
+import java.util.List;
+
 public class SugarApp extends Application {
 
     @Override
@@ -16,4 +18,8 @@ public void onTerminate() {
         SugarContext.terminate();
     }
 
+    public List<Class> getModelsClassesList() {
+        return null;
+    }
+
 }

library/src/main/java/com/orm/SugarContext.java

@@ -5,6 +5,7 @@
 import com.orm.util.ContextUtil;
 
 import java.util.Collections;
+import java.util.List;
 import java.util.Map;
 import java.util.WeakHashMap;
 
@@ -15,8 +16,8 @@ public class SugarContext {
     private SugarDb sugarDb;
     private Map<Object, Long> entitiesMap;
 
-    private SugarContext() {
-        this.sugarDb = SugarDb.getInstance();
+    private SugarContext(List<Class> modelsClasses) {
+        this.sugarDb = SugarDb.getInstance(modelsClasses);
         this.entitiesMap = Collections.synchronizedMap(new WeakHashMap<Object, Long>());
     }
 
@@ -29,7 +30,13 @@ public static SugarContext getSugarContext() {
 
     public static void init(Context context) {
         ContextUtil.init(context);
-        instance = new SugarContext();
+        instance = new SugarContext(null);
+        dbConfiguration = null;
+    }
+
+    public static void init(Context context, List<Class> modelsClasses) {
+        ContextUtil.init(context);
+        instance = new SugarContext(modelsClasses);
         dbConfiguration = null;
     }
 

library/src/main/java/com/orm/SugarDb.java

@@ -8,6 +8,8 @@
 import com.orm.helper.ManifestHelper;
 import com.orm.util.SugarCursorFactory;
 
+import java.util.List;
+
 import static com.orm.util.ContextUtil.getContext;
 import static com.orm.helper.ManifestHelper.getDatabaseVersion;
 import static com.orm.helper.ManifestHelper.getDbName;
@@ -19,20 +21,26 @@ public class SugarDb extends SQLiteOpenHelper {
     private final SchemaGenerator schemaGenerator;
     private SQLiteDatabase sqLiteDatabase;
     private int openedConnections = 0;
+    private List<Class> classes;
 
     //Prevent instantiation
-    private SugarDb() {
+    private SugarDb(List<Class> modelsClasses) {
         super(getContext(), getDbName(), new SugarCursorFactory(ManifestHelper.isDebugEnabled()), getDatabaseVersion());
+        this.classes = modelsClasses;
         schemaGenerator = SchemaGenerator.getInstance();
     }
 
     public static SugarDb getInstance() {
-        return new SugarDb();
+        return new SugarDb(null);
+    }
+
+    public static SugarDb getInstance(List<Class> modelsClasses) {
+        return new SugarDb(modelsClasses);
     }
 
     @Override
     public void onCreate(SQLiteDatabase sqLiteDatabase) {
-        schemaGenerator.createDatabase(sqLiteDatabase);
+        schemaGenerator.createDatabase(sqLiteDatabase, classes);
     }
 
     @Override
@@ -50,7 +58,7 @@ public void onConfigure(SQLiteDatabase db) {
 
     @Override
     public void onUpgrade(SQLiteDatabase sqLiteDatabase, int oldVersion, int newVersion) {
-        schemaGenerator.doUpgrade(sqLiteDatabase, oldVersion, newVersion);
+        schemaGenerator.doUpgrade(sqLiteDatabase, oldVersion, newVersion, classes);
     }
 
     public synchronized SQLiteDatabase getDB() {

library/src/test/java/com/orm/SchemaGeneratorTest.java

@@ -131,7 +131,7 @@ public void testAllTableCreation() {
         SQLiteDatabase sqLiteDatabase = SugarContext.getSugarContext().getSugarDb().getDB();
         SchemaGenerator schemaGenerator = SchemaGenerator.getInstance();
 
-        schemaGenerator.createDatabase(sqLiteDatabase);
+        schemaGenerator.createDatabase(sqLiteDatabase, null);
         String sql = "select count(*) from sqlite_master where type='table';";
 
         Cursor c = sqLiteDatabase.rawQuery(sql, null);
@@ -150,7 +150,7 @@ public void testDeleteAllTables() {
         SQLiteDatabase sqLiteDatabase = SugarContext.getSugarContext().getSugarDb().getDB();
         SchemaGenerator schemaGenerator = SchemaGenerator.getInstance();
 
-        schemaGenerator.createDatabase(sqLiteDatabase);
+        schemaGenerator.createDatabase(sqLiteDatabase, null);
         schemaGenerator.deleteTables(sqLiteDatabase);
 
         String sql = "select count(*) from sqlite_master where type='table';";