Element Plus實現Affix 固釘

<template>
  <div ref="root" class="el-affix" :style="rootStyle">
    <div :class="{'el-affix--fixed': state.fixed}" :style="affixStyle">
      <slot></slot>
    </div>
  </div>
</template>

// 部分核心代碼，代碼順序有所調整
setup(props, { emit }) {
    // target容器 ref
    const target = ref(null) 
    // 固釘ref，與template中的ref屬性配合，得到HTML元素
    const root = ref(null)
    // 滾動容器ref
    const scrollContainer = ref(null)
    
    // 固釘狀態
    const state = reactive({
      fixed: false,
      height: 0, // height of root
      width: 0, // width of root
      scrollTop: 0, // scrollTop of documentElement
      clientHeight: 0, // clientHeight of documentElement
      transform: 0,
    })
    
    onMounted(() => {
      // 根據傳入的target確定 target容器
      if (props.target) {
        target.value = document.querySelector(props.target)
        if (!target.value) {
          throw new Error(`target is not existed: ${props.target}`)
        }
      } else {
        target.value = document.documentElement
      }
      
      // 根據固釘元素，向上尋找滾動容器
      scrollContainer.value = getScrollContainer(root.value)
      // 監聽滾動容器的scroll事件
      on(scrollContainer.value, 'scroll', onScroll)
      // 監聽固釘元素的resize事件
      addResizeListener(root.value, updateState)
    })
    
    // 滾動容器的scroll事件的響應函數
    const onScroll = () => {
      // 更新固釘狀態
      updateState()
      
      emit('scroll', {
        scrollTop: state.scrollTop,
        fixed: state.fixed,
      })
    }
    
    // 更新固釘狀態函數
    const updateState = () => {
      const rootRect = root.value.getBoundingClientRect()
      const targetRect = target.value.getBoundingClientRect()
      state.height = rootRect.height
      state.width = rootRect.width
      state.scrollTop = scrollContainer.value === window ? document.documentElement.scrollTop : scrollContainer.value.scrollTop
      state.clientHeight = document.documentElement.clientHeight

      if (props.position === 'top') {
        if (props.target) {
          const difference = targetRect.bottom - props.offset - state.height
          // targetRect.bottom > 0 對應的是讓固釘始終保持在容器內，超過范圍則隱藏
          state.fixed = props.offset > rootRect.top && targetRect.bottom > 0
          // 用於處理場景：滾動過程中，target容器可視區域不足以顯示整個固釘，則固釘應相應偏移，隻展示部分
          state.transform = difference < 0 ? difference : 0
        } else {
          state.fixed = props.offset > rootRect.top
        }
      } else {
        if (props.target) {
          const difference = state.clientHeight - targetRect.top - props.offset - state.height
          state.fixed = state.clientHeight - props.offset < rootRect.bottom && state.clientHeight > targetRect.top
          state.transform = difference < 0 ? -difference : 0
        } else {
          state.fixed = state.clientHeight - props.offset < rootRect.bottom
        }
      }
    }
    // 監測固釘fixed狀態變化，並對外emit change事件
    watch(() => state.fixed, () => {
      emit('change', state.fixed)
    })
    
    // 計算屬性，通過固釘的狀態自動更新固釘的樣式
    const affixStyle = computed(() => {
      if (!state.fixed) {
        return
      }
      const offset = props.offset ? `${props.offset}px` : 0
      const transform = state.transform ? `translateY(${state.transform}px)` : ''

      return {
        height: `${state.height}px`,
        width: `${state.width}px`,
        top: props.position === 'top' ? offset : '',
        bottom: props.position === 'bottom' ? offset : '',
        transform: transform,
        zIndex: props.zIndex,
      }
    })
}