ZhongMianAdmin/__test__/bim/js/io/WebdavService.js

/*
 * WebdavService.js
 *
 * @author realor
 */

import { FileService, Metadata, Result } from './FileService.js'
import { ServiceManager } from './ServiceManager.js'
import { WebUtils } from '../utils/WebUtils.js'

class WebdavService extends FileService {
    constructor(name, description, url, username, password) {
        super(name, description, url, username, password)
    }

    open(path, readyCallback, progressCallback) {
        const OK = Result.OK
        const ERROR = Result.ERROR
        const COLLECTION = Metadata.COLLECTION
        const FILE = Metadata.FILE

        let url = this.getUrl(path)

        let baseUri = url
        let index = baseUri.indexOf('://')
        if (index !== -1) {
            baseUri = baseUri.substring(index + 3)
            index = baseUri.indexOf('/')
            if (index !== -1) {
                baseUri = baseUri.substring(index)
                if (baseUri.lastIndexOf('/') !== baseUri.length - 1) {
                    baseUri += '/'
                }
            }
        }

        // **** HTTP PROPFIND REQUEST ****

        let request = new XMLHttpRequest()
        request.onerror = () => {
            // ERROR
            readyCallback(new Result(ERROR, 'Connection error'))
        }
        request.onload = () => {
            if (request.status === 200 || request.status === 207) {
                try {
                    // OK
                    let xml = request.responseXML
                    let multiNode = xml.childNodes[0]
                    let responseNodes = multiNode.childNodes
                    let metadata = new Metadata()
                    let entries = []
                    for (let i = 0; i < responseNodes.length; i++) {
                        let responseNode = responseNodes[i]
                        if (responseNode.localName === 'response') {
                            let hrefNode = responseNode.querySelector('href')
                            let hrefValue = hrefNode.textContent
                            let fileName = hrefValue.substring(baseUri.length)
                            let isCollectionNode = responseNode.querySelector('propstat prop resourcetype collection') !== null
                            let contentLengthNode = responseNode.querySelector('propstat prop getcontentlength')
                            let lastModifiedNode = responseNode.querySelector('propstat prop getlastmodified')

                            let fileSize = contentLengthNode ? parseInt(contentLengthNode.textContent) : 0
                            let lastModified = lastModifiedNode ? parseInt(lastModifiedNode.textContent) : 0

                            if (fileName.indexOf('/') === 0) fileName = fileName.substring(1)

                            if (fileName.length === 0) {
                                // requested resource
                                let index = hrefValue.lastIndexOf('/')
                                metadata.name = hrefValue.substring(index + 1)
                                metadata.description = metadata.name
                                metadata.type = isCollectionNode ? COLLECTION : FILE
                                metadata.size = fileSize
                                metadata.lastModified = lastModified
                            } else {
                                let entry = new Metadata(fileName, fileName, isCollectionNode ? COLLECTION : FILE, fileSize, lastModified)
                                entries.push(entry)
                            }
                        }
                    }
                    if (metadata.type === COLLECTION) {
                        readyCallback(new Result(OK, '', path, metadata, entries, null))
                    } // download file
                    else {
                        let request = new XMLHttpRequest()
                        request.open('GET', url, true)
                        request.onload = () => {
                            if (request.status === 200) {
                                if (progressCallback) {
                                    progressCallback({ progress: 100, message: 'Download completed.' })
                                }
                                setTimeout(() => readyCallback(new Result(OK, '', path, metadata, null, request.response)), 100)
                            }
                        }
                        request.onerror = error => {
                            readyCallback(new Result(ERROR, error))
                        }
                        if (progressCallback) {
                            request.onprogress = event => {
                                let progress = Math.round((100 * event.loaded) / event.total)
                                let message = 'Downloading file...'
                                progressCallback({ progress: progress, message: message })
                            }
                        }
                        WebUtils.setBasicAuthorization(request, this.username, this.password)
                        request.send()
                    }
                } catch (ex) {
                    readyCallback(new Result(ERROR, ex))
                }
            } else {
                readyCallback(this.createError("Can't open", request.status))
            }
        }
        request.open('PROPFIND', url, true)
        request.setRequestHeader('depth', '1')
        WebUtils.setBasicAuthorization(request, this.username, this.password)
        request.send()
    }

    save(path, data, readyCallback, progressCallback) {
        const OK = Result.OK
        const ERROR = Result.ERROR

        const url = this.getUrl(path)
        const request = new XMLHttpRequest()
        request.onerror = error => {
            // ERROR
            readyCallback(new Result(ERROR, 'Connection error'))
        }
        request.onload = () => {
            if (request.status === 200) {
                readyCallback(new Result(OK))
            } else {
                readyCallback(this.createError('Save failed', request.status))
            }
        }
        if (progressCallback) {
            request.onprogress = event => {
                let progress = Math.round((100 * event.loaded) / event.total)
                let message = 'Uploading file...'
                progressCallback({ progress: progress, message: message })
            }
        }
        request.open('PUT', url, true)
        WebUtils.setBasicAuthorization(request, this.username, this.password)
        request.send(data)
    }

    remove(path, readyCallback, progressCallback) {
        const OK = Result.OK
        const ERROR = Result.ERROR

        const url = this.getUrl(path)
        const request = new XMLHttpRequest()
        request.onerror = error => {
            // ERROR
            readyCallback(new Result(ERROR, 'Connection error'))
        }
        request.onload = () => {
            if (request.status === 200) {
                readyCallback(new Result(OK))
            } else {
                readyCallback(this.createError('Delete failed', request.status))
            }
        }
        request.open('DELETE', url, true)
        WebUtils.setBasicAuthorization(request, this.username, this.password)
        request.send()
    }

    makeCollection(path, readyCallback, progressCallback) {
        const OK = Result.OK
        const ERROR = Result.ERROR

        const url = this.getUrl(path)
        const request = new XMLHttpRequest()
        request.onerror = error => {
            readyCallback(new Result(ERROR, 'Connection error'))
        }
        request.onload = () => {
            if (request.status === 200 || request.status === 201) {
                readyCallback(new Result(OK))
            } else {
                readyCallback(this.createError('Folder creation failed', request.status))
            }
        }
        request.open('MKCOL', url, true)
        WebUtils.setBasicAuthorization(request, this.username, this.password)
        request.send()
    }

    createError(message, status) {
        let statusMessage = WebUtils.getHttpStatusMessage(status)
        if (statusMessage.length > 0) {
            message += ': ' + statusMessage
        }
        message += ' (HTTP ' + status + ').'

        let resultStatus
        switch (status) {
            case 401:
                resultStatus = Result.INVALID_CREDENTIALS
                break
            case 403:
                resultStatus = Result.FORBIDDEN
                break
            default:
                resultStatus = Result.ERROR
        }
        return new Result(resultStatus, message)
    }

    getUrl(path) {
        let url = this.url || ''

        if (url && url.endsWith('/')) {
            url = url.substring(0, url.length - 1)
        }

        if (path && !path.startsWith('/')) {
            path = '/' + path
        }
        return url + path
    }
}

ServiceManager.addClass(WebdavService)

export { WebdavService }