Subversion Repositories SmartDukaan

package com.smartdukaan.cron.scheduled;

import nu.pattern.OpenCV;

import org.apache.commons.io.FileUtils;

import org.apache.logging.log4j.LogManager;

import org.apache.logging.log4j.Logger;

import org.opencv.core.Mat;

import org.opencv.imgcodecs.Imgcodecs;

import org.opencv.imgproc.Imgproc;

import org.openqa.selenium.*;

import org.openqa.selenium.chrome.ChromeDriver;

import org.openqa.selenium.chrome.ChromeOptions;

import org.openqa.selenium.interactions.Actions;

import org.openqa.selenium.support.ui.ExpectedConditions;

import org.openqa.selenium.support.ui.WebDriverWait;

import org.springframework.stereotype.Component;

import javax.imageio.ImageIO;

import java.awt.image.BufferedImage;

import java.io.File;

import java.time.LocalDate;

import java.time.format.DateTimeFormatter;

import java.util.HashMap;

import java.util.List;

import java.util.Map;

import java.util.concurrent.TimeUnit;

import java.util.stream.Collectors;

@Component

public class CheckOppoWarrantyTask {

    private static final Logger LOGGER = LogManager.getLogger(CheckOppoWarrantyTask.class);

    public Map<String, LocalDate> checkWarranty(List<String> imeis) {

        Map<String, LocalDate> dateMap = new HashMap<>();

        //System.out.println(System.getProperty("os.name"));

        //System.out.println(System.getProperty("os.arch"));

        OpenCV.loadShared();

        LOGGER.info("Initiating webdriver...");

        System.setProperty("webdriver.chrome.driver", "/usr/bin/chromedriver");

        Map<String, Object> prefsMap = new HashMap<>();

        prefsMap.put("profile.default_content_settings.popups", 0);

        prefsMap.put("download.prompt", false);

        prefsMap.put("download.directory_upgrade", true);

        //I need this

        //System.setProperty("webdriver.chrome.driver", "/usr/local/bin/chromedriver");

        ChromeOptions options = new ChromeOptions();

        options.setExperimentalOption("prefs", prefsMap);

        //options.setExperimentalOption("detach", true);

        options.addArguments("--headless");

        options.addArguments("--user-agent=\"Mozilla/5.0 (Windows Phone 10.0; Android 4.2.1; Microsoft; Lumia 640 XL LTE) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/42.0.2311.135 Mobile Safari/537.36 Edge/12.10166\"");

        options.addArguments("--no-sandbox");

        options.addArguments("start-maximized");

        options.addArguments("disable-infobars");

        options.addArguments("--disable-extensions");

        options.addArguments("--force-device-scale-factor=1");

        options.addArguments("--test-type");

        WebDriver driver = new ChromeDriver(options);

        driver.manage().timeouts().setScriptTimeout(10, TimeUnit.SECONDS);

        driver.manage().window().setSize(new Dimension(1600, 900));

        driver.manage().window().maximize();

        // Deleting all the cookies

        driver.manage().deleteAllCookies();

        // Specifiying pageLoadTimeout and Implicit wait

        driver.manage().timeouts().pageLoadTimeout(30, TimeUnit.SECONDS);

        driver.manage().timeouts().implicitlyWait(15, TimeUnit.SECONDS);

        int mainCount = 0;

        List<String> remainingImeis = imeis;

        WebDriverWait wait10Sec = new WebDriverWait(driver, 10);

        WebDriverWait wait5Sec = new WebDriverWait(driver, 5);

        WebElement slideButton;

        Actions actionProvider = new Actions(driver);

        do {

            driver.get("https://support.oppo.com/in/warranty-check/");

            for (int i = 0; i < remainingImeis.size(); i++) {

                String imei = remainingImeis.get(i);

                try {

                    System.out.println("Starting fresh with new IMEI " + imei);

                    if (driver.findElement(By.className("el-input__inner")).getAttribute("value").length() > 0) {

                        System.out.println("Darn.. leaving" + driver.findElement(By.className("el-input__inner")).getAttribute("value"));

                        driver.get("https://support.oppo.com/in/warranty-check/");

                        break;

                    }

                    driver.findElement(By.className("el-input__inner")).sendKeys(imei);

                    driver.findElement(By.className("el-button--primary")).click();

                    // Perform click-and-hold action on the element

                    int counter = 1;

                    boolean captchaNotBroken = true;

                    do {

                        String fileName = "/tmp/oppo-simple-i-" + counter + " " + System.currentTimeMillis() + ".png";

                        try {

                            System.out.println("Starting Do");

                            wait10Sec.until(ExpectedConditions.visibilityOfElementLocated(By.id("dx_captcha_basic_wrapper_1")));

                            wait10Sec.until(ExpectedConditions.visibilityOfElementLocated(By.id("dx_captcha_basic_slider_1")));

                            slideButton = driver.findElement(By.id("dx_captcha_basic_slider_1"));

                            System.out.println("Slide button - " + slideButton);

                            actionProvider.moveToElement(slideButton).clickAndHold().moveByOffset(10, 0).perform();

                        } catch (Exception e) {

                            System.out.println(e.getMessage());

                            driver.get("https://support.oppo.com/in/warranty-check/");

                            break;

                        }

                        Mat circles = getMatCircles(driver, fileName);

                        if (circles.cols() == 2) {

                            double firstCircleX = circles.get(0, 0)[0];

                            double secondCircleX = circles.get(0, 1)[0];

                            double distance = Math.abs(firstCircleX - secondCircleX);

                            System.out.println("Starting Checkdistance method");

                            int moveBy = checkDistance(driver, wait10Sec, actionProvider, distance);

                            if (moveBy > 0) {

                                System.out.println("Ending Checkdistance method " + moveBy);

                                actionProvider.moveByOffset(moveBy, 0).release().perform();

                                System.out.println("Move click performed");

                                try {

                                    try {

                                        wait5Sec.until(ExpectedConditions.visibilityOfElementLocated(By.className("dx_captcha_basic_bar-inform")));

                                        System.out.println("Failed = " + counter);

                                    } catch (Exception notFailedException) {

                                        System.out.println("Success  at attempt " + counter);

                                        List<WebElement> warrantyDateElements = driver.findElements(By.className("warranty-service--result_info__dateValue"));

                                        if (warrantyDateElements.size() > 0) {

                                            String warrantyDate = warrantyDateElements.get(0).getText();

                                            System.out.println("Date --  " + warrantyDate);

                                            try {

                                                dateMap.put(imei, LocalDate.parse(warrantyDate, DateTimeFormatter.ofPattern("yyyy/MM/dd")).minusYears(1));

                                            } catch (Exception e) {

                                                dateMap.put(imei, null);

                                            }

                                        } else {

                                            System.out.println("Could not capture date");

                                            dateMap.put(imei, null);

                                        }

                                        WebElement checkAgainButton = driver.findElement(By.className("van-button--normal"));

                                        captchaNotBroken = false;

                                        checkAgainButton.click();

                                        break;

                                    }

                                } catch (Throwable e) {

                                    System.out.println("Failed = " + counter + " ----------" + e.getMessage());

                                }

                            }

                        }

                        System.out.println("After circles conditions");

                        wait10Sec.until(ExpectedConditions.visibilityOfElementLocated(By.id("dx_captcha_basic_wrapper_1")));

                        wait10Sec.until(ExpectedConditions.visibilityOfElementLocated(By.id("dx_captcha_basic_slider_1")));

                        slideButton = driver.findElement(By.id("dx_captcha_basic_slider_1"));

                        actionProvider.moveToElement(slideButton).doubleClick().perform();

                        System.out.println("Ending Do" + counter++);

                    } while (captchaNotBroken && counter <= 20);

                    System.out.println("Counter reached upto 20");

                } catch (Exception e) {

                    System.out.println("Caught unknown - for imei " + imei);

                    e.printStackTrace();

                }

                driver.get("https://support.oppo.com/in/warranty-check/");

            }

            mainCount++;

            remainingImeis = imeis.stream().filter(x -> !dateMap.containsKey(x)).collect(Collectors.toList());

        } while (mainCount < 5);

        driver.close();

        driver.quit();

        return dateMap;

    }

