src/main/ui/TouchBarManager.js - master - githubmirror/Motrix

src/main/ui/TouchBarManager.js
109 строк2 KB
Dr_rOot
refactor: code format
06 май 2023, 13:16
06 май 2023, 13:161f033d1
import { EventEmitter } from 'node:events'
import { join } from 'node:path'
import { TouchBar, nativeImage } from 'electron'

import { handleCommand } from '../utils/menu'
import logger from '../core/Logger'

const { TouchBarButton, TouchBarLabel, TouchBarSpacer, TouchBarGroup } = TouchBar

export default class TouchBarManager extends EventEmitter {
  constructor (options) {
    super()
    this.options = options
    this.bars = {}
    this.load()
  }

  load () {
    this.template = require('../menus/touchBar.json')
  }

  getClickFn (item) {
    let fn = () => {}
    if (item.command) {
      fn = () => {
        handleCommand(item)
      }
    }
    return fn
  }

  getIconImage (icon) {
    if (!icon) {
      return
    }
    const img = join(__static, `./icons/${icon}.png`)
    return nativeImage.createFromPath(img)
  }

  buildItem (type, options) {
    let result = null
    const { label, backgroundColor, textColor, size } = options

    switch (type) {
    case 'button':
      result = new TouchBarButton({
        label,
        backgroundColor,
        icon: this.getIconImage(options.icon),
        click: this.getClickFn(options)
      })
      break
    case 'label':
      result = new TouchBarLabel({
        label,
        textColor
      })
      break
    case 'spacer':
      result = new TouchBarSpacer({ size })
      break
    case 'group':
      result = new TouchBarGroup({
        items: new TouchBar({
          items: options.items
        })
      })
      break
    default:
      result = null
    }

    return result
  }

  build (template) {
    const result = []

    template.forEach(tpl => {
      const { id, type, ...rest } = tpl
      let options = { ...rest }
      if (type === 'group') {
        options = { type, items: this.build(options.items) }
      }
      const item = this.buildItem(type, options)
      result.push(item)
    })
    return result
  }

  getTouchBarByPage (page) {
    let bar = this.bars[page] || null
    if (!bar) {
      try {
        const items = this.build(this.template)
        bar = new TouchBar({ items })
        this.bars[page] = bar
      } catch (e) {
        logger.info('getTouchBarByPage fail', e)
      }
    }
    return bar
  }

  setup (page, window) {
    const bar = this.getTouchBarByPage(page)
    window.setTouchBar(bar)
  }
}