Subversion Repositories SmartDukaan

package com.smartdukaan.cron.scheduled;

import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;
import org.json.JSONObject;
import org.openqa.selenium.By;
import org.openqa.selenium.Dimension;
import org.openqa.selenium.JavascriptExecutor;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.chrome.ChromeDriver;
import org.openqa.selenium.chrome.ChromeOptions;
import org.openqa.selenium.support.ui.ExpectedConditions;
import org.openqa.selenium.support.ui.WebDriverWait;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import java.time.Instant;
import java.time.LocalDate;
import java.time.ZoneId;
import java.time.format.DateTimeFormatter;
import java.util.HashMap;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import java.util.concurrent.TimeUnit;

/**
 * Website-based activation-date pull for Motorola, mirroring
 * CheckOppoWarrantyTask / CheckRealmeWarrantyTask.
 *
 * Differs from those two in one important way: there is no slider. The page
 * (https://en-in.support.motorola.com/app/warranty/check) is Oracle Service
 * Cloud, so there is no DingXiang widget, no canvas, and OpenCV is not needed
 * at all. The lookup is a single AJAX call and the whole job is to read its
 * JSON -- the same "read the JSON, not the DOM" lesson that realme taught.
 *
 * What gates it instead is reCAPTCHA v2, raised by the endpoint itself as an
 * "==CHALLENGE REQUIRED==" body. See MotorolaChallengeSolver.
 */
@Component
public class CheckMotorolaWarrantyTask {

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

    private static final String PAGE_URL = "https://en-in.support.motorola.com/app/warranty/check";

    /** Field and submit, confirmed against the live page. The submit is a
     *  type=button; the only type=submit on the page is a FOOTER control. */
    private static final String IMEI_INPUT = "#rn_warrantylookup_3_serial";
    private static final String SUBMIT_BTN = ".my-products__detect-device-button";

    /** RightNow routes the lookup through this widget endpoint. */
    private static final String LOOKUP_ENDPOINT = "/ci/ajax/widget/custom/warranty/warrantylookup";

    private static final String CHALLENGE_MARKER = "==CHALLENGE REQUIRED==";

    /** Optional: no bean supplied means challenges cannot be answered. */
    @Autowired(required = false)
    private MotorolaChallengeSolver challengeSolver;

    /** Logged once per batch so a first successful payload reveals its shape
     *  without spamming the log for every imei. */
    private boolean loggedSamplePayload = false;

    public Map<String, LocalDate> checkWarranty(List<String> imeis) {
        Map<String, LocalDate> dateMap = new HashMap<>();
        if (imeis == null || imeis.isEmpty()) {
            return dateMap;
        }
        loggedSamplePayload = false;

        ChromeOptions options = new ChromeOptions();
        Map<String, Object> prefsMap = new HashMap<>();
        prefsMap.put("profile.default_content_settings.popups", 0);
        prefsMap.put("download.prompt", false);
        options.setExperimentalOption("prefs", prefsMap);
        // Prod path, same as the oppo/realme drivers. Left overridable so the job
        // can be exercised on a dev box against Chrome for Testing without
        // editing the file:
        //   -Dwebdriver.chrome.driver=/Users/<u>/cft/chromedriver/chromedriver
        //   -Dwebdriver.chrome.binary="/Users/<u>/cft/chrome/Google Chrome for Testing.app/Contents/MacOS/Google Chrome for Testing"
        if (System.getProperty("webdriver.chrome.driver") == null) {
            System.setProperty("webdriver.chrome.driver", "/usr/bin/chromedriver");
        }
        String binary = System.getProperty("webdriver.chrome.binary");
        if (binary != null && !binary.isEmpty()) {
            options.setBinary(binary);
        }
        options.addArguments("--headless");
        options.addArguments("--no-sandbox");
        options.addArguments("--disable-extensions");
        options.addArguments("--test-type");
        // Same memory posture as the oppo/realme drivers: this box co-hosts a
        // 9.4GB tomcat with ~3GB available and has been OOM-killed twice, with
        // tomcat the victim both times. Each driver tree is roughly 850MB, so
        // do not let this job overlap the oppo/realme window.
        options.addArguments("--disable-gpu");
        options.addArguments("--disable-dev-shm-usage");
        options.addArguments("--disable-software-rasterizer");

        WebDriver driver = null;
        try {
            driver = new ChromeDriver(options);
            driver.manage().timeouts().pageLoadTimeout(30, TimeUnit.SECONDS);
            driver.manage().timeouts().implicitlyWait(8, TimeUnit.SECONDS);
            driver.manage().window().setSize(new Dimension(1600, 900));
            driver.manage().deleteAllCookies();

            WebDriverWait wait = new WebDriverWait(driver, 12);

            driver.get(PAGE_URL);
            installResponseCapture(driver);
            wait.until(ExpectedConditions.visibilityOfElementLocated(By.cssSelector(IMEI_INPUT)));

            for (Iterator<String> it = imeis.iterator(); it.hasNext(); ) {
                String imei = it.next();
                try {
                    LocalDate date = lookupOne(driver, wait, imei);
                    // A null date is still a result: the caller writes it back
                    // as a null activation, which bumps createTimestamp and so
                    // defers this imei to the next pool sweep.
                    dateMap.put(imei, date);
                } catch (ChallengeUnansweredException ce) {
                    // Abandon the whole batch. Every subsequent lookup in this
                    // session would hit the same wall, and continuing would be
                    // pure load on Motorola for no data.
                    LOGGER.warn("Motorola raised a reCAPTCHA challenge and no solver is configured "
                            + "- abandoning the batch after {} of {} imeis", dateMap.size(), imeis.size());
                    break;
                } catch (Exception e) {
                    LOGGER.warn("Motorola lookup failed for {} - {}", imei, e.getMessage());
                }
            }
        } catch (Exception e) {
            LOGGER.error("Motorola warranty run failed", e);
        } finally {
            if (driver != null) {
                try {
                    driver.quit();
                } catch (Exception ignored) {
                }
            }
        }
        LOGGER.info("Motorola: resolved {} of {} imeis", dateMap.size(), imeis.size());
        return dateMap;
    }

