Ombi/src/Ombi/ClientApp/src/app/media-details/components/tv/panels/tv-request-grid/tv-request-grid.component.ts

import { Component, Input } from "@angular/core";
import { IChildRequests, IEpisodesRequests, INewSeasonRequests, IRequestEngineResult, ISeasonsViewModel, ITvRequestViewModelV2, RequestType } from "../../../../../interfaces";
import { RequestService } from "../../../../../services/request.service";
import { MessageService } from "../../../../../services";
import { DenyDialogComponent } from "../../../shared/deny-dialog/deny-dialog.component";
import { ISearchTvResultV2 } from "../../../../../interfaces/ISearchTvResultV2";
import { TranslateService } from "@ngx-translate/core";
import { MatDialog } from "@angular/material/dialog";
import { SelectionModel } from "@angular/cdk/collections";
import { RequestServiceV2 } from "../../../../../services/requestV2.service";
import { AdminRequestDialogComponent } from "../../../../../shared/admin-request-dialog/admin-request-dialog.component";

@Component({
    templateUrl: "./tv-request-grid.component.html",
    styleUrls: ["./tv-request-grid.component.scss"],
    selector: "tv-request-grid"
})
export class TvRequestGridComponent {
    @Input() public tv: ISearchTvResultV2;
    @Input() public tvRequest: IChildRequests[];
    @Input() public isAdmin: boolean;
    public selection = new SelectionModel<IEpisodesRequests>(true, []);

    public get requestable() {
        return this.tv?.seasonRequests?.length > 0
    }

    public displayedColumns: string[] = ['select', 'number', 'title', 'airDate', 'status'];

    constructor(private requestService: RequestService, private requestServiceV2: RequestServiceV2, private notificationService: MessageService,
        private dialog: MatDialog, private translate: TranslateService) {

    }

    public async submitRequests() {
        // Make sure something has been selected
        const selected = this.selection.hasValue();
        if (!selected && !this.tv.requestAll && !this.tv.firstSeason && !this.tv.latestSeason) {
            this.notificationService.send(this.translate.instant("Requests.NeedToSelectEpisodes"));
            return;
        }

        this.tv.requested = true;

        const viewModel = <ITvRequestViewModelV2>{
            firstSeason: this.tv.firstSeason, latestSeason: this.tv.latestSeason, requestAll: this.tv.requestAll, theMovieDbId: this.tv.id,
            requestOnBehalf: null, languageCode: this.translate.currentLang
        };
        viewModel.seasons = [];
        this.tv.seasonRequests.forEach((season) => {
            const seasonsViewModel = <ISeasonsViewModel>{ seasonNumber: season.seasonNumber, episodes: [] };
            if (!this.tv.latestSeason && !this.tv.requestAll && !this.tv.firstSeason) {
                season.episodes.forEach(ep => {
                    if (this.selection.isSelected(ep)) {
                        ep.requested = true;
                        ep.requestStatus = "Common.PendingApproval";
                        seasonsViewModel.episodes.push({ episodeNumber: ep.episodeNumber });
                    }
                });
            }
            viewModel.seasons.push(seasonsViewModel);
        });

        if (this.isAdmin) {
            const dialog = this.dialog.open(AdminRequestDialogComponent, { width: "700px", data: { type: RequestType.tvShow, id: this.tv.id, is4k: null }, panelClass: 'modal-panel' });
            dialog.afterClosed().subscribe(async (result) => {
                if (result) {
                    viewModel.requestOnBehalf = result.username?.id;
                    viewModel.qualityPathOverride = result?.sonarrPathId;
                    viewModel.rootFolderOverride = result?.sonarrFolderId;
                    viewModel.languageProfile = result?.sonarrLanguageId;

                    const requestResult = await this.requestServiceV2.requestTv(viewModel).toPromise();
                    this.postRequest(requestResult);
                }
            });
        } else {
            const requestResult = await this.requestServiceV2.requestTv(viewModel).toPromise();
            this.postRequest(requestResult);
        }
    }

    public async requestAllSeasons() {
        this.tv.requestAll = true;
        await this.submitRequests();
    }

    public async requestFirstSeason() {
        this.tv.firstSeason = true;
        await this.submitRequests();
    }

    public async requestLatestSeason() {
        this.tv.latestSeason = true;
        await this.submitRequests();
    }

      /** Whether the number of selected elements matches the total number of rows. */
  public isAllSelected(dataSource: IEpisodesRequests[]) {
    const numSelected = this.selection.selected.length;
    const numRows = dataSource.length;
    return numSelected === numRows;
  }

  /** Selects all rows if they are not all selected; otherwise clear selection. */
  public masterToggle(dataSource: IEpisodesRequests[]) {
    this.isAllSelected(dataSource) ?
        this.selection.clear() :
        dataSource.forEach(row => {
            if (!row.available && !row.requested && !row.approved) {
                this.selection.select(row)
            }
        });
  }

  public isSeasonCheckable(season: INewSeasonRequests) {
      const seasonAvailable = season.episodes.every((ep) => {
        return ep.available || ep.requested || ep.approved;
      });
      return !seasonAvailable;
    }

    public getStatusClass(season: INewSeasonRequests): string {
        const seasonAvailable = season.episodes.every((ep) => {
            return ep.available;
          });
        if (seasonAvailable) {
            return "available";
        }

        const allDenied = season.episodes.every((ep) => {
            return ep.denied;
        });
        if (allDenied) {
            return "denied";
        }

        const seasonPending = season.episodes.some((ep) => {
            return ep.requested && !ep.approved
          });
        if (seasonPending) {
            return "requested";
        }

        const seasonApproved = season.episodes.some((ep) => {
            return ep.requested && ep.approved
          });
        if (seasonApproved) {
            return "approved";
        }
        return "";
    }

    public getEpisodeStatusClass(ep: IEpisodesRequests): string {
        if (ep.available) {
            return "available";
        }

        if (ep.denied) {
            return "denied";
        }

        if (ep.requested && !ep.approved) {
            return "requested";
        }

        if (ep.requested && ep.approved) {
            return "approved";
        }
        return "";
    }

    private postRequest(requestResult: IRequestEngineResult) {
        if (requestResult.result) {
            this.notificationService.send(
                this.translate.instant("Requests.RequestAddedSuccessfully", { title:this.tv.title }));

            this.selection.clear();

            if (this.tv.firstSeason) {
                this.tv.seasonRequests[0].episodes.forEach(ep => {
                    ep.requested = true;
                    ep.requestStatus = "Common.PendingApproval";
                });
            }
            if (this.tv.requestAll) {
                this.tv.seasonRequests.forEach(season => {
                    season.episodes.forEach(ep => {
                        ep.requested = true;
                        ep.requestStatus = "Common.PendingApproval";
                    });
                });
            }
            if (this.tv.latestSeason) {
                this.tv.seasonRequests[this.tv.seasonRequests.length - 1].episodes.forEach(ep => {
                    ep.requested = true;
                    ep.requestStatus = "Common.PendingApproval";
                });
            }

        } else {
            this.notificationService.sendRequestEngineResultError(requestResult);
        }
    }
}