import { RuleOption } from '../rule-options';
import { IPipeRule, IPipeRunFn, PipeRule } from './interfaces';

export interface IByHourOfDayRuleOptions {
  byHourOfDay: RuleOption.ByHourOfDay[];
}

export class ByHourOfDayPipe extends PipeRule<IByHourOfDayRuleOptions>
  implements IPipeRule<IByHourOfDayRuleOptions> {
  run(args: IPipeRunFn) {
    if (args.invalidDate) {
      return this.nextPipe.run(args);
    }

    let { date } = args;

    const currentHour = date.get('hour');

    for (const hour of this.options.byHourOfDay) {
      if (currentHour > hour) continue;

      if (currentHour === hour) return this.nextPipe.run({ date });

      return this.nextValidDate(args, date.granularity('day').set('hour', hour));
    }

    date = date
      .granularity('day')
      .add(1, 'day')
      .set('hour', this.options.byHourOfDay[0]);

    return this.nextValidDate(args, date);
  }
}