]> git.ktnx.net Git - mobile-ledger.git/blob - app/src/main/java/net/ktnx/mobileledger/async/SendTransactionTask.java
equality check for Profile
[mobile-ledger.git] / app / src / main / java / net / ktnx / mobileledger / async / SendTransactionTask.java
1 /*
2  * Copyright © 2020 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.async;
19
20 import android.os.AsyncTask;
21 import android.util.Log;
22
23 import net.ktnx.mobileledger.json.API;
24 import net.ktnx.mobileledger.json.ApiNotSupportedException;
25 import net.ktnx.mobileledger.json.Gateway;
26 import net.ktnx.mobileledger.model.LedgerTransaction;
27 import net.ktnx.mobileledger.model.LedgerTransactionAccount;
28 import net.ktnx.mobileledger.model.MobileLedgerProfile;
29 import net.ktnx.mobileledger.utils.Globals;
30 import net.ktnx.mobileledger.utils.Logger;
31 import net.ktnx.mobileledger.utils.NetworkUtil;
32 import net.ktnx.mobileledger.utils.SimpleDate;
33 import net.ktnx.mobileledger.utils.UrlEncodedFormData;
34
35 import java.io.BufferedReader;
36 import java.io.IOException;
37 import java.io.InputStream;
38 import java.io.InputStreamReader;
39 import java.io.OutputStream;
40 import java.net.HttpURLConnection;
41 import java.nio.charset.StandardCharsets;
42 import java.util.List;
43 import java.util.Locale;
44 import java.util.Map;
45 import java.util.regex.Matcher;
46 import java.util.regex.Pattern;
47
48 import static android.os.SystemClock.sleep;
49 import static net.ktnx.mobileledger.utils.Logger.debug;
50
51 /* TODO: get rid of the custom session/cookie and auth code?
52  *       (the last problem with the POST was the missing content-length header)
53  *       This will resolve itself when hledger-web 1.14+ is released with Debian/stable,
54  *       at which point the HTML form emulation can be dropped entirely
55  */
56
57 public class SendTransactionTask extends AsyncTask<LedgerTransaction, Void, Void> {
58     private final TaskCallback taskCallback;
59     private final MobileLedgerProfile mProfile;
60     private final boolean simulate;
61     protected String error;
62     private String token;
63     private String session;
64     private LedgerTransaction transaction;
65
66     public SendTransactionTask(TaskCallback callback, MobileLedgerProfile profile,
67                                boolean simulate) {
68         taskCallback = callback;
69         mProfile = profile;
70         this.simulate = simulate;
71     }
72     public SendTransactionTask(TaskCallback callback, MobileLedgerProfile profile) {
73         taskCallback = callback;
74         mProfile = profile;
75         simulate = false;
76     }
77     private void sendOK(API apiVersion) throws IOException, ApiNotSupportedException {
78         HttpURLConnection http = NetworkUtil.prepareConnection(mProfile, "add");
79         http.setRequestMethod("PUT");
80         http.setRequestProperty("Content-Type", "application/json");
81         http.setRequestProperty("Accept", "*/*");
82
83         Gateway gateway = Gateway.forApiVersion(apiVersion);
84         String body = gateway.transactionSaveRequest(transaction);
85
86         Logger.debug("network", "Sending using API " + apiVersion);
87         sendRequest(http, body);
88     }
89     private void sendRequest(HttpURLConnection http, String body)
90             throws IOException, ApiNotSupportedException {
91         if (simulate) {
92             debug("network", "The request would be: " + body);
93             try {
94                 Thread.sleep(1500);
95                 if (Math.random() > 0.3)
96                     throw new RuntimeException("Simulated test exception");
97             }
98             catch (InterruptedException ex) {
99                 Logger.debug("network", ex.toString());
100             }
101
102             return;
103         }
104
105         byte[] bodyBytes = body.getBytes(StandardCharsets.UTF_8);
106         http.setDoOutput(true);
107         http.setDoInput(true);
108         http.addRequestProperty("Content-Length", String.valueOf(bodyBytes.length));
109
110         debug("network", "request header: " + http.getRequestProperties()
111                                                   .toString());
112
113         try (OutputStream req = http.getOutputStream()) {
114             debug("network", "Request body: " + body);
115             req.write(bodyBytes);
116
117             final int responseCode = http.getResponseCode();
118             debug("network", String.format(Locale.US, "Response: %d %s", responseCode,
119                     http.getResponseMessage()));
120
121             try (InputStream resp = http.getErrorStream()) {
122
123                 switch (responseCode) {
124                     case 200:
125                     case 201:
126                         break;
127                     case 400:
128                     case 405: {
129                         BufferedReader reader = new BufferedReader(new InputStreamReader(resp));
130                         StringBuilder errorLines = new StringBuilder();
131                         int count = 0;
132                         while (count <= 5) {
133                             String line = reader.readLine();
134                             if (line == null)
135                                 break;
136                             Logger.debug("network", line);
137
138                             if (errorLines.length() != 0)
139                                 errorLines.append("\n");
140
141                             errorLines.append(line);
142                             count++;
143                         }
144                         throw new ApiNotSupportedException(errorLines.toString());
145                     }
146                     default:
147                         BufferedReader reader = new BufferedReader(new InputStreamReader(resp));
148                         String line = reader.readLine();
149                         debug("network", "Response content: " + line);
150                         throw new IOException(
151                                 String.format("Error response code %d", responseCode));
152                 }
153             }
154         }
155     }
156     private boolean legacySendOK() throws IOException {
157         HttpURLConnection http = NetworkUtil.prepareConnection(mProfile, "add");
158         http.setRequestMethod("POST");
159         http.setRequestProperty("Content-Type", "application/x-www-form-urlencoded");
160         http.setRequestProperty("Accept", "*/*");
161         if ((session != null) && !session.isEmpty()) {
162             http.setRequestProperty("Cookie", String.format("_SESSION=%s", session));
163         }
164         http.setDoOutput(true);
165         http.setDoInput(true);
166
167         UrlEncodedFormData params = new UrlEncodedFormData();
168         params.addPair("_formid", "identify-add");
169         if (token != null)
170             params.addPair("_token", token);
171
172         SimpleDate transactionDate = transaction.getDateIfAny();
173         if (transactionDate == null)
174             transactionDate = SimpleDate.today();
175
176         params.addPair("date", Globals.formatLedgerDate(transactionDate));
177         params.addPair("description", transaction.getDescription());
178         for (LedgerTransactionAccount acc : transaction.getAccounts()) {
179             params.addPair("account", acc.getAccountName());
180             if (acc.isAmountSet())
181                 params.addPair("amount", String.format(Locale.US, "%1.2f", acc.getAmount()));
182             else
183                 params.addPair("amount", "");
184         }
185
186         String body = params.toString();
187         http.addRequestProperty("Content-Length", String.valueOf(body.length()));
188
189         debug("network", "request header: " + http.getRequestProperties()
190                                                   .toString());
191
192         try (OutputStream req = http.getOutputStream()) {
193             debug("network", "Request body: " + body);
194             req.write(body.getBytes(StandardCharsets.US_ASCII));
195
196             try (InputStream resp = http.getInputStream()) {
197                 debug("update_accounts", String.valueOf(http.getResponseCode()));
198                 if (http.getResponseCode() == 303) {
199                     // everything is fine
200                     return true;
201                 }
202                 else if (http.getResponseCode() == 200) {
203                     // get the new cookie
204                     {
205                         Pattern reSessionCookie = Pattern.compile("_SESSION=([^;]+);.*");
206
207                         Map<String, List<String>> header = http.getHeaderFields();
208                         List<String> cookieHeader = header.get("Set-Cookie");
209                         if (cookieHeader != null) {
210                             String cookie = cookieHeader.get(0);
211                             Matcher m = reSessionCookie.matcher(cookie);
212                             if (m.matches()) {
213                                 session = m.group(1);
214                                 debug("network", "new session is " + session);
215                             }
216                             else {
217                                 debug("network", "set-cookie: " + cookie);
218                                 Log.w("network",
219                                         "Response Set-Cookie headers is not a _SESSION one");
220                             }
221                         }
222                         else {
223                             Log.w("network", "Response has no Set-Cookie header");
224                         }
225                     }
226                     // the token needs to be updated
227                     BufferedReader reader = new BufferedReader(new InputStreamReader(resp));
228                     Pattern re = Pattern.compile(
229                             "<input type=\"hidden\" name=\"_token\" value=\"([^\"]+)\">");
230                     String line;
231                     while ((line = reader.readLine()) != null) {
232                         //debug("dump", line);
233                         Matcher m = re.matcher(line);
234                         if (m.matches()) {
235                             token = m.group(1);
236                             debug("save-transaction", line);
237                             debug("save-transaction", "Token=" + token);
238                             return false;       // retry
239                         }
240                     }
241                     throw new IOException("Can't find _token string");
242                 }
243                 else {
244                     throw new IOException(
245                             String.format("Error response code %d", http.getResponseCode()));
246                 }
247             }
248         }
249     }
250     @Override
251     protected Void doInBackground(LedgerTransaction... ledgerTransactions) {
252         error = null;
253         try {
254             transaction = ledgerTransactions[0];
255
256             final API profileApiVersion = mProfile.getApiVersion();
257             switch (profileApiVersion) {
258                 case auto:
259                     boolean sendOK = false;
260                     for (API ver : API.allVersions) {
261                         Logger.debug("network", "Trying version " + ver);
262                         try {
263                             sendOK(ver);
264                             sendOK = true;
265                             Logger.debug("network", "Version " + ver + " request succeeded");
266
267                             break;
268                         }
269                         catch (ApiNotSupportedException e) {
270                             Logger.debug("network", "Version " + ver + " seems not supported");
271                         }
272                     }
273
274                     if (!sendOK) {
275                         Logger.debug("network", "Trying HTML form emulation");
276                         legacySendOkWithRetry();
277                     }
278                     break;
279                 case html:
280                     legacySendOkWithRetry();
281                     break;
282                 case v1_14:
283                 case v1_15:
284                 case v1_19_1:
285                     sendOK(profileApiVersion);
286                     break;
287                 default:
288                     throw new IllegalStateException("Unexpected API version: " + profileApiVersion);
289             }
290         }
291         catch (ApiNotSupportedException | Exception e) {
292             e.printStackTrace();
293             error = e.getMessage();
294         }
295
296         return null;
297     }
298     private void legacySendOkWithRetry() throws IOException {
299         int tried = 0;
300         while (!legacySendOK()) {
301             tried++;
302             if (tried >= 2)
303                 throw new IOException(String.format("aborting after %d tries", tried));
304             sleep(100);
305         }
306     }
307     @Override
308     protected void onPostExecute(Void aVoid) {
309         super.onPostExecute(aVoid);
310         taskCallback.done(error);
311     }
312
313 }