mind-map/simple-mind-map/src/KeyCommand.js

import { keyMap } from './utils/keyMap';
/** 
 * @Author: 王林 
 * @Date: 2021-04-24 15:20:46 
 * @Desc: 快捷按键、命令处理类 
 */
export default class KeyCommand {
    /** 
     * @Author: 王林 
     * @Date: 2021-04-24 15:21:32 
     * @Desc: 构造函数 
     */
    constructor(opt) {
        this.opt = opt
        this.mindMap = opt.mindMap
        this.shortcutMap = {
            //Enter: [fn]
        }
        this.shortcutMapCache = {}
        this.isPause = false
        this.bindEvent()
    }

    /** 
     * @Author: 王林 
     * @Date: 2022-08-14 08:57:55 
     * @Desc: 暂停快捷键响应 
     */
    pause() {
        this.isPause = true
    }

    /** 
     * @Author: 王林 
     * @Date: 2022-08-14 08:58:43 
     * @Desc: 恢复快捷键响应 
     */
    recovery() {
        this.isPause = false
    }

    /** 
     * javascript comment 
     * @Author: 王林25 
     * @Date: 2022-08-16 16:29:01 
     * @Desc: 保存当前注册的快捷键数据，然后清空快捷键数据
     */
    save() {
        this.shortcutMapCache = this.shortcutMap
        this.shortcutMap = {}
    }

    /** 
     * javascript comment 
     * @Author: 王林25 
     * @Date: 2022-08-16 16:29:38 
     * @Desc: 恢复保存的快捷键数据，然后清空缓存数据 
     */
    restore() {
        this.shortcutMap = this.shortcutMapCache
        this.shortcutMapCache = {}
    }

    /** 
     * @Author: 王林 
     * @Date: 2021-04-24 15:23:22 
     * @Desc: 绑定事件 
     */
    bindEvent() {
        window.addEventListener('keydown', (e) => {
            if (this.isPause) {
                return
            }
            Object.keys(this.shortcutMap).forEach((key) => {
                if (this.checkKey(e, key)) {
                    e.stopPropagation()
                    e.preventDefault()
                    this.shortcutMap[key].forEach((fn) => {
                        fn()
                    })
                }
            })
        })
    }

    /** 
     * @Author: 王林 
     * @Date: 2021-04-24 19:24:53 
     * @Desc: 检查键值是否符合 
     */
    checkKey(e, key) {
        let o = this.getOriginEventCodeArr(e)
        let k = this.getKeyCodeArr(key)
        if (o.length !== k.length) {
            return false
        }
        for (let i = 0; i < o.length; i++) {
            let index = k.findIndex((item) => {
                return item === o[i];
            })
            if (index === -1) {
                return false
            } else {
                k.splice(index, 1)
            }
        }
        return true
    }

    /** 
     * @Author: 王林 
     * @Date: 2021-04-24 19:15:19 
     * @Desc: 获取事件对象里的键值数组 
     */
    getOriginEventCodeArr(e) {
        let arr = []
        if (e.ctrlKey || e.metaKey) {
            arr.push(keyMap['Control'])
        }
        if (e.altKey) {
            arr.push(keyMap['Alt'])
        }
        if (e.shiftKey) {
            arr.push(keyMap['Shift'])
        }
        if (!arr.includes(e.keyCode)) {
            arr.push(e.keyCode)
        }
        return arr
    }

    /** 
     * @Author: 王林 
     * @Date: 2021-04-24 19:40:11 
     * @Desc: 获取快捷键对应的键值数组 
     */
    getKeyCodeArr(key) {
        let keyArr = key.split(/\s*\+\s*/)
        let arr = []
        keyArr.forEach((item) => {
            arr.push(keyMap[item])
        })
        return arr
    }

    /** 
     * @Author: 王林 
     * @Date: 2021-04-24 15:23:00 
     * @Desc: 添加快捷键命令 
     * Enter
     * Tab | Insert
     * Shift + a
     */
    addShortcut(key, fn) {
        key.split(/\s*\|\s*/).forEach((item) => {
            if (this.shortcutMap[item]) {
                this.shortcutMap[item].push(fn)
            } else {
                this.shortcutMap[item] = [fn]
            }
        })
    }

    /** 
     * javascript comment 
     * @Author: 王林25 
     * @Date: 2021-07-27 14:06:16 
     * @Desc: 移除快捷键命令 
     */
    removeShortcut(key, fn) {
        key.split(/\s*\|\s*/).forEach((item) => {
            if (this.shortcutMap[item]) {
                if (fn) {
                    let index = this.shortcutMap[item].findIndex((f) => {
                        return f === fn
                    })
                    if (index !== -1) {
                        this.shortcutMap[item].splice(index, 1)
                    }
                } else {
                    this.shortcutMap[item] = []
                    delete this.shortcutMap[item]
                }
            }
        })
    }

    /** 
     * @Author: 王林 
     * @Date: 2022-08-14 08:49:58 
     * @Desc: 获取指定快捷键的处理函数 
     */
    getShortcutFn(key) {
        let res = []
        key.split(/\s*\|\s*/).forEach((item) => {
            res = this.shortcutMap[item] || []
        })
        return res
    }
}