6147514757

Committed 11 Sep 2023 02:05PM UTC coverage: 89.892% (-0.09%) from 89.985%

Build # 6147514757

Build Type

Pull #873

github

Committed by

web-flow

Commit Message

Merge 3a80eaf16 into 664a57cd7

Pull Request Pull Request #873: Edr provider (TEST)

Run Details

1389 of 1389 new or added lines in 3 files covered. (100.0%)

107997 of 120141 relevant lines covered (89.89%)

60267.52 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

93.02

/services/src/api/model/db_types.rs

use super::{
    datatypes::{
        BoundingBox2D, Breakpoint, ClassificationMeasurement, Colorizer, ContinuousMeasurement,
        Coordinate2D, DataProviderId, DateTimeParseFormat, DefaultColors, FeatureDataType,
        LinearGradient, LogarithmicGradient, Measurement, MultiLineString, MultiPoint,
        MultiPolygon, NoGeometry, OverUnderColors, Palette, RgbaColor, SpatialReferenceOption,
        TimeInstance, TimeInterval, TimeStep, VectorDataType,
    },
    operators::{
        CsvHeader, FileNotFoundHandling, FormatSpecifics, GdalConfigOption,
        GdalDatasetGeoTransform, GdalDatasetParameters, GdalMetaDataList, GdalMetaDataRegular,
        GdalMetaDataStatic, GdalMetadataMapping, GdalMetadataNetCdfCf, GdalSourceTimePlaceholder,
        MockDatasetDataSourceLoadingInfo, MockMetaData, OgrMetaData, OgrSourceColumnSpec,
        OgrSourceDataset, OgrSourceDatasetTimeType, OgrSourceDurationSpec, OgrSourceErrorSpec,
        OgrSourceTimeFormat, PlotResultDescriptor, RasterResultDescriptor, TypedGeometry,
        TypedResultDescriptor, UnixTimeStampType, VectorColumnInfo, VectorResultDescriptor,
    },
};
use crate::{
    datasets::{
        external::{
            aruna::ArunaDataProviderDefinition,
            edr::{EdrDataProviderDefinition, EdrVectorSpec},
            gbif::GbifDataProviderDefinition,
            gfbio_abcd::GfbioAbcdDataProviderDefinition,
            gfbio_collections::GfbioCollectionsDataProviderDefinition,
            netcdfcf::{EbvPortalDataProviderDefinition, NetCdfCfDataProviderDefinition},
            pangaea::PangaeaDataProviderDefinition,
        },
        listing::Provenance,
        storage::MetaDataDefinition,
    },
    error::Error,
    layers::external::TypedDataProviderDefinition,
    projects::{
        ColorParam, DerivedColor, DerivedNumber, LineSymbology, NumberParam, PointSymbology,
        PolygonSymbology, RasterSymbology, Symbology,
    },
    util::postgres::DatabaseConnectionConfig,
};
use fallible_iterator::FallibleIterator;
use geoengine_datatypes::primitives::CacheTtlSeconds;
use ordered_float::NotNan;
use postgres_types::{FromSql, ToSql};
use std::{collections::HashMap, marker::PhantomData};

#[derive(Debug, ToSql, FromSql)]
#[postgres(name = "DefaultColors")]
pub struct DefaultColorsDbType {
    pub default_color: Option<RgbaColor>,
    pub over_color: Option<RgbaColor>,
    pub under_color: Option<RgbaColor>,
}

impl From<&DefaultColors> for DefaultColorsDbType {
    fn from(value: &DefaultColors) -> Self {
        match value {
            DefaultColors::DefaultColor { default_color } => Self {
                default_color: Some(*default_color),
                over_color: None,
                under_color: None,
            },
            DefaultColors::OverUnder(over_under) => Self {
                default_color: None,
                over_color: Some(over_under.over_color),
                under_color: Some(over_under.under_color),
            },
        }
    }
}

impl TryFrom<DefaultColorsDbType> for DefaultColors {
    type Error = Error;

    fn try_from(value: DefaultColorsDbType) -> Result<Self, Self::Error> {
        match (value.default_color, value.over_color, value.under_color) {
            (Some(default_color), None, None) => Ok(Self::DefaultColor { default_color }),
            (None, Some(over_color), Some(under_color)) => Ok(Self::OverUnder(OverUnderColors {
                over_color,
                under_color,
            })),
            _ => Err(Error::UnexpectedInvalidDbTypeConversion),
        }
    }
}

#[derive(Debug, ToSql, FromSql)]
#[postgres(name = "Colorizer")]
pub struct ColorizerDbType {
    r#type: ColorizerTypeDbType,
    breakpoints: Option<Vec<Breakpoint>>,
    no_data_color: Option<RgbaColor>,
    color_fields: Option<DefaultColorsDbType>,
    default_color: Option<RgbaColor>,
}

#[derive(Debug, PartialEq, ToSql, FromSql)]
// TODO: use #[postgres(rename_all = "camelCase")]
#[postgres(name = "ColorizerType")]
pub enum ColorizerTypeDbType {
    LinearGradient,
    LogarithmicGradient,
    Palette,
    Rgba,
}

impl From<&Colorizer> for ColorizerDbType {
    fn from(value: &Colorizer) -> Self {
        match value {
            Colorizer::LinearGradient(gradient) => ColorizerDbType {
                r#type: ColorizerTypeDbType::LinearGradient,
                breakpoints: Some(gradient.breakpoints.clone()),
                no_data_color: Some(gradient.no_data_color),
                color_fields: Some((&gradient.color_fields).into()),
                default_color: None,
            },
            Colorizer::LogarithmicGradient(gradient) => ColorizerDbType {
                r#type: ColorizerTypeDbType::LogarithmicGradient,
                breakpoints: Some(gradient.breakpoints.clone()),
                no_data_color: Some(gradient.no_data_color),
                color_fields: Some((&gradient.color_fields).into()),
                default_color: None,
            },
            Colorizer::Palette {
                colors,
                no_data_color,
                default_color,
            } => ColorizerDbType {
                r#type: ColorizerTypeDbType::Palette,
                breakpoints: Some(
                    colors
                        .0
                        .iter()
                        .map(|(k, v)| Breakpoint {
                            value: (*k).into(),
                            color: *v,
                        })
                        .collect(),
                ),
                no_data_color: Some(*no_data_color),
                color_fields: None,
                default_color: Some(*default_color),
            },
            Colorizer::Rgba => ColorizerDbType {
                r#type: ColorizerTypeDbType::Rgba,
                breakpoints: None,
                no_data_color: None,
                color_fields: None,
                default_color: None,
            },
        }
    }
}

impl TryFrom<ColorizerDbType> for Colorizer {
    type Error = Error;

    fn try_from(value: ColorizerDbType) -> Result<Self, Self::Error> {
        match value {
            ColorizerDbType {
                r#type: ColorizerTypeDbType::LinearGradient,
                breakpoints: Some(breakpoints),
                no_data_color: Some(no_data_color),
                color_fields: Some(color_fields),
                default_color: None,
            } => Ok(Self::LinearGradient(LinearGradient {
                breakpoints,
                no_data_color,
                color_fields: color_fields.try_into()?,
            })),
            ColorizerDbType {
                r#type: ColorizerTypeDbType::LogarithmicGradient,
                breakpoints: Some(breakpoints),
                no_data_color: Some(no_data_color),
                color_fields: Some(color_fields),
                default_color: None,
            } => Ok(Self::LogarithmicGradient(LogarithmicGradient {
                breakpoints,
                no_data_color,
                color_fields: color_fields.try_into()?,
            })),
            ColorizerDbType {
                r#type: ColorizerTypeDbType::Palette,
                breakpoints: Some(breakpoints),
                no_data_color: Some(no_data_color),
                color_fields: None,
                default_color: Some(default_color),
            } => Ok(Self::Palette {
                colors: Palette(
                    breakpoints
                        .into_iter()
                        .map(|b| (NotNan::<f64>::from(b.value), b.color))
                        .collect(),
                ),
                no_data_color,
                default_color,
            }),
            ColorizerDbType {
                r#type: ColorizerTypeDbType::Rgba,
                breakpoints: None,
                no_data_color: None,
                color_fields: None,
                default_color: None,
            } => Ok(Self::Rgba),
            _ => Err(Error::UnexpectedInvalidDbTypeConversion),
        }
    }
}

#[derive(Debug, ToSql, FromSql)]
#[postgres(name = "ColorParam")]
pub struct ColorParamDbType {
    color: Option<RgbaColor>,
    attribute: Option<String>,
    colorizer: Option<Colorizer>,
}

impl From<&ColorParam> for ColorParamDbType {
    fn from(value: &ColorParam) -> Self {
        match value {
            ColorParam::Static { color } => Self {
                color: Some((*color).into()),
                attribute: None,
                colorizer: None,
            },
            ColorParam::Derived(DerivedColor {
                attribute,
                colorizer,
            }) => Self {
                color: None,
                attribute: Some(attribute.clone()),
                colorizer: Some(colorizer.clone()),
            },
        }
    }
}

impl TryFrom<ColorParamDbType> for ColorParam {
    type Error = Error;

