/*
 * Copyright (c) 2024 Proton AG
 *
 * This file is part of ProtonVPN.
 *
 * ProtonVPN is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * ProtonVPN is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with ProtonVPN.  If not, see <https://www.gnu.org/licenses/>.
 */

using ProtonVPN.UI.Tests.UiTools;
using ProtonVPN.UI.Tests.TestsHelper;

namespace ProtonVPN.UI.Tests.Robots;
public class UpsellCarrouselRobot
{
    protected Element UpgradeButton = Element.ByName("Upgrade");
    protected Element CloseButton = Element.ByAutomationId("Close");
    protected Element NextUpsellButton = Element.ByAutomationId("MoveToNextUpsellFeatureButton");
    protected Element BackUpsellButton = Element.ByAutomationId("MoveToPreviousUpsellFeatureButton");
    protected Element ServersUpsellDescription = Element.ByName("Select any country from our worldwide network");
    protected Element SpeedUpsellTitle = Element.ByName("Browse at even higher speeds");
    protected Element StreamingUpsellTitle = Element.ByName("Stream your favorite movies and TV shows");
    protected Element NetshieldUpsellTitle = Element.ByName("Enjoy ad-free browsing");
    protected Element SecureCoreUpsellTitle = Element.ByName("Add another layer of protection to your connection");
    protected Element P2PUpsellTitle = Element.ByName("Unlock peer-to-peer downloads and file sharing (P2P)");
    protected Element P2PTorrentInProgressUpsellTitle = Element.ByName("P2P is not allowed on free VPN servers");
    protected Element StreamingInProgressUpsellTitle = Element.ByName("Upgrade to bypass VPN blocks");
    protected Element TenDevicesUpsellTitle = Element.ByName("Connect up to 10 devices at once");
    protected Element TorUpsellTitle = Element.ByName("Access the Tor network for extra privacy");
    protected Element SplitTunnelingUpsellTitle = Element.ByName("Get the best of both worlds");
    protected Element ProfilesUpsellTitle = Element.ByName("Get quick access to frequent connections");
    protected Element AdvancedSettingsUpsellTitle = Element.ByName("Unlock advanced VPN customization");

    public UpsellCarrouselRobot NextUpsell()
    {
        NextUpsellButton.Click();
        return this;
    }

    public UpsellCarrouselRobot GoBackUpsell()
    {
        BackUpsellButton.Click();
        return this;
    }

    public UpsellCarrouselRobot CloseModal()
    {
        CloseButton.Click(); 
        return this;
    }

    public class Verifications : UpsellCarrouselRobot
    {
        public Verifications IsServersUpsellDisplayed()
        {
            ServersUpsellDescription.WaitUntilDisplayed();
            UpgradeButton.WaitUntilDisplayed();
            return this;
        }

        public Verifications IsServersSpeedUpsellDisplayed()
        {
            SpeedUpsellTitle.WaitUntilDisplayed();
            UpgradeButton.WaitUntilDisplayed();
            return this;
        }

        public Verifications IsStreamingUpsellDisplayed()
        {
            StreamingUpsellTitle.WaitUntilDisplayed();
            UpgradeButton.WaitUntilDisplayed();
            return this;
        }

        public Verifications IsNetshieldUpsellDisplayed()
        {
            NetshieldUpsellTitle.WaitUntilDisplayed();
            UpgradeButton.WaitUntilDisplayed();
            return this;
        }

        public Verifications IsSecureCoreUpsellDisplayed()
        {
            SecureCoreUpsellTitle.WaitUntilDisplayed();
            UpgradeButton.WaitUntilDisplayed();
            return this;
        }

        public Verifications IsP2PUpsellDisplayed()
        {
            P2PUpsellTitle.WaitUntilDisplayed();
            UpgradeButton.WaitUntilDisplayed();
            return this;
        }

        public Verifications IsP2PTorrentInProgressUpsellDisplayed()
        {
            P2PTorrentInProgressUpsellTitle.WaitUntilDisplayed(TestConstants.ThirtySecondsTimeout);
            UpgradeButton.WaitUntilDisplayed();
            return this;
        }

        public Verifications IsStreamingInProgressUpsellDisplayed()
        {
            StreamingInProgressUpsellTitle.WaitUntilDisplayed(TestConstants.ThirtySecondsTimeout);
            UpgradeButton.WaitUntilDisplayed();
            return this;
        }

        public Verifications IsTenDevicesUpsellDisplayed()
        {
            TenDevicesUpsellTitle.WaitUntilDisplayed();
            UpgradeButton.WaitUntilDisplayed();
            return this;
        }

        public Verifications IsTorUpsellDisplayed()
        {
            TorUpsellTitle.WaitUntilDisplayed();
            UpgradeButton.WaitUntilDisplayed();
            return this;
        }

        public Verifications IsSplitTunnelingUpsellDisplayed()
        {
            SplitTunnelingUpsellTitle.WaitUntilDisplayed();
            UpgradeButton.WaitUntilDisplayed();
            return this;
        }

        public Verifications IsProfilesUpsellDisplayed()
        {
            ProfilesUpsellTitle.WaitUntilDisplayed();
            UpgradeButton.WaitUntilDisplayed();
            return this;
        }

        public Verifications IsAdvancedSettingsUpsellDisplayed()
        {
            AdvancedSettingsUpsellTitle.WaitUntilDisplayed();
            UpgradeButton.WaitUntilDisplayed();
            return this;
        }

        public Verifications IsUpgradeButtonDisplayed()
        {
            IsUpgradeButtonDisplayed();
            UpgradeButton.WaitUntilDisplayed();
            return this;
        }
    }

    public Verifications Verify => new();
}