Subversion Repositories SmartDukaan

package com.smartdukaan.cron.scheduled;

import com.spice.profitmandi.dao.model.ImeiActivationTimestampModel;

import com.spice.profitmandi.dao.repository.fofo.ActivatedImeiRepository;

import com.spice.profitmandi.service.inventory.InventoryService;

import okhttp3.*;

import org.apache.logging.log4j.LogManager;

import org.apache.logging.log4j.Logger;

import org.json.JSONObject;

import org.springframework.beans.factory.annotation.Autowired;

import org.springframework.stereotype.Component;

import java.nio.file.Files;

import java.nio.file.Paths;

import java.time.LocalDate;

import java.time.format.DateTimeFormatter;

import java.util.ArrayList;

import java.util.HashMap;

import java.util.List;

import java.util.Map;

@Component

class VivoImeiActivationService {

    @Autowired

    InventoryService inventoryService;

    @Autowired

    CaptchaService captchaService;

    @Autowired

    ActivatedImeiRepository activatedImeiRepository;

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

    private final Map<String, List<Cookie>> cookieStore = new HashMap<>();

    public void checkImeiActivation() throws Exception {

        if (LocalDate.now().getDayOfMonth() == 1) return;

        List<ImeiActivationTimestampModel> imeisActivationList = activatedImeiRepository

                .selectImeiActivationPendingByBrand("Vivo", 2);

        LOGGER.info("Secondary imeisActivationList = {}, size = {}", imeisActivationList, imeisActivationList.size());

        processImeis(imeisActivationList);

    }

    public void checkImeiActivationTertiary() throws Exception {

        if (LocalDate.now().getDayOfMonth() == 1) return;

        List<ImeiActivationTimestampModel> imeisActivationList = activatedImeiRepository

                .selectImeiActivationPendingByBrandTertiary("Vivo", 2);

        LOGGER.info("Tertiary imeisActivationList = {}, size = {}", imeisActivationList, imeisActivationList.size());

        processImeis(imeisActivationList);

    }

    private void processImeis(List<ImeiActivationTimestampModel> imeisActivationList) throws Exception {

        if (imeisActivationList.isEmpty()) return;

        cookieStore.clear();

        OkHttpClient okHttpClient = new OkHttpClient.Builder().cookieJar(new CookieJar() {

            @Override

            public void saveFromResponse(HttpUrl httpUrl, List<Cookie> list) {

                LOGGER.info("SAVE fROM - {}", httpUrl.host());

                cookieStore.put(httpUrl.host(), list);

            }

            @Override

            public List<Cookie> loadForRequest(HttpUrl httpUrl) {

                List<Cookie> cookies = cookieStore.get(httpUrl.host());

                LOGGER.info("load from - {}", httpUrl.host());

                return cookies != null ? cookies : new ArrayList<>();

            }

        }).build();

        Request request = new Request.Builder().url("https://www.vivo.com/in/support/IMEI").build();

        try (Response seedResponse = okHttpClient.newCall(request).execute()) {

            // Consume response to establish cookies

        }

        for (ImeiActivationTimestampModel imeiActivationTimestampModel : imeisActivationList) {

            String captchaBreak = this.getNewCaptcha(okHttpClient);

            this.getImeiActivation(captchaBreak, okHttpClient, imeiActivationTimestampModel);

        }

    }

    public String getNewCaptcha(OkHttpClient okHttpClient) throws Exception {

        LOGGER.info("okHttpClient" + okHttpClient);

        HttpUrl vivoSupportUrl = HttpUrl.parse("https://www.vivo.com/in/support/generatingCodes");

        Request request = new Request.Builder().url(vivoSupportUrl).build();

        String filePath = "/tmp/captcha.jpg";

        try (Response response = okHttpClient.newCall(request).execute()) {

            LOGGER.info("cookie {}", this.cookieStore.values());

            LOGGER.info("vivoSupportUrl" + vivoSupportUrl);

            Files.write(Paths.get(filePath), response.body().bytes());

        }

        String captchaBreak = captchaService.getCaptchaCode(filePath);

        System.out.println("Captcha is " + captchaBreak);

        return captchaBreak;

    }

    public void getImeiActivation(String captchaBreak, OkHttpClient okHttpClient,

                                  ImeiActivationTimestampModel vivoImeiAndActivationTimeStampModel) throws Exception {

        String imei = vivoImeiAndActivationTimeStampModel.getSerialNumber();

        RequestBody formBody = new FormBody.Builder().add("imei", imei).add("code", captchaBreak).build();

        Request request1 = new Request.Builder().url("https://www.vivo.com/in/support/checkCode").post(formBody)

                .build();

        JSONObject imeiActivationJson;

        try (Response response2 = okHttpClient.newCall(request1).execute()) {

            imeiActivationJson = new JSONObject(response2.body().string());

        }

        JSONObject data = imeiActivationJson.getJSONObject("data");

        int status = data.getInt("status");

        if (status == 0) {

            LOGGER.info("Found invalid captcha");

            return;

        } else if (status == 2) {

            LOGGER.info("Received status 2 for {}, data {}", imei, data);

            activatedImeiRepository.saveActivation(imei, null);

            return;

        }

        JSONObject imeiQueryDto = data.getJSONObject("imeiQueryDto");

        LOGGER.info("imeiQueryDto - {}", imeiQueryDto);

        String dueTimeString = imeiQueryDto.getString("dueTime");

        if (dueTimeString.equals("")) {

            activatedImeiRepository.saveActivation(imei, null);

            return;

        }

        DateTimeFormatter formatter = DateTimeFormatter.ofPattern("dd/MM/yyyy");

        dueTimeString = dueTimeString.split(" ")[0];

        LocalDate dueDate = LocalDate.parse(dueTimeString, formatter);

        activatedImeiRepository.saveActivation(imei, dueDate.atStartOfDay().minusYears(1));

        LOGGER.info("activatedImei saved for {} with dueDate {}", imei, dueDate.atStartOfDay());

    }

}