Compare commits
58 Commits
Author | SHA1 | Date |
---|---|---|
zxbu | d5b8f01e32 | |
zxbu | 6ade2dbfe8 | |
liuyonghui | f0bdb77674 | |
zxbu | c9e87bfcd7 | |
Delphi Lin | 2407802e5b | |
zxbu | 121ec196c9 | |
zxbu | ef3cb8540d | |
zhouxin | fa8b890e26 | |
zxbu | 08124d17e7 | |
kid101x | fef74ee99c | |
zxbu | ce08dc0566 | |
zuodajiang | b8f183be5e | |
zxbu | 04a97ea189 | |
zhouxin | da567fd6c2 | |
zhouxin | 044ec20f59 | |
zhouxin | 1718782b60 | |
zxbu | 62260137cb | |
zhouxin | 394eb7b6db | |
zxbu | eca5f7da36 | |
zxbu | f480c894f8 | |
zxbu | ccb7552422 | |
zhouxin | fec2079f5c | |
zxbu | 83283a2133 | |
zhouxin | 55dfacd3ee | |
zhouxin | 453e91f87c | |
zhouxin | ae620231fe | |
zhouxin | 410b742959 | |
zhouxin | cb9dc7e777 | |
zxbu | f52c1816eb | |
zxbu | 0e9cce633d | |
zxbu | f707d1e32e | |
zxbu | 8bffa83ce9 | |
zxbu | abbcc08168 | |
zxbu | b87280265c | |
zxbu | 45cdc14920 | |
zxbu | 0ed2dd288d | |
zxbu | 4f6bc46c4d | |
zxbu | b6c044a3fa | |
zxbu | 18e12a51df | |
zxbu | 0597402246 | |
expoli | c2d5037a13 | |
zxbu | 770136e659 | |
Jason | 9fa7be47d0 | |
zxbu | 0596f90050 | |
zhouxin | b567880ff0 | |
zxbu | ce5e68407d | |
zhouxin | baa24401fa | |
zhouxin | fa85bca764 | |
zhouxin | 0016c3e759 | |
zhouxin | 553d456dc3 | |
zhouxin | 31c0876e58 | |
zhouxin | c21e604690 | |
zhouxin | ca4da58520 | |
zxbu | 9f4a81bb9f | |
zxbu | 0a3403ca85 | |
zhouxin | 50f71bd7c8 | |
zxbu | 9c49d27ca7 | |
zxbu | a94ab69db5 |
|
@ -0,0 +1,75 @@
|
|||
# This is a basic workflow to help you get started with Actions
|
||||
|
||||
name: aliyundriver-Build
|
||||
|
||||
# Controls when the action will run.
|
||||
on:
|
||||
# Triggers the workflow on push or pull request events but only for the main branch
|
||||
push:
|
||||
branches: [ main ]
|
||||
pull_request:
|
||||
branches: [ main ]
|
||||
# schedule:
|
||||
# - cron: "0 0 * * */3"
|
||||
|
||||
# Allows you to run this workflow manually from the Actions tab
|
||||
workflow_dispatch:
|
||||
inputs:
|
||||
logLevel:
|
||||
description: 'Log level'
|
||||
required: true
|
||||
default: 'warning'
|
||||
tags:
|
||||
description: 'Test scenario tags'
|
||||
|
||||
# A workflow run is made up of one or more jobs that can run sequentially or in parallel
|
||||
jobs:
|
||||
buildx:
|
||||
runs-on: ubuntu-latest
|
||||
steps:
|
||||
-
|
||||
name: Checkout
|
||||
uses: actions/checkout@v2
|
||||
-
|
||||
name: Get current date
|
||||
id: date
|
||||
run: echo "::set-output name=today::$(date +'%Y-%m-%d')"
|
||||
-
|
||||
name: Set up QEMU
|
||||
uses: docker/setup-qemu-action@v1
|
||||
with:
|
||||
platforms: all
|
||||
-
|
||||
name: Set up Docker Buildx
|
||||
id: buildx
|
||||
uses: docker/setup-buildx-action@v1
|
||||
-
|
||||
name: Available platforms
|
||||
run: echo ${{ steps.buildx.outputs.platforms }}
|
||||
-
|
||||
name: Cache Docker layers
|
||||
uses: actions/cache@v2
|
||||
with:
|
||||
path: /tmp/.buildx-cache
|
||||
key: ${{ runner.os }}-buildx-${{ github.sha }}
|
||||
restore-keys: |
|
||||
${{ runner.os }}-buildx-
|
||||
-
|
||||
name: Login to DockerHub
|
||||
uses: docker/login-action@v1
|
||||
with:
|
||||
username: ${{ secrets.DOCKERHUB_USERNAME }}
|
||||
password: ${{ secrets.DOCKERHUB_TOKEN }}
|
||||
-
|
||||
name: Build and push
|
||||
uses: docker/build-push-action@v2
|
||||
with:
|
||||
context: .
|
||||
file: ./Dockerfile
|
||||
platforms: linux/amd64,linux/arm64
|
||||
push: ${{ github.event_name != 'pull_request' }}
|
||||
cache-from: type=local,src=/tmp/.buildx-cache
|
||||
cache-to: type=local,dest=/tmp/.buildx-cache
|
||||
tags: |
|
||||
zx5253/webdav-aliyundriver:latest
|
||||
zx5253/webdav-aliyundriver:${{ steps.date.outputs.today }}
|
|
@ -7,4 +7,5 @@ RUN cd /tmp/code && mvn clean package -Dmaven.test.skip=true -Dmaven.javadoc.ski
|
|||
FROM openjdk:11-jdk-oracle
|
||||
COPY --from=maven /tmp/code/target/*.jar /webdav.jar
|
||||
EXPOSE 8080
|
||||
ENTRYPOINT ["java","-Djava.security.egd=file:/dev/./urandom","-jar","/webdav.jar"]
|
||||
ENV JAVA_OPTS="-Xmx1g"
|
||||
ENTRYPOINT ["sh", "-c", "java $JAVA_OPTS -Djava.security.egd=file:/dev/./urandom -jar /webdav.jar"]
|
||||
|
|
122
README.md
122
README.md
|
@ -1,7 +1,32 @@
|
|||
|
||||
说明:[1.1.0版本](https://github.com/zxbu/webdav-aliyundriver/releases/tag/v1.1.0)支持阿里Teambition网盘的webdav协议,后续的2.x版本仅支持阿里云盘,不再维护Teambition网盘版本
|
||||
|
||||
- [webdav-aliyundriver](#webdav-aliyundriver)
|
||||
- [如何使用](#如何使用)
|
||||
- [Jar包运行](#jar包运行)
|
||||
- [容器运行](#容器运行)
|
||||
- [Docker-Compose](#docker-compose)
|
||||
- [参数说明](#参数说明)
|
||||
- [QQ群](#qq群)
|
||||
- [新手教程](#新手教程)
|
||||
- [群晖](#群晖)
|
||||
- [Windows10](#windows10)
|
||||
- [Linux](#linux)
|
||||
- [Mac](#mac)
|
||||
- [客户端兼容性](#客户端兼容性)
|
||||
- [浏览器获取refreshToken方式](#浏览器获取refreshtoken方式)
|
||||
- [功能说明](#功能说明)
|
||||
- [支持的功能](#支持的功能)
|
||||
- [暂不支持的功能](#暂不支持的功能)
|
||||
- [已知问题](#已知问题)
|
||||
- [TODO](#todo)
|
||||
- [免责声明](#免责声明)
|
||||
|
||||
# webdav-aliyundriver
|
||||
本项目实现了阿里云盘的webdav协议,只需要简单的配置一下,就可以让阿里云盘变身为webdav协议的文件服务器。
|
||||
基于此,你可以把阿里云盘挂载为Windows、Linux、Mac系统的磁盘,可以通过NAS系统做文件管理或文件同步,更多玩法等你挖掘
|
||||
|
||||
|
||||
# 如何使用
|
||||
支持refreshToken登录方式,具体看参数说明
|
||||
## Jar包运行
|
||||
|
@ -12,11 +37,39 @@ java -jar webdav.jar --aliyundrive.refresh-token="your refreshToken"
|
|||
```
|
||||
## 容器运行
|
||||
```bash
|
||||
docker run -d --name=webdav-aliyundriver --restart=always -p 8080:8080 -v /etc/localtime:/etc/localtime -v /etc/aliyun-driver/:/etc/aliyun-driver/ -e TZ="Asia/Shanghai" -e ALIYUNDRIVE_REFRESH_TOKEN="your refreshToken" zx5253/webdav-aliyundriver
|
||||
docker run -d --name=webdav-aliyundriver --restart=always -p 8080:8080 -v /etc/localtime:/etc/localtime -v /etc/aliyun-driver/:/etc/aliyun-driver/ -e TZ="Asia/Shanghai" -e ALIYUNDRIVE_REFRESH_TOKEN="your refreshToken" -e ALIYUNDRIVE_AUTH_PASSWORD="admin" -e JAVA_OPTS="-Xmx1g" zx5253/webdav-aliyundriver
|
||||
|
||||
# /etc/aliyun-driver/ 挂载卷自动维护了最新的refreshToken,建议挂载
|
||||
# ALIYUNDRIVE_AUTH_PASSWORD 是admin账户的密码,建议修改
|
||||
# JAVA_OPTS 可修改最大内存占用,比如 -e JAVA_OPTS="-Xmx512m" 表示最大内存限制为512m
|
||||
```
|
||||
|
||||
## Docker-Compose
|
||||
```yml
|
||||
version: "3.0"
|
||||
services:
|
||||
webdav-aliyundriver:
|
||||
image: zx5253/webdav-aliyundriver
|
||||
container_name: aliyundriver
|
||||
environment:
|
||||
- TZ=Asia/Shanghai
|
||||
- ALIYUNDRIVE_REFRESH_TOKEN=refreshToken
|
||||
- ALIYUNDRIVE_AUTH_USER_NAME=admin
|
||||
- ALIYUNDRIVE_AUTH_PASSWORD=admin
|
||||
- JAVA_OPTS=-Xmx1g
|
||||
volumes:
|
||||
- /etc/aliyun-driver/:/etc/aliyun-driver/
|
||||
ports:
|
||||
- 6666:8080
|
||||
restart: always
|
||||
|
||||
# “refreshToken”请根据下文说明自行获取。
|
||||
# “ALIYUNDRIVE_AUTH_USER-NAME”和“ALIYUNDRIVE_AUTH_PASSWORD”为连接用户名和密码,建议更改。
|
||||
# “/etc/aliyun-driver/:/etc/aliyun-driver/”,可以把冒号前改为指定目录,比如“/homes/USER/docker/alidriver/:/etc/aliyun-driver/”。
|
||||
# 删除了“/etc/localtime:/etc/localtime”,如有需要同步时间请自行添加在environment下。
|
||||
# 端口6666可自行按需更改,此端口为WebDAV连接端口,8080为容器内配置端口,修改请量力而为。
|
||||
# 建议不要保留这些中文注释,以防报错,比如QNAP。
|
||||
```
|
||||
|
||||
# 参数说明
|
||||
```bash
|
||||
|
@ -24,9 +77,49 @@ docker run -d --name=webdav-aliyundriver --restart=always -p 8080:8080 -v /etc/
|
|||
阿里云盘的refreshToken,获取方式见下文
|
||||
--server.port
|
||||
非必填,服务器端口号,默认为8080
|
||||
--aliyundrive.auth.enable=true
|
||||
是否开启WebDav账户验证,默认开启
|
||||
--aliyundrive.auth.user-name=admin
|
||||
WebDav账户,默认admin
|
||||
--aliyundrive.auth.password=admin
|
||||
WebDav密码,默认admin
|
||||
--aliyundrive.work-dir=/etc/aliyun-driver/
|
||||
token挂载路径(如果多开的话,需修改此配置)
|
||||
|
||||
```
|
||||
# QQ群
|
||||
> 群号:789738128
|
||||
> 群号(已满):789738128
|
||||
|
||||
> 二群群号(已满):979024890
|
||||
|
||||
> 三群群号(已满):212673498
|
||||
|
||||
> 四群群号(已满):752067171
|
||||
|
||||
> 五群群号:555954095
|
||||
|
||||
# 新手教程
|
||||
## 群晖
|
||||
TODO
|
||||
|
||||
## Windows10
|
||||
TODO
|
||||
|
||||
## Linux
|
||||
TODO
|
||||
|
||||
## Mac
|
||||
TODO
|
||||
|
||||
# 客户端兼容性
|
||||
| 客户端 | 下载 | 上传 | 备注 |
|
||||
| :-----| ----: | :----: | :----: |
|
||||
| 群辉Cloud Sync | 可用 | 可用 | 使用单向同步非常稳定 |
|
||||
| Rclone | 可用 | 可用 | 推荐,支持各个系统 |
|
||||
| Mac原生 | 可用 | 可用 | |
|
||||
| Windows原生 | 可用 | 有点小问题 | 不建议,适配有点问题,上传报错 |
|
||||
| RaiDrive | 可用 | 可用 | Windows平台下建议用这个 |
|
||||
|
||||
|
||||
# 浏览器获取refreshToken方式
|
||||
1. 先通过浏览器(建议chrome)打开阿里云盘官网并登录:https://www.aliyundrive.com/drive/
|
||||
|
@ -42,16 +135,27 @@ docker run -d --name=webdav-aliyundriver --restart=always -p 8080:8080 -v /etc/
|
|||
4. 文件下载
|
||||
5. 文件删除
|
||||
6. 文件上传(支持大文件自动分批上传)
|
||||
7. 支持超大文件上传(官方限制30G)
|
||||
8. 支持WebDav权限校验(默认账户密码:admin/admin)
|
||||
9. 文件下载断点续传
|
||||
10. Webdav下的流媒体播放等功能
|
||||
## 暂不支持的功能
|
||||
1. 权限校验
|
||||
2. 移动文件到其他目录的同时,修改文件名。比如 /a.zip 移动到 /b/a1.zip,是不支持的
|
||||
3. 文件上传断点续传
|
||||
4. 文件下载断点续传
|
||||
5. 同级目录下文件数量不能超过10000个(建议不超过100,否则性能比较差)
|
||||
1. 移动文件到其他目录的同时,修改文件名。比如 /a.zip 移动到 /b/a1.zip,是不支持的
|
||||
2. 文件上传断点续传
|
||||
3. 部分客户端兼容性不好
|
||||
## 已知问题
|
||||
1. 没有做文件sha1校验,不保证上传文件的100%准确性(一般场景下,是没问题的)
|
||||
2. 通过文件名和文件大小判断是否重复。也就是说如果一个文件即使发生了更新,但其大小没有任何改变,是不会自动上传的
|
||||
3. 超大文件上传存在问题,具体上限还不清楚,我自己实测5G+大小的文件无法顺利上传
|
||||
3. 不支持文件名包含 `/` 字符
|
||||
|
||||
## TODO
|
||||
1. 支持更多登录方式(验证码、账号密码等)
|
||||
2. 支持权限校验
|
||||
|
||||
|
||||
# 免责声明
|
||||
1. 本软件为免费开源项目,无任何形式的盈利行为。
|
||||
2. 本软件服务于阿里云盘,旨在让阿里云盘功能更强大。如有侵权,请与我联系,会及时处理。
|
||||
3. 本软件皆调用官方接口实现,无任何“Hack”行为,无破坏官方接口行为。
|
||||
5. 本软件仅做流量转发,不拦截、存储、篡改任何用户数据。
|
||||
6. 严禁使用本软件进行盈利、损坏官方、散落任何违法信息等行为。
|
||||
7. 本软件不作任何稳定性的承诺,如因使用本软件导致的文件丢失、文件破坏等意外情况,均与本软件无关。
|
||||
|
|
|
@ -0,0 +1,17 @@
|
|||
version: '3'
|
||||
services:
|
||||
# https://github.com/zxbu/webdav-aliyundriver
|
||||
webdav-aliyundriver:
|
||||
image: zx5253/webdav-aliyundriver
|
||||
container_name: webdav-aliyundriver
|
||||
restart: always
|
||||
volumes:
|
||||
- /etc/localtime:/etc/localtime
|
||||
- ./docker/etc/aliyun-driver/:/etc/aliyun-driver/
|
||||
ports:
|
||||
- "8080:8080"
|
||||
tty: true
|
||||
environment:
|
||||
- TZ=Asia/Shanghai
|
||||
- ALIYUNDRIVE_REFRESH_TOKEN=<change me>
|
||||
- ALIYUNDRIVE_AUTH_PASSWORD=<change me>
|
8
pom.xml
8
pom.xml
|
@ -10,7 +10,7 @@
|
|||
</parent>
|
||||
<groupId>com.github.zxbu</groupId>
|
||||
<artifactId>webdav-aliyundriver</artifactId>
|
||||
<version>2.2.0</version>
|
||||
<version>2.4.0</version>
|
||||
<name>webdav</name>
|
||||
<description>Demo project for Spring Boot</description>
|
||||
|
||||
|
@ -30,6 +30,12 @@
|
|||
<version>3.14.9</version>
|
||||
</dependency>
|
||||
|
||||
<dependency>
|
||||
<groupId>com.github.ben-manes.caffeine</groupId>
|
||||
<artifactId>caffeine</artifactId>
|
||||
<version>2.7.0</version>
|
||||
</dependency>
|
||||
|
||||
|
||||
<dependency>
|
||||
<groupId>org.springframework.boot</groupId>
|
||||
|
|
|
@ -8,11 +8,13 @@ import org.springframework.boot.autoconfigure.SpringBootApplication;
|
|||
import org.springframework.boot.web.servlet.FilterRegistrationBean;
|
||||
import org.springframework.boot.web.servlet.ServletRegistrationBean;
|
||||
import org.springframework.context.annotation.Bean;
|
||||
import org.springframework.scheduling.annotation.EnableScheduling;
|
||||
|
||||
import java.util.LinkedHashMap;
|
||||
import java.util.Map;
|
||||
|
||||
@SpringBootApplication
|
||||
@EnableScheduling
|
||||
public class WebdavTeambitionApplication {
|
||||
|
||||
public static void main(String[] args) {
|
||||
|
|
|
@ -7,7 +7,9 @@ import okhttp3.*;
|
|||
import org.slf4j.Logger;
|
||||
import org.slf4j.LoggerFactory;
|
||||
import org.springframework.util.Assert;
|
||||
import org.springframework.util.StringUtils;
|
||||
|
||||
import javax.servlet.http.HttpServletRequest;
|
||||
import java.io.IOException;
|
||||
import java.io.InputStream;
|
||||
import java.nio.charset.StandardCharsets;
|
||||
|
@ -42,7 +44,14 @@ public class AliYunDriverClient {
|
|||
@Override
|
||||
public Request authenticate(Route route, Response response) throws IOException {
|
||||
if (response.code() == 401 && response.body() != null && response.body().string().contains("AccessToken")) {
|
||||
String refreshTokenResult = post("https://websv.aliyundrive.com/token/refresh", Collections.singletonMap("refresh_token", readRefreshToken()));
|
||||
String refreshTokenResult;
|
||||
try {
|
||||
refreshTokenResult = post("https://websv.aliyundrive.com/token/refresh", Collections.singletonMap("refresh_token", readRefreshToken()));
|
||||
} catch (Exception e) {
|
||||
// 如果置换token失败,先清空原token文件,再尝试一次
|
||||
deleteRefreshTokenFile();
|
||||
refreshTokenResult = post("https://websv.aliyundrive.com/token/refresh", Collections.singletonMap("refresh_token", readRefreshToken()));
|
||||
}
|
||||
String accessToken = (String) JsonUtil.getJsonNodeValue(refreshTokenResult, "access_token");
|
||||
String refreshToken = (String) JsonUtil.getJsonNodeValue(refreshTokenResult, "refresh_token");
|
||||
Assert.hasLength(accessToken, "获取accessToken失败");
|
||||
|
@ -85,12 +94,32 @@ public class AliYunDriverClient {
|
|||
}
|
||||
|
||||
|
||||
public InputStream download(String url) {
|
||||
Request request = new Request.Builder().url(url).build();
|
||||
public Response download(String url, HttpServletRequest httpServletRequest, long size ) {
|
||||
Request.Builder builder = new Request.Builder().header("referer", "https://www.aliyundrive.com/");
|
||||
String range = httpServletRequest.getHeader("range");
|
||||
if (range != null) {
|
||||
// 如果range最后 >= size, 则去掉
|
||||
String[] split = range.split("-");
|
||||
if (split.length == 2) {
|
||||
String end = split[1];
|
||||
if (Long.parseLong(end) >= size) {
|
||||
range = range.substring(0, range.lastIndexOf('-') + 1);
|
||||
}
|
||||
}
|
||||
builder.header("range", range);
|
||||
}
|
||||
|
||||
String ifRange = httpServletRequest.getHeader("if-range");
|
||||
if (ifRange != null) {
|
||||
builder.header("if-range", ifRange);
|
||||
}
|
||||
|
||||
|
||||
Request request = builder.url(url).build();
|
||||
Response response = null;
|
||||
try {
|
||||
response = okHttpClient.newCall(request).execute();
|
||||
return response.body().byteStream();
|
||||
return response;
|
||||
} catch (IOException e) {
|
||||
throw new WebdavException(e);
|
||||
}
|
||||
|
@ -178,6 +207,16 @@ public class AliYunDriverClient {
|
|||
return aliYunDriveProperties.getUrl() + url;
|
||||
}
|
||||
|
||||
private void deleteRefreshTokenFile() {
|
||||
String refreshTokenPath = aliYunDriveProperties.getWorkDir() + "refresh-token";
|
||||
Path path = Paths.get(refreshTokenPath);
|
||||
try {
|
||||
Files.delete(path);
|
||||
} catch (IOException e) {
|
||||
e.printStackTrace();
|
||||
}
|
||||
}
|
||||
|
||||
private String readRefreshToken() {
|
||||
String refreshTokenPath = aliYunDriveProperties.getWorkDir() + "refresh-token";
|
||||
Path path = Paths.get(refreshTokenPath);
|
||||
|
|
|
@ -0,0 +1,28 @@
|
|||
package com.github.zxbu.webdavteambition.config;
|
||||
|
||||
import com.github.zxbu.webdavteambition.model.result.TFile;
|
||||
import com.github.zxbu.webdavteambition.store.AliYunDriverClientService;
|
||||
import org.springframework.beans.factory.annotation.Autowired;
|
||||
import org.springframework.scheduling.annotation.Scheduled;
|
||||
import org.springframework.stereotype.Component;
|
||||
|
||||
@Component
|
||||
public class AliYunDriverCronTask {
|
||||
|
||||
@Autowired
|
||||
private AliYunDriverClientService aliYunDriverClientService;
|
||||
|
||||
/**
|
||||
* 每隔5分钟请求一下接口,保证token不过期
|
||||
*/
|
||||
@Scheduled(initialDelay = 30 * 1000, fixedDelay = 5 * 60 * 1000)
|
||||
public void refreshToken() {
|
||||
try {
|
||||
TFile root = aliYunDriverClientService.getTFileByPath("/");
|
||||
aliYunDriverClientService.getTFiles(root.getFile_id());
|
||||
} catch (Exception e) {
|
||||
// nothing
|
||||
}
|
||||
|
||||
}
|
||||
}
|
|
@ -1,6 +1,8 @@
|
|||
package com.github.zxbu.webdavteambition.config;
|
||||
|
||||
import org.apache.catalina.authenticator.DigestAuthenticator;
|
||||
import org.apache.catalina.CredentialHandler;
|
||||
import org.apache.catalina.authenticator.AuthenticatorBase;
|
||||
import org.apache.catalina.authenticator.BasicAuthenticator;
|
||||
import org.apache.catalina.realm.GenericPrincipal;
|
||||
import org.apache.catalina.realm.MessageDigestCredentialHandler;
|
||||
import org.apache.catalina.realm.RealmBase;
|
||||
|
@ -19,7 +21,7 @@ import java.util.Collections;
|
|||
|
||||
@Component
|
||||
@ConditionalOnProperty(prefix = "aliyundrive.auth", name = "enable", matchIfMissing = true)
|
||||
public class EmbeddedTomcatConfig implements WebServerFactoryCustomizer<ConfigurableServletWebServerFactory>, Ordered {
|
||||
public class AuthTomcatConfig implements WebServerFactoryCustomizer<ConfigurableServletWebServerFactory>, Ordered {
|
||||
|
||||
@Autowired
|
||||
private AliYunDriveProperties aliYunDriveProperties;
|
||||
|
@ -37,7 +39,7 @@ public class EmbeddedTomcatConfig implements WebServerFactoryCustomizer<Configur
|
|||
if (aliYunDriveProperties.getAuth().getUserName().equals(username)) {
|
||||
return aliYunDriveProperties.getAuth().getPassword();
|
||||
}
|
||||
return "";
|
||||
return null;
|
||||
}
|
||||
|
||||
@Override
|
||||
|
@ -46,11 +48,11 @@ public class EmbeddedTomcatConfig implements WebServerFactoryCustomizer<Configur
|
|||
}
|
||||
};
|
||||
|
||||
MessageDigestCredentialHandler credentialHandler = new MessageDigestCredentialHandler();
|
||||
CredentialHandler credentialHandler = new MessageDigestCredentialHandler();
|
||||
realm.setCredentialHandler(credentialHandler);
|
||||
context.setRealm(realm);
|
||||
|
||||
DigestAuthenticator digestAuthenticator = new DigestAuthenticator();
|
||||
AuthenticatorBase digestAuthenticator = new BasicAuthenticator();
|
||||
SecurityConstraint securityConstraint = new SecurityConstraint();
|
||||
securityConstraint.setAuthConstraint(true);
|
||||
securityConstraint.addAuthRole("**");
|
||||
|
@ -67,4 +69,4 @@ public class EmbeddedTomcatConfig implements WebServerFactoryCustomizer<Configur
|
|||
public int getOrder() {
|
||||
return Ordered.LOWEST_PRECEDENCE;
|
||||
}
|
||||
}
|
||||
}
|
|
@ -0,0 +1,31 @@
|
|||
package com.github.zxbu.webdavteambition.model;
|
||||
|
||||
public class DownloadRequest {
|
||||
private String drive_id;
|
||||
private String file_id;
|
||||
private Integer expire_sec = 14400;
|
||||
|
||||
public String getDrive_id() {
|
||||
return drive_id;
|
||||
}
|
||||
|
||||
public void setDrive_id(String drive_id) {
|
||||
this.drive_id = drive_id;
|
||||
}
|
||||
|
||||
public String getFile_id() {
|
||||
return file_id;
|
||||
}
|
||||
|
||||
public void setFile_id(String file_id) {
|
||||
this.file_id = file_id;
|
||||
}
|
||||
|
||||
public Integer getExpire_sec() {
|
||||
return expire_sec;
|
||||
}
|
||||
|
||||
public void setExpire_sec(Integer expire_sec) {
|
||||
this.expire_sec = expire_sec;
|
||||
}
|
||||
}
|
|
@ -1,17 +1,17 @@
|
|||
package com.github.zxbu.webdavteambition.model;
|
||||
|
||||
public class Page {
|
||||
private int offset;
|
||||
private String marker;
|
||||
private int limit;
|
||||
private String order_by;
|
||||
private String order_direction;
|
||||
|
||||
public int getOffset() {
|
||||
return offset;
|
||||
public String getMarker() {
|
||||
return marker;
|
||||
}
|
||||
|
||||
public void setOffset(int offset) {
|
||||
this.offset = offset;
|
||||
public void setMarker(String marker) {
|
||||
this.marker = marker;
|
||||
}
|
||||
|
||||
public int getLimit() {
|
||||
|
|
|
@ -0,0 +1,42 @@
|
|||
package com.github.zxbu.webdavteambition.model;
|
||||
|
||||
import java.util.List;
|
||||
|
||||
public class RefreshUploadUrlRequest {
|
||||
private String drive_id;
|
||||
private List<UploadPreRequest.PartInfo> part_info_list;
|
||||
private String file_id;
|
||||
private String upload_id;
|
||||
|
||||
public String getDrive_id() {
|
||||
return drive_id;
|
||||
}
|
||||
|
||||
public void setDrive_id(String drive_id) {
|
||||
this.drive_id = drive_id;
|
||||
}
|
||||
|
||||
public List<UploadPreRequest.PartInfo> getPart_info_list() {
|
||||
return part_info_list;
|
||||
}
|
||||
|
||||
public void setPart_info_list(List<UploadPreRequest.PartInfo> part_info_list) {
|
||||
this.part_info_list = part_info_list;
|
||||
}
|
||||
|
||||
public String getFile_id() {
|
||||
return file_id;
|
||||
}
|
||||
|
||||
public void setFile_id(String file_id) {
|
||||
this.file_id = file_id;
|
||||
}
|
||||
|
||||
public String getUpload_id() {
|
||||
return upload_id;
|
||||
}
|
||||
|
||||
public void setUpload_id(String upload_id) {
|
||||
this.upload_id = upload_id;
|
||||
}
|
||||
}
|
|
@ -4,6 +4,7 @@ import java.util.List;
|
|||
|
||||
public class TFileListResult<T> {
|
||||
private List<T> items;
|
||||
private String next_marker;
|
||||
|
||||
public List<T> getItems() {
|
||||
return items;
|
||||
|
@ -12,4 +13,12 @@ public class TFileListResult<T> {
|
|||
public void setItems(List<T> items) {
|
||||
this.items = items;
|
||||
}
|
||||
|
||||
public String getNext_marker() {
|
||||
return next_marker;
|
||||
}
|
||||
|
||||
public void setNext_marker(String next_marker) {
|
||||
this.next_marker = next_marker;
|
||||
}
|
||||
}
|
||||
|
|
|
@ -2,6 +2,8 @@ package com.github.zxbu.webdavteambition.store;
|
|||
|
||||
import com.fasterxml.jackson.core.type.TypeReference;
|
||||
import com.fasterxml.jackson.databind.ObjectMapper;
|
||||
import com.github.benmanes.caffeine.cache.Cache;
|
||||
import com.github.benmanes.caffeine.cache.Caffeine;
|
||||
import com.github.zxbu.webdavteambition.client.AliYunDriverClient;
|
||||
import com.github.zxbu.webdavteambition.model.*;
|
||||
import com.github.zxbu.webdavteambition.model.result.TFile;
|
||||
|
@ -9,16 +11,21 @@ import com.github.zxbu.webdavteambition.model.result.TFileListResult;
|
|||
import com.github.zxbu.webdavteambition.model.result.UploadPreResult;
|
||||
import com.github.zxbu.webdavteambition.util.JsonUtil;
|
||||
import net.sf.webdav.exceptions.WebdavException;
|
||||
import okhttp3.HttpUrl;
|
||||
import okhttp3.Response;
|
||||
import org.apache.tomcat.util.http.fileupload.IOUtils;
|
||||
import org.slf4j.Logger;
|
||||
import org.slf4j.LoggerFactory;
|
||||
import org.springframework.beans.factory.annotation.Autowired;
|
||||
import org.springframework.stereotype.Service;
|
||||
import org.springframework.util.StringUtils;
|
||||
|
||||
import javax.servlet.http.HttpServletRequest;
|
||||
import java.io.IOException;
|
||||
import java.io.InputStream;
|
||||
import java.util.*;
|
||||
import java.util.concurrent.ConcurrentHashMap;
|
||||
import java.util.concurrent.TimeUnit;
|
||||
|
||||
@Service
|
||||
public class AliYunDriverClientService {
|
||||
|
@ -27,44 +34,64 @@ public class AliYunDriverClientService {
|
|||
private static String rootPath = "/";
|
||||
private static int chunkSize = 10485760; // 10MB
|
||||
private TFile rootTFile = null;
|
||||
private ThreadLocal<Map<String, Set<TFile>>> tFilesCache = new ThreadLocal<>();
|
||||
|
||||
private static Cache<String, Set<TFile>> tFilesCache = Caffeine.newBuilder()
|
||||
.initialCapacity(128)
|
||||
.maximumSize(1024)
|
||||
.expireAfterWrite(1, TimeUnit.MINUTES)
|
||||
.build();
|
||||
|
||||
private final AliYunDriverClient client;
|
||||
|
||||
@Autowired
|
||||
private VirtualTFileService virtualTFileService;
|
||||
|
||||
public AliYunDriverClientService(AliYunDriverClient aliYunDriverClient) {
|
||||
this.client = aliYunDriverClient;
|
||||
AliYunDriverFileSystemStore.setBean(this);
|
||||
}
|
||||
|
||||
public Set<TFile> getTFiles(String nodeId) {
|
||||
Map<String, Set<TFile>> map = tFilesCache.get();
|
||||
if (map == null) {
|
||||
map = new ConcurrentHashMap<>();
|
||||
tFilesCache.set(map);
|
||||
}
|
||||
return map.computeIfAbsent(nodeId, key -> {
|
||||
FileListRequest listQuery = new FileListRequest();
|
||||
listQuery.setOffset(0);
|
||||
listQuery.setLimit(10000);
|
||||
listQuery.setOrder_by("updated_at");
|
||||
listQuery.setOrder_direction("DESC");
|
||||
listQuery.setDrive_id(client.getDriveId());
|
||||
listQuery.setParent_file_id(nodeId);
|
||||
String json = client.post("/file/list", listQuery);
|
||||
TFileListResult<TFile> tFileListResult = JsonUtil.readValue(json, new TypeReference<TFileListResult<TFile>>() {
|
||||
});
|
||||
List<TFile> tFileList = tFileListResult.getItems();
|
||||
tFileList.sort(Comparator.comparing(TFile::getUpdated_at).reversed());
|
||||
Set<TFile> tFileSets = new LinkedHashSet<>();
|
||||
for (TFile tFile : tFileList) {
|
||||
if (!tFileSets.add(tFile)) {
|
||||
LOGGER.info("当前目录下{} 存在同名文件:{},文件大小:{}", nodeId, tFile.getName(), tFile.getSize());
|
||||
}
|
||||
}
|
||||
// 对文件名进行去重,只保留最新的一个
|
||||
return tFileSets;
|
||||
Set<TFile> tFiles = tFilesCache.get(nodeId, key -> {
|
||||
// 获取真实的文件列表
|
||||
return getTFiles2(nodeId);
|
||||
});
|
||||
Set<TFile> all = new LinkedHashSet<>(tFiles);
|
||||
// 获取上传中的文件列表
|
||||
Collection<TFile> virtualTFiles = virtualTFileService.list(nodeId);
|
||||
all.addAll(virtualTFiles);
|
||||
return all;
|
||||
}
|
||||
|
||||
private Set<TFile> getTFiles2(String nodeId) {
|
||||
List<TFile> tFileList = fileListFromApi(nodeId, null, new ArrayList<>());
|
||||
tFileList.sort(Comparator.comparing(TFile::getUpdated_at).reversed());
|
||||
Set<TFile> tFileSets = new LinkedHashSet<>();
|
||||
for (TFile tFile : tFileList) {
|
||||
if (!tFileSets.add(tFile)) {
|
||||
LOGGER.info("当前目录下{} 存在同名文件:{},文件大小:{}", nodeId, tFile.getName(), tFile.getSize());
|
||||
}
|
||||
}
|
||||
// 对文件名进行去重,只保留最新的一个
|
||||
return tFileSets;
|
||||
}
|
||||
|
||||
private List<TFile> fileListFromApi(String nodeId, String marker, List<TFile> all) {
|
||||
FileListRequest listQuery = new FileListRequest();
|
||||
listQuery.setMarker(marker);
|
||||
listQuery.setLimit(100);
|
||||
listQuery.setOrder_by("updated_at");
|
||||
listQuery.setOrder_direction("DESC");
|
||||
listQuery.setDrive_id(client.getDriveId());
|
||||
listQuery.setParent_file_id(nodeId);
|
||||
String json = client.post("/file/list", listQuery);
|
||||
TFileListResult<TFile> tFileListResult = JsonUtil.readValue(json, new TypeReference<TFileListResult<TFile>>() {
|
||||
});
|
||||
all.addAll(tFileListResult.getItems());
|
||||
if (!StringUtils.hasLength(tFileListResult.getNext_marker())) {
|
||||
return all;
|
||||
}
|
||||
return fileListFromApi(nodeId, tFileListResult.getNext_marker(), all);
|
||||
}
|
||||
|
||||
|
||||
|
@ -125,20 +152,43 @@ public class AliYunDriverClientService {
|
|||
UploadPreResult uploadPreResult = JsonUtil.readValue(json, UploadPreResult.class);
|
||||
List<UploadPreRequest.PartInfo> partInfoList = uploadPreResult.getPart_info_list();
|
||||
if (partInfoList != null) {
|
||||
if (size > 0) {
|
||||
virtualTFileService.createTFile(parent.getFile_id(), uploadPreResult);
|
||||
}
|
||||
LOGGER.info("文件预处理成功,开始上传。文件名:{},上传URL数量:{}", path, partInfoList.size());
|
||||
|
||||
byte[] buffer = new byte[chunkSize];
|
||||
for (int i = 0; i < partInfoList.size(); i++) {
|
||||
UploadPreRequest.PartInfo partInfo = partInfoList.get(i);
|
||||
|
||||
long expires = Long.parseLong(Objects.requireNonNull(Objects.requireNonNull(HttpUrl.parse(partInfo.getUpload_url())).queryParameter("x-oss-expires")));
|
||||
if (System.currentTimeMillis() / 1000 + 10 >= expires) {
|
||||
// 已过期,重新置换UploadUrl
|
||||
RefreshUploadUrlRequest refreshUploadUrlRequest = new RefreshUploadUrlRequest();
|
||||
refreshUploadUrlRequest.setDrive_id(client.getDriveId());
|
||||
refreshUploadUrlRequest.setUpload_id(uploadPreResult.getUpload_id());
|
||||
refreshUploadUrlRequest.setFile_id(uploadPreResult.getFile_id());
|
||||
refreshUploadUrlRequest.setPart_info_list(part_info_list);
|
||||
String refreshJson = client.post("/file/get_upload_url", refreshUploadUrlRequest);
|
||||
UploadPreResult refreshResult = JsonUtil.readValue(refreshJson, UploadPreResult.class);
|
||||
for (int j = i; j < partInfoList.size(); j++) {
|
||||
UploadPreRequest.PartInfo oldInfo = partInfoList.get(j);
|
||||
UploadPreRequest.PartInfo newInfo = refreshResult.getPart_info_list().stream().filter(p -> p.getPart_number().equals(oldInfo.getPart_number())).findAny().orElseThrow(NullPointerException::new);
|
||||
oldInfo.setUpload_url(newInfo.getUpload_url());
|
||||
}
|
||||
}
|
||||
|
||||
try {
|
||||
int read = IOUtils.read(inputStream, buffer, 0, buffer.length);
|
||||
if (read == -1) {
|
||||
LOGGER.info("文件上传结束。文件名:{},当前进度:{}/{}", path, (i + 1), partInfoList.size());
|
||||
return;
|
||||
}
|
||||
client.upload(partInfo.getUpload_url(), buffer, 0, read);
|
||||
LOGGER.info("文件正在上传。文件名:{},当前进度:{}/{}", path, (i+1), partInfoList.size());
|
||||
|
||||
virtualTFileService.updateLength(parent.getFile_id(), uploadPreResult.getFile_id(), buffer.length);
|
||||
LOGGER.info("文件正在上传。文件名:{},当前进度:{}/{}", path, (i + 1), partInfoList.size());
|
||||
} catch (IOException e) {
|
||||
virtualTFileService.remove(parent.getFile_id(), uploadPreResult.getFile_id());
|
||||
throw new WebdavException(e);
|
||||
}
|
||||
}
|
||||
|
@ -152,29 +202,12 @@ public class AliYunDriverClientService {
|
|||
uploadFinalRequest.setUpload_id(uploadPreResult.getUpload_id());
|
||||
|
||||
client.post("/file/complete", uploadFinalRequest);
|
||||
virtualTFileService.remove(parent.getFile_id(), uploadPreResult.getFile_id());
|
||||
LOGGER.info("文件上传成功。文件名:{}", path);
|
||||
// if (!uploadPreResult.getName().equals(pathInfo.getName())) {
|
||||
// LOGGER.info("上传文件名{}与原文件名{}不同,对文件进行重命名", uploadPreResult.getName(), pathInfo.getName());
|
||||
// TFile oldFile = getNodeIdByPath2(path);
|
||||
// // 如果旧文件存在,则先删除
|
||||
// if (oldFile != null) {
|
||||
// LOGGER.info("旧文件{}还存在,大小为{},进行删除操作,可前往网页版的回收站查看", path, oldFile.getSize());
|
||||
// remove(path);
|
||||
// try {
|
||||
// Thread.sleep(1500);
|
||||
// } catch (InterruptedException e) {
|
||||
// // no
|
||||
// }
|
||||
// }
|
||||
// RenameRequest renameRequest = new RenameRequest();
|
||||
// renameRequest.setDrive_id(client.getDriveId());
|
||||
// renameRequest.setFile_id(oldFile.getFile_id());
|
||||
// renameRequest.setName(pathInfo.getName());
|
||||
// client.post("/file/update", renameRequest);
|
||||
// }
|
||||
clearCache();
|
||||
}
|
||||
|
||||
|
||||
public void rename(String sourcePath, String newName) {
|
||||
sourcePath = normalizingPath(sourcePath);
|
||||
TFile tFile = getTFileByPath(sourcePath);
|
||||
|
@ -248,9 +281,15 @@ public class AliYunDriverClientService {
|
|||
return getNodeIdByPath2(path);
|
||||
}
|
||||
|
||||
public InputStream download(String path) {
|
||||
// String downloadUrl = getTFileByPath(path).getDownloadUrl();
|
||||
return client.download("downloadUrl");
|
||||
public Response download(String path, HttpServletRequest request, long size ) {
|
||||
TFile file = getTFileByPath(path);
|
||||
DownloadRequest downloadRequest = new DownloadRequest();
|
||||
downloadRequest.setDrive_id(client.getDriveId());
|
||||
downloadRequest.setFile_id(file.getFile_id());
|
||||
String json = client.post("/file/get_download_url", downloadRequest);
|
||||
Object url = JsonUtil.getJsonNodeValue(json, "url");
|
||||
LOGGER.debug("{} url = {}", path, url);
|
||||
return client.download(url.toString(), request, size);
|
||||
}
|
||||
|
||||
private TFile getNodeIdByPath2(String path) {
|
||||
|
@ -324,11 +363,7 @@ public class AliYunDriverClientService {
|
|||
return path;
|
||||
}
|
||||
|
||||
public void clearCache() {
|
||||
Map<String, Set<TFile>> map = tFilesCache.get();
|
||||
if (map != null) {
|
||||
map.clear();
|
||||
}
|
||||
|
||||
private void clearCache() {
|
||||
tFilesCache.invalidateAll();
|
||||
}
|
||||
}
|
||||
|
|
|
@ -3,9 +3,12 @@ package com.github.zxbu.webdavteambition.store;
|
|||
import com.github.zxbu.webdavteambition.model.FileType;
|
||||
import com.github.zxbu.webdavteambition.model.PathInfo;
|
||||
import com.github.zxbu.webdavteambition.model.result.TFile;
|
||||
import net.sf.webdav.*;
|
||||
import net.sf.webdav.exceptions.UnauthenticatedException;
|
||||
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 okhttp3.Response;
|
||||
import org.slf4j.Logger;
|
||||
import org.slf4j.LoggerFactory;
|
||||
|
||||
|
@ -16,6 +19,7 @@ import java.io.IOException;
|
|||
import java.io.InputStream;
|
||||
import java.security.Principal;
|
||||
import java.util.Enumeration;
|
||||
import java.util.Optional;
|
||||
import java.util.Set;
|
||||
|
||||
public class AliYunDriverFileSystemStore implements IWebdavStore {
|
||||
|
@ -35,28 +39,23 @@ public class AliYunDriverFileSystemStore implements IWebdavStore {
|
|||
|
||||
@Override
|
||||
public void destroy() {
|
||||
LOGGER.debug("destroy");
|
||||
LOGGER.info("destroy");
|
||||
|
||||
}
|
||||
|
||||
@Override
|
||||
public ITransaction begin(Principal principal, HttpServletRequest req, HttpServletResponse resp) {
|
||||
LOGGER.debug("begin");
|
||||
aliYunDriverClientService.clearCache();
|
||||
return new Transaction(principal, req, resp);
|
||||
}
|
||||
|
||||
@Override
|
||||
public void checkAuthentication(ITransaction transaction) {
|
||||
LOGGER.debug("checkAuthentication");
|
||||
if (transaction.getPrincipal() == null) {
|
||||
throw new UnauthenticatedException(WebdavStatus.SC_UNAUTHORIZED);
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
public void commit(ITransaction transaction) {
|
||||
aliYunDriverClientService.clearCache();
|
||||
LOGGER.debug("commit");
|
||||
}
|
||||
|
||||
|
@ -81,8 +80,24 @@ public class AliYunDriverFileSystemStore implements IWebdavStore {
|
|||
|
||||
@Override
|
||||
public InputStream getResourceContent(ITransaction transaction, String resourceUri) {
|
||||
LOGGER.debug("getResourceContent: {}", resourceUri);
|
||||
return aliYunDriverClientService.download(resourceUri);
|
||||
LOGGER.info("getResourceContent: {}", resourceUri);
|
||||
Enumeration<String> headerNames = transaction.getRequest().getHeaderNames();
|
||||
while (headerNames.hasMoreElements()) {
|
||||
String s = headerNames.nextElement();
|
||||
LOGGER.debug("{} request: {} = {}",resourceUri, s, transaction.getRequest().getHeader(s));
|
||||
}
|
||||
HttpServletResponse response = transaction.getResponse();
|
||||
long size = getResourceLength(transaction, resourceUri);
|
||||
Response downResponse = aliYunDriverClientService.download(resourceUri, transaction.getRequest(), size);
|
||||
response.setContentLengthLong(downResponse.body().contentLength());
|
||||
LOGGER.debug("{} code = {}", resourceUri, downResponse.code());
|
||||
for (String name : downResponse.headers().names()) {
|
||||
LOGGER.debug("{} downResponse: {} = {}", resourceUri, name, downResponse.header(name));
|
||||
response.addHeader(name, downResponse.header(name));
|
||||
}
|
||||
response.setStatus(downResponse.code());
|
||||
return downResponse.body().byteStream();
|
||||
|
||||
}
|
||||
|
||||
@Override
|
||||
|
@ -93,7 +108,8 @@ public class AliYunDriverFileSystemStore implements IWebdavStore {
|
|||
|
||||
long contentLength = request.getContentLength();
|
||||
if (contentLength < 0) {
|
||||
contentLength = Long.parseLong(request.getHeader("content-length"));
|
||||
contentLength = Long.parseLong(Optional.ofNullable(request.getHeader("content-length"))
|
||||
.orElse(request.getHeader("X-Expected-Entity-Length")));
|
||||
}
|
||||
aliYunDriverClientService.uploadPre(resourceUri, contentLength, content);
|
||||
|
||||
|
@ -115,7 +131,7 @@ public class AliYunDriverFileSystemStore implements IWebdavStore {
|
|||
|
||||
@Override
|
||||
public String[] getChildrenNames(ITransaction transaction, String folderUri) {
|
||||
LOGGER.debug("getChildrenNames: {}", folderUri);
|
||||
LOGGER.info("getChildrenNames: {}", folderUri);
|
||||
TFile tFile = aliYunDriverClientService.getTFileByPath(folderUri);
|
||||
if (tFile.getType().equals(FileType.file.name())) {
|
||||
return new String[0];
|
||||
|
@ -128,7 +144,11 @@ public class AliYunDriverFileSystemStore implements IWebdavStore {
|
|||
|
||||
@Override
|
||||
public long getResourceLength(ITransaction transaction, String path) {
|
||||
LOGGER.debug("getResourceLength: {}", path);
|
||||
return getResourceLength2(transaction, path);
|
||||
}
|
||||
|
||||
public long getResourceLength2(ITransaction transaction, String path) {
|
||||
LOGGER.info("getResourceLength: {}", path);
|
||||
TFile tFile = aliYunDriverClientService.getTFileByPath(path);
|
||||
if (tFile == null || tFile.getSize() == null) {
|
||||
return 384;
|
||||
|
@ -166,12 +186,12 @@ public class AliYunDriverFileSystemStore implements IWebdavStore {
|
|||
public StoredObject getStoredObject(ITransaction transaction, String uri) {
|
||||
|
||||
|
||||
LOGGER.debug("getStoredObject: {}", uri);
|
||||
LOGGER.info("getStoredObject: {}", uri);
|
||||
TFile tFile = aliYunDriverClientService.getTFileByPath(uri);
|
||||
if (tFile != null) {
|
||||
StoredObject so = new StoredObject();
|
||||
so.setFolder(tFile.getType().equalsIgnoreCase("folder"));
|
||||
so.setResourceLength(getResourceLength(transaction, uri));
|
||||
so.setResourceLength(getResourceLength2(transaction, uri));
|
||||
so.setCreationDate(tFile.getCreated_at());
|
||||
so.setLastModified(tFile.getUpdated_at());
|
||||
return so;
|
||||
|
|
|
@ -0,0 +1,68 @@
|
|||
package com.github.zxbu.webdavteambition.store;
|
||||
|
||||
import com.github.zxbu.webdavteambition.model.FileType;
|
||||
import com.github.zxbu.webdavteambition.model.result.TFile;
|
||||
import com.github.zxbu.webdavteambition.model.result.UploadPreResult;
|
||||
import org.springframework.stereotype.Service;
|
||||
|
||||
import java.util.Collection;
|
||||
import java.util.Collections;
|
||||
import java.util.Date;
|
||||
import java.util.Map;
|
||||
import java.util.concurrent.ConcurrentHashMap;
|
||||
|
||||
/**
|
||||
* 虚拟文件(用于上传时,列表展示)
|
||||
*/
|
||||
@Service
|
||||
public class VirtualTFileService {
|
||||
private final Map<String, Map<String, TFile>> virtualTFileMap = new ConcurrentHashMap<>();
|
||||
|
||||
/**
|
||||
* 创建文件
|
||||
*/
|
||||
public void createTFile(String parentId, UploadPreResult uploadPreResult) {
|
||||
Map<String, TFile> tFileMap = virtualTFileMap.computeIfAbsent(parentId, s -> new ConcurrentHashMap<>());
|
||||
tFileMap.put(uploadPreResult.getFile_id(), convert(uploadPreResult));
|
||||
}
|
||||
|
||||
public void updateLength(String parentId, String fileId, long length) {
|
||||
Map<String, TFile> tFileMap = virtualTFileMap.get(parentId);
|
||||
if (tFileMap == null) {
|
||||
return;
|
||||
}
|
||||
TFile tFile = tFileMap.get(fileId);
|
||||
if (tFile == null) {
|
||||
return;
|
||||
}
|
||||
tFile.setSize(tFile.getSize() + length);
|
||||
tFile.setUpdated_at(new Date());
|
||||
}
|
||||
|
||||
public void remove(String parentId, String fileId) {
|
||||
Map<String, TFile> tFileMap = virtualTFileMap.get(parentId);
|
||||
if (tFileMap == null) {
|
||||
return;
|
||||
}
|
||||
tFileMap.remove(fileId);
|
||||
}
|
||||
|
||||
public Collection<TFile> list(String parentId) {
|
||||
Map<String, TFile> tFileMap = virtualTFileMap.get(parentId);
|
||||
if (tFileMap == null) {
|
||||
return Collections.emptyList();
|
||||
}
|
||||
return tFileMap.values();
|
||||
}
|
||||
|
||||
private TFile convert(UploadPreResult uploadPreResult) {
|
||||
TFile tFile = new TFile();
|
||||
tFile.setCreated_at(new Date());
|
||||
tFile.setFile_id(uploadPreResult.getFile_id());
|
||||
tFile.setName(uploadPreResult.getFile_name());
|
||||
tFile.setType(FileType.file.name());
|
||||
tFile.setUpdated_at(new Date());
|
||||
tFile.setSize(0L);
|
||||
return tFile;
|
||||
}
|
||||
}
|
|
@ -18,6 +18,7 @@ package net.sf.webdav.methods;
|
|||
import java.io.IOException;
|
||||
import java.io.InputStream;
|
||||
import java.io.OutputStream;
|
||||
|
||||
import java.text.DateFormat;
|
||||
import java.text.SimpleDateFormat;
|
||||
import java.util.Arrays;
|
||||
|
@ -32,6 +33,8 @@ import net.sf.webdav.IWebdavStore;
|
|||
import net.sf.webdav.StoredObject;
|
||||
import net.sf.webdav.WebdavStatus;
|
||||
import net.sf.webdav.locking.ResourceLocks;
|
||||
import org.apache.tomcat.util.http.fileupload.IOUtils;
|
||||
import org.springframework.web.util.UriUtils;
|
||||
|
||||
public class DoGet extends DoHead {
|
||||
|
||||
|
@ -61,30 +64,33 @@ public class DoGet extends DoHead {
|
|||
OutputStream out = resp.getOutputStream();
|
||||
InputStream in = _store.getResourceContent(transaction, path);
|
||||
try {
|
||||
int read = -1;
|
||||
byte[] copyBuffer = new byte[BUF_SIZE];
|
||||
|
||||
while ((read = in.read(copyBuffer, 0, copyBuffer.length)) != -1) {
|
||||
out.write(copyBuffer, 0, read);
|
||||
if (in != null) {
|
||||
LOG.debug("开始 {}, ", path);
|
||||
IOUtils.copyLarge(in, out);
|
||||
LOG.debug("结束 {}", path);
|
||||
}
|
||||
} finally {
|
||||
// flushing causes a IOE if a file is opened on the webserver
|
||||
// client disconnected before server finished sending response
|
||||
try {
|
||||
in.close();
|
||||
if (in != null) {
|
||||
in.close();
|
||||
}
|
||||
} catch (Exception e) {
|
||||
LOG.warn("Closing InputStream causes Exception!\n"
|
||||
+ e.toString());
|
||||
LOG.warn("{} Closing InputStream causes Exception!\n", path
|
||||
,e);
|
||||
}
|
||||
try {
|
||||
out.flush();
|
||||
out.close();
|
||||
} catch (Exception e) {
|
||||
LOG.warn("Flushing OutputStream causes Exception!\n"
|
||||
+ e.toString());
|
||||
LOG.warn("{} Flushing OutputStream causes Exception!\n", path
|
||||
,e);
|
||||
}
|
||||
}
|
||||
} catch (Exception e) {
|
||||
LOG.warn("{} doBody causes Exception!\n", path
|
||||
,e);
|
||||
LOG.trace(e.toString());
|
||||
}
|
||||
}
|
||||
|
@ -141,7 +147,7 @@ public class DoGet extends DoHead {
|
|||
childrenTemp.append("\">");
|
||||
childrenTemp.append("<td>");
|
||||
childrenTemp.append("<a href=\"");
|
||||
childrenTemp.append(child);
|
||||
childrenTemp.append(UriUtils.encode(child, "utf-8"));
|
||||
StoredObject obj= _store.getStoredObject(transaction, path+"/"+child);
|
||||
if (obj == null)
|
||||
{
|
||||
|
@ -204,7 +210,7 @@ public class DoGet extends DoHead {
|
|||
/**
|
||||
* Return the CSS styles used to display the HTML representation
|
||||
* of the webdav content.
|
||||
*
|
||||
*
|
||||
* @return
|
||||
*/
|
||||
protected String getCSS()
|
||||
|
@ -266,7 +272,7 @@ public class DoGet extends DoHead {
|
|||
|
||||
/**
|
||||
* Return the header to be displayed in front of the folder content
|
||||
*
|
||||
*
|
||||
* @param transaction
|
||||
* @param path
|
||||
* @param resp
|
||||
|
@ -281,7 +287,7 @@ public class DoGet extends DoHead {
|
|||
|
||||
/**
|
||||
* Return the footer to be displayed after the folder content
|
||||
*
|
||||
*
|
||||
* @param transaction
|
||||
* @param path
|
||||
* @param resp
|
||||
|
|
|
@ -209,6 +209,9 @@ public class DoLock extends AbstractMethod {
|
|||
return;
|
||||
}
|
||||
nullSo = _store.getStoredObject(transaction, _path);
|
||||
if (nullSo == null) {
|
||||
nullSo = new StoredObject();
|
||||
}
|
||||
// define the newly created resource as null-resource
|
||||
nullSo.setNullResource(true);
|
||||
|
||||
|
@ -416,8 +419,13 @@ public class DoLock extends AbstractMethod {
|
|||
|
||||
if (currentNode.getNodeType() == Node.ELEMENT_NODE
|
||||
|| currentNode.getNodeType() == Node.TEXT_NODE) {
|
||||
_lockOwner = currentNode.getFirstChild()
|
||||
.getNodeValue();
|
||||
if (currentNode.getFirstChild() != null) {
|
||||
_lockOwner = currentNode.getFirstChild()
|
||||
.getNodeValue();
|
||||
} else {
|
||||
_lockOwner = currentNode.toString();
|
||||
}
|
||||
|
||||
}
|
||||
}
|
||||
}
|
||||
|
|
|
@ -72,6 +72,11 @@ public class DoUnlock extends DeterminableMethod {
|
|||
if (_resourceLocks.unlock(transaction, lockId, owner)) {
|
||||
StoredObject so = _store.getStoredObject(
|
||||
transaction, path);
|
||||
if (so == null) {
|
||||
resp.sendError(HttpServletResponse.SC_NOT_FOUND);
|
||||
return;
|
||||
}
|
||||
|
||||
if (so.isNullResource()) {
|
||||
_store.removeObject(transaction, path);
|
||||
}
|
||||
|
|
|
@ -5,3 +5,5 @@ logging.file.name=webdav.log
|
|||
aliyundrive.auth.enable=true
|
||||
aliyundrive.auth.user-name=admin
|
||||
aliyundrive.auth.password=admin
|
||||
|
||||
server.tomcat.connection-timeout=1d
|
Loading…
Reference in New Issue