Subversion Repositories SmartDukaan

package com.smartdukaan.cron.scheduled;

import com.spice.profitmandi.common.web.client.RestClient;
import org.apache.commons.io.FileUtils;
import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import java.io.File;
import java.util.Base64;
import java.util.HashMap;
import java.util.regex.Pattern;

@Service
public class CaptchaService {

        @Autowired
        RestClient restClient;
        private static final Logger LOGGER = LogManager.getLogger(CaptchaService.class);

        /** The 31 classes the solver model can emit: 123456789ABCDEFGHKMNPQRSTUVWXYZ */
        private static final Pattern CAPTCHA_CODE = Pattern.compile("^[1-9A-HK-NP-Z]{4}$");

        /**
         * @return the 4-character captcha code, or null if the solver did not return
         *         a usable one. Callers must skip the IMEI when this is null rather
         *         than submitting the value to Vivo.
         */
        public String getCaptchaCode(String filePath) throws Exception {
                byte[] fileContent = FileUtils.readFileToByteArray(new File(filePath));
                String encodedString = Base64.getEncoder().encodeToString(fileContent);
                Base64Image base64Image = new Base64Image();
                base64Image.setImage(encodedString);

                String response = restClient.postJson("http://45.79.121.178/uploader", base64Image, new HashMap<>());

                // RestClient returns the body for ANY status, so an nginx 502 page or a
                // solver error payload arrives here looking like a normal result. Anything
                // that is not a real code is rejected instead of being sent to Vivo.
                if (response == null || !CAPTCHA_CODE.matcher(response.trim()).matches()) {
                        LOGGER.error("Captcha solver returned an unusable response: {}",
                                        response == null ? "null" : response.substring(0, Math.min(response.length(), 200)));
                        return null;
                }
                return response.trim();
        }

        class Base64Image {
                private String image;

                public String getImage() {
                        return image;
                }

                public void setImage(String image) {
                        this.image = image;
                }
        }

}