/******************************************************************************
 *
 * Project:  Feather Translator
 * Purpose:  Implements OGRFeatherDriver.
 * Author:   Even Rouault, <even.rouault at spatialys.com>
 *
 ******************************************************************************
 * Copyright (c) 2022, Planet Labs
 *
 * Permission is hereby granted, free of charge, to any person obtaining a
 * copy of this software and associated documentation files (the "Software"),
 * to deal in the Software without restriction, including without limitation
 * the rights to use, copy, modify, merge, publish, distribute, sublicense,
 * and/or sell copies of the Software, and to permit persons to whom the
 * Software is furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included
 * in all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
 * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
 * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
 * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
 * DEALINGS IN THE SOFTWARE.
 ****************************************************************************/

#include "cpl_json.h"
#include "cpl_time.h"
#include "gdal_pam.h"
#include "ogrsf_frmts.h"
#include "ogr_p.h"

#include <cinttypes>
#include <limits>
#include <map>
#include <set>
#include <utility>

#include "ogr_feather.h"

#include "../arrow_common/ograrrowlayer.hpp"
#include "../arrow_common/ograrrowdataset.hpp"

/************************************************************************/
/*                        OGRFeatherLayer()                             */
/************************************************************************/

OGRFeatherLayer::OGRFeatherLayer(OGRFeatherDataset* poDS,
                                 const char* pszLayerName,
                                 std::shared_ptr<arrow::ipc::RecordBatchFileReader>& poRecordBatchFileReader):
    OGRArrowLayer(poDS, pszLayerName),
    m_poDS(poDS),
    m_poRecordBatchFileReader(poRecordBatchFileReader)
{
    EstablishFeatureDefn();
    CPLAssert( static_cast<int>(m_aeGeomEncoding.size()) == m_poFeatureDefn->GetGeomFieldCount() );
}

/************************************************************************/
/*                        OGRFeatherLayer()                             */
/************************************************************************/

OGRFeatherLayer::OGRFeatherLayer(OGRFeatherDataset* poDS,
                                 const char* pszLayerName,
                                 std::shared_ptr<arrow::io::RandomAccessFile> poFile,
                                 bool bSeekable,
                                 const arrow::ipc::IpcReadOptions& oOptions,
                                 std::shared_ptr<arrow::ipc::RecordBatchStreamReader>& poRecordBatchStreamReader):
    OGRArrowLayer(poDS, pszLayerName),
    m_poDS(poDS),
    m_poFile(poFile),
    m_bSeekable(bSeekable),
    m_oOptions(oOptions),
    m_poRecordBatchReader(poRecordBatchStreamReader)
{
    EstablishFeatureDefn();
    CPLAssert( static_cast<int>(m_aeGeomEncoding.size()) == m_poFeatureDefn->GetGeomFieldCount() );
}

/************************************************************************/
/*                          LoadGeoMetadata()                           */
/************************************************************************/

void OGRFeatherLayer::LoadGeoMetadata(const arrow::KeyValueMetadata* kv_metadata,
                                      const std::string& key)
{
    if( kv_metadata && kv_metadata->Contains(key) )
    {
        auto geo = kv_metadata->Get(key);
        if( geo.ok() )
        {
            CPLJSONDocument oDoc;
            if( oDoc.LoadMemory(*geo) )
            {
                auto oRoot = oDoc.GetRoot();
                const auto osVersion = oRoot.GetString("schema_version");
                if( key != GDAL_GEO_FOOTER_KEY && osVersion != "0.1.0" )
                {
                    CPLDebug("FEATHER",
                             "schema_version = %s not explicitly handled by the driver",
                             osVersion.c_str());
                }
                auto oColumns = oRoot.GetObj("columns");
                if( oColumns.IsValid() )
                {
                    for( const auto oColumn: oColumns.GetChildren() )
                    {
                        m_oMapGeometryColumns[oColumn.GetName()] = oColumn;
                    }
                }
            }
            else
            {
                CPLError(CE_Warning, CPLE_AppDefined,
                         "Cannot parse 'geo' metadata");
            }
        }
    }
}

