Merge branch 'feature/mnistCaptcha' into 'develop'

add mnist capctha

See merge request o2oa/o2oa!1115

o2server/x_processplatform_service_processing/src/main/java/com/x/processplatform/service/processing/processor/AbstractProcessor.java

@@ -101,18 +101,12 @@ public abstract class AbstractProcessor extends AbstractBaseProcessor {
 				aeiObjects.entityManagerContainer().commit();
 			}
 			return work.getId();
-		} catch (
-
-		Exception e) {
+		} catch (Exception e) {
 			LOGGER.error(e);
 			return null;
 		}
 	}
 
-//	private void arriveCleanManualTaskIdentityList(AeiObjects aeiObjects) {
-//		aeiObjects.getWork().setManualTaskIdentityList(new ArrayList<>());
-//	}
-
 	private void arriveCleanManualTaskIdentityMatrix(AeiObjects aeiObjects) {
 		aeiObjects.getWork().setManualTaskIdentityMatrix(new ManualTaskIdentityMatrix());
 	}
@@ -332,9 +326,7 @@ public abstract class AbstractProcessor extends AbstractBaseProcessor {
 				// 执行AfterInquireScript中的代码可能修改了data数据.
 				aeiObjects.entityManagerContainer().commit();
 			}
-		} catch (
-
-		Exception e) {
+		} catch (Exception e) {
 			LOGGER.error(e);
 		}
 		return results;

o2server/x_program_center/src/main/java/com/x/program/center/jaxrs/captcha/ExceptionCreateCaptcha.java

+package com.x.program.center.jaxrs.captcha;
+
+import com.x.base.core.project.exception.LanguagePromptException;
+
+class ExceptionCreateCaptcha extends LanguagePromptException {
+
+	private static final long serialVersionUID = -326121174563758109L;
+
+	ExceptionCreateCaptcha() {
+		super("create captcha error.");
+	}
+}

o2server/x_program_center/src/main/java/com/x/program/center/jaxrs/captcha/V2Create.java

 package com.x.program.center.jaxrs.captcha;
 
+import java.awt.geom.AffineTransform;
 import java.awt.image.BufferedImage;
 import java.io.ByteArrayOutputStream;
+import java.io.File;
+import java.io.IOException;
+import java.security.SecureRandom;
+import java.util.Arrays;
 import java.util.Base64;
+import java.util.Optional;
 import java.util.Properties;
+import java.util.Random;
+import java.util.function.Function;
+import java.util.stream.Stream;
 
 import javax.imageio.ImageIO;
 
+import org.apache.commons.lang3.ArrayUtils;
+
 import com.google.code.kaptcha.Constants;
 import com.google.code.kaptcha.Producer;
 import com.google.code.kaptcha.impl.DefaultKaptcha;
 import com.google.code.kaptcha.util.Config;
+import com.google.common.collect.Streams;
 import com.x.base.core.container.EntityManagerContainer;
 import com.x.base.core.container.factory.EntityManagerContainerFactory;
 import com.x.base.core.entity.annotation.CheckPersistType;
 import com.x.base.core.project.annotation.FieldDescribe;
+import com.x.base.core.project.bean.tuple.Pair;
 import com.x.base.core.project.http.ActionResult;
 import com.x.base.core.project.logger.Logger;
 import com.x.base.core.project.logger.LoggerFactory;
