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.
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.
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/>.
18 package net.ktnx.mobileledger.async;
20 import android.content.SharedPreferences;
21 import android.database.sqlite.SQLiteDatabase;
22 import android.os.OperationCanceledException;
23 import android.util.Log;
25 import net.ktnx.mobileledger.R;
26 import net.ktnx.mobileledger.model.Data;
27 import net.ktnx.mobileledger.model.LedgerAccount;
28 import net.ktnx.mobileledger.ui.activity.MainActivity;
29 import net.ktnx.mobileledger.utils.MLDB;
30 import net.ktnx.mobileledger.utils.NetworkUtil;
32 import java.io.BufferedReader;
33 import java.io.FileNotFoundException;
34 import java.io.IOException;
35 import java.io.InputStream;
36 import java.io.InputStreamReader;
37 import java.lang.ref.WeakReference;
38 import java.net.HttpURLConnection;
39 import java.net.MalformedURLException;
40 import java.net.URLDecoder;
41 import java.util.regex.Matcher;
42 import java.util.regex.Pattern;
44 public class RetrieveAccountsTask extends android.os.AsyncTask<Void, Integer, Void> {
46 WeakReference<MainActivity> mContext;
47 private SharedPreferences pref;
49 public RetrieveAccountsTask(WeakReference<MainActivity> context) {
54 public void setPref(SharedPreferences pref) {
58 protected Void doInBackground(Void... params) {
59 Data.backgroundTaskCount.incrementAndGet();
61 HttpURLConnection http = NetworkUtil.prepare_connection(pref, "add");
63 try (SQLiteDatabase db = MLDB.getWritableDatabase()) {
64 try (InputStream resp = http.getInputStream()) {
65 Log.d("update_accounts", String.valueOf(http.getResponseCode()));
66 if (http.getResponseCode() != 200) {
67 throw new IOException(
68 String.format("HTTP error: %d %s", http.getResponseCode(),
69 http.getResponseMessage()));
72 if (db.inTransaction()) throw new AssertionError();
74 db.beginTransaction();
77 db.execSQL("update account_values set keep=0;");
78 db.execSQL("update accounts set keep=0;");
81 String last_account_name = null;
83 new BufferedReader(new InputStreamReader(resp, "UTF-8"));
85 Pattern account_name_re =
86 Pattern.compile("/register\\?q=inacct%3A([a-zA-Z0-9%]+)\"");
87 Pattern account_value_re = Pattern.compile(
88 "<span class=\"[^\"]*\\bamount\\b[^\"]*\">\\s*([-+]?[\\d.,]+)(?:\\s+(\\S+))?</span>");
89 Pattern tr_end_re = Pattern.compile("</tr>");
90 Pattern descriptions_line_re =
91 Pattern.compile("\\bdescriptionsSuggester\\s*=\\s*new\\b");
92 Pattern description_items_re =
93 Pattern.compile("\"value\":\"([^\"]+)\"");
95 while ((line = buf.readLine()) != null) {
98 Matcher m = account_name_re.matcher(line);
100 String acct_encoded = m.group(1);
101 String acct_name = URLDecoder.decode(acct_encoded, "UTF-8");
102 acct_name = acct_name.replace("\"", "");
103 Log.d("account-parser", acct_name);
105 addAccount(db, acct_name);
106 publishProgress(++count);
108 last_account_name = acct_name;
113 Matcher tr_m = tr_end_re.matcher(line);
115 Log.d("account-parser", "<tr> - another account expected");
116 last_account_name = null;
120 if (last_account_name != null) {
121 m = account_value_re.matcher(line);
122 boolean match_found = false;
127 String value = m.group(1);
128 String currency = m.group(2);
129 if (currency == null) currency = "";
130 value = value.replace(',', '.');
131 Log.d("db", "curr=" + currency + ", value=" + value);
133 "insert or replace into account_values(account, currency, value, keep) values(?, ?, ?, 1);",
134 new Object[]{last_account_name, currency,
139 if (match_found) continue;
142 m = descriptions_line_re.matcher(line);
144 db.execSQL("update description_history set keep=0;");
145 m = description_items_re.matcher(line);
149 String description = m.group(1);
150 if (description.isEmpty()) continue;
152 Log.d("db", String.format("Stored description: %s",
154 db.execSQL("insert or replace into description_history" +
155 "(description, description_upper, keep) " +
157 new Object[]{description, description.toUpperCase()
163 db.execSQL("delete from account_values where keep=0;");
164 db.execSQL("delete from accounts where keep=0;");
165 db.setTransactionSuccessful();
167 catch (OperationCanceledException e) {
168 Log.w("async", "Account retrieval cancelled");
177 catch (MalformedURLException e) {
178 error = R.string.err_bad_backend_url;
181 catch (FileNotFoundException e) {
182 error = R.string.err_bad_auth;
185 catch (IOException e) {
186 error = R.string.err_net_io_error;
189 catch (Exception e) {
190 error = R.string.err_net_error;
194 Log.d("RAT", "decrementing background task count");
195 Data.backgroundTaskCount.decrementAndGet();
200 private void throwIfCancelled() {
201 if (isCancelled()) throw new OperationCanceledException(null);
204 private void addAccount(SQLiteDatabase db, String name) {
206 LedgerAccount acc = new LedgerAccount(name);
207 db.execSQL("update accounts set level = ?, keep = 1 where name = ?",
208 new Object[]{acc.getLevel(), name});
209 db.execSQL("insert into accounts(name, name_upper, parent_name, level) select ?,?," +
210 "?,? " + "where (select changes() = 0)",
211 new Object[]{name, name.toUpperCase(), acc.getParentName(), acc.getLevel()});
212 name = acc.getParentName();
213 } while (name != null);
216 protected void onPostExecute(Void result) {
217 MainActivity ctx = mContext.get();
218 if (ctx == null) return;
219 ctx.onRetrieveDone(this.error == 0);