/************************************************************************/
/*                        EstablishFeatureDefn()                        */
/************************************************************************/

void OGRFeatherLayer::EstablishFeatureDefn()
{
    m_poSchema = m_poRecordBatchFileReader ?
        m_poRecordBatchFileReader->schema() : m_poRecordBatchReader->schema();
    const auto& kv_metadata = m_poSchema->metadata();

#ifdef DEBUG
    if( kv_metadata )
    {
        for(const auto& keyValue: kv_metadata->sorted_pairs() )
        {
            CPLDebug("FEATHER", "%s = %s",
                     keyValue.first.c_str(),
                     keyValue.second.c_str());
        }
    }
#endif

    auto poFooterMetadata = m_poRecordBatchFileReader ?
        m_poRecordBatchFileReader->metadata() : nullptr;
    if( poFooterMetadata && poFooterMetadata->Contains(GDAL_GEO_FOOTER_KEY) &&
        CPLTestBool(CPLGetConfigOption("OGR_ARROW_READ_GDAL_FOOTER", "YES")) )
    {
        LoadGeoMetadata(poFooterMetadata.get(), GDAL_GEO_FOOTER_KEY);
    }
    else
    {
        LoadGeoMetadata(kv_metadata.get(), "geo");
    }
    const auto oMapFieldNameToGDALSchemaFieldDefn = LoadGDALMetadata(kv_metadata.get());

    const auto fields = m_poSchema->fields();
    for( int i = 0; i < m_poSchema->num_fields(); ++i )
    {
        const auto& field = fields[i];
        const auto& fieldName = field->name();

        const auto& field_kv_metadata = field->metadata();
        std::string osExtensionName;
        if( field_kv_metadata )
        {
            auto extension_name = field_kv_metadata->Get("ARROW:extension:name");
            if( extension_name.ok() )
            {
                osExtensionName = *extension_name;
            }
#ifdef DEBUG
            CPLDebug("FEATHER", "Metadata field %s:", fieldName.c_str());
            for(const auto& keyValue: field_kv_metadata->sorted_pairs() )
            {
                CPLDebug("FEATHER", "  %s = %s",
                         keyValue.first.c_str(),
                         keyValue.second.c_str());
            }
#endif
        }

        if( !m_osFIDColumn.empty() &&
            fieldName == m_osFIDColumn )
        {
            m_iFIDArrowColumn = i;
            continue;
        }

        bool bRegularField = true;
        auto oIter = m_oMapGeometryColumns.find(fieldName);
        if( oIter != m_oMapGeometryColumns.end() ||
            !osExtensionName.empty() )
        {
            CPLJSONObject oJSONDef;
            if( oIter != m_oMapGeometryColumns.end() )
                oJSONDef = oIter->second;
            auto osEncoding = oJSONDef.GetString("encoding");
            if( osEncoding.empty() && !osExtensionName.empty() )
                osEncoding = osExtensionName;

            OGRwkbGeometryType eGeomType = wkbUnknown;
            auto eGeomEncoding = OGRArrowGeomEncoding::WKB;
            if( IsValidGeometryEncoding(field, osEncoding, eGeomType, eGeomEncoding) )
            {
                bRegularField = false;
                OGRGeomFieldDefn oField(fieldName.c_str(), wkbUnknown);

                const auto osWKT = oJSONDef.GetString("crs");
                if( osWKT.empty() )
                {
#if 0
                    CPLError(CE_Warning, CPLE_AppDefined,
                             "Missing required 'crs' field for geometry column %s",
                             fieldName.c_str());
#endif
                }
                else
                {
                    OGRSpatialReference* poSRS = new OGRSpatialReference();
                    poSRS->SetAxisMappingStrategy(OAMS_TRADITIONAL_GIS_ORDER);

                    if( poSRS->importFromWkt(osWKT.c_str()) == OGRERR_NONE )
                    {
                        const double dfCoordEpoch = oJSONDef.GetDouble("epoch");
                        if( dfCoordEpoch > 0 )
                            poSRS->SetCoordinateEpoch(dfCoordEpoch);

                        oField.SetSpatialRef(poSRS);
                    }
                    poSRS->Release();
                }

                // m_aeGeomEncoding be filled before calling ComputeGeometryColumnType()
                m_aeGeomEncoding.push_back(eGeomEncoding);
                if( eGeomType == wkbUnknown )
                {
                    auto osType = oJSONDef.GetString("geometry_type");
                    if( osType.empty() )
                        osType = oJSONDef.GetString("gdal:geometry_type");
                    if( m_bSeekable &&
                        osType.empty() && CPLTestBool(CPLGetConfigOption(
                                "OGR_ARROW_COMPUTE_GEOMETRY_TYPE", "YES")) )
                    {
                        eGeomType = ComputeGeometryColumnType(
                            m_poFeatureDefn->GetGeomFieldCount(), i);
                        if( m_poRecordBatchReader )
                            ResetRecordBatchReader();
                    }
                    else
                        eGeomType = GetGeometryTypeFromString(osType);
                }

                oField.SetType(eGeomType);
                oField.SetNullable(field->nullable());
                m_poFeatureDefn->AddGeomFieldDefn(&oField);
                m_anMapGeomFieldIndexToArrowColumn.push_back(i);
            }
        }

        if( bRegularField )
        {
            CreateFieldFromSchema(field, {i},
                                  oMapFieldNameToGDALSchemaFieldDefn);
        }
    }

    CPLAssert( static_cast<int>(m_anMapFieldIndexToArrowColumn.size()) == m_poFeatureDefn->GetFieldCount() );
    CPLAssert( static_cast<int>(m_anMapGeomFieldIndexToArrowColumn.size()) == m_poFeatureDefn->GetGeomFieldCount() );
}

