app/src/main/java/net/ktnx/mobileledger/dao/TransactionDAO.java

   1 /*
   2  * Copyright © 2021 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.dao;
  19
  20 import android.os.AsyncTask;
  21
  22 import androidx.annotation.NonNull;
  23 import androidx.lifecycle.LiveData;
  24 import androidx.room.ColumnInfo;
  25 import androidx.room.Dao;
  26 import androidx.room.Delete;
  27 import androidx.room.Insert;
  28 import androidx.room.OnConflictStrategy;
  29 import androidx.room.Query;
  30 import androidx.room.Update;
  31
  32 import net.ktnx.mobileledger.db.Account;
  33 import net.ktnx.mobileledger.db.AccountValue;
  34 import net.ktnx.mobileledger.db.DB;
  35 import net.ktnx.mobileledger.db.Transaction;
  36 import net.ktnx.mobileledger.db.TransactionAccount;
  37 import net.ktnx.mobileledger.db.TransactionWithAccounts;
  38 import net.ktnx.mobileledger.model.LedgerAccount;
  39 import net.ktnx.mobileledger.utils.Logger;
  40 import net.ktnx.mobileledger.utils.Misc;
  41
  42 import java.util.ArrayList;
  43 import java.util.List;
  44 import java.util.Locale;
  45
  46 @Dao
  47 public abstract class TransactionDAO extends BaseDAO<Transaction> {
  48     static public List<String> unbox(List<DescriptionContainer> list) {
  49         ArrayList<String> result = new ArrayList<>(list.size());
  50         for (DescriptionContainer item : list) {
  51             result.add(item.description);
  52         }
  53
  54         return result;
  55     }
  56     @Insert(onConflict = OnConflictStrategy.REPLACE)
  57     public abstract long insertSync(Transaction item);
  58
  59     @Update
  60     public abstract void updateSync(Transaction item);
  61
  62     @Delete
  63     public abstract void deleteSync(Transaction item);
  64
  65     @Delete
  66     public abstract void deleteSync(Transaction... items);
  67
  68     @Delete
  69     public abstract void deleteSync(List<Transaction> items);
  70
  71     @Query("SELECT * FROM transactions WHERE id = :id")
  72     public abstract LiveData<Transaction> getById(long id);
  73
  74     @androidx.room.Transaction
  75     @Query("SELECT * FROM transactions WHERE id = :transactionId")
  76     public abstract LiveData<TransactionWithAccounts> getByIdWithAccounts(long transactionId);
  77
  78     @androidx.room.Transaction
  79     @Query("SELECT * FROM transactions WHERE id = :transactionId")
  80     public abstract TransactionWithAccounts getByIdWithAccountsSync(long transactionId);
  81
  82     @Query("SELECT DISTINCT description, CASE WHEN description_uc LIKE :term||'%%' THEN 1 " +
  83            "               WHEN description_uc LIKE '%%:'||:term||'%%' THEN 2 " +
  84            "               WHEN description_uc LIKE '%% '||:term||'%%' THEN 3 " +
  85            "               ELSE 9 END AS ordering FROM transactions " +
  86            "WHERE description_uc LIKE '%%'||:term||'%%' ORDER BY ordering, description_uc, rowid ")
  87     public abstract List<DescriptionContainer> lookupDescriptionSync(@NonNull String term);
  88
  89     @androidx.room.Transaction
  90     @Query("SELECT * from transactions WHERE description = :description ORDER BY year desc, month" +
  91            " desc, day desc LIMIT 1")
  92     public abstract TransactionWithAccounts getFirstByDescriptionSync(@NonNull String description);
  93
  94     @androidx.room.Transaction
  95     @Query("SELECT tr.id, tr.profile_id, tr.ledger_id, tr.description, tr.data_hash, tr.comment, " +
  96            "tr.year, tr.month, tr.day, tr.generation from transactions tr JOIN " +
  97            "transaction_accounts t_a ON t_a.transaction_id = tr.id WHERE tr.description = " +
  98            ":description AND t_a.account_name LIKE '%'||:accountTerm||'%' ORDER BY year desc, " +
  99            "month desc, day desc, tr.ledger_id desc LIMIT 1")
 100     public abstract TransactionWithAccounts getFirstByDescriptionHavingAccountSync(
 101             @NonNull String description, @NonNull String accountTerm);
 102
 103     @Query("SELECT * from transactions WHERE profile_id = :profileId")
 104     public abstract List<Transaction> getAllForProfileUnorderedSync(long profileId);
 105
 106     @Query("SELECT generation FROM transactions WHERE profile_id = :profileId LIMIT 1")
 107     protected abstract TransactionGenerationContainer getGenerationPOJOSync(long profileId);
 108
 109     @androidx.room.Transaction
 110     @Query("SELECT * FROM transactions WHERE profile_id = :profileId ORDER BY year " +
 111            " asc, month asc, day asc, ledger_id asc")
 112     public abstract LiveData<List<TransactionWithAccounts>> getAllWithAccounts(long profileId);
 113
 114     @androidx.room.Transaction
 115     @Query("SELECT distinct(tr.id), tr.ledger_id, tr.profile_id, tr.data_hash, tr.year, tr.month," +
 116            " tr.day, tr.description, tr.comment, tr.generation FROM transactions tr JOIN " +
 117            "transaction_accounts ta ON ta.transaction_id=tr.id WHERE ta.account_name LIKE " +
 118            ":accountName||'%' AND ta.amount <> 0 AND tr.profile_id = :profileId ORDER BY tr.year " +
 119            "asc, tr.month asc, tr.day asc, tr.ledger_id asc")
 120     public abstract LiveData<List<TransactionWithAccounts>> getAllWithAccountsFiltered(
 121             long profileId, String accountName);
 122
 123     @Query("DELETE FROM transactions WHERE profile_id = :profileId AND generation <> " +
 124            ":currentGeneration")
 125     public abstract int purgeOldTransactionsSync(long profileId, long currentGeneration);
 126
 127     @Query("DELETE FROM transaction_accounts WHERE EXISTS (SELECT 1 FROM transactions tr WHERE tr" +
 128            ".id=transaction_accounts.transaction_id AND tr.profile_id=:profileId) AND generation " +
 129            "<> :currentGeneration")
 130     public abstract int purgeOldTransactionAccountsSync(long profileId, long currentGeneration);
 131
 132     @Query("DELETE FROM transactions WHERE profile_id = :profileId")
 133     public abstract int deleteAllSync(long profileId);
 134
 135     @Query("SELECT * FROM transactions where profile_id = :profileId AND ledger_id = :ledgerId")
 136     public abstract Transaction getByLedgerId(long profileId, long ledgerId);
 137
 138     @Query("UPDATE transactions SET generation = :newGeneration WHERE id = :transactionId")
 139     public abstract int updateGeneration(long transactionId, long newGeneration);
 140
 141     @Query("UPDATE transaction_accounts SET generation = :newGeneration WHERE transaction_id = " +
 142            ":transactionId")
 143     public abstract int updateAccountsGeneration(long transactionId, long newGeneration);
 144
 145     @Query("SELECT max(ledger_id) as ledger_id FROM transactions WHERE profile_id = :profileId")
 146     public abstract LedgerIdContainer getMaxLedgerIdPOJOSync(long profileId);
 147     @androidx.room.Transaction
 148     public void updateGenerationWithAccounts(long transactionId, long newGeneration) {
 149         updateGeneration(transactionId, newGeneration);
 150         updateAccountsGeneration(transactionId, newGeneration);
 151     }
 152     public long getGenerationSync(long profileId) {
 153         TransactionGenerationContainer result = getGenerationPOJOSync(profileId);
 154
 155         if (result == null)
 156             return 0;
 157         return result.generation;
 158     }
 159     public long getMaxLedgerIdSync(long profileId) {
 160         LedgerIdContainer result = getMaxLedgerIdPOJOSync(profileId);
 161
 162         if (result == null)
 163             return 0;
 164         return result.ledgerId;
 165     }
 166     @androidx.room.Transaction
 167     public void storeTransactionsSync(List<TransactionWithAccounts> list, long profileId) {
 168         long generation = getGenerationSync(profileId) + 1;
 169
 170         for (TransactionWithAccounts tr : list) {
 171             tr.transaction.setGeneration(generation);
 172             tr.transaction.setProfileId(profileId);
 173
 174             storeSync(tr);
 175         }
 176
 177         Logger.debug("Transaction", "Purging old transactions");
 178         int removed = purgeOldTransactionsSync(profileId, generation);
 179         Logger.debug("Transaction", String.format(Locale.ROOT, "Purged %d transactions", removed));
 180
 181         removed = purgeOldTransactionAccountsSync(profileId, generation);
 182         Logger.debug("Transaction",
 183                 String.format(Locale.ROOT, "Purged %d transaction accounts", removed));
 184     }
 185     @androidx.room.Transaction
 186     void storeSync(TransactionWithAccounts rec) {
 187         TransactionAccountDAO trAccDao = DB.get()
 188                                            .getTransactionAccountDAO();
 189
 190         Transaction transaction = rec.transaction;
 191         Transaction existing = getByLedgerId(transaction.getProfileId(), transaction.getLedgerId());
 192         if (existing != null) {
 193             if (Misc.equalStrings(transaction.getDataHash(), existing.getDataHash())) {
 194                 updateGenerationWithAccounts(existing.getId(), rec.transaction.getGeneration());
 195                 return;
 196             }
 197
 198             existing.copyDataFrom(transaction);
 199             updateSync(existing);
 200
 201             transaction = existing;
 202         }
 203         else
 204             transaction.setId(insertSync(transaction));
 205
 206         for (TransactionAccount trAcc : rec.accounts) {
 207             trAcc.setTransactionId(transaction.getId());
 208             trAcc.setGeneration(transaction.getGeneration());
 209             TransactionAccount existingAcc =
 210                     trAccDao.getByOrderNoSync(trAcc.getTransactionId(), trAcc.getOrderNo());
 211             if (existingAcc != null) {
 212                 existingAcc.copyDataFrom(trAcc);
 213                 trAccDao.updateSync(existingAcc);
 214             }
 215             else
 216                 trAcc.setId(trAccDao.insertSync(trAcc));
 217         }
 218     }
 219     public void storeLast(TransactionWithAccounts rec) {
 220         AsyncTask.execute(() -> appendSync(rec));
 221     }
 222     @androidx.room.Transaction
 223     public void appendSync(TransactionWithAccounts rec) {
 224         TransactionAccountDAO trAccDao = DB.get()
 225                                            .getTransactionAccountDAO();
 226         AccountDAO accDao = DB.get()
 227                               .getAccountDAO();
 228         AccountValueDAO accValDao = DB.get()
 229                                       .getAccountValueDAO();
 230
 231         Transaction transaction = rec.transaction;
 232         final long profileId = transaction.getProfileId();
 233         transaction.setGeneration(getGenerationSync(profileId));
 234         transaction.setLedgerId(getMaxLedgerIdSync(profileId) + 1);
 235         transaction.setId(insertSync(transaction));
 236
 237         for (TransactionAccount trAcc : rec.accounts) {
 238             trAcc.setTransactionId(transaction.getId());
 239             trAcc.setGeneration(transaction.getGeneration());
 240             trAcc.setId(trAccDao.insertSync(trAcc));
 241
 242             String accName = trAcc.getAccountName();
 243             while (accName != null) {
 244                 Account acc = accDao.getByNameSync(profileId, accName);
 245                 if (acc == null) {
 246                     acc = new Account();
 247                     acc.setProfileId(profileId);
 248                     acc.setName(accName);
 249                     acc.setNameUpper(accName.toUpperCase());
 250                     acc.setParentName(LedgerAccount.extractParentName(accName));
 251                     acc.setLevel(LedgerAccount.determineLevel(acc.getName()));
 252                     acc.setGeneration(trAcc.getGeneration());
 253
 254                     acc.setId(accDao.insertSync(acc));
 255                 }
 256
 257                 AccountValue accVal = accValDao.getByCurrencySync(acc.getId(), trAcc.getCurrency());
 258                 if (accVal == null) {
 259                     accVal = new AccountValue();
 260                     accVal.setAccountId(acc.getId());
 261                     accVal.setGeneration(trAcc.getGeneration());
 262                     accVal.setCurrency(trAcc.getCurrency());
 263                     accVal.setValue(trAcc.getAmount());
 264                     accVal.setId(accValDao.insertSync(accVal));
 265                 }
 266                 else {
 267                     accVal.setValue(accVal.getValue() + trAcc.getAmount());
 268                     accValDao.updateSync(accVal);
 269                 }
 270
 271                 accName = LedgerAccount.extractParentName(accName);
 272             }
 273         }
 274     }
 275     static class TransactionGenerationContainer {
 276         @ColumnInfo
 277         long generation;
 278         public TransactionGenerationContainer(long generation) {
 279             this.generation = generation;
 280         }
 281     }
 282
 283     static class LedgerIdContainer {
 284         @ColumnInfo(name = "ledger_id")
 285         long ledgerId;
 286         public LedgerIdContainer(long ledgerId) {
 287             this.ledgerId = ledgerId;
 288         }
 289     }
 290
 291     static public class DescriptionContainer {
 292         @ColumnInfo
 293         public String description;
 294         @ColumnInfo
 295         public int ordering;
 296     }
 297 }