/******************************************************************************
 *
 * Project:  DRDC Ottawa GEOINT
 * Purpose:  Radarsat Constellation Mission - XML Products (product.xml) driver
 * Author:   Roberto Caron, MDA
 *           on behalf of DRDC Ottawa
 *
 ******************************************************************************
 * Copyright (c) 2020, DRDC Ottawa
 *
 * Based on the RS2 Dataset Class
 *
 * SPDX-License-Identifier: MIT
 ****************************************************************************/

#include "gdal_frmts.h"
#include "gdalplugindriverproxy.h"

#include "rcmdrivercore.h"

int RCMDatasetIdentify(GDALOpenInfo *poOpenInfo)
{
    /* Check for the case where we're trying to read the calibrated data: */
    if (STARTS_WITH_CI(poOpenInfo->pszFilename, szLayerCalibration) &&
        poOpenInfo->pszFilename[strlen(szLayerCalibration)] == chLayerSeparator)
    {
        return TRUE;
    }

    if (poOpenInfo->bIsDirectory)
    {
        const auto IsRCM = [](const std::string &osMDFilename)
        {
            CPLXMLNode *psProduct = CPLParseXMLFile(osMDFilename.c_str());
            if (psProduct == nullptr)
                return FALSE;

            CPLXMLNode *psProductAttributes =
                CPLGetXMLNode(psProduct, "=product");
            if (psProductAttributes == nullptr)
            {
                CPLDestroyXMLNode(psProduct);
                return FALSE;
            }

            /* Check the namespace only, should be rcm */
            const char *szNamespace =
                CPLGetXMLValue(psProductAttributes, "xmlns", "");

            if (strstr(szNamespace, "rcm") == nullptr)
            {
                /* Invalid namespace */
                CPLDestroyXMLNode(psProduct);
                return FALSE;
            }

            CPLDestroyXMLNode(psProduct);
            return TRUE;
        };

        /* Check for directory access when there is a product.xml file in the
        directory. */
        const std::string osMDFilename = CPLFormCIFilenameSafe(
            poOpenInfo->pszFilename, "product.xml", nullptr);

        VSIStatBufL sStat;
        if (VSIStatL(osMDFilename.c_str(), &sStat) == 0)
        {
            return IsRCM(osMDFilename);
        }

        /* If not, check for directory extra 'metadata' access when there is a
        product.xml file in the directory. */

        const std::string osMDFilenameMetadata = CPLFormCIFilenameSafe(
            poOpenInfo->pszFilename, GetMetadataProduct(), nullptr);

        VSIStatBufL sStatMetadata;
        if (VSIStatL(osMDFilenameMetadata.c_str(), &sStatMetadata) == 0)
        {
            return IsRCM(osMDFilenameMetadata);
        }

        return FALSE;
    }

    /* otherwise, do our normal stuff */
    if (strlen(poOpenInfo->pszFilename) < 11 ||
        !EQUAL(poOpenInfo->pszFilename + strlen(poOpenInfo->pszFilename) - 11,
               "product.xml"))
        return FALSE;

    if (poOpenInfo->nHeaderBytes < 100)
        return FALSE;

    /* The RCM schema location is rcm_prod_product.xsd */
    const char *pszHeader =
        reinterpret_cast<const char *>(poOpenInfo->pabyHeader);
    return strstr(pszHeader, "/rcm") && strstr(pszHeader, "<product");
}

/************************************************************************/
/*                     RCMDriverSetCommonMetadata()                     */
/************************************************************************/

void RCMDriverSetCommonMetadata(GDALDriver *poDriver)
{
    poDriver->SetDescription(RCM_DRIVER_NAME);
    poDriver->SetMetadataItem(GDAL_DCAP_RASTER, "YES");
    poDriver->SetMetadataItem(GDAL_DMD_LONGNAME,
                              "Radarsat Constellation Mission XML Product");
    poDriver->SetMetadataItem(GDAL_DMD_HELPTOPIC, "drivers/raster/rcm.html");
    poDriver->SetMetadataItem(GDAL_DMD_SUBDATASETS, "YES");
    poDriver->pfnIdentify = RCMDatasetIdentify;
    poDriver->SetMetadataItem(GDAL_DCAP_OPEN, "YES");
    poDriver->SetMetadataItem(GDAL_DCAP_VIRTUALIO, "YES");
}

/************************************************************************/
/*                      DeclareDeferredRCMPlugin()                      */
/************************************************************************/

#ifdef PLUGIN_FILENAME
void DeclareDeferredRCMPlugin()
{
    if (GDALGetDriverByName(RCM_DRIVER_NAME) != nullptr)
    {
        return;
    }

    auto poDriver = new GDALPluginDriverProxy(PLUGIN_FILENAME);
#ifdef PLUGIN_INSTALLATION_MESSAGE
    poDriver->SetMetadataItem(GDAL_DMD_PLUGIN_INSTALLATION_MESSAGE,
                              PLUGIN_INSTALLATION_MESSAGE);
#endif
    RCMDriverSetCommonMetadata(poDriver);
    GetGDALDriverManager()->DeclareDeferredPluginDriver(poDriver);
}
#endif
