#pragma warning disable RS0030 // Do not use banned APIs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Threading;
using Jellyfin.Database.Implementations;
using Jellyfin.Database.Implementations.Entities;
using MediaBrowser.Controller.Configuration;
using MediaBrowser.Controller.Dto;
using MediaBrowser.Controller.Entities;
using MediaBrowser.Controller.Library;
using MediaBrowser.Model.Dto;
using MediaBrowser.Model.Entities;
using Microsoft.EntityFrameworkCore;
using AudioBook = MediaBrowser.Controller.Entities.AudioBook;
using Book = MediaBrowser.Controller.Entities.Book;

namespace Emby.Server.Implementations.Library
{
    /// <summary>
    /// Class UserDataManager.
    /// </summary>
    public class UserDataManager : IUserDataManager
    {
        private readonly IServerConfigurationManager _config;
        private readonly IDbContextFactory<JellyfinDbContext> _repository;

        /// <summary>
        /// Initializes a new instance of the <see cref="UserDataManager"/> class.
        /// </summary>
        /// <param name="config">Instance of the <see cref="IServerConfigurationManager"/> interface.</param>
        /// <param name="repository">Instance of the <see cref="IDbContextFactory{JellyfinDbContext}"/> interface.</param>
        public UserDataManager(
            IServerConfigurationManager config,
            IDbContextFactory<JellyfinDbContext> repository)
        {
            _config = config;
            _repository = repository;
        }

        /// <inheritdoc />
        public event EventHandler<UserDataSaveEventArgs>? UserDataSaved;

        /// <inheritdoc />
        public void SaveUserData(User user, BaseItem item, UserItemData userData, UserDataSaveReason reason, CancellationToken cancellationToken)
        {
            ArgumentNullException.ThrowIfNull(userData);

            ArgumentNullException.ThrowIfNull(item);

            cancellationToken.ThrowIfCancellationRequested();

            var keys = item.GetUserDataKeys();

            using var dbContext = _repository.CreateDbContext();
            using var transaction = dbContext.Database.BeginTransaction();

            foreach (var key in keys)
            {
                userData.Key = key;
                var userDataEntry = Map(userData, user.Id, item.Id);
                if (dbContext.UserData.Any(f => f.ItemId == userDataEntry.ItemId && f.UserId == userDataEntry.UserId && f.CustomDataKey == userDataEntry.CustomDataKey))
                {
                    dbContext.UserData.Attach(userDataEntry).State = EntityState.Modified;
                }
                else
                {
                    dbContext.UserData.Add(userDataEntry);
                }
            }

            dbContext.SaveChanges();
            transaction.Commit();

            UserDataSaved?.Invoke(this, new UserDataSaveEventArgs
            {
                Keys = keys,
                UserData = userData,
                SaveReason = reason,
                UserId = user.Id,
                Item = item
            });
        }

        /// <inheritdoc />
        public void SaveUserData(User user, BaseItem item, UpdateUserItemDataDto userDataDto, UserDataSaveReason reason)
        {
            ArgumentNullException.ThrowIfNull(user);
            ArgumentNullException.ThrowIfNull(item);
            ArgumentNullException.ThrowIfNull(userDataDto);

            var userData = GetUserData(user, item) ?? throw new InvalidOperationException("UserData should not be null.");

            if (userDataDto.PlaybackPositionTicks.HasValue)
            {
                userData.PlaybackPositionTicks = userDataDto.PlaybackPositionTicks.Value;
            }

            if (userDataDto.PlayCount.HasValue)
            {
                userData.PlayCount = userDataDto.PlayCount.Value;
            }

            if (userDataDto.IsFavorite.HasValue)
            {
                userData.IsFavorite = userDataDto.IsFavorite.Value;
            }

            if (userDataDto.Likes.HasValue)
            {
                userData.Likes = userDataDto.Likes.Value;
            }

            if (userDataDto.Played.HasValue)
            {
                userData.Played = userDataDto.Played.Value;
            }

            if (userDataDto.LastPlayedDate.HasValue)
            {
                userData.LastPlayedDate = userDataDto.LastPlayedDate.Value;
            }

            if (userDataDto.Rating.HasValue)
            {
                userData.Rating = userDataDto.Rating.Value;
            }

            SaveUserData(user, item, userData, reason, CancellationToken.None);
        }

