/*
 * Copyright (C) 2022 StApps
 * This program is free software: you can redistribute it and/or modify it
 * under the terms of the GNU General Public License as published by the Free
 * Software Foundation, version 3.
 *
 * This program is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
 * more details.
 *
 * You should have received a copy of the GNU General Public License along with
 * this program. If not, see <https://www.gnu.org/licenses/>.
 */
import {SCGeoFilter} from '@openstapps/core';
import {QueryDslSpecificQueryContainer} from '../../types/util.js';

/**
 * Converts a geo filter to elasticsearch syntax
 * @param filter A search filter for the retrieval of the data
 */
export function buildGeoFilter(filter: SCGeoFilter): QueryDslSpecificQueryContainer<'bool'> {
  return {
    bool: {
      should: [
        {
          geo_shape: {
            ignore_unmapped: true,
            [`${filter.arguments.field}.polygon`]: {
              shape: filter.arguments.shape,
              relation: filter.arguments.spatialRelation,
            },
          },
        } satisfies QueryDslSpecificQueryContainer<'geo_shape'>,
        {
          geo_shape: {
            ignore_unmapped: true,
            [`${filter.arguments.field}.point`]: {
              shape: filter.arguments.shape,
              relation: filter.arguments.spatialRelation,
            },
          },
        } satisfies QueryDslSpecificQueryContainer<'geo_shape'>,
      ],
    },
  };
}
