lampac-ukraine/LME.StreamData/Controller.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Threading.Tasks;
using System.Web;
using LME.StreamData.Models;
using Microsoft.AspNetCore.Mvc;
using Shared;
using Shared.Engine;
using Shared.Models;
using Shared.Models.Online.Settings;
using Shared.Models.Templates;

namespace LME.StreamData.Controllers
{
    public class Controller : BaseOnlineController
    {
        ProxyManager proxyManager;

        public Controller() : base(ModInit.Settings)
        {
            proxyManager = new ProxyManager(ModInit.StreamDataSettings);
        }

        /// <summary>
        /// Головний ендпоінт модуля StreamData
        /// Працює виключно через TMDB ID (параметр id)
        /// </summary>
        [HttpGet]
        [Route("lite/lme_streamdata")]
        async public Task<ActionResult> Index(long id, string imdb_id, long kinopoisk_id, string title, string original_title, string original_language, int year, string source, int serial, string account_email, string t, int s = -1, int e = -1, bool play = false, bool rjson = false, string href = null, bool checksearch = false)
        {
            await UpdateService.ConnectAsync(host);

            var init = loadKit(ModInit.StreamDataSettings);
            if (!init.enable)
                return Forbid();

            var invoke = new StreamDataInvoke(init, hybridCache, OnLog, proxyManager, httpHydra);

            // checksearch — перевірка доступності
            if (checksearch)
            {
                if (!IsCheckOnlineSearchEnabled())
                    return OnError("lme_streamdata", refresh_proxy: true);

                if (id > 0)
                    return Content("data-json=", "text/plain; charset=utf-8");

                return OnError("lme_streamdata", refresh_proxy: true);
            }

            // play — повернути стрім для конкретного епізоду (call метод)
            if (play)
            {
                return await HandlePlay(invoke, init, id, s, e, title, original_title);
            }

            // Фільм
            if (serial != 1)
            {
                return await HandleMovie(invoke, init, id, title, original_title, rjson);
            }

            // Серіал
            return await HandleSerial(invoke, init, id, title, original_title, s, e, t, rjson);
        }

        /// <summary>
        /// Обробка фільму: отримуємо всі stream_urls та показуємо їх
        /// </summary>
        private async Task<ActionResult> HandleMovie(StreamDataInvoke invoke, OnlinesSettings init, long tmdbId, string title, string originalTitle, bool rjson)
        {
            var response = await invoke.GetMovie(tmdbId);
            if (response?.data?.stream_urls == null || response.data.stream_urls.Count == 0)
                return OnError("lme_streamdata", refresh_proxy: true);

            var streamUrls = response.data.stream_urls;
            var subs = CollectSubtitles(response);

            var displayTitle = !string.IsNullOrEmpty(title) ? title : (!string.IsNullOrEmpty(originalTitle) ? originalTitle : response.data.title);
            var tpl = new MovieTpl(displayTitle, originalTitle);

            int index = 1;
            foreach (var streamUrl in streamUrls)
            {
                if (string.IsNullOrWhiteSpace(streamUrl))
                    continue;

                string hostname = StreamDataInvoke.ExtractHostname(streamUrl);
                string label = streamUrls.Count > 1 ? $"Джерело {index} ({hostname})" : hostname;
                string processedUrl = BuildStreamUrl(init, streamUrl);
                tpl.Append(label, processedUrl, subtitles: subs);
                index++;
            }

            if (tpl.data == null || tpl.data.Count == 0)
                return OnError("lme_streamdata", refresh_proxy: true);

            return Content(
                rjson ? tpl.ToJson() : tpl.ToHtml(),
                rjson ? "application/json; charset=utf-8" : "text/html; charset=utf-8"
            );
        }

        /// <summary>
        /// Обробка серіалу: eps структура → сезони → епізоди → play
        /// </summary>
        private async Task<ActionResult> HandleSerial(StreamDataInvoke invoke, OnlinesSettings init, long tmdbId, string title, string originalTitle, int s, int e, string t, bool rjson)
        {
            var response = await invoke.GetTvSeries(tmdbId);
            if (response?.data?.eps == null || response.data.eps.Count == 0)
                return OnError("lme_streamdata", refresh_proxy: true);

            var eps = response.data.eps;
            var seasons = eps.Keys
                .Select(k => int.TryParse(k, out int sn) ? sn : 0)
                .Where(sn => sn > 0)
                .OrderBy(sn => sn)
                .ToList();

            if (seasons.Count == 0)
                return OnError("lme_streamdata", refresh_proxy: true);

            // Якщо сезон не вибрано — показуємо список сезонів
            if (s <= 0)
            {
                var displayTitle = !string.IsNullOrEmpty(title) ? title : (!string.IsNullOrEmpty(originalTitle) ? originalTitle : response.data.title);
                var seasonTpl = new SeasonTpl(seasons.Count);
                foreach (var season in seasons)
                {
                    string seasonLink = $"{host}/lite/lme_streamdata?id={tmdbId}&serial=1&s={season}&title={HttpUtility.UrlEncode(displayTitle)}&original_title={HttpUtility.UrlEncode(originalTitle)}";
                    seasonTpl.Append($"Сезон {season}", seasonLink, season.ToString());
                }

                return Content(
                    rjson ? seasonTpl.ToJson() : seasonTpl.ToHtml(),
                    rjson ? "application/json; charset=utf-8" : "text/html; charset=utf-8"
                );
            }

            // Якщо вибрано сезон але не вибрано епізод — показуємо епізоди
            string seasonKey = s.ToString();
            if (!eps.ContainsKey(seasonKey) || eps[seasonKey] == null || eps[seasonKey].Count == 0)
                return OnError("lme_streamdata", refresh_proxy: true);

            if (e <= 0)
            {
                var episodes = eps[seasonKey]
                    .Select(ep => int.TryParse(ep, out int en) ? en : 0)
                    .Where(en => en > 0)
                    .OrderBy(en => en)
                    .ToList();

                if (episodes.Count == 0)
                    return OnError("lme_streamdata", refresh_proxy: true);

                var displayTitle = !string.IsNullOrEmpty(title) ? title : (!string.IsNullOrEmpty(originalTitle) ? originalTitle : response.data.title);
                var episodeTpl = new EpisodeTpl(episodes.Count);

                foreach (var episode in episodes)
                {
                    string episodeName = $"Епізод {episode}";
                    string callUrl = $"{host}/lite/lme_streamdata?id={tmdbId}&serial=1&s={s}&e={episode}&play=true&title={HttpUtility.UrlEncode(displayTitle)}&original_title={HttpUtility.UrlEncode(originalTitle)}";
                    episodeTpl.Append(episodeName, displayTitle, s.ToString(), episode.ToString("D2"), accsArgs(callUrl), "call");
                }

                return Content(
                    rjson ? episodeTpl.ToJson() : episodeTpl.ToHtml(),
                    rjson ? "application/json; charset=utf-8" : "text/html; charset=utf-8"
                );
            }

            // Якщо є play=true з s та e — це вже оброблено вище в play-гілці
            // Сюди не дійдемо, але на всяк випадок:
            return OnError("lme_streamdata", refresh_proxy: true);
        }

