diff --git a/pom.xml b/pom.xml index ff8ca25..23cb209 100644 --- a/pom.xml +++ b/pom.xml @@ -66,6 +66,7 @@ org.springframework.cloud spring-cloud-starter-vault-config + 4.2.0 org.postgresql @@ -198,7 +199,7 @@ 2.22.2 - file:org_ccalm_jwt.yml + file:org-ccalm-jwt.yml diff --git a/src/main/java/org/ccalm/jwt/MainController.java b/src/main/java/org/ccalm/jwt/MainController.java index 72e3bd8..1828c51 100644 --- a/src/main/java/org/ccalm/jwt/MainController.java +++ b/src/main/java/org/ccalm/jwt/MainController.java @@ -65,6 +65,7 @@ import java.util.Properties; import java.io.IOException; import java.util.stream.Collectors; import org.springframework.security.core.Authentication; +import org.springframework.core.env.Environment; @Controller public class MainController implements ServletContextAware { @@ -111,6 +112,7 @@ public class MainController implements ServletContextAware { private ServletContext context; private final NamedParameterJdbcTemplate jdbcTemplate; + private final Environment environment; private HikariDataSource dataSource; public Storage storage = new Storage(); @@ -119,16 +121,26 @@ public class MainController implements ServletContextAware { this.context = servletContext; } + //@Autowired + //public void DatabaseService(HikariDataSource dataSource) { + // this.dataSource = dataSource; + //} + //--------------------------------------------------------------------------- @Autowired - public void DatabaseService(HikariDataSource dataSource) { + public MainController(NamedParameterJdbcTemplate jdbcTemplate,HikariDataSource dataSource,Environment environment) { + this.jdbcTemplate = jdbcTemplate; + this.environment = environment; this.dataSource = dataSource; } - - @Autowired - public MainController(NamedParameterJdbcTemplate jdbcTemplate) { - this.jdbcTemplate = jdbcTemplate; + //--------------------------------------------------------------------------- + public boolean isProduction() { + for (String profile : environment.getActiveProfiles()) { + if ("prod".equalsIgnoreCase(profile)) { + return true; + } + } + return false; } - //--------------------------------------------------------------------------- public String createStrJSONError(int code, String message, String setting, String marker) { JSONObject json = new JSONObject(); @@ -509,9 +521,14 @@ public class MainController implements ServletContextAware { json.put("error_code",0); json.put("error_message",""); - //Генерю Captcha - ImageCaptcha imageCaptcha = new ImageCaptcha.Builder(400, 100) - .addContent(new LatinContentProducer(7), + ImageCaptcha.Builder builder; + if(Tools.isInteger(email_model.getWidth())) { + builder = new ImageCaptcha.Builder(Integer.valueOf(email_model.getWidth()), 100); + }else{ + builder = new ImageCaptcha.Builder(400, 100); + } + + ImageCaptcha imageCaptcha = builder.addContent(new LatinContentProducer(7), new DefaultWordRenderer.Builder() .randomColor(Color.BLACK, Color.BLUE, Color.CYAN, Color.RED) .build()) @@ -520,7 +537,9 @@ public class MainController implements ServletContextAware { .build(); BufferedImage img = imageCaptcha.getImage(); - json.put("code",imageCaptcha.getContent());//json.put("code",""); + if(!isProduction()) + json.put("code",imageCaptcha.getContent());//json.put("code",""); + try { ByteArrayOutputStream baos = new ByteArrayOutputStream(); ImageIO.write(img, "jpeg", baos); @@ -1411,8 +1430,6 @@ public class MainController implements ServletContextAware { Translation trt = new Translation(language_id,jdbcTemplate); String result=createStrJSONError(10000,trt.trt("Request_not_processed"), (String)null, (String)null); try{ - //Connection conn = getConnection(); - int index = restore.getToken().indexOf("."); String payload = restore.getToken().substring(0, index); String signature1 = restore.getToken().substring(index+1); @@ -1440,7 +1457,7 @@ public class MainController implements ServletContextAware { if(token.getLong("exp") + + + @@ -25,10 +28,11 @@ - - + + +