package com.amazon.tests;

import java.util.ArrayList;
import java.util.HashSet;
import java.util.List;
import java.util.Set;

import org.openqa.selenium.By;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.WebElement;
import org.openqa.selenium.chrome.ChromeDriver;
import org.testng.annotations.AfterClass;
import org.testng.annotations.BeforeClass;
import org.testng.annotations.Test;

public class Grabber {
    private WebDriver driver;
    private List<String> internalLinks = new ArrayList<String>();
    private Set<String> noDuplicates = new HashSet<>();
    private List<WebElement> linksOnPage = new ArrayList<>();
    private List<String> listOld = new ArrayList<>();
    private List<String> listNew = new ArrayList<>();
    private List<String> listTemp = new ArrayList<>();

    @BeforeClass
    public void setup() {
        System.setProperty("webdriver.chrome.driver","C:/TEST/LIB/chromedriver.exe");
        driver = new ChromeDriver();
    }

    @AfterClass
    public void teardown() {
        driver.close();
    }

    @Test
    public void grabber()
    {
        String hrefvalue = null;
        boolean check = false;


        driver.get("https://www.redbullmobile.pl/");
        linksOnPage = driver.findElements(By.tagName("a"));


        for (int i = 0; i<linksOnPage.size(); i++)
        {
            hrefvalue = linksOnPage.get(i).getAttribute("href");

            if(hrefvalue != null)
            {
                if(hrefvalue.contains("redbullmobile") && !hrefvalue.contains("facebook") && !hrefvalue.contains("redbullmobile.play.pl") && !hrefvalue.contains("redbullmobilecollect") && !hrefvalue.contains("fb.com") && !hrefvalue.contains(".pdf"))
                {
                    noDuplicates.add(linksOnPage.get(i).getAttribute("href"));
                }
            }
        }
        listOld.addAll(noDuplicates);
        hrefvalue = null;
        linksOnPage.clear();
        noDuplicates.clear();


    do
    {
        listTemp.clear();
        for (int i = 0; i<listOld.size(); i++)
        {
            driver.get(listOld.get(i));
            linksOnPage = driver.findElements(By.tagName("a"));
            for (int i1 = 0; i1<linksOnPage.size(); i1++)
            {
                hrefvalue = linksOnPage.get(i1).getAttribute("href");
                if(hrefvalue != null)
                {
                    if(hrefvalue.contains("redbullmobile") && !hrefvalue.contains("facebook") && !hrefvalue.contains("redbullmobile.play.pl") && !hrefvalue.contains("redbullmobilecollect") && !hrefvalue.contains("fb.com") && !hrefvalue.contains(".pdf"))
                    {
                        noDuplicates.add(linksOnPage.get(i1).getAttribute("href"));
                    }
                }
            }
        }
        //When all links from listOld are visited compare listOld and listNew, every link listNew that's not on List is added to listTemp.

        //Links from ListTemp and ListOld are added to ListFinal
        internalLinks.addAll(listOld);
        internalLinks.addAll(listTemp);
        //listOld and listNew are cleared, listTemp content is added to listOld
        listOld.clear();
        listNew.clear();
        listOld.addAll(listTemp);
        //Go back to 3rd step
        //Program end when at then end of 6th step listTemp = null (so simple do...while with check, check changes to true when at the end of 6th step listTemp = null)
        if (listTemp.isEmpty())
        {
            check = true;
        }
    }while(check);
    }
}