TmMolStar/src/mol-util/file-info.ts

/**
 * Copyright (c) 2018 mol* contributors, licensed under MIT, See LICENSE file for more info.
 *
 * @author Alexander Rose <alexander.rose@weirdbyte.de>
 */

/** A File or Blob object or a URL string */
export type FileInput = File | Blob | string

// TODO only support compressed files for which uncompression support is available???
// TODO store globally with decompression plugins?
const compressedExtList = [ 'gz', 'zip' ]

// TODO store globally with parser plugins?
const binaryExtList = [ 'bcif', 'ccp4', 'dcd' ]

export interface FileInfo {
    path: string
    name: string
    ext: string
    base: string
    dir: string
    compressed: string | boolean
    binary: boolean
    protocol: string
    query: string
    src: FileInput
}

export function getFileInfo (file: FileInput): FileInfo {
    let path: string
    let compressed: string|false
    let protocol = ''

    if (file instanceof File) {
        path = file.name
    } else if (file instanceof Blob) {
        path = ''
    } else {
        path = file
    }
    const queryIndex = path.lastIndexOf('?')
    const query = queryIndex !== -1 ? path.substring(queryIndex) : ''
    path = path.substring(0, queryIndex === -1 ? path.length : queryIndex)

    const name = path.replace(/^.*[\\/]/, '')
    let base = name.substring(0, name.lastIndexOf('.'))

    const nameSplit = name.split('.')
    let ext = nameSplit.length > 1 ? (nameSplit.pop() || '').toLowerCase() : ''

    const protocolMatch = path.match(/^(.+):\/\/(.+)$/)
    if (protocolMatch) {
        protocol = protocolMatch[ 1 ].toLowerCase()
        path = protocolMatch[ 2 ] || ''
    }

    const dir = path.substring(0, path.lastIndexOf('/') + 1)

    if (compressedExtList.includes(ext)) {
        compressed = ext
        const n = path.length - ext.length - 1
        ext = (path.substr(0, n).split('.').pop() || '').toLowerCase()
        const m = base.length - ext.length - 1
        base = base.substr(0, m)
    } else {
        compressed = false
    }

    const binary = binaryExtList.includes(ext)

    return { path, name, ext, base, dir, compressed, binary, protocol, query, src: file }
}