    fn try_from(value: ColorParamDbType) -> Result<Self, Self::Error> {
        match value {
            ColorParamDbType {
                color: Some(color),
                attribute: None,
                colorizer: None,
            } => Ok(Self::Static {
                color: color.into(),
            }),
            ColorParamDbType {
                color: None,
                attribute: Some(attribute),
                colorizer: Some(colorizer),
            } => Ok(Self::Derived(DerivedColor {
                attribute,
                colorizer,
            })),
            _ => Err(Error::UnexpectedInvalidDbTypeConversion),
        }
    }
}

#[derive(Debug, ToSql, FromSql)]
#[postgres(name = "NumberParam")]
pub struct NumberParamDbType {
    value: Option<i64>,
    attribute: Option<String>,
    factor: Option<f64>,
    default_value: Option<f64>,
}

impl From<&NumberParam> for NumberParamDbType {
    fn from(value: &NumberParam) -> Self {
        match value {
            NumberParam::Static { value } => Self {
                value: Some(*value as i64),
                attribute: None,
                factor: None,
                default_value: None,
            },
            NumberParam::Derived(DerivedNumber {
                attribute,
                factor,
                default_value,
            }) => Self {
                value: None,
                attribute: Some(attribute.clone()),
                factor: Some(*factor),
                default_value: Some(*default_value),
            },
        }
    }
}

impl TryFrom<NumberParamDbType> for NumberParam {
    type Error = Error;

    fn try_from(value: NumberParamDbType) -> Result<Self, Self::Error> {
        match value {
            NumberParamDbType {
                value: Some(value),
                attribute: None,
                factor: None,
                default_value: None,
            } => Ok(Self::Static {
                value: value as usize,
            }),
            NumberParamDbType {
                value: None,
                attribute: Some(attribute),
                factor: Some(factor),
                default_value: Some(default_value),
            } => Ok(Self::Derived(DerivedNumber {
                attribute,
                factor,
                default_value,
            })),
            _ => Err(Error::UnexpectedInvalidDbTypeConversion),
        }
    }
}

#[derive(Debug, ToSql, FromSql)]
#[postgres(name = "Symbology")]
pub struct SymbologyDbType {
    raster: Option<RasterSymbology>,
    point: Option<PointSymbology>,
    line: Option<LineSymbology>,
    polygon: Option<PolygonSymbology>,
}

impl From<&Symbology> for SymbologyDbType {
    fn from(symbology: &Symbology) -> Self {
        match symbology {
            Symbology::Raster(raster) => SymbologyDbType {
                raster: Some(raster.clone()),
                point: None,
                line: None,
                polygon: None,
            },
            Symbology::Point(point) => SymbologyDbType {
                raster: None,
                point: Some(point.clone()),
                line: None,
                polygon: None,
            },
            Symbology::Line(line) => SymbologyDbType {
                raster: None,
                point: None,
                line: Some(line.clone()),
                polygon: None,
            },
            Symbology::Polygon(polygon) => SymbologyDbType {
                raster: None,
                point: None,
                line: None,
                polygon: Some(polygon.clone()),
            },
        }
    }
}

impl TryFrom<SymbologyDbType> for Symbology {
    type Error = Error;

    fn try_from(symbology: SymbologyDbType) -> Result<Self, Self::Error> {
        match symbology {
            SymbologyDbType {
                raster: Some(raster),
                point: None,
                line: None,
                polygon: None,
            } => Ok(Self::Raster(raster)),
            SymbologyDbType {
                raster: None,
                point: Some(point),
                line: None,
                polygon: None,
            } => Ok(Self::Point(point)),
            SymbologyDbType {
                raster: None,
                point: None,
                line: Some(line),
                polygon: None,
            } => Ok(Self::Line(line)),
            SymbologyDbType {
                raster: None,
                point: None,
                line: None,
                polygon: Some(polygon),
            } => Ok(Self::Polygon(polygon)),
            _ => Err(Error::UnexpectedInvalidDbTypeConversion),
        }
    }
}

#[derive(Debug, ToSql, FromSql)]
#[postgres(name = "Measurement")]
pub struct MeasurementDbType {
    continuous: Option<ContinuousMeasurement>,
    classification: Option<ClassificationMeasurementDbType>,
}

#[derive(Debug, ToSql, FromSql)]
struct SmallintTextKeyValue {
    key: i16,
    value: String,
}

#[derive(Debug, ToSql, FromSql)]
#[postgres(name = "ClassificationMeasurement")]
pub struct ClassificationMeasurementDbType {
    measurement: String,
    classes: Vec<SmallintTextKeyValue>,
}

impl From<&Measurement> for MeasurementDbType {
    fn from(measurement: &Measurement) -> Self {
        match measurement {
            Measurement::Unitless => Self {
                continuous: None,
                classification: None,
            },
            Measurement::Continuous(measurement) => Self {
                continuous: Some(measurement.clone()),
                classification: None,
            },
            Measurement::Classification(measurement) => Self {
                continuous: None,
                classification: Some(ClassificationMeasurementDbType {
                    measurement: measurement.measurement.clone(),
                    classes: measurement
                        .classes
                        .iter()
                        .map(|(key, value)| SmallintTextKeyValue {
                            key: i16::from(*key),
                            value: value.clone(),
                        })
                        .collect(),
                }),
            },
        }
    }
}

impl TryFrom<MeasurementDbType> for Measurement {
    type Error = Error;

    fn try_from(measurement: MeasurementDbType) -> Result<Self, Self::Error> {
        match measurement {
            MeasurementDbType {
                continuous: None,
                classification: None,
            } => Ok(Self::Unitless),
            MeasurementDbType {
                continuous: Some(continuous),
                classification: None,
            } => Ok(Self::Continuous(continuous)),
            MeasurementDbType {
                continuous: None,
                classification: Some(classification),
            } => {
                let mut classes = HashMap::with_capacity(classification.classes.len());
                for SmallintTextKeyValue { key, value } in classification.classes {
                    classes.insert(
                        u8::try_from(key).map_err(|_| Error::UnexpectedInvalidDbTypeConversion)?,
                        value,
                    );
                }

                Ok(Self::Classification(ClassificationMeasurement {
                    measurement: classification.measurement,
                    classes,
                }))
            }
            _ => Err(Error::UnexpectedInvalidDbTypeConversion),
        }
    }
}

#[derive(Debug, FromSql, ToSql)]
#[postgres(name = "VectorColumnInfo")]
pub struct VectorColumnInfoDbType {
    pub column: String,
    pub data_type: FeatureDataType,
    pub measurement: Measurement,
}

#[derive(Debug, FromSql, ToSql)]
#[postgres(name = "VectorResultDescriptor")]
pub struct VectorResultDescriptorDbType {
    pub data_type: VectorDataType,
    pub spatial_reference: SpatialReferenceOption,
    pub columns: Vec<VectorColumnInfoDbType>,
    pub time: Option<TimeInterval>,
    pub bbox: Option<BoundingBox2D>,
}

impl From<&VectorResultDescriptor> for VectorResultDescriptorDbType {
    fn from(result_descriptor: &VectorResultDescriptor) -> Self {
        Self {
            data_type: result_descriptor.data_type,
            spatial_reference: result_descriptor.spatial_reference,
            columns: result_descriptor
                .columns
                .iter()
                .map(|(column, info)| VectorColumnInfoDbType {
                    column: column.clone(),
                    data_type: info.data_type,
                    measurement: info.measurement.clone(),
                })
                .collect(),
            time: result_descriptor.time,
            bbox: result_descriptor.bbox,
        }
    }
}

impl TryFrom<VectorResultDescriptorDbType> for VectorResultDescriptor {
    type Error = Error;

    fn try_from(result_descriptor: VectorResultDescriptorDbType) -> Result<Self, Self::Error> {
        Ok(Self {
            data_type: result_descriptor.data_type,
            spatial_reference: result_descriptor.spatial_reference,
            columns: result_descriptor
                .columns
                .into_iter()
                .map(|info| {
                    (
                        info.column,
                        VectorColumnInfo {
                            data_type: info.data_type,
                            measurement: info.measurement,
                        },
                    )
                })
                .collect(),
            time: result_descriptor.time,
            bbox: result_descriptor.bbox,
        })
    }
}

#[derive(Debug, ToSql, FromSql)]
#[postgres(name = "ResultDescriptor")]
pub struct TypedResultDescriptorDbType {
    raster: Option<RasterResultDescriptor>,
    vector: Option<VectorResultDescriptor>,
    plot: Option<PlotResultDescriptor>,
}

impl From<&TypedResultDescriptor> for TypedResultDescriptorDbType {
    fn from(result_descriptor: &TypedResultDescriptor) -> Self {
        match result_descriptor {
            TypedResultDescriptor::Raster(raster) => Self {
                raster: Some(raster.clone()),
                vector: None,
                plot: None,
            },
            TypedResultDescriptor::Vector(vector) => Self {
                raster: None,
                vector: Some(vector.clone()),
                plot: None,
            },
            TypedResultDescriptor::Plot(plot) => Self {
                raster: None,
                vector: None,
                plot: Some(*plot),
            },
        }
    }
}

impl TryFrom<TypedResultDescriptorDbType> for TypedResultDescriptor {
    type Error = Error;

