id-scanner-lib 1.3.0 → 1.3.3

package/src/id-recognition/ocr-worker.ts

@@ -4,143 +4,153 @@
  * @module OCRWorker
  */
 
-import type { IDCardInfo } from '../utils/types';
+import type { IDCardInfo } from "../utils/types"
 
 /**
  * OCR处理输入接口
  */
 export interface OCRProcessInput {
-  imageBase64: string;
-  tessWorkerOptions?: any;
+  imageBase64: string
+  tessWorkerOptions?: any
 }
 
 /**
  * OCR处理输出接口
  */
 export interface OCRProcessOutput {
-  idCardInfo: IDCardInfo;
-  processingTime: number;
+  idCardInfo: IDCardInfo
+  processingTime: number
 }
 
 /**
  * 在Web Worker中执行OCR处理的函数
- * 
+ *
  * 该函数用于在使用 createWorker 创建的 Worker 中执行
- * 
+ *
  * @param input OCR处理输入数据
  * @returns OCR处理结果
  */
-export async function processOCRInWorker(input: OCRProcessInput): Promise<OCRProcessOutput> {
+export async function processOCRInWorker(
+  input: OCRProcessInput
+): Promise<OCRProcessOutput> {
   // 计时开始
-  const startTime = performance.now();
-  
+  const startTime = performance.now()
+
   // 加载Tesseract.js (Worker 环境下动态导入)
-  const { createWorker } = await import('tesseract.js');
-  
+  const { createWorker } = await import("tesseract.js")
+
   // 创建OCR Worker
-  const worker = await createWorker(input.tessWorkerOptions || {
-    logger: (m: any) => console.log(m)
-  }) as any; // 添加类型断言，避免TypeScript错误
-  
+  const worker = (await createWorker(
+    input.tessWorkerOptions || {
+      logger: (m: any) => console.log(m),
+    }
+  )) as any // 添加类型断言，避免TypeScript错误
+
   try {
     // 初始化OCR引擎
-    await worker.load();
-    await worker.loadLanguage('chi_sim');
-    await worker.initialize('chi_sim');
+    await worker.load()
+    await worker.loadLanguage("chi_sim")
+    await worker.initialize("chi_sim")
     await worker.setParameters({
-      tessedit_char_whitelist: '0123456789X-年月日一二三四五六七八九十零壹贰叁肆伍陆柒捌玖拾ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz民族汉族满族回族维吾尔族藏族苗族彝族壮族朝鲜族侗族瑶族白族土家族哈尼族哈萨克族傣族黎族傈僳族佤族高山族拉祜族水族东乡族钠西族景颇族柯尔克孜族士族达斡尔族仫佬族羌族布朗族撒拉族毛南族仡佬族锡伯族阿昌族普米族塔吉克族怒族乌孜别克族俄罗斯族鄂温克族德昂族保安族裕固族京族塔塔尔族独龙族鄂伦春族赫哲族门巴族珞巴族基诺族男女性别住址出生公民身份号码签发机关有效期'
-    });
-    
+      tessedit_char_whitelist:
+        "0123456789X-年月日一二三四五六七八九十零壹贰叁肆伍陆柒捌玖拾ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz民族汉族满族回族维吾尔族藏族苗族彝族壮族朝鲜族侗族瑶族白族土家族哈尼族哈萨克族傣族黎族傈僳族佤族高山族拉祜族水族东乡族钠西族景颇族柯尔克孜族士族达斡尔族仫佬族羌族布朗族撒拉族毛南族仡佬族锡伯族阿昌族普米族塔吉克族怒族乌孜别克族俄罗斯族鄂温克族德昂族保安族裕固族京族塔塔尔族独龙族鄂伦春族赫哲族门巴族珞巴族基诺族男女性别住址出生公民身份号码签发机关有效期",
+    })
+
     // 识别图像
-    const { data } = await worker.recognize(input.imageBase64);
-    
+    const { data } = await worker.recognize(input.imageBase64)
+
     // 解析识别结果
-    const idCardInfo = parseIDCardText(data.text);
-    
+    const idCardInfo = parseIDCardText(data.text)
+
     // 处理完成后终止worker
-    await worker.terminate();
-    
+    await worker.terminate()
+
     // 计算处理时间
-    const processingTime = performance.now() - startTime;
-    
+    const processingTime = performance.now() - startTime
+
     // 返回处理结果
     return {
       idCardInfo,
-      processingTime
-    };
+      processingTime,
+    }
   } catch (error) {
     // 确保资源被释放
-    await worker.terminate();
-    throw error;
+    await worker.terminate()
+    throw error
   }
 }
 
 /**
  * 解析身份证文本信息
- * 
+ *
  * 从OCR识别到的文本中提取结构化的身份证信息
- * 
+ *
  * @private
  * @param {string} text - OCR识别到的文本
  * @returns {IDCardInfo} 提取到的身份证信息对象
  */
 function parseIDCardText(text: string): IDCardInfo {
-  const info: IDCardInfo = {};
-  
+  const info: IDCardInfo = {}
+
   // 拆分为行
-  const lines = text.split('\n').filter(line => line.trim());
-  
+  const lines = text.split("\n").filter((line) => line.trim())
+
   // 解析身份证号码（最容易识别的部分）
-  const idNumberRegex = /(\d{17}[\dX])/;
-  const idNumberMatch = text.match(idNumberRegex);
+  const idNumberRegex = /(\d{17}[\dX])/
+  const idNumberMatch = text.match(idNumberRegex)
   if (idNumberMatch) {
-    info.idNumber = idNumberMatch[1];
+    info.idNumber = idNumberMatch[1]
   }
-  
+
   // 解析姓名
   for (const line of lines) {
-    if (line.includes('姓名') || line.length < 10 && line.length > 1 && !/\d/.test(line)) {
-      info.name = line.replace('姓名', '').trim();
-      break;
+    if (
+      line.includes("姓名") ||
+      (line.length < 10 && line.length > 1 && !/\d/.test(line))
+    ) {
+      info.name = line.replace("姓名", "").trim()
+      break
     }
   }
-  
+
   // 解析性别和民族
-  const genderNationalityRegex = /(男|女).*(族)/;
-  const genderMatch = text.match(genderNationalityRegex);
+  const genderNationalityRegex = /(男|女).*(族)/
+  const genderMatch = text.match(genderNationalityRegex)
   if (genderMatch) {
-    info.gender = genderMatch[1];
-    const nationalityText = genderMatch[0];
-    info.nationality = nationalityText.substring(nationalityText.indexOf(genderMatch[1]) + 1).trim();
+    info.gender = genderMatch[1]
+    const nationalityText = genderMatch[0]
+    info.nationality = nationalityText
+      .substring(nationalityText.indexOf(genderMatch[1]) + 1)
+      .trim()
   }
-  
+
   // 解析出生日期
-  const birthDateRegex = /(\d{4})年(\d{1,2})月(\d{1,2})日/;
-  const birthDateMatch = text.match(birthDateRegex);
+  const birthDateRegex = /(\d{4})年(\d{1,2})月(\d{1,2})日/
+  const birthDateMatch = text.match(birthDateRegex)
   if (birthDateMatch) {
-    info.birthDate = `${birthDateMatch[1]}-${birthDateMatch[2]}-${birthDateMatch[3]}`;
+    info.birthDate = `${birthDateMatch[1]}-${birthDateMatch[2]}-${birthDateMatch[3]}`
   }
-  
+
   // 解析地址
-  const addressRegex = /住址([\s\S]*?)公民身份号码/;
-  const addressMatch = text.match(addressRegex);
+  const addressRegex = /住址([\s\S]*?)公民身份号码/
+  const addressMatch = text.match(addressRegex)
   if (addressMatch) {
-    info.address = addressMatch[1].replace(/\n/g, '').trim();
+    info.address = addressMatch[1].replace(/\n/g, "").trim()
   }
-  
+
   // 解析签发机关
-  const authorityRegex = /签发机关([\s\S]*?)有效期/;
-  const authorityMatch = text.match(authorityRegex);
+  const authorityRegex = /签发机关([\s\S]*?)有效期/
+  const authorityMatch = text.match(authorityRegex)
   if (authorityMatch) {
-    info.issuingAuthority = authorityMatch[1].replace(/\n/g, '').trim();
+    info.issuingAuthority = authorityMatch[1].replace(/\n/g, "").trim()
   }
-  
+
   // 解析有效期限
-  const validPeriodRegex = /有效期限([\s\S]*?)(-|至)/;
-  const validPeriodMatch = text.match(validPeriodRegex);
+  const validPeriodRegex = /有效期限([\s\S]*?)(-|至)/
+  const validPeriodMatch = text.match(validPeriodRegex)
   if (validPeriodMatch) {
-    info.validPeriod = validPeriodMatch[0].replace('有效期限', '').trim();
+    info.validPeriod = validPeriodMatch[0].replace("有效期限", "").trim()
   }
-  
-  return info;
-} 
+
+  return info
+}

