MyQRCode/MyQrCode/Utils/QRCodeGenerator.swift

import SwiftUI
import CoreImage
import QRCode

// MARK: - 二维码生成工具类
class QRCodeGenerator {
    
    // MARK: - 生成基本二维码（使用Core Image）
    static func generateBasicQRCode(content: String) -> UIImage {
        let data = content.data(using: .utf8)
        let qrFilter = CIFilter.qrCodeGenerator()
        qrFilter.setValue(data, forKey: "inputMessage")
        qrFilter.setValue("H", forKey: "inputCorrectionLevel")
        
        guard let outputImage = qrFilter.outputImage else {
            return UIImage(systemName: "qrcode") ?? UIImage()
        }
        
        let context = CIContext()
        guard let cgImage = context.createCGImage(outputImage, from: outputImage.extent) else {
            return UIImage(systemName: "qrcode") ?? UIImage()
        }
        
        return UIImage(cgImage: cgImage)
    }
    
    // MARK: - 生成高质量二维码（使用QRCode库）
    static func generateHighQualityQRCode(
        content: String,
        styleData: QRCodeStyleData? = nil
    ) -> UIImage {
        do {
            let document = try createQRCodeDocument(content: content, styleData: styleData)
            let imageData = try document.pngData(dimension: 600)
            return UIImage(data: imageData) ?? generateBasicQRCode(content: content)
        } catch {
            print("生成高质量二维码失败：\(error.localizedDescription)")
            return generateBasicQRCode(content: content)
        }
    }
    
    // MARK: - 创建QRCode文档
    private static func createQRCodeDocument(
        content: String,
        styleData: QRCodeStyleData?
    ) throws -> QRCode.Document {
        let document = try QRCode.Document(engine: QRCodeEngineExternal())
        
        // 设置内容
        document.utf8String = content
        
        // 如果没有样式数据，使用默认样式
        guard let styleData = styleData else {
            // 使用默认样式
            document.design.backgroundColor(CGColor(red: 1, green: 1, blue: 1, alpha: 1)) // 白色背景
            document.design.style.eye = QRCode.FillStyle.Solid(CGColor(red: 0, green: 0, blue: 0, alpha: 1)) // 黑色眼睛
            document.design.style.eyeBackground = CGColor(red: 1, green: 1, blue: 1, alpha: 1) // 白色眼睛背景
            document.design.shape.onPixels = QRCode.PixelShape.Square() // 方形点
            document.design.style.onPixels = QRCode.FillStyle.Solid(CGColor(red: 0, green: 0, blue: 0, alpha: 1)) // 黑色点
            document.design.style.onPixelsBackground = CGColor(red: 1, green: 1, blue: 1, alpha: 1) // 白色点背景
            document.design.shape.offPixels = QRCode.PixelShape.Square() // 方形空白
            document.design.style.offPixels = QRCode.FillStyle.Solid(CGColor(red: 1, green: 1, blue: 1, alpha: 1)) // 白色空白
            document.design.style.offPixelsBackground = CGColor(red: 1, green: 1, blue: 1, alpha: 1) // 白色空白背景
            document.design.shape.eye = QRCode.EyeShape.Square() // 方形眼睛
            return document
        }
        
        // 应用样式数据
        applyStyleData(styleData, to: document)
        
        return document
    }
    
    // MARK: - 应用样式数据
    private static func applyStyleData(_ styleData: QRCodeStyleData, to document: QRCode.Document) {
        // 设置背景色
        if let backgroundColor = getColor(from: styleData.backgroundColor) {
            document.design.backgroundColor(backgroundColor)
        }
        
        // 设置前景色
        let foregroundColor = getColor(from: styleData.foregroundColor) ?? CGColor(red: 0, green: 0, blue: 0, alpha: 1)
        
        // 设置眼睛样式
        document.design.style.eye = QRCode.FillStyle.Solid(foregroundColor)
        document.design.style.eyeBackground = getColor(from: styleData.backgroundColor) ?? CGColor(red: 1, green: 1, blue: 1, alpha: 1)
        
        // 设置点样式
        let dotType = getDotType(from: styleData.dotType)
        document.design.shape.onPixels = dotType
        document.design.style.onPixels = QRCode.FillStyle.Solid(foregroundColor)
        document.design.style.onPixelsBackground = getColor(from: styleData.backgroundColor) ?? CGColor(red: 1, green: 1, blue: 1, alpha: 1)
        
        document.design.shape.offPixels = dotType
        document.design.style.offPixels = QRCode.FillStyle.Solid(getColor(from: styleData.backgroundColor) ?? CGColor(red: 1, green: 1, blue: 1, alpha: 1))
        document.design.style.offPixelsBackground = getColor(from: styleData.backgroundColor) ?? CGColor(red: 1, green: 1, blue: 1, alpha: 1)
        
        // 设置眼睛形状
        let eyeType = getEyeType(from: styleData.eyeType)
        document.design.shape.eye = eyeType
        
        // 设置Logo
        if let logo = styleData.logo {
            applyLogo(logo, hasCustomLogo: styleData.hasCustomLogo, customLogoFileName: styleData.customLogoFileName, to: document)
        }
    }
    