    fn try_from(result_descriptor: TypedResultDescriptorDbType) -> Result<Self, Self::Error> {
        match result_descriptor {
            TypedResultDescriptorDbType {
                raster: Some(raster),
                vector: None,
                plot: None,
            } => Ok(Self::Raster(raster)),
            TypedResultDescriptorDbType {
                raster: None,
                vector: Some(vector),
                plot: None,
            } => Ok(Self::Vector(vector)),
            TypedResultDescriptorDbType {
                raster: None,
                vector: None,
                plot: Some(plot),
            } => Ok(Self::Plot(plot)),
            _ => Err(Error::UnexpectedInvalidDbTypeConversion),
        }
    }
}

#[derive(Debug, PartialEq, ToSql, FromSql)]
pub struct TextTextKeyValue {
    key: String,
    value: String,
}

#[derive(Debug, PartialEq, ToSql, FromSql)]
#[postgres(transparent)]
pub struct HashMapTextTextDbType(pub Vec<TextTextKeyValue>);

impl From<&HashMap<String, String>> for HashMapTextTextDbType {
    fn from(map: &HashMap<String, String>) -> Self {
        Self(
            map.iter()
                .map(|(key, value)| TextTextKeyValue {
                    key: key.clone(),
                    value: value.clone(),
                })
                .collect(),
        )
    }
}

impl<S: std::hash::BuildHasher + std::default::Default> From<HashMapTextTextDbType>
    for HashMap<String, String, S>
{
    fn from(map: HashMapTextTextDbType) -> Self {
        map.0
            .into_iter()
            .map(|TextTextKeyValue { key, value }| (key, value))
            .collect()
    }
}

#[derive(Debug, ToSql, FromSql)]
#[postgres(name = "OgrSourceTimeFormat")]
pub struct OgrSourceTimeFormatDbType {
    custom: Option<OgrSourceTimeFormatCustomDbType>,
    unix_time_stamp: Option<OgrSourceTimeFormatUnixTimeStampDbType>,
}

#[derive(Debug, ToSql, FromSql)]
#[postgres(name = "OgrSourceTimeFormatCustom")]
pub struct OgrSourceTimeFormatCustomDbType {
    custom_format: DateTimeParseFormat,
}

#[derive(Debug, ToSql, FromSql)]
#[postgres(name = "OgrSourceTimeFormatUnixTimeStamp")]
pub struct OgrSourceTimeFormatUnixTimeStampDbType {
    timestamp_type: UnixTimeStampType,
    fmt: DateTimeParseFormat,
}

impl From<&OgrSourceTimeFormat> for OgrSourceTimeFormatDbType {
    fn from(other: &OgrSourceTimeFormat) -> Self {
        match other {
            OgrSourceTimeFormat::Custom { custom_format } => Self {
                custom: Some(OgrSourceTimeFormatCustomDbType {
                    custom_format: custom_format.clone(),
                }),
                unix_time_stamp: None,
            },
            OgrSourceTimeFormat::UnixTimeStamp {
                timestamp_type,
                fmt,
            } => Self {
                custom: None,
                unix_time_stamp: Some(OgrSourceTimeFormatUnixTimeStampDbType {
                    timestamp_type: *timestamp_type,
                    fmt: fmt.clone(),
                }),
            },
            OgrSourceTimeFormat::Auto => Self {
                custom: None,
                unix_time_stamp: None,
            },
        }
    }
}

impl TryFrom<OgrSourceTimeFormatDbType> for OgrSourceTimeFormat {
    type Error = Error;

    fn try_from(other: OgrSourceTimeFormatDbType) -> Result<Self, Self::Error> {
        match other {
            OgrSourceTimeFormatDbType {
                custom: Some(custom),
                unix_time_stamp: None,
            } => Ok(Self::Custom {
                custom_format: custom.custom_format,
            }),
            OgrSourceTimeFormatDbType {
                custom: None,
                unix_time_stamp: Some(unix_time_stamp),
            } => Ok(Self::UnixTimeStamp {
                timestamp_type: unix_time_stamp.timestamp_type,
                fmt: unix_time_stamp.fmt,
            }),
            OgrSourceTimeFormatDbType {
                custom: None,
                unix_time_stamp: None,
            } => Ok(Self::Auto),
            _ => Err(Error::UnexpectedInvalidDbTypeConversion),
        }
    }
}

#[derive(Debug, ToSql, FromSql)]
#[postgres(name = "OgrSourceDurationSpec")]
pub struct OgrSourceDurationSpecDbType {
    infinite: bool,
    zero: bool,
    value: Option<TimeStep>,
}

impl From<&OgrSourceDurationSpec> for OgrSourceDurationSpecDbType {
    fn from(other: &OgrSourceDurationSpec) -> Self {
        match other {
            OgrSourceDurationSpec::Infinite => Self {
                infinite: true,
                zero: false,
                value: None,
            },
            OgrSourceDurationSpec::Zero => Self {
                infinite: false,
                zero: true,
                value: None,
            },
            OgrSourceDurationSpec::Value(value) => Self {
                infinite: false,
                zero: false,
                value: Some(*value),
            },
        }
    }
}

impl TryFrom<OgrSourceDurationSpecDbType> for OgrSourceDurationSpec {
    type Error = Error;

    fn try_from(other: OgrSourceDurationSpecDbType) -> Result<Self, Self::Error> {
        match other {
            OgrSourceDurationSpecDbType {
                infinite: true,
                zero: false,
                value: None,
            } => Ok(Self::Infinite),
            OgrSourceDurationSpecDbType {
                infinite: false,
                zero: true,
                value: None,
            } => Ok(Self::Zero),
            OgrSourceDurationSpecDbType {
                infinite: false,
                zero: false,
                value: Some(value),
            } => Ok(Self::Value(value)),
            _ => Err(Error::UnexpectedInvalidDbTypeConversion),
        }
    }
}

#[derive(Debug, ToSql, FromSql)]
#[postgres(name = "OgrSourceDatasetTimeType")]
pub struct OgrSourceDatasetTimeTypeDbType {
    start: Option<OgrSourceDatasetTimeTypeStartDbType>,
    start_end: Option<OgrSourceDatasetTimeTypeStartEndDbType>,
    start_duration: Option<OgrSourceDatasetTimeTypeStartDurationDbType>,
}

#[derive(Debug, ToSql, FromSql)]
#[postgres(name = "OgrSourceDatasetTimeTypeStart")]
pub struct OgrSourceDatasetTimeTypeStartDbType {
    start_field: String,
    start_format: OgrSourceTimeFormat,
    duration: OgrSourceDurationSpec,
}

#[derive(Debug, ToSql, FromSql)]
#[postgres(name = "OgrSourceDatasetTimeTypeStartEnd")]
pub struct OgrSourceDatasetTimeTypeStartEndDbType {
    start_field: String,
    start_format: OgrSourceTimeFormat,
    end_field: String,
    end_format: OgrSourceTimeFormat,
}

#[derive(Debug, ToSql, FromSql)]
#[postgres(name = "OgrSourceDatasetTimeTypeStartDuration")]
pub struct OgrSourceDatasetTimeTypeStartDurationDbType {
    start_field: String,
    start_format: OgrSourceTimeFormat,
    duration_field: String,
}

impl From<&OgrSourceDatasetTimeType> for OgrSourceDatasetTimeTypeDbType {
    fn from(other: &OgrSourceDatasetTimeType) -> Self {
        match other {
            OgrSourceDatasetTimeType::None => Self {
                start: None,
                start_end: None,
                start_duration: None,
            },
            OgrSourceDatasetTimeType::Start {
                start_field,
                start_format,
                duration,
            } => Self {
                start: Some(OgrSourceDatasetTimeTypeStartDbType {
                    start_field: start_field.clone(),
                    start_format: start_format.clone(),
                    duration: *duration,
                }),
                start_end: None,
                start_duration: None,
            },
            OgrSourceDatasetTimeType::StartEnd {
                start_field,
                start_format,
                end_field,
                end_format,
            } => Self {
                start: None,
                start_end: Some(OgrSourceDatasetTimeTypeStartEndDbType {
                    start_field: start_field.clone(),
                    start_format: start_format.clone(),
                    end_field: end_field.clone(),
                    end_format: end_format.clone(),
                }),
                start_duration: None,
            },
            OgrSourceDatasetTimeType::StartDuration {
                start_field,
                start_format,
                duration_field,
            } => Self {
                start: None,
                start_end: None,
                start_duration: Some(OgrSourceDatasetTimeTypeStartDurationDbType {
                    start_field: start_field.clone(),
                    start_format: start_format.clone(),
                    duration_field: duration_field.clone(),
                }),
            },
        }
    }
}

impl TryFrom<OgrSourceDatasetTimeTypeDbType> for OgrSourceDatasetTimeType {
    type Error = Error;

