app/src/main/java/net/ktnx/mobileledger/utils/MLDB.java

   1 /*
   2  * Copyright © 2019 Damyan Ivanov.
   3  * This file is part of MoLe.
   4  * MoLe is free software: you can distribute it and/or modify it
   5  * under the term of the GNU General Public License as published by
   6  * the Free Software Foundation, either version 3 of the License, or
   7  * (at your opinion), any later version.
   8  *
   9  * MoLe is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  12  * GNU General Public License terms for details.
  13  *
  14  * You should have received a copy of the GNU General Public License
  15  * along with MoLe. If not, see <https://www.gnu.org/licenses/>.
  16  */
  17
  18 package net.ktnx.mobileledger.utils;
  19
  20 import android.annotation.TargetApi;
  21 import android.app.Application;
  22 import android.content.Context;
  23 import android.content.res.Resources;
  24 import android.database.Cursor;
  25 import android.database.MatrixCursor;
  26 import android.database.SQLException;
  27 import android.database.sqlite.SQLiteDatabase;
  28 import android.database.sqlite.SQLiteOpenHelper;
  29 import android.os.Build;
  30 import android.provider.FontsContract;
  31 import android.util.Log;
  32 import android.view.View;
  33 import android.widget.AutoCompleteTextView;
  34 import android.widget.FilterQueryProvider;
  35 import android.widget.SimpleCursorAdapter;
  36
  37 import net.ktnx.mobileledger.async.DescriptionSelectedCallback;
  38 import net.ktnx.mobileledger.model.Data;
  39
  40 import org.jetbrains.annotations.NonNls;
  41
  42 import java.io.BufferedReader;
  43 import java.io.IOException;
  44 import java.io.InputStream;
  45 import java.io.InputStreamReader;
  46 import java.util.Locale;
  47
  48 import static net.ktnx.mobileledger.utils.MLDB.DatabaseMode.READ;
  49 import static net.ktnx.mobileledger.utils.MLDB.DatabaseMode.WRITE;
  50
  51 public final class MLDB {
  52     public static final String ACCOUNTS_TABLE = "accounts";
  53     public static final String DESCRIPTION_HISTORY_TABLE = "description_history";
  54     public static final String OPT_LAST_SCRAPE = "last_scrape";
  55     @NonNls
  56     public static final String OPT_PROFILE_UUID = "profile_uuid";
  57     private static final String NO_PROFILE = "-";
  58     private static MobileLedgerDatabase helperForReading, helperForWriting;
  59     private static Application context;
  60     private static void checkState() {
  61         if (context == null)
  62             throw new IllegalStateException("First call init with a valid context");
  63     }
  64     public static synchronized SQLiteDatabase getDatabase(DatabaseMode mode) {
  65         checkState();
  66
  67         SQLiteDatabase db;
  68
  69         if (mode == READ) {
  70             if (helperForReading == null) helperForReading = new MobileLedgerDatabase(context);
  71             db = helperForReading.getReadableDatabase();
  72         }
  73         else {
  74             if (helperForWriting == null) helperForWriting = new MobileLedgerDatabase(context);
  75             db = helperForWriting.getWritableDatabase();
  76         }
  77
  78         db.execSQL("pragma case_sensitive_like=ON;");
  79         return db;
  80     }
  81     public static SQLiteDatabase getReadableDatabase() {
  82         return getDatabase(READ);
  83     }
  84     public static SQLiteDatabase getWritableDatabase() {
  85         return getDatabase(WRITE);
  86     }
  87     static public int getIntOption(String name, int default_value) {
  88         String s = getOption(name, String.valueOf(default_value));
  89         try {
  90             return Integer.parseInt(s);
  91         }
  92         catch (Exception e) {
  93             Log.d("db", "returning default int value of " + name, e);
  94             return default_value;
  95         }
  96     }
  97     static public long getLongOption(String name, long default_value) {
  98         String s = getOption(name, String.valueOf(default_value));
  99         try {
 100             return Long.parseLong(s);
 101         }
 102         catch (Exception e) {
 103             Log.d("db", "returning default long value of " + name, e);
 104             return default_value;
 105         }
 106     }
 107     static public String getOption(String name, String default_value) {
 108         Log.d("db", "about to fetch option " + name);
 109         SQLiteDatabase db = getReadableDatabase();
 110         try (Cursor cursor = db.rawQuery("select value from options where profile = ? and name=?",
 111                 new String[]{NO_PROFILE, name}))
 112         {
 113             if (cursor.moveToFirst()) {
 114                 String result = cursor.getString(0);
 115
 116                 if (result == null) result = default_value;
 117
 118                 Log.d("db", "option " + name + "=" + result);
 119                 return result;
 120             }
 121             else return default_value;
 122         }
 123         catch (Exception e) {
 124             Log.d("db", "returning default value for " + name, e);
 125             return default_value;
 126         }
 127     }
 128     static public void setOption(String name, String value) {
 129         Log.d("option", String.format("%s := %s", name, value));
 130         SQLiteDatabase db = MLDB.getWritableDatabase();
 131         db.execSQL("insert or replace into options(profile, name, value) values(?, ?, ?);",
 132                 new String[]{NO_PROFILE, name, value});
 133     }
 134     static public void setLongOption(String name, long value) {
 135         setOption(name, String.valueOf(value));
 136     }
 137     @TargetApi(Build.VERSION_CODES.N)
 138     public static void hookAutocompletionAdapter(final Context context,
 139                                                  final AutoCompleteTextView view,
 140                                                  final String table, final String field,
 141                                                  final boolean profileSpecific) {
 142         hookAutocompletionAdapter(context, view, table, field, profileSpecific, null, null);
 143     }
 144     @TargetApi(Build.VERSION_CODES.N)
 145     public static void hookAutocompletionAdapter(final Context context,
 146                                                  final AutoCompleteTextView view,
 147                                                  final String table, final String field,
 148                                                  final boolean profileSpecific,
 149                                                  final View nextView,
 150                                                  final DescriptionSelectedCallback callback) {
 151         String[] from = {field};
 152         int[] to = {android.R.id.text1};
 153         SimpleCursorAdapter adapter =
 154                 new SimpleCursorAdapter(context, android.R.layout.simple_dropdown_item_1line, null,
 155                         from, to, 0);
 156         adapter.setStringConversionColumn(1);
 157
 158         FilterQueryProvider provider = constraint -> {
 159             if (constraint == null) return null;
 160
 161             String str = constraint.toString().toUpperCase();
 162             Log.d("autocompletion", "Looking for " + str);
 163             String[] col_names = {FontsContract.Columns._ID, field};
 164             MatrixCursor c = new MatrixCursor(col_names);
 165
 166             String sql;
 167             String[] params;
 168             if (profileSpecific) {
 169                 sql = String.format("SELECT %s as a, case when %s_upper LIKE ?||'%%' then 1 " +
 170                                     "WHEN %s_upper LIKE '%%:'||?||'%%' then 2 " +
 171                                     "WHEN %s_upper LIKE '%% '||?||'%%' then 3 else 9 end " +
 172                                     "FROM %s " +
 173                                     "WHERE profile=? AND %s_upper LIKE '%%'||?||'%%' " +
 174                                     "ORDER BY 2, 1;", field, field, field, field, table, field);
 175                 params = new String[]{str, str, str, Data.profile.get().getUuid(), str};
 176             }
 177             else {
 178                 sql = String.format("SELECT %s as a, case when %s_upper LIKE ?||'%%' then 1 " +
 179                                     "WHEN %s_upper LIKE '%%:'||?||'%%' then 2 " +
 180                                     "WHEN %s_upper LIKE '%% '||?||'%%' then 3 " + "else 9 end " +
 181                                     "FROM %s " + "WHERE %s_upper LIKE '%%'||?||'%%' " +
 182                                     "ORDER BY 2, 1;", field, field, field, field, table, field);
 183                 params = new String[]{str, str, str, str};
 184             }
 185             Log.d("autocompletion", sql);
 186             SQLiteDatabase db = MLDB.getReadableDatabase();
 187
 188             try (Cursor matches = db.rawQuery(sql, params)) {
 189                 int i = 0;
 190                 while (matches.moveToNext()) {
 191                     String match = matches.getString(0);
 192                     int order = matches.getInt(1);
 193                     Log.d("autocompletion", String.format("match: %s |%d", match, order));
 194                     c.newRow().add(i++).add(match);
 195                 }
 196             }
 197
 198             return c;
 199
 200         };
 201
 202         adapter.setFilterQueryProvider(provider);
 203
 204         view.setAdapter(adapter);
 205
 206         if (nextView != null) {
 207             view.setOnItemClickListener((parent, itemView, position, id) -> {
 208                 nextView.requestFocus(View.FOCUS_FORWARD);
 209                 if (callback != null) {
 210                     callback.descriptionSelected(String.valueOf(view.getText()));
 211                 }
 212             });
 213         }
 214     }
 215     public static void init(Application context) {
 216         MLDB.context = context;
 217     }
 218     public static void done() {
 219         if (helperForReading != null) helperForReading.close();
 220
 221         if ((helperForWriting != helperForReading) && (helperForWriting != null))
 222             helperForWriting.close();
 223     }
 224
 225     public enum DatabaseMode {READ, WRITE}
 226 }
 227
 228 class MobileLedgerDatabase extends SQLiteOpenHelper implements AutoCloseable {
 229     public static final String DB_NAME = "MoLe.db";
 230     public static final int LATEST_REVISION = 20;
 231
 232     private final Application mContext;
 233
 234     public MobileLedgerDatabase(Application context) {
 235         super(context, DB_NAME, null, LATEST_REVISION);
 236         Log.d("db", "creating helper instance");
 237         mContext = context;
 238         super.setWriteAheadLoggingEnabled(true);
 239     }
 240
 241     @Override
 242     public void onCreate(SQLiteDatabase db) {
 243         Log.d("db", "onCreate called");
 244         onUpgrade(db, -1, LATEST_REVISION);
 245     }
 246
 247     @Override
 248     public void onUpgrade(SQLiteDatabase db, int oldVersion, int newVersion) {
 249         Log.d("db", "onUpgrade called");
 250         for (int i = oldVersion + 1; i <= newVersion; i++) applyRevision(db, i);
 251     }
 252
 253     private void applyRevision(SQLiteDatabase db, int rev_no) {
 254         final Resources rm = mContext.getResources();
 255         String rev_file = String.format(Locale.US, "sql_%d", rev_no);
 256
 257         int res_id = rm.getIdentifier(rev_file, "raw", mContext.getPackageName());
 258         if (res_id == 0)
 259             throw new SQLException(String.format(Locale.US, "No resource for revision %d", rev_no));
 260         db.beginTransaction();
 261         try (InputStream res = rm.openRawResource(res_id)) {
 262             Log.d("db", "Applying revision " + String.valueOf(rev_no));
 263             InputStreamReader isr = new InputStreamReader(res);
 264             BufferedReader reader = new BufferedReader(isr);
 265
 266             String line;
 267             int line_no = 1;
 268             while ((line = reader.readLine()) != null) {
 269                 if (line.startsWith("--")) {
 270                     line_no++;
 271                     continue;
 272                 }
 273                 if (line.isEmpty()) {
 274                     line_no++;
 275                     continue;
 276                 }
 277                 try {
 278                     db.execSQL(line);
 279                 }
 280                 catch (Exception e) {
 281                     throw new RuntimeException(
 282                             String.format("Error applying revision %d, line %d", rev_no, line_no),
 283                             e);
 284                 }
 285                 line_no++;
 286             }
 287
 288             db.setTransactionSuccessful();
 289         }
 290         catch (IOException e) {
 291             Log.e("db", String.format("Error opening raw resource for revision %d", rev_no));
 292             e.printStackTrace();
 293         }
 294         finally {
 295             db.endTransaction();
 296         }
 297     }
 298 }