    // MARK: - 应用Logo
    private static func applyLogo(
        _ logoIdentifier: String,
        hasCustomLogo: Bool,
        customLogoFileName: String?,
        to document: QRCode.Document
    ) {
        if hasCustomLogo, let fileName = customLogoFileName {
            // 加载自定义Logo
            if let customLogoImage = loadCustomLogoImage(fileName: fileName),
               let cgImage = customLogoImage.cgImage {
                document.logoTemplate = QRCode.LogoTemplate.CircleCenter(image: cgImage, inset: 0)
            }
        } else {
            // 加载预设Logo
            if let qrCodeLogo = QRCodeLogo(rawValue: logoIdentifier),
               let logoImage = qrCodeLogo.image,
               let cgImage = logoImage.cgImage {
                document.logoTemplate = QRCode.LogoTemplate.CircleCenter(image: cgImage)
            }
        }
    }
    
    // MARK: - 加载自定义Logo图片
    private static func loadCustomLogoImage(fileName: String) -> UIImage? {
        guard let documentsPath = FileManager.default.urls(for: .documentDirectory, in: .userDomainMask).first else {
            return nil
        }
        
        let customLogosPath = documentsPath.appendingPathComponent("CustomLogos")
        let imagePath = customLogosPath.appendingPathComponent(fileName)
        
        return UIImage(contentsOfFile: imagePath.path)
    }
    
    // MARK: - 颜色转换
    private static func getColor(from colorString: String) -> CGColor? {
        switch colorString.lowercased() {
        case "black": return CGColor(red: 0, green: 0, blue: 0, alpha: 1)
        case "white": return CGColor(red: 1, green: 1, blue: 1, alpha: 1)
        case "red": return CGColor(red: 1, green: 0, blue: 0, alpha: 1)
        case "blue": return CGColor(red: 0, green: 0, blue: 1, alpha: 1)
        case "green": return CGColor(red: 0, green: 1, blue: 0, alpha: 1)
        case "yellow": return CGColor(red: 1, green: 1, blue: 0, alpha: 1)
        case "purple": return CGColor(red: 0.5, green: 0, blue: 0.5, alpha: 1)
        case "orange": return CGColor(red: 1, green: 0.5, blue: 0, alpha: 1)
        case "pink": return CGColor(red: 1, green: 0.75, blue: 0.8, alpha: 1)
        case "cyan": return CGColor(red: 0, green: 1, blue: 1, alpha: 1)
        case "magenta": return CGColor(red: 1, green: 0, blue: 1, alpha: 1)
        case "brown": return CGColor(red: 0.6, green: 0.4, blue: 0.2, alpha: 1)
        case "gray": return CGColor(red: 0.5, green: 0.5, blue: 0.5, alpha: 1)
        case "navy": return CGColor(red: 0, green: 0, blue: 0.5, alpha: 1)
        case "teal": return CGColor(red: 0, green: 0.5, blue: 0.5, alpha: 1)
        case "indigo": return CGColor(red: 0.3, green: 0, blue: 0.7, alpha: 1)
        case "lime": return CGColor(red: 0.5, green: 1, blue: 0, alpha: 1)
        case "maroon": return CGColor(red: 0.5, green: 0, blue: 0, alpha: 1)
        case "olive": return CGColor(red: 0.5, green: 0.5, blue: 0, alpha: 1)
        case "silver": return CGColor(red: 0.75, green: 0.75, blue: 0.75, alpha: 1)
        default: return nil
        }
    }
    
    // MARK: - 点类型转换
    private static func getDotType(from dotTypeString: String) -> QRCodePixelShapeGenerator {
        // 根据字符串匹配到对应的QRCodeDotType
        if let dotType = QRCodeDotType.allCases.first(where: { $0.rawValue == dotTypeString }) {
            return dotType.pixelShape
        }
        
        // 如果没有找到匹配的类型，返回默认的方形
        return QRCode.PixelShape.Square()
    }
    
    // MARK: - 眼睛类型转换
    private static func getEyeType(from eyeTypeString: String) -> QRCodeEyeShapeGenerator {
        // 根据字符串匹配到对应的QRCodeEyeType
        if let eyeType = QRCodeEyeType.allCases.first(where: { $0.rawValue == eyeTypeString }) {
            return eyeType.eyeShape
        }
        
        // 如果没有找到匹配的类型，返回默认的方形
        return QRCode.EyeShape.Square()
    }
}