/************************************************************************/
/*                       ResetRecordBatchReader()                       */
/************************************************************************/

bool OGRFeatherLayer::ResetRecordBatchReader()
{
    const auto nPos = *(m_poFile->Tell());
    CPL_IGNORE_RET_VAL(m_poFile->Seek(0));
    auto result = arrow::ipc::RecordBatchStreamReader::Open(m_poFile, m_oOptions);
    if( !result.ok() )
    {
        CPLError(CE_Failure, CPLE_AppDefined,
                 "RecordBatchStreamReader::Open() failed with %s",
                 result.status().message().c_str());
        CPL_IGNORE_RET_VAL(m_poFile->Seek(nPos));
        return false;
    }
    else
    {
        m_poRecordBatchReader = *result;
        return true;
    }
}

/************************************************************************/
/*                     ComputeGeometryColumnType()                      */
/************************************************************************/

OGRwkbGeometryType OGRFeatherLayer::ComputeGeometryColumnType(int iGeomCol,
                                                              int iCol) const
{
    // Compute type of geometry column by iterating over each geometry, and
    // looking at the WKB geometry type in the first 5 bytes of each geometry.

    OGRwkbGeometryType eGeomType = wkbNone;

    if( m_poRecordBatchReader != nullptr )
    {
        std::shared_ptr<arrow::RecordBatch> poBatch;
        while( true )
        {
            auto status = m_poRecordBatchReader->ReadNext(&poBatch);
            if( !status.ok() )
            {
                CPLError(CE_Failure, CPLE_AppDefined,
                         "ReadNext() failed: %s",
                         status.message().c_str());
                break;
            }
            else if( !poBatch )
                break;
            eGeomType = ComputeGeometryColumnTypeProcessBatch(poBatch,
                                                              iGeomCol, iCol,
                                                              eGeomType);
            if( eGeomType == wkbUnknown )
                break;
        }
    }
    else
    {
        for(int iBatch = 0; iBatch < m_poRecordBatchFileReader->num_record_batches(); ++iBatch )
        {
            auto result = m_poRecordBatchFileReader->ReadRecordBatch(iBatch);
            if( !result.ok() )
            {
                CPLError(CE_Failure, CPLE_AppDefined,
                         "ReadRecordBatch() failed: %s",
                         result.status().message().c_str());
                break;
            }
            eGeomType = ComputeGeometryColumnTypeProcessBatch(*result,
                                                              iGeomCol, iCol,
                                                              eGeomType);
            if( eGeomType == wkbUnknown )
                break;
        }
    }

    return eGeomType == wkbNone ? wkbUnknown : eGeomType;
}

