import { Filter, FindOptions, MongoProvider } from './MongoProvider';
import {
  asJson,
  Condition,
  Gateway,
  Id,
  ifDefined,
  isDefined,
  isPresent,
  Json,
  JsonValue,
  List,
  LogicalCondition,
  Optional,
  PageList,
  toArray,
} from '@thisisagile/easy';
import { Collection } from './Collection';
import { stages } from './Stages';

export class MongoGateway implements Gateway<FindOptions> {
  constructor(readonly collection: Collection, readonly provider: MongoProvider = collection.provider) {}

  all(options?: FindOptions): Promise<PageList<Json>> {
    return this.provider.all(options).then(js => js.map(j => this.collection.in(j)));
  }

  byId(id: Id): Promise<Json | undefined> {
    return this.provider.byId(id).then(j => ifDefined(j, this.collection.in(j)));
  }

  by(key: string, value: JsonValue, options?: FindOptions): Promise<PageList<Json>> {
    return this.provider.by(key, value, options).then(js => js.map(j => this.collection.in(j)));
  }

  byIds(...ids: Id[]): Promise<List<Json>> {
    return this.find(this.collection.id.isIn(...ids));
  }

  find(q: JsonValue | Condition | LogicalCondition, options?: FindOptions): Promise<PageList<Json>> {
    return this.provider.find(asJson(q), options).then(js => js.map(j => this.collection.in(j)));
  }

  search(q: JsonValue, options?: FindOptions): Promise<PageList<Json>> {
    return this.find(this.collection.google(q), options);
  }

  filter(options?: FindOptions): Promise<PageList<Json>> {
    return this.all(options);
  }

  exists(id: Id): Promise<boolean> {
    return this.provider.byId(id).then(i => isDefined(i));
  }

  aggregate(...filters: Optional<Filter>[]): Promise<PageList<Json>> {
    return this.provider.aggregate(toArray(...filters).filter(isPresent) as Filter[]);
  }

  count(...filters: Optional<Filter>[]): Promise<number> {
    return this.aggregate(...filters, { $count: 'total' }).then(d => (d.first()?.total as number) ?? 0);
  }

  match(f: Filter): Promise<PageList<Json>> {
    return this.aggregate(stages.match.match(f));
  }

  add(item: Json): Promise<Json> {
    return this.provider.add(this.collection.out(item)).then(j => this.collection.in(j));
  }

  update(item: Json): Promise<Json> {
    return this.provider.update(this.collection.out(item)).then(j => this.collection.in(j));
  }

  remove(id: Id): Promise<boolean> {
    return this.provider.remove(id);
  }
}