rucaptcha 3.1.4 → 3.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 9abb0e5b8aa7d18c9d35262ee5d6c07e625fa70e215980a2f2e6622e10d6b66b
-  data.tar.gz: b599d194a27a5434ef79b53ee5adf10893a96c40f1a795b47313a51c0bc7e322
+  metadata.gz: 17ac95708dbda9b6245714405a31f1f9958692c3bc6d2b8d0956e3716a755a40
+  data.tar.gz: b3bfcf8d2be6b81030e6f226ba855c44c20284393453a861d6bd60e832d4b60c
 SHA512:
-  metadata.gz: f0c316a599ffbf7dbcfcba6a3fc89928633d4ba8f3f5e737bd137d2317518c23276e755dc63a7ea600c95ddef9c45974157cda7e90196f30a763c5b3aebb52b8
-  data.tar.gz: b86d81032b44ccef793b82c6ca37bd20c60dec3948034e70a31a16ace0f38480285d2df8f8b16836a40dd8e584ccac019bb1a1aafd3ca2ca0e19a8f3ec784548
+  metadata.gz: 644f9a58788af95d68767af7003fad4a204f2ec452fc9bcd5502b9e07c2995467e958d6aa0dd9636b29fe648fe53e6d1f6e2d7bbbc6bd4e718119bac252a8011
+  data.tar.gz: df7ae75db0eae30213b785bbef10758e9fe1ff1f9821f9eda6104a11362053d1bd1f5631992d760cc3ae57b60643fe41b86b07cb52097855954722be9f579b13

data/README.md

@@ -37,6 +37,7 @@ Create `config/initializers/rucaptcha.rb`
 RuCaptcha.configure do
   # Custom captcha code expire time if you need, default: 2 minutes
   # self.expires_in = 120
+
   # [Requirement / 重要]
   # Store Captcha code where, this config more like Rails config.cache_store
   # default: Read config info from `Rails.application.config.cache_store`
@@ -44,10 +45,22 @@ RuCaptcha.configure do
   # 默认：会从 Rails 配置的 cache_store 里面读取相同的配置信息，并尝试用可以运行的方式，用于存储验证码字符
   # 但如果是 [:null_store, :memory_store, :file_store] 之类的，你可以通过下面的配置项单独给 RuCaptcha 配置 cache_store
   self.cache_store = :mem_cache_store
+
+  # If you wants disable `cache_store` check warning, you can do it, default: false
   # 如果想要 disable cache_store 的 warning，就设置为 true，default false
   # self.skip_cache_store_check = true
+
   # Chars length, default: 5, allows: [3 - 7]
   # self.length = 5