/************************************************************************/
/*                         CreateFieldFromSchema()                      */
/************************************************************************/

void OGRFeatherLayer::CreateFieldFromSchema(
    const std::shared_ptr<arrow::Field>& field,
    const std::vector<int>& path,
    const std::map<std::string, std::unique_ptr<OGRFieldDefn>>& oMapFieldNameToGDALSchemaFieldDefn)
{
    OGRFieldDefn oField(field->name().c_str(), OFTString);
    OGRFieldType eType = OFTString;
    OGRFieldSubType eSubType = OFSTNone;
    bool bTypeOK = true;

    auto type = field->type();
    if( type->id() == arrow::Type::DICTIONARY && path.size() == 1 )
    {
        const auto dictionaryType = std::static_pointer_cast<arrow::DictionaryType>(field->type());
        const auto indexType = dictionaryType->index_type();
        if( dictionaryType->value_type()->id() == arrow::Type::STRING &&
            IsIntegerArrowType(indexType->id()) )
        {
            std::string osDomainName(field->name() + "Domain");
            m_poDS->RegisterDomainName(osDomainName, m_poFeatureDefn->GetFieldCount());
            oField.SetDomainName(osDomainName);
            type = indexType;
        }
        else
        {
            bTypeOK = false;
        }
    }

    if( type->id() == arrow::Type::STRUCT )
    {
        const auto subfields = field->Flatten();
        auto newpath = path;
        newpath.push_back(0);
        for( int j = 0; j < static_cast<int>(subfields.size()); j++ )
        {
            const auto& subfield = subfields[j];
            newpath.back() = j;
            CreateFieldFromSchema(subfield,
                                  newpath, oMapFieldNameToGDALSchemaFieldDefn);
        }
    }
    else if( bTypeOK )
    {
        MapArrowTypeToOGR(type, field, oField, eType, eSubType,
                          path, oMapFieldNameToGDALSchemaFieldDefn);
    }
}

/************************************************************************/
/*                          BuildDomain()                               */
/************************************************************************/

std::unique_ptr<OGRFieldDomain> OGRFeatherLayer::BuildDomain(const std::string& osDomainName,
                                                             int iFieldIndex) const
{
    const int iArrowCol = m_anMapFieldIndexToArrowColumn[iFieldIndex][0];
    CPLAssert( m_poSchema->fields()[iArrowCol]->type()->id() == arrow::Type::DICTIONARY );

    if( m_poRecordBatchReader )
    {
        if( m_poBatch )
        {
            return BuildDomainFromBatch(osDomainName, m_poBatch, iArrowCol);
        }
    }
    else if( m_poRecordBatchFileReader )
    {
        auto result = m_poRecordBatchFileReader->ReadRecordBatch(0);
        if( !result.ok() )
        {
            CPLError(CE_Failure, CPLE_AppDefined,
                     "ReadRecordBatch() failed: %s",
                     result.status().message().c_str());
        }
        auto poBatch = *result;
        if( poBatch )
        {
            return BuildDomainFromBatch(osDomainName, poBatch, iArrowCol);
        }
    }

    return nullptr;
}