        private UserData Map(UserItemData dto, Guid userId, Guid itemId)
        {
            return new UserData()
            {
                ItemId = itemId,
                CustomDataKey = dto.Key,
                Item = null,
                User = null,
                AudioStreamIndex = dto.AudioStreamIndex,
                IsFavorite = dto.IsFavorite,
                LastPlayedDate = dto.LastPlayedDate,
                Likes = dto.Likes,
                PlaybackPositionTicks = dto.PlaybackPositionTicks,
                PlayCount = dto.PlayCount,
                Played = dto.Played,
                Rating = dto.Rating,
                UserId = userId,
                SubtitleStreamIndex = dto.SubtitleStreamIndex,
            };
        }

        private static UserItemData Map(UserData dto)
        {
            return new UserItemData()
            {
                Key = dto.CustomDataKey!,
                AudioStreamIndex = dto.AudioStreamIndex,
                IsFavorite = dto.IsFavorite,
                LastPlayedDate = dto.LastPlayedDate,
                Likes = dto.Likes,
                PlaybackPositionTicks = dto.PlaybackPositionTicks,
                PlayCount = dto.PlayCount,
                Played = dto.Played,
                Rating = dto.Rating,
                SubtitleStreamIndex = dto.SubtitleStreamIndex,
            };
        }

        /// <inheritdoc />
        public Dictionary<Guid, UserItemData> GetUserDataBatch(IReadOnlyList<BaseItem> items, User user)
        {
            ArgumentNullException.ThrowIfNull(items);
            ArgumentNullException.ThrowIfNull(user);

            var result = new Dictionary<Guid, UserItemData>(items.Count);
            if (items.Count == 0)
            {
                return result;
            }

            // Fetch rows by item alone so rows kept under keys from older metadata resolve the same
            // way as the single item path.
            var itemIds = items.Select(e => e.Id).Distinct().ToList();
            using var context = _repository.CreateDbContext();
            var userDataByItem = context.UserData
                .AsNoTracking()
                .Where(e => e.UserId.Equals(user.Id))
                .WhereOneOrMany(itemIds, e => e.ItemId)
                .ToArray()
                .GroupBy(e => e.ItemId)
                .ToDictionary(g => g.Key, g => g.ToArray());

            foreach (var item in items)
            {
                if (result.ContainsKey(item.Id))
                {
                    continue;
                }

                var row = userDataByItem.TryGetValue(item.Id, out var itemUserData)
                    ? ResolveUserDataRow(item, itemUserData)
                    : null;

                result[item.Id] = row is not null
                    ? Map(row)
                    : new UserItemData { Key = item.GetUserDataKeys().FirstOrDefault() ?? string.Empty };
            }

            return result;
        }

        /// <inheritdoc />
        public VersionResumeData? GetResumeUserData(User user, BaseItem item)
        {
            return GetResumeUserDataBatch([item], user).GetValueOrDefault(item.Id);
        }