        /// <summary>
        /// Обробка play-запиту: API з season/episode → JSON зі стрімом
        /// </summary>
        private async Task<ActionResult> HandlePlay(StreamDataInvoke invoke, OnlinesSettings init, long tmdbId, int season, int episode, string title, string originalTitle)
        {
            if (tmdbId <= 0 || season <= 0 || episode <= 0)
                return OnError("lme_streamdata", refresh_proxy: true);

            var response = await invoke.GetEpisode(tmdbId, season, episode);
            if (response?.data?.stream_urls == null || response.data.stream_urls.Count == 0)
                return OnError("lme_streamdata", refresh_proxy: true);

            var firstUrl = response.data.stream_urls.FirstOrDefault(u => !string.IsNullOrWhiteSpace(u));
            if (string.IsNullOrEmpty(firstUrl))
                return OnError("lme_streamdata", refresh_proxy: true);

            var subs = CollectSubtitles(response);
            string displayTitle = !string.IsNullOrEmpty(title) ? title : (!string.IsNullOrEmpty(originalTitle) ? originalTitle : response.data.title);
            string streamUrl = BuildStreamUrl(init, firstUrl);

            return UpdateService.Validate(Content(
                VideoTpl.ToJson("play", streamUrl, displayTitle, subtitles: subs),
                "application/json; charset=utf-8"
            ));
        }

        /// <summary>
        /// Зібрати субтитри з відповіді API
        /// </summary>
        private SubtitleTpl CollectSubtitles(StreamDataResponse response)
        {
            if (response?.default_subs == null || response.default_subs.Count == 0)
                return null;

            var tpl = new SubtitleTpl(response.default_subs.Count);
            foreach (var sub in response.default_subs)
            {
                if (!string.IsNullOrWhiteSpace(sub?.url) && !string.IsNullOrWhiteSpace(sub?.lang))
                {
                    tpl.Append(sub.lang, sub.url);
                }
            }

            return tpl;
        }

        string BuildStreamUrl(OnlinesSettings init, string streamLink)
        {
            string link = StripLampacArgs(streamLink?.Trim());
            if (string.IsNullOrEmpty(link))
                return link;

            if (ApnHelper.IsEnabled(init))
            {
                if (ModInit.ApnHostProvided)
                    return ApnHelper.WrapUrl(init, link);

                var noApn = (OnlinesSettings)init.Clone();
                noApn.apnstream = false;
                noApn.apn = null;
                return HostStreamProxy(noApn, link);
            }

            return HostStreamProxy(init, link);
        }

        private static string StripLampacArgs(string url)
        {
            if (string.IsNullOrEmpty(url))
                return url;

            string cleaned = System.Text.RegularExpressions.Regex.Replace(
                url,
                @"([?&])(account_email|uid|nws_id)=[^&]*",
                "$1",
                System.Text.RegularExpressions.RegexOptions.IgnoreCase
            );

            cleaned = cleaned.Replace("?&", "?").Replace("&&", "&").TrimEnd('?', '&');
            return cleaned;
        }

        private static bool IsCheckOnlineSearchEnabled()
        {
            try
            {
                var onlineType = Type.GetType("Online.ModInit");
                if (onlineType == null)
                {
                    foreach (var asm in AppDomain.CurrentDomain.GetAssemblies())
                    {
                        onlineType = asm.GetType("Online.ModInit");
                        if (onlineType != null)
                            break;
                    }
                }
                var confField = onlineType?.GetField("conf", System.Reflection.BindingFlags.Public | System.Reflection.BindingFlags.Static);
                var conf = confField?.GetValue(null);
                var checkProp = conf?.GetType().GetProperty("checkOnlineSearch", System.Reflection.BindingFlags.Public | System.Reflection.BindingFlags.Instance);

                if (checkProp?.GetValue(conf) is bool enabled)
                    return enabled;
            }
            catch
            {
            }

            return true;
        }

        private static void OnLog(string message)
        {
            System.Console.WriteLine(message);
        }
    }
}