/************************************************************************/
/*                           ResetReading()                             */
/************************************************************************/

void OGRFeatherLayer::ResetReading()
{
    if( m_poRecordBatchReader != nullptr && m_iRecordBatch > 0 )
    {
        if( m_iRecordBatch == 1 && m_poBatchIdx1 )
        {
            // do nothing
        }
        else
        {
            m_bResetRecordBatchReaderAsked = true;
        }
    }
    OGRArrowLayer::ResetReading();
}

/************************************************************************/
/*                           ReadNextBatch()                            */
/************************************************************************/

bool OGRFeatherLayer::ReadNextBatch()
{
    if( m_poRecordBatchFileReader == nullptr )
    {
        return ReadNextBatchStream();
    }
    else
    {
        return ReadNextBatchFile();
    }
}

/************************************************************************/
/*                         ReadNextBatchFile()                          */
/************************************************************************/

bool OGRFeatherLayer::ReadNextBatchFile()
{
    ++m_iRecordBatch;
    if( m_iRecordBatch == m_poRecordBatchFileReader->num_record_batches() )
    {
        if( m_iRecordBatch == 1 )
            m_iRecordBatch = 0;
        else
            m_poBatch.reset();
        return false;
    }

    m_nIdxInBatch = 0;

    auto result = m_poRecordBatchFileReader->ReadRecordBatch(m_iRecordBatch);
    if( !result.ok() )
    {
        CPLError(CE_Failure, CPLE_AppDefined,
                 "ReadRecordBatch() failed: %s",
                 result.status().message().c_str());
        m_poBatch.reset();
        return false;
    }
    SetBatch(*result);

    return true;
}

/************************************************************************/
/*                         ReadNextBatchStream()                        */
/************************************************************************/

bool OGRFeatherLayer::ReadNextBatchStream()
{
    m_nIdxInBatch = 0;

    if( m_iRecordBatch == 0 && m_poBatchIdx0 )
    {
        SetBatch(m_poBatchIdx0);
        m_iRecordBatch = 1;
        return true;
    }

    else if( m_iRecordBatch == 1 && m_poBatchIdx1 )
    {
        SetBatch(m_poBatchIdx1);
        m_iRecordBatch = 2;
        return true;
    }

    else if( m_bSingleBatch )
    {
        CPLAssert( m_iRecordBatch == 0);
        CPLAssert( m_poBatch != nullptr);
        return false;
    }

    if( m_bResetRecordBatchReaderAsked )
    {
        if( !m_bSeekable )
        {
            CPLError(CE_Failure, CPLE_NotSupported,
                     "Attempting to rewind non-seekable stream");
            return false;
        }
        if( !ResetRecordBatchReader() )
            return false;
        m_bResetRecordBatchReaderAsked = false;
    }

    CPLAssert(m_poRecordBatchReader);

    ++m_iRecordBatch;

    std::shared_ptr<arrow::RecordBatch> poNextBatch;
    auto status = m_poRecordBatchReader->ReadNext(&poNextBatch);
    if( !status.ok() )
    {
        CPLError(CE_Failure, CPLE_AppDefined,
                 "ReadNext() failed: %s",
                 status.message().c_str());
        poNextBatch.reset();
    }
    if( poNextBatch == nullptr )
    {
        if( m_iRecordBatch == 1 )
        {
            m_iRecordBatch = 0;
            m_bSingleBatch = true;
        }
        else
        {
            m_poBatch.reset();
            m_poBatchColumns.clear();
        }
        return false;
    }
    SetBatch(poNextBatch);

    return true;
}

/************************************************************************/
/*                     TryToCacheFirstTwoBatches()                      */
/************************************************************************/

