suncheng/NasRobot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Refactor ChineseNfoRegistry to remove redundant permission setting and improve episode processing logic
All checks were successful
Docker Build & Deploy / Build Docker Image (push) Successful in 15s
Details
Docker Build & Deploy / Deploy to Production (push) Successful in 4s
Details

Browse Source
This commit is contained in:
孙诚
2025-04-03 11:28:26 +08:00
parent b008a54fdb
commit 14e59b48f6
1 changed files with 337 additions and 150 deletions
Download Patch File Download Diff File Expand all files Collapse all files

487
src/Service/Jobs/ChineseNfoRegistry.cs
View File

@@ -64,15 +64,6 @@ public class ChineseNfoRegistry : Registry, IChineseNfoRegistry
var failedCount = 0;
var skippedCount = 0;
// 使用 root 角色执行 chmod -R 666 设置全部文件的读写权限
var permission = new ProcessStartInfo("chmod", "-R 777 " + tvFolder)
{
CreateNoWindow = true,
UseShellExecute = false
};
Process.Start(permission);
// 扫描 tvshow.nfo 文件
var tvShowFiles = Directory.GetFiles(tvFolder, "tvshow.nfo", SearchOption.AllDirectories);
@@ -96,6 +87,10 @@ public class ChineseNfoRegistry : Registry, IChineseNfoRegistry
continue;
}
await SetTvInfo(tvShowFile, tvXml, tmdbId);
await Task.Delay(10000);
// 获取 tvShowFile 的文件夹
var folderPath = Path.GetDirectoryName(tvShowFile);
@@ -112,155 +107,347 @@ public class ChineseNfoRegistry : Registry, IChineseNfoRegistry
foreach (var episodeFile in seasonFiles)
{
var episodeContent = await File.ReadAllTextAsync(episodeFile);
var episodeXml = new XmlDocument();
episodeXml.LoadXml(episodeContent);
// 判断有无 completed 节点
var completedNode = episodeXml.SelectSingleNode("//completed");
if (completedNode != null && ignoreCompleted == false)
{
skippedCount++;
Console.WriteLine($"{episodeFile} & 已完成");
continue;
}
// 判断 locked == true
var lockedNode = episodeXml.SelectSingleNode("//lockdata");
if (lockedNode != null && lockedNode.InnerText == "true" && ignoreLocked == false)
{
skippedCount++;
Console.WriteLine($"{episodeFile} & 已锁定");
continue;
}
// 读取 <season>1</season>
var seasonNode = episodeXml.SelectSingleNode("//season");
if (seasonNode == null)
{
failedCount++;
Console.WriteLine($"{episodeFile} & 未找到 season");
continue;
}
// 读取 <episode>1</episode>
var episodeNode = episodeXml.SelectSingleNode("//episode");
if (episodeNode == null)
{
failedCount++;
Console.WriteLine($"{episodeFile} & 未找到 episode");
continue;
}
if (!int.TryParse(seasonNode.InnerText, out var season))
{
failedCount++;
Console.WriteLine($"{episodeFile} & season 不是数字");
continue;
}
if (!int.TryParse(episodeNode.InnerText, out var episode))
{
failedCount++;
Console.WriteLine($"{episodeFile} & episode 不是数字");
continue;
}
// 设置 title
var titleNode = episodeXml.SelectSingleNode("//title");
if (titleNode == null)
{
failedCount++;
Console.WriteLine($"{episodeFile} & 未找到 title");
continue;
}
var json = await GetTmdbEpisode(
episodeFile,
tmdbId,
season,
episode);
if (json == null)
{
failedCount++;
continue;
}
var title = json["name"]?.ToString();
var overview = json["overview"]?.ToString();
var isUpdate = false;
if (!string.IsNullOrEmpty(title))
{
title = $"<![CDATA[{title}]]>";
if (titleNode.InnerXml != title)
{
isUpdate = true;
// 写入且不转义
titleNode.InnerXml = title;
}
}
var plotNode = episodeXml.SelectSingleNode("//plot");
if (!string.IsNullOrEmpty(overview) && plotNode != null)
{
overview = $"<![CDATA[{overview}]]>";
if (plotNode.InnerXml != overview)
{
isUpdate = true;
plotNode.InnerXml = overview;
}
}
if (!isUpdate)
{
skippedCount++;
Console.WriteLine($"{episodeFile} & 无更新");
continue;
}
successCount++;
// 添加一个已完成节点
if (completedNode == null)
{
episodeXml.DocumentElement?.AppendChild(episodeXml.CreateElement("completed"));
}
// 锁定
if (lockedNode == null)
{
lockedNode = episodeXml.CreateElement("lockdata");
episodeXml.DocumentElement?.AppendChild(lockedNode);
lockedNode.InnerText = "true";
}
else
{
lockedNode.InnerText = "true";
}
try
{
episodeXml.Save(episodeFile);
}
catch (Exception e)
{
Console.WriteLine(e);
continue;
}
Console.WriteLine($"{episodeFile} & {title} & {overview}");
await SetEpisodeInfo(episodeFile, tmdbId);
await Task.Delay(10000);
}
}
await WxNotify.SendCommonAsync($"ChineseNfoRegistry.Job() success: {successCount}, failed: {failedCount}, skipped: {skippedCount}");
async Task SetTvInfo(string tvShowFile,XmlDocument tvXml, int tmdbId)
{
// 检查 lockdata
var lockedNode = tvXml.SelectSingleNode("//lockdata");
if (lockedNode != null && lockedNode.InnerText == "true" && ignoreLocked == false)
{
skippedCount++;
Console.WriteLine($"{tvShowFile} & 已锁定");
return;
}
// 获取tv info
var tvJson = await GetTmdbTv(tvShowFile, tmdbId);
if (tvJson == null)
{
failedCount++;
return;
}
var title = tvJson["name"]?.ToString();
var overview = tvJson["overview"]?.ToString();
var poster =tvJson["poster_path"]?.ToString();
var genres = tvJson["genres"]
?.AsArray()
.Select(x => x?["name"]?.ToString())
.Where(x=> !string.IsNullOrEmpty(x))
.ToList();
if (!string.IsNullOrEmpty(title))
{
title = $"<![CDATA[{title}]]>";
var titleNode = tvXml.SelectSingleNode("//title");
if (titleNode != null)
{
if (titleNode.InnerXml != title)
{
titleNode.InnerXml = title;
successCount++;
}
}
var sorttitleNode = tvXml.SelectSingleNode("//sorttitle");
if (sorttitleNode != null)
{
if (sorttitleNode.InnerXml != title)
{
sorttitleNode.InnerXml = title;
successCount++;
}
}
}
if (!string.IsNullOrEmpty(overview))
{
overview = $"<![CDATA[{overview}]]>";
var plotNode = tvXml.SelectSingleNode("//plot");
if (plotNode != null)
{
if (plotNode.InnerXml != overview)
{
plotNode.InnerXml = overview;
successCount++;
}
}
var outlineNode = tvXml.SelectSingleNode("//outline");
if (outlineNode != null)
{
if (outlineNode.InnerXml != overview)
{
outlineNode.InnerXml = overview;
successCount++;
}
}
}
if(!string.IsNullOrEmpty(poster))
{
poster = $"https://image.tmdb.org/t/p/w1280/{poster}";
// 下载并覆盖海报
var posterPath = tvShowFile.Replace("tvshow.nfo", $"poster.{poster.Split('.').Last()}");
if (File.Exists(posterPath))
{
File.Delete(posterPath);
}
using var client = new HttpClient();
var response = await client.GetAsync(poster);
if (response.IsSuccessStatusCode)
{
var bytes = await response.Content.ReadAsByteArrayAsync();
await File.WriteAllBytesAsync(posterPath, bytes);
successCount++;
}
}
if (genres?.Any() == true)
{
// 删除原有的 <genre> 节点
var genreNodes = tvXml.SelectNodes("//genre");
if (genreNodes != null)
{
foreach (XmlNode genreNode in genreNodes)
{
tvXml.DocumentElement?.RemoveChild(genreNode);
}
}
// 添加新的 <genre> 节点
foreach (var genre in genres)
{
var genreNode = tvXml.CreateElement("genre");
genreNode.InnerText = genre!;
tvXml.DocumentElement?.AppendChild(genreNode);
}
successCount++;
}
if (successCount == 0)
{
skippedCount++;
Console.WriteLine($"{tvShowFile} & 无更新");
return;
}
// 锁定节点
if(lockedNode == null)
{
lockedNode = tvXml.CreateElement("lockdata");
tvXml.DocumentElement?.AppendChild(lockedNode);
lockedNode.InnerText = "true";
}
else
{
lockedNode.InnerText = "true";
}
// 添加一个已完成节点
var completedNode = tvXml.SelectSingleNode("//completed");
if (completedNode == null)
{
completedNode = tvXml.CreateElement("completed");
tvXml.DocumentElement?.AppendChild(completedNode);
}
completedNode.InnerText = "true";
try
{
tvXml.Save(tvShowFile);
}
catch (Exception e)
{
Console.WriteLine(e);
failedCount++;
}
}
async Task SetEpisodeInfo(string episodeFile, int tmdbId)
{
var episodeContent = await File.ReadAllTextAsync(episodeFile);
var episodeXml = new XmlDocument();
episodeXml.LoadXml(episodeContent);
// 判断有无 completed 节点
var completedNode = episodeXml.SelectSingleNode("//completed");
if (completedNode != null && ignoreCompleted == false)
{
skippedCount++;
Console.WriteLine($"{episodeFile} & 已完成");
return;
}
// 判断 locked == true
var lockedNode = episodeXml.SelectSingleNode("//lockdata");
if (lockedNode != null && lockedNode.InnerText == "true" && ignoreLocked == false)
{
skippedCount++;
Console.WriteLine($"{episodeFile} & 已锁定");
return;
}
// 读取 <season>1</season>
var seasonNode = episodeXml.SelectSingleNode("//season");
if (seasonNode == null)
{
failedCount++;
Console.WriteLine($"{episodeFile} & 未找到 season");
return;
}
// 读取 <episode>1</episode>
var episodeNode = episodeXml.SelectSingleNode("//episode");
if (episodeNode == null)
{
failedCount++;
Console.WriteLine($"{episodeFile} & 未找到 episode");
return;
}
if (!int.TryParse(seasonNode.InnerText, out var season))
{
failedCount++;
Console.WriteLine($"{episodeFile} & season 不是数字");
return;
}
if (!int.TryParse(episodeNode.InnerText, out var episode))
{
failedCount++;
Console.WriteLine($"{episodeFile} & episode 不是数字");
return;
}
// 设置 title
var titleNode = episodeXml.SelectSingleNode("//title");
if (titleNode == null)
{
failedCount++;
Console.WriteLine($"{episodeFile} & 未找到 title");
return;
}
var json = await GetTmdbEpisode(
episodeFile,
tmdbId,
season,
episode);
if (json == null)
{
failedCount++;
return;
}
var title = json["name"]?.ToString();
var overview = json["overview"]?.ToString();
var isUpdate = false;
if (!string.IsNullOrEmpty(title))
{
title = $"<![CDATA[{title}]]>";
if (titleNode.InnerXml != title)
{
isUpdate = true;
// 写入且不转义
titleNode.InnerXml = title;
}
}
var plotNode = episodeXml.SelectSingleNode("//plot");
if (!string.IsNullOrEmpty(overview) && plotNode != null)
{
overview = $"<![CDATA[{overview}]]>";
if (plotNode.InnerXml != overview)
{
isUpdate = true;
plotNode.InnerXml = overview;
}
}
if (!isUpdate)
{
skippedCount++;
Console.WriteLine($"{episodeFile} & 无更新");
return;
}
successCount++;
// 添加一个已完成节点
if (completedNode == null)
{
episodeXml.DocumentElement?.AppendChild(episodeXml.CreateElement("completed"));
}
// 锁定
if (lockedNode == null)
{
lockedNode = episodeXml.CreateElement("lockdata");
episodeXml.DocumentElement?.AppendChild(lockedNode);
lockedNode.InnerText = "true";
}
else
{
lockedNode.InnerText = "true";
}
try
{
episodeXml.Save(episodeFile);
}
catch (Exception e)
{
Console.WriteLine(e);
return;
}
Console.WriteLine($"{episodeFile} & {title} & {overview}");
}
}
private async Task<JsonObject?> GetTmdbTv(
string path,
int tmdbId)
{
const string tvUrl = "/3/tv/{0}?api_key=e28e1bc408db7adefc8bacce225c5085&language=zh-CN";
var requestUrl = string.Format(tvUrl, tmdbId);
try
{
var response = await _client.GetAsync(requestUrl);
if (!response.IsSuccessStatusCode)
{
Console.WriteLine($"{requestUrl} & {path} & {response.StatusCode}");
return null;
}
var content = await response.Content.ReadAsStringAsync();
var json = JsonNode.Parse(content);
return json as JsonObject;
}
catch (Exception e)
{
Console.WriteLine($"{requestUrl} & {path} \r {e}");
return null;
}
}
private async Task<JsonObject?> GetTmdbEpisode(
string path,
int tmdbId,