    private int checkDistance(WebDriver driver, WebDriverWait wait3Sec, Actions actionProvider, double distance) throws Exception {

        //actionProvider.moveToElement(slideButton).clickAndHold().moveByOffset(1, 0).perform();

        actionProvider.moveByOffset(15, 0).perform();//

        String fileName2 = "/tmp/" + Thread.currentThread().getName() + "-moved.png";

        Mat movedCircles = getMatCircles(driver, fileName2);

        if (movedCircles.cols() != 2) return 0;

        double movedFirst = movedCircles.get(0, 0)[0];

        double movedSecond = movedCircles.get(0, 1)[0];

        //System.out.println("Test Offset Move " + testMove);

        System.out.println("Found Moved " + movedFirst + " ," + movedSecond);

        double movedDistance = Math.abs(movedFirst - movedSecond);

        double moved = distance - movedDistance;

		/*System.out.println("Original distance " + distance);

		System.out.println("Moved Distance " + movedDistance);

		System.out.println("Moved " + moved);*/

        if (moved == 0) return 0;

        int moveBy = (int) ((movedDistance) * (15 / moved));

        //int moveBy = (int) distance - 21;

        System.out.println("Move by = " + moveBy);

        return moveBy;

    }

    private Mat getMatCircles(WebDriver driver, String fileName) throws Exception {

        File screenshot = ((TakesScreenshot) driver).getScreenshotAs(OutputType.FILE);

        BufferedImage fullImg = ImageIO.read(screenshot);

        // Get the location of element on the page

        WebElement imageElement = driver.findElement(By.id("dx_captcha_basic_bg_1"));

        Point point = imageElement.getLocation();

        // Get width and height of the element

        int eleWidth = imageElement.getSize().getWidth();

        int eleHeight = imageElement.getSize().getHeight();

        // Crop the entire page screenshot to get only element screenshot

        BufferedImage eleScreenshot = fullImg.getSubimage(point.getX(), point.getY(),

                eleWidth, eleHeight);

        ImageIO.write(eleScreenshot, "png", screenshot);

        // Copy the element screenshot to disk

        File screenshotLocation = new File(fileName);

        FileUtils.copyFile(screenshot, screenshotLocation);

        Imgcodecs imageCodecs = new Imgcodecs();

        Mat grayMatrix = imageCodecs.imread(fileName, Imgcodecs.IMREAD_COLOR);

        Imgproc.cvtColor(grayMatrix, grayMatrix, Imgproc.COLOR_BGR2GRAY);

        Mat medianBlur = new Mat();

        Imgproc.medianBlur(grayMatrix, medianBlur, 5);

        Mat circles = new Mat();

        Imgproc.HoughCircles(medianBlur, circles, Imgproc.HOUGH_GRADIENT, 1, 15, 50, 20, 17, 25);

        return circles;

    }

}