
/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
/*
 * This file is part of the LibreOffice project.
 *
 * This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/.
 *
 * This file incorporates work covered by the following license notice:
 *
 *   Licensed to the Apache Software Foundation (ASF) under one or more
 *   contributor license agreements. See the NOTICE file distributed
 *   with this work for additional information regarding copyright
 *   ownership. The ASF licenses this file to you under the Apache
 *   License, Version 2.0 (the "License"); you may not use this file
 *   except in compliance with the License. You may obtain a copy of
 *   the License at http://www.apache.org/licenses/LICENSE-2.0 .
 */

#include <DExport.hxx>
#include <TokenWriter.hxx>
#include <com/sun/star/sdbc/XColumnLocate.hpp>
#include <com/sun/star/sdbc/XResultSetMetaDataSupplier.hpp>
#include <sal/log.hxx>
#include <osl/diagnose.h>
#include <core_resource.hxx>
#include <strings.hrc>
#include <strings.hxx>
#include <sqlmessage.hxx>
#include <com/sun/star/sdbc/XRowUpdate.hpp>

using namespace dbaui;
using namespace ::com::sun::star::uno;
using namespace ::com::sun::star::beans;
using namespace ::com::sun::star::sdbc;

// export data
ORowSetImportExport::ORowSetImportExport(weld::Window* pParent,
                                         const Reference< XResultSetUpdate >& xResultSetUpdate,
                                         const svx::ODataAccessDescriptor& aDataDescriptor,
                                         const Reference< XComponentContext >& rM)
                                            : ODatabaseImportExport(aDataDescriptor,rM,nullptr)
                                            ,m_xTargetResultSetUpdate(xResultSetUpdate)
                                            ,m_xTargetRowUpdate(xResultSetUpdate,UNO_QUERY)
                                            ,m_pParent(pParent)
                                            ,m_bAlreadyAsked(false)
{
    OSL_ENSURE(pParent,"Window can't be null!");
}

void ORowSetImportExport::initialize()
{
    ODatabaseImportExport::initialize();
    // do namemapping
    Reference<XColumnLocate> xColumnLocate(m_xResultSet,UNO_QUERY);
    OSL_ENSURE(xColumnLocate.is(),"The rowset normally should support this");

    m_xTargetResultSetMetaData = Reference<XResultSetMetaDataSupplier>(m_xTargetResultSetUpdate,UNO_QUERY_THROW)->getMetaData();
    if(!m_xTargetResultSetMetaData.is() || !xColumnLocate.is() || !m_xResultSetMetaData.is() )
        throw SQLException(DBA_RES(STR_UNEXPECTED_ERROR),*this,u"S1000"_ustr,0,Any());

    sal_Int32 nCount = m_xTargetResultSetMetaData->getColumnCount();
    m_aColumnMapping.reserve(nCount);
    m_aColumnTypes.reserve(nCount);
    for (sal_Int32 i = 1;i <= nCount; ++i)
    {
        sal_Int32 nPos = COLUMN_POSITION_NOT_FOUND; // means column is autoincrement or doesn't exist
        if(!m_xTargetResultSetMetaData->isAutoIncrement(i))
        {
            try
            {
                OUString sColumnName = m_xTargetResultSetMetaData->getColumnName(i);
                nPos = xColumnLocate->findColumn(sColumnName);
            }
            catch(const SQLException&)
            {
                if(m_xTargetResultSetMetaData->isNullable(i))
                    nPos = 0; // column doesn't exist but we could set it to null
            }
        }

        m_aColumnMapping.push_back(nPos);
        if(nPos > 0)
            m_aColumnTypes.push_back(m_xResultSetMetaData->getColumnType(nPos));
        else
            m_aColumnTypes.push_back(DataType::OTHER);
    }
}

bool ORowSetImportExport::Write()
{
    return true;
}

