mirror of
https://github.com/cxfksword/jellyfin-plugin-danmu.git
synced 2026-02-02 17:59:58 +08:00
300 lines
9.1 KiB
C#
300 lines
9.1 KiB
C#
using System;
|
||
using System.Threading;
|
||
using System.Threading.Tasks;
|
||
using MediaBrowser.Controller.Entities;
|
||
using Microsoft.Extensions.Logging;
|
||
using Jellyfin.Plugin.Danmu.Scrapers.Entity;
|
||
using System.Collections.Generic;
|
||
using Jellyfin.Plugin.Danmu.Core.Extensions;
|
||
using System.Linq;
|
||
|
||
namespace Jellyfin.Plugin.Danmu.Scrapers.Dandan;
|
||
|
||
public class Dandan : AbstractScraper
|
||
{
|
||
public const string ScraperProviderName = "弹弹play";
|
||
public const string ScraperProviderId = "DandanID";
|
||
|
||
private readonly DandanApi _api;
|
||
|
||
public Dandan(ILoggerFactory logManager)
|
||
: base(logManager.CreateLogger<Dandan>())
|
||
{
|
||
_api = new DandanApi(logManager);
|
||
}
|
||
|
||
public override int DefaultOrder => 2;
|
||
|
||
public override bool DefaultEnable => true;
|
||
|
||
public override string Name => "弹弹play";
|
||
|
||
public override string ProviderName => ScraperProviderName;
|
||
|
||
public override string ProviderId => ScraperProviderId;
|
||
|
||
public override async Task<List<ScraperSearchInfo>> Search(BaseItem item)
|
||
{
|
||
var list = new List<ScraperSearchInfo>();
|
||
var isMovieItemType = item is MediaBrowser.Controller.Entities.Movies.Movie;
|
||
var searchName = this.NormalizeSearchName(item.Name);
|
||
var animes = await this._api.SearchAsync(searchName, CancellationToken.None).ConfigureAwait(false);
|
||
var matches = await this._api.MatchAsync(item, CancellationToken.None).ConfigureAwait(false);
|
||
|
||
foreach (var match in matches)
|
||
{
|
||
var anime = await this._api.GetAnimeAsync(match.AnimeId, CancellationToken.None).ConfigureAwait(false);
|
||
if (anime == null)
|
||
{
|
||
continue;
|
||
}
|
||
|
||
animes.Add(anime);
|
||
}
|
||
|
||
animes = animes.DistinctBy(x => x.AnimeId).ToList();
|
||
|
||
foreach (var anime in animes)
|
||
{
|
||
var animeId = anime.AnimeId;
|
||
var title = anime.AnimeTitle;
|
||
var pubYear = anime.Year;
|
||
|
||
if (isMovieItemType && anime.Type != "movie")
|
||
{
|
||
continue;
|
||
}
|
||
|
||
if (!isMovieItemType && anime.Type == "movie")
|
||
{
|
||
continue;
|
||
}
|
||
|
||
list.Add(new ScraperSearchInfo()
|
||
{
|
||
Id = $"{animeId}",
|
||
Name = title,
|
||
Category = anime.TypeDescription,
|
||
Year = pubYear,
|
||
EpisodeSize = anime.EpisodeCount ?? 0,
|
||
});
|
||
}
|
||
|
||
return list;
|
||
}
|
||
|
||
public override async Task<string?> SearchMediaId(BaseItem item)
|
||
{
|
||
var isMovieItemType = item is MediaBrowser.Controller.Entities.Movies.Movie;
|
||
var searchName = this.NormalizeSearchName(item.Name);
|
||
var animes = await this._api.SearchAsync(searchName, CancellationToken.None).ConfigureAwait(false);
|
||
foreach (var anime in animes)
|
||
{
|
||
var animeId = anime.AnimeId;
|
||
var title = anime.AnimeTitle;
|
||
var pubYear = anime.Year;
|
||
|
||
if (isMovieItemType && anime.Type != "movie")
|
||
{
|
||
continue;
|
||
}
|
||
|
||
if (!isMovieItemType && anime.Type == "movie")
|
||
{
|
||
continue;
|
||
}
|
||
|
||
// 检测标题是否相似（越大越相似）
|
||
var score = searchName.Distance(title);
|
||
if (score < 0.7)
|
||
{
|
||
log.LogDebug("[{0}] 标题差异太大，忽略处理. 搜索词：{1}, score: {2}", title, searchName, score);
|
||
continue;
|
||
}
|
||
|
||
// 检测年份是否一致
|
||
var itemPubYear = item.ProductionYear ?? 0;
|
||
if (itemPubYear > 0 && pubYear > 0 && itemPubYear != pubYear)
|
||
{
|
||
log.LogDebug("[{0}] 发行年份不一致，忽略处理. dandan：{1} jellyfin: {2}", title, pubYear, itemPubYear);
|
||
continue;
|
||
}
|
||
|
||
return $"{animeId}";
|
||
}
|
||
|
||
return null;
|
||
}
|
||
|
||
public override async Task<string?> SearchMediaIdByFile(Video video)
|
||
{
|
||
var isMovieItemType = video is MediaBrowser.Controller.Entities.Movies.Movie;
|
||
var matches = await _api.MatchAsync(video, CancellationToken.None).ConfigureAwait(false);
|
||
foreach (var match in matches)
|
||
{
|
||
if (isMovieItemType && match.Type != "movie")
|
||
{
|
||
continue;
|
||
}
|
||
|
||
if (!isMovieItemType && match.Type == "movie")
|
||
{
|
||
continue;
|
||
}
|
||
|
||
log.LogInformation("通过文件特征匹配到动画: {Title}, AnimeId: {AnimeId}", match.AnimeTitle, match.AnimeId);
|
||
return $"{match.AnimeId}";
|
||
}
|
||
|
||
return null;
|
||
}
|
||
|
||
public override async Task<ScraperMedia?> GetMedia(BaseItem item, string id)
|
||
{
|
||
var animeId = id.ToLong();
|
||
if (animeId <= 0)
|
||
{
|
||
return null;
|
||
}
|
||
|
||
|
||
var anime = await _api.GetAnimeAsync(animeId, CancellationToken.None).ConfigureAwait(false);
|
||
if (anime == null)
|
||
{
|
||
log.LogInformation("[{0}]获取不到视频信息：id={1}", this.Name, animeId);
|
||
return null;
|
||
}
|
||
|
||
var isMovieItemType = item is MediaBrowser.Controller.Entities.Movies.Movie;
|
||
var media = new ScraperMedia();
|
||
|
||
media.Id = id;
|
||
if (isMovieItemType && anime.Episodes != null && anime.Episodes.Count > 0)
|
||
{
|
||
media.CommentId = $"{anime.Episodes[0].EpisodeId}";
|
||
}
|
||
if (anime.Episodes != null && anime.Episodes.Count > 0)
|
||
{
|
||
foreach (var ep in anime.Episodes)
|
||
{
|
||
media.Episodes.Add(new ScraperEpisode() { Id = $"{ep.EpisodeId}", CommentId = $"{ep.EpisodeId}" });
|
||
}
|
||
}
|
||
|
||
|
||
return media;
|
||
}
|
||
|
||
public override async Task<ScraperEpisode?> GetMediaEpisode(BaseItem item, string id)
|
||
{
|
||
var isMovieItemType = item is MediaBrowser.Controller.Entities.Movies.Movie;
|
||
if (isMovieItemType)
|
||
{
|
||
// id是animeId
|
||
var anime = await _api.GetAnimeAsync(id.ToLong(), CancellationToken.None).ConfigureAwait(false);
|
||
if (anime == null || anime.Episodes == null || anime.Episodes.Count <= 0)
|
||
{
|
||
return null;
|
||
}
|
||
|
||
return new ScraperEpisode() { Id = id, CommentId = $"{anime.Episodes[0].EpisodeId}" };
|
||
}
|
||
else
|
||
{
|
||
// id是episodeId
|
||
var epId = id.ToLong();
|
||
if (epId <= 0)
|
||
{
|
||
return null;
|
||
}
|
||
|
||
return new ScraperEpisode() { Id = id, CommentId = id };
|
||
}
|
||
}
|
||
|
||
public override async Task<ScraperDanmaku?> GetDanmuContent(BaseItem item, string commentId)
|
||
{
|
||
var cid = commentId.ToLong();
|
||
if (cid <= 0)
|
||
{
|
||
return null;
|
||
}
|
||
|
||
var comments = await _api.GetCommentsAsync(cid, CancellationToken.None).ConfigureAwait(false);
|
||
var danmaku = new ScraperDanmaku();
|
||
danmaku.ChatId = cid;
|
||
danmaku.ChatServer = "api.dandanplay.net";
|
||
foreach (var comment in comments)
|
||
{
|
||
var danmakuText = new ScraperDanmakuText();
|
||
var arr = comment.P.Split(",");
|
||
danmakuText.Progress = (int)(Convert.ToDouble(arr[0]) * 1000);
|
||
danmakuText.Mode = Convert.ToInt32(arr[1]);
|
||
danmakuText.Color = Convert.ToUInt32(arr[2]);
|
||
danmakuText.MidHash = arr[3];
|
||
danmakuText.Id = comment.Cid;
|
||
danmakuText.Content = comment.Text;
|
||
|
||
danmaku.Items.Add(danmakuText);
|
||
|
||
}
|
||
|
||
return danmaku;
|
||
}
|
||
|
||
public override async Task<List<ScraperSearchInfo>> SearchForApi(string keyword)
|
||
{
|
||
var list = new List<ScraperSearchInfo>();
|
||
var animes = await this._api.SearchAsync(keyword, CancellationToken.None).ConfigureAwait(false);
|
||
foreach (var anime in animes)
|
||
{
|
||
var animeId = anime.AnimeId;
|
||
var title = anime.AnimeTitle;
|
||
var pubYear = anime.Year;
|
||
|
||
list.Add(new ScraperSearchInfo()
|
||
{
|
||
Id = $"{animeId}",
|
||
Name = title,
|
||
Category = anime.TypeDescription,
|
||
Year = pubYear,
|
||
EpisodeSize = anime.EpisodeCount ?? 0,
|
||
});
|
||
}
|
||
|
||
return list;
|
||
}
|
||
|
||
public override async Task<List<ScraperEpisode>> GetEpisodesForApi(string id)
|
||
{
|
||
var list = new List<ScraperEpisode>();
|
||
var animeId = id.ToLong();
|
||
if (animeId <= 0)
|
||
{
|
||
return list;
|
||
}
|
||
|
||
var anime = await this._api.GetAnimeAsync(animeId, CancellationToken.None).ConfigureAwait(false);
|
||
if (anime == null)
|
||
{
|
||
return list;
|
||
}
|
||
|
||
if (anime.Episodes != null && anime.Episodes.Count > 0)
|
||
{
|
||
foreach (var ep in anime.Episodes)
|
||
{
|
||
list.Add(new ScraperEpisode() { Id = $"{ep.EpisodeId}", CommentId = $"{ep.EpisodeId}", Title = ep.EpisodeTitle });
|
||
}
|
||
}
|
||
|
||
|
||
return list;
|
||
}
|
||
|
||
public override async Task<ScraperDanmaku?> DownloadDanmuForApi(string commentId)
|
||
{
|
||
return await this.GetDanmuContent(null, commentId).ConfigureAwait(false);
|
||
}
|
||
}
|