• Home
• net.xdow
• aliyundrive-webdav-internal
• 2.0.0
• source code
• AliyunDriveFileSystemStore.java

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

com.github.zxbu.webdavteambition.store.AliyunDriveFileSystemStore Maven / Gradle / Ivy

package com.github.zxbu.webdavteambition.store;

import com.fujieid.jap.http.JapHttpRequest;
import com.fujieid.jap.http.JapHttpResponse;
import com.github.zxbu.webdavteambition.bean.AFileReqInfo;
import com.github.zxbu.webdavteambition.bean.PathInfo;
import com.github.zxbu.webdavteambition.config.AliyunDriveProperties;
import com.github.zxbu.webdavteambition.handler.GetRequestHandlerHolder;
import com.github.zxbu.webdavteambition.handler.IGetRequestHandler;
import com.google.common.net.HttpHeaders;
import com.google.common.net.UrlEscapers;
import com.google.common.util.concurrent.UncheckedExecutionException;
import lombok.Getter;
import net.sf.webdav.ITransaction;
import net.sf.webdav.IWebdavStore;
import net.sf.webdav.StoredObject;
import net.sf.webdav.Transaction;
import net.sf.webdav.exceptions.WebdavException;
import net.sf.webdav.util.ClientIdentifyUtils;
import net.sf.webdav.util.DateTimeUtils;
import net.xdow.aliyundrive.bean.AliyunDriveEnum;
import net.xdow.aliyundrive.bean.AliyunDriveFileInfo;
import net.xdow.aliyundrive.exception.NotAuthenticatedException;
import net.xdow.aliyundrive.util.JsonUtils;
import net.xdow.aliyundrive.webapi.impl.AliyunDriveWebApiImplV1;
import okhttp3.Response;
import okhttp3.ResponseBody;
import org.apache.commons.codec.digest.DigestUtils;
import org.apache.commons.lang3.StringUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.io.ByteArrayInputStream;
import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.security.Principal;
import java.util.*;

public class AliyunDriveFileSystemStore implements IWebdavStore {

    /**
     * OS X attempts to create ".DS_Store" files to store a folder's icon positions and background image. We choose not to store
     * these in our implementation, so we ignore requests to create them.
     */
    private static final String DS_STORE_SUFFIX = ".DS_Store";

    private static final Logger LOGGER = LoggerFactory.getLogger(AliyunDriveFileSystemStore.class);

    @Getter
    private AliyunDriveClientService aliyunDriveClientService;

    public AliyunDriveFileSystemStore(Object[] args, File root) {
        this.aliyunDriveClientService = (AliyunDriveClientService) args[0];
    }

    @Override
    public void destroy() {
        LOGGER.info("destroy");
    }

    @Override
    public ITransaction begin(Principal principal, JapHttpRequest req, JapHttpResponse resp) {
        LOGGER.debug("begin");
        return new Transaction(principal, req, resp);
    }

    @Override
    public void checkAuthentication(ITransaction transaction) {
        LOGGER.debug("checkAuthentication");
    }

    @Override
    public void commit(ITransaction transaction) {
        LOGGER.debug("commit");
    }

    @Override
    public void rollback(ITransaction transaction) {
        LOGGER.debug("rollback");
    }

    @Override
    public void createFolder(ITransaction transaction, String folderUri) {
        LOGGER.info("createFolder {}", folderUri);
        aliyunDriveClientService.createFolder(folderUri);
    }

    @Override
    public void createResource(ITransaction transaction, String resourceUri) {
        LOGGER.info("createResource {}", resourceUri);
    }