    /** One imei: clear, type, submit, wait for the widget's JSON. */
    private LocalDate lookupOne(WebDriver driver, WebDriverWait wait, String imei) throws Exception {
        clearCapture(driver);

        // Re-navigating per imei is what oppo/realme do, and it is what keeps a
        // stale previous result from being read as this imei's answer.
        driver.get(PAGE_URL);
        installResponseCapture(driver);
        wait.until(ExpectedConditions.visibilityOfElementLocated(By.cssSelector(IMEI_INPUT)));

        // The field is React-backed: a plain sendKeys can be discarded on
        // re-render, so drive the native setter and fire the events the widget
        // actually listens for.
        ((JavascriptExecutor) driver).executeScript(
                "var f = document.querySelector(arguments[0]);"
              + "var s = Object.getOwnPropertyDescriptor(window.HTMLInputElement.prototype,'value').set;"
              + "s.call(f, arguments[1]);"
              + "f.dispatchEvent(new Event('input',{bubbles:true}));"
              + "f.dispatchEvent(new Event('change',{bubbles:true}));",
                IMEI_INPUT, imei);

        driver.findElement(By.cssSelector(SUBMIT_BTN)).click();

        String body = awaitResponse(driver, 15000);
        if (body == null) {
            LOGGER.info("Motorola: no lookup response captured for {}", imei);
            return null;
        }

        if (body.contains(CHALLENGE_MARKER)) {
            String siteKey = extractSiteKey(body);
            if (challengeSolver == null) {
                throw new ChallengeUnansweredException(siteKey);
            }
            String token = challengeSolver.solve(siteKey, PAGE_URL);
            if (token == null || token.isEmpty()) {
                throw new ChallengeUnansweredException(siteKey);
            }
            body = submitWithToken(driver, wait, token);
            if (body == null || body.contains(CHALLENGE_MARKER)) {
                LOGGER.warn("Motorola rejected the supplied challenge token for {}", imei);
                return null;
            }
        }

        if (!loggedSamplePayload) {
            // The success payload shape is unverified -- every observed response
            // so far has been the challenge, so no real warranty body has been
            // seen. Log the first one in full so the parser can be tightened to
            // the actual field names instead of guessing.
            LOGGER.info("Motorola sample lookup payload for {}: {}", imei,
                    body.substring(0, Math.min(body.length(), 1500)));
            loggedSamplePayload = true;
        }
        return parseActivationDate(body);
    }

    /** Replays the lookup with a challenge token attached. */
    private String submitWithToken(WebDriver driver, WebDriverWait wait, String token) {
        try {
            clearCapture(driver);
            ((JavascriptExecutor) driver).executeScript(
                    "var el = document.getElementById('g-recaptcha-response');"
                  + "if (!el) { el = document.createElement('textarea');"
                  + "  el.id = 'g-recaptcha-response'; el.name = 'g-recaptcha-response';"
                  + "  el.style.display='none'; document.body.appendChild(el); }"
                  + "el.value = arguments[0];", token);
            driver.findElement(By.cssSelector(SUBMIT_BTN)).click();
            return awaitResponse(driver, 15000);
        } catch (Exception e) {
            LOGGER.warn("Challenge replay failed - {}", e.getMessage());
            return null;
        }
    }

    /** reCAPTCHA sitekey out of the challenge blob, for the solver. */
    private String extractSiteKey(String body) {
        try {
            int i = body.indexOf("publicKey:");
            if (i < 0) return null;
            int q1 = body.indexOf('"', i);
            int q2 = body.indexOf('"', q1 + 1);
            if (q1 < 0 || q2 < 0) return null;
            return body.substring(q1 + 1, q2).replace("\\", "");
        } catch (Exception e) {
            return null;
        }
    }