bool ORowSetImportExport::Read()
{
    if (!m_xResultSet)
        return false;
    // check if there is any column to copy
    if(std::none_of(m_aColumnMapping.begin(),m_aColumnMapping.end(),
                        [](sal_Int32 n) { return n > 0; }))
        return false;
    if(m_aSelection.hasElements())
    {
        for (auto& any : m_aSelection)
        {
            sal_Int32 nPos = -1;
            any >>= nPos;
            OSL_ENSURE(nPos != -1,"Invalid position!");
            if (!m_xResultSet->absolute(nPos) || !insertNewRow())
                break;
        }
    }
    else
    {
        Reference<XPropertySet> xProp(m_xResultSet,UNO_QUERY);
        sal_Int32 nRowCount = 0;
        if ( xProp.is() && xProp->getPropertySetInfo()->hasPropertyByName(PROPERTY_ISROWCOUNTFINAL) )
        {
            bool bFinal = false;
            xProp->getPropertyValue(PROPERTY_ISROWCOUNTFINAL) >>= bFinal;
            if ( !bFinal )
                m_xResultSet->afterLast();
            xProp->getPropertyValue(PROPERTY_ROWCOUNT) >>= nRowCount;
        }
        if ( !nRowCount )
        {
            m_xResultSet->afterLast();
            nRowCount = m_xResultSet->getRow();
        }
        OSL_ENSURE(nRowCount,"RowCount is 0!");
        m_xResultSet->beforeFirst();
        while(m_xResultSet.is() && m_xResultSet->next() && nRowCount )
        {
            --nRowCount;
            if (!insertNewRow())
                break;
        }
    }
    return true;
}

bool ORowSetImportExport::insertNewRow()
{
    try
    {
        m_xTargetResultSetUpdate->moveToInsertRow();
        sal_Int32 i = 1;
        for (auto const& column : m_aColumnMapping)
        {
            if(column > 0)
            {
                Any aValue;
                switch(m_aColumnTypes[i-1])
                {
                    case DataType::CHAR:
                    case DataType::VARCHAR:
                        aValue <<= m_xRow->getString(column);
                        break;
                    case DataType::DECIMAL:
                    case DataType::NUMERIC:
                        aValue <<= m_xRow->getDouble(column);
                        break;
                    case DataType::BIGINT:
                        aValue <<= m_xRow->getLong(column);
                        break;
                    case DataType::FLOAT:
                        aValue <<= m_xRow->getFloat(column);
                        break;
                    case DataType::DOUBLE:
                        aValue <<= m_xRow->getDouble(column);
                        break;
                    case DataType::LONGVARCHAR:
                        aValue <<= m_xRow->getString(column);
                        break;
                    case DataType::LONGVARBINARY:
                        aValue <<= m_xRow->getBytes(column);
                        break;
                    case DataType::DATE:
                        aValue <<= m_xRow->getDate(column);
                        break;
                    case DataType::TIME:
                        aValue <<= m_xRow->getTime(column);
                        break;
                    case DataType::TIMESTAMP:
                        aValue <<= m_xRow->getTimestamp(column);
                        break;
                    case DataType::BIT:
                    case DataType::BOOLEAN:
                        aValue <<= m_xRow->getBoolean(column);
                        break;
                    case DataType::TINYINT:
                        aValue <<= m_xRow->getByte(column);
                        break;
                    case DataType::SMALLINT:
                        aValue <<= m_xRow->getShort(column);
                        break;
                    case DataType::INTEGER:
                        aValue <<= m_xRow->getInt(column);
                        break;
                    case DataType::REAL:
                        aValue <<= m_xRow->getDouble(column);
                        break;
                    case DataType::BINARY:
                    case DataType::VARBINARY:
                        aValue <<= m_xRow->getBytes(column);
                        break;
                    case DataType::BLOB:
                        aValue <<= m_xRow->getBlob(column);
                        break;
                    case DataType::CLOB:
                        aValue <<= m_xRow->getClob(column);
                        break;
                    default:
                        SAL_WARN("dbaccess.ui", "Unknown type");
                }
                if(m_xRow->wasNull())
                    m_xTargetRowUpdate->updateNull(i);
                else
                    m_xTargetRowUpdate->updateObject(i,aValue);
            }
            else if(column == 0)//now we have know that we to set this column to null
                m_xTargetRowUpdate->updateNull(i);
            ++i;
        }
        m_xTargetResultSetUpdate->insertRow();
    }
    catch(const SQLException&)
    {
        if(!m_bAlreadyAsked)
        {
            OUString sAskIfContinue = DBA_RES(STR_ERROR_OCCURRED_WHILE_COPYING);
            OSQLWarningBox aDlg(m_pParent, sAskIfContinue, MessBoxStyle::YesNo | MessBoxStyle::DefaultYes);
            if (aDlg.run() == RET_YES)
                m_bAlreadyAsked = true;
            else
                return false;
        }
    }
    return true;
}

/* vim:set shiftwidth=4 softtabstop=4 expandtab: */
