diff options
Diffstat (limited to 'Api')
| -rw-r--r-- | Api/JCoverSharedController.cs | 168 | ||||
| -rw-r--r-- | Api/JCoverStaticProvider.cs | 40 | ||||
| -rw-r--r-- | Api/ScriptInjector.cs | 74 | ||||
| -rw-r--r-- | Api/coverscript.js | 93 |
4 files changed, 375 insertions, 0 deletions
diff --git a/Api/JCoverSharedController.cs b/Api/JCoverSharedController.cs new file mode 100644 index 0000000..dea10c4 --- /dev/null +++ b/Api/JCoverSharedController.cs @@ -0,0 +1,168 @@ +using System; +using System.Collections.Generic; +using System.ComponentModel.DataAnnotations; +using System.Linq; +using System.Text.Json; +using System.Text.Json.Nodes; +using System.Threading; +using System.Threading.Tasks; +using System.Web; +using Jellyfin.Api; +using MediaBrowser.Controller; +using MediaBrowser.Controller.Dto; +using MediaBrowser.Controller.Entities; +using MediaBrowser.Controller.Entities.TV; +using MediaBrowser.Controller.Library; +using MediaBrowser.Controller.Providers; +using MediaBrowser.Model.Entities; +using Microsoft.AspNetCore.Authorization; +using Microsoft.AspNetCore.Http; +using Microsoft.AspNetCore.Mvc; +using Microsoft.Extensions.Logging; + +namespace Jellyfin.Plugin.JCoverXtremePro.Api; + +[ApiController] +[Route("JCoverXtreme")] +// [Authorize(Policy = "RequiresElevation")] +public class JCoverSharedController : BaseJellyfinApiController +{ + private readonly IProviderManager _providerManager; + private readonly ILibraryManager _libraryManager; + + /// <summary> + /// This key is appended to image URLs to inform the frontend about the presence of a potential mass-download. + /// Keep in sync with coverscript.js#URL_META_KEY. + /// </summary> + public static string URL_META_KEY = "JCoverXtremeProMeta"; + + public JCoverSharedController( + IProviderManager providerManager, + IServerApplicationPaths applicationPaths, + ILibraryManager libraryManager) + { + _providerManager = providerManager; + _libraryManager = libraryManager; + } + + public static string AppendUrlMeta(string baseUrl, string key, string value) + { + return baseUrl + (baseUrl.Contains('?', StringComparison.InvariantCulture) ? "&" : "?") + + HttpUtility.UrlEncode(key) + "=" + HttpUtility.UrlEncode(value); + } + + public class SetMeta + { + public Guid seriesId { get; set; } + public string setId { get; set; } + } + + public static string PackSetInfo(string baseUrl, Series series, POJO.Set set) + { + return AppendUrlMeta( + baseUrl, + URL_META_KEY, + JsonSerializer.Serialize(new SetMeta + { + setId = set.id, + seriesId = series.Id + })); + } + + private static Dictionary<(int, int), POJO.File> CreateCoverFileLUT(POJO.Set set) + { + Dictionary<string, (int, int)> episodeIdToEpisodeNumber = new(); + foreach (var showSeason in set.show.seasons) + { + episodeIdToEpisodeNumber[showSeason.id] = (showSeason.season_number, -10); + foreach (var showEpisode in showSeason.episodes) + { + episodeIdToEpisodeNumber[showEpisode.id] = (showSeason.season_number, showEpisode.episode_number); + } + } + + Dictionary<(int, int), POJO.File> episodeNumberToFile = new(); + foreach (var file in set.files) + { + string id = string.Empty; + if (file.episode_id != null) + { + id = file.episode_id.id; + } + + if (file.season_id != null) + { + id = file.season_id.id; + } + + var tup = episodeIdToEpisodeNumber.GetValueOrDefault(id); + episodeNumberToFile[tup] = file; + } + + return episodeNumberToFile; + } + + [HttpPost("DownloadSeries")] + [ProducesResponseType(StatusCodes.Status204NoContent)] + public async Task<ActionResult> DownloadEntireSeriesImages( + [FromBody, Required] JsonObject setMeta + ) + { + // TODO: handle missing fields, local seasons missing, series missing, etc. + var setMetaObj = JsonSerializer.Deserialize<SetMeta>(setMeta); + var series = _libraryManager.GetItemById(setMetaObj.seriesId) as Series; + var jsonMeta = await MediuxDownloader.instance.GetMediuxMetadata($"https://mediux.pro/sets/{setMetaObj.setId}") + .ConfigureAwait(false); + var set = JsonSerializer.Deserialize<POJO.SetData>(jsonMeta).set; + var files = CreateCoverFileLUT(set); + foreach (var item in series.GetSeasons(null, new DtoOptions(true))) + { + var season = item as Season; + var seasonNumber = season.GetLookupInfo().IndexNumber.Value; + Plugin.Logger.LogInformation($"Season id: {seasonNumber}:"); + await TryDownloadEpisode(season, files, (seasonNumber, -10)) + .ConfigureAwait(false); + foreach (var itemAgain in season.GetEpisodes()) + { + var episode = itemAgain as Episode; + var episodeNumber = episode.GetLookupInfo().IndexNumber.Value; + Plugin.Logger.LogInformation($" * Episode id: {episodeNumber} {episode.Name}"); + await TryDownloadEpisode(episode, files, (seasonNumber, episodeNumber)) + .ConfigureAwait(false); + } + } + + return Empty; + } + + private async Task TryDownloadEpisode( + BaseItem item, + Dictionary<(int, int), POJO.File> files, + (int, int) episodeNumber) + { + POJO.File file; + if (files.TryGetValue(episodeNumber, out file)) + { + Plugin.Logger.LogInformation($" Found cover: {file.downloadUrl}"); + await SaveCoverFileForItem(item, file.downloadUrl) + .ConfigureAwait(false); + } + } + + private async Task SaveCoverFileForItem( + BaseItem item, + string downloadUrl + ) + { + await _providerManager.SaveImage( + item, downloadUrl, + // Note: this needs to be updated if SeriesImageProvider ever supports more image types + ImageType.Primary, + null, + CancellationToken.None + ).ConfigureAwait(false); + + await item.UpdateToRepositoryAsync(ItemUpdateType.ImageUpdate, CancellationToken.None) + .ConfigureAwait(false); + } +}
\ No newline at end of file diff --git a/Api/JCoverStaticProvider.cs b/Api/JCoverStaticProvider.cs new file mode 100644 index 0000000..e4587ba --- /dev/null +++ b/Api/JCoverStaticProvider.cs @@ -0,0 +1,40 @@ +using Microsoft.Extensions.Logging; + +namespace Jellyfin.Plugin.JCoverXtremePro.Api; + +using System.Reflection; +using Microsoft.AspNetCore.Http; +using Microsoft.AspNetCore.Mvc; + +/// <summary> +/// Static file server for the JavaScript snippet injected by <see cref="ScriptInjector"/> +/// </summary> +[ApiController] +[Route("JCoverXtremeProStatic")] +public class JCoverStaticProvider : ControllerBase +{ + private readonly Assembly assembly; + private readonly string scriptPath; + + public JCoverStaticProvider() + { + assembly = Assembly.GetExecutingAssembly(); + scriptPath = GetType().Namespace + ".coverscript.js"; + } + + [HttpGet("ClientScript")] + [ProducesResponseType(StatusCodes.Status200OK)] + [ProducesResponseType(StatusCodes.Status404NotFound)] + [Produces("application/javascript")] + public ActionResult GetClientScript() + { + Plugin.Logger.LogInformation($"Requesting ClientScript {scriptPath}"); + var scriptStream = assembly.GetManifestResourceStream(scriptPath); + if (scriptStream == null) + { + return NotFound(); + } + + return File(scriptStream, "application/javascript"); + } +}
\ No newline at end of file diff --git a/Api/ScriptInjector.cs b/Api/ScriptInjector.cs new file mode 100644 index 0000000..d6b1865 --- /dev/null +++ b/Api/ScriptInjector.cs @@ -0,0 +1,74 @@ +namespace Jellyfin.Plugin.JCoverXtremePro.Api; + +using System; +using System.IO; +using System.Text.RegularExpressions; +using MediaBrowser.Common.Configuration; +using MediaBrowser.Controller.Configuration; +using Microsoft.Extensions.Logging; + +/// <summary> +/// Utility for injecting a JavaScript script tag into the Jellyfin web frontend. +/// </summary> +public static class ScriptInjector +{ + public static void PerformInjection( + IApplicationPaths applicationPaths, + IServerConfigurationManager configurationManager + ) + { + var indexHtmlFilePath = Path.Combine(applicationPaths.WebPath, "index.html"); + if (!File.Exists(indexHtmlFilePath)) + { + Plugin.Logger.LogWarning("Could not find index html file"); + return; + } + + var html = File.ReadAllText(indexHtmlFilePath); + var snippet = GetInjectedSnippet(GetHTTPBasePath(configurationManager)); + if (html.Contains(snippet, StringComparison.InvariantCulture)) + { + Plugin.Logger.LogInformation("Not injecting existing HTML snippet."); + return; + } + + html = Regex.Replace(html, $"<script[^>]*guid=\"{Plugin.GUID}\"[^>]*></script>", string.Empty); + var bodyEnd = html.LastIndexOf("</body>", StringComparison.InvariantCulture); + if (bodyEnd < 0) + { + Plugin.Logger.LogError("Could not find end of body to inject script"); + return; + } + + html = html.Insert(bodyEnd, snippet); + try + { + File.WriteAllText(indexHtmlFilePath, html); + Plugin.Logger.LogInformation("Injected index.html"); + } + catch (Exception e) + { + Plugin.Logger.LogError(e, "Failed to write patched index.html"); + } + } + + public static string GetHTTPBasePath(IServerConfigurationManager configurationManager) + { + var networkConfig = configurationManager.GetConfiguration("network"); + var configType = networkConfig.GetType(); + var baseUrlField = configType.GetProperty("BaseUrl"); + var baseUrl = baseUrlField!.GetValue(networkConfig)!.ToString()!.Trim('/'); + return baseUrl; + } + + public static string GetScriptUrl(string basePath) + { + return basePath + "/JCoverXtremeProStatic/ClientScript"; + } + + public static string GetInjectedSnippet(string basePath) + { + return + $"<script guid=\"{Plugin.GUID}\" plugin=\"{Plugin.Instance!.Name}\" src=\"{GetScriptUrl(basePath)}\" defer></script>"; + } +}
\ No newline at end of file diff --git a/Api/coverscript.js b/Api/coverscript.js new file mode 100644 index 0000000..a4f0c04 --- /dev/null +++ b/Api/coverscript.js @@ -0,0 +1,93 @@ +(function () { + /** + * + * @param {HTMLElement} element + * @param {string} selector + * @returns {HTMLElement | null} + */ + function findParent(element, selector) { + let p = element + while (p) { + if (p.matches(selector)) return p + p = p.parentNode + } + return null + } + + const injectionMarker = "JCoverXtremePro-injection-marker"; + + /** + * + * @param {HTMLElement} cloneFrom + * @param {string} setMeta + * @return {HTMLElement} + */ + function createDownloadSeriesButton( + cloneFrom, + setMeta) { + /*<button is="paper-icon-button-light" class="btnDownloadRemoteImage autoSize paper-icon-button-light" raised"="" title="Download"><span class="material-icons cloud_download" aria-hidden="true"></span></button>*/ + //import LayersIcon from '@mui/icons-material/Layers'; + //import CloudDownloadIcon from '@mui/icons-material/CloudDownload'; + const element = document.createElement("button") + element.classList.add(...cloneFrom.classList) + element.classList.add(injectionMarker) + element.title = "Download Series" + const icon = document.createElement("span") + icon.classList.add("material-icons", "burst_mode") + icon.setAttribute("aria-hidden", "true") + element.appendChild(icon) + element.addEventListener("click", ev => { + ev.preventDefault() + console.log("Executing mass covering event! We will try to download the entirety of set " + setMeta) + fetch("/JCoverXtreme/DownloadSeries", + { + method: 'POST', + body: setMeta, + headers: { + "content-type": "application/json" + } + }).then(console.log) // TODO: check out the root somehow. for now just assume / + }) + return element + } + + /** + * Keep in sync with JCoverSharedController.URL_META_KEY + * @type {string} + */ + const URL_META_KEY = "JCoverXtremeProMeta" + + /** + * Extract the JCoverXtremePro metadata from an image url. + * + * @param {string} url + * @return {string} + */ + function extractSetMeta(url) { + return new URL(url).searchParams.get(URL_META_KEY) + } + + const observer = new MutationObserver(() => { + console.log("JCoverXtremePro observation was triggered!") + console.log("Listing all download buttons") + /** + * @type {NodeListOf<Element>} + */ + const buttons = document.querySelectorAll(".imageEditorCard .cardFooter .btnDownloadRemoteImage") + + buttons.forEach(element => { + const downloadRowContainer = findParent(element, ".cardText") + const cardContainer = findParent(element, ".cardBox") + const cardImage = cardContainer.querySelector("a.cardImageContainer[href]") + const setMeta = extractSetMeta(cardImage.href) + if (!setMeta) return; + if (downloadRowContainer.querySelector(`.${injectionMarker}`)) return + downloadRowContainer.appendChild(createDownloadSeriesButton(element, setMeta)) + }) + + }) + observer.observe(document.body, {// TODO: selectively observe the body if at all possible + subtree: true, + childList: true, + }); +})()
\ No newline at end of file |