void OGRFeatherLayer::TryToCacheFirstTwoBatches()
{
    if( m_poRecordBatchReader != nullptr && m_iRecordBatch <= 0 &&
        !m_bSingleBatch && m_poBatchIdx0 == nullptr )
    {
        ResetReading();
        if( !m_poBatch )
        {
            CPL_IGNORE_RET_VAL(ReadNextBatchStream());
        }
        if( m_poBatch )
        {
            auto poBatchIdx0 = m_poBatch;
            if( ReadNextBatchStream() )
            {
                CPLAssert(m_iRecordBatch == 1);
                m_poBatchIdx0 = poBatchIdx0;
                m_poBatchIdx1 = m_poBatch;
                SetBatch(poBatchIdx0);
                ResetReading();
            }
            ResetReading();
        }
    }
}

/************************************************************************/
/*                        GetFeatureCount()                             */
/************************************************************************/

GIntBig OGRFeatherLayer::GetFeatureCount(int bForce)
{
    if( m_poRecordBatchFileReader != nullptr &&
        m_poAttrQuery == nullptr && m_poFilterGeom == nullptr )
    {
        auto result = m_poRecordBatchFileReader->CountRows();
        if( result.ok() )
            return *result;
    }
    else if( m_poRecordBatchReader != nullptr )
    {
        if( !m_bSeekable && !bForce )
        {
            if( m_poAttrQuery == nullptr && m_poFilterGeom == nullptr )
            {
                TryToCacheFirstTwoBatches();
            }

            if( !m_bSingleBatch )
            {
                CPLError(CE_Failure, CPLE_AppDefined,
                         "GetFeatureCount() cannot be run in non-forced mode on "
                         "a non-seekable file made of several batches");
                return -1;
            }
        }

        if( m_poAttrQuery == nullptr && m_poFilterGeom == nullptr )
        {
            GIntBig nFeatures = 0;
            ResetReading();
            if( !m_poBatch )
                ReadNextBatchStream();
            while( m_poBatch )
            {
                nFeatures += m_poBatch->num_rows();
                if( !ReadNextBatchStream() )
                    break;
            }
            ResetReading();
            return nFeatures;
        }
    }
    return OGRLayer::GetFeatureCount(bForce);
}

/************************************************************************/
/*                       CanRunNonForcedGetExtent()                     */
/************************************************************************/

bool OGRFeatherLayer::CanRunNonForcedGetExtent()
{
    if( m_bSeekable )
        return true;
    TryToCacheFirstTwoBatches();
    if( !m_bSingleBatch )
    {
        CPLError(CE_Failure, CPLE_AppDefined,
                 "GetExtent() cannot be run in non-forced mode on "
                 "a non-seekable file made of several batches");
        return false;
    }
    return true;
}

/************************************************************************/
/*                         TestCapability()                             */
/************************************************************************/

int OGRFeatherLayer::TestCapability(const char* pszCap)
{
    if( EQUAL(pszCap, OLCFastFeatureCount) )
    {
        return m_bSeekable &&
               m_poAttrQuery == nullptr && m_poFilterGeom == nullptr;
    }

    if( EQUAL(pszCap, OLCFastGetExtent) )
    {
        for(int i = 0; i < m_poFeatureDefn->GetGeomFieldCount(); i++ )
        {
            auto oIter = m_oMapGeometryColumns.find(
            m_poFeatureDefn->GetGeomFieldDefn(i)->GetNameRef() );
            if( oIter == m_oMapGeometryColumns.end() )
            {
                return false;
            }
            const auto& oJSONDef = oIter->second;
            const auto oBBox = oJSONDef.GetArray("bbox");
            if( !(oBBox.IsValid() && (oBBox.Size() == 4 || oBBox.Size() == 6)) )
            {
                return false;
            }
        }
        return true;
    }

    if( EQUAL(pszCap, OLCStringsAsUTF8) )
        return true;

    if( EQUAL(pszCap, OLCMeasuredGeometries) )
        return true;

    return false;
}