    fn try_from(other: OgrSourceDatasetTimeTypeDbType) -> Result<Self, Self::Error> {
        match other {
            OgrSourceDatasetTimeTypeDbType {
                start: None,
                start_end: None,
                start_duration: None,
            } => Ok(Self::None),
            OgrSourceDatasetTimeTypeDbType {
                start: Some(start),
                start_end: None,
                start_duration: None,
            } => Ok(Self::Start {
                start_field: start.start_field,
                start_format: start.start_format,
                duration: start.duration,
            }),
            OgrSourceDatasetTimeTypeDbType {
                start: None,
                start_end: Some(start_end),
                start_duration: None,
            } => Ok(Self::StartEnd {
                start_field: start_end.start_field,
                start_format: start_end.start_format,
                end_field: start_end.end_field,
                end_format: start_end.end_format,
            }),
            OgrSourceDatasetTimeTypeDbType {
                start: None,
                start_end: None,
                start_duration: Some(start_duration),
            } => Ok(Self::StartDuration {
                start_field: start_duration.start_field,
                start_format: start_duration.start_format,
                duration_field: start_duration.duration_field,
            }),
            _ => Err(Error::UnexpectedInvalidDbTypeConversion),
        }
    }
}

#[derive(Debug, ToSql, FromSql)]
#[postgres(name = "FormatSpecifics")]
pub struct FormatSpecificsDbType {
    csv: Option<FormatSpecificsCsvDbType>,
}

#[derive(Debug, ToSql, FromSql)]
#[postgres(name = "FormatSpecificsCsv")]
pub struct FormatSpecificsCsvDbType {
    header: CsvHeader,
}

impl From<&FormatSpecifics> for FormatSpecificsDbType {
    fn from(other: &FormatSpecifics) -> Self {
        match other {
            FormatSpecifics::Csv { header } => Self {
                csv: Some(FormatSpecificsCsvDbType { header: *header }),
            },
        }
    }
}

impl TryFrom<FormatSpecificsDbType> for FormatSpecifics {
    type Error = Error;

    fn try_from(other: FormatSpecificsDbType) -> Result<Self, Self::Error> {
        match other {
            FormatSpecificsDbType {
                csv: Some(FormatSpecificsCsvDbType { header }),
            } => Ok(Self::Csv { header }),
            _ => Err(Error::UnexpectedInvalidDbTypeConversion),
        }
    }
}

#[derive(Debug, ToSql, FromSql)]
#[postgres(name = "OgrSourceColumnSpec")]
pub struct OgrSourceColumnSpecDbType {
    pub format_specifics: Option<FormatSpecifics>,
    pub x: String,
    pub y: Option<String>,
    pub int: Vec<String>,
    pub float: Vec<String>,
    pub text: Vec<String>,
    pub bool: Vec<String>,
    pub datetime: Vec<String>,
    pub rename: Option<HashMapTextTextDbType>,
}

impl From<&OgrSourceColumnSpec> for OgrSourceColumnSpecDbType {
    fn from(other: &OgrSourceColumnSpec) -> Self {
        Self {
            format_specifics: other.format_specifics.clone(),
            x: other.x.clone(),
            y: other.y.clone(),
            int: other.int.clone(),
            float: other.float.clone(),
            text: other.text.clone(),
            bool: other.bool.clone(),
            datetime: other.datetime.clone(),
            rename: other.rename.as_ref().map(Into::into),
        }
    }
}

impl TryFrom<OgrSourceColumnSpecDbType> for OgrSourceColumnSpec {
    type Error = Error;

    fn try_from(other: OgrSourceColumnSpecDbType) -> Result<Self, Self::Error> {
        Ok(Self {
            format_specifics: other.format_specifics,
            x: other.x,
            y: other.y,
            int: other.int,
            float: other.float,
            text: other.text,
            bool: other.bool,
            datetime: other.datetime,
            rename: other.rename.map(Into::into),
        })
    }
}

#[derive(Debug)]
pub struct PolygonRef<'p> {
    pub rings: &'p [Vec<Coordinate2D>],
}

#[derive(Debug)]
pub struct PolygonOwned {
    pub rings: Vec<Vec<Coordinate2D>>,
}

impl ToSql for PolygonRef<'_> {
    fn to_sql(
        &self,
        ty: &postgres_types::Type,
        w: &mut bytes::BytesMut,
    ) -> Result<postgres_types::IsNull, Box<dyn std::error::Error + Sync + Send>> {
        let postgres_types::Kind::Domain(domain_type) = ty.kind() else {
            panic!("expected domain type");
        };

        let postgres_types::Kind::Array(member_type) = domain_type.kind() else {
            panic!("expected array type");
        };

        let dimension = postgres_protocol::types::ArrayDimension {
            len: self.rings.len() as i32,
            lower_bound: 1, // arrays are one-indexed
        };

        postgres_protocol::types::array_to_sql(
            Some(dimension),
            member_type.oid(),
            self.rings.iter(),
            |coordinates, w| {
                postgres_protocol::types::path_to_sql(
                    true,
                    coordinates.iter().map(|p| (p.x, p.y)),
                    w,
                )?;

                Ok(postgres_protocol::IsNull::No)
            },
            w,
        )?;

        Ok(postgres_types::IsNull::No)
    }

    fn accepts(ty: &postgres_types::Type) -> bool {
        if ty.name() != "Polygon" {
            return false;
        }

        let postgres_types::Kind::Domain(inner_type) = ty.kind() else {
            return false;
        };

        let postgres_types::Kind::Array(inner_type) = inner_type.kind() else {
            return false;
        };

        matches!(inner_type, &postgres_types::Type::PATH)
    }

    postgres_types::to_sql_checked!();
}

impl<'a> FromSql<'a> for PolygonOwned {
    fn from_sql(
        _ty: &postgres_types::Type,
        raw: &'a [u8],
    ) -> Result<Self, Box<dyn std::error::Error + Sync + Send>> {
        let array = postgres_protocol::types::array_from_sql(raw)?;
        if array.dimensions().count()? > 1 {
            return Err("array contains too many dimensions".into());
        }

        let rings = array
            .values()
            .map(|raw| {
                let Some(raw) = raw else {
                    return Err("array contains NULL values".into());
                };
                let path = postgres_protocol::types::path_from_sql(raw)?;

                let coordinates = path
                    .points()
                    .map(|point| {
                        Ok(Coordinate2D {
                            x: point.x(),
                            y: point.y(),
                        })
                    })
                    .collect()?;
                Ok(coordinates)
            })
            .collect()?;

        Ok(Self { rings })
    }

    fn accepts(ty: &postgres_types::Type) -> bool {
        if ty.name() != "Polygon" {
            return false;
        }

        let postgres_types::Kind::Domain(inner_type) = ty.kind() else {
            return false;
        };

        let postgres_types::Kind::Array(inner_type) = inner_type.kind() else {
            return false;
        };

        matches!(inner_type, &postgres_types::Type::PATH)
    }
}

#[derive(Debug, ToSql, FromSql)]
#[postgres(name = "TypedGeometry")]
pub struct TypedGeometryDbType {
    data: bool,
    multi_point: Option<MultiPoint>,
    multi_line_string: Option<MultiLineString>,
    multi_polygon: Option<MultiPolygon>,
}

impl From<&TypedGeometry> for TypedGeometryDbType {
    fn from(other: &TypedGeometry) -> Self {
        match other {
            TypedGeometry::Data(_) => Self {
                data: true,
                multi_point: None,
                multi_line_string: None,
                multi_polygon: None,
            },
            TypedGeometry::MultiPoint(points) => Self {
                data: false,
                multi_point: Some(points.clone()),
                multi_line_string: None,
                multi_polygon: None,
            },
            TypedGeometry::MultiLineString(lines) => Self {
                data: false,
                multi_point: None,
                multi_line_string: Some(lines.clone()),
                multi_polygon: None,
            },
            TypedGeometry::MultiPolygon(polygons) => Self {
                data: false,
                multi_point: None,
                multi_line_string: None,
                multi_polygon: Some(polygons.clone()),
            },
        }
    }
}

impl TryFrom<TypedGeometryDbType> for TypedGeometry {
    type Error = Error;

    fn try_from(other: TypedGeometryDbType) -> Result<Self, Self::Error> {
        match other {
            TypedGeometryDbType {
                data: true,
                multi_point: None,
                multi_line_string: None,
                multi_polygon: None,
            } => Ok(TypedGeometry::Data(NoGeometry)),
            TypedGeometryDbType {
                data: false,
                multi_point: Some(multi_point),
                multi_line_string: None,
                multi_polygon: None,
            } => Ok(TypedGeometry::MultiPoint(multi_point)),
            TypedGeometryDbType {
                data: false,
                multi_point: None,
                multi_line_string: Some(multi_line_string),
                multi_polygon: None,
            } => Ok(TypedGeometry::MultiLineString(multi_line_string)),
            TypedGeometryDbType {
                data: false,
                multi_point: None,
                multi_line_string: None,
                multi_polygon: Some(multi_polygon),
            } => Ok(TypedGeometry::MultiPolygon(multi_polygon)),
            _ => Err(Error::UnexpectedInvalidDbTypeConversion),
        }
    }
}

#[derive(Debug, ToSql, FromSql)]
#[postgres(name = "OgrSourceDataset")]
pub struct OgrSourceDatasetDbType {
    pub file_name: String,
    pub layer_name: String,
    pub data_type: Option<VectorDataType>,
    pub time: OgrSourceDatasetTimeType,
    pub default_geometry: Option<TypedGeometry>,
    pub columns: Option<OgrSourceColumnSpec>,
    pub force_ogr_time_filter: bool,
    pub force_ogr_spatial_filter: bool,
    pub on_error: OgrSourceErrorSpec,
    pub sql_query: Option<String>,
    pub attribute_query: Option<String>,
    pub cache_ttl: CacheTtlSeconds,
}

