Subversion Repositories SmartDukaan

package com.smartdukaan.cron.scheduled;

import com.spice.profitmandi.common.exception.ProfitMandiBusinessException;
import com.spice.profitmandi.dao.entity.fofo.ActivatedImei;
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.io.IOException;
import java.nio.file.Files;
import java.nio.file.Paths;
import java.time.LocalDate;
import java.time.LocalDateTime;
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 ProfitMandiBusinessException, IOException, Exception {

                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();
                okHttpClient.newCall(request).execute();


                List<ImeiActivationTimestampModel> imeisActivationList = activatedImeiRepository
                                .selectImeiActivationPendingByBrand("Vivo", 2);
                imeisActivationList = imeisActivationList.size() > 20 ? imeisActivationList.subList(0, 20) : imeisActivationList;
                LOGGER.info("imeisActivationList = {}", imeisActivationList);

                LOGGER.info("Complete list size is - {}", imeisActivationList.size());
                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();

                Response response = okHttpClient.newCall(request).execute();
                LOGGER.info("cookie {}", this.cookieStore.values());

                LOGGER.info("vivoSupportUrl" + vivoSupportUrl);

                String filePath = "/tmp/captcha.jpg";
                Files.write(Paths.get(filePath), response.body().bytes());

                // Get all relevent imes from activatedImei and inventoryItem for vivo

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

                Response response2 = okHttpClient.newCall(request1).execute();
                JSONObject 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);
                        return;
                }

                JSONObject imeiQueryDto = data.getJSONObject("imeiQueryDto");
                LOGGER.info("imeiQueryDto - {}", imeiQueryDto);
                String dueTimeString = imeiQueryDto.getString("dueTime");

                ActivatedImei activatedImei = activatedImeiRepository.selectBySerialNumber(imei);
                if (activatedImei == null) {
                        LOGGER.info("Adding activation for - {}", imei);
                        activatedImei = new ActivatedImei();
                        activatedImei.setSerialNumber(imei);
                        activatedImei.setCreateTimestamp(LocalDateTime.now());
                        activatedImeiRepository.persist(activatedImei);
                }
                activatedImei.setCreateTimestamp(LocalDateTime.now());
                if (dueTimeString.equals("")) return;

                DateTimeFormatter formatter = DateTimeFormatter.ofPattern("dd/MM/yyyy");
                dueTimeString = dueTimeString.split(" ")[0];
                LocalDate dueDate = LocalDate.parse(dueTimeString, formatter);
                activatedImei.setActivationTimestamp(dueDate.atStartOfDay().minusYears(1));

                LOGGER.info("activatedImei" + activatedImei);
                LOGGER.info("vivoImeiStatusModel" + dueDate.atStartOfDay());


        }
}