PaperInvoiceNotificationTask.java

/*
 jBilling - The Enterprise Open Source Billing System
 Copyright (C) 2003-2011 Enterprise jBilling Software Ltd. and Emiliano Conde

 This file is part of jbilling.

 jbilling is free software: you can redistribute it and/or modify
 it under the terms of the GNU Affero General Public License as published by
 the Free Software Foundation, either version 3 of the License, or
 (at your option) any later version.

 jbilling 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 Affero General Public License for more details.

 You should have received a copy of the GNU Affero General Public License
 along with jbilling.  If not, see <http://www.gnu.org/licenses/>.
 */

package com.sapienter.jbilling.server.pluggableTask;

import java.util.ArrayList;
import java.util.List;

import org.apache.log4j.Logger;

import com.sapienter.jbilling.common.SessionInternalError;
import com.sapienter.jbilling.server.invoice.PaperInvoiceBatchBL;
import com.sapienter.jbilling.server.invoice.db.InvoiceDAS;
import com.sapienter.jbilling.server.invoice.db.InvoiceDTO;
import com.sapienter.jbilling.server.notification.MessageDTO;
import com.sapienter.jbilling.server.notification.NotificationBL;
import com.sapienter.jbilling.server.pluggableTask.admin.ParameterDescription;
import com.sapienter.jbilling.server.process.db.PaperInvoiceBatchDTO;
import com.sapienter.jbilling.server.user.ContactBL;
import com.sapienter.jbilling.server.user.ContactDTOEx;
import com.sapienter.jbilling.server.user.db.UserDTO;

/**
 * @author Emil
 */
public class PaperInvoiceNotificationTask
        extends PluggableTask implements NotificationTask {

    private static final Logger LOG = Logger.getLogger(PaperInvoiceNotificationTask.class);
    // pluggable task parameters names
    public static final ParameterDescription PARAMETER_DESIGN = 
    	new ParameterDescription("design", true, ParameterDescription.Type.STR);
    public static final ParameterDescription PARAMETER_LANGUAGE_OPTIONAL = 
    	new ParameterDescription("language", false, ParameterDescription.Type.STR);
    public static final ParameterDescription PARAMETER_SQL_QUERY_OPTIONAL = 
    	new ParameterDescription("sql_query", false, ParameterDescription.Type.STR);


    //initializer for pluggable params
    { 
    	descriptions.add(PARAMETER_DESIGN);
        descriptions.add(PARAMETER_LANGUAGE_OPTIONAL);
        descriptions.add(PARAMETER_SQL_QUERY_OPTIONAL);
    }



    private String design;
    private boolean language;
    private boolean sqlQuery;
    private ContactBL contact;
    private ContactDTOEx to;
    private Integer entityId;
    private InvoiceDTO invoice;
    private ContactDTOEx from;

    /* (non-Javadoc)
     * @see com.sapienter.jbilling.server.pluggableTask.NotificationTask#deliver(com.sapienter.betty.interfaces.UserEntityLocal, com.sapienter.betty.server.notification.MessageDTO)
     */
    private void init(UserDTO user, MessageDTO message)
            throws TaskException {
        design = (String) parameters.get(PARAMETER_DESIGN.getName());

        language = Boolean.valueOf((String) parameters.get(
                PARAMETER_LANGUAGE_OPTIONAL.getName()));

        sqlQuery = Boolean.valueOf((String) parameters.get(
                PARAMETER_SQL_QUERY_OPTIONAL.getName()));

        invoice = (InvoiceDTO) message.getParameters().get(
                "invoiceDto");
        try {
            contact = new ContactBL();
            contact.setInvoice(invoice.getId());
            to = contact.getDTO();
            entityId = user.getEntity().getId();
            contact.setEntity(entityId);
            from = contact.getDTO();
        } catch (Exception e) {
            throw new TaskException(e);
        }
    }

    public void deliver(UserDTO user, MessageDTO message)
            throws TaskException {
        if (!message.getTypeId().equals(MessageDTO.TYPE_INVOICE_PAPER)) {
            // this task is only to notify about invoices
            return;
        }
        try {
            init(user, message);
            NotificationBL.generatePaperInvoiceAsFile(getDesign(user), sqlQuery,
                    invoice, from, to, message.getContent()[0].getContent(),
                    message.getContent()[1].getContent(), entityId,
                    user.getUserName(), user.getPassword());
            // update the batch record
            Integer processId = (Integer) message.getParameters().get(
                    "processId");
            PaperInvoiceBatchBL batchBL = new PaperInvoiceBatchBL();
            PaperInvoiceBatchDTO record = batchBL.createGet(processId);
            record.setTotalInvoices(record.getTotalInvoices() + 1);
            // link the batch to this invoice
            // lock the row, the payment MDB will update too
            InvoiceDTO myInvoice = new InvoiceDAS().findForUpdate(invoice.getId());
            myInvoice.setPaperInvoiceBatch(record);
            record.getInvoices().add(myInvoice);
        } catch (Exception e) {
            throw new TaskException(e);
        }
    }

    public byte[] getPDF(UserDTO user, MessageDTO message)
            throws SessionInternalError {
        try {
            init(user, message);
            LOG.debug("now message1 = " + message.getContent()[0].getContent());
            return NotificationBL.generatePaperInvoiceAsStream(getDesign(user),
                    sqlQuery, invoice, from, to, 
                    message.getContent()[0].getContent(),
                    message.getContent()[1].getContent(), entityId,
                    user.getUserName(), user.getPassword());
        } catch (Exception e) {
            throw new SessionInternalError(e);
        }
    }

    public String getPDFFile(UserDTO user, MessageDTO message)
            throws SessionInternalError {
        try {
            init(user, message);
            return NotificationBL.generatePaperInvoiceAsFile(getDesign(user),
                    sqlQuery, invoice, from, to, 
                    message.getContent()[0].getContent(),
                    message.getContent()[1].getContent(), entityId,
                    user.getUserName(), user.getPassword());
        } catch (Exception e) {
            throw new SessionInternalError(e);
        }
    }

    public int getSections() {
        return 2;
    }

    private String getDesign(UserDTO user) {
        if (language) {
            return design + user.getLanguage().getCode();
        } else {
            return design;
        }
    }
}