element-plus/packages/components/loading/src/directive.ts

import Loading from './index'

import type { DirectiveBinding } from 'vue'
import type { ILoadingInstance } from './loading.type'

const INSTANCE_NAME = 'ElLoading'

export interface ElementLoading extends HTMLElement {
  [INSTANCE_NAME]?: ILoadingInstance
}

const createInstance = (el: ElementLoading, binding: DirectiveBinding) => {
  const textExr = el.getAttribute('element-loading-text')
  const spinnerExr = el.getAttribute('element-loading-spinner')
  const svgExr = el.getAttribute('element-loading-svg')
  const svgViewBoxExr = el.getAttribute('element-loading-svg-view-box')
  const backgroundExr = el.getAttribute('element-loading-background')
  const customClassExr = el.getAttribute('element-loading-custom-class')
  const vm = binding.instance
  el[INSTANCE_NAME] = Loading({
    text: (vm && vm[textExr]) || textExr,
    svg: (vm && vm[svgExr]) || svgExr,
    svgViewBox: (vm && vm[svgViewBoxExr]) || svgViewBoxExr,
    spinner: (vm && vm[spinnerExr]) || spinnerExr,
    background: (vm && vm[backgroundExr]) || backgroundExr,
    customClass: (vm && vm[customClassExr]) || customClassExr,
    fullscreen: !!binding.modifiers.fullscreen,
    target: binding.modifiers.fullscreen ? null : el,
    body: !!binding.modifiers.body,
    visible: true,
    lock: !!binding.modifiers.lock,
  })
}

const vLoading = {
  mounted(el: ElementLoading, binding: DirectiveBinding) {
    if (binding.value) {
      createInstance(el, binding)
    }
  },
  updated(el: ElementLoading, binding: DirectiveBinding) {
    const instance = el[INSTANCE_NAME]
    if (binding.oldValue !== binding.value) {
      if (binding.value) {
        createInstance(el, binding)
      } else {
        instance?.close()
      }
    }
  },
  unmounted(el: ElementLoading) {
    el[INSTANCE_NAME]?.close()
  },
}

export default vLoading