/**
 * Copyright 2023 Google LLC
 *
 * Licensed under the Apache License, Version 2.0 (the "License"); you may not
 * use this file except in compliance with the License. You may obtain a copy of
 * the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
 * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
 * License for the specific language governing permissions and limitations under
 * the License.
 */

import {Marker, MarkerCollection, MarkerOptions} from '@googlemaps/marker';

export type PlaceMarkerOptions = {
  place?: google.maps.places.PlaceResult;
} & MarkerOptions<google.maps.places.PlaceResult>;

/**
 * A special marker class for data from the Google Maps Places API. Uses the
 * recommended colors and icons from a place result.
 */
export class PlaceMarker extends Marker<google.maps.places.PlaceResult> {
  private place_: google.maps.places.PlaceResult | null = null;

  constructor(
    options: PlaceMarkerOptions,
    data?: google.maps.places.PlaceResult
  ) {
    if (options.place) {
      data = options.place;
    }

    super(options, data);

    this.attributeDefaults_ = {
      position: ({data}) => data?.geometry?.location?.toJSON(),
      color: ({data}) => data?.icon_background_color as string,
      glyphColor: () => '#ffff',
      glyph: ({data}) =>
        data?.icon_mask_base_uri && new URL(`${data?.icon_mask_base_uri}.svg`)
    };
  }

  get place(): google.maps.places.PlaceResult | null {
    return this.place_;
  }

  set place(place: google.maps.places.PlaceResult | null) {
    this.place_ = place;
    if (place) {
      this.setData(place);
    }
  }
}

/**
 * An extended MarkerCollection that can directly be used with results from the
 * PlacesService.
 *
 * @example
 *   const markers = new PlaceMarkerCollection({map});
 *   const placesService = new google.maps.places.PlacesService(map);
 *
 *   const searchQuery = {keyword: 'coffee', radius: 2000};
 *
 *   placesService.nearbySearch(searchQuery, (result, status) => {
 *     if (status !== 'OK' || !result) return;
 *
 *     markers.setData(result);
 *   });
 */
export class PlaceMarkerCollection extends MarkerCollection<google.maps.places.PlaceResult> {
  protected createMarker(
    options: MarkerOptions<google.maps.places.PlaceResult>,
    data: google.maps.places.PlaceResult
  ): Marker<google.maps.places.PlaceResult> {
    return new PlaceMarker(options, data);
  }
}