using System.Collections.Generic;
using System.Linq;

using UnityEditor.Localization;

using Crowdin.Plugin.Models.Settings;
using Crowdin.Plugin.Server_API.Helpers;

namespace Crowdin.Plugin.Helpers.UnityLocalization
{
    public static class LocalizationTablesProvider
    {
        public static IEnumerable<AssetTableCollection> GetAssetTableCollections()
        {
            TablesSelection settings = SettingsHelper.GetAssetTablesSelection();

            if (settings.allSelected)
            {
                return LocalizationEditorSettings.GetAssetTableCollections();
            }
            else
            {
                return LocalizationEditorSettings.GetAssetTableCollections()
                    .Where(collection => settings.selectedTables
                        .Contains(collection.TableCollectionName));
            }
        }

        public static IEnumerable<StringTableCollection> GetStringTableCollections()
        {
            TablesSelection settings = SettingsHelper.GetStringTablesSelection();

            if (settings.allSelected)
            {
                return LocalizationEditorSettings.GetStringTableCollections();
            }
            else
            {
                return LocalizationEditorSettings.GetStringTableCollections()
                    .Where(collection => settings.selectedTables
                        .Contains(collection.TableCollectionName));
            }
        }
    }
}