Add check method and demo

zzfly256 · zzfly256 · commit dc957887e4de · 2019-02-12T01:44:16.000+08:00
diff --git a/Captcha.php b/Captcha.php
@@ -10,24 +10,27 @@ class Captcha
 {
     private $str = "ABCDEFGHIJKLMNOPQRSTUVWXYZaabcdefghijklmnopqrstuvwxyz1234567890";
     private $font;
-    private $im;
+    private $imageResource;
     private $level;
     private $code;
+    private $caseSensitive;
 
     /**
      * Captcha constructor.
      * @param int $level
+     * @param boolean $caseSensitive
      */
-    public function __construct($level = 2)
+    public function __construct($level = 2, $caseSensitive = false)
     {
         $this->level = $level;
+        $this->caseSensitive = $caseSensitive;
 
-        // load the truetype font
-        $this->font = realpath('./arial.ttf');
+        // load the TrueType font
+        $this->font = __DIR__.'/arial.ttf';
 
         // create image
-        $this->im = imagecreate(200, 77);
-        imagecolorallocate($this->im, 255, 255, 255);
+        $this->imageResource = imagecreate(200, 77);
+        imagecolorallocate($this->imageResource, 255, 255, 255);
 
         // generate
         $this->generate();
@@ -43,7 +46,7 @@ public function __construct($level = 2)
      */
     public function getImageResource()
     {
-        return $this->im;
+        return $this->imageResource;
     }
 
     /**
@@ -61,8 +64,8 @@ public function render()
     {
         header('Content-Type: image/png');
 
-        imagepng($this->im);
-        imagedestroy($this->im);
+        imagepng($this->imageResource);
+        imagedestroy($this->imageResource);
     }
 
     /**
@@ -81,10 +84,24 @@ private function generate()
 
             // get font size
             $fontSize = ($this->level > 1) ? rand(20, 48) : 28;
-            imagettftext($this->im, $fontSize, rand(-35, 35), 35 * $i, 55, imagecolorallocate($this->im, rand(0, 240), rand(0, 240), rand(0, 240)), $this->font, $char);
+            imagettftext($this->imageResource, $fontSize, rand(-35, 35), 35 * $i, 55, imagecolorallocate($this->imageResource, rand(0, 240), rand(0, 240), rand(0, 240)), $this->font, $char);
         }
 
-        $this->code = $output;
+        $this->code =  ($this->caseSensitive) ? $output : strtolower($output);
+    }
+
+    /**
+     * @param $str
+     * @return bool
+     * check the code
+     */
+    public function check($str){
+        if (!$this->caseSensitive){
+            return strtolower($str) == strtolower($this->code);
+        }
+        else{
+            return $str == $this->code;
+        }
     }
 
     /**
@@ -95,11 +112,11 @@ private function addLines()
 
         $lines = rand(1, 3);
         for ($i = 0; $i < $lines; $i++) {
-            imageline($this->im, rand(0, 200), rand(0, -77), rand(0, 200), rand(77, 144), imagecolorallocate($this->im, rand(0, 240), rand(0, 240), rand(0, 240)));
+            imageline($this->imageResource, rand(0, 200), rand(0, -77), rand(0, 200), rand(77, 144), imagecolorallocate($this->imageResource, rand(0, 240), rand(0, 240), rand(0, 240)));
         }
 
         for ($i = 0; $i < 5 - $lines; $i++) {
-            imageline($this->im, rand(0, -200), rand(0, 77), rand(200, 400), rand(0, 77), imagecolorallocate($this->im, rand(0, 240), rand(0, 240), rand(0, 240)));
+            imageline($this->imageResource, rand(0, -200), rand(0, 77), rand(200, 400), rand(0, 77), imagecolorallocate($this->imageResource, rand(0, 240), rand(0, 240), rand(0, 240)));
         }
     }
 }
diff --git a/demo/action.php b/demo/action.php
@@ -0,0 +1,7 @@
+<?php
+session_start();
+if ($_SESSION['code'] == strtolower($_POST['code'])){
+    echo "true";
+} else {
+    echo "false";
+}
diff --git a/demo/image.php b/demo/image.php
@@ -0,0 +1,8 @@
+<?php
+
+session_start();
+require_once './../Captcha.php';
+
+$captcha = new Captcha();
+$_SESSION['code'] = $captcha->getCode();
+$captcha->render();
diff --git a/demo/index.html b/demo/index.html
@@ -0,0 +1,18 @@
+<!doctype html>
+<html lang="en">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport"
+          content="width=device-width, user-scalable=no, initial-scale=1.0, maximum-scale=1.0, minimum-scale=1.0">
+    <meta http-equiv="X-UA-Compatible" content="ie=edge">
+    <title>Captcha Demo</title>
+
+</head>
+<body>
+<form action="/demo/action.php" method="post">
+    <input type="text" name="code" placeholder="please input the captcha">
+    <img src="/demo/image.php" alt="">
+    <input type="submit">
+</form>
+</body>
+</html>