    /**
     * Pulls an activation/warranty-start date out of the payload.
     *
     * Deliberately tolerant about field names and formats: the real response
     * has never been observed past the challenge, so this tries the keys such
     * payloads normally carry and falls back to logging rather than guessing
     * wrong. Tighten it once the sample payload above appears in the log.
     */
    LocalDate parseActivationDate(String json) {
        if (json == null || json.trim().isEmpty()) return null;
        String[] keys = {"activationDate", "warrantyStartDate", "startDate",
                         "purchaseDate", "regDate", "inWarrantyStartDate", "shipDate"};
        try {
            JSONObject root = new JSONObject(json);
            JSONObject scope = root.optJSONObject("data") != null ? root.optJSONObject("data") : root;
            for (String key : keys) {
                if (!scope.has(key) || scope.isNull(key)) continue;
                LocalDate d = toLocalDate(scope.get(key).toString());
                if (d != null) return d;
            }
        } catch (Exception e) {
            LOGGER.debug("Motorola payload was not the JSON shape expected - {}", e.getMessage());
        }
        return null;
    }

    /** Epoch millis, or the date formats these portals typically emit. */
    private LocalDate toLocalDate(String raw) {
        if (raw == null) return null;
        String v = raw.trim();
        if (v.isEmpty() || "null".equalsIgnoreCase(v)) return null;
        try {
            if (v.matches("\\d{10,13}")) {
                long epoch = Long.parseLong(v);
                if (v.length() == 10) epoch *= 1000L;
                return Instant.ofEpochMilli(epoch).atZone(ZoneId.systemDefault()).toLocalDate();
            }
        } catch (Exception ignored) {
        }
        String[] patterns = {"yyyy-MM-dd", "dd-MM-yyyy", "MM/dd/yyyy", "dd/MM/yyyy", "yyyy/MM/dd"};
        for (String p : patterns) {
            try {
                return LocalDate.parse(v.length() > 10 ? v.substring(0, 10) : v,
                        DateTimeFormatter.ofPattern(p));
            } catch (Exception ignored) {
            }
        }
        return null;
    }

    /**
     * Captures the widget's AJAX body.
     *
     * Hooks XHR, fetch AND Response.prototype. The last one matters: a client
     * that grabbed globalThis.fetch when its bundle loaded sails straight past
     * a window.fetch wrapper, which is exactly what hid realme's lookup.
     */
    private void installResponseCapture(WebDriver driver) {
        try {
            ((JavascriptExecutor) driver).executeScript(
                "if (!window.__motoCapInstalled) {"
              + "  window.__motoCapInstalled = true; window.__motoResp = null;"
              + "  var hit = function(u){ return String(u).indexOf('" + LOOKUP_ENDPOINT + "') !== -1; };"
              + "  var O = XMLHttpRequest.prototype.open;"
              + "  XMLHttpRequest.prototype.open = function(m,u){"
              + "    var self=this;"
              + "    this.addEventListener('load', function(){"
              + "      try { if (hit(u)) window.__motoResp = self.responseText; } catch(e){}"
              + "    });"
              + "    return O.apply(this, arguments);"
              + "  };"
              + "  var F = window.fetch;"
              + "  if (F) { window.fetch = function(){"
              + "    var u = arguments[0]; u = (u && u.url) ? u.url : String(u);"
              + "    return F.apply(this, arguments).then(function(r){"
              + "      try { if (hit(u)) r.clone().text().then(function(t){ window.__motoResp = t; }); } catch(e){}"
              + "      return r;"
              + "    });"
              + "  }; }"
              + "  ['json','text'].forEach(function(name){"
              + "    var orig = Response.prototype[name];"
              + "    if (!orig) return;"
              + "    Response.prototype[name] = function(){"
              + "      try { var u = this.url || '';"
              + "        if (hit(u) && !this.__motoSeen) {"
              + "          var c = this.clone(); this.__motoSeen = 1;"
              + "          orig.call(c).then(function(t){"
              + "            window.__motoResp = (typeof t === 'string') ? t : JSON.stringify(t); });"
              + "        } } catch(e){}"
              + "      return orig.apply(this, arguments);"
              + "    };"
              + "  });"
              + "}");
        } catch (Exception e) {
            LOGGER.warn("Could not install the Motorola response capture hook: {}", e.getMessage());
        }
    }

    /** Poll rather than read once: the click can return before the AJAX lands. */
    private String awaitResponse(WebDriver driver, long timeoutMs) {
        long deadline = System.currentTimeMillis() + timeoutMs;
        while (System.currentTimeMillis() < deadline) {
            try {
                Object v = ((JavascriptExecutor) driver).executeScript("return window.__motoResp || null;");
                if (v != null && !v.toString().trim().isEmpty()) {
                    return v.toString();
                }
            } catch (Exception ignored) {
            }
            try {
                Thread.sleep(300);
            } catch (InterruptedException ie) {
                Thread.currentThread().interrupt();
                return null;
            }
        }
        return null;
    }

    private void clearCapture(WebDriver driver) {
        try {
            ((JavascriptExecutor) driver).executeScript("window.__motoResp = null;");
        } catch (Exception ignored) {
        }
    }

    /** Raised when the page challenges and nothing can answer it. */
    static class ChallengeUnansweredException extends Exception {
        ChallengeUnansweredException(String siteKey) {
            super("reCAPTCHA challenge raised (sitekey=" + siteKey + ") and no solver is configured");
        }
    }
}