dialog-chart-export.c

/***************************************************************************
 *            dialog-chart-export.c
 *
 *  Sun Feb 27 14:19:12 2005
 *  Copyright  2005  Neil Williams
 *  linux@codehelp.co.uk
 *  Copyright (c) 2006 David Hampton <hampton@employees.org>
 ****************************************************************************/

/*
 *  This program is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation; either version 2 of the License, or
 *  (at your option) any later version.
 *
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU Library General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this program; if not, write to the Free Software
 *  Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
 *  02110-1301, USA.
 */
 
#include "config.h"

#include <gtk/gtk.h>
#include <glib/gi18n.h>
#include <time.h>

#include "qof.h"
#include "AccountP.h"
#include "Transaction.h"
#include "dialog-chart-export.h"
#include "gnc-ui-util.h"
#include "dialog-utils.h"
#include "gnc-engine.h"
#include "gnc-file.h"
#include "gnc-ui.h"
#include "gnc-session.h"

#define EQUITY_ACCOUNT_NAME  _("Opening Balances")
#define OPENING_BALANCE_DESC _("Opening Balance")

void chart_export_response_cb (GtkDialog *dialog, gint response, gpointer user_data);

typedef struct chart_data_s
{
        GladeXML *xml;
        GtkWidget *dialog;
        GtkWidget *calendar;
        time_t chart_time_t;
        QofSession *chart_session;
        Account *equity_account;
        GList      *param_ref_list;
}chart_data;

static void
chart_collection_cb(QofInstance *ent, gpointer user_data)
{
        chart_data *data;
        Account *acc;
        gboolean success;
        const GUID *guid;
        QofCollection *copy_coll;
        QofBook *book;

        g_return_if_fail(user_data != NULL);
        data = (chart_data*)user_data;
        acc = (Account*)ent;
        if(0 == safe_strcmp(EQUITY_ACCOUNT_NAME, xaccAccountGetName(acc)) 
                && (xaccAccountGetType(acc) == ACCT_TYPE_EQUITY))
        {
                success = qof_instance_copy_to_session(data->chart_session, ent);
                if(!success) { return; }
                guid = qof_entity_get_guid(ent);
                book = qof_session_get_book(data->chart_session);
                copy_coll = qof_book_get_collection(book, GNC_ID_ACCOUNT);
                data->equity_account = (Account*)qof_collection_lookup_entity(copy_coll, guid);
                return;
        }
}

static void
chart_reference_cb(QofInstance *ent, gpointer user_data)
{
        QofInstanceReference *reference;
        QofParam     *ref_param;
        chart_data   *data;

        g_return_if_fail(user_data != NULL);
        data = (chart_data*)user_data;
        while(data->param_ref_list != NULL) {
                ref_param = data->param_ref_list->data;
                reference = qof_instance_get_reference_from(ent, ref_param);
                qof_session_update_reference_list(data->chart_session, reference);
                data->param_ref_list = data->param_ref_list->next;
        }
}

static void
chart_entity_cb(QofInstance *ent, gpointer user_data)
{
        chart_data *data;
        Account *acc_ent, *equity_account;
        Transaction *trans;
        Split *split;
        gnc_numeric balance;
        QofBook *book;
        QofCollection *coll;
        const GUID *guid;
        time_t trans_time;
        GList *ref;
        QofInstanceReference *ent_ref;
        
        g_return_if_fail(user_data != NULL);
        data = (chart_data*)user_data;
        trans_time = data->chart_time_t;
        data->param_ref_list = NULL;
        guid = qof_entity_get_guid(ent);
        acc_ent = (Account*)ent;
        ref = NULL;
        equity_account = data->equity_account;
        g_return_if_fail(equity_account != NULL);
        balance = xaccAccountGetBalanceAsOfDate(acc_ent, data->chart_time_t);
        qof_instance_copy_to_session(data->chart_session, ent);
        book = qof_session_get_book(data->chart_session);
        coll = qof_book_get_collection(book, GNC_ID_ACCOUNT);
        acc_ent = (Account*)qof_collection_lookup_entity(coll, guid);
        if(xaccAccountGetCommodity(acc_ent) == NULL)
        {
                xaccAccountSetCommodity(acc_ent, gnc_default_currency());
        }
        /* can't use gnc_account_create_opening_balance directly - the partial 
        QofBook doesn't have an AccountGroup that is used to locate the Equity Account. */
        xaccAccountBeginEdit (acc_ent);
        xaccAccountBeginEdit (equity_account);
        trans = xaccMallocTransaction (book);
        xaccTransBeginEdit (trans);
        xaccTransSetCurrency (trans, xaccAccountGetCommodity (acc_ent));
        xaccTransSetDateSecs (trans, trans_time);
        xaccTransSetDateEnteredSecs (trans, trans_time);
        xaccTransSetDescription (trans, OPENING_BALANCE_DESC);
        /* User account split */
        split = xaccMallocSplit (book);
        xaccTransAppendSplit (trans, split);
        xaccAccountInsertSplit (acc_ent, split);
        xaccSplitSetAmount (split, balance);
        xaccSplitSetValue (split, balance);
        ref = qof_class_get_referenceList(GNC_ID_SPLIT);
        while(ref != NULL) {
                ent_ref = qof_instance_get_reference_from(QOF_INSTANCE(split), ref->data);
                qof_session_update_reference_list(data->chart_session, ent_ref);
                ref = g_list_next(ref);
        }
        g_list_free(ref);
        balance = gnc_numeric_neg (balance);
        /* Equity account split */
        split = xaccMallocSplit (book);
        xaccTransAppendSplit (trans, split);
        xaccAccountInsertSplit (equity_account, split);
        xaccSplitSetAmount (split, balance);
        xaccSplitSetValue (split, balance);
        xaccTransCommitEdit (trans);
        xaccAccountCommitEdit (equity_account);
        xaccAccountCommitEdit (acc_ent);
        ref = qof_class_get_referenceList(GNC_ID_TRANS);
        while(ref != NULL) {
                ent_ref = qof_instance_get_reference_from(QOF_INSTANCE(trans), ref->data);
                qof_session_update_reference_list(data->chart_session, ent_ref);
                ref = g_list_next(ref);
        }
        g_list_free(ref);
        ref = qof_class_get_referenceList(GNC_ID_SPLIT);
        while(ref != NULL) {
                ent_ref = qof_instance_get_reference_from(QOF_INSTANCE(split), ref->data);
                qof_session_update_reference_list(data->chart_session, ent_ref);
                ref = g_list_next(ref);
        }
        g_list_free(ref);
}

