PDF_Office/PDF Office/PDF Master/KMClass/Tools/Common/KMEnumExtensions.swift

//
//  KMEnumExtensions.swift
//  PDF Reader Pro
//
//  Created by tangchao on 2023/7/14.
//

import Foundation

@objc enum KMSubscribeWaterMarkType: Int {
    case none       = 0
    case stamp      = 1
    case link
    case sign
    case editText
    case editImage
    case insert
    case extract
    case replace
    case split
    case delete
    case rotate
    case copy
    case toWord
    case toExcel
    case toPPT
    case toRTF
    case toCSV
    case toHTML
    case toText
    case toImage
    case compress
    case merge
    case setPassword
    case removePassword
    case crop
    case aiTranslate
    case aiRewrite
    case aiCorrect
    case save
    case searchReplace
    case adsWeb
    case openTab
    case imageToPDF
    case batch
    case pdfEdit
    case print
    case ocr
    case fileCompare
    case watermark
    case background
    case headerfooter
    case bates
    case redact
    case digitalSign
    case form
    case share
    case measure
    case themeColor
    
    case invalid
    
    func isConvertType() -> Bool {
        if (self == .toWord || self == .toExcel || self == .toPPT || self == .toRTF || self == .toCSV || self == .toHTML || self == .toText || self == .toImage) {
            return true
        }
        return false
    }
}

protocol KMSubscribeWaterMarkTypeConvertProtocol {
    func toSubscribeWaterMarkType() -> KMSubscribeWaterMarkType
}

extension KMToolbarType: KMSubscribeWaterMarkTypeConvertProtocol {
    func toSubscribeWaterMarkType() -> KMSubscribeWaterMarkType {
        switch self {
        case .word:
            return .toWord
        case .excel:
            return .toExcel
        case .ppt:
            return .toPPT
        case .rtf:
            return .toRTF
        case .csv:
            return .toCSV
        case .html:
            return .toHTML
        case .conversion_text:
            return .toText
        case .conversion_image:
            return .toImage
        case .compress:
            return .compress
        case .merge:
            return .merge
        default:
            return .none
        }
    }
}

extension KMPDFConvertType: KMSubscribeWaterMarkTypeConvertProtocol {
    func toSubscribeWaterMarkType() -> KMSubscribeWaterMarkType {
        switch self {
        case .word:
            return .toWord
        case .excel:
            return .toExcel
        case .ppt:
            return .toPPT
        case .rtf:
            return .toRTF
        case .csv:
            return .toCSV
        case .html:
            return .toHTML
        case .text:
            return .toText
        case .image:
            return .toImage
        default:
            return .none
        }
    }
}

extension KMItemKey: KMSubscribeWaterMarkTypeConvertProtocol {
    func toSubscribeWaterMarkType() -> KMSubscribeWaterMarkType {
        switch self {
        case .leftRotate:
            return .rotate
        case .rightRotate:
            return .rotate
        case .paste:
            return .copy
        case .delete:
            return .delete
        default:
            return .none
        }
    }
}

extension KMPageEditType: KMSubscribeWaterMarkTypeConvertProtocol {
    func toSubscribeWaterMarkType() -> KMSubscribeWaterMarkType {
        switch self {
        case .leftRotate:
            return .rotate
        case .rightRotate:
            return .rotate
        case .insert:
            return .insert
        case .extract:
            return .extract
        case .replace:
            return .replace
        case .split:
            return .split
        case .reverse:
            return .none
        case .delete:
            return .delete
        case .paste:
            return .copy
        default:
            return .none
        }
    }
}