import {
    Directive,
    ElementRef,
    OnInit,
} from '@angular/core';

@Directive({
    selector: '[fade-in-on-load]',
})
export class FadeInOnLoadDirective implements OnInit {
    constructor(
        private _elementRef: ElementRef,
    ) {}

    public ngOnInit() {
        this._elementRef.nativeElement.classList.add(
            'u-fade-in-on-load',
        );
        this._elementRef.nativeElement.onload = () => {
            this._elementRef.nativeElement.classList.add(
                'u-fade-in-on-load--loaded',
            );
        };
    }
}