extensions/sd-webui-infinite-image-browsing/vue/src/page/fileTransfer/fileSort.ts

import type { FileNodeInfo } from '@/api/files'
import { t } from '@/i18n'
import { SearchSelectConv } from 'vue3-ts-util'

export const sortMethodMap = (): Record<SortMethod, string> => ({
  'date-asc': t('sortByDateAscending'),
  'date-desc': t('sortByDateDescending'),
  'name-asc': t('sortByNameAscending'),
  'name-desc': t('sortByNameDescending'),
  'size-asc': t('sortBySizeAscending'),
  'size-desc': t('sortBySizeDescending'),
  'created-time-asc': t('sortByCreatedDateAscending'),
  'created-time-desc': t('sortByCreatedDateDescending')
})
export enum SortMethod {
  DATE_ASC = 'date-asc',
  DATE_DESC = 'date-desc',
  NAME_ASC = 'name-asc',
  NAME_DESC = 'name-desc',
  SIZE_ASC = 'size-asc',
  SIZE_DESC = 'size-desc',
  CREATED_TIME_ASC = 'created-time-asc',
  CREATED_TIME_DESC = 'created-time-desc'
}

export const sortMethods = Object.values(SortMethod) as SortMethod[]


export const sortMethodConv: SearchSelectConv<SortMethod> = {
  value: (v) => v,
  text: (v) => sortMethodMap()[v].toLocaleLowerCase()
}
type FileList = FileNodeInfo[]

const compareByType = (a: FileNodeInfo, b: FileNodeInfo) => {
  const sa = a.type === 'dir' ? 1 : 0
  const sb = b.type === 'dir' ? 1 : 0
  return sb - sa
}

const compareByDate = (a: FileNodeInfo, b: FileNodeInfo) => {
  const da = Date.parse(a.date)
  const db = Date.parse(b.date)
  return da - db
}

const compareByCreatedDate = (a: FileNodeInfo, b: FileNodeInfo) => {
  const da = Date.parse(a.created_time)
  const db = Date.parse(b.created_time)
  return da - db
}

const compareByName = (a: FileNodeInfo, b: FileNodeInfo) => {
  const an = a.name.toLowerCase()
  const bn = b.name.toLowerCase()
  return an.localeCompare(bn)
}

const compareBySize = (a: FileNodeInfo, b: FileNodeInfo) => {
  return a.bytes - b.bytes
}

export const sortFiles = (files: FileList, method: SortMethod) => {
  const compare = (a: FileNodeInfo, b: FileNodeInfo) => {
    switch (method) {
      case SortMethod.DATE_ASC:
        return compareByDate(a, b)
      case SortMethod.DATE_DESC:
        return compareByDate(b, a)
      case SortMethod.CREATED_TIME_ASC:
        return compareByCreatedDate(a, b)
      case SortMethod.CREATED_TIME_DESC:
        return compareByCreatedDate(b, a)
      case SortMethod.NAME_ASC:
        return compareByName(a, b)
      case SortMethod.NAME_DESC:
        return compareByName(b, a)
      case SortMethod.SIZE_ASC:
        return compareBySize(a, b)
      case SortMethod.SIZE_DESC:
        return compareBySize(b, a)
      default:
        throw new Error(`Invalid sort method: ${method}`)
    }
  }
  return files.slice().sort((a, b) => compareByType(a, b) || compare(a, b))
}