koel/resources/assets/js/composables/useUpload.ts

import isMobile from 'ismobilejs'
import { computed, toRef } from 'vue'
import { useAuthorization } from '@/composables/useAuthorization'
import { settingStore } from '@/stores'
import { acceptedMediaTypes, UploadFile } from '@/config'
import { uploadService } from '@/services'
import { getAllFileEntries, pluralize, requireInjection } from '@/utils'
import { ActiveScreenKey, MessageToasterKey } from '@/symbols'
import router from '@/router'

export const useUpload = () => {
  const { isAdmin } = useAuthorization()

  const activeScreen = requireInjection(ActiveScreenKey)
  const toaster = requireInjection(MessageToasterKey)
  const mediaPath = toRef(settingStore.state, 'media_path')

  const mediaPathSetUp = computed(() => Boolean(mediaPath.value))
  const allowsUpload = computed(() => isAdmin.value && !isMobile.any)

  const fileEntryToFile = async (entry: FileSystemEntry) => new Promise<File>(resolve => entry.file(resolve))

  const queueFilesForUpload = (files: Array<File>) => {
    const uploadCandidates = files
      .filter(file => acceptedMediaTypes.includes(file.type))
      .map((file): UploadFile => ({
        file,
        id: `${file.name}-${file.size}`, // for simplicity, a file's identity is determined by its name and size
        status: 'Ready',
        name: file.name,
        progress: 0
      }))

    uploadService.queue(uploadCandidates)

    return uploadCandidates
  }

  const handleDropEvent = async (event: DragEvent) => {
    if (!event.dataTransfer) {
      return
    }

    const fileEntries = await getAllFileEntries(event.dataTransfer.items)
    const files = await Promise.all(fileEntries.map(async entry => await fileEntryToFile(entry)))
    const queuedFiles = queueFilesForUpload(files)

    if (queuedFiles.length) {
      toaster.value.success(`Queued ${pluralize(queuedFiles, 'file')} for upload`)
      activeScreen.value === 'Upload' || router.go('upload')
    } else {
      toaster.value.warning('No files applicable for upload')
    }
  }

  return {
    mediaPathSetUp,
    allowsUpload,
    handleDropEvent,
    queueFilesForUpload
  }
}