import { FlutterDriver } from '../../driver';

const waitForConstructor = (command: `waitForAbsent` | `waitFor` | `waitForTappable`) => async (
  self: FlutterDriver,
  elementBase64: string,
  durationMilliseconds?: number,
): Promise<string> => {

  let args = {};

  if (typeof durationMilliseconds === `number`) {
    args = {
      timeout: durationMilliseconds,
    };
  } else if (typeof durationMilliseconds !== `undefined`) {
    // @todo BaseDriver's errors.InvalidArgumentError();
    throw new Error(`durationMilliseconds is not a valid options`);
  }

  await self.executeElementCommand(command, elementBase64, args);
  return elementBase64;
};

export const waitForAbsent = waitForConstructor(`waitForAbsent`);

export const waitFor = waitForConstructor(`waitFor`);

export const waitForTappable = waitForConstructor(`waitForTappable`);