impl From<&OgrSourceDataset> for OgrSourceDatasetDbType {
    fn from(other: &OgrSourceDataset) -> Self {
        Self {
            file_name: other.file_name.to_string_lossy().to_string(),
            layer_name: other.layer_name.clone(),
            data_type: other.data_type,
            time: other.time.clone(),
            default_geometry: other.default_geometry.clone(),
            columns: other.columns.clone(),
            force_ogr_time_filter: other.force_ogr_time_filter,
            force_ogr_spatial_filter: other.force_ogr_spatial_filter,
            on_error: other.on_error,
            sql_query: other.sql_query.clone(),
            attribute_query: other.attribute_query.clone(),
            cache_ttl: other.cache_ttl,
        }
    }
}

impl TryFrom<OgrSourceDatasetDbType> for OgrSourceDataset {
    type Error = Error;

    fn try_from(other: OgrSourceDatasetDbType) -> Result<Self, Self::Error> {
        Ok(Self {
            file_name: other.file_name.into(),
            layer_name: other.layer_name,
            data_type: other.data_type,
            time: other.time,
            default_geometry: other.default_geometry,
            columns: other.columns,
            force_ogr_time_filter: other.force_ogr_time_filter,
            force_ogr_spatial_filter: other.force_ogr_spatial_filter,
            on_error: other.on_error,
            sql_query: other.sql_query,
            attribute_query: other.attribute_query,
            cache_ttl: other.cache_ttl,
        })
    }
}

#[derive(Debug, ToSql, FromSql)]
#[postgres(name = "OgrMetaData")]
pub struct OgrMetaDataDbType {
    pub loading_info: OgrSourceDataset,
    pub result_descriptor: VectorResultDescriptor,
}

impl From<&OgrMetaData> for OgrMetaDataDbType {
    fn from(other: &OgrMetaData) -> Self {
        Self {
            loading_info: other.loading_info.clone(),
            result_descriptor: other.result_descriptor.clone(),
        }
    }
}

impl TryFrom<OgrMetaDataDbType> for OgrMetaData {
    type Error = Error;

    fn try_from(other: OgrMetaDataDbType) -> Result<Self, Self::Error> {
        Ok(Self {
            loading_info: other.loading_info,
            result_descriptor: other.result_descriptor,
            phantom: PhantomData,
        })
    }
}

#[derive(Debug, ToSql, FromSql)]
#[postgres(name = "MockMetaData")]
pub struct MockMetaDataDbType {
    pub loading_info: MockDatasetDataSourceLoadingInfo,
    pub result_descriptor: VectorResultDescriptor,
}

impl From<&MockMetaData> for MockMetaDataDbType {
    fn from(other: &MockMetaData) -> Self {
        Self {
            loading_info: other.loading_info.clone(),
            result_descriptor: other.result_descriptor.clone(),
        }
    }
}

impl TryFrom<MockMetaDataDbType> for MockMetaData {
    type Error = Error;

    fn try_from(other: MockMetaDataDbType) -> Result<Self, Self::Error> {
        Ok(Self {
            loading_info: other.loading_info,
            result_descriptor: other.result_descriptor,
            phantom: PhantomData,
        })
    }
}

#[derive(Debug, ToSql, FromSql)]
#[postgres(name = "GdalDatasetParameters")]
pub struct GdalDatasetParametersDbType {
    pub file_path: String,
    pub rasterband_channel: i64,
    pub geo_transform: GdalDatasetGeoTransform,
    pub width: i64,
    pub height: i64,
    pub file_not_found_handling: FileNotFoundHandling,
    pub no_data_value: Option<f64>,
    pub properties_mapping: Option<Vec<GdalMetadataMapping>>,
    pub gdal_open_options: Option<Vec<String>>,
    pub gdal_config_options: Option<Vec<GdalConfigOption>>,
    pub allow_alphaband_as_mask: bool,
}

impl From<&GdalDatasetParameters> for GdalDatasetParametersDbType {
    fn from(other: &GdalDatasetParameters) -> Self {
        Self {
            file_path: other.file_path.to_string_lossy().to_string(),
            rasterband_channel: other.rasterband_channel as i64,
            geo_transform: other.geo_transform,
            width: other.width as i64,
            height: other.height as i64,
            file_not_found_handling: other.file_not_found_handling,
            no_data_value: other.no_data_value,
            properties_mapping: other.properties_mapping.clone(),
            gdal_open_options: other.gdal_open_options.clone(),
            gdal_config_options: other.gdal_config_options.clone(),
            allow_alphaband_as_mask: other.allow_alphaband_as_mask,
        }
    }
}

impl TryFrom<GdalDatasetParametersDbType> for GdalDatasetParameters {
    type Error = Error;

    fn try_from(other: GdalDatasetParametersDbType) -> Result<Self, Self::Error> {
        Ok(Self {
            file_path: other.file_path.into(),
            rasterband_channel: other.rasterband_channel as usize,
            geo_transform: other.geo_transform,
            width: other.width as usize,
            height: other.height as usize,
            file_not_found_handling: other.file_not_found_handling,
            no_data_value: other.no_data_value,
            properties_mapping: other.properties_mapping,
            gdal_open_options: other.gdal_open_options,
            gdal_config_options: other.gdal_config_options,
            allow_alphaband_as_mask: other.allow_alphaband_as_mask,
        })
    }
}

#[derive(Debug, ToSql, FromSql, PartialEq)]
pub struct TextGdalSourceTimePlaceholderKeyValue {
    pub key: String,
    pub value: GdalSourceTimePlaceholder,
}

#[derive(Debug, ToSql, FromSql)]
#[postgres(name = "GdalMetaDataRegular")]
pub struct GdalMetaDataRegularDbType {
    pub result_descriptor: RasterResultDescriptor,
    pub params: GdalDatasetParameters,
    pub time_placeholders: Vec<TextGdalSourceTimePlaceholderKeyValue>,
    pub data_time: TimeInterval,
    pub step: TimeStep,
    pub cache_ttl: CacheTtlSeconds,
}

impl From<&GdalMetaDataRegular> for GdalMetaDataRegularDbType {
    fn from(other: &GdalMetaDataRegular) -> Self {
        Self {
            result_descriptor: other.result_descriptor.clone(),
            params: other.params.clone(),
            time_placeholders: other
                .time_placeholders
                .iter()
                .map(|(key, value)| TextGdalSourceTimePlaceholderKeyValue {
                    key: key.clone(),
                    value: value.clone(),
                })
                .collect(),
            data_time: other.data_time,
            step: other.step,
            cache_ttl: other.cache_ttl,
        }
    }
}

impl TryFrom<GdalMetaDataRegularDbType> for GdalMetaDataRegular {
    type Error = Error;

    fn try_from(other: GdalMetaDataRegularDbType) -> Result<Self, Self::Error> {
        Ok(Self {
            result_descriptor: other.result_descriptor,
            params: other.params,
            time_placeholders: other
                .time_placeholders
                .iter()
                .map(|item| (item.key.clone(), item.value.clone()))
                .collect(),
            data_time: other.data_time,
            step: other.step,
            cache_ttl: other.cache_ttl,
        })
    }
}

#[derive(Debug, ToSql, FromSql)]
#[postgres(name = "GdalMetadataNetCdfCf")]
pub struct GdalMetadataNetCdfCfDbType {
    pub result_descriptor: RasterResultDescriptor,
    pub params: GdalDatasetParameters,
    pub start: TimeInstance,
    pub end: TimeInstance,
    pub step: TimeStep,
    pub band_offset: i64,
    pub cache_ttl: CacheTtlSeconds,
}

impl From<&GdalMetadataNetCdfCf> for GdalMetadataNetCdfCfDbType {
    fn from(other: &GdalMetadataNetCdfCf) -> Self {
        Self {
            result_descriptor: other.result_descriptor.clone(),
            params: other.params.clone(),
            start: other.start,
            end: other.end,
            step: other.step,
            band_offset: other.band_offset as i64,
            cache_ttl: other.cache_ttl,
        }
    }
}

impl TryFrom<GdalMetadataNetCdfCfDbType> for GdalMetadataNetCdfCf {
    type Error = Error;

    fn try_from(other: GdalMetadataNetCdfCfDbType) -> Result<Self, Self::Error> {
        Ok(Self {
            result_descriptor: other.result_descriptor,
            params: other.params,
            start: other.start,
            end: other.end,
            step: other.step,
            band_offset: other.band_offset as usize,
            cache_ttl: other.cache_ttl,
        })
    }
}

#[derive(Debug, ToSql, FromSql)]
#[postgres(name = "MetaDataDefinition")]
pub struct MetaDataDefinitionDbType {
    mock_meta_data: Option<MockMetaData>,
    ogr_meta_data: Option<OgrMetaData>,
    gdal_meta_data_regular: Option<GdalMetaDataRegular>,
    gdal_static: Option<GdalMetaDataStatic>,
    gdal_metadata_net_cdf_cf: Option<GdalMetadataNetCdfCf>,
    gdal_meta_data_list: Option<GdalMetaDataList>,
}