/************************************************************************/
/*                         GetMetadataItem()                            */
/************************************************************************/

const char* OGRFeatherLayer::GetMetadataItem( const char* pszName,
                                              const char* pszDomain )
{
    // Mostly for unit test purposes
    if( pszDomain != nullptr && EQUAL(pszDomain, "_ARROW_") )
    {
        if( EQUAL(pszName, "FORMAT") )
        {
            return m_poRecordBatchFileReader ? "FILE": "STREAM";
        }
        if( m_poRecordBatchFileReader != nullptr )
        {
            int iBatch = -1;
            if( EQUAL(pszName, "NUM_RECORD_BATCHES") )
            {
                return CPLSPrintf("%d", m_poRecordBatchFileReader->num_record_batches());
            }
            else if( sscanf(pszName, "RECORD_BATCHES[%d]", &iBatch) == 1 &&
                     strstr(pszName, ".NUM_ROWS") )
            {
                auto result = m_poRecordBatchFileReader->ReadRecordBatch(iBatch);
                if( !result.ok() )
                {
                    return nullptr;
                }
                return CPLSPrintf("%" PRId64, (*result)->num_rows());
            }
        }
        return nullptr;
    }
    else if( pszDomain != nullptr && EQUAL(pszDomain, "_ARROW_METADATA_") )
    {
        const auto kv_metadata = (m_poRecordBatchFileReader ?
            m_poRecordBatchFileReader->schema() : m_poRecordBatchReader->schema())->metadata();
        if( kv_metadata && kv_metadata->Contains(pszName) )
        {
            auto metadataItem = kv_metadata->Get(pszName);
            if( metadataItem.ok() )
            {
                return CPLSPrintf("%s", metadataItem->c_str());
            }
        }
        return nullptr;
    }
    else if( m_poRecordBatchFileReader != nullptr &&
             pszDomain != nullptr && EQUAL(pszDomain, "_ARROW_FOOTER_METADATA_") )
    {
        const auto kv_metadata = m_poRecordBatchFileReader->metadata();
        if( kv_metadata && kv_metadata->Contains(pszName) )
        {
            auto metadataItem = kv_metadata->Get(pszName);
            if( metadataItem.ok() )
            {
                return CPLSPrintf("%s", metadataItem->c_str());
            }
        }
        return nullptr;
    }
    return OGRLayer::GetMetadataItem(pszName, pszDomain);
}

/************************************************************************/
/*                           GetMetadata()                              */
/************************************************************************/

char** OGRFeatherLayer::GetMetadata( const char* pszDomain )
{
    // Mostly for unit test purposes
    if( pszDomain != nullptr && EQUAL(pszDomain, "_ARROW_METADATA_") )
    {
        m_aosFeatherMetadata.Clear();
        const auto kv_metadata = (m_poRecordBatchFileReader ?
            m_poRecordBatchFileReader->schema() : m_poRecordBatchReader->schema())->metadata();
        if( kv_metadata )
        {
            for( const auto& kv: kv_metadata->sorted_pairs() )
            {
                m_aosFeatherMetadata.SetNameValue(kv.first.c_str(), kv.second.c_str());
            }
        }
        return m_aosFeatherMetadata.List();
    }
    if( m_poRecordBatchFileReader != nullptr &&
        pszDomain != nullptr && EQUAL(pszDomain, "_ARROW_FOOTER_METADATA_") )
    {
        m_aosFeatherMetadata.Clear();
        const auto kv_metadata = m_poRecordBatchFileReader->metadata();
        if( kv_metadata )
        {
            for( const auto& kv: kv_metadata->sorted_pairs() )
            {
                m_aosFeatherMetadata.SetNameValue(kv.first.c_str(), kv.second.c_str());
            }
        }
        return m_aosFeatherMetadata.List();
    }
    return OGRLayer::GetMetadata(pszDomain);
}
