import Subject from '../Subject';
import Subscriber from './Subscriber';
import Subscription from '../Subscription';
import Scheduler from '../Scheduler';
import TestMessage from './TestMessage';
import SubscriptionLog from './SubscriptionLog';
import SubscriptionLoggable from './SubscriptionLoggable';
import applyMixins from '../util/applyMixins';

export default class HotObservable<T> extends Subject<T> implements SubscriptionLoggable {
  public subscriptions: SubscriptionLog[] = [];
  scheduler: Scheduler;
  logSubscribedFrame: () => number;
  logUnsubscribedFrame: (index: number) => void;

  constructor(public messages: TestMessage[],
              scheduler: Scheduler) {
    super();
    this.scheduler = scheduler;
  }

  _subscribe(subscriber: Subscriber<any>): Subscription<T> {
    const subject: HotObservable<T> = this;
    const index = subject.logSubscribedFrame();
    subscriber.add(new Subscription(() => {
      subject.logUnsubscribedFrame(index);
    }));
    return super._subscribe(subscriber);
  }

  setup() {
    const subject = this;
    const messagesLength = subject.messages.length;
    for (let i = 0; i < messagesLength; i++) {
      const message = subject.messages[i];
      this.scheduler.schedule(
        () => { message.notification.observe(subject); },
        message.frame
      );
    }
  }
}
applyMixins(HotObservable, [SubscriptionLoggable]);