    @Override
    public InputStream getResourceContent(ITransaction transaction, String resourceUri) {
        LOGGER.info("getResourceContent: {}", resourceUri);
        Enumeration headerNames = transaction.getRequest().getHeaderNames();
        while (headerNames.hasMoreElements()) {
            String s = headerNames.nextElement();
            LOGGER.debug("{} request: {} = {}", resourceUri, s, transaction.getRequest().getHeader(s));
        }
        JapHttpResponse response = transaction.getResponse();
        long size = getResourceLength(transaction, resourceUri);
        Response downResponse = aliyunDriveClientService.download(resourceUri, transaction.getRequest(), size);
        LOGGER.debug("{} code = {}", resourceUri, downResponse.code());
        Set names = downResponse.headers().names();
        boolean isPDF = resourceUri.toLowerCase().endsWith(".pdf");
        for (String name : names) {
            if (isPDF) {
                if (HttpHeaders.CONTENT_TYPE.equalsIgnoreCase(name)) {
                    continue;
                }
            }
            //Fix Winscp Invalid Content-Length in response
            if (HttpHeaders.CONTENT_LENGTH.equalsIgnoreCase(name)) {
                continue;
            }
            LOGGER.debug("{} downResponse: {} = {}", resourceUri, name, downResponse.header(name));
            if (HttpHeaders.CONTENT_DISPOSITION.equalsIgnoreCase(name)) {
                if (isPDF) {
                    String value = downResponse.header(name);
                    response.addHeader(name, String.valueOf(value).replace("attachment;", "inline;"));
                    continue;
                }
            }
            response.addHeader(name, downResponse.header(name));
        }
        if (isPDF) {
            response.setContentType("application/pdf");
        }

        ResponseBody body = downResponse.body();
        if (body == null) {
            response.setContentLengthLong(0L);
            response.setStatus(downResponse.code());
            return new ByteArrayInputStream(new byte[0]);
        }
        response.setContentLengthLong(body.contentLength());
        response.setStatus(downResponse.code());
        return body.byteStream();
    }

    @Override
    public long setResourceContent(ITransaction transaction, String resourceUri, InputStream content, String contentType, String characterEncoding) {
        LOGGER.info("setResourceContent {}", resourceUri);
        // Mac OS X workaround from Drools Guvnor
        if (resourceUri.endsWith(DS_STORE_SUFFIX)) return 0;
        String resourceName = resourceNameFromResourcePath(resourceUri);
        if (resourceName.startsWith("._")) {
            return -1;
        }
        JapHttpRequest request = transaction.getRequest();
        JapHttpResponse response = transaction.getResponse();

        long contentLength = -1;
        try {
            contentLength = request.getContentLength();
        } catch (IOException e) {
        }
        if (contentLength < 0) {
            contentLength = Long.parseLong(StringUtils.defaultIfEmpty(request.getHeader("content-length"), "-1"));
            if (contentLength < 0) {
                contentLength = Long.parseLong(StringUtils.defaultIfEmpty(request.getHeader("X-Expected-Entity-Length"), "-1"));
            }
        }
        if (LOGGER.isTraceEnabled()) {
            Enumeration headerNames = request.getHeaderNames();
            while (headerNames.hasMoreElements()) {
                String key = headerNames.nextElement();
                LOGGER.trace("header key: {} value: {}", key, request.getHeader(key));
            }
        }
        String sha1Sum = StringUtils.defaultIfEmpty(request.getHeader("oc-checksum"), "").toLowerCase();
        sha1Sum = sha1Sum.startsWith("sha1:") ? sha1Sum.toLowerCase().replace("sha1:", "") : null;
        long modifyTimeSec = Long.parseLong(StringUtils.defaultIfEmpty(request.getHeader("x-oc-mtime"), "-1"));
        if (modifyTimeSec != -1) {
            response.setHeader("X-OC-MTime", "accepted");
        }

        aliyunDriveClientService.uploadPre(resourceUri, contentLength, content, sha1Sum, modifyTimeSec, response);

        if (contentLength == 0) {
            String expect = request.getHeader("Expect");

            // 支持大文件上传
            if ("100-continue".equalsIgnoreCase(expect)) {
                try {
                    response.sendError(100, "Continue");
                } catch (IOException e) {
                    e.printStackTrace();
                }
                return 0;
            }
        }
        return contentLength;
    }


