package net.ktnx.mobileledger.db;
+import android.content.res.Resources;
+import android.database.Cursor;
+import android.database.SQLException;
+
import androidx.annotation.NonNull;
+import androidx.lifecycle.MutableLiveData;
import androidx.room.Database;
import androidx.room.Room;
import androidx.room.RoomDatabase;
import androidx.sqlite.db.SupportSQLiteDatabase;
import net.ktnx.mobileledger.App;
+import net.ktnx.mobileledger.dao.AccountDAO;
+import net.ktnx.mobileledger.dao.AccountValueDAO;
import net.ktnx.mobileledger.dao.CurrencyDAO;
-import net.ktnx.mobileledger.dao.PatternAccountDAO;
-import net.ktnx.mobileledger.dao.PatternHeaderDAO;
-import net.ktnx.mobileledger.utils.MobileLedgerDatabase;
+import net.ktnx.mobileledger.dao.DescriptionHistoryDAO;
+import net.ktnx.mobileledger.dao.OptionDAO;
+import net.ktnx.mobileledger.dao.ProfileDAO;
+import net.ktnx.mobileledger.dao.TemplateAccountDAO;
+import net.ktnx.mobileledger.dao.TemplateHeaderDAO;
+import net.ktnx.mobileledger.dao.TransactionAccountDAO;
+import net.ktnx.mobileledger.dao.TransactionDAO;
+import net.ktnx.mobileledger.utils.Logger;
+
+import java.io.BufferedReader;
+import java.io.IOException;
+import java.io.InputStream;
+import java.io.InputStreamReader;
+import java.util.Locale;
+import java.util.regex.Matcher;
+import java.util.regex.Pattern;
-@Database(version = 52, entities = {PatternHeader.class, PatternAccount.class, Currency.class})
+import static net.ktnx.mobileledger.utils.Logger.debug;
+
+@Database(version = DB.REVISION,
+ entities = {TemplateHeader.class, TemplateAccount.class, Currency.class, Account.class,
+ Profile.class, Option.class, AccountValue.class, DescriptionHistory.class,
+ Transaction.class, TransactionAccount.class
+ })
abstract public class DB extends RoomDatabase {
+ public static final int REVISION = 59;
+ public static final String DB_NAME = "MoLe.db";
+ public static final MutableLiveData<Boolean> initComplete = new MutableLiveData<>(false);
private static DB instance;
public static DB get() {
if (instance != null)
if (instance != null)
return instance;
- return instance =
- Room.databaseBuilder(App.instance, DB.class, MobileLedgerDatabase.DB_NAME)
- .addMigrations(new Migration[]{new Migration(51, 52) {
- @Override
- public void migrate(@NonNull SupportSQLiteDatabase db) {
- db.beginTransaction();
- try {
- db.execSQL("create index fk_pattern_accounts_pattern on " +
- "pattern_accounts(pattern_id);");
- db.execSQL("create index fk_pattern_accounts_currency on " +
- "pattern_accounts(currency);");
- db.setTransactionSuccessful();
- }
- finally {
- db.endTransaction();
- }
+ RoomDatabase.Builder<DB> builder =
+ Room.databaseBuilder(App.instance, DB.class, DB_NAME);
+ builder.addMigrations(
+ new Migration[]{singleVersionMigration(17), singleVersionMigration(18),
+ singleVersionMigration(19), singleVersionMigration(20),
+ multiVersionMigration(20, 22), multiVersionMigration(22, 30),
+ multiVersionMigration(30, 32), multiVersionMigration(32, 34),
+ multiVersionMigration(34, 40), singleVersionMigration(41),
+ multiVersionMigration(41, 58), singleVersionMigration(59)
+ })
+ .addCallback(new Callback() {
+ @Override
+ public void onOpen(@NonNull SupportSQLiteDatabase db) {
+ super.onOpen(db);
+ db.execSQL("PRAGMA foreign_keys = ON");
+ db.execSQL("pragma case_sensitive_like" + "=ON;");
+
+ }
+ });
+
+// if (BuildConfig.DEBUG)
+// builder.setQueryCallback(((sqlQuery, bindArgs) -> Logger.debug("room", sqlQuery)),
+// Executors.newSingleThreadExecutor());
+
+ return instance = builder.build();
+ }
+ }
+ private static Migration singleVersionMigration(int toVersion) {
+ return new Migration(toVersion - 1, toVersion) {
+ @Override
+ public void migrate(@NonNull SupportSQLiteDatabase db) {
+ String fileName = String.format(Locale.US, "db_%d", toVersion);
+
+ applyRevisionFile(db, fileName);
+
+ // when migrating to version 59, migrate profile/theme options to the
+ // SharedPreferences
+ if (toVersion == 59) {
+ try (Cursor c = db.query(
+ "SELECT p.id, p.theme_hue FROM profiles p WHERE p.id=(SELECT o.value " +
+ "FROM options WHERE o.profile_uid IS NULL AND o.name=?",
+ new Object[]{"profile_id"}))
+ {
+ if (c.moveToFirst()) {
+ long currentProfileId = c.getLong(0);
+ int currentTheme = c.getInt(1);
+
+ if (currentProfileId >= 0 && currentTheme >= 0) {
+ App.storeStartupProfileAndTheme(currentProfileId, currentTheme);
}
}
- })
- .build();
+ }
+ }
+ }
+ };
+ }
+ private static Migration dummyVersionMigration(int toVersion) {
+ return new Migration(toVersion - 1, toVersion) {
+ @Override
+ public void migrate(@NonNull SupportSQLiteDatabase db) {
+ Logger.debug("db",
+ String.format(Locale.ROOT, "Dummy DB migration to version %d", toVersion));
+ }
+ };
+ }
+ private static Migration multiVersionMigration(int fromVersion, int toVersion) {
+ return new Migration(fromVersion, toVersion) {
+ @Override
+ public void migrate(@NonNull SupportSQLiteDatabase db) {
+ String fileName = String.format(Locale.US, "db_%d_%d", fromVersion, toVersion);
+
+ applyRevisionFile(db, fileName);
+ }
+ };
+ }
+ public static void applyRevisionFile(@NonNull SupportSQLiteDatabase db, String fileName) {
+ final Resources rm = App.instance.getResources();
+ int res_id = rm.getIdentifier(fileName, "raw", App.instance.getPackageName());
+ if (res_id == 0)
+ throw new SQLException(String.format(Locale.US, "No resource for %s", fileName));
+
+ try (InputStream res = rm.openRawResource(res_id)) {
+ debug("db", "Applying " + fileName);
+ InputStreamReader isr = new InputStreamReader(res);
+ BufferedReader reader = new BufferedReader(isr);
+
+ Pattern endOfStatement = Pattern.compile(";\\s*(?:--.*)?$");
+
+ String line;
+ String sqlStatement = null;
+ int lineNo = 0;
+ while ((line = reader.readLine()) != null) {
+ lineNo++;
+ if (line.startsWith("--"))
+ continue;
+ if (line.isEmpty())
+ continue;
+
+ if (sqlStatement == null)
+ sqlStatement = line;
+ else
+ sqlStatement = sqlStatement.concat(" " + line);
+
+ Matcher m = endOfStatement.matcher(line);
+ if (!m.find())
+ continue;
+
+ try {
+ db.execSQL(sqlStatement);
+ sqlStatement = null;
+ }
+ catch (Exception e) {
+ throw new RuntimeException(
+ String.format("Error applying %s, line %d, statement: %s", fileName,
+ lineNo, sqlStatement), e);
+ }
+ }
+
+ if (sqlStatement != null)
+ throw new RuntimeException(String.format(
+ "Error applying %s: EOF after continuation. Line %s, Incomplete " +
+ "statement: %s", fileName, lineNo, sqlStatement));
+
+ }
+ catch (IOException e) {
+ throw new RuntimeException(String.format("Error opening raw resource for %s", fileName),
+ e);
}
}
- public abstract PatternHeaderDAO getPatternDAO();
+ public abstract TemplateHeaderDAO getTemplateDAO();
- public abstract PatternAccountDAO getPatternAccountDAO();
+ public abstract TemplateAccountDAO getTemplateAccountDAO();
public abstract CurrencyDAO getCurrencyDAO();
+
+ public abstract AccountDAO getAccountDAO();
+
+ public abstract AccountValueDAO getAccountValueDAO();
+
+ public abstract TransactionDAO getTransactionDAO();
+
+ public abstract TransactionAccountDAO getTransactionAccountDAO();
+
+ public abstract OptionDAO getOptionDAO();
+
+ public abstract DescriptionHistoryDAO getDescriptionHistoryDAO();
+
+ public abstract ProfileDAO getProfileDAO();
}