        /// <inheritdoc />
        public IReadOnlyDictionary<Guid, VersionResumeData> GetResumeUserDataBatch(IReadOnlyList<BaseItem> items, User user)
        {
            ArgumentNullException.ThrowIfNull(user);

            var result = new Dictionary<Guid, VersionResumeData>();

            // Candidate primaries: a directly queried version (PrimaryVersionId set) keeps its own data.
            // Linked alternates are already known in memory; only the local-alternate existence check
            // would otherwise hit the database (one query per item via Video.HasLocalAlternateVersions),
            // so collect those ids and resolve them all in a single query below.
            List<Video>? candidates = null;
            List<Guid>? localProbeIds = null;
            foreach (var item in items)
            {
                if (item is not Video video || video.PrimaryVersionId.HasValue)
                {
                    continue;
                }

                (candidates ??= []).Add(video);

                if (video.LinkedAlternateVersions.Length == 0)
                {
                    (localProbeIds ??= []).Add(video.Id);
                }
            }

            if (candidates is null)
            {
                return result;
            }

            HashSet<Guid>? withLocalAlternates = null;
            if (localProbeIds is not null)
            {
                using var dbContext = _repository.CreateDbContext();
                withLocalAlternates = dbContext.LinkedChildren
                    .Where(lc => lc.ChildType == Jellyfin.Database.Implementations.Entities.LinkedChildType.LocalAlternateVersion)
                    .WhereOneOrMany(localProbeIds, lc => lc.ParentId)
                    .Select(lc => lc.ParentId)
                    .Distinct()
                    .ToHashSet();
            }

            List<(Guid PrimaryId, IReadOnlyList<Video> Versions)>? versionGroups = null;
            List<BaseItem>? allVersions = null;

            foreach (var video in candidates)
            {
                // Only items that actually have alternate versions aggregate over them.
                if (video.LinkedAlternateVersions.Length == 0
                    && (withLocalAlternates is null || !withLocalAlternates.Contains(video.Id)))
                {
                    continue;
                }

                var versions = video.GetAllVersions();
                if (versions.Count < 2)
                {
                    continue;
                }

                (versionGroups ??= []).Add((video.Id, versions));
                (allVersions ??= []).AddRange(versions);
            }

            if (versionGroups is null)
            {
                return result;
            }

            var userDataByVersion = GetUserDataBatch(allVersions!.DistinctBy(i => i.Id).ToList(), user);

            foreach (var (primaryId, versions) in versionGroups)
            {
                // Consider both in-progress and completed versions so a finished alternate still marks the primary as played.
                var resumeVersion = VersionPlaybackSelector.SelectMostRecentlyPlayed(
                    versions,
                    version => userDataByVersion.GetValueOrDefault(version.Id),
                    data => data.PlaybackPositionTicks > 0 || data.Played);

                if (resumeVersion is not null)
                {
                    result[primaryId] = new VersionResumeData(resumeVersion.Id, userDataByVersion[resumeVersion.Id]);
                }
            }

            return result;
        }

        /// <inheritdoc />
        public UserItemData? GetUserData(User user, BaseItem item)
        {
            ArgumentNullException.ThrowIfNull(user);
            ArgumentNullException.ThrowIfNull(item);

            using var dbContext = _repository.CreateDbContext();
            var rows = dbContext.UserData
                .AsNoTracking()
                .Where(e => e.ItemId == item.Id && e.UserId == user.Id)
                .ToArray();

            var row = ResolveUserDataRow(item, rows);
            return row is not null ? Map(row) : new UserItemData()
            {
                Key = item.GetUserDataKeys()[0],
            };
        }

        /// <summary>
        /// Picks the row matching the item's current user data keys, in key order, so rows left behind
        /// under keys from older metadata don't take priority over the rows the write path updates.
        /// </summary>
        /// <param name="item">The item whose keys to match.</param>
        /// <param name="rows">The candidate user data rows for a single user.</param>
        /// <returns>The best matching row, or <c>null</c> when there are none.</returns>
        private static UserData? ResolveUserDataRow(BaseItem item, IEnumerable<UserData>? rows)
        {
            var candidates = rows?.ToList();
            if (candidates is null || candidates.Count == 0)
            {
                return null;
            }

            foreach (var key in item.GetUserDataKeys())
            {
                var match = candidates.Find(e => string.Equals(e.CustomDataKey, key, StringComparison.Ordinal));
                if (match is not null)
                {
                    return match;
                }
            }

            return candidates[0];
        }

        /// <inheritdoc />
        public UserItemDataDto? GetUserDataDto(BaseItem item, User user)
            => GetUserDataDto(item, null, user, new DtoOptions());

        /// <inheritdoc />
        public UserItemDataDto? GetUserDataDto(BaseItem item, BaseItemDto? itemDto, User user, DtoOptions options)
        {
            var userData = GetUserData(user, item);
            if (userData is null)
            {
                return null;
            }

            var dto = GetUserItemDataDto(userData, item.Id);

            item.FillUserDataDtoValues(dto, userData, itemDto, user, options);

            // For an item with alternate versions, surface the most recently played version's resume point.
            GetResumeUserData(user, item)?.ApplyTo(dto);

            return dto;
        }

