]> git.ktnx.net Git - mobile-ledger.git/blob - MLDB.java
1073a88f58edc8c7745452de5e0a30ae7ae3c3fb
[mobile-ledger.git] / MLDB.java
1 /*
2  * Copyright © 2018 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.content.Context;
21 import android.content.res.Resources;
22 import android.database.Cursor;
23 import android.database.SQLException;
24 import android.database.sqlite.SQLiteDatabase;
25 import android.database.sqlite.SQLiteOpenHelper;
26 import android.util.Log;
27
28 import java.io.BufferedReader;
29 import java.io.IOException;
30 import java.io.InputStream;
31 import java.io.InputStreamReader;
32 import java.util.Locale;
33
34 import static net.ktnx.mobileledger.utils.MLDB.DatabaseMode.READ;
35 import static net.ktnx.mobileledger.utils.MLDB.DatabaseMode.WRITE;
36
37 public final class MLDB {
38     public enum DatabaseMode {READ, WRITE}
39
40     public static final String ACCOUNTS_TABLE = "accounts";
41     public static final String DESCRIPTION_HISTORY_TABLE = "description_history";
42     public static final String OPT_TRANSACTION_LIST_STAMP = "transaction_list_last_update";
43     private static MobileLedgerDatabase helperForReading, helperForWriting;
44
45     public static synchronized SQLiteDatabase getDatabase(Context context, DatabaseMode mode) {
46         if (mode == READ) {
47             if (helperForReading == null) helperForReading = new MobileLedgerDatabase(context);
48             return helperForReading.getReadableDatabase();
49         }
50         else {
51             if (helperForWriting == null) helperForWriting = new MobileLedgerDatabase(context);
52             return helperForWriting.getWritableDatabase();
53         }
54     }
55     public static SQLiteDatabase getReadableDatabase(Context context) {
56         return getDatabase(context, READ);
57     }
58     public static SQLiteDatabase getWritableDatabase(Context context) {
59         return getDatabase(context, WRITE);
60     }
61     static public int get_option_value(Context context, String name, int default_value) {
62         String s = get_option_value(context, name, String.valueOf(default_value));
63         try {
64             return Integer.parseInt(s);
65         }
66         catch (Exception e) {
67             Log.d("db", "returning default int value of " + name, e);
68             return default_value;
69         }
70     }
71
72     static public long get_option_value(Context context, String name, long default_value) {
73         String s = get_option_value(context, name, String.valueOf(default_value));
74         try {
75             return Long.parseLong(s);
76         }
77         catch (Exception e) {
78             Log.d("db", "returning default long value of " + name, e);
79             return default_value;
80         }
81     }
82
83     static public String get_option_value(Context context, String name, String default_value) {
84         Log.d("db", "about to fetch option " + name);
85         try (SQLiteDatabase db = getReadableDatabase(context)) {
86             try (Cursor cursor = db
87                     .rawQuery("select value from options where name=?", new String[]{name}))
88             {
89                 if (cursor.moveToFirst()) {
90                     String result = cursor.getString(0);
91
92                     if (result == null) result = default_value;
93
94                     Log.d("db", "option " + name + "=" + result);
95                     return result;
96                 }
97                 else return default_value;
98             }
99             catch (Exception e) {
100                 Log.d("db", "returning default value for " + name, e);
101                 return default_value;
102             }
103         }
104     }
105
106     static public void set_option_value(Context context, String name, String value) {
107         Log.d("db", "setting option " + name + "=" + value);
108         try (SQLiteDatabase db = getWritableDatabase(context)) {
109             db.execSQL("insert or replace into options(name, value) values(?, ?);",
110                     new String[]{name, value});
111         }
112     }
113
114     static public void set_option_value(Context context, String name, long value) {
115         set_option_value(context, name, String.valueOf(value));
116     }
117 }
118
119 class MobileLedgerDatabase extends SQLiteOpenHelper implements AutoCloseable {
120     public static final String DB_NAME = "mobile-ledger.db";
121     public static final int LATEST_REVISION = 10;
122
123     private final Context mContext;
124
125     public MobileLedgerDatabase(Context context) {
126         super(context, DB_NAME, null, LATEST_REVISION);
127         Log.d("db", "creating helper instance");
128         mContext = context;
129         super.setWriteAheadLoggingEnabled(true);
130     }
131
132     @Override
133     public void onCreate(SQLiteDatabase db) {
134         Log.d("db", "onCreate called");
135         onUpgrade(db, -1, LATEST_REVISION);
136     }
137
138     @Override
139     public void onUpgrade(SQLiteDatabase db, int oldVersion, int newVersion) {
140         Log.d("db", "onUpgrade called");
141         for (int i = oldVersion + 1; i <= newVersion; i++) applyRevision(db, i);
142     }
143
144     private void applyRevision(SQLiteDatabase db, int rev_no) {
145         final Resources rm = mContext.getResources();
146         String rev_file = String.format(Locale.US, "sql_%d", rev_no);
147
148         int res_id = rm.getIdentifier(rev_file, "raw", mContext.getPackageName());
149         if (res_id == 0) throw new SQLException(
150                 String.format(Locale.US, "No resource for revision %d", rev_no));
151         db.beginTransaction();
152         try (InputStream res = rm.openRawResource(res_id)) {
153             Log.d("db", "Applying revision " + String.valueOf(rev_no));
154             InputStreamReader isr = new InputStreamReader(res);
155             BufferedReader reader = new BufferedReader(isr);
156
157             String line;
158             while ((line = reader.readLine()) != null) {
159                 db.execSQL(line);
160             }
161
162             db.setTransactionSuccessful();
163         }
164         catch (IOException e) {
165             Log.e("db", String.format("Error opening raw resource for revision %d", rev_no));
166             e.printStackTrace();
167         }
168         finally {
169             db.endTransaction();
170         }
171     }
172 }