void
gnc_main_window_chart_export(void)
{
        GladeXML *xml;
        chart_data *data;

        xml = gnc_glade_xml_new ("chart-export.glade", "chart-export");
        data = g_new0(chart_data, 1);
        data->xml = xml;
        data->dialog = glade_xml_get_widget(xml, "chart-export");
        data->calendar = glade_xml_get_widget(xml, "chart-calendar");
        glade_xml_signal_autoconnect_full(xml,
                                          gnc_glade_autoconnect_full_func,
                                          data);
        gtk_widget_show(data->dialog);
}

static void
on_dateok_clicked (chart_data  *data)
{
        guint year, month, day;
        struct tm *chart_tm;
        gchar *filename;
        QofSession *current_session, *chart_session;
        QofBook *book;
        QofCollection *coll;

        data->chart_time_t = time(NULL);
        chart_tm = gmtime(&data->chart_time_t);
        /* set today - calendar will omit any zero/NULL values */
        year = chart_tm->tm_year + 1900;
        month = chart_tm->tm_mon + 1;
        day = chart_tm->tm_mday;
        gtk_calendar_get_date(GTK_CALENDAR(data->calendar),
                              &year, &month, &day);
        if((year + 1900) != chart_tm->tm_year) { 
        chart_tm->tm_year = year - 1900;
        }
        if(month != chart_tm->tm_mon) { 
        chart_tm->tm_mon = month;
        }
        if(day != chart_tm->tm_yday) { 
                chart_tm->tm_mday = day; 
        }
        data->chart_time_t = mktime(chart_tm);
        current_session = gnc_get_current_session();
        book = qof_session_get_book(current_session);
        chart_session = qof_session_new();
        filename = gnc_file_dialog(_("Export Chart of Accounts to QSF XML"),
                                   NULL, NULL, GNC_FILE_DIALOG_EXPORT);
        if (filename)
        {
                gnc_set_busy_cursor(NULL, TRUE);
                qof_event_suspend();
                qof_session_begin(chart_session, filename, TRUE, TRUE);
                data->chart_session = chart_session;
                data->equity_account = NULL;
                coll = qof_book_get_collection(book, GNC_ID_ACCOUNT);
                qof_collection_foreach(coll, chart_collection_cb, data);
                if(data->equity_account == NULL)
                  {
                        data->equity_account = xaccMallocAccount (qof_session_get_book(chart_session));
                        xaccAccountBeginEdit (data->equity_account);
                        xaccAccountSetName (data->equity_account, EQUITY_ACCOUNT_NAME);
                        xaccAccountSetDescription(data->equity_account, EQUITY_ACCOUNT_NAME);
                        xaccAccountSetType (data->equity_account, ACCT_TYPE_EQUITY);
                        xaccAccountSetCommodity (data->equity_account, gnc_default_currency());
                  }
                qof_object_foreach(GNC_ID_ACCOUNT, book, chart_entity_cb, data);
                data->param_ref_list = qof_class_get_referenceList(GNC_ID_TRANS);
                qof_object_foreach(GNC_ID_TRANS, book, chart_reference_cb, data);
                g_list_free(data->param_ref_list);
                data->param_ref_list = qof_class_get_referenceList(GNC_ID_SPLIT);
                qof_object_foreach(GNC_ID_SPLIT, book, chart_reference_cb, data);
                g_list_free(data->param_ref_list);
                qof_session_save(chart_session, NULL);
                show_session_error(qof_session_get_error(chart_session),
                                   filename, GNC_FILE_DIALOG_EXPORT);
                qof_event_resume();
                gnc_unset_busy_cursor(NULL);
        }
        qof_session_end(chart_session);
        gnc_set_current_session(current_session);
}

void
chart_export_response_cb (GtkDialog *dialog, gint response, gpointer user_data)
{
        chart_data  *data;
        data = (chart_data*)user_data;
        switch (response) {
          case GTK_RESPONSE_OK:
            gtk_widget_hide(data->dialog);
            on_dateok_clicked(data);
            break;

          default:
            /* do nothing */
            break;
        }

        gtk_widget_destroy(data->dialog);
        g_object_unref(data->xml);
        g_free(data);
}

---