impl From<&MetaDataDefinition> for MetaDataDefinitionDbType {
    fn from(other: &MetaDataDefinition) -> Self {
        match other {
            MetaDataDefinition::MockMetaData(meta_data) => Self {
                mock_meta_data: Some(meta_data.clone().into()),
                ogr_meta_data: None,
                gdal_meta_data_regular: None,
                gdal_static: None,
                gdal_metadata_net_cdf_cf: None,
                gdal_meta_data_list: None,
            },
            MetaDataDefinition::OgrMetaData(meta_data) => Self {
                mock_meta_data: None,
                ogr_meta_data: Some(meta_data.clone().into()),
                gdal_meta_data_regular: None,
                gdal_static: None,
                gdal_metadata_net_cdf_cf: None,
                gdal_meta_data_list: None,
            },
            MetaDataDefinition::GdalMetaDataRegular(meta_data) => Self {
                mock_meta_data: None,
                ogr_meta_data: None,
                gdal_meta_data_regular: Some(meta_data.clone().into()),
                gdal_static: None,
                gdal_metadata_net_cdf_cf: None,
                gdal_meta_data_list: None,
            },
            MetaDataDefinition::GdalStatic(meta_data) => Self {
                mock_meta_data: None,
                ogr_meta_data: None,
                gdal_meta_data_regular: None,
                gdal_static: Some(meta_data.clone().into()),
                gdal_metadata_net_cdf_cf: None,
                gdal_meta_data_list: None,
            },
            MetaDataDefinition::GdalMetadataNetCdfCf(meta_data) => Self {
                mock_meta_data: None,
                ogr_meta_data: None,
                gdal_meta_data_regular: None,
                gdal_static: None,
                gdal_metadata_net_cdf_cf: Some(meta_data.clone().into()),
                gdal_meta_data_list: None,
            },
            MetaDataDefinition::GdalMetaDataList(meta_data) => Self {
                mock_meta_data: None,
                ogr_meta_data: None,
                gdal_meta_data_regular: None,
                gdal_static: None,
                gdal_metadata_net_cdf_cf: None,
                gdal_meta_data_list: Some(meta_data.clone().into()),
            },
        }
    }
}

impl TryFrom<MetaDataDefinitionDbType> for MetaDataDefinition {
    type Error = Error;

    fn try_from(other: MetaDataDefinitionDbType) -> Result<Self, Self::Error> {
        match other {
            MetaDataDefinitionDbType {
                mock_meta_data: Some(meta_data),
                ogr_meta_data: None,
                gdal_meta_data_regular: None,
                gdal_static: None,
                gdal_metadata_net_cdf_cf: None,
                gdal_meta_data_list: None,
            } => Ok(MetaDataDefinition::MockMetaData(meta_data.into())),
            MetaDataDefinitionDbType {
                mock_meta_data: None,
                ogr_meta_data: Some(meta_data),
                gdal_meta_data_regular: None,
                gdal_static: None,
                gdal_metadata_net_cdf_cf: None,
                gdal_meta_data_list: None,
            } => Ok(MetaDataDefinition::OgrMetaData(meta_data.into())),
            MetaDataDefinitionDbType {
                mock_meta_data: None,
                ogr_meta_data: None,
                gdal_meta_data_regular: Some(meta_data),
                gdal_static: None,
                gdal_metadata_net_cdf_cf: None,
                gdal_meta_data_list: None,
            } => Ok(MetaDataDefinition::GdalMetaDataRegular(meta_data.into())),
            MetaDataDefinitionDbType {
                mock_meta_data: None,
                ogr_meta_data: None,
                gdal_meta_data_regular: None,
                gdal_static: Some(meta_data),
                gdal_metadata_net_cdf_cf: None,
                gdal_meta_data_list: None,
            } => Ok(MetaDataDefinition::GdalStatic(meta_data.into())),
            MetaDataDefinitionDbType {
                mock_meta_data: None,
                ogr_meta_data: None,
                gdal_meta_data_regular: None,
                gdal_static: None,
                gdal_metadata_net_cdf_cf: Some(meta_data),
                gdal_meta_data_list: None,
            } => Ok(MetaDataDefinition::GdalMetadataNetCdfCf(meta_data.into())),
            MetaDataDefinitionDbType {
                mock_meta_data: None,
                ogr_meta_data: None,
                gdal_meta_data_regular: None,
                gdal_static: None,
                gdal_metadata_net_cdf_cf: None,
                gdal_meta_data_list: Some(meta_data),
            } => Ok(MetaDataDefinition::GdalMetaDataList(meta_data.into())),
            _ => Err(Error::UnexpectedInvalidDbTypeConversion),
        }
    }
}

#[derive(Debug, ToSql, FromSql)]
#[postgres(name = "DatabaseConnectionConfig")]
pub struct DatabaseConnectionConfigDbType {
    pub host: String,
    pub port: i32,
    pub database: String,
    pub schema: String,
    pub user: String,
    pub password: String,
}

impl From<&DatabaseConnectionConfig> for DatabaseConnectionConfigDbType {
    fn from(other: &DatabaseConnectionConfig) -> Self {
        Self {
            host: other.host.clone(),
            port: i32::from(other.port),
            database: other.database.clone(),
            schema: other.schema.clone(),
            user: other.user.clone(),
            password: other.password.clone(),
        }
    }
}

impl TryFrom<DatabaseConnectionConfigDbType> for DatabaseConnectionConfig {
    type Error = Error;

    fn try_from(other: DatabaseConnectionConfigDbType) -> Result<Self, Self::Error> {
        Ok(Self {
            host: other.host,
            port: other.port as u16,
            database: other.database,
            schema: other.schema,
            user: other.user,
            password: other.password,
        })
    }
}

#[derive(Debug, ToSql, FromSql)]
#[postgres(name = "GfbioCollectionsDataProviderDefinition")]
pub struct GfbioCollectionsDataProviderDefinitionDbType {
    pub name: String,
    pub collection_api_url: String,
    pub collection_api_auth_token: String,
    pub abcd_db_config: DatabaseConnectionConfig,
    pub pangaea_url: String,
    pub cache_ttl: CacheTtlSeconds,
}

impl From<&GfbioCollectionsDataProviderDefinition>
    for GfbioCollectionsDataProviderDefinitionDbType
{
    fn from(other: &GfbioCollectionsDataProviderDefinition) -> Self {
        Self {
            name: other.name.clone(),
            collection_api_url: other.collection_api_url.clone().into(),
            collection_api_auth_token: other.collection_api_auth_token.clone(),
            abcd_db_config: other.abcd_db_config.clone(),
            pangaea_url: other.pangaea_url.clone().into(),
            cache_ttl: other.cache_ttl,
        }
    }
}

impl TryFrom<GfbioCollectionsDataProviderDefinitionDbType>
    for GfbioCollectionsDataProviderDefinition
{
    type Error = Error;

    fn try_from(other: GfbioCollectionsDataProviderDefinitionDbType) -> Result<Self, Self::Error> {
        Ok(Self {
            name: other.name,
            collection_api_url: other.collection_api_url.as_str().try_into()?,
            collection_api_auth_token: other.collection_api_auth_token,
            abcd_db_config: other.abcd_db_config,
            pangaea_url: other.pangaea_url.as_str().try_into()?,
            cache_ttl: other.cache_ttl,
        })
    }
}

#[derive(Debug, ToSql, FromSql)]
#[postgres(name = "EbvPortalDataProviderDefinition")]
pub struct EbvPortalDataProviderDefinitionDbType {
    pub name: String,
    pub path: String,
    pub base_url: String,
    pub overviews: String,
    pub cache_ttl: CacheTtlSeconds,
}

impl From<&EbvPortalDataProviderDefinition> for EbvPortalDataProviderDefinitionDbType {
    fn from(other: &EbvPortalDataProviderDefinition) -> Self {
        Self {
            name: other.name.clone(),
            path: other.path.to_string_lossy().to_string(),
            base_url: other.base_url.clone().into(),
            overviews: other.overviews.to_string_lossy().to_string(),
            cache_ttl: other.cache_ttl,
        }
    }
}

impl TryFrom<EbvPortalDataProviderDefinitionDbType> for EbvPortalDataProviderDefinition {
    type Error = Error;

    fn try_from(other: EbvPortalDataProviderDefinitionDbType) -> Result<Self, Self::Error> {
        Ok(Self {
            name: other.name,
            path: other.path.into(),
            base_url: other.base_url.as_str().try_into()?,
            overviews: other.overviews.into(),
            cache_ttl: other.cache_ttl,
        })
    }
}

#[derive(Debug, ToSql, FromSql)]
#[postgres(name = "NetCdfCfDataProviderDefinition")]
pub struct NetCdfCfDataProviderDefinitionDbType {
    pub name: String,
    pub path: String,
    pub overviews: String,
    pub cache_ttl: CacheTtlSeconds,
}