        /// <summary>
        /// Converts a UserItemData to a DTOUserItemData.
        /// </summary>
        /// <param name="data">The data.</param>
        /// <param name="itemId">The reference key to an Item.</param>
        /// <returns>DtoUserItemData.</returns>
        /// <exception cref="ArgumentNullException"><paramref name="data"/> is <c>null</c>.</exception>
        private UserItemDataDto GetUserItemDataDto(UserItemData data, Guid itemId)
        {
            ArgumentNullException.ThrowIfNull(data);

            return new UserItemDataDto
            {
                IsFavorite = data.IsFavorite,
                Likes = data.Likes,
                PlaybackPositionTicks = data.PlaybackPositionTicks,
                PlayCount = data.PlayCount,
                Rating = data.Rating,
                Played = data.Played,
                LastPlayedDate = data.LastPlayedDate,
                ItemId = itemId,
                Key = data.Key
            };
        }

        /// <inheritdoc />
        public bool UpdatePlayState(BaseItem item, UserItemData data, long? reportedPositionTicks)
        {
            var playedToCompletion = false;

            var runtimeTicks = item.GetRunTimeTicksForPlayState();

            var positionTicks = reportedPositionTicks ?? runtimeTicks;
            var hasRuntime = runtimeTicks > 0;

            // If a position has been reported, and if we know the duration
            if (positionTicks > 0 && hasRuntime && item is not AudioBook && item is not Book)
            {
                var pctIn = decimal.Divide(positionTicks, runtimeTicks) * 100;

                if (pctIn < _config.Configuration.MinResumePct)
                {
                    // ignore progress during the beginning
                    positionTicks = 0;
                }
                else if (pctIn > _config.Configuration.MaxResumePct || positionTicks >= (runtimeTicks - TimeSpan.TicksPerSecond))
                {
                    // mark as completed close to the end
                    positionTicks = 0;
                    data.Played = playedToCompletion = true;
                }
                else
                {
                    // Enforce MinResumeDuration
                    var durationSeconds = TimeSpan.FromTicks(runtimeTicks).TotalSeconds;
                    if (durationSeconds < _config.Configuration.MinResumeDurationSeconds)
                    {
                        positionTicks = 0;
                        data.Played = playedToCompletion = true;
                    }
                }
            }
            else if (positionTicks > 0 && hasRuntime && item is AudioBook)
            {
                var playbackPositionInMinutes = TimeSpan.FromTicks(positionTicks).TotalMinutes;
                var remainingTimeInMinutes = TimeSpan.FromTicks(runtimeTicks - positionTicks).TotalMinutes;

                if (playbackPositionInMinutes < _config.Configuration.MinAudiobookResume)
                {
                    // ignore progress during the beginning
                    positionTicks = 0;
                }
                else if (remainingTimeInMinutes < _config.Configuration.MaxAudiobookResume || positionTicks >= runtimeTicks)
                {
                    // mark as completed close to the end
                    positionTicks = 0;
                    data.Played = playedToCompletion = true;
                }
            }
            else if (!hasRuntime)
            {
                // If we don't know the runtime we'll just have to assume it was fully played
                data.Played = playedToCompletion = true;
                positionTicks = 0;
            }

            if (!item.SupportsPlayedStatus)
            {
                positionTicks = 0;
                data.Played = false;
            }

            if (!item.SupportsPositionTicksResume)
            {
                positionTicks = 0;
            }

            data.PlaybackPositionTicks = positionTicks;

            return playedToCompletion;
        }

        /// <inheritdoc />
        public void ResetPlaybackStreamSelections(User user, BaseItem item)
        {
            ArgumentNullException.ThrowIfNull(user);
            ArgumentNullException.ThrowIfNull(item);

            using var dbContext = _repository.CreateDbContext();
            var rows = dbContext.UserData
                .Where(e => e.ItemId == item.Id && e.UserId == user.Id
                            && (e.AudioStreamIndex != null || e.SubtitleStreamIndex != null))
                .ToList();

            if (rows.Count == 0)
            {
                return;
            }

            foreach (var row in rows)
            {
                row.AudioStreamIndex = null;
                row.SubtitleStreamIndex = null;
            }

            dbContext.SaveChanges();
        }
    }
}