SmartTagsTask.cs

using MediaBrowser.Model.Tasks;
using MediaBrowser.Controller.Library;
using MediaBrowser.Model.Serialization;
using MediaBrowser.Common.Configuration;
using MediaBrowser.Model.Logging;
using MediaBrowser.Common.Net;
using MediaBrowser.Controller.Entities;     
using MediaBrowser.Model.Entities;          
using System.Threading;
using System.Threading.Tasks;
using System.Collections.Generic;
using System;
using System.Linq;
using System.IO;

namespace SmartTags;

public class SmartTagsTask : IScheduledTask
{
    private readonly ILogger _logger;
    private readonly ILibraryManager _libraryManager;
    private readonly IJsonSerializer _jsonSerializer;
    private readonly IHttpClient _httpClient;
    private readonly IApplicationPaths _appPaths;

    public SmartTagsTask(
        ILogger logger, 
        ILibraryManager libraryManager, 
        IJsonSerializer jsonSerializer, 
        IHttpClient httpClient,
        IApplicationPaths appPaths)
    {
        _logger = logger;
        _libraryManager = libraryManager;
        _jsonSerializer = jsonSerializer;
        _httpClient = httpClient;
        _appPaths = appPaths;
    }

    public string Name => "SmartTags Update";
    public string Key => "SmartTagsTask";
    public string Description => "基于 TMDB 数据更新原产地、年代及 IMDb Top 标签。";
    public string Category => "SmartTags";

    public IEnumerable<TaskTriggerInfo> GetDefaultTriggers()
    {
        yield return new TaskTriggerInfo { Type = TaskTriggerInfo.TriggerDaily, TimeOfDayTicks = TimeSpan.FromHours(4).Ticks };
    }

    public async Task Execute(CancellationToken cancellationToken, IProgress<double> progress)
    {
        var config = Plugin.Instance?.Configuration;
        if (config == null || string.IsNullOrEmpty(config.TmdbApiKey))
        {
            _logger.Warn("[SmartTags] 未配置 TMDB API Key，跳过任务。");
            return;
        }

        _logger.Info("[SmartTags] 任务开始。正在初始化...");

        // 1. 准备 IMDb Top 250 数据 (如果启用)
        HashSet<string> imdbTopIds = new HashSet<string>(StringComparer.OrdinalIgnoreCase);
        if (config.EnableImdbTopTags)
        {
            _logger.Debug("[SmartTags] 正在获取 IMDb Top 250 列表...");
            imdbTopIds = await GetImdbTop250IdsAsync(cancellationToken);
            _logger.Info($"[SmartTags] 加载了 {imdbTopIds.Count} 个 IMDb Top 250 条目。");
        }

        // 2. 初始化数据管理器和打标服务
        var dataManager = new TmdbDataManager(_appPaths, _jsonSerializer, _httpClient);
        var service = new TaggingService(_logger, dataManager, imdbTopIds);

        // 3. 查询
        var query = new InternalItemsQuery
        {
            IncludeItemTypes = new[] { "Movie", "Series" },
            Recursive = true,
            IsVirtualItem = false
        };
        var items = _libraryManager.GetItemList(query);
        var total = items.Length;
        
        _logger.Info($"[SmartTags] 共扫描到 {total} 个媒体项。");
        
        int processed = 0;
        int updatedCount = 0;

        foreach (var item in items)
        {
            cancellationToken.ThrowIfCancellationRequested();
            processed++;
            if (processed % 50 == 0) progress.Report((double)processed / total * 100);

            // 复用 TaggingService 统一处理所有标签逻辑
            bool isModified = await service.ProcessItemAsync(item, config, cancellationToken);

            if (isModified)
            {
                item.UpdateToRepository(ItemUpdateType.MetadataEdit);
                updatedCount++;
            }
        }

        // 任务结束时刷新缓存到磁盘
        dataManager.Flush();

        _logger.Info($"[SmartTags] 任务完成。处理 {total} 个项目，实际更新 {updatedCount} 个。");
    }

    // --- 辅助方法: IMDb Top 250 ---
    private async Task<HashSet<string>> GetImdbTop250IdsAsync(CancellationToken cancellationToken)
    {
        var cachePath = Path.Combine(_appPaths.PluginConfigurationsPath, "SmartTags_ImdbTop250.json");
        var ids = new HashSet<string>(StringComparer.OrdinalIgnoreCase);

        // 1. 尝试从网络下载
        bool needDownload = true;
        if (File.Exists(cachePath))
        {
            var lastWrite = File.GetLastWriteTimeUtc(cachePath);
            // 缓存有效期 24 小时
            if ((DateTime.UtcNow - lastWrite).TotalHours < 24)
            {
                needDownload = false;
                _logger.Debug("[SmartTags] IMDb Top 250 缓存未过期，使用本地文件。");
            }
        }

        if (needDownload)
        {
            try 
            {
                var url = "https://raw.githubusercontent.com/theapache64/top250/master/top250_min.json";
                var options = new HttpRequestOptions 
                { 
                    Url = url, 
                    CancellationToken = cancellationToken, 
                    BufferContent = true,
                    EnableHttpCompression = true
                };
                
                using var response = await _httpClient.GetResponse(options).ConfigureAwait(false);
                using var stream = response.Content;
                using var fileStream = new FileStream(cachePath, FileMode.Create, FileAccess.Write);
                await stream.CopyToAsync(fileStream);
                _logger.Info("[SmartTags] IMDb Top 250 数据已从 GitHub 更新。");
            }
            catch (Exception ex)
            {
                _logger.Error($"[SmartTags] 下载 IMDb Top 250 失败: {ex.Message}。将尝试使用旧缓存。");
            }
        }

        // 2. 读取本地缓存并解析
        if (File.Exists(cachePath))
        {
            try
            {
                var list = _jsonSerializer.DeserializeFromFile<List<ImdbSimpleItem>>(cachePath);
                
                if (list != null)
                {
                    foreach (var i in list)
                    {
                        // JSON 格式: "/title/tt0055630/"
                        if (!string.IsNullOrEmpty(i.ImdbUrl))
                        {
                            var parts = i.ImdbUrl.Split('/');
                            var ttId = parts.FirstOrDefault(p => p.StartsWith("tt", StringComparison.OrdinalIgnoreCase));
                            
                            if (!string.IsNullOrEmpty(ttId))
                            {
                                ids.Add(ttId);
                            }
                        }
                    }
                    _logger.Info($"[SmartTags] 成功解析 {ids.Count} 个 IMDb Top 250 ID。");
                }
            }
            catch (Exception ex)
            {
                _logger.Error($"[SmartTags] 解析 IMDb 缓存失败: {ex.Message}");
            }
        }

        return ids;
    }
}