@tai-kun/surrealdb/src/data-types/standard/geometry-polygon.ts

The SurrealDB SDK for JavaScript

import {
  type GeoJsonPolygon,
  GeometryPolygonBase as Base,
  type GeometryPolygonSource as GeometryPolygonSourceBase,
  type GeometryPolygonTypes as GeometryPolygonTypesBase,
} from "@tai-kun/surrealdb/encodable-datatypes";
import { type Coord, isGeometryPolygon, map } from "../_internals/geometry";
import {
  GeometryLine,
  type GeometryLineBase,
  type GeometryLineTypes,
} from "./geometry-line";
import type { GeometryPointBase, GeometryPointTypes } from "./geometry-point";

type PointBase = new(
  source: any,
) => GeometryPointBase<GeometryPointTypes<Coord>>;

type LineBase = new(
  source: any,
) => GeometryLineBase<GeometryLineTypes<PointBase>>;

export type GeometryPolygonTypes<TLine extends LineBase = LineBase> =
  GeometryPolygonTypesBase<TLine>;

export type GeometryPolygonSource<
  TTypes extends GeometryPolygonTypes = GeometryPolygonTypes,
> = GeometryPolygonSourceBase<TTypes>;

export type { GeoJsonPolygon };

export class GeometryPolygonBase<
  TTypes extends GeometryPolygonTypes = GeometryPolygonTypes,
> extends Base<TTypes> {
  // @ts-expect-error readonly を外すだけ。
  override polygon: [
    InstanceType<TTypes["Line"]>,
    ...InstanceType<TTypes["Line"]>[],
  ];

  override get coordinates(): [
    InstanceType<TTypes["Line"]>["coordinates"],
    ...InstanceType<TTypes["Line"]>["coordinates"][],
  ] {
    return map(this.polygon, l => l.coordinates);
  }

  override set coordinates(source: GeometryPolygonSource<TTypes>) {
    this.polygon = map(
      source,
      (l: any) =>
        (l instanceof this.types.Line
          ? l
          : new this.types.Line(l)) as InstanceType<TTypes["Line"]>,
    );
  }

  get exteriorRing(): InstanceType<TTypes["Line"]> {
    return this.polygon[0];
  }

  set exteriorRing(v: InstanceType<TTypes["Line"]>) {
    this.polygon = [v, ...this.interiorRings];
  }

  get interiorRings(): InstanceType<TTypes["Line"]>[] {
    return this.polygon.slice(1);
  }

  set interiorRings(v: InstanceType<TTypes["Line"]>[]) {
    this.polygon = [this.exteriorRing, ...v];
  }

  clone(): this {
    const This = this.constructor as typeof GeometryPolygonBase;

    return new This(map(this.polygon, l => l.clone()), this.types) as this;
  }

  equals(other: unknown): boolean {
    return isGeometryPolygon<GeometryPolygonBase<GeometryPolygonTypes>>(other)
      && other.polygon.length === this.polygon.length
      && other.polygon.every((l, i) => this.polygon[i]!.equals(l));
  }
}

/**
 * [API Reference](https://tai-kun.github.io/surrealdb.js/v2/api/data/geometry-polygon)
 */
export class GeometryPolygon
  extends GeometryPolygonBase<GeometryPolygonTypes<typeof GeometryLine>>
{
  static readonly Line = GeometryLine;

  constructor(source: GeometryPolygonSource<typeof GeometryPolygon>) {
    super(source, GeometryPolygon);
  }
}