    @Override
    public String[] getChildrenNames(ITransaction transaction, String folderUri) {
        if (LOGGER.isTraceEnabled()) {
            LOGGER.trace("getChildrenNames: {}", folderUri);
        }
        AliyunDriveFileInfo tFile = this.aliyunDriveClientService.getTFileByPath(folderUri);
        if (tFile == null) {
            return new String[0];
        }
        if (tFile.getType() == AliyunDriveEnum.Type.File) {
            return new String[0];
        }
        try {
            AFileReqInfo tFileInfo = AFileReqInfo.from(tFile);
            Set tFileList = this.aliyunDriveClientService.getTFileListCached(tFileInfo);
            List nameList = new ArrayList<>();
            for (AliyunDriveFileInfo file : tFileList) {
                tFile = file;
                nameList.add(tFile.getName());
            }
            return nameList.toArray(new String[nameList.size()]);
        } catch (UncheckedExecutionException e) {
            if (e.getCause() instanceof NotAuthenticatedException) {
                return new String[]{"授权已失效" + e.getMessage()};

            }
        }
        return new String[0];
    }

    @Override
    public long getResourceLength(ITransaction transaction, String path) {
        return getResourceLength2(transaction, path);
    }

    private long getResourceLength2(AliyunDriveFileInfo tFile) {
        if (tFile == null || tFile.getSize() == null) {
            return 384;
        }
        return tFile.getSize();
    }
    private long getResourceLength2(ITransaction transaction, String path) {
        long size = 0;
        try {
            AliyunDriveFileInfo tFile = aliyunDriveClientService.getTFileByPath(path);
            return size = getResourceLength2(tFile);
        } finally {
            if (LOGGER.isTraceEnabled()) {
                LOGGER.trace("getResourceLength: {} size: {}", path, size);
            }
        }
    }

    @Override
    public void removeObject(ITransaction transaction, String uri) {
        LOGGER.info("removeObject: {}", uri);
        String resourceName = resourceNameFromResourcePath(uri);
        if (resourceName.startsWith("._")) {
            return;
        }
        aliyunDriveClientService.removeByPath(uri);
    }

    @Override
    public boolean moveObject(ITransaction transaction, String destinationPath, String sourcePath) {
        LOGGER.info("moveObject, destinationPath={}, sourcePath={}", destinationPath, sourcePath);

        PathInfo destinationPathInfo = aliyunDriveClientService.getPathInfo(destinationPath);
        PathInfo sourcePathInfo = aliyunDriveClientService.getPathInfo(sourcePath);
        // 名字相同，说明是移动目录
        if (sourcePathInfo.getName().equals(destinationPathInfo.getName())) {
            aliyunDriveClientService.move(sourcePath, destinationPathInfo.getParentPath());
        } else {
            if (!destinationPathInfo.getParentPath().equals(sourcePathInfo.getParentPath())) {
                throw new WebdavException("不支持目录和名字同时修改");
            }
            // 名字不同，说明是修改名字。不考虑目录和名字同时修改的情况
            aliyunDriveClientService.rename(sourcePath, destinationPathInfo.getName());
        }
        return true;
    }

    @Override
    public StoredObject getStoredObject(ITransaction transaction, String uri) {
        StoredObject result = null;
        try {

            if ("/favicon.ico".equals(uri)) {
                return result = null;
            }
            String resourceName = resourceNameFromResourcePath(uri);
            if (resourceName.startsWith("._")) {
                // OS-X uses these hidden files ...
                return null;
            }
            AliyunDriveFileInfo tFile = aliyunDriveClientService.getTFileByPath(uri);
            if (tFile != null) {
                StoredObject so = new StoredObject();
                so.setFolder(tFile.getType() == AliyunDriveEnum.Type.Folder);
                so.setResourceLength(getResourceLength2(tFile));
                Date localCreatedTime = tFile.getLocalCreatedAt();
                so.setCreationDate(localCreatedTime != null ? localCreatedTime : tFile.getCreatedAt());
                Date localModifyTime = tFile.getLocalModifiedAt();
                so.setLastModified(localModifyTime != null ? localModifyTime : tFile.getUpdatedAt());
                so.setSha1sum(tFile.getContentHash());
                so.setMimeType(tFile.getMimeType());
                so.setThumbnailUrl(tFile.getThumbnail());
                return result = so;
            }
            return result = null;
        } finally {
            LOGGER.debug("getStoredObject: {} result: {}", uri, JsonUtils.toJson(result));
        }
    }