package/src/index.ts

@@ -2,7 +2,7 @@
  * @file ID扫描识别库主入口文件
  * @description 提供身份证识别与二维码、条形码扫描功能的纯前端TypeScript库
  * @module IDScannerLib
- * @version 1.0.0
+ * @version 1.3.0
  * @license MIT
  */
 
@@ -18,6 +18,12 @@ import {
 import type { QRScannerOptions } from "./scanner/qr-scanner"
 import type { BarcodeScannerOptions } from "./scanner/barcode-scanner"
 
+// 导入防伪检测器
+import {
+  AntiFakeDetector,
+  AntiFakeDetectionResult,
+} from "./id-recognition/anti-fake-detector"
+
 /**
  * IDScanner配置选项接口
  */
@@ -29,6 +35,7 @@ export interface IDScannerOptions {
   onBarcodeScanned?: (result: string) => void
   onIDCardScanned?: (info: IDCardInfo) => void
   onImageProcessed?: (processedImage: ImageData | File) => void
+  onAntiFakeDetected?: (result: AntiFakeDetectionResult) => void
   onError?: (error: Error) => void
 }
 
@@ -42,15 +49,17 @@ export class IDScanner {
   private camera: Camera
   private scanMode: "qr" | "barcode" | "idcard" = "qr"
   private videoElement: HTMLVideoElement | null = null
-
-  // 延迟加载的模块
+  private scanning = false
   private qrModule: any = null
   private ocrModule: any = null
-
-  // 模块加载状态
+  private scanTimer: number | null = null
   private isQRModuleLoaded: boolean = false
   private isOCRModuleLoaded: boolean = false
 
+  // 新增防伪检测器
+  private antiFakeDetector: AntiFakeDetector | null = null
+  private isAntiFakeModuleLoaded: boolean = false
+
   /**
    * 构造函数
    * @param options 配置选项
@@ -61,7 +70,7 @@ export class IDScanner {
 
   /**
    * 初始化模块
-   * 根据需要初始化OCR引擎
+   * 根据需要初始化OCR引擎和防伪检测模块
    */
   async initialize(): Promise<void> {
     try {
@@ -80,13 +89,44 @@ export class IDScanner {
         await this.ocrModule.initialize()
       }
 
-      console.log("IDScanner initialized")
+      // 初始化防伪检测模块
+      if (!this.isAntiFakeModuleLoaded) {
+        this.antiFakeDetector = new AntiFakeDetector()
+        this.isAntiFakeModuleLoaded = true
+      }
+
+      console.log("IDScanner初始化完成")
     } catch (error) {
+      console.error("初始化失败:", error)
       this.handleError(error as Error)
       throw error
     }
   }
 
+  /**
+   * 初始化OCR模块
+   */
+  private async initOCRModule(): Promise<void> {
+    if (this.isOCRModuleLoaded) return
+
+    try {
+      // 动态导入OCR模块
+      const OCRModule = await import("./ocr-module").then((m) => m.OCRModule)
+      this.ocrModule = new OCRModule({
+        cameraOptions: this.options.cameraOptions,
+        onIDCardScanned: this.options.onIDCardScanned,
+        onError: this.options.onError,
+      })
+      this.isOCRModuleLoaded = true
+
+      // 初始化OCR模块
+      await this.ocrModule.initialize()
+    } catch (error) {
+      console.error("OCR模块初始化失败:", error)
+      throw error
+    }
+  }
+
   /**
    * 启动二维码扫描
    * @param videoElement HTML视频元素
@@ -194,7 +234,7 @@ export class IDScanner {
     if (this.options.onError) {
       this.options.onError(error)
     } else {
-      console.error("IDScanner error:", error)
+      console.error("IDScanner错误:", error)
     }
   }
 
@@ -216,6 +256,13 @@ export class IDScanner {
       this.qrModule = null
       this.isQRModuleLoaded = false
     }
+
+    // 释放防伪检测资源
+    if (this.antiFakeDetector) {
+      this.antiFakeDetector.dispose()
+      this.antiFakeDetector = null
+      this.isAntiFakeModuleLoaded = false
+    }
   }
 
   /**
@@ -388,12 +435,11 @@ export class IDScanner {
   async processIDCardImage(
     imageSource: HTMLImageElement | HTMLCanvasElement | string | File
   ): Promise<IDCardInfo> {
-    try {
-      // 检查OCR模块是否已加载，若未加载则自动初始化
-      if (!this.isOCRModuleLoaded) {
-        await this.initialize()
-      }
+    if (!this.isOCRModuleLoaded) {
+      await this.initOCRModule()
+    }
 
+    try {
       // 处理不同类型的图片源
       let imageElement: HTMLImageElement
 
@@ -460,7 +506,26 @@ export class IDScanner {
       })
 
       // 使用OCR模块处理图像
-      return this.ocrModule.processIDCard(enhancedImageData)
+      const idInfo = await this.ocrModule.processIDCard(enhancedImageData)
+
+      // 进行防伪检测并将结果添加到身份证信息中
+      if (this.isAntiFakeModuleLoaded && this.antiFakeDetector) {
+        try {
+          const result = await this.antiFakeDetector.detect(enhancedImageData)
+          // 将防伪检测结果添加到身份证信息对象中
+          const extendedInfo = idInfo as any
+          extendedInfo.antiFakeResult = result
+
+          // 触发防伪检测回调
+          if (this.options.onAntiFakeDetected) {
+            this.options.onAntiFakeDetected(result)
+          }
+        } catch (error) {
+          console.warn("身份证防伪检测失败:", error)
+        }
+      }
+
+      return idInfo
     } catch (error) {
       this.handleError(error as Error)
       throw error
@@ -596,6 +661,89 @@ export class IDScanner {
       throw error
     }
   }
+
+  /**
+   * 身份证防伪检测
+   * @param imageSource 图片源
+   * @returns 防伪检测结果
+   */
+  async detectIDCardAntiFake(
+    imageSource: HTMLImageElement | HTMLCanvasElement | string | File
+  ): Promise<AntiFakeDetectionResult> {
+    if (!this.isAntiFakeModuleLoaded || !this.antiFakeDetector) {
+      await this.initialize()
+      if (!this.antiFakeDetector) {
+        throw new Error("防伪检测模块初始化失败")
+      }
+    }
+
+    try {
+      // 转换输入为ImageData
+      let imageData: ImageData
+
+      if (typeof imageSource === "string") {
+        // 处理URL或Base64
+        const img = new Image()
+        await new Promise<void>((resolve, reject) => {
+          img.onload = () => resolve()
+          img.onerror = () => reject(new Error("图像加载失败"))
+          img.src = imageSource
+        })
+
+        const canvas = document.createElement("canvas")
+        canvas.width = img.width
+        canvas.height = img.height
+        const ctx = canvas.getContext("2d")
+        if (!ctx) {
+          throw new Error("无法创建Canvas上下文")
+        }
+
+        ctx.drawImage(img, 0, 0)
+        imageData = ctx.getImageData(0, 0, canvas.width, canvas.height)
+      } else if (imageSource instanceof File) {
+        // 处理文件
+        imageData = await ImageProcessor.createImageDataFromFile(imageSource)
+      } else if (imageSource instanceof HTMLImageElement) {
+        // 处理Image元素
+        const canvas = document.createElement("canvas")
+        canvas.width = imageSource.width
+        canvas.height = imageSource.height
+        const ctx = canvas.getContext("2d")
+        if (!ctx) {
+          throw new Error("无法创建Canvas上下文")
+        }
+
+        ctx.drawImage(imageSource, 0, 0)
+        imageData = ctx.getImageData(0, 0, canvas.width, canvas.height)
+      } else {
+        // 处理Canvas元素
+        const ctx = (imageSource as HTMLCanvasElement).getContext("2d")
+        if (!ctx) {
+          throw new Error("无法获取Canvas上下文")
+        }
+
+        imageData = ctx.getImageData(
+          0,
+          0,
+          imageSource.width,
+          imageSource.height
+        )
+      }
+
+      // 执行防伪检测
+      const result = await this.antiFakeDetector.detect(imageData)
+
+      // 触发回调
+      if (this.options.onAntiFakeDetected) {
+        this.options.onAntiFakeDetected(result)
+      }
+
+      return result
+    } catch (error) {
+      this.handleError(error as Error)
+      throw error
+    }
+  }
 }
 
 // 导出工具类和类型
@@ -607,6 +755,9 @@ export {
 } from "./utils/image-processing"
 export { IDCardInfo, DetectionResult } from "./utils/types"
 
+// 导出防伪检测相关类型和类
+export { AntiFakeDetector, AntiFakeDetectionResult }
+
 // 为了向后兼容，我们创建一个演示类
 export class IDScannerDemo {
   private scanner: IDScanner
@@ -759,7 +910,7 @@ export class IDScannerDemo {
   private handleIDCardResult(info: IDCardInfo): void {
     // 格式化显示身份证信息
     const infoHtml = Object.entries(info)
-      .filter(([key, value]) => value) // 过滤掉空值
+      .filter(([key, value]) => value && key !== "antiFakeResult") // 过滤掉空值和防伪结果
       .map(([key, value]) => {
         // 转换键名为中文显示
         const keyMap: { [key: string]: string } = {
@@ -778,9 +929,32 @@ export class IDScannerDemo {
       })
       .join("")
 
+    // 检查是否有防伪检测结果
+    let antiFakeHtml = ""
+    const anyInfo = info as any
+    if (anyInfo.antiFakeResult) {
+      const antiFakeResult = anyInfo.antiFakeResult
+      antiFakeHtml = `
+        <h3>防伪检测结果:</h3>
+        <div style="color: ${antiFakeResult.isAuthentic ? "green" : "red"}">
+          ${
+            antiFakeResult.isAuthentic
+              ? "✓ 身份证真实"
+              : "⚠ 警告：可能为伪造证件"
+          }
+        </div>
+        <div>置信度: ${(antiFakeResult.confidence * 100).toFixed(1)}%</div>
+        <div>检测到的特征: ${
+          antiFakeResult.detectedFeatures.join(", ") || "无"
+        }</div>
+        <div>${antiFakeResult.message}</div>
+      `
+    }
+
     this.updateResultDisplay(`
       <h3>身份证信息:</h3>
       ${infoHtml}
+      ${antiFakeHtml}
     `)
   }
 

package/src/types/browser-image-compression.d.ts

@@ -0,0 +1,19 @@
+/**
+ * Type definitions for browser-image-compression
+ */
+
+declare module 'browser-image-compression' {
+  export interface Options {
+    maxSizeMB?: number;
+    maxWidthOrHeight?: number;
+    useWebWorker?: boolean;
+    maxIteration?: number;
+    quality?: number;
+    fileType?: string;
+    onProgress?: (progress: number) => void;
+  }
+
+  function imageCompression(file: File, options?: Options): Promise<File>;
+  
+  export default imageCompression;
+}

package/src/types/tesseract.d.ts

@@ -0,0 +1,37 @@
+/**
+ * Type definitions for tesseract.js
+ */
+
+declare module 'tesseract.js' {
+  export interface WorkerOptions {
+    langPath?: string;
+    corePath?: string;
+    workerPath?: string;
+    logger?: (message: any) => void;
+    errorHandler?: (error: Error) => void;
+  }
+
+  export interface RecognizeResult {
+    data: {
+      text: string;
+      hocr: string;
+      tsv: string;
+      confidence: number;
+      blocks: any[];
+      lines: any[];
+      words: any[];
+      symbols: any[];
+    };
+  }
+
+  export interface Worker {
+    load(): Promise<any>;
+    loadLanguage(lang: string): Promise<any>;
+    initialize(lang: string): Promise<any>;
+    setParameters(params: Record<string, any>): Promise<any>;
+    recognize(image: HTMLCanvasElement | ImageData | Buffer | string): Promise<RecognizeResult>;
+    terminate(): Promise<any>;
+  }
+
+  export function createWorker(options?: WorkerOptions): Worker;
+}