impl From<&NetCdfCfDataProviderDefinition> for NetCdfCfDataProviderDefinitionDbType {
    fn from(other: &NetCdfCfDataProviderDefinition) -> Self {
        Self {
            name: other.name.clone(),
            path: other.path.to_string_lossy().to_string(),
            overviews: other.overviews.to_string_lossy().to_string(),
            cache_ttl: other.cache_ttl,
        }
    }
}

impl TryFrom<NetCdfCfDataProviderDefinitionDbType> for NetCdfCfDataProviderDefinition {
    type Error = Error;

    fn try_from(other: NetCdfCfDataProviderDefinitionDbType) -> Result<Self, Self::Error> {
        Ok(Self {
            name: other.name,
            path: other.path.into(),
            overviews: other.overviews.into(),
            cache_ttl: other.cache_ttl,
        })
    }
}

#[derive(Debug, ToSql, FromSql)]
#[postgres(name = "PangaeaDataProviderDefinition")]
pub struct PangaeaDataProviderDefinitionDbType {
    pub name: String,
    pub base_url: String,
    pub cache_ttl: CacheTtlSeconds,
}

impl From<&PangaeaDataProviderDefinition> for PangaeaDataProviderDefinitionDbType {
    fn from(other: &PangaeaDataProviderDefinition) -> Self {
        Self {
            name: other.name.clone(),
            base_url: other.base_url.clone().into(),
            cache_ttl: other.cache_ttl,
        }
    }
}

impl TryFrom<PangaeaDataProviderDefinitionDbType> for PangaeaDataProviderDefinition {
    type Error = Error;

    fn try_from(other: PangaeaDataProviderDefinitionDbType) -> Result<Self, Self::Error> {
        Ok(Self {
            name: other.name,
            base_url: other.base_url.as_str().try_into()?,
            cache_ttl: other.cache_ttl,
        })
    }
}

#[derive(Debug, ToSql, FromSql)]
#[postgres(name = "EdrVectorSpec")]
pub struct EdrVectorSpecDbType {
    x: String,
    y: Option<String>,
    time: String,
}

impl From<&EdrVectorSpec> for EdrVectorSpecDbType {
    fn from(other: &EdrVectorSpec) -> Self {
        Self {
            x: other.x.clone(),
            y: other.y.clone(),
            time: other.time.clone(),
        }
    }
}

impl TryFrom<EdrVectorSpecDbType> for EdrVectorSpec {
    type Error = Error;

    fn try_from(other: EdrVectorSpecDbType) -> Result<Self, Self::Error> {
        Ok(Self {
            x: other.x,
            y: other.y,
            time: other.time,
        })
    }
}

#[derive(Debug, ToSql, FromSql)]
#[postgres(name = "EdrDataProviderDefinition")]
pub struct EdrDataProviderDefinitionDbType {
    name: String,
    id: DataProviderId,
    base_url: String,
    vector_spec: Option<EdrVectorSpec>,
    cache_ttl: CacheTtlSeconds,
    /// List of vertical reference systems with a discrete scale
    discrete_vrs: Vec<String>,
    provenance: Option<Vec<Provenance>>,
}

impl From<&EdrDataProviderDefinition> for EdrDataProviderDefinitionDbType {
    fn from(other: &EdrDataProviderDefinition) -> Self {
        Self {
            name: other.name.clone(),
            id: other.id,
            base_url: other.base_url.clone().into(),
            vector_spec: other.vector_spec.clone(),
            cache_ttl: other.cache_ttl,
            discrete_vrs: other.discrete_vrs.clone(),
            provenance: other.provenance.clone(),
        }
    }
}

impl TryFrom<EdrDataProviderDefinitionDbType> for EdrDataProviderDefinition {
    type Error = Error;

    fn try_from(other: EdrDataProviderDefinitionDbType) -> Result<Self, Self::Error> {
        Ok(Self {
            name: other.name,
            id: other.id,
            base_url: other.base_url.as_str().try_into()?,
            vector_spec: other.vector_spec,
            cache_ttl: other.cache_ttl,
            discrete_vrs: other.discrete_vrs,
            provenance: other.provenance,
        })
    }
}

#[derive(Debug, ToSql, FromSql)]
#[postgres(name = "DataProviderDefinition")]
pub struct TypedDataProviderDefinitionDbType {
    aruna_data_provider_definition: Option<ArunaDataProviderDefinition>,
    gbif_data_provider_definition: Option<GbifDataProviderDefinition>,
    gfbio_abcd_data_provider_definition: Option<GfbioAbcdDataProviderDefinition>,
    gfbio_collections_data_provider_definition: Option<GfbioCollectionsDataProviderDefinition>,
    ebv_portal_data_provider_definition: Option<EbvPortalDataProviderDefinition>,
    net_cdf_cf_data_provider_definition: Option<NetCdfCfDataProviderDefinition>,
    pangaea_data_provider_definition: Option<PangaeaDataProviderDefinition>,
    edr_data_provider_definition: Option<EdrDataProviderDefinition>,
}

impl From<&TypedDataProviderDefinition> for TypedDataProviderDefinitionDbType {
    fn from(other: &TypedDataProviderDefinition) -> Self {
        match other {
            TypedDataProviderDefinition::ArunaDataProviderDefinition(data_provider_definition) => {
                Self {
                    aruna_data_provider_definition: Some(data_provider_definition.clone()),
                    gbif_data_provider_definition: None,
                    gfbio_abcd_data_provider_definition: None,
                    gfbio_collections_data_provider_definition: None,
                    ebv_portal_data_provider_definition: None,
                    net_cdf_cf_data_provider_definition: None,
                    pangaea_data_provider_definition: None,
                    edr_data_provider_definition: None,
                }
            }
            TypedDataProviderDefinition::GbifDataProviderDefinition(data_provider_definition) => {
                Self {
                    aruna_data_provider_definition: None,
                    gbif_data_provider_definition: Some(data_provider_definition.clone()),
                    gfbio_abcd_data_provider_definition: None,
                    gfbio_collections_data_provider_definition: None,
                    ebv_portal_data_provider_definition: None,
                    net_cdf_cf_data_provider_definition: None,
                    pangaea_data_provider_definition: None,
                    edr_data_provider_definition: None,
                }
            }
            TypedDataProviderDefinition::GfbioAbcdDataProviderDefinition(
                data_provider_definition,
            ) => Self {
                aruna_data_provider_definition: None,
                gbif_data_provider_definition: None,
                gfbio_abcd_data_provider_definition: Some(data_provider_definition.clone()),
                gfbio_collections_data_provider_definition: None,
                ebv_portal_data_provider_definition: None,
                net_cdf_cf_data_provider_definition: None,
                pangaea_data_provider_definition: None,
                edr_data_provider_definition: None,
            },
            TypedDataProviderDefinition::GfbioCollectionsDataProviderDefinition(
                data_provider_definition,
            ) => Self {
                aruna_data_provider_definition: None,
                gbif_data_provider_definition: None,
                gfbio_abcd_data_provider_definition: None,
                gfbio_collections_data_provider_definition: Some(data_provider_definition.clone()),
                ebv_portal_data_provider_definition: None,
                net_cdf_cf_data_provider_definition: None,
                pangaea_data_provider_definition: None,
                edr_data_provider_definition: None,
            },
            TypedDataProviderDefinition::EbvPortalDataProviderDefinition(
                data_provider_definition,
            ) => Self {
                aruna_data_provider_definition: None,
                gbif_data_provider_definition: None,
                gfbio_abcd_data_provider_definition: None,
                gfbio_collections_data_provider_definition: None,
                ebv_portal_data_provider_definition: Some(data_provider_definition.clone()),
                net_cdf_cf_data_provider_definition: None,
                pangaea_data_provider_definition: None,
                edr_data_provider_definition: None,
            },
            TypedDataProviderDefinition::NetCdfCfDataProviderDefinition(
                data_provider_definition,
            ) => Self {
                aruna_data_provider_definition: None,
                gbif_data_provider_definition: None,
                gfbio_abcd_data_provider_definition: None,
                gfbio_collections_data_provider_definition: None,
                ebv_portal_data_provider_definition: None,
                net_cdf_cf_data_provider_definition: Some(data_provider_definition.clone()),
                pangaea_data_provider_definition: None,
                edr_data_provider_definition: None,
            },
            TypedDataProviderDefinition::PangaeaDataProviderDefinition(
                data_provider_definition,
            ) => Self {
                aruna_data_provider_definition: None,
                gbif_data_provider_definition: None,
                gfbio_abcd_data_provider_definition: None,
                gfbio_collections_data_provider_definition: None,
                ebv_portal_data_provider_definition: None,
                net_cdf_cf_data_provider_definition: None,
                pangaea_data_provider_definition: Some(data_provider_definition.clone()),
                edr_data_provider_definition: None,
            },
            TypedDataProviderDefinition::EdrDataProviderDefinition(data_provider_definition) => {
                Self {
                    aruna_data_provider_definition: None,
                    gbif_data_provider_definition: None,
                    gfbio_abcd_data_provider_definition: None,
                    gfbio_collections_data_provider_definition: None,
                    ebv_portal_data_provider_definition: None,
                    net_cdf_cf_data_provider_definition: None,
                    pangaea_data_provider_definition: None,
                    edr_data_provider_definition: Some(data_provider_definition.clone()),
                }
            }
        }
    }
}