+
+  # Enable or disable Strikethrough, default: true
+  # self.line = true
+
+  # Enable or disable noise, default: false
+  # self.noise = false
+
+  # Set the image format, default: png, allows: [jpeg, png, webp]
+  # self.format = 'png'
 end
 ```
 

data/Rakefile

@@ -27,9 +27,14 @@ end
 RSpec::Core::RakeTask.new(:spec)
 task default: :spec
 
+def create_captcha(length = 5, difficulty = 5)
+  require "rucaptcha"
+  RuCaptchaCore.create(length, difficulty, false, false, "png")
+end
+
 task :preview do
   require "rucaptcha"
-  res = RuCaptchaCore.create(5, 5)
+  res = create_captcha()
   warn "-------------------------\n#{res[0]}"
   puts res[1].pack("c*")
 end
@@ -38,11 +43,11 @@ task :memory do
   require "rucaptcha"
   require "memory_profiler"
 
-  RuCaptchaCore.create(5, 5)
+  create_captcha()
 
   report = MemoryProfiler.report do
     1000.times do
-      RuCaptchaCore.create(5, 5)
+      create_captcha()
     end
   end
 
@@ -57,10 +62,10 @@ task :benchmark do
   require "rucaptcha"
   require "benchmark/ips"
 
-  RuCaptchaCore.create(5, 5)
+  RuCaptchaCore.create(5, 5, true, true, "png")
 
   Benchmark.ips do |x|
-    x.report("Generate image") { RuCaptchaCore.create(5, 5) }
+    x.report("Generate image") { RuCaptchaCore.create(5, 5, true, true, "png") }
     x.compare!
   end
 end

data/ext/rucaptcha/src/captcha.rs

@@ -1,6 +1,4 @@
-use image::{ImageBuffer, Rgb};
-use imageproc::drawing::{draw_cubic_bezier_curve_mut, draw_text_mut};
-use imageproc::noise::gaussian_noise_mut;
+use image::{ImageBuffer, Rgba};
 use rand::{thread_rng, Rng};
 use rusttype::{Font, Scale};
 use std::io::Cursor;
@@ -15,21 +13,21 @@ static FONT_BYTES1: &[u8; 145008] = include_bytes!("../fonts/FuzzyBubbles-Regula
 static FONT_BYTES2: &[u8; 37792] = include_bytes!("../fonts/Handlee-Regular.ttf");
 
 // https://coolors.co/cc0b8f-7c0abe-5700c8-3c2ea4-3d56a8-3fa67e-45bb30-69d003-a0d003-d8db02
-static COLORS: [(u8, u8, u8); 14] = [
-    (197, 166, 3),
-    (187, 87, 5),
-    (176, 7, 7),
-    (186, 9, 56),
-    (204, 11, 143),
-    (124, 10, 190),
-    (87, 0, 200),
-    (61, 86, 168),
-    (63, 166, 126),
-    (69, 187, 48),
-    (105, 208, 3),
-    (160, 208, 3),
-    (216, 219, 2),
-    (50, 50, 50),
+static COLORS: [(u8, u8, u8, u8); 14] = [
+    (197, 166, 3, 255),
+    (187, 87, 5, 255),
+    (176, 7, 7, 255),
+    (186, 9, 56, 255),
+    (204, 11, 143, 255),
+    (124, 10, 190, 255),
+    (87, 0, 200, 255),
+    (61, 86, 168, 255),
+    (63, 166, 126, 255),
+    (69, 187, 48, 255),
+    (105, 208, 3, 255),
+    (160, 208, 3, 255),
+    (216, 219, 2, 255),
+    (50, 50, 50, 255),
 ];
 
 static SCALE_SM: u32 = 32;
@@ -51,18 +49,18 @@ fn get_captcha(len: usize) -> Vec<String> {
 }
 
 #[allow(unused)]
-fn get_color() -> Rgb<u8> {
+fn get_color() -> Rgba<u8> {
     let rnd = rand_num(COLORS.len() - 1);
     let c = COLORS[rnd];
-    Rgb([c.0, c.1, c.2])
+    Rgba([c.0, c.1, c.2, c.3])
 }
 
-fn get_colors(num: usize) -> Vec<Rgb<u8>> {
+fn get_colors(num: usize) -> Vec<Rgba<u8>> {
     let rnd = rand_num(COLORS.len());
     let mut out = vec![];
     for i in 0..num {
         let c = COLORS[(rnd + i) % COLORS.len()];
-        out.push(Rgb([c.0, c.1, c.2]))
+        out.push(Rgba([c.0, c.1, c.2, c.3]))
     }
 
     out
@@ -80,13 +78,13 @@ fn get_font() -> Font<'static> {
     }
 }
 
-fn get_image(width: usize, height: usize) -> ImageBuffer<Rgb<u8>, Vec<u8>> {
+fn get_image(width: usize, height: usize) -> ImageBuffer<Rgba<u8>, Vec<u8>> {
     ImageBuffer::from_fn(width as u32, height as u32, |_, _| {
-        image::Rgb([255, 255, 255])
+        image::Rgba([255, 255, 255, 255])
     })
 }
 
-fn cyclic_write_character(res: &[String], image: &mut ImageBuffer<Rgb<u8>, Vec<u8>>) {
+fn cyclic_write_character(res: &[String], image: &mut ImageBuffer<Rgba<u8>, Vec<u8>>, lines: bool) {
     let c = (image.width() - 20) / res.len() as u32;
     let y = image.height() / 3 - 15;
 
@@ -107,7 +105,10 @@ fn cyclic_write_character(res: &[String], image: &mut ImageBuffer<Rgb<u8>, Vec<u
     // Draw line, ellipse first as background
     for (i, _) in res.iter().enumerate() {
         let line_color = line_colors[i];
-        draw_interference_line(1, image, line_color);
+
+        if lines {
+            draw_interference_line(1, image, line_color);
+        }
         draw_interference_ellipse(1, image, line_color);
     }
 
@@ -121,7 +122,7 @@ fn cyclic_write_character(res: &[String], image: &mut ImageBuffer<Rgb<u8>, Vec<u
         for j in 0..(rand_num(3) + 1) as i32 {
             // Draw text again with offset
             let offset = j * (rand_num(2) as i32);
-            draw_text_mut(
+            imageproc::drawing::draw_text_mut(
                 image,
                 color,
                 10 + offset + (i as u32 * c) as i32,
@@ -137,7 +138,7 @@ fn cyclic_write_character(res: &[String], image: &mut ImageBuffer<Rgb<u8>, Vec<u
     }
 }
 
-fn draw_interference_line(num: usize, image: &mut ImageBuffer<Rgb<u8>, Vec<u8>>, color: Rgb<u8>) {
+fn draw_interference_line(num: usize, image: &mut ImageBuffer<Rgba<u8>, Vec<u8>>, color: Rgba<u8>) {
     for _ in 0..num {
         let width = image.width();
         let height = image.height();
@@ -153,7 +154,7 @@ fn draw_interference_line(num: usize, image: &mut ImageBuffer<Rgb<u8>, Vec<u8>>,
         let ctrl_x2 = get_next((width / 12) as f32, width / 12 * 3);
         let ctrl_y2 = get_next(x1, height - 5);
         // Randomly draw bezier curves
-        draw_cubic_bezier_curve_mut(
+        imageproc::drawing::draw_cubic_bezier_curve_mut(
             image,
             (x1, y1),
             (x2, y2),
@@ -166,8 +167,8 @@ fn draw_interference_line(num: usize, image: &mut ImageBuffer<Rgb<u8>, Vec<u8>>,
 
 fn draw_interference_ellipse(
     num: usize,
-    image: &mut ImageBuffer<Rgb<u8>, Vec<u8>>,
-    color: Rgb<u8>,
+    image: &mut ImageBuffer<Rgba<u8>, Vec<u8>>,
+    color: Rgba<u8>,
 ) {
     for _ in 0..num {
         // max cycle width 20px
@@ -189,6 +190,9 @@ pub struct CaptchaBuilder {
     width: usize,
     height: usize,
     complexity: usize,
+    line: bool,
+    noise: bool,
+    format: image::ImageFormat,
 }
 
 impl CaptchaBuilder {
@@ -198,6 +202,9 @@ impl CaptchaBuilder {
             width: 220,
             height: 70,
             complexity: 5,
+            line: true,
+            noise: false,
+            format: image::ImageFormat::Png,
         }
     }
 
@@ -206,15 +213,26 @@ impl CaptchaBuilder {
         self
     }
 
-    // pub fn width(mut self, width: usize) -> Self {
-    //     self.width = width;
-    //     self
-    // }
+    pub fn line(mut self, line: bool) -> Self {
+        self.line = line;
+        self
+    }
+
+    pub fn noise(mut self, noise: bool) -> Self {
+        self.noise = noise;
+        self
+    }
+
+    pub fn format(mut self, format: &str) -> Self {
+        self.format = match format {
+            "png" => image::ImageFormat::Png,
+            "jpg" | "jpeg" => image::ImageFormat::Jpeg,
+            "webp" => image::ImageFormat::WebP,
+            _ => image::ImageFormat::Png,
+        };
 
-    // pub fn height(mut self, height: usize) -> Self {
-    //     self.height = height;
-    //     self
-    // }
+        self
+    }
 
     pub fn complexity(mut self, complexity: usize) -> Self {
         let mut complexity = complexity;
@@ -238,20 +256,86 @@ impl CaptchaBuilder {
         let mut image = get_image(self.width, self.height);
 
         // Loop to write the verification code string into the background image
-        cyclic_write_character(&res, &mut image);
-
-        gaussian_noise_mut(
-            &mut image,
-            (self.complexity - 1) as f64,
-            ((10 * self.complexity) - 10) as f64,
-            ((5 * self.complexity) - 5) as u64,
-        );
+        cyclic_write_character(&res, &mut image, self.line);
+
+        if self.noise {
+            imageproc::noise::gaussian_noise_mut(
+                &mut image,
+                (self.complexity - 1) as f64,
+                ((10 * self.complexity) - 10) as f64,
+                ((5 * self.complexity) - 5) as u64,
+            );
+        }
 
         let mut bytes: Vec<u8> = Vec::new();
         image
-            .write_to(&mut Cursor::new(&mut bytes), image::ImageFormat::Jpeg)
+            .write_to(&mut Cursor::new(&mut bytes), image::ImageFormat::Png)
             .unwrap();
 
         Captcha { text, image: bytes }
     }
 }
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn test_format() {
+        let mut builder = CaptchaBuilder::new();
+        assert_eq!(builder.format, image::ImageFormat::Png);
+
+        builder = builder.format("jpg");
+        assert_eq!(builder.format, image::ImageFormat::Jpeg);
+        builder = builder.format("jpeg");
+        assert_eq!(builder.format, image::ImageFormat::Jpeg);
+        builder = builder.format("webp");
+        assert_eq!(builder.format, image::ImageFormat::WebP);
+        builder = builder.format("png");
+        assert_eq!(builder.format, image::ImageFormat::Png);
+        builder = builder.format("gif");
+        assert_eq!(builder.format, image::ImageFormat::Png);
+    }
+
+    #[test]
+    fn test_line() {
+        let mut builder = CaptchaBuilder::new();
+        assert!(builder.line);
+
+        builder = builder.line(false);
+        assert!(!builder.line);
+    }
+
+    #[test]
+    fn test_noise() {
+        let mut builder = CaptchaBuilder::new();
+        assert!(!builder.noise);
+
+        builder = builder.noise(true);
+        assert!(builder.noise);
+    }
+
+    #[test]
+    fn test_difficulty() {
+        let mut builder = CaptchaBuilder::new();
+        assert_eq!(builder.complexity, 5);
+
+        builder = builder.complexity(10);
+        assert_eq!(builder.complexity, 10);
+
+        builder = builder.complexity(11);
+        assert_eq!(builder.complexity, 10);
+
+        builder = builder.complexity(0);
+        assert_eq!(builder.complexity, 1);
+    }
+
+    #[test]
+    fn test_length() {
+        let mut builder = CaptchaBuilder::new();
+        assert_eq!(builder.length, 4);
+
+        builder = builder.length(10);
+        assert_eq!(builder.length, 10);
+    }
+}

data/ext/rucaptcha/src/lib.rs

@@ -2,9 +2,21 @@ use magnus::{define_class, function, Error, Object};
 
 mod captcha;
 
-pub fn create(len: usize, difficulty: usize) -> (String, Vec<u8>) {
+pub fn create(
+    len: usize,
+    difficulty: usize,
+    line: bool,
+    noise: bool,
+    format: String,
+) -> (String, Vec<u8>) {
     let c = captcha::CaptchaBuilder::new();
-    let out = c.complexity(difficulty).length(len).build();
+    let out = c
+        .complexity(difficulty)
+        .length(len)
+        .line(line)
+        .noise(noise)
+        .format(&format)
+        .build();
 
     (out.text, out.image)
 }
@@ -12,7 +24,7 @@ pub fn create(len: usize, difficulty: usize) -> (String, Vec<u8>) {
 #[magnus::init]
 fn init() -> Result<(), Error> {
     let class = define_class("RuCaptchaCore", Default::default())?;
-    class.define_singleton_method("create", function!(create, 2))?;
+    class.define_singleton_method("create", function!(create, 5))?;
 
     Ok(())
 }

data/lib/rucaptcha/configuration.rb

@@ -9,6 +9,12 @@ module RuCaptcha
     attr_accessor :length
     # Hard mode, default: 5, allows: [1..10]
     attr_accessor :difficulty
+    # Enable or disable strikethrough lines on captcha image, default: true
+    attr_accessor :line
+    # Enable or disable noise on captcha image, default: false
+    attr_accessor :noise
+    # Image format allow: ['jpeg', 'png', 'webp'], default: 'png'
+    attr_accessor :format
     # skip_cache_store_check, default: false
     attr_accessor :skip_cache_store_check
   end

data/lib/rucaptcha/version.rb

@@ -1,3 +1,3 @@
 module RuCaptcha
-  VERSION = "3.1.4"
+  VERSION = "3.2.0"
 end

data/lib/rucaptcha.rb

@@ -28,6 +28,9 @@ module RuCaptcha
       @config.difficulty = 3
       @config.expires_in = 2.minutes
       @config.skip_cache_store_check = false
+      @config.line = true
+      @config.noise = true
+      @config.format = "png"
 
       @config.cache_store = if Rails.application
                               Rails.application.config.cache_store
@@ -47,7 +50,7 @@ module RuCaptcha
 
       raise RuCaptcha::Errors::Configuration, "length config error, value must in 3..7" unless length.in?(3..7)
 
-      result = RuCaptchaCore.create(length, config.difficulty || 5)
+      result = RuCaptchaCore.create(length, config.difficulty || 5, config.line, config.noise, config.format)
       [result[0].downcase, result[1].pack("c*")]
     end
 

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: rucaptcha
 version: !ruby/object:Gem::Version
-  version: 3.1.4
+  version: 3.2.0
 platform: ruby
 authors:
 - Jason Lee
-autorequire: 
+autorequire:
 bindir: bin
 cert_chain: []
-date: 2023-01-11 00:00:00.000000000 Z
+date: 2023-06-09 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: railties
@@ -38,7 +38,7 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: 0.9.18
-description: 
+description:
 email: huacnlee@gmail.com
 executables: []
 extensions:
@@ -73,7 +73,7 @@ homepage: https://github.com/huacnlee/rucaptcha
 licenses:
 - MIT
 metadata: {}
-post_install_message: 
+post_install_message:
 rdoc_options: []
 require_paths:
 - lib
@@ -88,8 +88,8 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.4.1
-signing_key: 
+rubygems_version: 3.4.13
+signing_key:
 specification_version: 4
 summary: Captcha Gem for Rails, which generates captcha image by Rust.
 test_files: []