OF-DL/OF DL/Services/FileNameService.cs

using System.Reflection;
using HtmlAgilityPack;

namespace OF_DL.Services;

public class FileNameService(IAuthService authService) : IFileNameService
{
    public async Task<Dictionary<string, string>> GetFilename(object obj1, object obj2, object obj3,
        List<string> selectedProperties, string username, Dictionary<string, long> users = null)
    {
        Dictionary<string, string> values = new();
        Type type1 = obj1.GetType();
        Type type2 = obj2.GetType();
        PropertyInfo[] properties1 = type1.GetProperties();
        PropertyInfo[] properties2 = type2.GetProperties();

        foreach (string propertyName in selectedProperties)
        {
            if (propertyName.Contains("media"))
            {
                object drmProperty = null;
                object fileProperty = GetNestedPropertyValue(obj2, "files");
                if (fileProperty != null)
                {
                    drmProperty = GetNestedPropertyValue(obj2, "files.drm");
                }

                if (fileProperty != null && drmProperty != null && propertyName == "mediaCreatedAt")
                {
                    object mpdurl = GetNestedPropertyValue(obj2, "files.drm.manifest.dash");
                    object policy = GetNestedPropertyValue(obj2, "files.drm.signature.dash.CloudFrontPolicy");
                    object signature = GetNestedPropertyValue(obj2, "files.drm.signature.dash.CloudFrontSignature");
                    object kvp = GetNestedPropertyValue(obj2, "files.drm.signature.dash.CloudFrontKeyPairId");
                    DateTime lastModified =
                        await DownloadService.GetDRMVideoLastModified(string.Join(",", mpdurl, policy, signature, kvp),
                            authService.CurrentAuth);
                    values.Add(propertyName, lastModified.ToString("yyyy-MM-dd"));
                    continue;
                }

                if ((fileProperty == null || drmProperty == null) && propertyName == "mediaCreatedAt")
                {
                    object source = GetNestedPropertyValue(obj2, "files.full.url");
                    if (source != null)
                    {
                        DateTime lastModified = await DownloadService.GetMediaLastModified(source.ToString());
                        values.Add(propertyName, lastModified.ToString("yyyy-MM-dd"));
                        continue;
                    }

                    object preview = GetNestedPropertyValue(obj2, "preview");
                    if (preview != null)
                    {
                        DateTime lastModified = await DownloadService.GetMediaLastModified(preview.ToString());
                        values.Add(propertyName, lastModified.ToString("yyyy-MM-dd"));
                        continue;
                    }
                }

                PropertyInfo? property = Array.Find(properties2,
                    p => p.Name.Equals(propertyName.Replace("media", ""), StringComparison.OrdinalIgnoreCase));
                if (property != null)
                {
                    object? propertyValue = property.GetValue(obj2);
                    if (propertyValue != null)
                    {
                        if (propertyValue is DateTime dateTimeValue)
                        {
                            values.Add(propertyName, dateTimeValue.ToString("yyyy-MM-dd"));
                        }
                        else
                        {
                            values.Add(propertyName, propertyValue.ToString());
                        }
                    }
                }
            }
            else if (propertyName.Contains("filename"))
            {
                string sourcePropertyPath = "files.full.url";
                object sourcePropertyValue = GetNestedPropertyValue(obj2, sourcePropertyPath);
                if (sourcePropertyValue != null)
                {
                    Uri uri = new(sourcePropertyValue.ToString());
                    string filename = Path.GetFileName(uri.LocalPath);
                    values.Add(propertyName, filename.Split(".")[0]);
                }
                else
                {
                    string propertyPath = "files.drm.manifest.dash";
                    object nestedPropertyValue = GetNestedPropertyValue(obj2, propertyPath);
                    if (nestedPropertyValue != null)
                    {
                        Uri uri = new(nestedPropertyValue.ToString());
                        string filename = Path.GetFileName(uri.LocalPath);
                        values.Add(propertyName, filename.Split(".")[0] + "_source");
                    }
                }
            }
            else if (propertyName.Contains("username"))
            {
                if (!string.IsNullOrEmpty(username))
                {
                    values.Add(propertyName, username);
                }
                else
                {
                    string propertyPath = "id";
                    object nestedPropertyValue = GetNestedPropertyValue(obj3, propertyPath);
                    if (nestedPropertyValue != null)
                    {
                        values.Add(propertyName,
                            users.FirstOrDefault(u => u.Value == Convert.ToInt32(nestedPropertyValue.ToString())).Key);
                    }
                }
            }
            else if (propertyName.Contains("text", StringComparison.OrdinalIgnoreCase))
            {
                PropertyInfo property = Array.Find(properties1,
                    p => p.Name.Equals(propertyName, StringComparison.OrdinalIgnoreCase));
                if (property != null)
                {
                    object propertyValue = property.GetValue(obj1);
                    if (propertyValue != null)
                    {
                        HtmlDocument pageDoc = new();
                        pageDoc.LoadHtml(propertyValue.ToString());
                        string str = pageDoc.DocumentNode.InnerText;
                        if (str.Length > 100) // todo: add length limit to config
                        {
                            str = str.Substring(0, 100);
                        }

                        values.Add(propertyName, str);
                    }
                }
            }
            else
            {
                PropertyInfo property = Array.Find(properties1,
                    p => p.Name.Equals(propertyName, StringComparison.OrdinalIgnoreCase));
                if (property != null)
                {
                    object propertyValue = property.GetValue(obj1);
                    if (propertyValue != null)
                    {
                        if (propertyValue is DateTime dateTimeValue)
                        {
                            values.Add(propertyName, dateTimeValue.ToString("yyyy-MM-dd"));
                        }
                        else
                        {
                            values.Add(propertyName, propertyValue.ToString());
                        }
                    }
                }
            }
        }

        return values;
    }

    public async Task<string> BuildFilename(string fileFormat, Dictionary<string, string> values)
    {
        foreach (KeyValuePair<string, string> kvp in values)
        {
            string placeholder = "{" + kvp.Key + "}";
            fileFormat = fileFormat.Replace(placeholder, kvp.Value);
        }

        return RemoveInvalidFileNameChars($"{fileFormat}");
    }

    private static object GetNestedPropertyValue(object source, string propertyPath)
    {
        object value = source;
        foreach (string propertyName in propertyPath.Split('.'))
        {
            PropertyInfo property = value.GetType().GetProperty(propertyName) ??
                                    throw new ArgumentException($"Property '{propertyName}' not found.");
            value = property.GetValue(value);
        }

        return value;
    }

    private static string RemoveInvalidFileNameChars(string fileName) => string.IsNullOrEmpty(fileName)
        ? fileName
        : string.Concat(fileName.Split(Path.GetInvalidFileNameChars()));
}