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<>();

		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);

		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("--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);

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

		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 {

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

				try {

					String imei = remainingImeis.get(i);

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

					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;

					String fileName = "/tmp/oppo-simple.png";

					boolean captchaBroken = false;

					do {

						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"));

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

						} catch (StaleElementReferenceException staleElementReferenceException) {

							System.out.println("Found stale element for " + imei + ", lets move ahead with fresh imei");

							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"));

										captchaBroken = true;

										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 (!captchaBroken);

				} catch (Exception e) {

					e.printStackTrace();

				}

			}

			mainCount++;

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

		} while (mainCount < 10 && dateMap.size() != imeis.size());

		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;

	}

}