Payment.cpp

/*
    This file is part of the Zero Reserve Plugin for Retroshare.

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

    Zero Reserve 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 Lesser General Public License for more details.

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

#include "Payment.h"
#include "MyOrders.h"
#include "zrtypes.h"
#include "zrdb.h"

#include "ZeroReservePlugin.h"

#include <QListWidget>
#include <QDateTime>

#include <sstream>

#include <stdlib.h>


QListWidget * Payment::txLogView = NULL;

Payment::Requests Payment::requestList;
Payment::Requests Payment::myRequests;


Payment::Payment( const std::string & counterparty, const ZR::ZR_Number & amount, const std::string & currency, Category category) :
    m_credit( counterparty, currency ),
    m_amount( amount ),
    m_category( category )
{
    m_credit.loadPeer();
}

void Payment::setCounterparty( const std::string & counterparty )
{
    m_credit.m_id = counterparty;
    m_credit.loadPeer();
}


const Payment::Request Payment::getRequest( const ZR::VirtualAddress & addr )
{
    Requests::iterator it = requestList.find( addr );
    if( it == requestList.end() )
        return Request( 0, Currency::INVALID );
    return (*it).second;
}

const Payment::Request Payment::getMyRequest( const ZR::VirtualAddress & addr )
{
    Requests::iterator it = myRequests.find( addr );
    if( it == myRequests.end() )
        return Request( 0, Currency::INVALID );
    return (*it).second;
}

/////// PaymentReceiver

PaymentReceiver::PaymentReceiver( const std::string & counterparty, const ZR::ZR_Number & amount, const std::string & currency, Category category) :
    Payment( counterparty, amount, currency, category)
{}


ZR::ZR_Number PaymentReceiver::newBalance() const
{
    return m_credit.m_balance + m_amount;
}


int PaymentReceiver::init()
{
    switch( m_category )
    {
    case PAYMENT:
        if( m_credit.getPeerAvailable() < m_amount ) return ZR::ZR_FAILURE;
        return ZR::ZR_SUCCESS;
    default:
        return ZR::ZR_FAILURE;
    }
}

int PaymentReceiver::commit()
{
    try{
        m_credit.loadPeer();
        m_credit.m_balance = newBalance();

        // TODO: make atomic !!!!
        ZrDB::Instance()->updatePeerCredit( m_credit, "balance", m_credit.m_balance );
        ZrDB::Instance()->appendTx( m_credit.m_id, m_credit.m_currency, m_amount );

        if( txLogView ){
            txLogView->insertItem( 0, QDateTime::currentDateTime().toString() + " : " + m_credit.m_currency.c_str() + " : +" + m_amount.toDecimalQString() );
        }
    }
    catch( std::exception e ){
        g_ZeroReservePlugin->placeMsg( std::string(  __func__ ) + ": Exception caught: " + e.what() );
        return ZR::ZR_FAILURE;
    }


    switch( m_category )
    {
    case PAYMENT:
        return ZR::ZR_SUCCESS;
    default:
        return ZR::ZR_FAILURE;
    }
}


/////// PaymentSpender

PaymentSpender::PaymentSpender(const std::string & counterparty, ZR::ZR_Number amount, const std::string & currency, Category category) :
    Payment( counterparty, amount, currency, category)
{}

ZR::ZR_Number PaymentSpender::newBalance() const
{
    return m_credit.m_balance - m_amount;
}

int PaymentSpender::init( )
{
    if( m_credit.getMyAvailable() < m_amount ){
        return ZR::ZR_FAILURE;
    }
    return ZR::ZR_SUCCESS;
}

int PaymentSpender::commit()
{
    try{
        m_credit.loadPeer();
        m_credit.m_balance = newBalance();
        // TODO: make atomic !!!!
        ZrDB::Instance()->updatePeerCredit( m_credit, "balance", m_credit.m_balance );
        ZrDB::Instance()->appendTx( m_credit.m_id, m_credit.m_currency,  -m_amount );

        if( txLogView ){
            txLogView->insertItem( 0, QDateTime::currentDateTime().toString() + " : " + m_credit.m_currency.c_str() + " : -" + m_amount.toDecimalQString() );
        }
    }
    catch( std::exception e ){
        g_ZeroReservePlugin->placeMsg( std::string(  __func__ ) + ": Exception caught: " + e.what() );
        return ZR::ZR_FAILURE;
    }

    switch( m_category )
    {
    case PAYMENT:
        return ZR::ZR_SUCCESS;
    default:
        return ZR::ZR_FAILURE;
    }
    return ZR::ZR_SUCCESS;
}