import { map } from 'rxjs/operators';
import { VesselGroupDto, VesselGroup } from 'app/modules/earnings/store/vesselGroups/vesselGroups.types';
import { MaerskApiClient } from '../../../common/clients/maerskApi.client';

export class VesselGroupsService {
  private apiClient: MaerskApiClient;

  constructor(apiClient) {
    this.apiClient = apiClient;
  }

  createVesselGroups = (vesselGroups: VesselGroupDto[]): VesselGroup[] => {
    return vesselGroups.map(
      (vesselGroup) => ({ id: vesselGroup.id, name: vesselGroup.name })
    );
  }

  getVesselGroups = (poolId: number) => {
    return this.apiClient.getVesselGroups(poolId).pipe(map(this.createVesselGroups));
  }
}