    @Override
    public long getQuotaAvailableBytes(ITransaction transaction) {
        return aliyunDriveClientService.getQuotaAvailableBytes();
    }

    @Override
    public long getQuotaUsedBytes(ITransaction transaction) {
        return aliyunDriveClientService.getQuotaUsedBytes();
    }

    @Override
    public String getFooter(ITransaction transaction) {
        if (this.aliyunDriveClientService.getAliyunDrive() instanceof AliyunDriveWebApiImplV1) {
            return "
\n" +
                    "  更换RefreshToken: \n" +
                    "  \n" +
                    "  \n" +
                    "
 ";
        }
        return "";
    }

    @Override
    public String getResourceDownloadUrlForRedirection(ITransaction transaction, String path) {
        JapHttpRequest request = transaction.getRequest();
        String userAgent = request.getHeader("User-Agent");
        String referer = request.getHeader("Referer");
        AliyunDriveProperties.DownloadProxyMode mode = this.aliyunDriveClientService.getProperties().getDownloadProxyMode();
        switch (mode) {
            case Proxy:
                return null;
            case Direct:
                if (ClientIdentifyUtils.isWinSCP5AndBelow(userAgent)
                        || ClientIdentifyUtils.isSynoCloudSync(userAgent)
                        || ClientIdentifyUtils.checkAliyunDriveRefererForProxyMode(referer)) {
                    throw new WebdavException("DirectModeUnsupportedCode",
                            "This client is not support Direct mode, please consider switch to Proxy mode, or Auto mode.");
                }
                break;
            default:
                if (ClientIdentifyUtils.isWinSCP5AndBelow(userAgent)
                        || ClientIdentifyUtils.isSynoCloudSync(userAgent)
                        || ClientIdentifyUtils.isKodi(userAgent)
                        || ClientIdentifyUtils.isAppleCoreMedia(userAgent)
                        || ClientIdentifyUtils.isVLC(userAgent)
                        || ClientIdentifyUtils.checkAliyunDriveRefererForProxyMode(referer)) {
                    LOGGER.warn("Using Proxy mode User-Agent: {} Referer: {}", userAgent, referer);
                    return null;
                }
        }

        String url = this.aliyunDriveClientService.getDownloadUrlByPath(path);
        if (!request.isSecure()) {
            url = url.replaceAll("^https://", "http://");
        }
        return url;
    }

    @Override
    public String getPublicLink(ITransaction transaction, String path) {
        path = UrlEscapers.urlFragmentEscaper().escape(path);
        AliyunDriveProperties properties = this.aliyunDriveClientService.getProperties();
        String shareToken = properties.getShareToken();
        long shareExpireSec = properties.getShareExpireSec();
        long expire = (shareExpireSec > 0) ? DateTimeUtils.getCurrentDateGMT().getTime() / 1000 + shareExpireSec: 0;
        String hash = DigestUtils.sha1Hex(path + "_" + shareToken + "_" + expire);
        return String.format("%s?p=%s%s", path, hash, expire > 0 ? "%26expire=" + expire: "");
    }

    @Override
    public boolean handleCustomGetRequest(ITransaction transaction, String path) {
        JapHttpRequest request = transaction.getRequest();
        String action = request.getParameter("action");
        if (StringUtils.isEmpty(action)){
            return false;
        }
        IGetRequestHandler handler = GetRequestHandlerHolder.INSTANCE.get(action);
        if (handler == null) {
            return false;
        }
        try {
            if (handler.handle(this, transaction, path)) {
                return true;
            }
        } catch (IOException e) {
            throw new WebdavException(e);
        }
        return false;
    }

    protected String resourceNameFromResourcePath( String path ) {
        int ind = path.lastIndexOf('/');
        return path.substring(ind + 1);
    }
}