+import com.x.program.center.ThisApplication;
 import com.x.program.center.core.entity.Captcha;
 
 class V2Create extends BaseAction {
@@ -27,33 +41,78 @@ class V2Create extends BaseAction {
 	ActionResult<Wo> execute(Integer width, Integer height) throws Exception {
 		try (EntityManagerContainer emc = EntityManagerContainerFactory.instance().create()) {
 			ActionResult<Wo> result = new ActionResult<>();
-			if (width <= 80 || width > 800) {
-				width = 80;
+			final Integer adjustWidth = (width <= 80 || width > 800) ? 80 : width;
+			final Integer adjustHeight = (height <= 30 || height > 800) ? 30 : height;
+
+			Optional<Optional<Pair<String, byte[]>>> optional = Stream.of(kaptcha, mnist)
+					.map(o -> o.apply(Pair.of(adjustWidth, adjustHeight))).filter(Optional::isPresent).findFirst();
+			if (optional.isEmpty()) {
+				throw new ExceptionCreateCaptcha();
 			}
-			if (height <= 30 || height > 800) {
-				height = 30;
+			Optional<Pair<String, byte[]>> opt = optional.get();
+			if (opt.isEmpty()) {
+				throw new ExceptionCreateCaptcha();
 			}
-			Producer producer = createProducer(width, height);
-			// 生成随机字符串
-			String verifyCode = producer.createText();
-			// 生成图片
-			BufferedImage bufferedImage = producer.createImage(verifyCode);
-			ByteArrayOutputStream stream = new ByteArrayOutputStream();
-			ImageIO.write(bufferedImage, "png", stream);
-
+			Pair<String, byte[]> pair = opt.get();
 			emc.beginTransaction(Captcha.class);
 			Captcha captcha = new Captcha();
-			captcha.setAnswer(verifyCode);
+			captcha.setAnswer(pair.first());
 			emc.persist(captcha, CheckPersistType.all);
 			emc.commit();
 			Wo wo = new Wo();
 			wo.setId(captcha.getId());
-			wo.setImage(Base64.getEncoder().encodeToString(stream.toByteArray()));
+			wo.setImage(Base64.getEncoder().encodeToString(pair.second()));
 			result.setData(wo);
 			return result;
 		}
 	}
 
+	private static Function<Pair<Integer, Integer>, Optional<Pair<String, byte[]>>> kaptcha = param -> {
+		Producer producer = createProducer(param.first(), param.second());
+		// 生成随机字符串
+		String verifyCode = producer.createText();
+		// 生成图片
+		try (ByteArrayOutputStream out = new ByteArrayOutputStream()) {
+			BufferedImage bufferedImage = producer.createImage(verifyCode);
+			ImageIO.write(bufferedImage, "png", out);
+			return Optional.of(Pair.of(verifyCode, out.toByteArray()));
+		} catch (Exception | Error e) {
+			// catch all exception
+			LOGGER.warn("can not create captcha, use mnist image instead, com.google.code.kaptcha not available:{}.",
+					e.getMessage());
+		}
+		return Optional.empty();
+	};
+
+	private static Function<Pair<Integer, Integer>, Optional<Pair<String, byte[]>>> mnist = param -> {
+		Random random = new SecureRandom();
+		String verifyCode = random.nextInt(10) + "" + random.nextInt(10) + "" + random.nextInt(10) + ""
+				+ random.nextInt(10);
+		try (ByteArrayOutputStream out = new ByteArrayOutputStream()) {
+			Character[] chars = ArrayUtils.toObject(verifyCode.toCharArray());
+			BufferedImage combined = new BufferedImage(28 * chars.length, 28, BufferedImage.TYPE_INT_RGB);
+			final File dir = new File(ThisApplication.context().path(), "WEB-INF/mnist");
+			Streams.mapWithIndex(Arrays.<Character>stream(chars), Pair::of).forEach(p -> {
+				BufferedImage image;
+				try {
+					image = ImageIO.read(new File(dir, p.first() + "_" + random.nextInt(10)));
+					combined.getGraphics().drawImage(image, (int) (28 * p.second()), 0, null);
+				} catch (IOException e) {
+					LOGGER.error(e);
+				}
+			});
+			BufferedImage overlay = new BufferedImage(param.first(), param.second(), BufferedImage.TYPE_INT_RGB);
+			overlay.createGraphics().drawRenderedImage(combined,
+					AffineTransform.getScaleInstance(overlay.getWidth() / (double) combined.getWidth(),
+							overlay.getHeight() / (double) combined.getHeight()));
+			ImageIO.write(overlay, "png", out);
+			return Optional.of(Pair.of(verifyCode, out.toByteArray()));
+		} catch (IOException e) {
+			LOGGER.error(e);
+		}
+		return Optional.empty();
+	};
+
 	public static class Wo {
 
 		@FieldDescribe("标识")
@@ -90,8 +149,6 @@ class V2Create extends BaseAction {
 		properties.setProperty(Constants.KAPTCHA_TEXTPRODUCER_CHAR_LENGTH, "4");
 		properties.setProperty(Constants.KAPTCHA_TEXTPRODUCER_FONT_SIZE, "40");
 		properties.setProperty(Constants.KAPTCHA_TEXTPRODUCER_CHAR_STRING, "1234567890");
-//		properties.setProperty(Constants.KAPTCHA_WORDRENDERER_IMPL,
-//				"com.x.base.core.project.kaptcha.SimpleWordRenderer");
 		Config config = new Config(properties);
 		DefaultKaptcha defaultKaptcha = new DefaultKaptcha();
 		defaultKaptcha.setConfig(config);