Resolvers

VehicleResolvers

import { IResolvers } from "apollo-server-express";
import { addCar, getAllVehicles, getCars } from "../dataService";
import { Boat, Car, Truck } from "../types/Entities";
import { GqlContext } from "./GqlContext";

const resolvers: IResolvers = {
  SearchResult: {
    __resolveType(obj: any, ctx: GqlContext, info: any) {
      if (obj.wheelCount && obj.passengerCount) return "Car";
      if (obj.wheelCount && obj.maxPayload) return "Truck";
      if (obj.passengerCount) return "Boat";
      return "Car";
    },
  },
  IVehicle: {
    __resolveType(obj: any, ctx: GqlContext, info: any) {
      if (obj.wheelCount && obj.passengerCount) return "Car";
      if (obj.wheelCount && obj.maxPayload) return "Truck";
      if (obj.passengerCount) return "Boat";
      return "Car";
    },
  },
  IPassengerVehicle: {
    __resolveType(obj: any, ctx: GqlContext, info: any) {
      if (obj.wheelCount && obj.passengerCount) {
        return "Car";
      }
      return "Boat";
    },
  },
  IWorkVehicle: {
    __resolveType(obj: any, ctx: GqlContext, info: any) {
      return "Truck";
    },
  },
  ILandVehicle: {
    __resolveType(obj: any, ctx: GqlContext, info: any) {
      if (obj.wheelCount && obj.passengerCount) {
        return "Car";
      }
      return "Truck";
    },
  },
  Query: {
    getAllVehicles: async (
      obj: any,
      args: null,
      ctx: GqlContext,
      info: any
    ): Promise<Array<Car | Truck | Boat>> => {
      return getAllVehicles();
    },
  },
};

export default resolvers;