443e0a9a5bdc2eedffeb3e55472d798c2be8065d
[mobile-ledger.git] / app / src / main / java / net / ktnx / mobileledger / utils / MLDB.java
1 /*
2  * Copyright © 2019 Damyan Ivanov.
3  * This file is part of Mobile-Ledger.
4  * Mobile-Ledger 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  * Mobile-Ledger 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 Mobile-Ledger. 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.content.Context;
22 import android.content.res.Resources;
23 import android.database.Cursor;
24 import android.database.MatrixCursor;
25 import android.database.SQLException;
26 import android.database.sqlite.SQLiteDatabase;
27 import android.database.sqlite.SQLiteOpenHelper;
28 import android.os.Build;
29 import android.provider.FontsContract;
30 import android.util.Log;
31 import android.widget.AutoCompleteTextView;
32 import android.widget.FilterQueryProvider;
33 import android.widget.SimpleCursorAdapter;
34
35 import java.io.BufferedReader;
36 import java.io.IOException;
37 import java.io.InputStream;
38 import java.io.InputStreamReader;
39 import java.util.Locale;
40
41 import static net.ktnx.mobileledger.utils.MLDB.DatabaseMode.READ;
42 import static net.ktnx.mobileledger.utils.MLDB.DatabaseMode.WRITE;
43
44 public final class MLDB {
45     public static final String ACCOUNTS_TABLE = "accounts";
46     public static final String DESCRIPTION_HISTORY_TABLE = "description_history";
47     public static final String OPT_TRANSACTION_LIST_STAMP = "transaction_list_last_update";
48     public static final String OPT_LAST_REFRESH = "last_refresh";
49     private static MobileLedgerDatabase helperForReading, helperForWriting;
50     private static Context context;
51     private static void checkState() {
52         if (context == null)
53             throw new IllegalStateException("First call init with a valid context");
54     }
55     public static synchronized SQLiteDatabase getDatabase(DatabaseMode mode) {
56         checkState();
57
58         if (mode == READ) {
59             if (helperForReading == null) helperForReading = new MobileLedgerDatabase(context);
60             return helperForReading.getReadableDatabase();
61         }
62         else {
63             if (helperForWriting == null) helperForWriting = new MobileLedgerDatabase(context);
64             return helperForWriting.getWritableDatabase();
65         }
66     }
67     public static SQLiteDatabase getReadableDatabase() {
68         return getDatabase(READ);
69     }
70     public static SQLiteDatabase getWritableDatabase() {
71         return getDatabase(WRITE);
72     }
73     static public int get_option_value(String name, int default_value) {
74         String s = get_option_value(name, String.valueOf(default_value));
75         try {
76             return Integer.parseInt(s);
77         }
78         catch (Exception e) {
79             Log.d("db", "returning default int value of " + name, e);
80             return default_value;
81         }
82     }
83     static public long get_option_value(String name, long default_value) {
84         String s = get_option_value(name, String.valueOf(default_value));
85         try {
86             return Long.parseLong(s);
87         }
88         catch (Exception e) {
89             Log.d("db", "returning default long value of " + name, e);
90             return default_value;
91         }
92     }
93     static public String get_option_value(String name, String default_value) {
94         Log.d("db", "about to fetch option " + name);
95         try (SQLiteDatabase db = getReadableDatabase()) {
96             try (Cursor cursor = db
97                     .rawQuery("select value from options where name=?", new String[]{name}))
98             {
99                 if (cursor.moveToFirst()) {
100                     String result = cursor.getString(0);
101
102                     if (result == null) result = default_value;
103
104                     Log.d("db", "option " + name + "=" + result);
105                     return result;
106                 }
107                 else return default_value;
108             }
109             catch (Exception e) {
110                 Log.d("db", "returning default value for " + name, e);
111                 return default_value;
112             }
113         }
114     }
115     static public void set_option_value(String name, String value) {
116         Log.d("db", "setting option " + name + "=" + value);
117         try (SQLiteDatabase db = getWritableDatabase()) {
118             db.execSQL("insert or replace into options(name, value) values(?, ?);",
119                     new String[]{name, value});
120         }
121     }
122     static public void set_option_value(String name, long value) {
123         set_option_value(name, String.valueOf(value));
124     }
125     @TargetApi(Build.VERSION_CODES.N)
126     public static void hook_autocompletion_adapter(final Context context,
127                                                    final AutoCompleteTextView view,
128                                                    final String table, final String field) {
129         String[] from = {field};
130         int[] to = {android.R.id.text1};
131         SimpleCursorAdapter adapter =
132                 new SimpleCursorAdapter(context, android.R.layout.simple_dropdown_item_1line, null,
133                         from, to, 0);
134         adapter.setStringConversionColumn(1);
135
136         FilterQueryProvider provider = new FilterQueryProvider() {
137             @Override
138             public Cursor runQuery(CharSequence constraint) {
139                 if (constraint == null) return null;
140
141                 String str = constraint.toString().toUpperCase();
142                 Log.d("autocompletion", "Looking for " + str);
143                 String[] col_names = {FontsContract.Columns._ID, field};
144                 MatrixCursor c = new MatrixCursor(col_names);
145
146                 try (SQLiteDatabase db = MLDB.getReadableDatabase()) {
147
148                     try (Cursor matches = db.rawQuery(String.format(
149                             "SELECT %s as a, case when %s_upper LIKE ?||'%%' then 1 " +
150                             "WHEN %s_upper LIKE '%%:'||?||'%%' then 2 " +
151                             "WHEN %s_upper LIKE '%% '||?||'%%' then 3 " + "else 9 end " +
152                             "FROM %s " + "WHERE %s_upper LIKE '%%'||?||'%%' " + "ORDER BY 2, 1;",
153                             field, field, field, field, table, field),
154                             new String[]{str, str, str, str}))
155                     {
156                         int i = 0;
157                         while (matches.moveToNext()) {
158                             String match = matches.getString(0);
159                             int order = matches.getInt(1);
160                             Log.d("autocompletion", String.format("match: %s |%d", match, order));
161                             c.newRow().add(i++).add(match);
162                         }
163                     }
164
165                     return c;
166                 }
167
168             }
169         };
170
171         adapter.setFilterQueryProvider(provider);
172
173         view.setAdapter(adapter);
174     }
175     public static void init(Context context) {
176         MLDB.context = context.getApplicationContext();
177     }
178     public enum DatabaseMode {READ, WRITE}
179 }
180
181 class MobileLedgerDatabase extends SQLiteOpenHelper implements AutoCloseable {
182     public static final String DB_NAME = "mobile-ledger.db";
183     public static final int LATEST_REVISION = 10;
184
185     private final Context mContext;
186
187     public MobileLedgerDatabase(Context context) {
188         super(context, DB_NAME, null, LATEST_REVISION);
189         Log.d("db", "creating helper instance");
190         mContext = context;
191         super.setWriteAheadLoggingEnabled(true);
192     }
193
194     @Override
195     public void onCreate(SQLiteDatabase db) {
196         Log.d("db", "onCreate called");
197         onUpgrade(db, -1, LATEST_REVISION);
198     }
199
200     @Override
201     public void onUpgrade(SQLiteDatabase db, int oldVersion, int newVersion) {
202         Log.d("db", "onUpgrade called");
203         for (int i = oldVersion + 1; i <= newVersion; i++) applyRevision(db, i);
204     }
205
206     private void applyRevision(SQLiteDatabase db, int rev_no) {
207         final Resources rm = mContext.getResources();
208         String rev_file = String.format(Locale.US, "sql_%d", rev_no);
209
210         int res_id = rm.getIdentifier(rev_file, "raw", mContext.getPackageName());
211         if (res_id == 0)
212             throw new SQLException(String.format(Locale.US, "No resource for revision %d", rev_no));
213         db.beginTransaction();
214         try (InputStream res = rm.openRawResource(res_id)) {
215             Log.d("db", "Applying revision " + String.valueOf(rev_no));
216             InputStreamReader isr = new InputStreamReader(res);
217             BufferedReader reader = new BufferedReader(isr);
218
219             String line;
220             while ((line = reader.readLine()) != null) {
221                 db.execSQL(line);
222             }
223
224             db.setTransactionSuccessful();
225         }
226         catch (IOException e) {
227             Log.e("db", String.format("Error opening raw resource for revision %d", rev_no));
228             e.printStackTrace();
229         }
230         finally {
231             db.endTransaction();
232         }
233     }
234 }