impl TryFrom<TypedDataProviderDefinitionDbType> for TypedDataProviderDefinition {
    type Error = Error;

    #[allow(clippy::too_many_lines)]
    fn try_from(result_descriptor: TypedDataProviderDefinitionDbType) -> Result<Self, Self::Error> {
        match result_descriptor {
            TypedDataProviderDefinitionDbType {
                aruna_data_provider_definition: Some(data_provider_definition),
                gbif_data_provider_definition: None,
                gfbio_abcd_data_provider_definition: None,
                gfbio_collections_data_provider_definition: None,
                ebv_portal_data_provider_definition: None,
                net_cdf_cf_data_provider_definition: None,
                pangaea_data_provider_definition: None,
                edr_data_provider_definition: None,
            } => Ok(TypedDataProviderDefinition::ArunaDataProviderDefinition(
                data_provider_definition,
            )),
            TypedDataProviderDefinitionDbType {
                aruna_data_provider_definition: None,
                gbif_data_provider_definition: Some(data_provider_definition),
                gfbio_abcd_data_provider_definition: None,
                gfbio_collections_data_provider_definition: None,
                ebv_portal_data_provider_definition: None,
                net_cdf_cf_data_provider_definition: None,
                pangaea_data_provider_definition: None,
                edr_data_provider_definition: None,
            } => Ok(TypedDataProviderDefinition::GbifDataProviderDefinition(
                data_provider_definition,
            )),
            TypedDataProviderDefinitionDbType {
                aruna_data_provider_definition: None,
                gbif_data_provider_definition: None,
                gfbio_abcd_data_provider_definition: Some(data_provider_definition),
                gfbio_collections_data_provider_definition: None,
                ebv_portal_data_provider_definition: None,
                net_cdf_cf_data_provider_definition: None,
                pangaea_data_provider_definition: None,
                edr_data_provider_definition: None,
            } => Ok(
                TypedDataProviderDefinition::GfbioAbcdDataProviderDefinition(
                    data_provider_definition,
                ),
            ),
            TypedDataProviderDefinitionDbType {
                aruna_data_provider_definition: None,
                gbif_data_provider_definition: None,
                gfbio_abcd_data_provider_definition: None,
                gfbio_collections_data_provider_definition: Some(data_provider_definition),
                ebv_portal_data_provider_definition: None,
                net_cdf_cf_data_provider_definition: None,
                pangaea_data_provider_definition: None,
                edr_data_provider_definition: None,
            } => Ok(
                TypedDataProviderDefinition::GfbioCollectionsDataProviderDefinition(
                    data_provider_definition,
                ),
            ),
            TypedDataProviderDefinitionDbType {
                aruna_data_provider_definition: None,
                gbif_data_provider_definition: None,
                gfbio_abcd_data_provider_definition: None,
                gfbio_collections_data_provider_definition: None,
                ebv_portal_data_provider_definition: Some(data_provider_definition),
                net_cdf_cf_data_provider_definition: None,
                pangaea_data_provider_definition: None,
                edr_data_provider_definition: None,
            } => Ok(
                TypedDataProviderDefinition::EbvPortalDataProviderDefinition(
                    data_provider_definition,
                ),
            ),
            TypedDataProviderDefinitionDbType {
                aruna_data_provider_definition: None,
                gbif_data_provider_definition: None,
                gfbio_abcd_data_provider_definition: None,
                gfbio_collections_data_provider_definition: None,
                ebv_portal_data_provider_definition: None,
                net_cdf_cf_data_provider_definition: Some(data_provider_definition),
                pangaea_data_provider_definition: None,
                edr_data_provider_definition: None,
            } => Ok(TypedDataProviderDefinition::NetCdfCfDataProviderDefinition(
                data_provider_definition,
            )),
            TypedDataProviderDefinitionDbType {
                aruna_data_provider_definition: None,
                gbif_data_provider_definition: None,
                gfbio_abcd_data_provider_definition: None,
                gfbio_collections_data_provider_definition: None,
                ebv_portal_data_provider_definition: None,
                net_cdf_cf_data_provider_definition: None,
                pangaea_data_provider_definition: Some(data_provider_definition),
                edr_data_provider_definition: None,
            } => Ok(TypedDataProviderDefinition::PangaeaDataProviderDefinition(
                data_provider_definition,
            )),
            TypedDataProviderDefinitionDbType {
                aruna_data_provider_definition: None,
                gbif_data_provider_definition: None,
                gfbio_abcd_data_provider_definition: None,
                gfbio_collections_data_provider_definition: None,
                ebv_portal_data_provider_definition: None,
                net_cdf_cf_data_provider_definition: None,
                pangaea_data_provider_definition: None,
                edr_data_provider_definition: Some(data_provider_definition),
            } => Ok(TypedDataProviderDefinition::EdrDataProviderDefinition(
                data_provider_definition,
            )),
            _ => Err(Error::UnexpectedInvalidDbTypeConversion),
        }
    }
}

/// A macro for quickly implementing `FromSql` and `ToSql` for `$RustType` if there is a `From` and `Into`
/// implementation for another type `$DbType` that already implements it.
///
/// # Usage
///
/// ```rust,ignore
/// delegate_from_to_sql!($RustType, $DbType)
/// ```
///
#[macro_export]
macro_rules! delegate_from_to_sql {
    ( $RustType:ty, $DbType:ty ) => {
        impl ToSql for $RustType {
            fn to_sql(
                &self,
                ty: &postgres_types::Type,
                w: &mut bytes::BytesMut,
            ) -> Result<postgres_types::IsNull, Box<dyn std::error::Error + Sync + Send>> {
                <$DbType as ToSql>::to_sql(&self.into(), ty, w)
            }

            fn accepts(ty: &postgres_types::Type) -> bool {
                <$DbType as FromSql>::accepts(ty)
            }

            postgres_types::to_sql_checked!();
        }

        impl<'a> FromSql<'a> for $RustType {
            fn from_sql(
                ty: &postgres_types::Type,
                raw: &'a [u8],
            ) -> Result<$RustType, Box<dyn std::error::Error + Sync + Send>> {
                Ok(<$DbType as FromSql>::from_sql(ty, raw)?.try_into()?)
            }

            fn accepts(ty: &postgres_types::Type) -> bool {
                <$DbType as FromSql>::accepts(ty)
            }
        }
    };
}

delegate_from_to_sql!(Colorizer, ColorizerDbType);
delegate_from_to_sql!(ColorParam, ColorParamDbType);
delegate_from_to_sql!(DatabaseConnectionConfig, DatabaseConnectionConfigDbType);
delegate_from_to_sql!(DefaultColors, DefaultColorsDbType);
delegate_from_to_sql!(
    EbvPortalDataProviderDefinition,
    EbvPortalDataProviderDefinitionDbType
);
delegate_from_to_sql!(FormatSpecifics, FormatSpecificsDbType);
delegate_from_to_sql!(GdalDatasetParameters, GdalDatasetParametersDbType);
delegate_from_to_sql!(GdalMetadataNetCdfCf, GdalMetadataNetCdfCfDbType);
delegate_from_to_sql!(GdalMetaDataRegular, GdalMetaDataRegularDbType);
delegate_from_to_sql!(
    GfbioCollectionsDataProviderDefinition,
    GfbioCollectionsDataProviderDefinitionDbType
);
delegate_from_to_sql!(Measurement, MeasurementDbType);
delegate_from_to_sql!(MetaDataDefinition, MetaDataDefinitionDbType);
delegate_from_to_sql!(MockMetaData, MockMetaDataDbType);
delegate_from_to_sql!(NumberParam, NumberParamDbType);
delegate_from_to_sql!(
    NetCdfCfDataProviderDefinition,
    NetCdfCfDataProviderDefinitionDbType
);
delegate_from_to_sql!(OgrMetaData, OgrMetaDataDbType);
delegate_from_to_sql!(OgrSourceColumnSpec, OgrSourceColumnSpecDbType);
delegate_from_to_sql!(OgrSourceDataset, OgrSourceDatasetDbType);
delegate_from_to_sql!(OgrSourceDatasetTimeType, OgrSourceDatasetTimeTypeDbType);
delegate_from_to_sql!(OgrSourceDurationSpec, OgrSourceDurationSpecDbType);
delegate_from_to_sql!(OgrSourceTimeFormat, OgrSourceTimeFormatDbType);
delegate_from_to_sql!(
    PangaeaDataProviderDefinition,
    PangaeaDataProviderDefinitionDbType
);
delegate_from_to_sql!(EdrDataProviderDefinition, EdrDataProviderDefinitionDbType);
delegate_from_to_sql!(EdrVectorSpec, EdrVectorSpecDbType);
delegate_from_to_sql!(Symbology, SymbologyDbType);
delegate_from_to_sql!(VectorResultDescriptor, VectorResultDescriptorDbType);
delegate_from_to_sql!(
    TypedDataProviderDefinition,
    TypedDataProviderDefinitionDbType
);
delegate_from_to_sql!(TypedGeometry, TypedGeometryDbType);
delegate_from_to_sql!(TypedResultDescriptor, TypedResultDescriptorDbType);

geo-engine / geoengine / 6147514757

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous