QifParser.java

/*
 * Copyright (c) 2011 Denis Solonenko.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the GNU Public License v2.0
 * which accompanies this distribution, and is available at
 * http://www.gnu.org/licenses/old-licenses/gpl-2.0.html
 */

package ru.orangesoftware.financisto2.export.qif;

import java.io.*;
import java.util.ArrayList;
import java.util.HashSet;
import java.util.List;
import java.util.Set;

import static ru.orangesoftware.financisto2.utils.Utils.isEmpty;
import static ru.orangesoftware.financisto2.utils.Utils.isNotEmpty;

/**
 * Created by IntelliJ IDEA.
 * User: Denis Solonenko
 * Date: 10/11/11 10:32 PM
 */
public class QifParser {

    private final QifBufferedReader r;
    private final QifDateFormat dateFormat;

    public final List<QifAccount> accounts = new ArrayList<QifAccount>();
    public final Set<QifCategory> categories = new HashSet<QifCategory>();
    public final Set<QifCategory> categoriesFromTransactions = new HashSet<QifCategory>();
    public final Set<String> payees = new HashSet<String>();
    public final Set<String> classes = new HashSet<String>();

    public QifParser(QifBufferedReader r, QifDateFormat dateFormat) {
        this.r = r;
        this.dateFormat = dateFormat;
    }

    public void parse() throws IOException {
        String line;
        while ((line = r.readLine()) != null) {
            if (line.startsWith("!Account")) {
                parseAccount();
            } else if (line.startsWith("!Type:Cat")) {
                parseCategories();
            }
        }
        categories.addAll(categoriesFromTransactions);
    }

    private void parseCategories() throws IOException {
        while (true) {
            QifCategory category = new QifCategory();
            category.readFrom(r);
            categories.add(category);
            if (shouldBreakCurrentBlock()) {
                break;
            }
        }
    }

    private void parseAccount() throws IOException {
        QifAccount account = new QifAccount();
        account.readFrom(r);
        accounts.add(account);
        String peek = r.peekLine();
        if (peek != null) {
            if (peek.startsWith("!Type:")) {
                applyAccountType(account, peek);
                r.readLine();
                while (true) {
                    QifTransaction t = new QifTransaction();
                    t.readFrom(r, dateFormat);
                    addPayeeFromTransaction(t);
                    addCategoryFromTransaction(t);
                    account.transactions.add(t);
                    if (shouldBreakCurrentBlock()) {
                        break;
                    }
                }
            }
        }
    }

    private void applyAccountType(QifAccount account, String peek) {
        if (isEmpty(account.type)) {
            account.type = peek.substring(6);
        }
    }

    private void addPayeeFromTransaction(QifTransaction t) {
        if (isNotEmpty(t.payee)) {
            payees.add(t.payee);
        }
    }

    private void addCategoryFromTransaction(QifTransaction t) {
        if (isNotEmpty(t.category)) {
            QifCategory c = new QifCategory(t.category, false);
            categoriesFromTransactions.add(c);
        }
        if (isNotEmpty(t.categoryClass)) {
            classes.add(t.categoryClass);
        }
    }

    private boolean shouldBreakCurrentBlock() throws IOException {
        String peek = r.peekLine();
        return peek == null || peek.startsWith("!");
    }

}