Compare commits
59 Commits
Author | SHA1 | Date | |
---|---|---|---|
6c0d54394f | |||
ce5dacbf3f | |||
08aaa5e2c0 | |||
42c0e438d5 | |||
e4df146043 | |||
27b7dae113 | |||
293d574ce7 | |||
56b3b35556 | |||
a7a0e85a46 | |||
95c0106fdd | |||
6612338fc1 | |||
c276a1541f | |||
cc96a5bbdb | |||
0810561a8a | |||
82a5c43b94 | |||
d38f36ef44 | |||
f9533440c7 | |||
41a186b051 | |||
4e6a44253c | |||
ebda77cd43 | |||
1a1e86521f | |||
1b4740dae3 | |||
91fc8df84e | |||
e6ecf1fa30 | |||
183a6f1b3a | |||
3c2d59e272 | |||
fd80e3eaf7 | |||
4928c331a8 | |||
3ad75e54cb | |||
a2cf3ab42e | |||
e24814ee2f | |||
37b42e6e17 | |||
30ebb0f4d4 | |||
8e059c64b5 | |||
395de069c2 | |||
4c22f37d54 | |||
a73a40133d | |||
6591af58ea | |||
58568d4ef6 | |||
5295593bf8 | |||
24d031d578 | |||
7141bf0358 | |||
c5d707cf0a | |||
dfcf66b43e | |||
fa6ee62cf0 | |||
1428d90361 | |||
c413c22201 | |||
9b6adecd62 | |||
b3540cf539 | |||
f8650c9c0b | |||
bf2e5768d6 | |||
18c82e79b5 | |||
d69d24a5b2 | |||
342729179d | |||
0537449335 | |||
df90311453 | |||
876579ea3b | |||
e83081380e | |||
9daeaf7562 |
8
.github/ISSUE_TEMPLATE/bug_report.yml
vendored
8
.github/ISSUE_TEMPLATE/bug_report.yml
vendored
@ -7,6 +7,14 @@ body:
|
|||||||
value: |
|
value: |
|
||||||
Thanks for taking the time to fill out this bug report, please **confirm that your issue is not a duplicate issue and not because of your operation or version issues**
|
Thanks for taking the time to fill out this bug report, please **confirm that your issue is not a duplicate issue and not because of your operation or version issues**
|
||||||
感谢您花时间填写此错误报告,请**务必确认您的issue不是重复的且不是因为您的操作或版本问题**
|
感谢您花时间填写此错误报告,请**务必确认您的issue不是重复的且不是因为您的操作或版本问题**
|
||||||
|
- type: checkboxes
|
||||||
|
attributes:
|
||||||
|
label: Please make sure of the following things
|
||||||
|
description: You may select more than one, even select all.
|
||||||
|
options:
|
||||||
|
- label: I have read the [documentation](https://alist-doc.nn.ci).
|
||||||
|
- label: I'm sure there are no duplicate issues or discussions.
|
||||||
|
- label: I'm sure it's due to `alist` and not something else(such as `Dependencies` or `Operational`).
|
||||||
- type: input
|
- type: input
|
||||||
id: version
|
id: version
|
||||||
attributes:
|
attributes:
|
||||||
|
9
.github/ISSUE_TEMPLATE/feature_request.yml
vendored
9
.github/ISSUE_TEMPLATE/feature_request.yml
vendored
@ -2,6 +2,15 @@ name: "Feature request"
|
|||||||
description: Feature request
|
description: Feature request
|
||||||
labels: ["enhancement: pending triage"]
|
labels: ["enhancement: pending triage"]
|
||||||
body:
|
body:
|
||||||
|
- type: checkboxes
|
||||||
|
attributes:
|
||||||
|
label: Please make sure of the following things
|
||||||
|
description: You may select more than one, even select all.
|
||||||
|
options:
|
||||||
|
- label: I have read the [documentation](https://alist-doc.nn.ci).
|
||||||
|
- label: I'm sure there are no duplicate issues or discussions.
|
||||||
|
- label: I'm sure this feature is not implemented.
|
||||||
|
- label: I'm sure it's a reasonable and popular requirement.
|
||||||
- type: textarea
|
- type: textarea
|
||||||
id: feature-description
|
id: feature-description
|
||||||
attributes:
|
attributes:
|
||||||
|
4
.github/workflows/build.yml
vendored
4
.github/workflows/build.yml
vendored
@ -11,7 +11,7 @@ jobs:
|
|||||||
strategy:
|
strategy:
|
||||||
matrix:
|
matrix:
|
||||||
platform: [ubuntu-latest]
|
platform: [ubuntu-latest]
|
||||||
go-version: [1.17]
|
go-version: [1.18]
|
||||||
name: Build
|
name: Build
|
||||||
runs-on: ${{ matrix.platform }}
|
runs-on: ${{ matrix.platform }}
|
||||||
steps:
|
steps:
|
||||||
@ -30,7 +30,7 @@ jobs:
|
|||||||
with:
|
with:
|
||||||
path: alist
|
path: alist
|
||||||
|
|
||||||
- name: Install upx
|
- name: Install dependencies
|
||||||
run: |
|
run: |
|
||||||
docker pull techknowlogick/xgo:latest
|
docker pull techknowlogick/xgo:latest
|
||||||
go install src.techknowlogick.com/xgo@latest
|
go install src.techknowlogick.com/xgo@latest
|
||||||
|
2
.github/workflows/build_docker.yml
vendored
2
.github/workflows/build_docker.yml
vendored
@ -3,8 +3,6 @@ name: build_docker
|
|||||||
on:
|
on:
|
||||||
push:
|
push:
|
||||||
branches: [ v2 ]
|
branches: [ v2 ]
|
||||||
pull_request:
|
|
||||||
branches: [ v2 ]
|
|
||||||
|
|
||||||
jobs:
|
jobs:
|
||||||
build_docker:
|
build_docker:
|
||||||
|
2
.github/workflows/release.yml
vendored
2
.github/workflows/release.yml
vendored
@ -10,7 +10,7 @@ jobs:
|
|||||||
strategy:
|
strategy:
|
||||||
matrix:
|
matrix:
|
||||||
platform: [ubuntu-latest]
|
platform: [ubuntu-latest]
|
||||||
go-version: [1.17]
|
go-version: [1.18]
|
||||||
name: Release
|
name: Release
|
||||||
runs-on: ${{ matrix.platform }}
|
runs-on: ${{ matrix.platform }}
|
||||||
steps:
|
steps:
|
||||||
|
1
.gitignore
vendored
1
.gitignore
vendored
@ -22,6 +22,7 @@ dist/
|
|||||||
# vendor/
|
# vendor/
|
||||||
bin/*
|
bin/*
|
||||||
/alist
|
/alist
|
||||||
|
/alist.exe
|
||||||
*.json
|
*.json
|
||||||
public/*.html
|
public/*.html
|
||||||
public/assets/
|
public/assets/
|
||||||
|
@ -30,7 +30,7 @@ English | [中文](./README_cn.md) | [Contributors](./CONTRIBUTORS.md) | [Contri
|
|||||||
- [x] [PikPak](https://www.mypikpak.com/)
|
- [x] [PikPak](https://www.mypikpak.com/)
|
||||||
- [x] [ShandianPan](https://shandianpan.com/)
|
- [x] [ShandianPan](https://shandianpan.com/)
|
||||||
- [x] [S3](https://aws.amazon.com/s3/)
|
- [x] [S3](https://aws.amazon.com/s3/)
|
||||||
- [x] WebDav
|
- [x] WebDav(Support OneDrive/SharePoint without API)
|
||||||
- [x] Teambition([China](https://www.teambition.com/ ),[International](https://us.teambition.com/ ))
|
- [x] Teambition([China](https://www.teambition.com/ ),[International](https://us.teambition.com/ ))
|
||||||
- [x] [Mediatrack](https://www.mediatrack.cn/)
|
- [x] [Mediatrack](https://www.mediatrack.cn/)
|
||||||
- [x] [139yun](https://yun.139.com/) (Personal, Family)
|
- [x] [139yun](https://yun.139.com/) (Personal, Family)
|
||||||
@ -38,6 +38,7 @@ English | [中文](./README_cn.md) | [Contributors](./CONTRIBUTORS.md) | [Contri
|
|||||||
- [x] [Baidu Disk](http://pan.baidu.com/)
|
- [x] [Baidu Disk](http://pan.baidu.com/)
|
||||||
- [x] [Quark](https://pan.quark.cn)
|
- [x] [Quark](https://pan.quark.cn)
|
||||||
- [x] [XunleiCloud](https://pan.xunlei.com/)
|
- [x] [XunleiCloud](https://pan.xunlei.com/)
|
||||||
|
- [x] SFTP
|
||||||
- [x] Easy to deploy and out-of-the-box
|
- [x] Easy to deploy and out-of-the-box
|
||||||
- [x] File preview (PDF, markdown, code, plain text, ...)
|
- [x] File preview (PDF, markdown, code, plain text, ...)
|
||||||
- [x] Image preview in gallery mode
|
- [x] Image preview in gallery mode
|
||||||
@ -86,4 +87,4 @@ The `AList` is open-source software licensed under the AGPL-3.0 license.
|
|||||||
|
|
||||||
---
|
---
|
||||||
|
|
||||||
> [@Blog](https://nn.ci/) · [@GitHub](https://github.com/Xhofe) · [@TelegramGroup](https://t.me/alist_chat) · [@QQGroup](https://jq.qq.com/?_wv=1027&k=OVPJcv2b)
|
> [@Blog](https://nn.ci/) · [@GitHub](https://github.com/Xhofe) · [@TelegramGroup](https://t.me/alist_chat) · [@QQGroup](https://jq.qq.com/?_wv=1027&k=YJJj2Gwb)
|
@ -30,7 +30,7 @@
|
|||||||
- [x] [PikPak](https://www.mypikpak.com/)
|
- [x] [PikPak](https://www.mypikpak.com/)
|
||||||
- [x] [闪电盘](https://shandianpan.com/)
|
- [x] [闪电盘](https://shandianpan.com/)
|
||||||
- [x] [S3](https://aws.amazon.com/cn/s3/)
|
- [x] [S3](https://aws.amazon.com/cn/s3/)
|
||||||
- [x] WebDav
|
- [x] WebDav(支持无API的OneDrive/SharePoint)
|
||||||
- [x] Teambition([中国](https://www.teambition.com/ ),[国际](https://us.teambition.com/ ))
|
- [x] Teambition([中国](https://www.teambition.com/ ),[国际](https://us.teambition.com/ ))
|
||||||
- [x] [分秒帧](https://www.mediatrack.cn/)
|
- [x] [分秒帧](https://www.mediatrack.cn/)
|
||||||
- [x] [和彩云](https://yun.139.com/) (个人云, 家庭云)
|
- [x] [和彩云](https://yun.139.com/) (个人云, 家庭云)
|
||||||
@ -38,6 +38,7 @@
|
|||||||
- [x] [百度网盘](http://pan.baidu.com/)
|
- [x] [百度网盘](http://pan.baidu.com/)
|
||||||
- [x] [夸克网盘](https://pan.quark.cn)
|
- [x] [夸克网盘](https://pan.quark.cn)
|
||||||
- [x] [迅雷云盘](https://pan.xunlei.com/)
|
- [x] [迅雷云盘](https://pan.xunlei.com/)
|
||||||
|
- [x] SFTP
|
||||||
- [x] 部署方便,开箱即用
|
- [x] 部署方便,开箱即用
|
||||||
- [x] 文件预览(PDF、markdown、代码、纯文本……)
|
- [x] 文件预览(PDF、markdown、代码、纯文本……)
|
||||||
- [x] 画廊模式下的图像预览
|
- [x] 画廊模式下的图像预览
|
||||||
@ -86,4 +87,4 @@
|
|||||||
|
|
||||||
---
|
---
|
||||||
|
|
||||||
> [@Blog](https://nn.ci/) · [@GitHub](https://github.com/Xhofe) · [@TelegramGroup](https://t.me/alist_chat) · [@QQGroup](https://jq.qq.com/?_wv=1027&k=OVPJcv2b)
|
> [@博客](https://nn.ci/) · [@GitHub](https://github.com/Xhofe) · [@Telegram群](https://t.me/alist_chat) · [@QQ群](https://jq.qq.com/?_wv=1027&k=YJJj2Gwb)
|
1
alist.go
1
alist.go
@ -12,7 +12,6 @@ import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
func Init() bool {
|
func Init() bool {
|
||||||
//bootstrap.InitLog()
|
|
||||||
bootstrap.InitConf()
|
bootstrap.InitConf()
|
||||||
bootstrap.InitCron()
|
bootstrap.InitCron()
|
||||||
bootstrap.InitModel()
|
bootstrap.InitModel()
|
||||||
|
@ -1,12 +0,0 @@
|
|||||||
package main
|
|
||||||
|
|
||||||
import (
|
|
||||||
"fmt"
|
|
||||||
"net/url"
|
|
||||||
"testing"
|
|
||||||
)
|
|
||||||
|
|
||||||
func TestUrl(t *testing.T) {
|
|
||||||
s,_ := url.QueryUnescape("/ali/%E7%8C%AA%E5%A4%B4%E7%9A%84%E6%96%87%E4%BB%B6%5B%E5%98%BF%E5%98%BF%5D/%E9%82%B9%E9%82%B9%E7%9A%84%E6%96%87%E4%BB%B6/%E6%A1%8C%E9%9D%A2%E5%A3%81%E7%BA%B8/v2-e8f266ba17ae387eefed1cb22b2b5e4e_r.jpg")
|
|
||||||
fmt.Print(s)
|
|
||||||
}
|
|
@ -7,6 +7,7 @@ import (
|
|||||||
log "github.com/sirupsen/logrus"
|
log "github.com/sirupsen/logrus"
|
||||||
"io/ioutil"
|
"io/ioutil"
|
||||||
"os"
|
"os"
|
||||||
|
"path/filepath"
|
||||||
)
|
)
|
||||||
|
|
||||||
// InitConf init config
|
// InitConf init config
|
||||||
@ -46,7 +47,11 @@ func InitConf() {
|
|||||||
if !conf.Conf.Force {
|
if !conf.Conf.Force {
|
||||||
confFromEnv()
|
confFromEnv()
|
||||||
}
|
}
|
||||||
err := os.MkdirAll(conf.Conf.TempDir, 0700)
|
err := os.RemoveAll(filepath.Join(conf.Conf.TempDir))
|
||||||
|
if err != nil {
|
||||||
|
log.Errorln("failed delete temp file:", err)
|
||||||
|
}
|
||||||
|
err = os.MkdirAll(conf.Conf.TempDir, 0700)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Fatalf("create temp dir error: %s", err.Error())
|
log.Fatalf("create temp dir error: %s", err.Error())
|
||||||
}
|
}
|
||||||
|
@ -74,9 +74,9 @@ func InitModel() {
|
|||||||
log.Infof("auto migrate model...")
|
log.Infof("auto migrate model...")
|
||||||
if databaseConfig.Type == "mysql" {
|
if databaseConfig.Type == "mysql" {
|
||||||
err = conf.DB.Set("gorm:table_options", "ENGINE=InnoDB CHARSET=utf8mb4").
|
err = conf.DB.Set("gorm:table_options", "ENGINE=InnoDB CHARSET=utf8mb4").
|
||||||
AutoMigrate(&model.SettingItem{}, &model.Account{}, &model.Meta{})
|
AutoMigrate(&model.SettingItem{}, &model.Account{}, &model.Meta{}, &model.SearchFile{})
|
||||||
} else {
|
} else {
|
||||||
err = conf.DB.AutoMigrate(&model.SettingItem{}, &model.Account{}, &model.Meta{})
|
err = conf.DB.AutoMigrate(&model.SettingItem{}, &model.Account{}, &model.Meta{}, &model.SearchFile{})
|
||||||
}
|
}
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Fatalf("failed to auto migrate: %s", err.Error())
|
log.Fatalf("failed to auto migrate: %s", err.Error())
|
||||||
|
@ -118,8 +118,8 @@ func InitSettings() {
|
|||||||
Group: model.FRONT,
|
Group: model.FRONT,
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
Key: "home readme url",
|
Key: "global readme url",
|
||||||
Description: "when have multiple, the readme file to show",
|
Description: "Default display when directory has no readme",
|
||||||
Type: "string",
|
Type: "string",
|
||||||
Access: model.PUBLIC,
|
Access: model.PUBLIC,
|
||||||
Group: model.FRONT,
|
Group: model.FRONT,
|
||||||
@ -258,6 +258,14 @@ func InitSettings() {
|
|||||||
Access: model.PRIVATE,
|
Access: model.PRIVATE,
|
||||||
Group: model.BACK,
|
Group: model.BACK,
|
||||||
},
|
},
|
||||||
|
{
|
||||||
|
Key: "enable search",
|
||||||
|
Value: "false",
|
||||||
|
Type: "bool",
|
||||||
|
Access: model.PUBLIC,
|
||||||
|
Group: model.BACK,
|
||||||
|
Description: "Experimental function, not recommended as it's still under development",
|
||||||
|
},
|
||||||
}
|
}
|
||||||
for i, _ := range settings {
|
for i, _ := range settings {
|
||||||
v := settings[i]
|
v := settings[i]
|
||||||
|
7
build.sh
7
build.sh
@ -51,6 +51,7 @@ BUILD() {
|
|||||||
gitTag=$(git describe --long --tags --dirty --always)
|
gitTag=$(git describe --long --tags --dirty --always)
|
||||||
webTag=$(wget -qO- -t1 -T2 "https://api.github.com/repos/alist-org/alist-web/releases/latest" | grep "tag_name" | head -n 1 | awk -F ":" '{print $2}' | sed 's/\"//g;s/,//g;s/ //g')
|
webTag=$(wget -qO- -t1 -T2 "https://api.github.com/repos/alist-org/alist-web/releases/latest" | grep "tag_name" | head -n 1 | awk -F ":" '{print $2}' | sed 's/\"//g;s/,//g;s/ //g')
|
||||||
echo "build version: $gitTag"
|
echo "build version: $gitTag"
|
||||||
|
|
||||||
ldflags="\
|
ldflags="\
|
||||||
-w -s \
|
-w -s \
|
||||||
-X 'github.com/Xhofe/alist/conf.BuiltAt=$builtAt' \
|
-X 'github.com/Xhofe/alist/conf.BuiltAt=$builtAt' \
|
||||||
@ -60,7 +61,7 @@ BUILD() {
|
|||||||
-X 'github.com/Xhofe/alist/conf.GitTag=$gitTag' \
|
-X 'github.com/Xhofe/alist/conf.GitTag=$gitTag' \
|
||||||
-X 'github.com/Xhofe/alist/conf.WebTag=$webTag' \
|
-X 'github.com/Xhofe/alist/conf.WebTag=$webTag' \
|
||||||
"
|
"
|
||||||
|
rm -rf .git/
|
||||||
if [ "$1" == "release" ]; then
|
if [ "$1" == "release" ]; then
|
||||||
xgo -out "$appName" -ldflags="$ldflags" -tags=jsoniter .
|
xgo -out "$appName" -ldflags="$ldflags" -tags=jsoniter .
|
||||||
else
|
else
|
||||||
@ -96,7 +97,7 @@ BUILD_MUSL() {
|
|||||||
gitTag=$(git describe --long --tags --dirty --always)
|
gitTag=$(git describe --long --tags --dirty --always)
|
||||||
webTag=$(wget -qO- -t1 -T2 "https://api.github.com/repos/alist-org/alist-web/releases/latest" | grep "tag_name" | head -n 1 | awk -F ":" '{print $2}' | sed 's/\"//g;s/,//g;s/ //g')
|
webTag=$(wget -qO- -t1 -T2 "https://api.github.com/repos/alist-org/alist-web/releases/latest" | grep "tag_name" | head -n 1 | awk -F ":" '{print $2}' | sed 's/\"//g;s/,//g;s/ //g')
|
||||||
ldflags="\
|
ldflags="\
|
||||||
-w -s \
|
-w -s --extldflags '-static -fpic' \
|
||||||
-X 'github.com/Xhofe/alist/conf.BuiltAt=$builtAt' \
|
-X 'github.com/Xhofe/alist/conf.BuiltAt=$builtAt' \
|
||||||
-X 'github.com/Xhofe/alist/conf.GoVersion=$goVersion' \
|
-X 'github.com/Xhofe/alist/conf.GoVersion=$goVersion' \
|
||||||
-X 'github.com/Xhofe/alist/conf.GitAuthor=$gitAuthor' \
|
-X 'github.com/Xhofe/alist/conf.GitAuthor=$gitAuthor' \
|
||||||
@ -148,8 +149,8 @@ elif [ "$1" = "docker" ]; then
|
|||||||
elif [ "$1" = "build" ]; then
|
elif [ "$1" = "build" ]; then
|
||||||
BUILD build
|
BUILD build
|
||||||
elif [ "$1" = "release" ]; then
|
elif [ "$1" = "release" ]; then
|
||||||
BUILD release
|
|
||||||
BUILD_MUSL
|
BUILD_MUSL
|
||||||
|
BUILD release
|
||||||
RELEASE
|
RELEASE
|
||||||
else
|
else
|
||||||
echo -e "${RED_COLOR} Parameter error ${RES}"
|
echo -e "${RED_COLOR} Parameter error ${RES}"
|
||||||
|
@ -85,6 +85,7 @@ var (
|
|||||||
"Visitor WebDAV username", "Visitor WebDAV password",
|
"Visitor WebDAV username", "Visitor WebDAV password",
|
||||||
"default page size", "load type",
|
"default page size", "load type",
|
||||||
"ocr api", "favicon",
|
"ocr api", "favicon",
|
||||||
|
"enable search",
|
||||||
}
|
}
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@ -12,58 +12,8 @@ import (
|
|||||||
log "github.com/sirupsen/logrus"
|
log "github.com/sirupsen/logrus"
|
||||||
"path/filepath"
|
"path/filepath"
|
||||||
"strconv"
|
"strconv"
|
||||||
"time"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
type BaseResp struct {
|
|
||||||
Code int `json:"code"`
|
|
||||||
Message string `json:"message"`
|
|
||||||
}
|
|
||||||
|
|
||||||
type Pan123TokenResp struct {
|
|
||||||
BaseResp
|
|
||||||
Data struct {
|
|
||||||
Token string `json:"token"`
|
|
||||||
} `json:"data"`
|
|
||||||
}
|
|
||||||
|
|
||||||
type Pan123File struct {
|
|
||||||
FileName string `json:"FileName"`
|
|
||||||
Size int64 `json:"Size"`
|
|
||||||
UpdateAt *time.Time `json:"UpdateAt"`
|
|
||||||
FileId int64 `json:"FileId"`
|
|
||||||
Type int `json:"Type"`
|
|
||||||
Etag string `json:"Etag"`
|
|
||||||
S3KeyFlag string `json:"S3KeyFlag"`
|
|
||||||
}
|
|
||||||
|
|
||||||
type Pan123Files struct {
|
|
||||||
BaseResp
|
|
||||||
Data struct {
|
|
||||||
InfoList []Pan123File `json:"InfoList"`
|
|
||||||
Next string `json:"Next"`
|
|
||||||
} `json:"data"`
|
|
||||||
}
|
|
||||||
|
|
||||||
type Pan123DownResp struct {
|
|
||||||
BaseResp
|
|
||||||
Data struct {
|
|
||||||
DownloadUrl string `json:"DownloadUrl"`
|
|
||||||
} `json:"data"`
|
|
||||||
}
|
|
||||||
|
|
||||||
type UploadResp struct {
|
|
||||||
BaseResp
|
|
||||||
Data struct {
|
|
||||||
AccessKeyId string `json:"AccessKeyId"`
|
|
||||||
Bucket string `json:"Bucket"`
|
|
||||||
Key string `json:"Key"`
|
|
||||||
SecretAccessKey string `json:"SecretAccessKey"`
|
|
||||||
SessionToken string `json:"SessionToken"`
|
|
||||||
FileId int64 `json:"FileId"`
|
|
||||||
} `json:"data"`
|
|
||||||
}
|
|
||||||
|
|
||||||
func (driver Pan123) Login(account *model.Account) error {
|
func (driver Pan123) Login(account *model.Account) error {
|
||||||
url := "https://www.123pan.com/api/user/sign_in"
|
url := "https://www.123pan.com/api/user/sign_in"
|
||||||
if account.APIProxyUrl != "" {
|
if account.APIProxyUrl != "" {
|
||||||
@ -98,11 +48,7 @@ func (driver Pan123) FormatFile(file *Pan123File) *model.File {
|
|||||||
Driver: driver.Config().Name,
|
Driver: driver.Config().Name,
|
||||||
UpdatedAt: file.UpdateAt,
|
UpdatedAt: file.UpdateAt,
|
||||||
}
|
}
|
||||||
if file.Type == 1 {
|
f.Type = file.GetType()
|
||||||
f.Type = conf.FOLDER
|
|
||||||
} else {
|
|
||||||
f.Type = utils.GetFileType(filepath.Ext(file.FileName))
|
|
||||||
}
|
|
||||||
return f
|
return f
|
||||||
}
|
}
|
||||||
|
|
||||||
|
72
drivers/123/types.go
Normal file
72
drivers/123/types.go
Normal file
@ -0,0 +1,72 @@
|
|||||||
|
package _23
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/Xhofe/alist/conf"
|
||||||
|
"github.com/Xhofe/alist/utils"
|
||||||
|
"path"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
type Pan123File struct {
|
||||||
|
FileName string `json:"FileName"`
|
||||||
|
Size int64 `json:"Size"`
|
||||||
|
UpdateAt *time.Time `json:"UpdateAt"`
|
||||||
|
FileId int64 `json:"FileId"`
|
||||||
|
Type int `json:"Type"`
|
||||||
|
Etag string `json:"Etag"`
|
||||||
|
S3KeyFlag string `json:"S3KeyFlag"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (f Pan123File) GetSize() uint64 {
|
||||||
|
return uint64(f.Size)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (f Pan123File) GetName() string {
|
||||||
|
return f.FileName
|
||||||
|
}
|
||||||
|
|
||||||
|
func (f Pan123File) GetType() int {
|
||||||
|
if f.Type == 1 {
|
||||||
|
return conf.FOLDER
|
||||||
|
}
|
||||||
|
return utils.GetFileType(path.Ext(f.FileName))
|
||||||
|
}
|
||||||
|
|
||||||
|
type BaseResp struct {
|
||||||
|
Code int `json:"code"`
|
||||||
|
Message string `json:"message"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type Pan123TokenResp struct {
|
||||||
|
BaseResp
|
||||||
|
Data struct {
|
||||||
|
Token string `json:"token"`
|
||||||
|
} `json:"data"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type Pan123Files struct {
|
||||||
|
BaseResp
|
||||||
|
Data struct {
|
||||||
|
InfoList []Pan123File `json:"InfoList"`
|
||||||
|
Next string `json:"Next"`
|
||||||
|
} `json:"data"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type Pan123DownResp struct {
|
||||||
|
BaseResp
|
||||||
|
Data struct {
|
||||||
|
DownloadUrl string `json:"DownloadUrl"`
|
||||||
|
} `json:"data"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type UploadResp struct {
|
||||||
|
BaseResp
|
||||||
|
Data struct {
|
||||||
|
AccessKeyId string `json:"AccessKeyId"`
|
||||||
|
Bucket string `json:"Bucket"`
|
||||||
|
Key string `json:"Key"`
|
||||||
|
SecretAccessKey string `json:"SecretAccessKey"`
|
||||||
|
SessionToken string `json:"SessionToken"`
|
||||||
|
FileId int64 `json:"FileId"`
|
||||||
|
} `json:"data"`
|
||||||
|
}
|
@ -18,7 +18,6 @@ import (
|
|||||||
"math"
|
"math"
|
||||||
"net/http"
|
"net/http"
|
||||||
"net/http/cookiejar"
|
"net/http/cookiejar"
|
||||||
"path/filepath"
|
|
||||||
"regexp"
|
"regexp"
|
||||||
"strconv"
|
"strconv"
|
||||||
"strings"
|
"strings"
|
||||||
@ -60,11 +59,9 @@ func (driver Cloud189) FormatFile(file *Cloud189File) *model.File {
|
|||||||
f.UpdatedAt = &lastOpTime
|
f.UpdatedAt = &lastOpTime
|
||||||
}
|
}
|
||||||
if file.Size == -1 {
|
if file.Size == -1 {
|
||||||
f.Type = conf.FOLDER
|
|
||||||
f.Size = 0
|
f.Size = 0
|
||||||
} else {
|
|
||||||
f.Type = utils.GetFileType(filepath.Ext(file.Name))
|
|
||||||
}
|
}
|
||||||
|
f.Type = file.GetType()
|
||||||
return f
|
return f
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -132,7 +129,7 @@ func (driver Cloud189) Login(account *model.Account) error {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
if lt == "" {
|
if lt == "" {
|
||||||
return fmt.Errorf("get empty login page")
|
return errors.New("get page: " + b)
|
||||||
}
|
}
|
||||||
captchaToken := regexp.MustCompile(`captchaToken' value='(.+?)'`).FindStringSubmatch(b)[1]
|
captchaToken := regexp.MustCompile(`captchaToken' value='(.+?)'`).FindStringSubmatch(b)[1]
|
||||||
returnUrl := regexp.MustCompile(`returnUrl = '(.+?)'`).FindStringSubmatch(b)[1]
|
returnUrl := regexp.MustCompile(`returnUrl = '(.+?)'`).FindStringSubmatch(b)[1]
|
||||||
|
@ -187,7 +187,7 @@ func (driver Cloud189) Link(args base.Args, account *model.Account) (*base.Link,
|
|||||||
link := base.Link{
|
link := base.Link{
|
||||||
Headers: []base.Header{
|
Headers: []base.Header{
|
||||||
{Name: "User-Agent", Value: base.UserAgent},
|
{Name: "User-Agent", Value: base.UserAgent},
|
||||||
{Name: "Authorization", Value: ""},
|
//{Name: "Authorization", Value: ""},
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
if res.StatusCode() == 302 {
|
if res.StatusCode() == 302 {
|
||||||
|
@ -1,5 +1,11 @@
|
|||||||
package _89
|
package _89
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/Xhofe/alist/conf"
|
||||||
|
"github.com/Xhofe/alist/utils"
|
||||||
|
"path"
|
||||||
|
)
|
||||||
|
|
||||||
type Cloud189Error struct {
|
type Cloud189Error struct {
|
||||||
ErrorCode string `json:"errorCode"`
|
ErrorCode string `json:"errorCode"`
|
||||||
ErrorMsg string `json:"errorMsg"`
|
ErrorMsg string `json:"errorMsg"`
|
||||||
@ -17,6 +23,24 @@ type Cloud189File struct {
|
|||||||
Url string `json:"url"`
|
Url string `json:"url"`
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (f Cloud189File) GetSize() uint64 {
|
||||||
|
if f.Size == -1 {
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
return uint64(f.Size)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (f Cloud189File) GetName() string {
|
||||||
|
return f.Name
|
||||||
|
}
|
||||||
|
|
||||||
|
func (f Cloud189File) GetType() int {
|
||||||
|
if f.Size == -1 {
|
||||||
|
return conf.FOLDER
|
||||||
|
}
|
||||||
|
return utils.GetFileType(path.Ext(f.Name))
|
||||||
|
}
|
||||||
|
|
||||||
type Cloud189Folder struct {
|
type Cloud189Folder struct {
|
||||||
Id int64 `json:"id"`
|
Id int64 `json:"id"`
|
||||||
LastOpTime string `json:"lastOpTime"`
|
LastOpTime string `json:"lastOpTime"`
|
||||||
|
@ -34,7 +34,7 @@ func GetState(account *model.Account) *State {
|
|||||||
SetHeaders(map[string]string{
|
SetHeaders(map[string]string{
|
||||||
"Accept": "application/json;charset=UTF-8",
|
"Accept": "application/json;charset=UTF-8",
|
||||||
"User-Agent": base.UserAgent,
|
"User-Agent": base.UserAgent,
|
||||||
}),
|
}).SetTimeout(base.DefaultTimeout),
|
||||||
}
|
}
|
||||||
userStateCache.States[account.Username] = state
|
userStateCache.States[account.Username] = state
|
||||||
return state
|
return state
|
||||||
@ -198,7 +198,7 @@ func (s *State) refreshSession(account *model.Account) error {
|
|||||||
"accessToken": s.AccessToken,
|
"accessToken": s.AccessToken,
|
||||||
}).
|
}).
|
||||||
SetHeader("X-Request-ID", uuid.NewString()).
|
SetHeader("X-Request-ID", uuid.NewString()).
|
||||||
Get("https://api.cloud.189.cn/getSessionForPC.action")
|
Get(API_URL + "/getSessionForPC.action")
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
@ -223,10 +223,8 @@ func (s *State) refreshSession(account *model.Account) error {
|
|||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *State) IsLogin() bool {
|
func (s *State) IsLogin(account *model.Account) bool {
|
||||||
_, err := s.Request("GET", API_URL+"/getUserInfo.action", nil, func(r *resty.Request) {
|
_, err := s.Request(http.MethodGet, API_URL+"/getUserInfo.action", nil, func(r *resty.Request) { r.SetQueryParams(clientSuffix()) }, account)
|
||||||
r.SetQueryParams(clientSuffix())
|
|
||||||
}, nil)
|
|
||||||
return err == nil
|
return err == nil
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -242,12 +240,12 @@ func (s *State) RefreshSession(account *model.Account) error {
|
|||||||
return s.refreshSession(account)
|
return s.refreshSession(account)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *State) Request(method string, fullUrl string, params url.Values, callback func(*resty.Request), account *model.Account) (*resty.Response, error) {
|
func (s *State) Request(method string, fullUrl string, params Params, callback func(*resty.Request), account *model.Account) (*resty.Response, error) {
|
||||||
s.Lock()
|
s.Lock()
|
||||||
dateOfGmt := getHttpDateStr()
|
dateOfGmt := getHttpDateStr()
|
||||||
sessionKey := s.SessionKey
|
sessionKey := s.SessionKey
|
||||||
sessionSecret := s.SessionSecret
|
sessionSecret := s.SessionSecret
|
||||||
if account != nil && isFamily(account) {
|
if isFamily(account) {
|
||||||
sessionKey = s.FamilySessionKey
|
sessionKey = s.FamilySessionKey
|
||||||
sessionSecret = s.FamilySessionSecret
|
sessionSecret = s.FamilySessionSecret
|
||||||
}
|
}
|
||||||
@ -267,25 +265,12 @@ func (s *State) Request(method string, fullUrl string, params url.Values, callba
|
|||||||
}
|
}
|
||||||
req.SetHeader("Signature", signatureOfHmac(sessionSecret, sessionKey, method, fullUrl, dateOfGmt, paramsData))
|
req.SetHeader("Signature", signatureOfHmac(sessionSecret, sessionKey, method, fullUrl, dateOfGmt, paramsData))
|
||||||
|
|
||||||
callback(req)
|
if callback != nil {
|
||||||
|
callback(req)
|
||||||
|
}
|
||||||
s.Unlock()
|
s.Unlock()
|
||||||
|
|
||||||
var err error
|
res, err := req.Execute(method, fullUrl)
|
||||||
var res *resty.Response
|
|
||||||
switch method {
|
|
||||||
case "GET":
|
|
||||||
res, err = req.Get(fullUrl)
|
|
||||||
case "POST":
|
|
||||||
res, err = req.Post(fullUrl)
|
|
||||||
case "DELETE":
|
|
||||||
res, err = req.Delete(fullUrl)
|
|
||||||
case "PATCH":
|
|
||||||
res, err = req.Patch(fullUrl)
|
|
||||||
case "PUT":
|
|
||||||
res, err = req.Put(fullUrl)
|
|
||||||
default:
|
|
||||||
return nil, base.ErrNotSupport
|
|
||||||
}
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
@ -298,33 +283,36 @@ func (s *State) Request(method string, fullUrl string, params url.Values, callba
|
|||||||
}
|
}
|
||||||
if erron.Code != "" && erron.Code != "SUCCESS" {
|
if erron.Code != "" && erron.Code != "SUCCESS" {
|
||||||
if erron.Msg == "" {
|
if erron.Msg == "" {
|
||||||
|
if erron.Message == "" {
|
||||||
|
return nil, fmt.Errorf(res.String())
|
||||||
|
}
|
||||||
return nil, fmt.Errorf(erron.Message)
|
return nil, fmt.Errorf(erron.Message)
|
||||||
}
|
}
|
||||||
return nil, fmt.Errorf(erron.Msg)
|
return nil, fmt.Errorf(erron.Msg)
|
||||||
}
|
}
|
||||||
if erron.ErrorCode != "" {
|
if erron.ErrorCode != "" {
|
||||||
return nil, fmt.Errorf(erron.ErrorMsg)
|
switch erron.ErrorCode {
|
||||||
}
|
case "InvalidSessionKey":
|
||||||
|
|
||||||
if account != nil {
|
|
||||||
switch utils.Json.Get(res.Body(), "res_code").ToInt64() {
|
|
||||||
case 11, 18:
|
|
||||||
if err := s.RefreshSession(account); err != nil {
|
if err := s.RefreshSession(account); err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
return s.Request(method, fullUrl, params, callback, account)
|
return s.Request(method, fullUrl, params, callback, account)
|
||||||
case 0:
|
|
||||||
if res.StatusCode() == http.StatusOK {
|
|
||||||
return res, nil
|
|
||||||
}
|
|
||||||
fallthrough
|
|
||||||
default:
|
|
||||||
return nil, fmt.Errorf(res.String())
|
|
||||||
}
|
}
|
||||||
|
return nil, fmt.Errorf(erron.ErrorMsg)
|
||||||
}
|
}
|
||||||
|
|
||||||
if utils.Json.Get(res.Body(), "res_code").ToInt64() != 0 {
|
switch utils.Json.Get(res.Body(), "res_code").ToInt64() {
|
||||||
return res, fmt.Errorf(utils.Json.Get(res.Body(), "res_message").ToString())
|
case 11, 18:
|
||||||
|
if err := s.RefreshSession(account); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return s.Request(method, fullUrl, params, callback, account)
|
||||||
|
case 0:
|
||||||
|
if res.StatusCode() == http.StatusOK {
|
||||||
|
return res, nil
|
||||||
|
}
|
||||||
|
return nil, fmt.Errorf(res.String())
|
||||||
|
default:
|
||||||
|
return nil, fmt.Errorf(utils.Json.Get(res.Body(), "res_message").ToString())
|
||||||
}
|
}
|
||||||
return res, nil
|
|
||||||
}
|
}
|
||||||
|
@ -1,16 +1,19 @@
|
|||||||
package _189
|
package _189
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"bytes"
|
||||||
"crypto/md5"
|
"crypto/md5"
|
||||||
|
"encoding/base64"
|
||||||
"encoding/hex"
|
"encoding/hex"
|
||||||
"fmt"
|
"fmt"
|
||||||
"io"
|
"io"
|
||||||
"io/ioutil"
|
"io/ioutil"
|
||||||
|
"math"
|
||||||
"net/http"
|
"net/http"
|
||||||
|
"net/url"
|
||||||
"os"
|
"os"
|
||||||
"path/filepath"
|
"path/filepath"
|
||||||
"strings"
|
"strings"
|
||||||
"time"
|
|
||||||
|
|
||||||
"github.com/Xhofe/alist/conf"
|
"github.com/Xhofe/alist/conf"
|
||||||
"github.com/Xhofe/alist/drivers/base"
|
"github.com/Xhofe/alist/drivers/base"
|
||||||
@ -50,10 +53,9 @@ func (driver Cloud189) Items() []base.Item {
|
|||||||
Description: "account password",
|
Description: "account password",
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
Name: "root_folder",
|
Name: "root_folder",
|
||||||
Label: "root folder file_id",
|
Label: "root folder file_id",
|
||||||
Type: base.TypeString,
|
Type: base.TypeString,
|
||||||
Required: true,
|
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
Name: "internal_type",
|
Name: "internal_type",
|
||||||
@ -63,10 +65,9 @@ func (driver Cloud189) Items() []base.Item {
|
|||||||
Values: "Personal,Family",
|
Values: "Personal,Family",
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
Name: "site_id",
|
Name: "site_id",
|
||||||
Label: "family id",
|
Label: "family id",
|
||||||
Type: base.TypeString,
|
Type: base.TypeString,
|
||||||
Required: true,
|
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
Name: "order_by",
|
Name: "order_by",
|
||||||
@ -82,6 +83,11 @@ func (driver Cloud189) Items() []base.Item {
|
|||||||
Values: "true,false",
|
Values: "true,false",
|
||||||
Required: true,
|
Required: true,
|
||||||
},
|
},
|
||||||
|
{
|
||||||
|
Name: "bool_1",
|
||||||
|
Label: "fast upload",
|
||||||
|
Type: base.TypeBool,
|
||||||
|
},
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -92,10 +98,14 @@ func (driver Cloud189) Save(account *model.Account, old *model.Account) error {
|
|||||||
|
|
||||||
if !isFamily(account) && account.RootFolder == "" {
|
if !isFamily(account) && account.RootFolder == "" {
|
||||||
account.RootFolder = "-11"
|
account.RootFolder = "-11"
|
||||||
|
account.SiteId = ""
|
||||||
|
}
|
||||||
|
if isFamily(account) && account.RootFolder == "-11" {
|
||||||
|
account.RootFolder = ""
|
||||||
}
|
}
|
||||||
|
|
||||||
state := GetState(account)
|
state := GetState(account)
|
||||||
if !state.IsLogin() {
|
if !state.IsLogin(account) {
|
||||||
if err := state.Login(account); err != nil {
|
if err := state.Login(account); err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
@ -121,7 +131,7 @@ func (driver Cloud189) Save(account *model.Account, old *model.Account) error {
|
|||||||
|
|
||||||
func (driver Cloud189) getFamilyInfoList(account *model.Account) ([]FamilyInfoResp, error) {
|
func (driver Cloud189) getFamilyInfoList(account *model.Account) ([]FamilyInfoResp, error) {
|
||||||
var resp FamilyInfoListResp
|
var resp FamilyInfoListResp
|
||||||
_, err := GetState(account).Request("GET", API_URL+"/family/manage/getFamilyList.action", nil, func(r *resty.Request) {
|
_, err := GetState(account).Request(http.MethodGet, API_URL+"/family/manage/getFamilyList.action", nil, func(r *resty.Request) {
|
||||||
r.SetQueryParams(clientSuffix())
|
r.SetQueryParams(clientSuffix())
|
||||||
r.SetResult(&resp)
|
r.SetResult(&resp)
|
||||||
}, account)
|
}, account)
|
||||||
@ -179,16 +189,16 @@ func (driver Cloud189) Files(path string, account *model.Account) ([]model.File,
|
|||||||
client := GetState(account)
|
client := GetState(account)
|
||||||
for pageNum := 1; ; pageNum++ {
|
for pageNum := 1; ; pageNum++ {
|
||||||
var resp Cloud189FilesResp
|
var resp Cloud189FilesResp
|
||||||
queryparam := map[string]string{
|
_, err = client.Request(http.MethodGet, fullUrl, nil, func(r *resty.Request) {
|
||||||
"folderId": file.Id,
|
r.SetQueryParams(clientSuffix()).
|
||||||
"fileType": "0",
|
SetQueryParams(map[string]string{
|
||||||
"mediaAttr": "0",
|
"folderId": file.Id,
|
||||||
"iconOption": "5",
|
"fileType": "0",
|
||||||
"pageNum": fmt.Sprint(pageNum),
|
"mediaAttr": "0",
|
||||||
"pageSize": "130",
|
"iconOption": "5",
|
||||||
}
|
"pageNum": fmt.Sprint(pageNum),
|
||||||
_, err = client.Request("GET", fullUrl, nil, func(r *resty.Request) {
|
"pageSize": "130",
|
||||||
r.SetQueryParams(clientSuffix()).SetQueryParams(queryparam)
|
})
|
||||||
if isFamily(account) {
|
if isFamily(account) {
|
||||||
r.SetQueryParams(map[string]string{
|
r.SetQueryParams(map[string]string{
|
||||||
"familyId": account.SiteId,
|
"familyId": account.SiteId,
|
||||||
@ -212,10 +222,6 @@ func (driver Cloud189) Files(path string, account *model.Account) ([]model.File,
|
|||||||
break
|
break
|
||||||
}
|
}
|
||||||
|
|
||||||
mustTime := func(str string) *time.Time {
|
|
||||||
time, _ := http.ParseTime(str)
|
|
||||||
return &time
|
|
||||||
}
|
|
||||||
for _, folder := range resp.FileListAO.FolderList {
|
for _, folder := range resp.FileListAO.FolderList {
|
||||||
files = append(files, model.File{
|
files = append(files, model.File{
|
||||||
Id: fmt.Sprint(folder.ID),
|
Id: fmt.Sprint(folder.ID),
|
||||||
@ -223,7 +229,7 @@ func (driver Cloud189) Files(path string, account *model.Account) ([]model.File,
|
|||||||
Size: 0,
|
Size: 0,
|
||||||
Type: conf.FOLDER,
|
Type: conf.FOLDER,
|
||||||
Driver: driver.Config().Name,
|
Driver: driver.Config().Name,
|
||||||
UpdatedAt: mustTime(folder.CreateDate),
|
UpdatedAt: MustParseTime(folder.LastOpTime),
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
for _, file := range resp.FileListAO.FileList {
|
for _, file := range resp.FileListAO.FileList {
|
||||||
@ -233,7 +239,7 @@ func (driver Cloud189) Files(path string, account *model.Account) ([]model.File,
|
|||||||
Size: file.Size,
|
Size: file.Size,
|
||||||
Type: utils.GetFileType(filepath.Ext(file.Name)),
|
Type: utils.GetFileType(filepath.Ext(file.Name)),
|
||||||
Driver: driver.Config().Name,
|
Driver: driver.Config().Name,
|
||||||
UpdatedAt: mustTime(file.CreateDate),
|
UpdatedAt: MustParseTime(file.LastOpTime),
|
||||||
Thumbnail: file.Icon.SmallUrl,
|
Thumbnail: file.Icon.SmallUrl,
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
@ -279,7 +285,7 @@ func (driver Cloud189) Link(args base.Args, account *model.Account) (*base.Link,
|
|||||||
var downloadUrl struct {
|
var downloadUrl struct {
|
||||||
URL string `json:"fileDownloadUrl"`
|
URL string `json:"fileDownloadUrl"`
|
||||||
}
|
}
|
||||||
_, err = GetState(account).Request("GET", fullUrl, nil, func(r *resty.Request) {
|
_, err = GetState(account).Request(http.MethodGet, fullUrl, nil, func(r *resty.Request) {
|
||||||
r.SetQueryParams(clientSuffix()).SetQueryParam("fileId", file.Id)
|
r.SetQueryParams(clientSuffix()).SetQueryParam("fileId", file.Id)
|
||||||
if isFamily(account) {
|
if isFamily(account) {
|
||||||
r.SetQueryParams(map[string]string{
|
r.SetQueryParams(map[string]string{
|
||||||
@ -324,7 +330,7 @@ func (driver Cloud189) MakeDir(path string, account *model.Account) error {
|
|||||||
}
|
}
|
||||||
fullUrl += "/createFolder.action"
|
fullUrl += "/createFolder.action"
|
||||||
|
|
||||||
_, err = GetState(account).Request("POST", fullUrl, nil, func(r *resty.Request) {
|
_, err = GetState(account).Request(http.MethodPost, fullUrl, nil, func(r *resty.Request) {
|
||||||
r.SetQueryParams(clientSuffix()).SetQueryParams(map[string]string{
|
r.SetQueryParams(clientSuffix()).SetQueryParams(map[string]string{
|
||||||
"folderName": name,
|
"folderName": name,
|
||||||
"relativePath": "",
|
"relativePath": "",
|
||||||
@ -354,7 +360,7 @@ func (driver Cloud189) Move(src string, dst string, account *model.Account) erro
|
|||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
_, err = GetState(account).Request("POST", API_URL+"/batch/createBatchTask.action", nil, func(r *resty.Request) {
|
_, err = GetState(account).Request(http.MethodPost, API_URL+"/batch/createBatchTask.action", nil, func(r *resty.Request) {
|
||||||
r.SetFormData(clientSuffix()).SetFormData(map[string]string{
|
r.SetFormData(clientSuffix()).SetFormData(map[string]string{
|
||||||
"type": "MOVE",
|
"type": "MOVE",
|
||||||
"taskInfos": string(MustToBytes(utils.Json.Marshal(
|
"taskInfos": string(MustToBytes(utils.Json.Marshal(
|
||||||
@ -390,10 +396,10 @@ func (driver Cloud189) Move(src string, dst string, account *model.Account) erro
|
|||||||
|
|
||||||
var queryParam map[string]string
|
var queryParam map[string]string
|
||||||
fullUrl := API_URL
|
fullUrl := API_URL
|
||||||
method := "POST"
|
method := http.MethodPost
|
||||||
if isFamily(account) {
|
if isFamily(account) {
|
||||||
fullUrl += "/family/file"
|
fullUrl += "/family/file"
|
||||||
method = "GET"
|
method = http.MethodGet
|
||||||
}
|
}
|
||||||
if srcFile.IsDir() {
|
if srcFile.IsDir() {
|
||||||
fullUrl += "/moveFolder.action"
|
fullUrl += "/moveFolder.action"
|
||||||
@ -431,10 +437,10 @@ func (driver Cloud189) Rename(src string, dst string, account *model.Account) er
|
|||||||
|
|
||||||
var queryParam map[string]string
|
var queryParam map[string]string
|
||||||
fullUrl := API_URL
|
fullUrl := API_URL
|
||||||
method := "POST"
|
method := http.MethodPost
|
||||||
if isFamily(account) {
|
if isFamily(account) {
|
||||||
fullUrl += "/family/file"
|
fullUrl += "/family/file"
|
||||||
method = "GET"
|
method = http.MethodGet
|
||||||
}
|
}
|
||||||
if srcFile.IsDir() {
|
if srcFile.IsDir() {
|
||||||
fullUrl += "/renameFolder.action"
|
fullUrl += "/renameFolder.action"
|
||||||
@ -470,7 +476,7 @@ func (driver Cloud189) Copy(src string, dst string, account *model.Account) erro
|
|||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
_, err = GetState(account).Request("POST", API_URL+"/batch/createBatchTask.action", nil, func(r *resty.Request) {
|
_, err = GetState(account).Request(http.MethodPost, API_URL+"/batch/createBatchTask.action", nil, func(r *resty.Request) {
|
||||||
r.SetFormData(clientSuffix()).SetFormData(map[string]string{
|
r.SetFormData(clientSuffix()).SetFormData(map[string]string{
|
||||||
"type": "COPY",
|
"type": "COPY",
|
||||||
"taskInfos": string(MustToBytes(utils.Json.Marshal(
|
"taskInfos": string(MustToBytes(utils.Json.Marshal(
|
||||||
@ -500,7 +506,7 @@ func (driver Cloud189) Delete(path string, account *model.Account) error {
|
|||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
_, err = GetState(account).Request("POST", API_URL+"/batch/createBatchTask.action", nil, func(r *resty.Request) {
|
_, err = GetState(account).Request(http.MethodPost, API_URL+"/batch/createBatchTask.action", nil, func(r *resty.Request) {
|
||||||
r.SetFormData(clientSuffix()).SetFormData(map[string]string{
|
r.SetFormData(clientSuffix()).SetFormData(map[string]string{
|
||||||
"type": "DELETE",
|
"type": "DELETE",
|
||||||
"taskInfos": string(MustToBytes(utils.Json.Marshal(
|
"taskInfos": string(MustToBytes(utils.Json.Marshal(
|
||||||
@ -535,12 +541,209 @@ func (driver Cloud189) Upload(file *model.FileStream, account *model.Account) er
|
|||||||
return base.ErrNotFolder
|
return base.ErrNotFolder
|
||||||
}
|
}
|
||||||
|
|
||||||
if isFamily(account) {
|
if account.Bool1 {
|
||||||
return driver.uploadFamily(file, parentFile, account)
|
return driver.FastUpload(file, parentFile, account)
|
||||||
}
|
}
|
||||||
return driver.uploadPerson(file, parentFile, account)
|
return driver.CommonUpload(file, parentFile, account)
|
||||||
|
/*
|
||||||
|
if isFamily(account) {
|
||||||
|
return driver.uploadFamily(file, parentFile, account)
|
||||||
|
}
|
||||||
|
return driver.uploadPerson(file, parentFile, account)
|
||||||
|
*/
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (driver Cloud189) CommonUpload(file *model.FileStream, parentFile *model.File, account *model.Account) error {
|
||||||
|
// 初始化上传
|
||||||
|
state := GetState(account)
|
||||||
|
const DEFAULT int64 = 10485760
|
||||||
|
count := int(math.Ceil(float64(file.Size) / float64(DEFAULT)))
|
||||||
|
|
||||||
|
params := Params{
|
||||||
|
"parentFolderId": parentFile.Id,
|
||||||
|
"fileName": url.PathEscape(file.Name),
|
||||||
|
"fileSize": fmt.Sprint(file.Size),
|
||||||
|
"sliceSize": fmt.Sprint(DEFAULT),
|
||||||
|
"lazyCheck": "1",
|
||||||
|
}
|
||||||
|
|
||||||
|
fullUrl := UPLOAD_URL
|
||||||
|
if isFamily(account) {
|
||||||
|
params.Set("familyId", account.SiteId)
|
||||||
|
fullUrl += "/family"
|
||||||
|
} else {
|
||||||
|
//params.Set("extend", `{"opScene":"1","relativepath":"","rootfolderid":""}`)
|
||||||
|
fullUrl += "/person"
|
||||||
|
}
|
||||||
|
|
||||||
|
var initMultiUpload InitMultiUploadResp
|
||||||
|
_, err := state.Request(http.MethodGet, fullUrl+"/initMultiUpload", params, func(r *resty.Request) { r.SetQueryParams(clientSuffix()).SetResult(&initMultiUpload) }, account)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
fileMd5 := md5.New()
|
||||||
|
silceMd5 := md5.New()
|
||||||
|
silceMd5Hexs := make([]string, 0, count)
|
||||||
|
byteData := bytes.NewBuffer(make([]byte, DEFAULT))
|
||||||
|
for i := 1; i <= count; i++ {
|
||||||
|
byteData.Reset()
|
||||||
|
silceMd5.Reset()
|
||||||
|
if n, err := io.CopyN(io.MultiWriter(fileMd5, silceMd5, byteData), file, DEFAULT); err != io.EOF && n == 0 {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
md5Bytes := silceMd5.Sum(nil)
|
||||||
|
silceMd5Hexs = append(silceMd5Hexs, strings.ToUpper(hex.EncodeToString(md5Bytes)))
|
||||||
|
silceMd5Base64 := base64.StdEncoding.EncodeToString(md5Bytes)
|
||||||
|
|
||||||
|
var uploadUrl UploadUrlsResp
|
||||||
|
_, err = state.Request(http.MethodGet, fullUrl+"/getMultiUploadUrls",
|
||||||
|
Params{"partInfo": fmt.Sprintf("%d-%s", i, silceMd5Base64), "uploadFileId": initMultiUpload.Data.UploadFileID},
|
||||||
|
func(r *resty.Request) { r.SetQueryParams(clientSuffix()).SetResult(&uploadUrl) },
|
||||||
|
account)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
uploadData := uploadUrl.UploadUrls[fmt.Sprint("partNumber_", i)]
|
||||||
|
req, _ := http.NewRequest(http.MethodPut, uploadData.RequestURL, byteData)
|
||||||
|
for k, v := range ParseHttpHeader(uploadData.RequestHeader) {
|
||||||
|
req.Header.Set(k, v)
|
||||||
|
}
|
||||||
|
for k, v := range clientSuffix() {
|
||||||
|
req.URL.RawQuery += fmt.Sprintf("&%s=%s", k, v)
|
||||||
|
}
|
||||||
|
r, err := base.HttpClient.Do(req)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if r.StatusCode != http.StatusOK {
|
||||||
|
data, _ := io.ReadAll(r.Body)
|
||||||
|
return fmt.Errorf(string(data))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fileMd5Hex := strings.ToUpper(hex.EncodeToString(fileMd5.Sum(nil)))
|
||||||
|
sliceMd5Hex := fileMd5Hex
|
||||||
|
if int64(file.Size) > DEFAULT {
|
||||||
|
sliceMd5Hex = strings.ToUpper(utils.GetMD5Encode(strings.Join(silceMd5Hexs, "\n")))
|
||||||
|
}
|
||||||
|
|
||||||
|
_, err = state.Request(http.MethodGet, fullUrl+"/commitMultiUploadFile",
|
||||||
|
Params{
|
||||||
|
"uploadFileId": initMultiUpload.Data.UploadFileID,
|
||||||
|
"fileMd5": fileMd5Hex,
|
||||||
|
"sliceMd5": sliceMd5Hex,
|
||||||
|
"lazyCheck": "1",
|
||||||
|
"isLog": "0",
|
||||||
|
"opertype": "3",
|
||||||
|
},
|
||||||
|
func(r *resty.Request) { r.SetQueryParams(clientSuffix()) }, account)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
func (driver Cloud189) FastUpload(file *model.FileStream, parentFile *model.File, account *model.Account) error {
|
||||||
|
tempFile, err := ioutil.TempFile(conf.Conf.TempDir, "file-*")
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
defer tempFile.Close()
|
||||||
|
defer os.Remove(tempFile.Name())
|
||||||
|
|
||||||
|
// 初始化上传
|
||||||
|
state := GetState(account)
|
||||||
|
|
||||||
|
const DEFAULT int64 = 10485760
|
||||||
|
count := int(math.Ceil(float64(file.Size) / float64(DEFAULT)))
|
||||||
|
|
||||||
|
// 优先计算所需信息
|
||||||
|
fileMd5 := md5.New()
|
||||||
|
silceMd5 := md5.New()
|
||||||
|
silceMd5Hexs := make([]string, 0, count)
|
||||||
|
silceMd5Base64s := make([]string, 0, count)
|
||||||
|
for i := 1; i <= count; i++ {
|
||||||
|
silceMd5.Reset()
|
||||||
|
if n, err := io.CopyN(io.MultiWriter(fileMd5, silceMd5, tempFile), file, DEFAULT); err != nil && n == 0 {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
md5Byte := silceMd5.Sum(nil)
|
||||||
|
silceMd5Hexs = append(silceMd5Hexs, strings.ToUpper(hex.EncodeToString(md5Byte)))
|
||||||
|
silceMd5Base64s = append(silceMd5Base64s, fmt.Sprint(i, "-", base64.StdEncoding.EncodeToString(md5Byte)))
|
||||||
|
}
|
||||||
|
fileMd5Hex := strings.ToUpper(hex.EncodeToString(fileMd5.Sum(nil)))
|
||||||
|
sliceMd5Hex := fileMd5Hex
|
||||||
|
if int64(file.Size) > DEFAULT {
|
||||||
|
sliceMd5Hex = strings.ToUpper(utils.GetMD5Encode(strings.Join(silceMd5Hexs, "\n")))
|
||||||
|
}
|
||||||
|
|
||||||
|
params := Params{
|
||||||
|
"parentFolderId": parentFile.Id,
|
||||||
|
"fileName": url.PathEscape(file.Name),
|
||||||
|
"fileSize": fmt.Sprint(file.Size),
|
||||||
|
"fileMd5": fileMd5Hex,
|
||||||
|
"sliceSize": fmt.Sprint(DEFAULT),
|
||||||
|
"sliceMd5": sliceMd5Hex,
|
||||||
|
}
|
||||||
|
|
||||||
|
fullUrl := UPLOAD_URL
|
||||||
|
if isFamily(account) {
|
||||||
|
params.Set("familyId", account.SiteId)
|
||||||
|
fullUrl += "/family"
|
||||||
|
} else {
|
||||||
|
//params.Set("extend", `{"opScene":"1","relativepath":"","rootfolderid":""}`)
|
||||||
|
fullUrl += "/person"
|
||||||
|
}
|
||||||
|
|
||||||
|
var uploadInfo InitMultiUploadResp
|
||||||
|
_, err = state.Request(http.MethodGet, fullUrl+"/initMultiUpload", params, func(r *resty.Request) { r.SetQueryParams(clientSuffix()).SetResult(&uploadInfo) }, account)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
if uploadInfo.Data.FileDataExists != 1 {
|
||||||
|
var uploadUrls UploadUrlsResp
|
||||||
|
_, err := state.Request(http.MethodGet, fullUrl+"/getMultiUploadUrls",
|
||||||
|
Params{
|
||||||
|
"uploadFileId": uploadInfo.Data.UploadFileID,
|
||||||
|
"partInfo": strings.Join(silceMd5Base64s, ","),
|
||||||
|
},
|
||||||
|
func(r *resty.Request) { r.SetQueryParams(clientSuffix()).SetResult(&uploadUrls) },
|
||||||
|
account)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
for i := 1; i <= count; i++ {
|
||||||
|
uploadData := uploadUrls.UploadUrls[fmt.Sprint("partNumber_", i)]
|
||||||
|
req, _ := http.NewRequest(http.MethodPut, uploadData.RequestURL, io.NewSectionReader(tempFile, int64(i-1)*DEFAULT, DEFAULT))
|
||||||
|
for k, v := range ParseHttpHeader(uploadData.RequestHeader) {
|
||||||
|
req.Header.Set(k, v)
|
||||||
|
}
|
||||||
|
for k, v := range clientSuffix() {
|
||||||
|
req.URL.RawQuery += fmt.Sprintf("&%s=%s", k, v)
|
||||||
|
}
|
||||||
|
r, err := base.HttpClient.Do(req)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if r.StatusCode != http.StatusOK {
|
||||||
|
data, _ := io.ReadAll(r.Body)
|
||||||
|
return fmt.Errorf(string(data))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
_, err = state.Request(http.MethodGet, fullUrl+"/commitMultiUploadFile",
|
||||||
|
Params{
|
||||||
|
"uploadFileId": uploadInfo.Data.UploadFileID,
|
||||||
|
"isLog": "0",
|
||||||
|
"opertype": "3",
|
||||||
|
},
|
||||||
|
func(r *resty.Request) { r.SetQueryParams(clientSuffix()) },
|
||||||
|
account)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
func (driver Cloud189) uploadFamily(file *model.FileStream, parentFile *model.File, account *model.Account) error {
|
func (driver Cloud189) uploadFamily(file *model.FileStream, parentFile *model.File, account *model.Account) error {
|
||||||
tempFile, err := ioutil.TempFile(conf.Conf.TempDir, "file-*")
|
tempFile, err := ioutil.TempFile(conf.Conf.TempDir, "file-*")
|
||||||
if err != nil {
|
if err != nil {
|
||||||
@ -557,7 +760,7 @@ func (driver Cloud189) uploadFamily(file *model.FileStream, parentFile *model.Fi
|
|||||||
|
|
||||||
client := GetState(account)
|
client := GetState(account)
|
||||||
var createUpload CreateUploadFileResult
|
var createUpload CreateUploadFileResult
|
||||||
_, err = client.Request("GET", API_URL+"/family/file/createFamilyFile.action", nil, func(r *resty.Request) {
|
_, err = client.Request(http.MethodGet, API_URL+"/family/file/createFamilyFile.action", nil, func(r *resty.Request) {
|
||||||
r.SetQueryParams(map[string]string{
|
r.SetQueryParams(map[string]string{
|
||||||
"fileMd5": hex.EncodeToString(fileMd5.Sum(nil)),
|
"fileMd5": hex.EncodeToString(fileMd5.Sum(nil)),
|
||||||
"fileName": file.Name,
|
"fileName": file.Name,
|
||||||
@ -579,7 +782,7 @@ func (driver Cloud189) uploadFamily(file *model.FileStream, parentFile *model.Fi
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
_, err = client.Request("GET", createUpload.FileCommitUrl, nil, func(r *resty.Request) {
|
_, err = client.Request(http.MethodGet, createUpload.FileCommitUrl, nil, func(r *resty.Request) {
|
||||||
r.SetQueryParams(clientSuffix())
|
r.SetQueryParams(clientSuffix())
|
||||||
r.SetHeaders(map[string]string{
|
r.SetHeaders(map[string]string{
|
||||||
"FamilyId": account.SiteId,
|
"FamilyId": account.SiteId,
|
||||||
@ -606,7 +809,7 @@ func (driver Cloud189) uploadPerson(file *model.FileStream, parentFile *model.Fi
|
|||||||
|
|
||||||
client := GetState(account)
|
client := GetState(account)
|
||||||
var createUpload CreateUploadFileResult
|
var createUpload CreateUploadFileResult
|
||||||
_, err = client.Request("POST", API_URL+"/createUploadFile.action", nil, func(r *resty.Request) {
|
_, err = client.Request(http.MethodPost, API_URL+"/createUploadFile.action", nil, func(r *resty.Request) {
|
||||||
r.SetQueryParams(clientSuffix())
|
r.SetQueryParams(clientSuffix())
|
||||||
r.SetFormData(clientSuffix()).SetFormData(map[string]string{
|
r.SetFormData(clientSuffix()).SetFormData(map[string]string{
|
||||||
"parentFolderId": parentFile.Id,
|
"parentFolderId": parentFile.Id,
|
||||||
@ -634,7 +837,7 @@ func (driver Cloud189) uploadPerson(file *model.FileStream, parentFile *model.Fi
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
_, err = client.Request("POST", createUpload.FileCommitUrl, nil, func(r *resty.Request) {
|
_, err = client.Request(http.MethodPost, createUpload.FileCommitUrl, nil, func(r *resty.Request) {
|
||||||
r.SetQueryParams(clientSuffix())
|
r.SetQueryParams(clientSuffix())
|
||||||
r.SetFormData(map[string]string{
|
r.SetFormData(map[string]string{
|
||||||
"uploadFileId": fmt.Sprint(createUpload.UploadFileId),
|
"uploadFileId": fmt.Sprint(createUpload.UploadFileId),
|
||||||
@ -689,7 +892,7 @@ func (driver Cloud189) getUploadFileState(uploadFileId int64, account *model.Acc
|
|||||||
fullUrl += "/getUploadFileStatus.action"
|
fullUrl += "/getUploadFileStatus.action"
|
||||||
}
|
}
|
||||||
var uploadFileState UploadFileStatusResult
|
var uploadFileState UploadFileStatusResult
|
||||||
_, err := GetState(account).Request("GET", fullUrl, nil, func(r *resty.Request) {
|
_, err := GetState(account).Request(http.MethodGet, fullUrl, nil, func(r *resty.Request) {
|
||||||
r.SetQueryParams(clientSuffix())
|
r.SetQueryParams(clientSuffix())
|
||||||
r.SetQueryParams(map[string]string{
|
r.SetQueryParams(map[string]string{
|
||||||
"uploadFileId": fmt.Sprint(uploadFileId),
|
"uploadFileId": fmt.Sprint(uploadFileId),
|
||||||
@ -704,128 +907,6 @@ func (driver Cloud189) getUploadFileState(uploadFileId int64, account *model.Acc
|
|||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
return &uploadFileState, nil
|
return &uploadFileState, nil
|
||||||
}
|
}*/
|
||||||
|
|
||||||
/*
|
|
||||||
暂时未解决
|
|
||||||
func (driver Cloud189) Upload(file *model.FileStream, account *model.Account) error {
|
|
||||||
if file == nil {
|
|
||||||
return base.ErrEmptyFile
|
|
||||||
}
|
|
||||||
|
|
||||||
parentFile, err := driver.File(file.ParentPath, account)
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
if !parentFile.IsDir() {
|
|
||||||
return base.ErrNotFolder
|
|
||||||
}
|
|
||||||
|
|
||||||
fullUrl := UPLOAD_URL
|
|
||||||
if isFamily(account) {
|
|
||||||
fullUrl += "/family"
|
|
||||||
} else {
|
|
||||||
fullUrl += "/person"
|
|
||||||
}
|
|
||||||
|
|
||||||
tempFile, err := ioutil.TempFile(conf.Conf.TempDir, "file-*")
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
defer tempFile.Close()
|
|
||||||
defer os.Remove(tempFile.Name())
|
|
||||||
|
|
||||||
// 初始化上传
|
|
||||||
const DEFAULT int64 = 10485760
|
|
||||||
count := int64(math.Ceil(float64(file.Size) / float64(DEFAULT)))
|
|
||||||
fileMd5 := md5.New()
|
|
||||||
silceMd5 := md5.New()
|
|
||||||
silceMd5Hexs := make([]string, 0, count)
|
|
||||||
silceMd5Base64s := make([]string, 0, count)
|
|
||||||
for i := int64(1); i <= count; i++ {
|
|
||||||
if _, err := io.CopyN(io.MultiWriter(fileMd5, silceMd5, tempFile), file, DEFAULT); err != io.EOF {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
md5Byte := silceMd5.Sum(nil)
|
|
||||||
silceMd5Hexs = append(silceMd5Hexs, strings.ToUpper(hex.EncodeToString(md5Byte)))
|
|
||||||
silceMd5Base64s = append(silceMd5Base64s, fmt.Sprint(i, "-", base64.StdEncoding.EncodeToString(md5Byte)))
|
|
||||||
}
|
|
||||||
fileMd5Hex := strings.ToUpper(hex.EncodeToString(fileMd5.Sum(nil)))
|
|
||||||
sliceMd5Hex := fileMd5Hex
|
|
||||||
if int64(file.Size) > DEFAULT {
|
|
||||||
sliceMd5Hex = strings.ToUpper(utils.GetMD5Encode(strings.Join(silceMd5Hexs, "\n")))
|
|
||||||
}
|
|
||||||
|
|
||||||
qID := uuid.NewString()
|
|
||||||
client := GetState(account)
|
|
||||||
param := MapToUrlValues(map[string]interface{}{
|
|
||||||
"parentFolderId": parentFile.Id,
|
|
||||||
"fileName": url.QueryEscape(file.Name),
|
|
||||||
"fileMd5": fileMd5Hex,
|
|
||||||
"fileSize": fmt.Sprint(file.Size),
|
|
||||||
"sliceMd5": sliceMd5Hex,
|
|
||||||
"sliceSize": fmt.Sprint(DEFAULT),
|
|
||||||
})
|
|
||||||
if isFamily(account) {
|
|
||||||
param.Set("familyId", account.SiteId)
|
|
||||||
}
|
|
||||||
|
|
||||||
var uploadInfo InitMultiUploadResp
|
|
||||||
_, err = client.Request("GET", fullUrl+"/initMultiUpload", param, func(r *resty.Request) {
|
|
||||||
r.SetQueryParams(clientSuffix())
|
|
||||||
r.SetHeader("X-Request-ID", qID)
|
|
||||||
r.SetResult(&uploadInfo)
|
|
||||||
}, account)
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
if uploadInfo.Data.FileDataExists != 1 {
|
|
||||||
param = MapToUrlValues(map[string]interface{}{
|
|
||||||
"uploadFileId": uploadInfo.Data.UploadFileID,
|
|
||||||
"partInfo": strings.Join(silceMd5Base64s, ","),
|
|
||||||
})
|
|
||||||
if isFamily(account) {
|
|
||||||
param.Set("familyId", account.SiteId)
|
|
||||||
}
|
|
||||||
var uploadUrls UploadUrlsResp
|
|
||||||
_, err := client.Request("GET", fullUrl+"/getMultiUploadUrls", param, func(r *resty.Request) {
|
|
||||||
r.SetQueryParams(clientSuffix())
|
|
||||||
r.SetHeader("X-Request-ID", qID).SetHeader("content-type", "application/x-www-form-urlencoded")
|
|
||||||
r.SetResult(&uploadUrls)
|
|
||||||
|
|
||||||
}, account)
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
var i int64
|
|
||||||
for _, uploadurl := range uploadUrls.UploadUrls {
|
|
||||||
req := resty.New().SetTLSClientConfig(&tls.Config{InsecureSkipVerify: true}).SetProxy("http://192.168.0.30:8888").R()
|
|
||||||
for _, header := range strings.Split(decodeURIComponent(uploadurl.RequestHeader), "&") {
|
|
||||||
i := strings.Index(header, "=")
|
|
||||||
req.SetHeader(header[0:i], header[i+1:])
|
|
||||||
}
|
|
||||||
_, err := req.SetBody(io.NewSectionReader(tempFile, i*DEFAULT, DEFAULT)).Put(uploadurl.RequestURL)
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
param = MapToUrlValues(map[string]interface{}{
|
|
||||||
"uploadFileId": uploadInfo.Data.UploadFileID,
|
|
||||||
"isLog": "0",
|
|
||||||
"opertype": "1",
|
|
||||||
})
|
|
||||||
if isFamily(account) {
|
|
||||||
param.Set("familyId", account.SiteId)
|
|
||||||
}
|
|
||||||
_, err = client.Request("GET", fullUrl+"/commitMultiUploadFile", param, func(r *resty.Request) {
|
|
||||||
r.SetHeader("X-Request-ID", qID)
|
|
||||||
r.SetQueryParams(clientSuffix())
|
|
||||||
}, account)
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
*/
|
|
||||||
var _ base.Driver = (*Cloud189)(nil)
|
var _ base.Driver = (*Cloud189)(nil)
|
||||||
|
@ -137,6 +137,7 @@ type BatchTaskInfo struct {
|
|||||||
//SrcParentId string `json:"srcParentId"`
|
//SrcParentId string `json:"srcParentId"`
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
type CreateUploadFileResult struct {
|
type CreateUploadFileResult struct {
|
||||||
// UploadFileId 上传文件请求ID
|
// UploadFileId 上传文件请求ID
|
||||||
UploadFileId int64 `json:"uploadFileId"`
|
UploadFileId int64 `json:"uploadFileId"`
|
||||||
@ -157,8 +158,8 @@ type UploadFileStatusResult struct {
|
|||||||
FileCommitUrl string `json:"fileCommitUrl"`
|
FileCommitUrl string `json:"fileCommitUrl"`
|
||||||
FileDataExists int `json:"fileDataExists"`
|
FileDataExists int `json:"fileDataExists"`
|
||||||
}
|
}
|
||||||
|
*/
|
||||||
|
|
||||||
/*
|
|
||||||
type InitMultiUploadResp struct {
|
type InitMultiUploadResp struct {
|
||||||
//Code string `json:"code"`
|
//Code string `json:"code"`
|
||||||
Data struct {
|
Data struct {
|
||||||
@ -177,4 +178,3 @@ type Part struct {
|
|||||||
RequestURL string `json:"requestURL"`
|
RequestURL string `json:"requestURL"`
|
||||||
RequestHeader string `json:"requestHeader"`
|
RequestHeader string `json:"requestHeader"`
|
||||||
}
|
}
|
||||||
*/
|
|
||||||
|
@ -15,6 +15,7 @@ import (
|
|||||||
rand2 "math/rand"
|
rand2 "math/rand"
|
||||||
"net/http"
|
"net/http"
|
||||||
"net/url"
|
"net/url"
|
||||||
|
"sort"
|
||||||
"strings"
|
"strings"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
@ -118,19 +119,17 @@ func toFamilyOrderBy(o string) string {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func MapToUrlValues(m map[string]interface{}) url.Values {
|
func ParseHttpHeader(str string) map[string]string {
|
||||||
url := make(url.Values, len(m))
|
header := make(map[string]string)
|
||||||
for k, v := range m {
|
for _, value := range strings.Split(str, "&") {
|
||||||
url.Add(k, fmt.Sprint(v))
|
i := strings.Index(value, "=")
|
||||||
|
header[strings.TrimSpace(value[0:i])] = strings.TrimSpace(value[i+1:])
|
||||||
}
|
}
|
||||||
return url
|
return header
|
||||||
}
|
}
|
||||||
|
|
||||||
func decodeURIComponent(str string) string {
|
func MustString(str string, err error) string {
|
||||||
r, _ := url.QueryUnescape(str)
|
return str
|
||||||
//r, _ := url.PathUnescape(str)
|
|
||||||
//r = strings.ReplaceAll(r, " ", "+")
|
|
||||||
return r
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func MustToBytes(b []byte, err error) []byte {
|
func MustToBytes(b []byte, err error) []byte {
|
||||||
@ -143,3 +142,36 @@ func BoolToNumber(b bool) int {
|
|||||||
}
|
}
|
||||||
return 0
|
return 0
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func MustParseTime(str string) *time.Time {
|
||||||
|
loc, _ := time.LoadLocation("Local")
|
||||||
|
lastOpTime, _ := time.ParseInLocation("2006-01-02 15:04:05", str, loc)
|
||||||
|
return &lastOpTime
|
||||||
|
}
|
||||||
|
|
||||||
|
type Params map[string]string
|
||||||
|
|
||||||
|
func (p Params) Set(k, v string) {
|
||||||
|
p[k] = v
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p Params) Encode() string {
|
||||||
|
if p == nil {
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
var buf strings.Builder
|
||||||
|
keys := make([]string, 0, len(p))
|
||||||
|
for k := range p {
|
||||||
|
keys = append(keys, k)
|
||||||
|
}
|
||||||
|
sort.Strings(keys)
|
||||||
|
for _, k := range keys {
|
||||||
|
if buf.Len() > 0 {
|
||||||
|
buf.WriteByte('&')
|
||||||
|
}
|
||||||
|
buf.WriteString(k)
|
||||||
|
buf.WriteByte('=')
|
||||||
|
buf.WriteString(p[k])
|
||||||
|
}
|
||||||
|
return buf.String()
|
||||||
|
}
|
||||||
|
@ -3,7 +3,6 @@ package alidrive
|
|||||||
import (
|
import (
|
||||||
"errors"
|
"errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
"github.com/Xhofe/alist/conf"
|
|
||||||
"github.com/Xhofe/alist/drivers/base"
|
"github.com/Xhofe/alist/drivers/base"
|
||||||
"github.com/Xhofe/alist/model"
|
"github.com/Xhofe/alist/model"
|
||||||
"github.com/Xhofe/alist/utils"
|
"github.com/Xhofe/alist/utils"
|
||||||
@ -11,36 +10,10 @@ import (
|
|||||||
jsoniter "github.com/json-iterator/go"
|
jsoniter "github.com/json-iterator/go"
|
||||||
log "github.com/sirupsen/logrus"
|
log "github.com/sirupsen/logrus"
|
||||||
"path/filepath"
|
"path/filepath"
|
||||||
"time"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
var aliClient = resty.New()
|
var aliClient = resty.New()
|
||||||
|
|
||||||
type AliRespError struct {
|
|
||||||
Code string `json:"code"`
|
|
||||||
Message string `json:"message"`
|
|
||||||
}
|
|
||||||
|
|
||||||
type AliFiles struct {
|
|
||||||
Items []AliFile `json:"items"`
|
|
||||||
NextMarker string `json:"next_marker"`
|
|
||||||
}
|
|
||||||
|
|
||||||
type AliFile struct {
|
|
||||||
DriveId string `json:"drive_id"`
|
|
||||||
CreatedAt *time.Time `json:"created_at"`
|
|
||||||
FileExtension string `json:"file_extension"`
|
|
||||||
FileId string `json:"file_id"`
|
|
||||||
Type string `json:"type"`
|
|
||||||
Name string `json:"name"`
|
|
||||||
Category string `json:"category"`
|
|
||||||
ParentFileId string `json:"parent_file_id"`
|
|
||||||
UpdatedAt *time.Time `json:"updated_at"`
|
|
||||||
Size int64 `json:"size"`
|
|
||||||
Thumbnail string `json:"thumbnail"`
|
|
||||||
Url string `json:"url"`
|
|
||||||
}
|
|
||||||
|
|
||||||
func (driver AliDrive) FormatFile(file *AliFile) *model.File {
|
func (driver AliDrive) FormatFile(file *AliFile) *model.File {
|
||||||
f := &model.File{
|
f := &model.File{
|
||||||
Id: file.FileId,
|
Id: file.FileId,
|
||||||
@ -51,17 +24,7 @@ func (driver AliDrive) FormatFile(file *AliFile) *model.File {
|
|||||||
Driver: driver.Config().Name,
|
Driver: driver.Config().Name,
|
||||||
Url: file.Url,
|
Url: file.Url,
|
||||||
}
|
}
|
||||||
if file.Type == "folder" {
|
f.Type = file.GetType()
|
||||||
f.Type = conf.FOLDER
|
|
||||||
} else {
|
|
||||||
f.Type = utils.GetFileType(file.FileExtension)
|
|
||||||
}
|
|
||||||
if file.Category == "video" {
|
|
||||||
f.Type = conf.VIDEO
|
|
||||||
}
|
|
||||||
if file.Category == "image" {
|
|
||||||
f.Type = conf.IMAGE
|
|
||||||
}
|
|
||||||
return f
|
return f
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -66,6 +66,11 @@ func (driver AliDrive) Items() []base.Item {
|
|||||||
Required: false,
|
Required: false,
|
||||||
Description: ">0 and <=200",
|
Description: ">0 and <=200",
|
||||||
},
|
},
|
||||||
|
{
|
||||||
|
Name: "bool_1",
|
||||||
|
Label: "fast upload",
|
||||||
|
Type: base.TypeBool,
|
||||||
|
},
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -371,7 +376,7 @@ func (driver AliDrive) Delete(path string, account *model.Account) error {
|
|||||||
}
|
}
|
||||||
return fmt.Errorf("%s", e.Message)
|
return fmt.Errorf("%s", e.Message)
|
||||||
}
|
}
|
||||||
if res.StatusCode() == 204 {
|
if res.StatusCode() < 400 {
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
return errors.New(res.String())
|
return errors.New(res.String())
|
||||||
@ -391,8 +396,7 @@ func (driver AliDrive) Upload(file *model.FileStream, account *model.Account) er
|
|||||||
if file == nil {
|
if file == nil {
|
||||||
return base.ErrEmptyFile
|
return base.ErrEmptyFile
|
||||||
}
|
}
|
||||||
const DEFAULT int64 = 10485760
|
|
||||||
var count = int64(math.Ceil(float64(file.GetSize()) / float64(DEFAULT)))
|
|
||||||
parentFile, err := driver.File(file.ParentPath, account)
|
parentFile, err := driver.File(file.ParentPath, account)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
@ -401,16 +405,14 @@ func (driver AliDrive) Upload(file *model.FileStream, account *model.Account) er
|
|||||||
return base.ErrNotFolder
|
return base.ErrNotFolder
|
||||||
}
|
}
|
||||||
|
|
||||||
|
const DEFAULT int64 = 10485760
|
||||||
|
var count = int(math.Ceil(float64(file.GetSize()) / float64(DEFAULT)))
|
||||||
|
|
||||||
partInfoList := make([]base.Json, 0, count)
|
partInfoList := make([]base.Json, 0, count)
|
||||||
var i int64
|
for i := 1; i <= count; i++ {
|
||||||
for i = 0; i < count; i++ {
|
partInfoList = append(partInfoList, base.Json{"part_number": i})
|
||||||
partInfoList = append(partInfoList, base.Json{
|
|
||||||
"part_number": i + 1,
|
|
||||||
})
|
|
||||||
}
|
}
|
||||||
|
|
||||||
buf := make([]byte, 1024)
|
|
||||||
n, _ := file.Read(buf[:])
|
|
||||||
reqBody := base.Json{
|
reqBody := base.Json{
|
||||||
"check_name_mode": "overwrite",
|
"check_name_mode": "overwrite",
|
||||||
"drive_id": account.DriveId,
|
"drive_id": account.DriveId,
|
||||||
@ -419,9 +421,17 @@ func (driver AliDrive) Upload(file *model.FileStream, account *model.Account) er
|
|||||||
"part_info_list": partInfoList,
|
"part_info_list": partInfoList,
|
||||||
"size": file.GetSize(),
|
"size": file.GetSize(),
|
||||||
"type": "file",
|
"type": "file",
|
||||||
"pre_hash": utils.GetSHA1Encode(string(buf[:n])),
|
|
||||||
}
|
}
|
||||||
fileReader := io.MultiReader(bytes.NewReader(buf[:n]), file.File)
|
|
||||||
|
if account.Bool1 {
|
||||||
|
buf := make([]byte, 1024)
|
||||||
|
n, _ := file.Read(buf[:])
|
||||||
|
reqBody["pre_hash"] = utils.GetSHA1Encode(string(buf[:n]))
|
||||||
|
file.File = io.NopCloser(io.MultiReader(bytes.NewReader(buf[:n]), file.File))
|
||||||
|
} else {
|
||||||
|
reqBody["content_hash_name"] = "none"
|
||||||
|
reqBody["proof_version"] = "v1"
|
||||||
|
}
|
||||||
|
|
||||||
var resp UploadResp
|
var resp UploadResp
|
||||||
var e AliRespError
|
var e AliRespError
|
||||||
@ -444,7 +454,7 @@ func (driver AliDrive) Upload(file *model.FileStream, account *model.Account) er
|
|||||||
return fmt.Errorf("%s", e.Message)
|
return fmt.Errorf("%s", e.Message)
|
||||||
}
|
}
|
||||||
|
|
||||||
if e.Code == "PreHashMatched" {
|
if e.Code == "PreHashMatched" && account.Bool1 {
|
||||||
tempFile, err := ioutil.TempFile(conf.Conf.TempDir, "file-*")
|
tempFile, err := ioutil.TempFile(conf.Conf.TempDir, "file-*")
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
@ -455,7 +465,7 @@ func (driver AliDrive) Upload(file *model.FileStream, account *model.Account) er
|
|||||||
|
|
||||||
delete(reqBody, "pre_hash")
|
delete(reqBody, "pre_hash")
|
||||||
h := sha1.New()
|
h := sha1.New()
|
||||||
if _, err = io.Copy(tempFile, io.TeeReader(fileReader, h)); err != nil {
|
if _, err = io.Copy(io.MultiWriter(tempFile, h), file.File); err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
reqBody["content_hash"] = hex.EncodeToString(h.Sum(nil))
|
reqBody["content_hash"] = hex.EncodeToString(h.Sum(nil))
|
||||||
@ -470,10 +480,11 @@ func (driver AliDrive) Upload(file *model.FileStream, account *model.Account) er
|
|||||||
o = i ? r.mod(i) : new gt.BigNumber(0);
|
o = i ? r.mod(i) : new gt.BigNumber(0);
|
||||||
(t.file.slice(o.toNumber(), Math.min(o.plus(8).toNumber(), t.file.size)))
|
(t.file.slice(o.toNumber(), Math.min(o.plus(8).toNumber(), t.file.size)))
|
||||||
*/
|
*/
|
||||||
|
buf := make([]byte, 8)
|
||||||
r, _ := new(big.Int).SetString(utils.GetMD5Encode(account.AccessToken)[:16], 16)
|
r, _ := new(big.Int).SetString(utils.GetMD5Encode(account.AccessToken)[:16], 16)
|
||||||
i := new(big.Int).SetUint64(file.Size)
|
i := new(big.Int).SetUint64(file.Size)
|
||||||
o := r.Mod(r, i)
|
o := r.Mod(r, i)
|
||||||
n, _ = io.NewSectionReader(tempFile, o.Int64(), 8).Read(buf[:8])
|
n, _ := io.NewSectionReader(tempFile, o.Int64(), 8).Read(buf[:8])
|
||||||
reqBody["proof_code"] = base64.StdEncoding.EncodeToString(buf[:n])
|
reqBody["proof_code"] = base64.StdEncoding.EncodeToString(buf[:n])
|
||||||
|
|
||||||
_, err = client.Post("https://api.aliyundrive.com/adrive/v2/file/createWithFolders")
|
_, err = client.Post("https://api.aliyundrive.com/adrive/v2/file/createWithFolders")
|
||||||
@ -491,11 +502,11 @@ func (driver AliDrive) Upload(file *model.FileStream, account *model.Account) er
|
|||||||
if _, err = tempFile.Seek(0, io.SeekStart); err != nil {
|
if _, err = tempFile.Seek(0, io.SeekStart); err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
fileReader = tempFile
|
file.File = tempFile
|
||||||
}
|
}
|
||||||
|
|
||||||
for i = 0; i < count; i++ {
|
for _, partInfo := range resp.PartInfoList {
|
||||||
req, err := http.NewRequest("PUT", resp.PartInfoList[i].UploadUrl, io.LimitReader(fileReader, DEFAULT))
|
req, err := http.NewRequest("PUT", partInfo.UploadUrl, io.LimitReader(file.File, DEFAULT))
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
@ -523,7 +534,7 @@ func (driver AliDrive) Upload(file *model.FileStream, account *model.Account) er
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
if e.Code != "" {
|
if e.Code != "" && e.Code != "PreHashMatched" {
|
||||||
//if e.Code == "AccessTokenInvalid" {
|
//if e.Code == "AccessTokenInvalid" {
|
||||||
// err = driver.RefreshToken(account)
|
// err = driver.RefreshToken(account)
|
||||||
// if err != nil {
|
// if err != nil {
|
||||||
|
53
drivers/alidrive/types.go
Normal file
53
drivers/alidrive/types.go
Normal file
@ -0,0 +1,53 @@
|
|||||||
|
package alidrive
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/Xhofe/alist/conf"
|
||||||
|
"github.com/Xhofe/alist/utils"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
type AliRespError struct {
|
||||||
|
Code string `json:"code"`
|
||||||
|
Message string `json:"message"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type AliFiles struct {
|
||||||
|
Items []AliFile `json:"items"`
|
||||||
|
NextMarker string `json:"next_marker"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type AliFile struct {
|
||||||
|
DriveId string `json:"drive_id"`
|
||||||
|
CreatedAt *time.Time `json:"created_at"`
|
||||||
|
FileExtension string `json:"file_extension"`
|
||||||
|
FileId string `json:"file_id"`
|
||||||
|
Type string `json:"type"`
|
||||||
|
Name string `json:"name"`
|
||||||
|
Category string `json:"category"`
|
||||||
|
ParentFileId string `json:"parent_file_id"`
|
||||||
|
UpdatedAt *time.Time `json:"updated_at"`
|
||||||
|
Size int64 `json:"size"`
|
||||||
|
Thumbnail string `json:"thumbnail"`
|
||||||
|
Url string `json:"url"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (f AliFile) GetSize() uint64 {
|
||||||
|
return uint64(f.Size)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (f AliFile) GetName() string {
|
||||||
|
return f.Name
|
||||||
|
}
|
||||||
|
|
||||||
|
func (f AliFile) GetType() int {
|
||||||
|
if f.Type == "folder" {
|
||||||
|
return conf.FOLDER
|
||||||
|
}
|
||||||
|
if f.Category == "video" {
|
||||||
|
return conf.VIDEO
|
||||||
|
}
|
||||||
|
if f.Category == "image" {
|
||||||
|
return conf.IMAGE
|
||||||
|
}
|
||||||
|
return utils.GetFileType(f.FileExtension)
|
||||||
|
}
|
@ -18,6 +18,7 @@ import (
|
|||||||
_ "github.com/Xhofe/alist/drivers/pikpak"
|
_ "github.com/Xhofe/alist/drivers/pikpak"
|
||||||
_ "github.com/Xhofe/alist/drivers/quark"
|
_ "github.com/Xhofe/alist/drivers/quark"
|
||||||
_ "github.com/Xhofe/alist/drivers/s3"
|
_ "github.com/Xhofe/alist/drivers/s3"
|
||||||
|
_ "github.com/Xhofe/alist/drivers/sftp"
|
||||||
_ "github.com/Xhofe/alist/drivers/shandian"
|
_ "github.com/Xhofe/alist/drivers/shandian"
|
||||||
_ "github.com/Xhofe/alist/drivers/teambition"
|
_ "github.com/Xhofe/alist/drivers/teambition"
|
||||||
_ "github.com/Xhofe/alist/drivers/uss"
|
_ "github.com/Xhofe/alist/drivers/uss"
|
||||||
|
@ -48,8 +48,7 @@ func (driver Baidu) refreshToken(account *model.Account) error {
|
|||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (driver Baidu) Request(pathname string, method int, headers, query, form map[string]string, data interface{}, resp interface{}, account *model.Account) ([]byte, error) {
|
func (driver Baidu) Request(fullurl string, method int, headers, query, form map[string]string, data interface{}, resp interface{}, account *model.Account) ([]byte, error) {
|
||||||
u := "https://pan.baidu.com/rest/2.0" + pathname
|
|
||||||
req := base.RestyClient.R()
|
req := base.RestyClient.R()
|
||||||
req.SetQueryParam("access_token", account.AccessToken)
|
req.SetQueryParam("access_token", account.AccessToken)
|
||||||
if headers != nil {
|
if headers != nil {
|
||||||
@ -71,15 +70,15 @@ func (driver Baidu) Request(pathname string, method int, headers, query, form ma
|
|||||||
var err error
|
var err error
|
||||||
switch method {
|
switch method {
|
||||||
case base.Get:
|
case base.Get:
|
||||||
res, err = req.Get(u)
|
res, err = req.Get(fullurl)
|
||||||
case base.Post:
|
case base.Post:
|
||||||
res, err = req.Post(u)
|
res, err = req.Post(fullurl)
|
||||||
case base.Patch:
|
case base.Patch:
|
||||||
res, err = req.Patch(u)
|
res, err = req.Patch(fullurl)
|
||||||
case base.Delete:
|
case base.Delete:
|
||||||
res, err = req.Delete(u)
|
res, err = req.Delete(fullurl)
|
||||||
case base.Put:
|
case base.Put:
|
||||||
res, err = req.Put(u)
|
res, err = req.Put(fullurl)
|
||||||
default:
|
default:
|
||||||
return nil, base.ErrNotSupport
|
return nil, base.ErrNotSupport
|
||||||
}
|
}
|
||||||
@ -94,7 +93,7 @@ func (driver Baidu) Request(pathname string, method int, headers, query, form ma
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
return driver.Request(pathname, method, headers, query, form, data, resp, account)
|
return driver.Request(fullurl, method, headers, query, form, data, resp, account)
|
||||||
}
|
}
|
||||||
return nil, fmt.Errorf("errno: %d, refer to https://pan.baidu.com/union/doc/", errno)
|
return nil, fmt.Errorf("errno: %d, refer to https://pan.baidu.com/union/doc/", errno)
|
||||||
}
|
}
|
||||||
@ -102,11 +101,11 @@ func (driver Baidu) Request(pathname string, method int, headers, query, form ma
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (driver Baidu) Get(pathname string, params map[string]string, resp interface{}, account *model.Account) ([]byte, error) {
|
func (driver Baidu) Get(pathname string, params map[string]string, resp interface{}, account *model.Account) ([]byte, error) {
|
||||||
return driver.Request(pathname, base.Get, nil, params, nil, nil, resp, account)
|
return driver.Request("https://pan.baidu.com/rest/2.0"+pathname, base.Get, nil, params, nil, nil, resp, account)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (driver Baidu) Post(pathname string, params map[string]string, data interface{}, resp interface{}, account *model.Account) ([]byte, error) {
|
func (driver Baidu) Post(pathname string, params map[string]string, data interface{}, resp interface{}, account *model.Account) ([]byte, error) {
|
||||||
return driver.Request(pathname, base.Post, nil, params, nil, data, resp, account)
|
return driver.Request("https://pan.baidu.com/rest/2.0"+pathname, base.Post, nil, params, nil, data, resp, account)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (driver Baidu) manage(opera string, filelist interface{}, account *model.Account) ([]byte, error) {
|
func (driver Baidu) manage(opera string, filelist interface{}, account *model.Account) ([]byte, error) {
|
||||||
|
@ -58,6 +58,14 @@ func (driver Baidu) Items() []base.Item {
|
|||||||
Default: "asc",
|
Default: "asc",
|
||||||
Required: false,
|
Required: false,
|
||||||
},
|
},
|
||||||
|
{
|
||||||
|
Name: "internal_type",
|
||||||
|
Label: "download api",
|
||||||
|
Type: base.TypeSelect,
|
||||||
|
Required: true,
|
||||||
|
Values: "official,crack",
|
||||||
|
Default: "official",
|
||||||
|
},
|
||||||
{
|
{
|
||||||
Name: "client_id",
|
Name: "client_id",
|
||||||
Label: "client id",
|
Label: "client id",
|
||||||
@ -125,6 +133,13 @@ func (driver Baidu) Files(path string, account *model.Account) ([]model.File, er
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (driver Baidu) Link(args base.Args, account *model.Account) (*base.Link, error) {
|
func (driver Baidu) Link(args base.Args, account *model.Account) (*base.Link, error) {
|
||||||
|
if account.InternalType == "crack" {
|
||||||
|
return driver.LinkCrack(args, account)
|
||||||
|
}
|
||||||
|
return driver.LinkOfficial(args, account)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (driver Baidu) LinkOfficial(args base.Args, account *model.Account) (*base.Link, error) {
|
||||||
file, err := driver.File(args.Path, account)
|
file, err := driver.File(args.Path, account)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
@ -157,6 +172,32 @@ func (driver Baidu) Link(args base.Args, account *model.Account) (*base.Link, er
|
|||||||
}}, nil
|
}}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (driver Baidu) LinkCrack(args base.Args, account *model.Account) (*base.Link, error) {
|
||||||
|
file, err := driver.File(args.Path, account)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if file.IsDir() {
|
||||||
|
return nil, base.ErrNotFile
|
||||||
|
}
|
||||||
|
var resp DownloadResp2
|
||||||
|
param := map[string]string{
|
||||||
|
"target": fmt.Sprintf("[\"%s\"]", utils.Join(account.RootFolder, args.Path)),
|
||||||
|
"dlink": "1",
|
||||||
|
"web": "5",
|
||||||
|
"origin": "dlna",
|
||||||
|
}
|
||||||
|
_, err = driver.Request("https://pan.baidu.com/api/filemetas", base.Get, nil, param, nil, nil, &resp, account)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return &base.Link{
|
||||||
|
Url: resp.Info[0].Dlink,
|
||||||
|
Headers: []base.Header{
|
||||||
|
{Name: "User-Agent", Value: "pan.baidu.com"},
|
||||||
|
}}, nil
|
||||||
|
}
|
||||||
|
|
||||||
func (driver Baidu) Path(path string, account *model.Account) (*model.File, []model.File, error) {
|
func (driver Baidu) Path(path string, account *model.Account) (*model.File, []model.File, error) {
|
||||||
file, err := driver.File(path, account)
|
file, err := driver.File(path, account)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
|
@ -74,6 +74,65 @@ type DownloadResp struct {
|
|||||||
RequestId string `json:"request_id"`
|
RequestId string `json:"request_id"`
|
||||||
}
|
}
|
||||||
|
|
||||||
|
type DownloadResp2 struct {
|
||||||
|
Errno int `json:"errno"`
|
||||||
|
Info []struct {
|
||||||
|
//ExtentTinyint4 int `json:"extent_tinyint4"`
|
||||||
|
//ExtentTinyint1 int `json:"extent_tinyint1"`
|
||||||
|
//Bitmap string `json:"bitmap"`
|
||||||
|
//Category int `json:"category"`
|
||||||
|
//Isdir int `json:"isdir"`
|
||||||
|
//Videotag int `json:"videotag"`
|
||||||
|
Dlink string `json:"dlink"`
|
||||||
|
//OperID int64 `json:"oper_id"`
|
||||||
|
//PathMd5 int `json:"path_md5"`
|
||||||
|
//Wpfile int `json:"wpfile"`
|
||||||
|
//LocalMtime int `json:"local_mtime"`
|
||||||
|
/*Thumbs struct {
|
||||||
|
Icon string `json:"icon"`
|
||||||
|
URL3 string `json:"url3"`
|
||||||
|
URL2 string `json:"url2"`
|
||||||
|
URL1 string `json:"url1"`
|
||||||
|
} `json:"thumbs"`*/
|
||||||
|
//PlaySource int `json:"play_source"`
|
||||||
|
//Share int `json:"share"`
|
||||||
|
//FileKey string `json:"file_key"`
|
||||||
|
//Errno int `json:"errno"`
|
||||||
|
//LocalCtime int `json:"local_ctime"`
|
||||||
|
//Rotate int `json:"rotate"`
|
||||||
|
//Metadata time.Time `json:"metadata"`
|
||||||
|
//Height int `json:"height"`
|
||||||
|
//SampleRate int `json:"sample_rate"`
|
||||||
|
//Width int `json:"width"`
|
||||||
|
//OwnerType int `json:"owner_type"`
|
||||||
|
//Privacy int `json:"privacy"`
|
||||||
|
//ExtentInt3 int64 `json:"extent_int3"`
|
||||||
|
//RealCategory string `json:"real_category"`
|
||||||
|
//SrcLocation string `json:"src_location"`
|
||||||
|
//MetaInfo string `json:"meta_info"`
|
||||||
|
//ID string `json:"id"`
|
||||||
|
//Duration int `json:"duration"`
|
||||||
|
//FileSize string `json:"file_size"`
|
||||||
|
//Channels int `json:"channels"`
|
||||||
|
//UseSegment int `json:"use_segment"`
|
||||||
|
//ServerCtime int `json:"server_ctime"`
|
||||||
|
//Resolution string `json:"resolution"`
|
||||||
|
//OwnerID int `json:"owner_id"`
|
||||||
|
//ExtraInfo string `json:"extra_info"`
|
||||||
|
//Size int `json:"size"`
|
||||||
|
//FsID int64 `json:"fs_id"`
|
||||||
|
//ExtentTinyint3 int `json:"extent_tinyint3"`
|
||||||
|
//Md5 string `json:"md5"`
|
||||||
|
//Path string `json:"path"`
|
||||||
|
//FrameRate int `json:"frame_rate"`
|
||||||
|
//ExtentTinyint2 int `json:"extent_tinyint2"`
|
||||||
|
//ServerFilename string `json:"server_filename"`
|
||||||
|
//ServerMtime int `json:"server_mtime"`
|
||||||
|
//TkbindID int `json:"tkbind_id"`
|
||||||
|
} `json:"info"`
|
||||||
|
RequestID int64 `json:"request_id"`
|
||||||
|
}
|
||||||
|
|
||||||
type PrecreateResp struct {
|
type PrecreateResp struct {
|
||||||
Path string `json:"path"`
|
Path string `json:"path"`
|
||||||
Uploadid string `json:"uploadid"`
|
Uploadid string `json:"uploadid"`
|
||||||
|
@ -1,7 +1,6 @@
|
|||||||
package base
|
package base
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
|
||||||
"github.com/Xhofe/alist/conf"
|
"github.com/Xhofe/alist/conf"
|
||||||
"github.com/Xhofe/alist/model"
|
"github.com/Xhofe/alist/model"
|
||||||
"github.com/Xhofe/alist/utils"
|
"github.com/Xhofe/alist/utils"
|
||||||
@ -9,12 +8,39 @@ import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
func KeyCache(path string, account *model.Account) string {
|
func KeyCache(path string, account *model.Account) string {
|
||||||
path = utils.ParsePath(path)
|
//path = utils.ParsePath(path)
|
||||||
return fmt.Sprintf("%s%s", account.Name, path)
|
key := utils.ParsePath(utils.Join(account.Name, path))
|
||||||
|
log.Debugln("cache key: ", key)
|
||||||
|
return key
|
||||||
}
|
}
|
||||||
|
|
||||||
func SetCache(path string, obj interface{}, account *model.Account) error {
|
func SaveSearchFiles[T model.ISearchFile](key string, obj []T) {
|
||||||
return conf.Cache.Set(conf.Ctx, KeyCache(path, account), obj, nil)
|
err := model.DeleteSearchFilesByPath(key)
|
||||||
|
if err != nil {
|
||||||
|
log.Errorln("failed create search files", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
files := make([]model.SearchFile, len(obj))
|
||||||
|
for i := 0; i < len(obj); i++ {
|
||||||
|
files[i] = model.SearchFile{
|
||||||
|
Path: key,
|
||||||
|
Name: obj[i].GetName(),
|
||||||
|
Size: obj[i].GetSize(),
|
||||||
|
Type: obj[i].GetType(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
err = model.CreateSearchFiles(files)
|
||||||
|
if err != nil {
|
||||||
|
log.Errorln("failed create search files", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func SetCache[T model.ISearchFile](path string, obj []T, account *model.Account) error {
|
||||||
|
key := KeyCache(path, account)
|
||||||
|
if conf.GetBool("enable search") {
|
||||||
|
go SaveSearchFiles(key, obj)
|
||||||
|
}
|
||||||
|
return conf.Cache.Set(conf.Ctx, key, obj, nil)
|
||||||
}
|
}
|
||||||
|
|
||||||
func GetCache(path string, account *model.Account) (interface{}, error) {
|
func GetCache(path string, account *model.Account) (interface{}, error) {
|
||||||
|
@ -19,8 +19,9 @@ type DriverConfig struct {
|
|||||||
}
|
}
|
||||||
|
|
||||||
type Args struct {
|
type Args struct {
|
||||||
Path string
|
Path string
|
||||||
IP string
|
IP string
|
||||||
|
Header http.Header
|
||||||
}
|
}
|
||||||
|
|
||||||
type Driver interface {
|
type Driver interface {
|
||||||
|
@ -3,6 +3,7 @@ package base
|
|||||||
import (
|
import (
|
||||||
"errors"
|
"errors"
|
||||||
"io"
|
"io"
|
||||||
|
"net/http"
|
||||||
)
|
)
|
||||||
|
|
||||||
var (
|
var (
|
||||||
@ -49,4 +50,6 @@ type Link struct {
|
|||||||
Headers []Header `json:"headers"`
|
Headers []Header `json:"headers"`
|
||||||
Data io.ReadCloser
|
Data io.ReadCloser
|
||||||
FilePath string `json:"path"` // for native
|
FilePath string `json:"path"` // for native
|
||||||
|
Status int
|
||||||
|
Header http.Header
|
||||||
}
|
}
|
||||||
|
@ -2,15 +2,10 @@ package google
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
"github.com/Xhofe/alist/conf"
|
|
||||||
"github.com/Xhofe/alist/drivers/base"
|
"github.com/Xhofe/alist/drivers/base"
|
||||||
"github.com/Xhofe/alist/model"
|
"github.com/Xhofe/alist/model"
|
||||||
"github.com/Xhofe/alist/utils"
|
|
||||||
"github.com/go-resty/resty/v2"
|
"github.com/go-resty/resty/v2"
|
||||||
log "github.com/sirupsen/logrus"
|
log "github.com/sirupsen/logrus"
|
||||||
"path/filepath"
|
|
||||||
"strconv"
|
|
||||||
"time"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
type TokenError struct {
|
type TokenError struct {
|
||||||
@ -44,19 +39,6 @@ func (driver GoogleDrive) RefreshToken(account *model.Account) error {
|
|||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
type File struct {
|
|
||||||
Id string `json:"id"`
|
|
||||||
Name string `json:"name"`
|
|
||||||
MimeType string `json:"mimeType"`
|
|
||||||
ModifiedTime *time.Time `json:"modifiedTime"`
|
|
||||||
Size string `json:"size"`
|
|
||||||
ThumbnailLink string `json:"thumbnailLink"`
|
|
||||||
}
|
|
||||||
|
|
||||||
func (driver GoogleDrive) IsDir(mimeType string) bool {
|
|
||||||
return mimeType == "application/vnd.google-apps.folder" || mimeType == "application/vnd.google-apps.shortcut"
|
|
||||||
}
|
|
||||||
|
|
||||||
func (driver GoogleDrive) FormatFile(file *File, account *model.Account) *model.File {
|
func (driver GoogleDrive) FormatFile(file *File, account *model.Account) *model.File {
|
||||||
f := &model.File{
|
f := &model.File{
|
||||||
Id: file.Id,
|
Id: file.Id,
|
||||||
@ -65,13 +47,8 @@ func (driver GoogleDrive) FormatFile(file *File, account *model.Account) *model.
|
|||||||
UpdatedAt: file.ModifiedTime,
|
UpdatedAt: file.ModifiedTime,
|
||||||
Url: "",
|
Url: "",
|
||||||
}
|
}
|
||||||
if driver.IsDir(file.MimeType) {
|
f.Size = int64(file.GetSize())
|
||||||
f.Type = conf.FOLDER
|
f.Type = file.GetType()
|
||||||
} else {
|
|
||||||
size, _ := strconv.ParseInt(file.Size, 10, 64)
|
|
||||||
f.Size = size
|
|
||||||
f.Type = utils.GetFileType(filepath.Ext(file.Name))
|
|
||||||
}
|
|
||||||
if file.ThumbnailLink != "" {
|
if file.ThumbnailLink != "" {
|
||||||
if account.APIProxyUrl != "" {
|
if account.APIProxyUrl != "" {
|
||||||
f.Thumbnail = fmt.Sprintf("%s/%s", account.APIProxyUrl, file.ThumbnailLink)
|
f.Thumbnail = fmt.Sprintf("%s/%s", account.APIProxyUrl, file.ThumbnailLink)
|
||||||
|
38
drivers/google/types.go
Normal file
38
drivers/google/types.go
Normal file
@ -0,0 +1,38 @@
|
|||||||
|
package google
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/Xhofe/alist/conf"
|
||||||
|
"github.com/Xhofe/alist/utils"
|
||||||
|
"path"
|
||||||
|
"strconv"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
type File struct {
|
||||||
|
Id string `json:"id"`
|
||||||
|
Name string `json:"name"`
|
||||||
|
MimeType string `json:"mimeType"`
|
||||||
|
ModifiedTime *time.Time `json:"modifiedTime"`
|
||||||
|
Size string `json:"size"`
|
||||||
|
ThumbnailLink string `json:"thumbnailLink"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (f File) GetSize() uint64 {
|
||||||
|
if f.GetType() == conf.FOLDER {
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
size, _ := strconv.ParseUint(f.Size, 10, 64)
|
||||||
|
return size
|
||||||
|
}
|
||||||
|
|
||||||
|
func (f File) GetName() string {
|
||||||
|
return f.Name
|
||||||
|
}
|
||||||
|
|
||||||
|
func (f File) GetType() int {
|
||||||
|
mimeType := f.MimeType
|
||||||
|
if mimeType == "application/vnd.google-apps.folder" || mimeType == "application/vnd.google-apps.shortcut" {
|
||||||
|
return conf.FOLDER
|
||||||
|
}
|
||||||
|
return utils.GetFileType(path.Ext(f.Name))
|
||||||
|
}
|
@ -2,28 +2,15 @@ package lanzou
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
"github.com/Xhofe/alist/conf"
|
|
||||||
"github.com/Xhofe/alist/drivers/base"
|
"github.com/Xhofe/alist/drivers/base"
|
||||||
"github.com/Xhofe/alist/model"
|
"github.com/Xhofe/alist/model"
|
||||||
"github.com/Xhofe/alist/utils"
|
|
||||||
log "github.com/sirupsen/logrus"
|
log "github.com/sirupsen/logrus"
|
||||||
"net/url"
|
"net/url"
|
||||||
"path/filepath"
|
|
||||||
"regexp"
|
"regexp"
|
||||||
"strconv"
|
"strconv"
|
||||||
"time"
|
"time"
|
||||||
)
|
)
|
||||||
|
|
||||||
type LanZouFile struct {
|
|
||||||
Name string `json:"name"`
|
|
||||||
NameAll string `json:"name_all"`
|
|
||||||
Id string `json:"id"`
|
|
||||||
FolId string `json:"fol_id"`
|
|
||||||
Size string `json:"size"`
|
|
||||||
Time string `json:"time"`
|
|
||||||
Folder bool
|
|
||||||
}
|
|
||||||
|
|
||||||
func (driver *Lanzou) FormatFile(file *LanZouFile) *model.File {
|
func (driver *Lanzou) FormatFile(file *LanZouFile) *model.File {
|
||||||
now := time.Now()
|
now := time.Now()
|
||||||
f := &model.File{
|
f := &model.File{
|
||||||
@ -35,12 +22,11 @@ func (driver *Lanzou) FormatFile(file *LanZouFile) *model.File {
|
|||||||
UpdatedAt: &now,
|
UpdatedAt: &now,
|
||||||
}
|
}
|
||||||
if file.Folder {
|
if file.Folder {
|
||||||
f.Type = conf.FOLDER
|
|
||||||
f.Id = file.FolId
|
f.Id = file.FolId
|
||||||
} else {
|
} else {
|
||||||
f.Name = file.NameAll
|
f.Name = file.NameAll
|
||||||
f.Type = utils.GetFileType(filepath.Ext(file.NameAll))
|
|
||||||
}
|
}
|
||||||
|
f.Type = file.GetType()
|
||||||
return f
|
return f
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -1,5 +1,39 @@
|
|||||||
package lanzou
|
package lanzou
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/Xhofe/alist/conf"
|
||||||
|
"github.com/Xhofe/alist/utils"
|
||||||
|
"path"
|
||||||
|
)
|
||||||
|
|
||||||
|
type LanZouFile struct {
|
||||||
|
Name string `json:"name"`
|
||||||
|
NameAll string `json:"name_all"`
|
||||||
|
Id string `json:"id"`
|
||||||
|
FolId string `json:"fol_id"`
|
||||||
|
Size string `json:"size"`
|
||||||
|
Time string `json:"time"`
|
||||||
|
Folder bool
|
||||||
|
}
|
||||||
|
|
||||||
|
func (f LanZouFile) GetSize() uint64 {
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
|
||||||
|
func (f LanZouFile) GetName() string {
|
||||||
|
if f.Folder {
|
||||||
|
return f.Name
|
||||||
|
}
|
||||||
|
return f.NameAll
|
||||||
|
}
|
||||||
|
|
||||||
|
func (f LanZouFile) GetType() int {
|
||||||
|
if f.Folder {
|
||||||
|
return conf.FOLDER
|
||||||
|
}
|
||||||
|
return utils.GetFileType(path.Ext(f.NameAll))
|
||||||
|
}
|
||||||
|
|
||||||
type DownPageResp struct {
|
type DownPageResp struct {
|
||||||
Zt int `json:"zt"`
|
Zt int `json:"zt"`
|
||||||
Info struct {
|
Info struct {
|
||||||
|
@ -71,7 +71,6 @@ func (driver Native) File(path string, account *model.Account) (*model.File, err
|
|||||||
time := f.ModTime()
|
time := f.ModTime()
|
||||||
file := &model.File{
|
file := &model.File{
|
||||||
Name: f.Name(),
|
Name: f.Name(),
|
||||||
Size: f.Size(),
|
|
||||||
UpdatedAt: &time,
|
UpdatedAt: &time,
|
||||||
Driver: driver.Config().Name,
|
Driver: driver.Config().Name,
|
||||||
}
|
}
|
||||||
@ -79,6 +78,7 @@ func (driver Native) File(path string, account *model.Account) (*model.File, err
|
|||||||
file.Type = conf.FOLDER
|
file.Type = conf.FOLDER
|
||||||
} else {
|
} else {
|
||||||
file.Type = utils.GetFileType(filepath.Ext(f.Name()))
|
file.Type = utils.GetFileType(filepath.Ext(f.Name()))
|
||||||
|
file.Size = f.Size()
|
||||||
}
|
}
|
||||||
return file, nil
|
return file, nil
|
||||||
}
|
}
|
||||||
@ -103,7 +103,6 @@ func (driver Native) Files(path string, account *model.Account) ([]model.File, e
|
|||||||
time := f.ModTime()
|
time := f.ModTime()
|
||||||
file := model.File{
|
file := model.File{
|
||||||
Name: f.Name(),
|
Name: f.Name(),
|
||||||
Size: f.Size(),
|
|
||||||
Type: 0,
|
Type: 0,
|
||||||
UpdatedAt: &time,
|
UpdatedAt: &time,
|
||||||
Driver: driver.Config().Name,
|
Driver: driver.Config().Name,
|
||||||
@ -112,9 +111,14 @@ func (driver Native) Files(path string, account *model.Account) ([]model.File, e
|
|||||||
file.Type = conf.FOLDER
|
file.Type = conf.FOLDER
|
||||||
} else {
|
} else {
|
||||||
file.Type = utils.GetFileType(filepath.Ext(f.Name()))
|
file.Type = utils.GetFileType(filepath.Ext(f.Name()))
|
||||||
|
file.Size = f.Size()
|
||||||
}
|
}
|
||||||
files = append(files, file)
|
files = append(files, file)
|
||||||
}
|
}
|
||||||
|
_, err = base.GetCache(path, account)
|
||||||
|
if len(files) != 0 && err != nil {
|
||||||
|
_ = base.SetCache(path, files, account)
|
||||||
|
}
|
||||||
return files, nil
|
return files, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -53,6 +53,7 @@ func (driver Onedrive) Items() []base.Item {
|
|||||||
Label: "redirect uri",
|
Label: "redirect uri",
|
||||||
Type: base.TypeString,
|
Type: base.TypeString,
|
||||||
Required: true,
|
Required: true,
|
||||||
|
Default: "https://tool.nn.ci/onedrive/callback",
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
Name: "refresh_token",
|
Name: "refresh_token",
|
||||||
|
@ -138,6 +138,7 @@ func (driver Quark) Link(args base.Args, account *model.Account) (*base.Link, er
|
|||||||
Url: resp.Data[0].DownloadUrl,
|
Url: resp.Data[0].DownloadUrl,
|
||||||
Headers: []base.Header{
|
Headers: []base.Header{
|
||||||
{Name: "Cookie", Value: account.AccessToken},
|
{Name: "Cookie", Value: account.AccessToken},
|
||||||
|
{Name: "Referer", Value: "https://pan.quark.cn"},
|
||||||
},
|
},
|
||||||
}, nil
|
}, nil
|
||||||
}
|
}
|
||||||
|
@ -103,7 +103,7 @@ func (driver Quark) GetFiles(parent string, account *model.Account) ([]model.Fil
|
|||||||
for _, f := range resp.Data.List {
|
for _, f := range resp.Data.List {
|
||||||
files = append(files, *driver.formatFile(&f))
|
files = append(files, *driver.formatFile(&f))
|
||||||
}
|
}
|
||||||
if page*size >= resp.Metadata.Count {
|
if page*size >= resp.Metadata.Total {
|
||||||
break
|
break
|
||||||
}
|
}
|
||||||
page++
|
page++
|
||||||
|
220
drivers/sftp/driver.go
Normal file
220
drivers/sftp/driver.go
Normal file
@ -0,0 +1,220 @@
|
|||||||
|
package template
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/Xhofe/alist/conf"
|
||||||
|
"github.com/Xhofe/alist/drivers/base"
|
||||||
|
"github.com/Xhofe/alist/model"
|
||||||
|
"github.com/Xhofe/alist/utils"
|
||||||
|
"io"
|
||||||
|
"path"
|
||||||
|
"path/filepath"
|
||||||
|
)
|
||||||
|
|
||||||
|
type SFTP struct {
|
||||||
|
}
|
||||||
|
|
||||||
|
func (driver SFTP) Config() base.DriverConfig {
|
||||||
|
return base.DriverConfig{
|
||||||
|
Name: "SFTP",
|
||||||
|
OnlyProxy: true,
|
||||||
|
OnlyLocal: true,
|
||||||
|
LocalSort: true,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (driver SFTP) Items() []base.Item {
|
||||||
|
// TODO fill need info
|
||||||
|
return []base.Item{
|
||||||
|
{
|
||||||
|
Name: "site_url",
|
||||||
|
Label: "ip/host",
|
||||||
|
Type: base.TypeString,
|
||||||
|
Required: true,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
Name: "limit",
|
||||||
|
Label: "port",
|
||||||
|
Type: base.TypeNumber,
|
||||||
|
Required: true,
|
||||||
|
Default: "22",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
Name: "username",
|
||||||
|
Label: "username",
|
||||||
|
Type: base.TypeString,
|
||||||
|
Required: true,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
Name: "password",
|
||||||
|
Label: "password",
|
||||||
|
Type: base.TypeString,
|
||||||
|
Required: true,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
Name: "root_folder",
|
||||||
|
Label: "root folder path",
|
||||||
|
Type: base.TypeString,
|
||||||
|
Default: "/",
|
||||||
|
Required: true,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (driver SFTP) Save(account *model.Account, old *model.Account) error {
|
||||||
|
if old != nil {
|
||||||
|
clientsMap.Lock()
|
||||||
|
defer clientsMap.Unlock()
|
||||||
|
delete(clientsMap.clients, old.Name)
|
||||||
|
}
|
||||||
|
if account == nil {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
_, err := GetClient(account)
|
||||||
|
if err != nil {
|
||||||
|
account.Status = err.Error()
|
||||||
|
} else {
|
||||||
|
account.Status = "work"
|
||||||
|
}
|
||||||
|
_ = model.SaveAccount(account)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
func (driver SFTP) File(path string, account *model.Account) (*model.File, error) {
|
||||||
|
path = utils.ParsePath(path)
|
||||||
|
if path == "/" {
|
||||||
|
return &model.File{
|
||||||
|
Id: account.RootFolder,
|
||||||
|
Name: account.Name,
|
||||||
|
Size: 0,
|
||||||
|
Type: conf.FOLDER,
|
||||||
|
Driver: driver.Config().Name,
|
||||||
|
UpdatedAt: account.UpdatedAt,
|
||||||
|
}, nil
|
||||||
|
}
|
||||||
|
dir, name := filepath.Split(path)
|
||||||
|
files, err := driver.Files(dir, account)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
for _, file := range files {
|
||||||
|
if file.Name == name {
|
||||||
|
return &file, nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return nil, base.ErrPathNotFound
|
||||||
|
}
|
||||||
|
|
||||||
|
func (driver SFTP) Files(path string, account *model.Account) ([]model.File, error) {
|
||||||
|
path = utils.ParsePath(path)
|
||||||
|
remotePath := utils.Join(account.RootFolder, path)
|
||||||
|
cache, err := base.GetCache(path, account)
|
||||||
|
if err == nil {
|
||||||
|
files, _ := cache.([]model.File)
|
||||||
|
return files, nil
|
||||||
|
}
|
||||||
|
client, err := GetClient(account)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
var files []model.File
|
||||||
|
rawFiles, err := client.Files(remotePath)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
for i := 0; i < len(rawFiles); i++ {
|
||||||
|
files = append(files, driver.formatFile(rawFiles[i]))
|
||||||
|
}
|
||||||
|
if len(files) > 0 {
|
||||||
|
_ = base.SetCache(path, files, account)
|
||||||
|
}
|
||||||
|
return files, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (driver SFTP) Link(args base.Args, account *model.Account) (*base.Link, error) {
|
||||||
|
client, err := GetClient(account)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
remoteFileName := utils.Join(account.RootFolder, args.Path)
|
||||||
|
remoteFile, err := client.Open(remoteFileName)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return &base.Link{
|
||||||
|
Data: remoteFile,
|
||||||
|
}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (driver SFTP) Path(path string, account *model.Account) (*model.File, []model.File, error) {
|
||||||
|
path = utils.ParsePath(path)
|
||||||
|
file, err := driver.File(path, account)
|
||||||
|
if err != nil {
|
||||||
|
return nil, nil, err
|
||||||
|
}
|
||||||
|
if !file.IsDir() {
|
||||||
|
return file, nil, nil
|
||||||
|
}
|
||||||
|
files, err := driver.Files(path, account)
|
||||||
|
if err != nil {
|
||||||
|
return nil, nil, err
|
||||||
|
}
|
||||||
|
return nil, files, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (driver SFTP) Preview(path string, account *model.Account) (interface{}, error) {
|
||||||
|
//TODO preview interface if driver support
|
||||||
|
return nil, base.ErrNotImplement
|
||||||
|
}
|
||||||
|
|
||||||
|
func (driver SFTP) MakeDir(path string, account *model.Account) error {
|
||||||
|
client, err := GetClient(account)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
return client.MkdirAll(utils.Join(account.RootFolder, path))
|
||||||
|
}
|
||||||
|
|
||||||
|
func (driver SFTP) Move(src string, dst string, account *model.Account) error {
|
||||||
|
return driver.Rename(src, dst, account)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (driver SFTP) Rename(src string, dst string, account *model.Account) error {
|
||||||
|
client, err := GetClient(account)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
return client.Rename(utils.Join(account.RootFolder, src), utils.Join(account.RootFolder, dst))
|
||||||
|
}
|
||||||
|
|
||||||
|
func (driver SFTP) Copy(src string, dst string, account *model.Account) error {
|
||||||
|
return base.ErrNotSupport
|
||||||
|
}
|
||||||
|
|
||||||
|
func (driver SFTP) Delete(path string, account *model.Account) error {
|
||||||
|
client, err := GetClient(account)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
return client.Remove(utils.Join(account.RootFolder, path))
|
||||||
|
}
|
||||||
|
|
||||||
|
func (driver SFTP) Upload(file *model.FileStream, account *model.Account) error {
|
||||||
|
if file == nil {
|
||||||
|
return base.ErrEmptyFile
|
||||||
|
}
|
||||||
|
client, err := GetClient(account)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
dstFile, err := client.Create(path.Join(account.RootFolder, file.ParentPath, file.Name))
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
defer func() {
|
||||||
|
_ = dstFile.Close()
|
||||||
|
}()
|
||||||
|
_, err = io.Copy(dstFile, file)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
var _ base.Driver = (*SFTP)(nil)
|
110
drivers/sftp/sftp.go
Normal file
110
drivers/sftp/sftp.go
Normal file
@ -0,0 +1,110 @@
|
|||||||
|
package template
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"github.com/Xhofe/alist/conf"
|
||||||
|
"github.com/Xhofe/alist/drivers/base"
|
||||||
|
"github.com/Xhofe/alist/model"
|
||||||
|
"github.com/Xhofe/alist/utils"
|
||||||
|
"github.com/pkg/sftp"
|
||||||
|
"golang.org/x/crypto/ssh"
|
||||||
|
"os"
|
||||||
|
"path"
|
||||||
|
"sync"
|
||||||
|
)
|
||||||
|
|
||||||
|
var clientsMap = struct {
|
||||||
|
sync.Mutex
|
||||||
|
clients map[string]*Client
|
||||||
|
}{clients: make(map[string]*Client)}
|
||||||
|
|
||||||
|
func GetClient(account *model.Account) (*Client, error) {
|
||||||
|
clientsMap.Lock()
|
||||||
|
defer clientsMap.Unlock()
|
||||||
|
if v, ok := clientsMap.clients[account.Name]; ok {
|
||||||
|
return v, nil
|
||||||
|
}
|
||||||
|
conn, err := ssh.Dial("tcp", fmt.Sprintf("%s:%d", account.SiteUrl, account.Limit), &ssh.ClientConfig{
|
||||||
|
User: account.Username,
|
||||||
|
Auth: []ssh.AuthMethod{ssh.Password(account.Password)},
|
||||||
|
HostKeyCallback: ssh.InsecureIgnoreHostKey(),
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
client, err := sftp.NewClient(conn)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
c := &Client{client}
|
||||||
|
clientsMap.clients[account.Name] = c
|
||||||
|
return c, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
type Client struct {
|
||||||
|
*sftp.Client
|
||||||
|
}
|
||||||
|
|
||||||
|
func (client *Client) Files(remotePath string) ([]os.FileInfo, error) {
|
||||||
|
return client.ReadDir(remotePath)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (client *Client) Remove(remotePath string) error {
|
||||||
|
f, err := client.Stat(remotePath)
|
||||||
|
if err != nil {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
if f.IsDir() {
|
||||||
|
return client.removeDirectory(remotePath)
|
||||||
|
} else {
|
||||||
|
return client.removeFile(remotePath)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (client *Client) removeDirectory(remotePath string) error {
|
||||||
|
//打不开,说明要么文件路径错误了,要么是第一次部署
|
||||||
|
remoteFiles, err := client.ReadDir(remotePath)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
for _, backupDir := range remoteFiles {
|
||||||
|
remoteFilePath := path.Join(remotePath, backupDir.Name())
|
||||||
|
if backupDir.IsDir() {
|
||||||
|
err := client.removeDirectory(remoteFilePath)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
err := client.Remove(path.Join(remoteFilePath))
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return client.RemoveDirectory(remotePath)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (client *Client) removeFile(remotePath string) error {
|
||||||
|
return client.Remove(utils.Join(remotePath))
|
||||||
|
}
|
||||||
|
|
||||||
|
func (driver SFTP) formatFile(f os.FileInfo) model.File {
|
||||||
|
t := f.ModTime()
|
||||||
|
file := model.File{
|
||||||
|
//Id: f.Id,
|
||||||
|
Name: f.Name(),
|
||||||
|
Size: f.Size(),
|
||||||
|
Driver: driver.Config().Name,
|
||||||
|
UpdatedAt: &t,
|
||||||
|
}
|
||||||
|
if f.IsDir() {
|
||||||
|
file.Type = conf.FOLDER
|
||||||
|
} else {
|
||||||
|
file.Type = utils.GetFileType(path.Ext(f.Name()))
|
||||||
|
}
|
||||||
|
return file
|
||||||
|
}
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
base.RegisterDriver(&SFTP{})
|
||||||
|
}
|
18
drivers/sftp/types.go
Normal file
18
drivers/sftp/types.go
Normal file
@ -0,0 +1,18 @@
|
|||||||
|
package template
|
||||||
|
|
||||||
|
import "time"
|
||||||
|
|
||||||
|
// write all struct here
|
||||||
|
|
||||||
|
type Resp struct {
|
||||||
|
Code int `json:"code"`
|
||||||
|
Message string `json:"message"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type File struct {
|
||||||
|
Id string `json:"id"`
|
||||||
|
FileName string `json:"file_name"`
|
||||||
|
Size int64 `json:"size"`
|
||||||
|
File bool `json:"file"`
|
||||||
|
UpdatedAt *time.Time `json:"updated_at"`
|
||||||
|
}
|
3
drivers/sftp/util.go
Normal file
3
drivers/sftp/util.go
Normal file
@ -0,0 +1,3 @@
|
|||||||
|
package template
|
||||||
|
|
||||||
|
// write util func here, such as cal sign
|
@ -3,8 +3,10 @@ package webdav
|
|||||||
import (
|
import (
|
||||||
"github.com/Xhofe/alist/conf"
|
"github.com/Xhofe/alist/conf"
|
||||||
"github.com/Xhofe/alist/drivers/base"
|
"github.com/Xhofe/alist/drivers/base"
|
||||||
|
"github.com/Xhofe/alist/drivers/webdav/odrvcookie"
|
||||||
"github.com/Xhofe/alist/model"
|
"github.com/Xhofe/alist/model"
|
||||||
"github.com/Xhofe/alist/utils"
|
"github.com/Xhofe/alist/utils"
|
||||||
|
"net/http"
|
||||||
"path/filepath"
|
"path/filepath"
|
||||||
)
|
)
|
||||||
|
|
||||||
@ -40,6 +42,15 @@ func (driver WebDav) Items() []base.Item {
|
|||||||
Type: base.TypeString,
|
Type: base.TypeString,
|
||||||
Required: true,
|
Required: true,
|
||||||
},
|
},
|
||||||
|
{
|
||||||
|
Name: "internal_type",
|
||||||
|
Label: "vendor",
|
||||||
|
Type: base.TypeSelect,
|
||||||
|
Required: true,
|
||||||
|
Default: "other",
|
||||||
|
Values: "sharepoint,other",
|
||||||
|
Description: "webdav vendor",
|
||||||
|
},
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -47,9 +58,17 @@ func (driver WebDav) Save(account *model.Account, old *model.Account) error {
|
|||||||
if account == nil {
|
if account == nil {
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
account.Status = "work"
|
var err error
|
||||||
|
if isSharePoint(account) {
|
||||||
|
_, err = odrvcookie.GetCookie(account.Username, account.Password, account.SiteUrl)
|
||||||
|
}
|
||||||
|
if err != nil {
|
||||||
|
account.Status = err.Error()
|
||||||
|
} else {
|
||||||
|
account.Status = "work"
|
||||||
|
}
|
||||||
_ = model.SaveAccount(account)
|
_ = model.SaveAccount(account)
|
||||||
return nil
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
func (driver WebDav) File(path string, account *model.Account) (*model.File, error) {
|
func (driver WebDav) File(path string, account *model.Account) (*model.File, error) {
|
||||||
@ -114,11 +133,27 @@ func (driver WebDav) Files(path string, account *model.Account) ([]model.File, e
|
|||||||
func (driver WebDav) Link(args base.Args, account *model.Account) (*base.Link, error) {
|
func (driver WebDav) Link(args base.Args, account *model.Account) (*base.Link, error) {
|
||||||
path := args.Path
|
path := args.Path
|
||||||
c := driver.NewClient(account)
|
c := driver.NewClient(account)
|
||||||
reader, err := c.ReadStream(driver.WebDavPath(path))
|
callback := func(r *http.Request) {
|
||||||
|
if args.Header.Get("Range") != "" {
|
||||||
|
r.Header.Set("Range", args.Header.Get("Range"))
|
||||||
|
}
|
||||||
|
if args.Header.Get("If-Range") != "" {
|
||||||
|
r.Header.Set("If-Range", args.Header.Get("If-Range"))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
reader, header, err := c.ReadStream(driver.WebDavPath(path), callback)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
return &base.Link{Data: reader}, nil
|
link := &base.Link{Data: reader}
|
||||||
|
if header.Get("Content-Range") != "" {
|
||||||
|
link.Status = 206
|
||||||
|
link.Header = http.Header{
|
||||||
|
"Content-Range": header.Values("Content-Range"),
|
||||||
|
"Content-Length": header.Values("Content-Length"),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return link, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (driver WebDav) Path(path string, account *model.Account) (*model.File, []model.File, error) {
|
func (driver WebDav) Path(path string, account *model.Account) (*model.File, []model.File, error) {
|
||||||
@ -178,7 +213,11 @@ func (driver WebDav) Upload(file *model.FileStream, account *model.Account) erro
|
|||||||
}
|
}
|
||||||
c := driver.NewClient(account)
|
c := driver.NewClient(account)
|
||||||
path := utils.Join(file.ParentPath, file.Name)
|
path := utils.Join(file.ParentPath, file.Name)
|
||||||
err := c.WriteStream(driver.WebDavPath(path), file, 0644)
|
callback := func(r *http.Request) {
|
||||||
|
r.Header.Set("Content-Type", file.GetMIMEType())
|
||||||
|
r.ContentLength = int64(file.GetSize())
|
||||||
|
}
|
||||||
|
err := c.WriteStream(driver.WebDavPath(path), file, 0644, callback)
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
|
47
drivers/webdav/odrvcookie/cookie.go
Normal file
47
drivers/webdav/odrvcookie/cookie.go
Normal file
@ -0,0 +1,47 @@
|
|||||||
|
package odrvcookie
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/Xhofe/alist/utils/cookie"
|
||||||
|
log "github.com/sirupsen/logrus"
|
||||||
|
"net/http"
|
||||||
|
"sync"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
type SpCookie struct {
|
||||||
|
Cookie string
|
||||||
|
expire time.Time
|
||||||
|
}
|
||||||
|
|
||||||
|
func (sp SpCookie) IsExpire() bool {
|
||||||
|
return time.Now().After(sp.expire)
|
||||||
|
}
|
||||||
|
|
||||||
|
var cookiesMap = struct {
|
||||||
|
sync.Mutex
|
||||||
|
m map[string]*SpCookie
|
||||||
|
}{m: make(map[string]*SpCookie)}
|
||||||
|
|
||||||
|
func GetCookie(username, password, siteUrl string) (string, error) {
|
||||||
|
cookiesMap.Lock()
|
||||||
|
defer cookiesMap.Unlock()
|
||||||
|
spCookie, ok := cookiesMap.m[username]
|
||||||
|
if ok {
|
||||||
|
if !spCookie.IsExpire() {
|
||||||
|
log.Debugln("sp use old cookie.")
|
||||||
|
return spCookie.Cookie, nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
log.Debugln("fetch new cookie")
|
||||||
|
ca := New(username, password, siteUrl)
|
||||||
|
tokenConf, err := ca.Cookies()
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
spCookie = &SpCookie{
|
||||||
|
Cookie: cookie.ToString([]*http.Cookie{&tokenConf.RtFa, &tokenConf.FedAuth}),
|
||||||
|
expire: time.Now().Add(time.Hour * 12),
|
||||||
|
}
|
||||||
|
cookiesMap.m[username] = spCookie
|
||||||
|
return spCookie.Cookie, nil
|
||||||
|
}
|
206
drivers/webdav/odrvcookie/fetch.go
Normal file
206
drivers/webdav/odrvcookie/fetch.go
Normal file
@ -0,0 +1,206 @@
|
|||||||
|
// Package odrvcookie can fetch authentication cookies for a sharepoint webdav endpoint
|
||||||
|
package odrvcookie
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
"encoding/xml"
|
||||||
|
"fmt"
|
||||||
|
"html/template"
|
||||||
|
"net/http"
|
||||||
|
"net/http/cookiejar"
|
||||||
|
"net/url"
|
||||||
|
"strings"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"golang.org/x/net/publicsuffix"
|
||||||
|
)
|
||||||
|
|
||||||
|
// CookieAuth hold the authentication information
|
||||||
|
// These are username and password as well as the authentication endpoint
|
||||||
|
type CookieAuth struct {
|
||||||
|
user string
|
||||||
|
pass string
|
||||||
|
endpoint string
|
||||||
|
}
|
||||||
|
|
||||||
|
// CookieResponse contains the requested cookies
|
||||||
|
type CookieResponse struct {
|
||||||
|
RtFa http.Cookie
|
||||||
|
FedAuth http.Cookie
|
||||||
|
}
|
||||||
|
|
||||||
|
// SuccessResponse hold a response from the sharepoint webdav
|
||||||
|
type SuccessResponse struct {
|
||||||
|
XMLName xml.Name `xml:"Envelope"`
|
||||||
|
Succ SuccessResponseBody `xml:"Body"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// SuccessResponseBody is the body of a success response, it holds the token
|
||||||
|
type SuccessResponseBody struct {
|
||||||
|
XMLName xml.Name
|
||||||
|
Type string `xml:"RequestSecurityTokenResponse>TokenType"`
|
||||||
|
Created time.Time `xml:"RequestSecurityTokenResponse>Lifetime>Created"`
|
||||||
|
Expires time.Time `xml:"RequestSecurityTokenResponse>Lifetime>Expires"`
|
||||||
|
Token string `xml:"RequestSecurityTokenResponse>RequestedSecurityToken>BinarySecurityToken"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// reqString is a template that gets populated with the user data in order to retrieve a "BinarySecurityToken"
|
||||||
|
const reqString = `<s:Envelope xmlns:s="http://www.w3.org/2003/05/soap-envelope"
|
||||||
|
xmlns:a="http://www.w3.org/2005/08/addressing"
|
||||||
|
xmlns:u="http://docs.oasis-open.org/wss/2004/01/oasis-200401-wss-wssecurity-utility-1.0.xsd">
|
||||||
|
<s:Header>
|
||||||
|
<a:Action s:mustUnderstand="1">http://schemas.xmlsoap.org/ws/2005/02/trust/RST/Issue</a:Action>
|
||||||
|
<a:ReplyTo>
|
||||||
|
<a:Address>http://www.w3.org/2005/08/addressing/anonymous</a:Address>
|
||||||
|
</a:ReplyTo>
|
||||||
|
<a:To s:mustUnderstand="1">{{ .LoginUrl }}</a:To>
|
||||||
|
<o:Security s:mustUnderstand="1"
|
||||||
|
xmlns:o="http://docs.oasis-open.org/wss/2004/01/oasis-200401-wss-wssecurity-secext-1.0.xsd">
|
||||||
|
<o:UsernameToken>
|
||||||
|
<o:Username>{{ .Username }}</o:Username>
|
||||||
|
<o:Password>{{ .Password }}</o:Password>
|
||||||
|
</o:UsernameToken>
|
||||||
|
</o:Security>
|
||||||
|
</s:Header>
|
||||||
|
<s:Body>
|
||||||
|
<t:RequestSecurityToken xmlns:t="http://schemas.xmlsoap.org/ws/2005/02/trust">
|
||||||
|
<wsp:AppliesTo xmlns:wsp="http://schemas.xmlsoap.org/ws/2004/09/policy">
|
||||||
|
<a:EndpointReference>
|
||||||
|
<a:Address>{{ .Address }}</a:Address>
|
||||||
|
</a:EndpointReference>
|
||||||
|
</wsp:AppliesTo>
|
||||||
|
<t:KeyType>http://schemas.xmlsoap.org/ws/2005/05/identity/NoProofKey</t:KeyType>
|
||||||
|
<t:RequestType>http://schemas.xmlsoap.org/ws/2005/02/trust/Issue</t:RequestType>
|
||||||
|
<t:TokenType>urn:oasis:names:tc:SAML:1.0:assertion</t:TokenType>
|
||||||
|
</t:RequestSecurityToken>
|
||||||
|
</s:Body>
|
||||||
|
</s:Envelope>`
|
||||||
|
|
||||||
|
// New creates a new CookieAuth struct
|
||||||
|
func New(pUser, pPass, pEndpoint string) CookieAuth {
|
||||||
|
retStruct := CookieAuth{
|
||||||
|
user: pUser,
|
||||||
|
pass: pPass,
|
||||||
|
endpoint: pEndpoint,
|
||||||
|
}
|
||||||
|
|
||||||
|
return retStruct
|
||||||
|
}
|
||||||
|
|
||||||
|
// Cookies creates a CookieResponse. It fetches the auth token and then
|
||||||
|
// retrieves the Cookies
|
||||||
|
func (ca *CookieAuth) Cookies() (CookieResponse, error) {
|
||||||
|
spToken, err := ca.getSPToken()
|
||||||
|
if err != nil {
|
||||||
|
return CookieResponse{}, err
|
||||||
|
}
|
||||||
|
return ca.getSPCookie(spToken)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ca *CookieAuth) getSPCookie(conf *SuccessResponse) (CookieResponse, error) {
|
||||||
|
spRoot, err := url.Parse(ca.endpoint)
|
||||||
|
if err != nil {
|
||||||
|
return CookieResponse{}, err
|
||||||
|
}
|
||||||
|
|
||||||
|
u, err := url.Parse("https://" + spRoot.Host + "/_forms/default.aspx?wa=wsignin1.0")
|
||||||
|
if err != nil {
|
||||||
|
return CookieResponse{}, err
|
||||||
|
}
|
||||||
|
|
||||||
|
// To authenticate with davfs or anything else we need two cookies (rtFa and FedAuth)
|
||||||
|
// In order to get them we use the token we got earlier and a cookieJar
|
||||||
|
jar, err := cookiejar.New(&cookiejar.Options{PublicSuffixList: publicsuffix.List})
|
||||||
|
if err != nil {
|
||||||
|
return CookieResponse{}, err
|
||||||
|
}
|
||||||
|
|
||||||
|
client := &http.Client{
|
||||||
|
Jar: jar,
|
||||||
|
}
|
||||||
|
|
||||||
|
// Send the previously aquired Token as a Post parameter
|
||||||
|
if _, err = client.Post(u.String(), "text/xml", strings.NewReader(conf.Succ.Token)); err != nil {
|
||||||
|
return CookieResponse{}, err
|
||||||
|
}
|
||||||
|
|
||||||
|
cookieResponse := CookieResponse{}
|
||||||
|
for _, cookie := range jar.Cookies(u) {
|
||||||
|
if (cookie.Name == "rtFa") || (cookie.Name == "FedAuth") {
|
||||||
|
switch cookie.Name {
|
||||||
|
case "rtFa":
|
||||||
|
cookieResponse.RtFa = *cookie
|
||||||
|
case "FedAuth":
|
||||||
|
cookieResponse.FedAuth = *cookie
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return cookieResponse, err
|
||||||
|
}
|
||||||
|
|
||||||
|
var loginUrlsMap = map[string]string{
|
||||||
|
"com": "https://login.microsoftonline.com",
|
||||||
|
"cn": "https://login.chinacloudapi.cn",
|
||||||
|
"us": "https://login.microsoftonline.us",
|
||||||
|
"de": "https://login.microsoftonline.de",
|
||||||
|
}
|
||||||
|
|
||||||
|
func getLoginUrl(endpoint string) (string, error) {
|
||||||
|
spRoot, err := url.Parse(endpoint)
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
domains := strings.Split(spRoot.Host, ".")
|
||||||
|
tld := domains[len(domains)-1]
|
||||||
|
loginUrl, ok := loginUrlsMap[tld]
|
||||||
|
if !ok {
|
||||||
|
return "", fmt.Errorf("tld %s is not supported", tld)
|
||||||
|
}
|
||||||
|
return loginUrl + "/extSTS.srf", nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ca *CookieAuth) getSPToken() (*SuccessResponse, error) {
|
||||||
|
loginUrl, err := getLoginUrl(ca.endpoint)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
reqData := map[string]string{
|
||||||
|
"Username": ca.user,
|
||||||
|
"Password": ca.pass,
|
||||||
|
"Address": ca.endpoint,
|
||||||
|
"LoginUrl": loginUrl,
|
||||||
|
}
|
||||||
|
|
||||||
|
t := template.Must(template.New("authXML").Parse(reqString))
|
||||||
|
|
||||||
|
buf := &bytes.Buffer{}
|
||||||
|
if err := t.Execute(buf, reqData); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
// Execute the first request which gives us an auth token for the sharepoint service
|
||||||
|
// With this token we can authenticate on the login page and save the returned cookies
|
||||||
|
req, err := http.NewRequest("POST", loginUrl, buf)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
client := &http.Client{}
|
||||||
|
resp, err := client.Do(req)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
defer resp.Body.Close()
|
||||||
|
|
||||||
|
respBuf := bytes.Buffer{}
|
||||||
|
respBuf.ReadFrom(resp.Body)
|
||||||
|
s := respBuf.Bytes()
|
||||||
|
|
||||||
|
var conf SuccessResponse
|
||||||
|
err = xml.Unmarshal(s, &conf)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
return &conf, err
|
||||||
|
}
|
7
drivers/webdav/util.go
Normal file
7
drivers/webdav/util.go
Normal file
@ -0,0 +1,7 @@
|
|||||||
|
package webdav
|
||||||
|
|
||||||
|
import "github.com/Xhofe/alist/model"
|
||||||
|
|
||||||
|
func isSharePoint(account *model.Account) bool {
|
||||||
|
return account.InternalType == "sharepoint"
|
||||||
|
}
|
@ -2,14 +2,26 @@ package webdav
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"github.com/Xhofe/alist/drivers/base"
|
"github.com/Xhofe/alist/drivers/base"
|
||||||
|
"github.com/Xhofe/alist/drivers/webdav/odrvcookie"
|
||||||
"github.com/Xhofe/alist/model"
|
"github.com/Xhofe/alist/model"
|
||||||
|
"github.com/Xhofe/alist/pkg/gowebdav"
|
||||||
"github.com/Xhofe/alist/utils"
|
"github.com/Xhofe/alist/utils"
|
||||||
"github.com/studio-b12/gowebdav"
|
"net/http"
|
||||||
"strings"
|
"strings"
|
||||||
)
|
)
|
||||||
|
|
||||||
func (driver WebDav) NewClient(account *model.Account) *gowebdav.Client {
|
func (driver WebDav) NewClient(account *model.Account) *gowebdav.Client {
|
||||||
return gowebdav.NewClient(account.SiteUrl, account.Username, account.Password)
|
c := gowebdav.NewClient(account.SiteUrl, account.Username, account.Password)
|
||||||
|
if isSharePoint(account) {
|
||||||
|
cookie, err := odrvcookie.GetCookie(account.Username, account.Password, account.SiteUrl)
|
||||||
|
if err == nil {
|
||||||
|
c.SetInterceptor(func(method string, rq *http.Request) {
|
||||||
|
rq.Header.Del("Authorization")
|
||||||
|
rq.Header.Set("Cookie", cookie)
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return c
|
||||||
}
|
}
|
||||||
|
|
||||||
func (driver WebDav) WebDavPath(path string) string {
|
func (driver WebDav) WebDavPath(path string) string {
|
||||||
|
@ -245,12 +245,18 @@ func (s *State) Request(method string, url string, callback func(*resty.Request)
|
|||||||
log.Debug(res.String())
|
log.Debug(res.String())
|
||||||
|
|
||||||
var e Erron
|
var e Erron
|
||||||
utils.Json.Unmarshal(res.Body(), &e)
|
err = utils.Json.Unmarshal(res.Body(), &e)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
switch e.ErrorCode {
|
switch e.ErrorCode {
|
||||||
case 9:
|
case 9:
|
||||||
s.newCaptchaToken(getAction(method, url), nil, account)
|
_, err = s.newCaptchaToken(getAction(method, url), nil, account)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
fallthrough
|
fallthrough
|
||||||
case 4122, 4121:
|
case 4122, 4121: // Authorization expired
|
||||||
return s.Request(method, url, callback, account)
|
return s.Request(method, url, callback, account)
|
||||||
case 0:
|
case 0:
|
||||||
if res.StatusCode() == http.StatusOK {
|
if res.StatusCode() == http.StatusOK {
|
||||||
|
12
go.mod
12
go.mod
@ -1,6 +1,6 @@
|
|||||||
module github.com/Xhofe/alist
|
module github.com/Xhofe/alist
|
||||||
|
|
||||||
go 1.17
|
go 1.18
|
||||||
|
|
||||||
require (
|
require (
|
||||||
github.com/aws/aws-sdk-go v1.27.0
|
github.com/aws/aws-sdk-go v1.27.0
|
||||||
@ -13,9 +13,9 @@ require (
|
|||||||
github.com/jlaffaye/ftp v0.0.0-20211117213618-11820403398b
|
github.com/jlaffaye/ftp v0.0.0-20211117213618-11820403398b
|
||||||
github.com/json-iterator/go v1.1.12
|
github.com/json-iterator/go v1.1.12
|
||||||
github.com/patrickmn/go-cache v2.1.0+incompatible
|
github.com/patrickmn/go-cache v2.1.0+incompatible
|
||||||
|
github.com/pkg/sftp v1.13.4
|
||||||
github.com/robfig/cron/v3 v3.0.0
|
github.com/robfig/cron/v3 v3.0.0
|
||||||
github.com/sirupsen/logrus v1.8.1
|
github.com/sirupsen/logrus v1.8.1
|
||||||
github.com/studio-b12/gowebdav v0.0.0-20211109083228-3f8721cd4b6f
|
|
||||||
github.com/upyun/go-sdk/v3 v3.0.2
|
github.com/upyun/go-sdk/v3 v3.0.2
|
||||||
golang.org/x/text v0.3.7
|
golang.org/x/text v0.3.7
|
||||||
gorm.io/driver/mysql v1.3.2
|
gorm.io/driver/mysql v1.3.2
|
||||||
@ -24,6 +24,8 @@ require (
|
|||||||
gorm.io/gorm v1.23.1
|
gorm.io/gorm v1.23.1
|
||||||
)
|
)
|
||||||
|
|
||||||
|
require github.com/kr/fs v0.1.0 // indirect
|
||||||
|
|
||||||
require (
|
require (
|
||||||
github.com/baiyubin/aliyun-sts-go-sdk v0.0.0-20180326062324-cfa1a18b161f // indirect
|
github.com/baiyubin/aliyun-sts-go-sdk v0.0.0-20180326062324-cfa1a18b161f // indirect
|
||||||
github.com/fatih/color v1.13.0
|
github.com/fatih/color v1.13.0
|
||||||
@ -73,9 +75,9 @@ require (
|
|||||||
go.opentelemetry.io/otel v0.20.0 // indirect
|
go.opentelemetry.io/otel v0.20.0 // indirect
|
||||||
go.opentelemetry.io/otel/metric v0.20.0 // indirect
|
go.opentelemetry.io/otel/metric v0.20.0 // indirect
|
||||||
go.opentelemetry.io/otel/trace v0.20.0 // indirect
|
go.opentelemetry.io/otel/trace v0.20.0 // indirect
|
||||||
golang.org/x/crypto v0.0.0-20220214200702-86341886e292 // indirect
|
golang.org/x/crypto v0.0.0-20220331220935-ae2d96664a29
|
||||||
golang.org/x/net v0.0.0-20211209124913-491a49abca63 // indirect
|
golang.org/x/net v0.0.0-20211209124913-491a49abca63
|
||||||
golang.org/x/sys v0.0.0-20211023085530-d6a326fbbf70 // indirect
|
golang.org/x/sys v0.0.0-20220330033206-e17cdc41300f // indirect
|
||||||
google.golang.org/protobuf v1.27.1 // indirect
|
google.golang.org/protobuf v1.27.1 // indirect
|
||||||
gopkg.in/natefinch/lumberjack.v2 v2.0.0 // indirect
|
gopkg.in/natefinch/lumberjack.v2 v2.0.0 // indirect
|
||||||
gopkg.in/tomb.v2 v2.0.0-20161208151619-d5d1b5820637 // indirect
|
gopkg.in/tomb.v2 v2.0.0-20161208151619-d5d1b5820637 // indirect
|
||||||
|
23
go.sum
23
go.sum
@ -46,7 +46,6 @@ github.com/bradfitz/gomemcache v0.0.0-20190913173617-a41fca850d0b/go.mod h1:H0wQ
|
|||||||
github.com/caarlos0/env/v6 v6.9.1 h1:zOkkjM0F6ltnQ5eBX6IPI41UP/KDGEK7rRPwGCNos8k=
|
github.com/caarlos0/env/v6 v6.9.1 h1:zOkkjM0F6ltnQ5eBX6IPI41UP/KDGEK7rRPwGCNos8k=
|
||||||
github.com/caarlos0/env/v6 v6.9.1/go.mod h1:hvp/ryKXKipEkcuYjs9mI4bBCg+UI0Yhgm5Zu0ddvwc=
|
github.com/caarlos0/env/v6 v6.9.1/go.mod h1:hvp/ryKXKipEkcuYjs9mI4bBCg+UI0Yhgm5Zu0ddvwc=
|
||||||
github.com/casbin/casbin/v2 v2.1.2/go.mod h1:YcPU1XXisHhLzuxH9coDNf2FbKpjGlbCg3n9yuLkIJQ=
|
github.com/casbin/casbin/v2 v2.1.2/go.mod h1:YcPU1XXisHhLzuxH9coDNf2FbKpjGlbCg3n9yuLkIJQ=
|
||||||
github.com/cenkalti/backoff v2.2.1+incompatible h1:tNowT99t7UNflLxfYYSlKYsBpXdEet03Pg2g16Swow4=
|
|
||||||
github.com/cenkalti/backoff v2.2.1+incompatible/go.mod h1:90ReRw6GdpyfrHakVjL/QHaoyV4aDUVVkXQJJJ3NXXM=
|
github.com/cenkalti/backoff v2.2.1+incompatible/go.mod h1:90ReRw6GdpyfrHakVjL/QHaoyV4aDUVVkXQJJJ3NXXM=
|
||||||
github.com/cenkalti/backoff/v4 v4.1.0 h1:c8LkOFQTzuO0WBM/ae5HdGQuZPfPxp7lqBRwQRm4fSc=
|
github.com/cenkalti/backoff/v4 v4.1.0 h1:c8LkOFQTzuO0WBM/ae5HdGQuZPfPxp7lqBRwQRm4fSc=
|
||||||
github.com/cenkalti/backoff/v4 v4.1.0/go.mod h1:scbssz8iZGpm3xbr14ovlUdkxfGXNInqkPWOWmG2CLw=
|
github.com/cenkalti/backoff/v4 v4.1.0/go.mod h1:scbssz8iZGpm3xbr14ovlUdkxfGXNInqkPWOWmG2CLw=
|
||||||
@ -226,7 +225,6 @@ github.com/hpcloud/tail v1.0.0/go.mod h1:ab1qPbhIpdTxEkNHXyeSf5vhxWSCs/tWer42PpO
|
|||||||
github.com/hudl/fargo v1.3.0/go.mod h1:y3CKSmjA+wD2gak7sUSXTAoopbhU08POFhmITJgmKTg=
|
github.com/hudl/fargo v1.3.0/go.mod h1:y3CKSmjA+wD2gak7sUSXTAoopbhU08POFhmITJgmKTg=
|
||||||
github.com/inconshreveable/mousetrap v1.0.0/go.mod h1:PxqpIevigyE2G7u3NXJIT2ANytuPF1OarO4DADm73n8=
|
github.com/inconshreveable/mousetrap v1.0.0/go.mod h1:PxqpIevigyE2G7u3NXJIT2ANytuPF1OarO4DADm73n8=
|
||||||
github.com/influxdata/influxdb1-client v0.0.0-20191209144304-8bf82d3c094d/go.mod h1:qj24IKcXYK6Iy9ceXlo3Tc+vtHo9lIhSX5JddghvEPo=
|
github.com/influxdata/influxdb1-client v0.0.0-20191209144304-8bf82d3c094d/go.mod h1:qj24IKcXYK6Iy9ceXlo3Tc+vtHo9lIhSX5JddghvEPo=
|
||||||
github.com/jackc/chunkreader v1.0.0 h1:4s39bBR8ByfqH+DKm8rQA3E1LHZWB9XWcrz8fqaZbe0=
|
|
||||||
github.com/jackc/chunkreader v1.0.0/go.mod h1:RT6O25fNZIuasFJRyZ4R/Y2BbhasbmZXF9QQ7T3kePo=
|
github.com/jackc/chunkreader v1.0.0/go.mod h1:RT6O25fNZIuasFJRyZ4R/Y2BbhasbmZXF9QQ7T3kePo=
|
||||||
github.com/jackc/chunkreader/v2 v2.0.0/go.mod h1:odVSm741yZoC3dpHEUXIqA9tQRhFrgOHwnPIn9lDKlk=
|
github.com/jackc/chunkreader/v2 v2.0.0/go.mod h1:odVSm741yZoC3dpHEUXIqA9tQRhFrgOHwnPIn9lDKlk=
|
||||||
github.com/jackc/chunkreader/v2 v2.0.1 h1:i+RDz65UE+mmpjTfyz0MoVTnzeYxroil2G82ki7MGG8=
|
github.com/jackc/chunkreader/v2 v2.0.1 h1:i+RDz65UE+mmpjTfyz0MoVTnzeYxroil2G82ki7MGG8=
|
||||||
@ -248,7 +246,6 @@ github.com/jackc/pgmock v0.0.0-20210724152146-4ad1a8207f65 h1:DadwsjnMwFjfWc9y5W
|
|||||||
github.com/jackc/pgmock v0.0.0-20210724152146-4ad1a8207f65/go.mod h1:5R2h2EEX+qri8jOWMbJCtaPWkrrNc7OHwsp2TCqp7ak=
|
github.com/jackc/pgmock v0.0.0-20210724152146-4ad1a8207f65/go.mod h1:5R2h2EEX+qri8jOWMbJCtaPWkrrNc7OHwsp2TCqp7ak=
|
||||||
github.com/jackc/pgpassfile v1.0.0 h1:/6Hmqy13Ss2zCq62VdNG8tM1wchn8zjSGOBJ6icpsIM=
|
github.com/jackc/pgpassfile v1.0.0 h1:/6Hmqy13Ss2zCq62VdNG8tM1wchn8zjSGOBJ6icpsIM=
|
||||||
github.com/jackc/pgpassfile v1.0.0/go.mod h1:CEx0iS5ambNFdcRtxPj5JhEz+xB6uRky5eyVu/W2HEg=
|
github.com/jackc/pgpassfile v1.0.0/go.mod h1:CEx0iS5ambNFdcRtxPj5JhEz+xB6uRky5eyVu/W2HEg=
|
||||||
github.com/jackc/pgproto3 v1.1.0 h1:FYYE4yRw+AgI8wXIinMlNjBbp/UitDJwfj5LqqewP1A=
|
|
||||||
github.com/jackc/pgproto3 v1.1.0/go.mod h1:eR5FA3leWg7p9aeAqi37XOTgTIbkABlvcPB3E5rlc78=
|
github.com/jackc/pgproto3 v1.1.0/go.mod h1:eR5FA3leWg7p9aeAqi37XOTgTIbkABlvcPB3E5rlc78=
|
||||||
github.com/jackc/pgproto3/v2 v2.0.0-alpha1.0.20190420180111-c116219b62db/go.mod h1:bhq50y+xrl9n5mRYyCBFKkpRVTLYJVWeCc+mEAI3yXA=
|
github.com/jackc/pgproto3/v2 v2.0.0-alpha1.0.20190420180111-c116219b62db/go.mod h1:bhq50y+xrl9n5mRYyCBFKkpRVTLYJVWeCc+mEAI3yXA=
|
||||||
github.com/jackc/pgproto3/v2 v2.0.0-alpha1.0.20190609003834-432c2951c711/go.mod h1:uH0AWtUmuShn0bcesswc4aBTWGvw0cAxIJp+6OB//Wg=
|
github.com/jackc/pgproto3/v2 v2.0.0-alpha1.0.20190609003834-432c2951c711/go.mod h1:uH0AWtUmuShn0bcesswc4aBTWGvw0cAxIJp+6OB//Wg=
|
||||||
@ -307,6 +304,8 @@ github.com/kisielk/gotool v1.0.0/go.mod h1:XhKaO+MFFWcvkIS/tQcRk01m1F5IRFswLeQ+o
|
|||||||
github.com/konsorten/go-windows-terminal-sequences v1.0.1/go.mod h1:T0+1ngSBFLxvqU3pZ+m/2kptfBszLMUkC4ZK/EgS/cQ=
|
github.com/konsorten/go-windows-terminal-sequences v1.0.1/go.mod h1:T0+1ngSBFLxvqU3pZ+m/2kptfBszLMUkC4ZK/EgS/cQ=
|
||||||
github.com/konsorten/go-windows-terminal-sequences v1.0.2/go.mod h1:T0+1ngSBFLxvqU3pZ+m/2kptfBszLMUkC4ZK/EgS/cQ=
|
github.com/konsorten/go-windows-terminal-sequences v1.0.2/go.mod h1:T0+1ngSBFLxvqU3pZ+m/2kptfBszLMUkC4ZK/EgS/cQ=
|
||||||
github.com/konsorten/go-windows-terminal-sequences v1.0.3/go.mod h1:T0+1ngSBFLxvqU3pZ+m/2kptfBszLMUkC4ZK/EgS/cQ=
|
github.com/konsorten/go-windows-terminal-sequences v1.0.3/go.mod h1:T0+1ngSBFLxvqU3pZ+m/2kptfBszLMUkC4ZK/EgS/cQ=
|
||||||
|
github.com/kr/fs v0.1.0 h1:Jskdu9ieNAYnjxsi0LbQp1ulIKZV1LAFgK1tWhpZgl8=
|
||||||
|
github.com/kr/fs v0.1.0/go.mod h1:FFnZGqtBN9Gxj7eW1uZ42v5BccTP0vu6NEaFoC2HwRg=
|
||||||
github.com/kr/logfmt v0.0.0-20140226030751-b84e30acd515/go.mod h1:+0opPa2QZZtGFBFZlji/RkVcI2GknAs/DXo4wKdlNEc=
|
github.com/kr/logfmt v0.0.0-20140226030751-b84e30acd515/go.mod h1:+0opPa2QZZtGFBFZlji/RkVcI2GknAs/DXo4wKdlNEc=
|
||||||
github.com/kr/pretty v0.1.0/go.mod h1:dAy3ld7l9f0ibDNOQOHHMYYIIbhfbHSm3C4ZsoJORNo=
|
github.com/kr/pretty v0.1.0/go.mod h1:dAy3ld7l9f0ibDNOQOHHMYYIIbhfbHSm3C4ZsoJORNo=
|
||||||
github.com/kr/pretty v0.2.1/go.mod h1:ipq/a2n7PKx3OHsz4KJII5eveXtPO4qwEXGdVfWzfnI=
|
github.com/kr/pretty v0.2.1/go.mod h1:ipq/a2n7PKx3OHsz4KJII5eveXtPO4qwEXGdVfWzfnI=
|
||||||
@ -420,6 +419,8 @@ github.com/pkg/errors v0.8.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINE
|
|||||||
github.com/pkg/errors v0.9.1 h1:FEBLx1zS214owpjy7qsBeixbURkuhQAwrK5UwLGTwt4=
|
github.com/pkg/errors v0.9.1 h1:FEBLx1zS214owpjy7qsBeixbURkuhQAwrK5UwLGTwt4=
|
||||||
github.com/pkg/errors v0.9.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=
|
github.com/pkg/errors v0.9.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=
|
||||||
github.com/pkg/profile v1.2.1/go.mod h1:hJw3o1OdXxsrSjjVksARp5W95eeEaEfptyVZyv6JUPA=
|
github.com/pkg/profile v1.2.1/go.mod h1:hJw3o1OdXxsrSjjVksARp5W95eeEaEfptyVZyv6JUPA=
|
||||||
|
github.com/pkg/sftp v1.13.4 h1:Lb0RYJCmgUcBgZosfoi9Y9sbl6+LJgOIgk/2Y4YjMFg=
|
||||||
|
github.com/pkg/sftp v1.13.4/go.mod h1:LzqnAvaD5TWeNBsZpfKxSYn1MbjWwOsCIAFFJbpIsK8=
|
||||||
github.com/pmezard/go-difflib v0.0.0-20151028094244-d8ed2627bdf0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
|
github.com/pmezard/go-difflib v0.0.0-20151028094244-d8ed2627bdf0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
|
||||||
github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
|
github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
|
||||||
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
|
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
|
||||||
@ -505,11 +506,8 @@ github.com/stretchr/testify v1.5.1/go.mod h1:5W2xD1RspED5o8YsWQXVCued0rvSQ+mT+I5
|
|||||||
github.com/stretchr/testify v1.6.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
|
github.com/stretchr/testify v1.6.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
|
||||||
github.com/stretchr/testify v1.7.0 h1:nwc3DEeHmmLAfoZucVR881uASk0Mfjw8xYJ99tb5CcY=
|
github.com/stretchr/testify v1.7.0 h1:nwc3DEeHmmLAfoZucVR881uASk0Mfjw8xYJ99tb5CcY=
|
||||||
github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
|
github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
|
||||||
github.com/studio-b12/gowebdav v0.0.0-20211109083228-3f8721cd4b6f h1:L2NE7BXnSlSLoNYZ0lCwZDjdnYjCNYC71k9ClZUTFTs=
|
|
||||||
github.com/studio-b12/gowebdav v0.0.0-20211109083228-3f8721cd4b6f/go.mod h1:bHA7t77X/QFExdeAnDzK6vKM34kEZAcE1OX4MfiwjkE=
|
|
||||||
github.com/tmc/grpc-websocket-proxy v0.0.0-20170815181823-89b8d40f7ca8/go.mod h1:ncp9v5uamzpCO7NfCPTXjqaC+bZgJeR0sMTm6dMHP7U=
|
github.com/tmc/grpc-websocket-proxy v0.0.0-20170815181823-89b8d40f7ca8/go.mod h1:ncp9v5uamzpCO7NfCPTXjqaC+bZgJeR0sMTm6dMHP7U=
|
||||||
github.com/ugorji/go v1.1.7/go.mod h1:kZn38zHttfInRq0xu/PH0az30d+z6vm202qpg1oXVMw=
|
github.com/ugorji/go v1.1.7/go.mod h1:kZn38zHttfInRq0xu/PH0az30d+z6vm202qpg1oXVMw=
|
||||||
github.com/ugorji/go v1.2.6 h1:tGiWC9HENWE2tqYycIqFTNorMmFRVhNwCpDOpWqnk8E=
|
|
||||||
github.com/ugorji/go v1.2.6/go.mod h1:anCg0y61KIhDlPZmnH+so+RQbysYVyDko0IMgJv0Nn0=
|
github.com/ugorji/go v1.2.6/go.mod h1:anCg0y61KIhDlPZmnH+so+RQbysYVyDko0IMgJv0Nn0=
|
||||||
github.com/ugorji/go/codec v1.1.7/go.mod h1:Ax+UKWsSmolVDwsd+7N3ZtXu+yMGCf907BLYF3GoBXY=
|
github.com/ugorji/go/codec v1.1.7/go.mod h1:Ax+UKWsSmolVDwsd+7N3ZtXu+yMGCf907BLYF3GoBXY=
|
||||||
github.com/ugorji/go/codec v1.2.6 h1:7kbGefxLoDBuYXOms4yD7223OpNMMPNPZxXk5TvFcyQ=
|
github.com/ugorji/go/codec v1.2.6 h1:7kbGefxLoDBuYXOms4yD7223OpNMMPNPZxXk5TvFcyQ=
|
||||||
@ -556,11 +554,12 @@ golang.org/x/crypto v0.0.0-20190820162420-60c769a6c586/go.mod h1:yigFU9vqHzYiE8U
|
|||||||
golang.org/x/crypto v0.0.0-20191011191535-87dc89f01550/go.mod h1:yigFU9vqHzYiE8UmvKecakEJjdnWj3jj499lnFckfCI=
|
golang.org/x/crypto v0.0.0-20191011191535-87dc89f01550/go.mod h1:yigFU9vqHzYiE8UmvKecakEJjdnWj3jj499lnFckfCI=
|
||||||
golang.org/x/crypto v0.0.0-20200622213623-75b288015ac9/go.mod h1:LzIPMQfyMNhhGPhUkYOs5KpL4U8rLKemX1yGLhDgUto=
|
golang.org/x/crypto v0.0.0-20200622213623-75b288015ac9/go.mod h1:LzIPMQfyMNhhGPhUkYOs5KpL4U8rLKemX1yGLhDgUto=
|
||||||
golang.org/x/crypto v0.0.0-20201203163018-be400aefbc4c/go.mod h1:jdWPYTVW3xRLrWPugEBEK3UY2ZEsg3UU495nc5E+M+I=
|
golang.org/x/crypto v0.0.0-20201203163018-be400aefbc4c/go.mod h1:jdWPYTVW3xRLrWPugEBEK3UY2ZEsg3UU495nc5E+M+I=
|
||||||
|
golang.org/x/crypto v0.0.0-20210421170649-83a5a9bb288b/go.mod h1:T9bdIzuCu7OtxOm1hfPfRQxPLYneinmdGuTeoZ9dtd4=
|
||||||
golang.org/x/crypto v0.0.0-20210616213533-5ff15b29337e/go.mod h1:GvvjBRRGRdwPK5ydBHafDWAxML/pGHZbMvKqRZ5+Abc=
|
golang.org/x/crypto v0.0.0-20210616213533-5ff15b29337e/go.mod h1:GvvjBRRGRdwPK5ydBHafDWAxML/pGHZbMvKqRZ5+Abc=
|
||||||
golang.org/x/crypto v0.0.0-20210711020723-a769d52b0f97/go.mod h1:GvvjBRRGRdwPK5ydBHafDWAxML/pGHZbMvKqRZ5+Abc=
|
golang.org/x/crypto v0.0.0-20210711020723-a769d52b0f97/go.mod h1:GvvjBRRGRdwPK5ydBHafDWAxML/pGHZbMvKqRZ5+Abc=
|
||||||
golang.org/x/crypto v0.0.0-20210921155107-089bfa567519/go.mod h1:GvvjBRRGRdwPK5ydBHafDWAxML/pGHZbMvKqRZ5+Abc=
|
golang.org/x/crypto v0.0.0-20210921155107-089bfa567519/go.mod h1:GvvjBRRGRdwPK5ydBHafDWAxML/pGHZbMvKqRZ5+Abc=
|
||||||
golang.org/x/crypto v0.0.0-20220214200702-86341886e292 h1:f+lwQ+GtmgoY+A2YaQxlSOnDjXcQ7ZRLWOHbC6HtRqE=
|
golang.org/x/crypto v0.0.0-20220331220935-ae2d96664a29 h1:tkVvjkPTB7pnW3jnid7kNyAMPVWllTNOf/qKDze4p9o=
|
||||||
golang.org/x/crypto v0.0.0-20220214200702-86341886e292/go.mod h1:IxCIyHEi3zRg3s0A5j5BB6A9Jmi73HwBIUl50j+osU4=
|
golang.org/x/crypto v0.0.0-20220331220935-ae2d96664a29/go.mod h1:IxCIyHEi3zRg3s0A5j5BB6A9Jmi73HwBIUl50j+osU4=
|
||||||
golang.org/x/exp v0.0.0-20190121172915-509febef88a4/go.mod h1:CJ0aWSM057203Lf6IL+f9T1iT9GByDxfZKAQTCR3kQA=
|
golang.org/x/exp v0.0.0-20190121172915-509febef88a4/go.mod h1:CJ0aWSM057203Lf6IL+f9T1iT9GByDxfZKAQTCR3kQA=
|
||||||
golang.org/x/lint v0.0.0-20181026193005-c67002cb31c3/go.mod h1:UVdnD1Gm6xHRNCYTkRU2/jEulfH38KcIWyp/GAMgvoE=
|
golang.org/x/lint v0.0.0-20181026193005-c67002cb31c3/go.mod h1:UVdnD1Gm6xHRNCYTkRU2/jEulfH38KcIWyp/GAMgvoE=
|
||||||
golang.org/x/lint v0.0.0-20190227174305-5b3e6a55c961/go.mod h1:wehouNa3lNwaWXcvxsM5YxQ5yQlVC4a0KAMCusXpPoU=
|
golang.org/x/lint v0.0.0-20190227174305-5b3e6a55c961/go.mod h1:wehouNa3lNwaWXcvxsM5YxQ5yQlVC4a0KAMCusXpPoU=
|
||||||
@ -595,7 +594,6 @@ golang.org/x/net v0.0.0-20201021035429-f5854403a974/go.mod h1:sp8m0HH+o8qH0wwXwY
|
|||||||
golang.org/x/net v0.0.0-20201202161906-c7110b5ffcbb/go.mod h1:sp8m0HH+o8qH0wwXwYZr8TS3Oi6o0r6Gce1SSxlDquU=
|
golang.org/x/net v0.0.0-20201202161906-c7110b5ffcbb/go.mod h1:sp8m0HH+o8qH0wwXwYZr8TS3Oi6o0r6Gce1SSxlDquU=
|
||||||
golang.org/x/net v0.0.0-20210226172049-e18ecbb05110/go.mod h1:m0MpNAwzfU5UDzcl9v0D8zg8gWTRqZa9RBIspLL5mdg=
|
golang.org/x/net v0.0.0-20210226172049-e18ecbb05110/go.mod h1:m0MpNAwzfU5UDzcl9v0D8zg8gWTRqZa9RBIspLL5mdg=
|
||||||
golang.org/x/net v0.0.0-20210405180319-a5a99cb37ef4/go.mod h1:p54w0d4576C0XHj96bSt6lcn1PtDYWL6XObtHCRCNQM=
|
golang.org/x/net v0.0.0-20210405180319-a5a99cb37ef4/go.mod h1:p54w0d4576C0XHj96bSt6lcn1PtDYWL6XObtHCRCNQM=
|
||||||
golang.org/x/net v0.0.0-20211112202133-69e39bad7dc2/go.mod h1:9nx3DQGgdP8bBQD5qxJ1jj9UTztislL4KSBs9R2vV5Y=
|
|
||||||
golang.org/x/net v0.0.0-20211209124913-491a49abca63 h1:iocB37TsdFuN6IBRZ+ry36wrkoV51/tl5vOWqkcPGvY=
|
golang.org/x/net v0.0.0-20211209124913-491a49abca63 h1:iocB37TsdFuN6IBRZ+ry36wrkoV51/tl5vOWqkcPGvY=
|
||||||
golang.org/x/net v0.0.0-20211209124913-491a49abca63/go.mod h1:9nx3DQGgdP8bBQD5qxJ1jj9UTztislL4KSBs9R2vV5Y=
|
golang.org/x/net v0.0.0-20211209124913-491a49abca63/go.mod h1:9nx3DQGgdP8bBQD5qxJ1jj9UTztislL4KSBs9R2vV5Y=
|
||||||
golang.org/x/oauth2 v0.0.0-20180821212333-d2e6202438be/go.mod h1:N/0e6XlmueqKjAGxoOufVs8QHGRruUQn6yWY3a++T0U=
|
golang.org/x/oauth2 v0.0.0-20180821212333-d2e6202438be/go.mod h1:N/0e6XlmueqKjAGxoOufVs8QHGRruUQn6yWY3a++T0U=
|
||||||
@ -644,13 +642,14 @@ golang.org/x/sys v0.0.0-20210112080510-489259a85091/go.mod h1:h1NjWce9XRLGQEsW7w
|
|||||||
golang.org/x/sys v0.0.0-20210124154548-22da62e12c0c/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
|
golang.org/x/sys v0.0.0-20210124154548-22da62e12c0c/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
|
||||||
golang.org/x/sys v0.0.0-20210309074719-68d13333faf2/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
|
golang.org/x/sys v0.0.0-20210309074719-68d13333faf2/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
|
||||||
golang.org/x/sys v0.0.0-20210330210617-4fbd30eecc44/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
|
golang.org/x/sys v0.0.0-20210330210617-4fbd30eecc44/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
|
||||||
golang.org/x/sys v0.0.0-20210423082822-04245dca01da/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
|
golang.org/x/sys v0.0.0-20210423185535-09eb48e85fd7/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
|
||||||
golang.org/x/sys v0.0.0-20210615035016-665e8c7367d1/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
golang.org/x/sys v0.0.0-20210615035016-665e8c7367d1/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||||
golang.org/x/sys v0.0.0-20210630005230-0f9fa26af87c/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
golang.org/x/sys v0.0.0-20210630005230-0f9fa26af87c/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||||
golang.org/x/sys v0.0.0-20210806184541-e5e7981a1069/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
golang.org/x/sys v0.0.0-20210806184541-e5e7981a1069/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||||
golang.org/x/sys v0.0.0-20211023085530-d6a326fbbf70 h1:SeSEfdIxyvwGJliREIJhRPPXvW6sDlLT+UQ3B0hD0NA=
|
golang.org/x/sys v0.0.0-20220330033206-e17cdc41300f h1:rlezHXNlxYWvBCzNses9Dlc7nGFaNMJeqLolcmQSSZY=
|
||||||
golang.org/x/sys v0.0.0-20211023085530-d6a326fbbf70/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
golang.org/x/sys v0.0.0-20220330033206-e17cdc41300f/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||||
golang.org/x/term v0.0.0-20201117132131-f5c789dd3221/go.mod h1:Nr5EML6q2oocZ2LXRh80K7BxOlk5/8JxuGnuhpl+muw=
|
golang.org/x/term v0.0.0-20201117132131-f5c789dd3221/go.mod h1:Nr5EML6q2oocZ2LXRh80K7BxOlk5/8JxuGnuhpl+muw=
|
||||||
|
golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1 h1:v+OssWQX+hTHEmOBgwxdZxK4zHq3yOs8F9J7mk0PY8E=
|
||||||
golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo=
|
golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo=
|
||||||
golang.org/x/text v0.0.0-20160726164857-2910a502d2bf/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=
|
golang.org/x/text v0.0.0-20160726164857-2910a502d2bf/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=
|
||||||
golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=
|
golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=
|
||||||
|
@ -207,6 +207,9 @@ func GetAccountsByPath(path string) []Account {
|
|||||||
if bIndex != -1 {
|
if bIndex != -1 {
|
||||||
name = name[:bIndex]
|
name = name[:bIndex]
|
||||||
}
|
}
|
||||||
|
if name == "/" {
|
||||||
|
name = ""
|
||||||
|
}
|
||||||
// 不是这个账号
|
// 不是这个账号
|
||||||
if path != name && !strings.HasPrefix(path, name+"/") {
|
if path != name && !strings.HasPrefix(path, name+"/") {
|
||||||
continue
|
continue
|
||||||
@ -253,6 +256,9 @@ func GetAccountFilesByPath(prefix string) []File {
|
|||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
full := utils.ParsePath(v.Name)
|
full := utils.ParsePath(v.Name)
|
||||||
|
if len(full) <= len(prefix) {
|
||||||
|
continue
|
||||||
|
}
|
||||||
// 不是以prefix为前缀
|
// 不是以prefix为前缀
|
||||||
if !strings.HasPrefix(full, prefix+"/") && prefix != "/" {
|
if !strings.HasPrefix(full, prefix+"/") && prefix != "/" {
|
||||||
continue
|
continue
|
||||||
|
@ -56,7 +56,7 @@ func ExtractFolder(files []File, account *Account) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
front := account.ExtractFolder == "front"
|
front := account.ExtractFolder == "front"
|
||||||
sort.Slice(files, func(i, j int) bool {
|
sort.SliceStable(files, func(i, j int) bool {
|
||||||
if files[i].IsDir() || files[j].IsDir() {
|
if files[i].IsDir() || files[j].IsDir() {
|
||||||
if !files[i].IsDir() {
|
if !files[i].IsDir() {
|
||||||
return !front
|
return !front
|
||||||
@ -84,3 +84,7 @@ func (f File) ModTime() time.Time {
|
|||||||
func (f File) IsDir() bool {
|
func (f File) IsDir() bool {
|
||||||
return f.Type == conf.FOLDER
|
return f.Type == conf.FOLDER
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (f File) GetType() int {
|
||||||
|
return f.Type
|
||||||
|
}
|
||||||
|
@ -12,6 +12,7 @@ type Meta struct {
|
|||||||
Hide string `json:"hide"`
|
Hide string `json:"hide"`
|
||||||
Upload bool `json:"upload"`
|
Upload bool `json:"upload"`
|
||||||
OnlyShows string `json:"only_shows"`
|
OnlyShows string `json:"only_shows"`
|
||||||
|
Readme string `json:"readme"`
|
||||||
}
|
}
|
||||||
|
|
||||||
func GetMetaByPath(path string) (*Meta, error) {
|
func GetMetaByPath(path string) (*Meta, error) {
|
||||||
|
35
model/search_file.go
Normal file
35
model/search_file.go
Normal file
@ -0,0 +1,35 @@
|
|||||||
|
package model
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"github.com/Xhofe/alist/conf"
|
||||||
|
)
|
||||||
|
|
||||||
|
type ISearchFile interface {
|
||||||
|
GetName() string
|
||||||
|
GetSize() uint64
|
||||||
|
GetType() int
|
||||||
|
}
|
||||||
|
|
||||||
|
type SearchFile struct {
|
||||||
|
Path string `json:"path" gorm:"index"`
|
||||||
|
Name string `json:"name"`
|
||||||
|
Size uint64 `json:"size"`
|
||||||
|
Type int `json:"type"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func CreateSearchFiles(files []SearchFile) error {
|
||||||
|
return conf.DB.Create(files).Error
|
||||||
|
}
|
||||||
|
|
||||||
|
func DeleteSearchFilesByPath(path string) error {
|
||||||
|
return conf.DB.Where(fmt.Sprintf("%s = ?", columnName("path")), path).Delete(&SearchFile{}).Error
|
||||||
|
}
|
||||||
|
|
||||||
|
func SearchByNameAndPath(path, keyword string) ([]SearchFile, error) {
|
||||||
|
var files []SearchFile
|
||||||
|
if err := conf.DB.Where(fmt.Sprintf("%s LIKE ? AND %s LIKE ?", columnName("path"), columnName("name")), fmt.Sprintf("%s%%", path), fmt.Sprintf("%%%s%%", keyword)).Find(&files).Error; err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return files, nil
|
||||||
|
}
|
21
pkg/gowebdav/.gitignore
vendored
Normal file
21
pkg/gowebdav/.gitignore
vendored
Normal file
@ -0,0 +1,21 @@
|
|||||||
|
# Folders to ignore
|
||||||
|
/src
|
||||||
|
/bin
|
||||||
|
/pkg
|
||||||
|
/gowebdav
|
||||||
|
/.idea
|
||||||
|
|
||||||
|
# Binaries for programs and plugins
|
||||||
|
*.exe
|
||||||
|
*.exe~
|
||||||
|
*.dll
|
||||||
|
*.so
|
||||||
|
*.dylib
|
||||||
|
|
||||||
|
# Test binary, build with `go test -c`
|
||||||
|
*.test
|
||||||
|
|
||||||
|
# Output of the go coverage tool, specifically when used with LiteIDE
|
||||||
|
*.out
|
||||||
|
|
||||||
|
.vscode/
|
10
pkg/gowebdav/.travis.yml
Normal file
10
pkg/gowebdav/.travis.yml
Normal file
@ -0,0 +1,10 @@
|
|||||||
|
language: go
|
||||||
|
|
||||||
|
go:
|
||||||
|
- "1.x"
|
||||||
|
|
||||||
|
install:
|
||||||
|
- go get ./...
|
||||||
|
|
||||||
|
script:
|
||||||
|
- go test -v --short ./...
|
27
pkg/gowebdav/LICENSE
Normal file
27
pkg/gowebdav/LICENSE
Normal file
@ -0,0 +1,27 @@
|
|||||||
|
Copyright (c) 2014, Studio B12 GmbH
|
||||||
|
All rights reserved.
|
||||||
|
|
||||||
|
Redistribution and use in source and binary forms, with or without
|
||||||
|
modification, are permitted provided that the following conditions are met:
|
||||||
|
|
||||||
|
1. Redistributions of source code must retain the above copyright notice, this
|
||||||
|
list of conditions and the following disclaimer.
|
||||||
|
|
||||||
|
2. Redistributions in binary form must reproduce the above copyright notice,
|
||||||
|
this list of conditions and the following disclaimer in the documentation
|
||||||
|
and/or other materials provided with the distribution.
|
||||||
|
|
||||||
|
3. Neither the name of the copyright holder nor the names of its contributors
|
||||||
|
may be used to endorse or promote products derived from this software without
|
||||||
|
specific prior written permission.
|
||||||
|
|
||||||
|
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
|
||||||
|
ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
|
||||||
|
WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
|
||||||
|
DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
|
||||||
|
FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
|
||||||
|
DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
|
||||||
|
SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
|
||||||
|
CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
|
||||||
|
OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
||||||
|
OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
33
pkg/gowebdav/Makefile
Normal file
33
pkg/gowebdav/Makefile
Normal file
@ -0,0 +1,33 @@
|
|||||||
|
BIN := gowebdav
|
||||||
|
SRC := $(wildcard *.go) cmd/gowebdav/main.go
|
||||||
|
|
||||||
|
all: test cmd
|
||||||
|
|
||||||
|
cmd: ${BIN}
|
||||||
|
|
||||||
|
${BIN}: ${SRC}
|
||||||
|
go build -o $@ ./cmd/gowebdav
|
||||||
|
|
||||||
|
test:
|
||||||
|
go test -v --short ./...
|
||||||
|
|
||||||
|
api:
|
||||||
|
@sed '/^## API$$/,$$d' -i README.md
|
||||||
|
@echo '## API' >> README.md
|
||||||
|
@godoc2md github.com/studio-b12/gowebdav | sed '/^$$/N;/^\n$$/D' |\
|
||||||
|
sed '2d' |\
|
||||||
|
sed 's/\/src\/github.com\/studio-b12\/gowebdav\//https:\/\/github.com\/studio-b12\/gowebdav\/blob\/master\//g' |\
|
||||||
|
sed 's/\/src\/target\//https:\/\/github.com\/studio-b12\/gowebdav\/blob\/master\//g' |\
|
||||||
|
sed 's/^#/##/g' >> README.md
|
||||||
|
|
||||||
|
check:
|
||||||
|
gofmt -w -s $(SRC)
|
||||||
|
@echo
|
||||||
|
gocyclo -over 15 .
|
||||||
|
@echo
|
||||||
|
golint ./...
|
||||||
|
|
||||||
|
clean:
|
||||||
|
@rm -f ${BIN}
|
||||||
|
|
||||||
|
.PHONY: all cmd clean test api check
|
564
pkg/gowebdav/README.md
Normal file
564
pkg/gowebdav/README.md
Normal file
@ -0,0 +1,564 @@
|
|||||||
|
# GoWebDAV
|
||||||
|
|
||||||
|
[](https://travis-ci.org/studio-b12/gowebdav)
|
||||||
|
[](https://godoc.org/github.com/studio-b12/gowebdav)
|
||||||
|
[](https://goreportcard.com/report/github.com/studio-b12/gowebdav)
|
||||||
|
|
||||||
|
A golang WebDAV client library.
|
||||||
|
|
||||||
|
## Main features
|
||||||
|
`gowebdav` library allows to perform following actions on the remote WebDAV server:
|
||||||
|
* [create path](#create-path-on-a-webdav-server)
|
||||||
|
* [get files list](#get-files-list)
|
||||||
|
* [download file](#download-file-to-byte-array)
|
||||||
|
* [upload file](#upload-file-from-byte-array)
|
||||||
|
* [get information about specified file/folder](#get-information-about-specified-filefolder)
|
||||||
|
* [move file to another location](#move-file-to-another-location)
|
||||||
|
* [copy file to another location](#copy-file-to-another-location)
|
||||||
|
* [delete file](#delete-file)
|
||||||
|
|
||||||
|
## Usage
|
||||||
|
|
||||||
|
First of all you should create `Client` instance using `NewClient()` function:
|
||||||
|
|
||||||
|
```go
|
||||||
|
root := "https://webdav.mydomain.me"
|
||||||
|
user := "user"
|
||||||
|
password := "password"
|
||||||
|
|
||||||
|
c := gowebdav.NewClient(root, user, password)
|
||||||
|
```
|
||||||
|
|
||||||
|
After you can use this `Client` to perform actions, described below.
|
||||||
|
|
||||||
|
**NOTICE:** we will not check errors in examples, to focus you on the `gowebdav` library's code, but you should do it in your code!
|
||||||
|
|
||||||
|
### Create path on a WebDAV server
|
||||||
|
```go
|
||||||
|
err := c.Mkdir("folder", 0644)
|
||||||
|
```
|
||||||
|
In case you want to create several folders you can use `c.MkdirAll()`:
|
||||||
|
```go
|
||||||
|
err := c.MkdirAll("folder/subfolder/subfolder2", 0644)
|
||||||
|
```
|
||||||
|
|
||||||
|
### Get files list
|
||||||
|
```go
|
||||||
|
files, _ := c.ReadDir("folder/subfolder")
|
||||||
|
for _, file := range files {
|
||||||
|
//notice that [file] has os.FileInfo type
|
||||||
|
fmt.Println(file.Name())
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
### Download file to byte array
|
||||||
|
```go
|
||||||
|
webdavFilePath := "folder/subfolder/file.txt"
|
||||||
|
localFilePath := "/tmp/webdav/file.txt"
|
||||||
|
|
||||||
|
bytes, _ := c.Read(webdavFilePath)
|
||||||
|
ioutil.WriteFile(localFilePath, bytes, 0644)
|
||||||
|
```
|
||||||
|
|
||||||
|
### Download file via reader
|
||||||
|
Also you can use `c.ReadStream()` method:
|
||||||
|
```go
|
||||||
|
webdavFilePath := "folder/subfolder/file.txt"
|
||||||
|
localFilePath := "/tmp/webdav/file.txt"
|
||||||
|
|
||||||
|
reader, _ := c.ReadStream(webdavFilePath)
|
||||||
|
|
||||||
|
file, _ := os.Create(localFilePath)
|
||||||
|
defer file.Close()
|
||||||
|
|
||||||
|
io.Copy(file, reader)
|
||||||
|
```
|
||||||
|
|
||||||
|
### Upload file from byte array
|
||||||
|
```go
|
||||||
|
webdavFilePath := "folder/subfolder/file.txt"
|
||||||
|
localFilePath := "/tmp/webdav/file.txt"
|
||||||
|
|
||||||
|
bytes, _ := ioutil.ReadFile(localFilePath)
|
||||||
|
|
||||||
|
c.Write(webdavFilePath, bytes, 0644)
|
||||||
|
```
|
||||||
|
|
||||||
|
### Upload file via writer
|
||||||
|
```go
|
||||||
|
webdavFilePath := "folder/subfolder/file.txt"
|
||||||
|
localFilePath := "/tmp/webdav/file.txt"
|
||||||
|
|
||||||
|
file, _ := os.Open(localFilePath)
|
||||||
|
defer file.Close()
|
||||||
|
|
||||||
|
c.WriteStream(webdavFilePath, file, 0644)
|
||||||
|
```
|
||||||
|
|
||||||
|
### Get information about specified file/folder
|
||||||
|
```go
|
||||||
|
webdavFilePath := "folder/subfolder/file.txt"
|
||||||
|
|
||||||
|
info := c.Stat(webdavFilePath)
|
||||||
|
//notice that [info] has os.FileInfo type
|
||||||
|
fmt.Println(info)
|
||||||
|
```
|
||||||
|
|
||||||
|
### Move file to another location
|
||||||
|
```go
|
||||||
|
oldPath := "folder/subfolder/file.txt"
|
||||||
|
newPath := "folder/subfolder/moved.txt"
|
||||||
|
isOverwrite := true
|
||||||
|
|
||||||
|
c.Rename(oldPath, newPath, isOverwrite)
|
||||||
|
```
|
||||||
|
|
||||||
|
### Copy file to another location
|
||||||
|
```go
|
||||||
|
oldPath := "folder/subfolder/file.txt"
|
||||||
|
newPath := "folder/subfolder/file-copy.txt"
|
||||||
|
isOverwrite := true
|
||||||
|
|
||||||
|
c.Copy(oldPath, newPath, isOverwrite)
|
||||||
|
```
|
||||||
|
|
||||||
|
### Delete file
|
||||||
|
```go
|
||||||
|
webdavFilePath := "folder/subfolder/file.txt"
|
||||||
|
|
||||||
|
c.Remove(webdavFilePath)
|
||||||
|
```
|
||||||
|
|
||||||
|
## Links
|
||||||
|
|
||||||
|
More details about WebDAV server you can read from following resources:
|
||||||
|
|
||||||
|
* [RFC 4918 - HTTP Extensions for Web Distributed Authoring and Versioning (WebDAV)](https://tools.ietf.org/html/rfc4918)
|
||||||
|
* [RFC 5689 - Extended MKCOL for Web Distributed Authoring and Versioning (WebDAV)](https://tools.ietf.org/html/rfc5689)
|
||||||
|
* [RFC 2616 - HTTP/1.1 Status Code Definitions](http://www.w3.org/Protocols/rfc2616/rfc2616-sec10.html "HTTP/1.1 Status Code Definitions")
|
||||||
|
* [WebDav: Next Generation Collaborative Web Authoring By Lisa Dusseaul](https://books.google.de/books?isbn=0130652083 "WebDav: Next Generation Collaborative Web Authoring By Lisa Dusseault")
|
||||||
|
|
||||||
|
**NOTICE**: RFC 2518 is obsoleted by RFC 4918 in June 2007
|
||||||
|
|
||||||
|
## Contributing
|
||||||
|
All contributing are welcome. If you have any suggestions or find some bug - please create an Issue to let us make this project better. We appreciate your help!
|
||||||
|
|
||||||
|
## License
|
||||||
|
This library is distributed under the BSD 3-Clause license found in the [LICENSE](https://github.com/studio-b12/gowebdav/blob/master/LICENSE) file.
|
||||||
|
## API
|
||||||
|
|
||||||
|
`import "github.com/studio-b12/gowebdav"`
|
||||||
|
|
||||||
|
* [Overview](#pkg-overview)
|
||||||
|
* [Index](#pkg-index)
|
||||||
|
* [Examples](#pkg-examples)
|
||||||
|
* [Subdirectories](#pkg-subdirectories)
|
||||||
|
|
||||||
|
### <a name="pkg-overview">Overview</a>
|
||||||
|
Package gowebdav is a WebDAV client library with a command line tool
|
||||||
|
included.
|
||||||
|
|
||||||
|
### <a name="pkg-index">Index</a>
|
||||||
|
* [func FixSlash(s string) string](#FixSlash)
|
||||||
|
* [func FixSlashes(s string) string](#FixSlashes)
|
||||||
|
* [func Join(path0 string, path1 string) string](#Join)
|
||||||
|
* [func PathEscape(path string) string](#PathEscape)
|
||||||
|
* [func ReadConfig(uri, netrc string) (string, string)](#ReadConfig)
|
||||||
|
* [func String(r io.Reader) string](#String)
|
||||||
|
* [type Authenticator](#Authenticator)
|
||||||
|
* [type BasicAuth](#BasicAuth)
|
||||||
|
* [func (b *BasicAuth) Authorize(req *http.Request, method string, path string)](#BasicAuth.Authorize)
|
||||||
|
* [func (b *BasicAuth) Pass() string](#BasicAuth.Pass)
|
||||||
|
* [func (b *BasicAuth) Type() string](#BasicAuth.Type)
|
||||||
|
* [func (b *BasicAuth) User() string](#BasicAuth.User)
|
||||||
|
* [type Client](#Client)
|
||||||
|
* [func NewClient(uri, user, pw string) *Client](#NewClient)
|
||||||
|
* [func (c *Client) Connect() error](#Client.Connect)
|
||||||
|
* [func (c *Client) Copy(oldpath, newpath string, overwrite bool) error](#Client.Copy)
|
||||||
|
* [func (c *Client) Mkdir(path string, _ os.FileMode) error](#Client.Mkdir)
|
||||||
|
* [func (c *Client) MkdirAll(path string, _ os.FileMode) error](#Client.MkdirAll)
|
||||||
|
* [func (c *Client) Read(path string) ([]byte, error)](#Client.Read)
|
||||||
|
* [func (c *Client) ReadDir(path string) ([]os.FileInfo, error)](#Client.ReadDir)
|
||||||
|
* [func (c *Client) ReadStream(path string) (io.ReadCloser, error)](#Client.ReadStream)
|
||||||
|
* [func (c *Client) ReadStreamRange(path string, offset, length int64) (io.ReadCloser, error)](#Client.ReadStreamRange)
|
||||||
|
* [func (c *Client) Remove(path string) error](#Client.Remove)
|
||||||
|
* [func (c *Client) RemoveAll(path string) error](#Client.RemoveAll)
|
||||||
|
* [func (c *Client) Rename(oldpath, newpath string, overwrite bool) error](#Client.Rename)
|
||||||
|
* [func (c *Client) SetHeader(key, value string)](#Client.SetHeader)
|
||||||
|
* [func (c *Client) SetInterceptor(interceptor func(method string, rq *http.Request))](#Client.SetInterceptor)
|
||||||
|
* [func (c *Client) SetTimeout(timeout time.Duration)](#Client.SetTimeout)
|
||||||
|
* [func (c *Client) SetTransport(transport http.RoundTripper)](#Client.SetTransport)
|
||||||
|
* [func (c *Client) Stat(path string) (os.FileInfo, error)](#Client.Stat)
|
||||||
|
* [func (c *Client) Write(path string, data []byte, _ os.FileMode) error](#Client.Write)
|
||||||
|
* [func (c *Client) WriteStream(path string, stream io.Reader, _ os.FileMode) error](#Client.WriteStream)
|
||||||
|
* [type DigestAuth](#DigestAuth)
|
||||||
|
* [func (d *DigestAuth) Authorize(req *http.Request, method string, path string)](#DigestAuth.Authorize)
|
||||||
|
* [func (d *DigestAuth) Pass() string](#DigestAuth.Pass)
|
||||||
|
* [func (d *DigestAuth) Type() string](#DigestAuth.Type)
|
||||||
|
* [func (d *DigestAuth) User() string](#DigestAuth.User)
|
||||||
|
* [type File](#File)
|
||||||
|
* [func (f File) ContentType() string](#File.ContentType)
|
||||||
|
* [func (f File) ETag() string](#File.ETag)
|
||||||
|
* [func (f File) IsDir() bool](#File.IsDir)
|
||||||
|
* [func (f File) ModTime() time.Time](#File.ModTime)
|
||||||
|
* [func (f File) Mode() os.FileMode](#File.Mode)
|
||||||
|
* [func (f File) Name() string](#File.Name)
|
||||||
|
* [func (f File) Path() string](#File.Path)
|
||||||
|
* [func (f File) Size() int64](#File.Size)
|
||||||
|
* [func (f File) String() string](#File.String)
|
||||||
|
* [func (f File) Sys() interface{}](#File.Sys)
|
||||||
|
* [type NoAuth](#NoAuth)
|
||||||
|
* [func (n *NoAuth) Authorize(req *http.Request, method string, path string)](#NoAuth.Authorize)
|
||||||
|
* [func (n *NoAuth) Pass() string](#NoAuth.Pass)
|
||||||
|
* [func (n *NoAuth) Type() string](#NoAuth.Type)
|
||||||
|
* [func (n *NoAuth) User() string](#NoAuth.User)
|
||||||
|
|
||||||
|
##### <a name="pkg-examples">Examples</a>
|
||||||
|
* [PathEscape](#example_PathEscape)
|
||||||
|
|
||||||
|
##### <a name="pkg-files">Package files</a>
|
||||||
|
[basicAuth.go](https://github.com/studio-b12/gowebdav/blob/master/basicAuth.go) [client.go](https://github.com/studio-b12/gowebdav/blob/master/client.go) [digestAuth.go](https://github.com/studio-b12/gowebdav/blob/master/digestAuth.go) [doc.go](https://github.com/studio-b12/gowebdav/blob/master/doc.go) [file.go](https://github.com/studio-b12/gowebdav/blob/master/file.go) [netrc.go](https://github.com/studio-b12/gowebdav/blob/master/netrc.go) [requests.go](https://github.com/studio-b12/gowebdav/blob/master/requests.go) [utils.go](https://github.com/studio-b12/gowebdav/blob/master/utils.go)
|
||||||
|
|
||||||
|
### <a name="FixSlash">func</a> [FixSlash](https://github.com/studio-b12/gowebdav/blob/master/utils.go?s=707:737#L45)
|
||||||
|
``` go
|
||||||
|
func FixSlash(s string) string
|
||||||
|
```
|
||||||
|
FixSlash appends a trailing / to our string
|
||||||
|
|
||||||
|
### <a name="FixSlashes">func</a> [FixSlashes](https://github.com/studio-b12/gowebdav/blob/master/utils.go?s=859:891#L53)
|
||||||
|
``` go
|
||||||
|
func FixSlashes(s string) string
|
||||||
|
```
|
||||||
|
FixSlashes appends and prepends a / if they are missing
|
||||||
|
|
||||||
|
### <a name="Join">func</a> [Join](https://github.com/studio-b12/gowebdav/blob/master/utils.go?s=992:1036#L62)
|
||||||
|
``` go
|
||||||
|
func Join(path0 string, path1 string) string
|
||||||
|
```
|
||||||
|
Join joins two paths
|
||||||
|
|
||||||
|
### <a name="PathEscape">func</a> [PathEscape](https://github.com/studio-b12/gowebdav/blob/master/utils.go?s=506:541#L36)
|
||||||
|
``` go
|
||||||
|
func PathEscape(path string) string
|
||||||
|
```
|
||||||
|
PathEscape escapes all segments of a given path
|
||||||
|
|
||||||
|
### <a name="ReadConfig">func</a> [ReadConfig](https://github.com/studio-b12/gowebdav/blob/master/netrc.go?s=428:479#L27)
|
||||||
|
``` go
|
||||||
|
func ReadConfig(uri, netrc string) (string, string)
|
||||||
|
```
|
||||||
|
ReadConfig reads login and password configuration from ~/.netrc
|
||||||
|
machine foo.com login username password 123456
|
||||||
|
|
||||||
|
### <a name="String">func</a> [String](https://github.com/studio-b12/gowebdav/blob/master/utils.go?s=1166:1197#L67)
|
||||||
|
``` go
|
||||||
|
func String(r io.Reader) string
|
||||||
|
```
|
||||||
|
String pulls a string out of our io.Reader
|
||||||
|
|
||||||
|
### <a name="Authenticator">type</a> [Authenticator](https://github.com/studio-b12/gowebdav/blob/master/client.go?s=388:507#L29)
|
||||||
|
``` go
|
||||||
|
type Authenticator interface {
|
||||||
|
Type() string
|
||||||
|
User() string
|
||||||
|
Pass() string
|
||||||
|
Authorize(*http.Request, string, string)
|
||||||
|
}
|
||||||
|
```
|
||||||
|
Authenticator stub
|
||||||
|
|
||||||
|
### <a name="BasicAuth">type</a> [BasicAuth](https://github.com/studio-b12/gowebdav/blob/master/basicAuth.go?s=106:157#L9)
|
||||||
|
``` go
|
||||||
|
type BasicAuth struct {
|
||||||
|
// contains filtered or unexported fields
|
||||||
|
}
|
||||||
|
```
|
||||||
|
BasicAuth structure holds our credentials
|
||||||
|
|
||||||
|
#### <a name="BasicAuth.Authorize">func</a> (\*BasicAuth) [Authorize](https://github.com/studio-b12/gowebdav/blob/master/basicAuth.go?s=473:549#L30)
|
||||||
|
``` go
|
||||||
|
func (b *BasicAuth) Authorize(req *http.Request, method string, path string)
|
||||||
|
```
|
||||||
|
Authorize the current request
|
||||||
|
|
||||||
|
#### <a name="BasicAuth.Pass">func</a> (\*BasicAuth) [Pass](https://github.com/studio-b12/gowebdav/blob/master/basicAuth.go?s=388:421#L25)
|
||||||
|
``` go
|
||||||
|
func (b *BasicAuth) Pass() string
|
||||||
|
```
|
||||||
|
Pass holds the BasicAuth password
|
||||||
|
|
||||||
|
#### <a name="BasicAuth.Type">func</a> (\*BasicAuth) [Type](https://github.com/studio-b12/gowebdav/blob/master/basicAuth.go?s=201:234#L15)
|
||||||
|
``` go
|
||||||
|
func (b *BasicAuth) Type() string
|
||||||
|
```
|
||||||
|
Type identifies the BasicAuthenticator
|
||||||
|
|
||||||
|
#### <a name="BasicAuth.User">func</a> (\*BasicAuth) [User](https://github.com/studio-b12/gowebdav/blob/master/basicAuth.go?s=297:330#L20)
|
||||||
|
``` go
|
||||||
|
func (b *BasicAuth) User() string
|
||||||
|
```
|
||||||
|
User holds the BasicAuth username
|
||||||
|
|
||||||
|
### <a name="Client">type</a> [Client](https://github.com/studio-b12/gowebdav/blob/master/client.go?s=172:364#L18)
|
||||||
|
``` go
|
||||||
|
type Client struct {
|
||||||
|
// contains filtered or unexported fields
|
||||||
|
}
|
||||||
|
```
|
||||||
|
Client defines our structure
|
||||||
|
|
||||||
|
#### <a name="NewClient">func</a> [NewClient](https://github.com/studio-b12/gowebdav/blob/master/client.go?s=1019:1063#L62)
|
||||||
|
``` go
|
||||||
|
func NewClient(uri, user, pw string) *Client
|
||||||
|
```
|
||||||
|
NewClient creates a new instance of client
|
||||||
|
|
||||||
|
#### <a name="Client.Connect">func</a> (\*Client) [Connect](https://github.com/studio-b12/gowebdav/blob/master/client.go?s=1843:1875#L87)
|
||||||
|
``` go
|
||||||
|
func (c *Client) Connect() error
|
||||||
|
```
|
||||||
|
Connect connects to our dav server
|
||||||
|
|
||||||
|
#### <a name="Client.Copy">func</a> (\*Client) [Copy](https://github.com/studio-b12/gowebdav/blob/master/client.go?s=6702:6770#L313)
|
||||||
|
``` go
|
||||||
|
func (c *Client) Copy(oldpath, newpath string, overwrite bool) error
|
||||||
|
```
|
||||||
|
Copy copies a file from A to B
|
||||||
|
|
||||||
|
#### <a name="Client.Mkdir">func</a> (\*Client) [Mkdir](https://github.com/studio-b12/gowebdav/blob/master/client.go?s=5793:5849#L272)
|
||||||
|
``` go
|
||||||
|
func (c *Client) Mkdir(path string, _ os.FileMode) error
|
||||||
|
```
|
||||||
|
Mkdir makes a directory
|
||||||
|
|
||||||
|
#### <a name="Client.MkdirAll">func</a> (\*Client) [MkdirAll](https://github.com/studio-b12/gowebdav/blob/master/client.go?s=6028:6087#L283)
|
||||||
|
``` go
|
||||||
|
func (c *Client) MkdirAll(path string, _ os.FileMode) error
|
||||||
|
```
|
||||||
|
MkdirAll like mkdir -p, but for webdav
|
||||||
|
|
||||||
|
#### <a name="Client.Read">func</a> (\*Client) [Read](https://github.com/studio-b12/gowebdav/blob/master/client.go?s=6876:6926#L318)
|
||||||
|
``` go
|
||||||
|
func (c *Client) Read(path string) ([]byte, error)
|
||||||
|
```
|
||||||
|
Read reads the contents of a remote file
|
||||||
|
|
||||||
|
#### <a name="Client.ReadDir">func</a> (\*Client) [ReadDir](https://github.com/studio-b12/gowebdav/blob/master/client.go?s=2869:2929#L130)
|
||||||
|
``` go
|
||||||
|
func (c *Client) ReadDir(path string) ([]os.FileInfo, error)
|
||||||
|
```
|
||||||
|
ReadDir reads the contents of a remote directory
|
||||||
|
|
||||||
|
#### <a name="Client.ReadStream">func</a> (\*Client) [ReadStream](https://github.com/studio-b12/gowebdav/blob/master/client.go?s=7237:7300#L336)
|
||||||
|
``` go
|
||||||
|
func (c *Client) ReadStream(path string) (io.ReadCloser, error)
|
||||||
|
```
|
||||||
|
ReadStream reads the stream for a given path
|
||||||
|
|
||||||
|
#### <a name="Client.ReadStreamRange">func</a> (\*Client) [ReadStreamRange](https://github.com/studio-b12/gowebdav/blob/master/client.go?s=8049:8139#L358)
|
||||||
|
``` go
|
||||||
|
func (c *Client) ReadStreamRange(path string, offset, length int64) (io.ReadCloser, error)
|
||||||
|
```
|
||||||
|
ReadStreamRange reads the stream representing a subset of bytes for a given path,
|
||||||
|
utilizing HTTP Range Requests if the server supports it.
|
||||||
|
The range is expressed as offset from the start of the file and length, for example
|
||||||
|
offset=10, length=10 will return bytes 10 through 19.
|
||||||
|
|
||||||
|
If the server does not support partial content requests and returns full content instead,
|
||||||
|
this function will emulate the behavior by skipping `offset` bytes and limiting the result
|
||||||
|
to `length`.
|
||||||
|
|
||||||
|
#### <a name="Client.Remove">func</a> (\*Client) [Remove](https://github.com/studio-b12/gowebdav/blob/master/client.go?s=5299:5341#L249)
|
||||||
|
``` go
|
||||||
|
func (c *Client) Remove(path string) error
|
||||||
|
```
|
||||||
|
Remove removes a remote file
|
||||||
|
|
||||||
|
#### <a name="Client.RemoveAll">func</a> (\*Client) [RemoveAll](https://github.com/studio-b12/gowebdav/blob/master/client.go?s=5407:5452#L254)
|
||||||
|
``` go
|
||||||
|
func (c *Client) RemoveAll(path string) error
|
||||||
|
```
|
||||||
|
RemoveAll removes remote files
|
||||||
|
|
||||||
|
#### <a name="Client.Rename">func</a> (\*Client) [Rename](https://github.com/studio-b12/gowebdav/blob/master/client.go?s=6536:6606#L308)
|
||||||
|
``` go
|
||||||
|
func (c *Client) Rename(oldpath, newpath string, overwrite bool) error
|
||||||
|
```
|
||||||
|
Rename moves a file from A to B
|
||||||
|
|
||||||
|
#### <a name="Client.SetHeader">func</a> (\*Client) [SetHeader](https://github.com/studio-b12/gowebdav/blob/master/client.go?s=1235:1280#L67)
|
||||||
|
``` go
|
||||||
|
func (c *Client) SetHeader(key, value string)
|
||||||
|
```
|
||||||
|
SetHeader lets us set arbitrary headers for a given client
|
||||||
|
|
||||||
|
#### <a name="Client.SetInterceptor">func</a> (\*Client) [SetInterceptor](https://github.com/studio-b12/gowebdav/blob/master/client.go?s=1387:1469#L72)
|
||||||
|
``` go
|
||||||
|
func (c *Client) SetInterceptor(interceptor func(method string, rq *http.Request))
|
||||||
|
```
|
||||||
|
SetInterceptor lets us set an arbitrary interceptor for a given client
|
||||||
|
|
||||||
|
#### <a name="Client.SetTimeout">func</a> (\*Client) [SetTimeout](https://github.com/studio-b12/gowebdav/blob/master/client.go?s=1571:1621#L77)
|
||||||
|
``` go
|
||||||
|
func (c *Client) SetTimeout(timeout time.Duration)
|
||||||
|
```
|
||||||
|
SetTimeout exposes the ability to set a time limit for requests
|
||||||
|
|
||||||
|
#### <a name="Client.SetTransport">func</a> (\*Client) [SetTransport](https://github.com/studio-b12/gowebdav/blob/master/client.go?s=1714:1772#L82)
|
||||||
|
``` go
|
||||||
|
func (c *Client) SetTransport(transport http.RoundTripper)
|
||||||
|
```
|
||||||
|
SetTransport exposes the ability to define custom transports
|
||||||
|
|
||||||
|
#### <a name="Client.Stat">func</a> (\*Client) [Stat](https://github.com/studio-b12/gowebdav/blob/master/client.go?s=4255:4310#L197)
|
||||||
|
``` go
|
||||||
|
func (c *Client) Stat(path string) (os.FileInfo, error)
|
||||||
|
```
|
||||||
|
Stat returns the file stats for a specified path
|
||||||
|
|
||||||
|
#### <a name="Client.Write">func</a> (\*Client) [Write](https://github.com/studio-b12/gowebdav/blob/master/client.go?s=9051:9120#L388)
|
||||||
|
``` go
|
||||||
|
func (c *Client) Write(path string, data []byte, _ os.FileMode) error
|
||||||
|
```
|
||||||
|
Write writes data to a given path
|
||||||
|
|
||||||
|
#### <a name="Client.WriteStream">func</a> (\*Client) [WriteStream](https://github.com/studio-b12/gowebdav/blob/master/client.go?s=9476:9556#L411)
|
||||||
|
``` go
|
||||||
|
func (c *Client) WriteStream(path string, stream io.Reader, _ os.FileMode) error
|
||||||
|
```
|
||||||
|
WriteStream writes a stream
|
||||||
|
|
||||||
|
### <a name="DigestAuth">type</a> [DigestAuth](https://github.com/studio-b12/gowebdav/blob/master/digestAuth.go?s=157:254#L14)
|
||||||
|
``` go
|
||||||
|
type DigestAuth struct {
|
||||||
|
// contains filtered or unexported fields
|
||||||
|
}
|
||||||
|
```
|
||||||
|
DigestAuth structure holds our credentials
|
||||||
|
|
||||||
|
#### <a name="DigestAuth.Authorize">func</a> (\*DigestAuth) [Authorize](https://github.com/studio-b12/gowebdav/blob/master/digestAuth.go?s=577:654#L36)
|
||||||
|
``` go
|
||||||
|
func (d *DigestAuth) Authorize(req *http.Request, method string, path string)
|
||||||
|
```
|
||||||
|
Authorize the current request
|
||||||
|
|
||||||
|
#### <a name="DigestAuth.Pass">func</a> (\*DigestAuth) [Pass](https://github.com/studio-b12/gowebdav/blob/master/digestAuth.go?s=491:525#L31)
|
||||||
|
``` go
|
||||||
|
func (d *DigestAuth) Pass() string
|
||||||
|
```
|
||||||
|
Pass holds the DigestAuth password
|
||||||
|
|
||||||
|
#### <a name="DigestAuth.Type">func</a> (\*DigestAuth) [Type](https://github.com/studio-b12/gowebdav/blob/master/digestAuth.go?s=299:333#L21)
|
||||||
|
``` go
|
||||||
|
func (d *DigestAuth) Type() string
|
||||||
|
```
|
||||||
|
Type identifies the DigestAuthenticator
|
||||||
|
|
||||||
|
#### <a name="DigestAuth.User">func</a> (\*DigestAuth) [User](https://github.com/studio-b12/gowebdav/blob/master/digestAuth.go?s=398:432#L26)
|
||||||
|
``` go
|
||||||
|
func (d *DigestAuth) User() string
|
||||||
|
```
|
||||||
|
User holds the DigestAuth username
|
||||||
|
|
||||||
|
### <a name="File">type</a> [File](https://github.com/studio-b12/gowebdav/blob/master/file.go?s=93:253#L10)
|
||||||
|
``` go
|
||||||
|
type File struct {
|
||||||
|
// contains filtered or unexported fields
|
||||||
|
}
|
||||||
|
```
|
||||||
|
File is our structure for a given file
|
||||||
|
|
||||||
|
#### <a name="File.ContentType">func</a> (File) [ContentType](https://github.com/studio-b12/gowebdav/blob/master/file.go?s=476:510#L31)
|
||||||
|
``` go
|
||||||
|
func (f File) ContentType() string
|
||||||
|
```
|
||||||
|
ContentType returns the content type of a file
|
||||||
|
|
||||||
|
#### <a name="File.ETag">func</a> (File) [ETag](https://github.com/studio-b12/gowebdav/blob/master/file.go?s=929:956#L56)
|
||||||
|
``` go
|
||||||
|
func (f File) ETag() string
|
||||||
|
```
|
||||||
|
ETag returns the ETag of a file
|
||||||
|
|
||||||
|
#### <a name="File.IsDir">func</a> (File) [IsDir](https://github.com/studio-b12/gowebdav/blob/master/file.go?s=1035:1061#L61)
|
||||||
|
``` go
|
||||||
|
func (f File) IsDir() bool
|
||||||
|
```
|
||||||
|
IsDir let us see if a given file is a directory or not
|
||||||
|
|
||||||
|
#### <a name="File.ModTime">func</a> (File) [ModTime](https://github.com/studio-b12/gowebdav/blob/master/file.go?s=836:869#L51)
|
||||||
|
``` go
|
||||||
|
func (f File) ModTime() time.Time
|
||||||
|
```
|
||||||
|
ModTime returns the modified time of a file
|
||||||
|
|
||||||
|
#### <a name="File.Mode">func</a> (File) [Mode](https://github.com/studio-b12/gowebdav/blob/master/file.go?s=665:697#L41)
|
||||||
|
``` go
|
||||||
|
func (f File) Mode() os.FileMode
|
||||||
|
```
|
||||||
|
Mode will return the mode of a given file
|
||||||
|
|
||||||
|
#### <a name="File.Name">func</a> (File) [Name](https://github.com/studio-b12/gowebdav/blob/master/file.go?s=378:405#L26)
|
||||||
|
``` go
|
||||||
|
func (f File) Name() string
|
||||||
|
```
|
||||||
|
Name returns the name of a file
|
||||||
|
|
||||||
|
#### <a name="File.Path">func</a> (File) [Path](https://github.com/studio-b12/gowebdav/blob/master/file.go?s=295:322#L21)
|
||||||
|
``` go
|
||||||
|
func (f File) Path() string
|
||||||
|
```
|
||||||
|
Path returns the full path of a file
|
||||||
|
|
||||||
|
#### <a name="File.Size">func</a> (File) [Size](https://github.com/studio-b12/gowebdav/blob/master/file.go?s=573:599#L36)
|
||||||
|
``` go
|
||||||
|
func (f File) Size() int64
|
||||||
|
```
|
||||||
|
Size returns the size of a file
|
||||||
|
|
||||||
|
#### <a name="File.String">func</a> (File) [String](https://github.com/studio-b12/gowebdav/blob/master/file.go?s=1183:1212#L71)
|
||||||
|
``` go
|
||||||
|
func (f File) String() string
|
||||||
|
```
|
||||||
|
String lets us see file information
|
||||||
|
|
||||||
|
#### <a name="File.Sys">func</a> (File) [Sys](https://github.com/studio-b12/gowebdav/blob/master/file.go?s=1095:1126#L66)
|
||||||
|
``` go
|
||||||
|
func (f File) Sys() interface{}
|
||||||
|
```
|
||||||
|
Sys ????
|
||||||
|
|
||||||
|
### <a name="NoAuth">type</a> [NoAuth](https://github.com/studio-b12/gowebdav/blob/master/client.go?s=551:599#L37)
|
||||||
|
``` go
|
||||||
|
type NoAuth struct {
|
||||||
|
// contains filtered or unexported fields
|
||||||
|
}
|
||||||
|
```
|
||||||
|
NoAuth structure holds our credentials
|
||||||
|
|
||||||
|
#### <a name="NoAuth.Authorize">func</a> (\*NoAuth) [Authorize](https://github.com/studio-b12/gowebdav/blob/master/client.go?s=894:967#L58)
|
||||||
|
``` go
|
||||||
|
func (n *NoAuth) Authorize(req *http.Request, method string, path string)
|
||||||
|
```
|
||||||
|
Authorize the current request
|
||||||
|
|
||||||
|
#### <a name="NoAuth.Pass">func</a> (\*NoAuth) [Pass](https://github.com/studio-b12/gowebdav/blob/master/client.go?s=812:842#L53)
|
||||||
|
``` go
|
||||||
|
func (n *NoAuth) Pass() string
|
||||||
|
```
|
||||||
|
Pass returns the current password
|
||||||
|
|
||||||
|
#### <a name="NoAuth.Type">func</a> (\*NoAuth) [Type](https://github.com/studio-b12/gowebdav/blob/master/client.go?s=638:668#L43)
|
||||||
|
``` go
|
||||||
|
func (n *NoAuth) Type() string
|
||||||
|
```
|
||||||
|
Type identifies the authenticator
|
||||||
|
|
||||||
|
#### <a name="NoAuth.User">func</a> (\*NoAuth) [User](https://github.com/studio-b12/gowebdav/blob/master/client.go?s=724:754#L48)
|
||||||
|
``` go
|
||||||
|
func (n *NoAuth) User() string
|
||||||
|
```
|
||||||
|
User returns the current user
|
||||||
|
|
||||||
|
- - -
|
||||||
|
Generated by [godoc2md](http://godoc.org/github.com/davecheney/godoc2md)
|
34
pkg/gowebdav/basicAuth.go
Normal file
34
pkg/gowebdav/basicAuth.go
Normal file
@ -0,0 +1,34 @@
|
|||||||
|
package gowebdav
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/base64"
|
||||||
|
"net/http"
|
||||||
|
)
|
||||||
|
|
||||||
|
// BasicAuth structure holds our credentials
|
||||||
|
type BasicAuth struct {
|
||||||
|
user string
|
||||||
|
pw string
|
||||||
|
}
|
||||||
|
|
||||||
|
// Type identifies the BasicAuthenticator
|
||||||
|
func (b *BasicAuth) Type() string {
|
||||||
|
return "BasicAuth"
|
||||||
|
}
|
||||||
|
|
||||||
|
// User holds the BasicAuth username
|
||||||
|
func (b *BasicAuth) User() string {
|
||||||
|
return b.user
|
||||||
|
}
|
||||||
|
|
||||||
|
// Pass holds the BasicAuth password
|
||||||
|
func (b *BasicAuth) Pass() string {
|
||||||
|
return b.pw
|
||||||
|
}
|
||||||
|
|
||||||
|
// Authorize the current request
|
||||||
|
func (b *BasicAuth) Authorize(req *http.Request, method string, path string) {
|
||||||
|
a := b.user + ":" + b.pw
|
||||||
|
auth := "Basic " + base64.StdEncoding.EncodeToString([]byte(a))
|
||||||
|
req.Header.Set("Authorization", auth)
|
||||||
|
}
|
447
pkg/gowebdav/client.go
Normal file
447
pkg/gowebdav/client.go
Normal file
@ -0,0 +1,447 @@
|
|||||||
|
package gowebdav
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
"encoding/xml"
|
||||||
|
"fmt"
|
||||||
|
"io"
|
||||||
|
"net/http"
|
||||||
|
"net/url"
|
||||||
|
"os"
|
||||||
|
pathpkg "path"
|
||||||
|
"strings"
|
||||||
|
"sync"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Client defines our structure
|
||||||
|
type Client struct {
|
||||||
|
root string
|
||||||
|
headers http.Header
|
||||||
|
interceptor func(method string, rq *http.Request)
|
||||||
|
c *http.Client
|
||||||
|
|
||||||
|
authMutex sync.Mutex
|
||||||
|
auth Authenticator
|
||||||
|
}
|
||||||
|
|
||||||
|
// Authenticator stub
|
||||||
|
type Authenticator interface {
|
||||||
|
Type() string
|
||||||
|
User() string
|
||||||
|
Pass() string
|
||||||
|
Authorize(*http.Request, string, string)
|
||||||
|
}
|
||||||
|
|
||||||
|
// NoAuth structure holds our credentials
|
||||||
|
type NoAuth struct {
|
||||||
|
user string
|
||||||
|
pw string
|
||||||
|
}
|
||||||
|
|
||||||
|
// Type identifies the authenticator
|
||||||
|
func (n *NoAuth) Type() string {
|
||||||
|
return "NoAuth"
|
||||||
|
}
|
||||||
|
|
||||||
|
// User returns the current user
|
||||||
|
func (n *NoAuth) User() string {
|
||||||
|
return n.user
|
||||||
|
}
|
||||||
|
|
||||||
|
// Pass returns the current password
|
||||||
|
func (n *NoAuth) Pass() string {
|
||||||
|
return n.pw
|
||||||
|
}
|
||||||
|
|
||||||
|
// Authorize the current request
|
||||||
|
func (n *NoAuth) Authorize(req *http.Request, method string, path string) {
|
||||||
|
}
|
||||||
|
|
||||||
|
// NewClient creates a new instance of client
|
||||||
|
func NewClient(uri, user, pw string) *Client {
|
||||||
|
return &Client{FixSlash(uri), make(http.Header), nil, &http.Client{}, sync.Mutex{}, &NoAuth{user, pw}}
|
||||||
|
}
|
||||||
|
|
||||||
|
// SetHeader lets us set arbitrary headers for a given client
|
||||||
|
func (c *Client) SetHeader(key, value string) {
|
||||||
|
c.headers.Add(key, value)
|
||||||
|
}
|
||||||
|
|
||||||
|
// SetInterceptor lets us set an arbitrary interceptor for a given client
|
||||||
|
func (c *Client) SetInterceptor(interceptor func(method string, rq *http.Request)) {
|
||||||
|
c.interceptor = interceptor
|
||||||
|
}
|
||||||
|
|
||||||
|
// SetTimeout exposes the ability to set a time limit for requests
|
||||||
|
func (c *Client) SetTimeout(timeout time.Duration) {
|
||||||
|
c.c.Timeout = timeout
|
||||||
|
}
|
||||||
|
|
||||||
|
// SetTransport exposes the ability to define custom transports
|
||||||
|
func (c *Client) SetTransport(transport http.RoundTripper) {
|
||||||
|
c.c.Transport = transport
|
||||||
|
}
|
||||||
|
|
||||||
|
// Connect connects to our dav server
|
||||||
|
func (c *Client) Connect() error {
|
||||||
|
rs, err := c.options("/")
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
err = rs.Body.Close()
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
if rs.StatusCode != 200 {
|
||||||
|
return newPathError("Connect", c.root, rs.StatusCode)
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
type props struct {
|
||||||
|
Status string `xml:"DAV: status"`
|
||||||
|
Name string `xml:"DAV: prop>displayname,omitempty"`
|
||||||
|
Type xml.Name `xml:"DAV: prop>resourcetype>collection,omitempty"`
|
||||||
|
Size string `xml:"DAV: prop>getcontentlength,omitempty"`
|
||||||
|
ContentType string `xml:"DAV: prop>getcontenttype,omitempty"`
|
||||||
|
ETag string `xml:"DAV: prop>getetag,omitempty"`
|
||||||
|
Modified string `xml:"DAV: prop>getlastmodified,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type response struct {
|
||||||
|
Href string `xml:"DAV: href"`
|
||||||
|
Props []props `xml:"DAV: propstat"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func getProps(r *response, status string) *props {
|
||||||
|
for _, prop := range r.Props {
|
||||||
|
if strings.Contains(prop.Status, status) {
|
||||||
|
return &prop
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// ReadDir reads the contents of a remote directory
|
||||||
|
func (c *Client) ReadDir(path string) ([]os.FileInfo, error) {
|
||||||
|
path = FixSlashes(path)
|
||||||
|
files := make([]os.FileInfo, 0)
|
||||||
|
skipSelf := true
|
||||||
|
parse := func(resp interface{}) error {
|
||||||
|
r := resp.(*response)
|
||||||
|
|
||||||
|
if skipSelf {
|
||||||
|
skipSelf = false
|
||||||
|
if p := getProps(r, "200"); p != nil && p.Type.Local == "collection" {
|
||||||
|
r.Props = nil
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
return newPathError("ReadDir", path, 405)
|
||||||
|
}
|
||||||
|
|
||||||
|
if p := getProps(r, "200"); p != nil {
|
||||||
|
f := new(File)
|
||||||
|
if ps, err := url.PathUnescape(r.Href); err == nil {
|
||||||
|
f.name = pathpkg.Base(ps)
|
||||||
|
} else {
|
||||||
|
f.name = p.Name
|
||||||
|
}
|
||||||
|
f.path = path + f.name
|
||||||
|
f.modified = parseModified(&p.Modified)
|
||||||
|
f.etag = p.ETag
|
||||||
|
f.contentType = p.ContentType
|
||||||
|
|
||||||
|
if p.Type.Local == "collection" {
|
||||||
|
f.path += "/"
|
||||||
|
f.size = 0
|
||||||
|
f.isdir = true
|
||||||
|
} else {
|
||||||
|
f.size = parseInt64(&p.Size)
|
||||||
|
f.isdir = false
|
||||||
|
}
|
||||||
|
|
||||||
|
files = append(files, *f)
|
||||||
|
}
|
||||||
|
|
||||||
|
r.Props = nil
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
err := c.propfind(path, false,
|
||||||
|
`<d:propfind xmlns:d='DAV:'>
|
||||||
|
<d:prop>
|
||||||
|
<d:displayname/>
|
||||||
|
<d:resourcetype/>
|
||||||
|
<d:getcontentlength/>
|
||||||
|
<d:getcontenttype/>
|
||||||
|
<d:getetag/>
|
||||||
|
<d:getlastmodified/>
|
||||||
|
</d:prop>
|
||||||
|
</d:propfind>`,
|
||||||
|
&response{},
|
||||||
|
parse)
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
if _, ok := err.(*os.PathError); !ok {
|
||||||
|
err = newPathErrorErr("ReadDir", path, err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return files, err
|
||||||
|
}
|
||||||
|
|
||||||
|
// Stat returns the file stats for a specified path
|
||||||
|
func (c *Client) Stat(path string) (os.FileInfo, error) {
|
||||||
|
var f *File
|
||||||
|
parse := func(resp interface{}) error {
|
||||||
|
r := resp.(*response)
|
||||||
|
if p := getProps(r, "200"); p != nil && f == nil {
|
||||||
|
f = new(File)
|
||||||
|
f.name = p.Name
|
||||||
|
f.path = path
|
||||||
|
f.etag = p.ETag
|
||||||
|
f.contentType = p.ContentType
|
||||||
|
|
||||||
|
if p.Type.Local == "collection" {
|
||||||
|
if !strings.HasSuffix(f.path, "/") {
|
||||||
|
f.path += "/"
|
||||||
|
}
|
||||||
|
f.size = 0
|
||||||
|
f.modified = time.Unix(0, 0)
|
||||||
|
f.isdir = true
|
||||||
|
} else {
|
||||||
|
f.size = parseInt64(&p.Size)
|
||||||
|
f.modified = parseModified(&p.Modified)
|
||||||
|
f.isdir = false
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
r.Props = nil
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
err := c.propfind(path, true,
|
||||||
|
`<d:propfind xmlns:d='DAV:'>
|
||||||
|
<d:prop>
|
||||||
|
<d:displayname/>
|
||||||
|
<d:resourcetype/>
|
||||||
|
<d:getcontentlength/>
|
||||||
|
<d:getcontenttype/>
|
||||||
|
<d:getetag/>
|
||||||
|
<d:getlastmodified/>
|
||||||
|
</d:prop>
|
||||||
|
</d:propfind>`,
|
||||||
|
&response{},
|
||||||
|
parse)
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
if _, ok := err.(*os.PathError); !ok {
|
||||||
|
err = newPathErrorErr("ReadDir", path, err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return f, err
|
||||||
|
}
|
||||||
|
|
||||||
|
// Remove removes a remote file
|
||||||
|
func (c *Client) Remove(path string) error {
|
||||||
|
return c.RemoveAll(path)
|
||||||
|
}
|
||||||
|
|
||||||
|
// RemoveAll removes remote files
|
||||||
|
func (c *Client) RemoveAll(path string) error {
|
||||||
|
rs, err := c.req("DELETE", path, nil, nil)
|
||||||
|
if err != nil {
|
||||||
|
return newPathError("Remove", path, 400)
|
||||||
|
}
|
||||||
|
err = rs.Body.Close()
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
if rs.StatusCode == 200 || rs.StatusCode == 204 || rs.StatusCode == 404 {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
return newPathError("Remove", path, rs.StatusCode)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Mkdir makes a directory
|
||||||
|
func (c *Client) Mkdir(path string, _ os.FileMode) (err error) {
|
||||||
|
path = FixSlashes(path)
|
||||||
|
status, err := c.mkcol(path)
|
||||||
|
if err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if status == 201 {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
return newPathError("Mkdir", path, status)
|
||||||
|
}
|
||||||
|
|
||||||
|
// MkdirAll like mkdir -p, but for webdav
|
||||||
|
func (c *Client) MkdirAll(path string, _ os.FileMode) (err error) {
|
||||||
|
path = FixSlashes(path)
|
||||||
|
status, err := c.mkcol(path)
|
||||||
|
if err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if status == 201 {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
if status == 409 {
|
||||||
|
paths := strings.Split(path, "/")
|
||||||
|
sub := "/"
|
||||||
|
for _, e := range paths {
|
||||||
|
if e == "" {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
sub += e + "/"
|
||||||
|
status, err = c.mkcol(sub)
|
||||||
|
if err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if status != 201 {
|
||||||
|
return newPathError("MkdirAll", sub, status)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
return newPathError("MkdirAll", path, status)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Rename moves a file from A to B
|
||||||
|
func (c *Client) Rename(oldpath, newpath string, overwrite bool) error {
|
||||||
|
return c.copymove("MOVE", oldpath, newpath, overwrite)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Copy copies a file from A to B
|
||||||
|
func (c *Client) Copy(oldpath, newpath string, overwrite bool) error {
|
||||||
|
return c.copymove("COPY", oldpath, newpath, overwrite)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Read reads the contents of a remote file
|
||||||
|
func (c *Client) Read(path string) ([]byte, error) {
|
||||||
|
var stream io.ReadCloser
|
||||||
|
var err error
|
||||||
|
|
||||||
|
if stream, _, err = c.ReadStream(path, nil); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
defer stream.Close()
|
||||||
|
|
||||||
|
buf := new(bytes.Buffer)
|
||||||
|
_, err = buf.ReadFrom(stream)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return buf.Bytes(), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// ReadStream reads the stream for a given path
|
||||||
|
func (c *Client) ReadStream(path string, callback func(rq *http.Request)) (io.ReadCloser, http.Header, error) {
|
||||||
|
rs, err := c.req("GET", path, nil, callback)
|
||||||
|
if err != nil {
|
||||||
|
return nil, nil, newPathErrorErr("ReadStream", path, err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if rs.StatusCode < 400 {
|
||||||
|
return rs.Body, rs.Header, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
rs.Body.Close()
|
||||||
|
return nil, nil, newPathError("ReadStream", path, rs.StatusCode)
|
||||||
|
}
|
||||||
|
|
||||||
|
// ReadStreamRange reads the stream representing a subset of bytes for a given path,
|
||||||
|
// utilizing HTTP Range Requests if the server supports it.
|
||||||
|
// The range is expressed as offset from the start of the file and length, for example
|
||||||
|
// offset=10, length=10 will return bytes 10 through 19.
|
||||||
|
//
|
||||||
|
// If the server does not support partial content requests and returns full content instead,
|
||||||
|
// this function will emulate the behavior by skipping `offset` bytes and limiting the result
|
||||||
|
// to `length`.
|
||||||
|
func (c *Client) ReadStreamRange(path string, offset, length int64) (io.ReadCloser, error) {
|
||||||
|
rs, err := c.req("GET", path, nil, func(r *http.Request) {
|
||||||
|
r.Header.Add("Range", fmt.Sprintf("bytes=%v-%v", offset, offset+length-1))
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
return nil, newPathErrorErr("ReadStreamRange", path, err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if rs.StatusCode == http.StatusPartialContent {
|
||||||
|
// server supported partial content, return as-is.
|
||||||
|
return rs.Body, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// server returned success, but did not support partial content, so we have the whole
|
||||||
|
// stream in rs.Body
|
||||||
|
if rs.StatusCode == 200 {
|
||||||
|
// discard first 'offset' bytes.
|
||||||
|
if _, err := io.Copy(io.Discard, io.LimitReader(rs.Body, offset)); err != nil {
|
||||||
|
return nil, newPathErrorErr("ReadStreamRange", path, err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// return a io.ReadCloser that is limited to `length` bytes.
|
||||||
|
return &limitedReadCloser{rs.Body, int(length)}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
rs.Body.Close()
|
||||||
|
return nil, newPathError("ReadStream", path, rs.StatusCode)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Write writes data to a given path
|
||||||
|
func (c *Client) Write(path string, data []byte, _ os.FileMode) (err error) {
|
||||||
|
s, err := c.put(path, bytes.NewReader(data), nil)
|
||||||
|
if err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
switch s {
|
||||||
|
|
||||||
|
case 200, 201, 204:
|
||||||
|
return nil
|
||||||
|
|
||||||
|
case 409:
|
||||||
|
err = c.createParentCollection(path)
|
||||||
|
if err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
s, err = c.put(path, bytes.NewReader(data), nil)
|
||||||
|
if err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if s == 200 || s == 201 || s == 204 {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return newPathError("Write", path, s)
|
||||||
|
}
|
||||||
|
|
||||||
|
// WriteStream writes a stream
|
||||||
|
func (c *Client) WriteStream(path string, stream io.Reader, _ os.FileMode, callback func(r *http.Request)) (err error) {
|
||||||
|
|
||||||
|
err = c.createParentCollection(path)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
s, err := c.put(path, stream, callback)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
switch s {
|
||||||
|
case 200, 201, 204:
|
||||||
|
return nil
|
||||||
|
|
||||||
|
default:
|
||||||
|
return newPathError("WriteStream", path, s)
|
||||||
|
}
|
||||||
|
}
|
103
pkg/gowebdav/cmd/gowebdav/README.md
Normal file
103
pkg/gowebdav/cmd/gowebdav/README.md
Normal file
@ -0,0 +1,103 @@
|
|||||||
|
# Description
|
||||||
|
Command line tool for [gowebdav](https://github.com/studio-b12/gowebdav) library.
|
||||||
|
|
||||||
|
# Prerequisites
|
||||||
|
## Software
|
||||||
|
* **OS**: all, which are supported by `Golang`
|
||||||
|
* **Golang**: version 1.x
|
||||||
|
* **Git**: version 2.14.2 at higher (required to install via `go get`)
|
||||||
|
|
||||||
|
# Install
|
||||||
|
```sh
|
||||||
|
go get -u github.com/studio-b12/gowebdav/cmd/gowebdav
|
||||||
|
```
|
||||||
|
|
||||||
|
# Usage
|
||||||
|
It is recommended to set following environment variables to improve your experience with this tool:
|
||||||
|
* `ROOT` is an URL of target WebDAV server (e.g. `https://webdav.mydomain.me/user_root_folder`)
|
||||||
|
* `USER` is a login to connect to specified server (e.g. `user`)
|
||||||
|
* `PASSWORD` is a password to connect to specified server (e.g. `p@s$w0rD`)
|
||||||
|
|
||||||
|
In following examples we suppose that:
|
||||||
|
* environment variable `ROOT` is set to `https://webdav.mydomain.me/ufolder`
|
||||||
|
* environment variable `USER` is set to `user`
|
||||||
|
* environment variable `PASSWORD` is set `p@s$w0rD`
|
||||||
|
* folder `/ufolder/temp` exists on the server
|
||||||
|
* file `/ufolder/temp/file.txt` exists on the server
|
||||||
|
* file `/ufolder/temp/document.rtf` exists on the server
|
||||||
|
* file `/tmp/webdav/to_upload.txt` exists on the local machine
|
||||||
|
* folder `/tmp/webdav/` is used to download files from the server
|
||||||
|
|
||||||
|
## Examples
|
||||||
|
|
||||||
|
#### Get content of specified folder
|
||||||
|
```sh
|
||||||
|
gowebdav -X LS temp
|
||||||
|
```
|
||||||
|
|
||||||
|
#### Get info about file/folder
|
||||||
|
```sh
|
||||||
|
gowebdav -X STAT temp
|
||||||
|
gowebdav -X STAT temp/file.txt
|
||||||
|
```
|
||||||
|
|
||||||
|
#### Create folder on the remote server
|
||||||
|
```sh
|
||||||
|
gowebdav -X MKDIR temp2
|
||||||
|
gowebdav -X MKDIRALL all/folders/which-you-want/to_create
|
||||||
|
```
|
||||||
|
|
||||||
|
#### Download file
|
||||||
|
```sh
|
||||||
|
gowebdav -X GET temp/document.rtf /tmp/webdav/document.rtf
|
||||||
|
```
|
||||||
|
|
||||||
|
You may do not specify target local path, in this case file will be downloaded to the current folder with the
|
||||||
|
|
||||||
|
#### Upload file
|
||||||
|
```sh
|
||||||
|
gowebdav -X PUT temp/uploaded.txt /tmp/webdav/to_upload.txt
|
||||||
|
```
|
||||||
|
|
||||||
|
#### Move file on the remote server
|
||||||
|
```sh
|
||||||
|
gowebdav -X MV temp/file.txt temp/moved_file.txt
|
||||||
|
```
|
||||||
|
|
||||||
|
#### Copy file to another location
|
||||||
|
```sh
|
||||||
|
gowebdav -X MV temp/file.txt temp/file-copy.txt
|
||||||
|
```
|
||||||
|
|
||||||
|
#### Delete file from the remote server
|
||||||
|
```sh
|
||||||
|
gowebdav -X DEL temp/file.txt
|
||||||
|
```
|
||||||
|
|
||||||
|
# Wrapper script
|
||||||
|
|
||||||
|
You can create wrapper script for your server (via `$EDITOR ./dav && chmod a+x ./dav`) and add following content to it:
|
||||||
|
```sh
|
||||||
|
#!/bin/sh
|
||||||
|
|
||||||
|
ROOT="https://my.dav.server/" \
|
||||||
|
USER="foo" \
|
||||||
|
PASSWORD="$(pass dav/foo@my.dav.server)" \
|
||||||
|
gowebdav $@
|
||||||
|
```
|
||||||
|
|
||||||
|
It allows you to use [pass](https://www.passwordstore.org/ "the standard unix password manager") or similar tools to retrieve the password.
|
||||||
|
|
||||||
|
## Examples
|
||||||
|
|
||||||
|
Using the `dav` wrapper:
|
||||||
|
|
||||||
|
```sh
|
||||||
|
$ ./dav -X LS /
|
||||||
|
|
||||||
|
$ echo hi dav! > hello && ./dav -X PUT /hello
|
||||||
|
$ ./dav -X STAT /hello
|
||||||
|
$ ./dav -X PUT /hello_dav hello
|
||||||
|
$ ./dav -X GET /hello_dav
|
||||||
|
$ ./dav -X GET /hello_dav hello.txt
|
||||||
|
```
|
263
pkg/gowebdav/cmd/gowebdav/main.go
Normal file
263
pkg/gowebdav/cmd/gowebdav/main.go
Normal file
@ -0,0 +1,263 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"errors"
|
||||||
|
"flag"
|
||||||
|
"fmt"
|
||||||
|
"io"
|
||||||
|
"io/fs"
|
||||||
|
"os"
|
||||||
|
"os/user"
|
||||||
|
"path"
|
||||||
|
"path/filepath"
|
||||||
|
"runtime"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
d "github.com/Xhofe/alist/pkg/gowebdav"
|
||||||
|
)
|
||||||
|
|
||||||
|
func main() {
|
||||||
|
root := flag.String("root", os.Getenv("ROOT"), "WebDAV Endpoint [ENV.ROOT]")
|
||||||
|
user := flag.String("user", os.Getenv("USER"), "User [ENV.USER]")
|
||||||
|
password := flag.String("pw", os.Getenv("PASSWORD"), "Password [ENV.PASSWORD]")
|
||||||
|
netrc := flag.String("netrc-file", filepath.Join(getHome(), ".netrc"), "read login from netrc file")
|
||||||
|
method := flag.String("X", "", `Method:
|
||||||
|
LS <PATH>
|
||||||
|
STAT <PATH>
|
||||||
|
|
||||||
|
MKDIR <PATH>
|
||||||
|
MKDIRALL <PATH>
|
||||||
|
|
||||||
|
GET <PATH> [<FILE>]
|
||||||
|
PUT <PATH> [<FILE>]
|
||||||
|
|
||||||
|
MV <OLD> <NEW>
|
||||||
|
CP <OLD> <NEW>
|
||||||
|
|
||||||
|
DEL <PATH>
|
||||||
|
`)
|
||||||
|
flag.Parse()
|
||||||
|
|
||||||
|
if *root == "" {
|
||||||
|
fail("Set WebDAV ROOT")
|
||||||
|
}
|
||||||
|
|
||||||
|
if argsLength := len(flag.Args()); argsLength == 0 || argsLength > 2 {
|
||||||
|
fail("Unsupported arguments")
|
||||||
|
}
|
||||||
|
|
||||||
|
if *password == "" {
|
||||||
|
if u, p := d.ReadConfig(*root, *netrc); u != "" && p != "" {
|
||||||
|
user = &u
|
||||||
|
password = &p
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
c := d.NewClient(*root, *user, *password)
|
||||||
|
|
||||||
|
cmd := getCmd(*method)
|
||||||
|
|
||||||
|
if e := cmd(c, flag.Arg(0), flag.Arg(1)); e != nil {
|
||||||
|
fail(e)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func fail(err interface{}) {
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println(err)
|
||||||
|
}
|
||||||
|
os.Exit(-1)
|
||||||
|
}
|
||||||
|
|
||||||
|
func getHome() string {
|
||||||
|
u, e := user.Current()
|
||||||
|
if e != nil {
|
||||||
|
return os.Getenv("HOME")
|
||||||
|
}
|
||||||
|
|
||||||
|
if u != nil {
|
||||||
|
return u.HomeDir
|
||||||
|
}
|
||||||
|
|
||||||
|
switch runtime.GOOS {
|
||||||
|
case "windows":
|
||||||
|
return ""
|
||||||
|
default:
|
||||||
|
return "~/"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func getCmd(method string) func(c *d.Client, p0, p1 string) error {
|
||||||
|
switch strings.ToUpper(method) {
|
||||||
|
case "LS", "LIST", "PROPFIND":
|
||||||
|
return cmdLs
|
||||||
|
|
||||||
|
case "STAT":
|
||||||
|
return cmdStat
|
||||||
|
|
||||||
|
case "GET", "PULL", "READ":
|
||||||
|
return cmdGet
|
||||||
|
|
||||||
|
case "DELETE", "RM", "DEL":
|
||||||
|
return cmdRm
|
||||||
|
|
||||||
|
case "MKCOL", "MKDIR":
|
||||||
|
return cmdMkdir
|
||||||
|
|
||||||
|
case "MKCOLALL", "MKDIRALL", "MKDIRP":
|
||||||
|
return cmdMkdirAll
|
||||||
|
|
||||||
|
case "RENAME", "MV", "MOVE":
|
||||||
|
return cmdMv
|
||||||
|
|
||||||
|
case "COPY", "CP":
|
||||||
|
return cmdCp
|
||||||
|
|
||||||
|
case "PUT", "PUSH", "WRITE":
|
||||||
|
return cmdPut
|
||||||
|
|
||||||
|
default:
|
||||||
|
return func(c *d.Client, p0, p1 string) (err error) {
|
||||||
|
return errors.New("Unsupported method: " + method)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func cmdLs(c *d.Client, p0, _ string) (err error) {
|
||||||
|
files, err := c.ReadDir(p0)
|
||||||
|
if err == nil {
|
||||||
|
fmt.Println(fmt.Sprintf("ReadDir: '%s' entries: %d ", p0, len(files)))
|
||||||
|
for _, f := range files {
|
||||||
|
fmt.Println(f)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func cmdStat(c *d.Client, p0, _ string) (err error) {
|
||||||
|
file, err := c.Stat(p0)
|
||||||
|
if err == nil {
|
||||||
|
fmt.Println(file)
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func cmdGet(c *d.Client, p0, p1 string) (err error) {
|
||||||
|
bytes, err := c.Read(p0)
|
||||||
|
if err == nil {
|
||||||
|
if p1 == "" {
|
||||||
|
p1 = filepath.Join(".", p0)
|
||||||
|
}
|
||||||
|
err = writeFile(p1, bytes, 0644)
|
||||||
|
if err == nil {
|
||||||
|
fmt.Println(fmt.Sprintf("Written %d bytes to: %s", len(bytes), p1))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func cmdRm(c *d.Client, p0, _ string) (err error) {
|
||||||
|
if err = c.Remove(p0); err == nil {
|
||||||
|
fmt.Println("Remove: " + p0)
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func cmdMkdir(c *d.Client, p0, _ string) (err error) {
|
||||||
|
if err = c.Mkdir(p0, 0755); err == nil {
|
||||||
|
fmt.Println("Mkdir: " + p0)
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func cmdMkdirAll(c *d.Client, p0, _ string) (err error) {
|
||||||
|
if err = c.MkdirAll(p0, 0755); err == nil {
|
||||||
|
fmt.Println("MkdirAll: " + p0)
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func cmdMv(c *d.Client, p0, p1 string) (err error) {
|
||||||
|
if err = c.Rename(p0, p1, true); err == nil {
|
||||||
|
fmt.Println("Rename: " + p0 + " -> " + p1)
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func cmdCp(c *d.Client, p0, p1 string) (err error) {
|
||||||
|
if err = c.Copy(p0, p1, true); err == nil {
|
||||||
|
fmt.Println("Copy: " + p0 + " -> " + p1)
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func cmdPut(c *d.Client, p0, p1 string) (err error) {
|
||||||
|
if p1 == "" {
|
||||||
|
p1 = path.Join(".", p0)
|
||||||
|
} else {
|
||||||
|
var fi fs.FileInfo
|
||||||
|
fi, err = c.Stat(p0)
|
||||||
|
if err != nil && !d.IsErrNotFound(err) {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if !d.IsErrNotFound(err) && fi.IsDir() {
|
||||||
|
p0 = path.Join(p0, p1)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
stream, err := getStream(p1)
|
||||||
|
if err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
defer stream.Close()
|
||||||
|
|
||||||
|
if err = c.WriteStream(p0, stream, 0644, nil); err == nil {
|
||||||
|
fmt.Println("Put: " + p1 + " -> " + p0)
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func writeFile(path string, bytes []byte, mode os.FileMode) error {
|
||||||
|
parent := filepath.Dir(path)
|
||||||
|
if _, e := os.Stat(parent); os.IsNotExist(e) {
|
||||||
|
if e := os.MkdirAll(parent, os.ModePerm); e != nil {
|
||||||
|
return e
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
f, err := os.Create(path)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
defer f.Close()
|
||||||
|
|
||||||
|
_, err = f.Write(bytes)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
func getStream(pathOrString string) (io.ReadCloser, error) {
|
||||||
|
|
||||||
|
fi, err := os.Stat(pathOrString)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
if fi.IsDir() {
|
||||||
|
return nil, &os.PathError{
|
||||||
|
Op: "Open",
|
||||||
|
Path: pathOrString,
|
||||||
|
Err: errors.New("Path: '" + pathOrString + "' is a directory"),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
f, err := os.Open(pathOrString)
|
||||||
|
if err == nil {
|
||||||
|
return f, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil, &os.PathError{
|
||||||
|
Op: "Open",
|
||||||
|
Path: pathOrString,
|
||||||
|
Err: err,
|
||||||
|
}
|
||||||
|
}
|
146
pkg/gowebdav/digestAuth.go
Normal file
146
pkg/gowebdav/digestAuth.go
Normal file
@ -0,0 +1,146 @@
|
|||||||
|
package gowebdav
|
||||||
|
|
||||||
|
import (
|
||||||
|
"crypto/md5"
|
||||||
|
"crypto/rand"
|
||||||
|
"encoding/hex"
|
||||||
|
"fmt"
|
||||||
|
"io"
|
||||||
|
"net/http"
|
||||||
|
"strings"
|
||||||
|
)
|
||||||
|
|
||||||
|
// DigestAuth structure holds our credentials
|
||||||
|
type DigestAuth struct {
|
||||||
|
user string
|
||||||
|
pw string
|
||||||
|
digestParts map[string]string
|
||||||
|
}
|
||||||
|
|
||||||
|
// Type identifies the DigestAuthenticator
|
||||||
|
func (d *DigestAuth) Type() string {
|
||||||
|
return "DigestAuth"
|
||||||
|
}
|
||||||
|
|
||||||
|
// User holds the DigestAuth username
|
||||||
|
func (d *DigestAuth) User() string {
|
||||||
|
return d.user
|
||||||
|
}
|
||||||
|
|
||||||
|
// Pass holds the DigestAuth password
|
||||||
|
func (d *DigestAuth) Pass() string {
|
||||||
|
return d.pw
|
||||||
|
}
|
||||||
|
|
||||||
|
// Authorize the current request
|
||||||
|
func (d *DigestAuth) Authorize(req *http.Request, method string, path string) {
|
||||||
|
d.digestParts["uri"] = path
|
||||||
|
d.digestParts["method"] = method
|
||||||
|
d.digestParts["username"] = d.user
|
||||||
|
d.digestParts["password"] = d.pw
|
||||||
|
req.Header.Set("Authorization", getDigestAuthorization(d.digestParts))
|
||||||
|
}
|
||||||
|
|
||||||
|
func digestParts(resp *http.Response) map[string]string {
|
||||||
|
result := map[string]string{}
|
||||||
|
if len(resp.Header["Www-Authenticate"]) > 0 {
|
||||||
|
wantedHeaders := []string{"nonce", "realm", "qop", "opaque", "algorithm", "entityBody"}
|
||||||
|
responseHeaders := strings.Split(resp.Header["Www-Authenticate"][0], ",")
|
||||||
|
for _, r := range responseHeaders {
|
||||||
|
for _, w := range wantedHeaders {
|
||||||
|
if strings.Contains(r, w) {
|
||||||
|
result[w] = strings.Trim(
|
||||||
|
strings.SplitN(r, `=`, 2)[1],
|
||||||
|
`"`,
|
||||||
|
)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return result
|
||||||
|
}
|
||||||
|
|
||||||
|
func getMD5(text string) string {
|
||||||
|
hasher := md5.New()
|
||||||
|
hasher.Write([]byte(text))
|
||||||
|
return hex.EncodeToString(hasher.Sum(nil))
|
||||||
|
}
|
||||||
|
|
||||||
|
func getCnonce() string {
|
||||||
|
b := make([]byte, 8)
|
||||||
|
io.ReadFull(rand.Reader, b)
|
||||||
|
return fmt.Sprintf("%x", b)[:16]
|
||||||
|
}
|
||||||
|
|
||||||
|
func getDigestAuthorization(digestParts map[string]string) string {
|
||||||
|
d := digestParts
|
||||||
|
// These are the correct ha1 and ha2 for qop=auth. We should probably check for other types of qop.
|
||||||
|
|
||||||
|
var (
|
||||||
|
ha1 string
|
||||||
|
ha2 string
|
||||||
|
nonceCount = 00000001
|
||||||
|
cnonce = getCnonce()
|
||||||
|
response string
|
||||||
|
)
|
||||||
|
|
||||||
|
// 'ha1' value depends on value of "algorithm" field
|
||||||
|
switch d["algorithm"] {
|
||||||
|
case "MD5", "":
|
||||||
|
ha1 = getMD5(d["username"] + ":" + d["realm"] + ":" + d["password"])
|
||||||
|
case "MD5-sess":
|
||||||
|
ha1 = getMD5(
|
||||||
|
fmt.Sprintf("%s:%v:%s",
|
||||||
|
getMD5(d["username"]+":"+d["realm"]+":"+d["password"]),
|
||||||
|
nonceCount,
|
||||||
|
cnonce,
|
||||||
|
),
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
// 'ha2' value depends on value of "qop" field
|
||||||
|
switch d["qop"] {
|
||||||
|
case "auth", "":
|
||||||
|
ha2 = getMD5(d["method"] + ":" + d["uri"])
|
||||||
|
case "auth-int":
|
||||||
|
if d["entityBody"] != "" {
|
||||||
|
ha2 = getMD5(d["method"] + ":" + d["uri"] + ":" + getMD5(d["entityBody"]))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 'response' value depends on value of "qop" field
|
||||||
|
switch d["qop"] {
|
||||||
|
case "":
|
||||||
|
response = getMD5(
|
||||||
|
fmt.Sprintf("%s:%s:%s",
|
||||||
|
ha1,
|
||||||
|
d["nonce"],
|
||||||
|
ha2,
|
||||||
|
),
|
||||||
|
)
|
||||||
|
case "auth", "auth-int":
|
||||||
|
response = getMD5(
|
||||||
|
fmt.Sprintf("%s:%s:%v:%s:%s:%s",
|
||||||
|
ha1,
|
||||||
|
d["nonce"],
|
||||||
|
nonceCount,
|
||||||
|
cnonce,
|
||||||
|
d["qop"],
|
||||||
|
ha2,
|
||||||
|
),
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
authorization := fmt.Sprintf(`Digest username="%s", realm="%s", nonce="%s", uri="%s", nc=%v, cnonce="%s", response="%s"`,
|
||||||
|
d["username"], d["realm"], d["nonce"], d["uri"], nonceCount, cnonce, response)
|
||||||
|
|
||||||
|
if d["qop"] != "" {
|
||||||
|
authorization += fmt.Sprintf(`, qop=%s`, d["qop"])
|
||||||
|
}
|
||||||
|
|
||||||
|
if d["opaque"] != "" {
|
||||||
|
authorization += fmt.Sprintf(`, opaque="%s"`, d["opaque"])
|
||||||
|
}
|
||||||
|
|
||||||
|
return authorization
|
||||||
|
}
|
3
pkg/gowebdav/doc.go
Normal file
3
pkg/gowebdav/doc.go
Normal file
@ -0,0 +1,3 @@
|
|||||||
|
// Package gowebdav is a WebDAV client library with a command line tool
|
||||||
|
// included.
|
||||||
|
package gowebdav
|
49
pkg/gowebdav/errors.go
Normal file
49
pkg/gowebdav/errors.go
Normal file
@ -0,0 +1,49 @@
|
|||||||
|
package gowebdav
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"os"
|
||||||
|
)
|
||||||
|
|
||||||
|
// StatusError implements error and wraps
|
||||||
|
// an erroneous status code.
|
||||||
|
type StatusError struct {
|
||||||
|
Status int
|
||||||
|
}
|
||||||
|
|
||||||
|
func (se StatusError) Error() string {
|
||||||
|
return fmt.Sprintf("%d", se.Status)
|
||||||
|
}
|
||||||
|
|
||||||
|
// IsErrCode returns true if the given error
|
||||||
|
// is an os.PathError wrapping a StatusError
|
||||||
|
// with the given status code.
|
||||||
|
func IsErrCode(err error, code int) bool {
|
||||||
|
if pe, ok := err.(*os.PathError); ok {
|
||||||
|
se, ok := pe.Err.(StatusError)
|
||||||
|
return ok && se.Status == code
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
// IsErrNotFound is shorthand for IsErrCode
|
||||||
|
// for status 404.
|
||||||
|
func IsErrNotFound(err error) bool {
|
||||||
|
return IsErrCode(err, 404)
|
||||||
|
}
|
||||||
|
|
||||||
|
func newPathError(op string, path string, statusCode int) error {
|
||||||
|
return &os.PathError{
|
||||||
|
Op: op,
|
||||||
|
Path: path,
|
||||||
|
Err: StatusError{statusCode},
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func newPathErrorErr(op string, path string, err error) error {
|
||||||
|
return &os.PathError{
|
||||||
|
Op: op,
|
||||||
|
Path: path,
|
||||||
|
Err: err,
|
||||||
|
}
|
||||||
|
}
|
77
pkg/gowebdav/file.go
Normal file
77
pkg/gowebdav/file.go
Normal file
@ -0,0 +1,77 @@
|
|||||||
|
package gowebdav
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"os"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
// File is our structure for a given file
|
||||||
|
type File struct {
|
||||||
|
path string
|
||||||
|
name string
|
||||||
|
contentType string
|
||||||
|
size int64
|
||||||
|
modified time.Time
|
||||||
|
etag string
|
||||||
|
isdir bool
|
||||||
|
}
|
||||||
|
|
||||||
|
// Path returns the full path of a file
|
||||||
|
func (f File) Path() string {
|
||||||
|
return f.path
|
||||||
|
}
|
||||||
|
|
||||||
|
// Name returns the name of a file
|
||||||
|
func (f File) Name() string {
|
||||||
|
return f.name
|
||||||
|
}
|
||||||
|
|
||||||
|
// ContentType returns the content type of a file
|
||||||
|
func (f File) ContentType() string {
|
||||||
|
return f.contentType
|
||||||
|
}
|
||||||
|
|
||||||
|
// Size returns the size of a file
|
||||||
|
func (f File) Size() int64 {
|
||||||
|
return f.size
|
||||||
|
}
|
||||||
|
|
||||||
|
// Mode will return the mode of a given file
|
||||||
|
func (f File) Mode() os.FileMode {
|
||||||
|
// TODO check webdav perms
|
||||||
|
if f.isdir {
|
||||||
|
return 0775 | os.ModeDir
|
||||||
|
}
|
||||||
|
|
||||||
|
return 0664
|
||||||
|
}
|
||||||
|
|
||||||
|
// ModTime returns the modified time of a file
|
||||||
|
func (f File) ModTime() time.Time {
|
||||||
|
return f.modified
|
||||||
|
}
|
||||||
|
|
||||||
|
// ETag returns the ETag of a file
|
||||||
|
func (f File) ETag() string {
|
||||||
|
return f.etag
|
||||||
|
}
|
||||||
|
|
||||||
|
// IsDir let us see if a given file is a directory or not
|
||||||
|
func (f File) IsDir() bool {
|
||||||
|
return f.isdir
|
||||||
|
}
|
||||||
|
|
||||||
|
// Sys ????
|
||||||
|
func (f File) Sys() interface{} {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// String lets us see file information
|
||||||
|
func (f File) String() string {
|
||||||
|
if f.isdir {
|
||||||
|
return fmt.Sprintf("Dir : '%s' - '%s'", f.path, f.name)
|
||||||
|
}
|
||||||
|
|
||||||
|
return fmt.Sprintf("File: '%s' SIZE: %d MODIFIED: %s ETAG: %s CTYPE: %s", f.path, f.size, f.modified.String(), f.etag, f.contentType)
|
||||||
|
}
|
54
pkg/gowebdav/netrc.go
Normal file
54
pkg/gowebdav/netrc.go
Normal file
@ -0,0 +1,54 @@
|
|||||||
|
package gowebdav
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bufio"
|
||||||
|
"fmt"
|
||||||
|
"net/url"
|
||||||
|
"os"
|
||||||
|
"regexp"
|
||||||
|
"strings"
|
||||||
|
)
|
||||||
|
|
||||||
|
func parseLine(s string) (login, pass string) {
|
||||||
|
fields := strings.Fields(s)
|
||||||
|
for i, f := range fields {
|
||||||
|
if f == "login" {
|
||||||
|
login = fields[i+1]
|
||||||
|
}
|
||||||
|
if f == "password" {
|
||||||
|
pass = fields[i+1]
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return login, pass
|
||||||
|
}
|
||||||
|
|
||||||
|
// ReadConfig reads login and password configuration from ~/.netrc
|
||||||
|
// machine foo.com login username password 123456
|
||||||
|
func ReadConfig(uri, netrc string) (string, string) {
|
||||||
|
u, err := url.Parse(uri)
|
||||||
|
if err != nil {
|
||||||
|
return "", ""
|
||||||
|
}
|
||||||
|
|
||||||
|
file, err := os.Open(netrc)
|
||||||
|
if err != nil {
|
||||||
|
return "", ""
|
||||||
|
}
|
||||||
|
defer file.Close()
|
||||||
|
|
||||||
|
re := fmt.Sprintf(`^.*machine %s.*$`, u.Host)
|
||||||
|
scanner := bufio.NewScanner(file)
|
||||||
|
for scanner.Scan() {
|
||||||
|
s := scanner.Text()
|
||||||
|
|
||||||
|
matched, err := regexp.MatchString(re, s)
|
||||||
|
if err != nil {
|
||||||
|
return "", ""
|
||||||
|
}
|
||||||
|
if matched {
|
||||||
|
return parseLine(s)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return "", ""
|
||||||
|
}
|
214
pkg/gowebdav/requests.go
Normal file
214
pkg/gowebdav/requests.go
Normal file
@ -0,0 +1,214 @@
|
|||||||
|
package gowebdav
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
"fmt"
|
||||||
|
"io"
|
||||||
|
"net/http"
|
||||||
|
"path"
|
||||||
|
"strings"
|
||||||
|
)
|
||||||
|
|
||||||
|
func (c *Client) req(method, path string, body io.Reader, intercept func(*http.Request)) (req *http.Response, err error) {
|
||||||
|
var r *http.Request
|
||||||
|
var retryBuf io.Reader
|
||||||
|
|
||||||
|
if body != nil {
|
||||||
|
// If the authorization fails, we will need to restart reading
|
||||||
|
// from the passed body stream.
|
||||||
|
// When body is seekable, use seek to reset the streams
|
||||||
|
// cursor to the start.
|
||||||
|
// Otherwise, copy the stream into a buffer while uploading
|
||||||
|
// and use the buffers content on retry.
|
||||||
|
if sk, ok := body.(io.Seeker); ok {
|
||||||
|
if _, err = sk.Seek(0, io.SeekStart); err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
retryBuf = body
|
||||||
|
} else {
|
||||||
|
buff := &bytes.Buffer{}
|
||||||
|
retryBuf = buff
|
||||||
|
body = io.TeeReader(body, buff)
|
||||||
|
}
|
||||||
|
r, err = http.NewRequest(method, PathEscape(Join(c.root, path)), body)
|
||||||
|
} else {
|
||||||
|
r, err = http.NewRequest(method, PathEscape(Join(c.root, path)), nil)
|
||||||
|
}
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
for k, vals := range c.headers {
|
||||||
|
for _, v := range vals {
|
||||||
|
r.Header.Add(k, v)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// make sure we read 'c.auth' only once since it will be substituted below
|
||||||
|
// and that is unsafe to do when multiple goroutines are running at the same time.
|
||||||
|
c.authMutex.Lock()
|
||||||
|
auth := c.auth
|
||||||
|
c.authMutex.Unlock()
|
||||||
|
|
||||||
|
auth.Authorize(r, method, path)
|
||||||
|
|
||||||
|
if intercept != nil {
|
||||||
|
intercept(r)
|
||||||
|
}
|
||||||
|
|
||||||
|
if c.interceptor != nil {
|
||||||
|
c.interceptor(method, r)
|
||||||
|
}
|
||||||
|
|
||||||
|
rs, err := c.c.Do(r)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
if rs.StatusCode == 401 && auth.Type() == "NoAuth" {
|
||||||
|
wwwAuthenticateHeader := strings.ToLower(rs.Header.Get("Www-Authenticate"))
|
||||||
|
|
||||||
|
if strings.Index(wwwAuthenticateHeader, "digest") > -1 {
|
||||||
|
c.authMutex.Lock()
|
||||||
|
c.auth = &DigestAuth{auth.User(), auth.Pass(), digestParts(rs)}
|
||||||
|
c.authMutex.Unlock()
|
||||||
|
} else if strings.Index(wwwAuthenticateHeader, "basic") > -1 {
|
||||||
|
c.authMutex.Lock()
|
||||||
|
c.auth = &BasicAuth{auth.User(), auth.Pass()}
|
||||||
|
c.authMutex.Unlock()
|
||||||
|
} else {
|
||||||
|
return rs, newPathError("Authorize", c.root, rs.StatusCode)
|
||||||
|
}
|
||||||
|
|
||||||
|
// retryBuf will be nil if body was nil initially so no check
|
||||||
|
// for body == nil is required here.
|
||||||
|
return c.req(method, path, retryBuf, intercept)
|
||||||
|
} else if rs.StatusCode == 401 {
|
||||||
|
return rs, newPathError("Authorize", c.root, rs.StatusCode)
|
||||||
|
}
|
||||||
|
|
||||||
|
return rs, err
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *Client) mkcol(path string) (status int, err error) {
|
||||||
|
rs, err := c.req("MKCOL", path, nil, nil)
|
||||||
|
if err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
defer rs.Body.Close()
|
||||||
|
|
||||||
|
status = rs.StatusCode
|
||||||
|
if status == 405 {
|
||||||
|
status = 201
|
||||||
|
}
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *Client) options(path string) (*http.Response, error) {
|
||||||
|
return c.req("OPTIONS", path, nil, func(rq *http.Request) {
|
||||||
|
rq.Header.Add("Depth", "0")
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *Client) propfind(path string, self bool, body string, resp interface{}, parse func(resp interface{}) error) error {
|
||||||
|
rs, err := c.req("PROPFIND", path, strings.NewReader(body), func(rq *http.Request) {
|
||||||
|
if self {
|
||||||
|
rq.Header.Add("Depth", "0")
|
||||||
|
} else {
|
||||||
|
rq.Header.Add("Depth", "1")
|
||||||
|
}
|
||||||
|
rq.Header.Add("Content-Type", "application/xml;charset=UTF-8")
|
||||||
|
rq.Header.Add("Accept", "application/xml,text/xml")
|
||||||
|
rq.Header.Add("Accept-Charset", "utf-8")
|
||||||
|
// TODO add support for 'gzip,deflate;q=0.8,q=0.7'
|
||||||
|
rq.Header.Add("Accept-Encoding", "")
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
defer rs.Body.Close()
|
||||||
|
|
||||||
|
if rs.StatusCode != 207 {
|
||||||
|
return newPathError("PROPFIND", path, rs.StatusCode)
|
||||||
|
}
|
||||||
|
|
||||||
|
return parseXML(rs.Body, resp, parse)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *Client) doCopyMove(
|
||||||
|
method string,
|
||||||
|
oldpath string,
|
||||||
|
newpath string,
|
||||||
|
overwrite bool,
|
||||||
|
) (
|
||||||
|
status int,
|
||||||
|
r io.ReadCloser,
|
||||||
|
err error,
|
||||||
|
) {
|
||||||
|
rs, err := c.req(method, oldpath, nil, func(rq *http.Request) {
|
||||||
|
rq.Header.Add("Destination", PathEscape(Join(c.root, newpath)))
|
||||||
|
if overwrite {
|
||||||
|
rq.Header.Add("Overwrite", "T")
|
||||||
|
} else {
|
||||||
|
rq.Header.Add("Overwrite", "F")
|
||||||
|
}
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
status = rs.StatusCode
|
||||||
|
r = rs.Body
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *Client) copymove(method string, oldpath string, newpath string, overwrite bool) (err error) {
|
||||||
|
s, data, err := c.doCopyMove(method, oldpath, newpath, overwrite)
|
||||||
|
if err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if data != nil {
|
||||||
|
defer data.Close()
|
||||||
|
}
|
||||||
|
|
||||||
|
switch s {
|
||||||
|
case 201, 204:
|
||||||
|
return nil
|
||||||
|
|
||||||
|
case 207:
|
||||||
|
// TODO handle multistat errors, worst case ...
|
||||||
|
log(fmt.Sprintf(" TODO handle %s - %s multistatus result %s", method, oldpath, String(data)))
|
||||||
|
|
||||||
|
case 409:
|
||||||
|
err := c.createParentCollection(newpath)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
return c.copymove(method, oldpath, newpath, overwrite)
|
||||||
|
}
|
||||||
|
|
||||||
|
return newPathError(method, oldpath, s)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *Client) put(path string, stream io.Reader, callback func(r *http.Request)) (status int, err error) {
|
||||||
|
rs, err := c.req("PUT", path, stream, callback)
|
||||||
|
if err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
defer rs.Body.Close()
|
||||||
|
//all, _ := io.ReadAll(rs.Body)
|
||||||
|
//logrus.Debugln("put res: ", string(all))
|
||||||
|
status = rs.StatusCode
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *Client) createParentCollection(itemPath string) (err error) {
|
||||||
|
parentPath := path.Dir(itemPath)
|
||||||
|
if parentPath == "." || parentPath == "/" {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
return c.MkdirAll(parentPath, 0755)
|
||||||
|
}
|
118
pkg/gowebdav/utils.go
Normal file
118
pkg/gowebdav/utils.go
Normal file
@ -0,0 +1,118 @@
|
|||||||
|
package gowebdav
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
"encoding/xml"
|
||||||
|
"fmt"
|
||||||
|
"io"
|
||||||
|
"net/url"
|
||||||
|
"strconv"
|
||||||
|
"strings"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
func log(msg interface{}) {
|
||||||
|
fmt.Println(msg)
|
||||||
|
}
|
||||||
|
|
||||||
|
// PathEscape escapes all segments of a given path
|
||||||
|
func PathEscape(path string) string {
|
||||||
|
s := strings.Split(path, "/")
|
||||||
|
for i, e := range s {
|
||||||
|
s[i] = url.PathEscape(e)
|
||||||
|
}
|
||||||
|
return strings.Join(s, "/")
|
||||||
|
}
|
||||||
|
|
||||||
|
// FixSlash appends a trailing / to our string
|
||||||
|
func FixSlash(s string) string {
|
||||||
|
if !strings.HasSuffix(s, "/") {
|
||||||
|
s += "/"
|
||||||
|
}
|
||||||
|
return s
|
||||||
|
}
|
||||||
|
|
||||||
|
// FixSlashes appends and prepends a / if they are missing
|
||||||
|
func FixSlashes(s string) string {
|
||||||
|
if !strings.HasPrefix(s, "/") {
|
||||||
|
s = "/" + s
|
||||||
|
}
|
||||||
|
|
||||||
|
return FixSlash(s)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Join joins two paths
|
||||||
|
func Join(path0 string, path1 string) string {
|
||||||
|
return strings.TrimSuffix(path0, "/") + "/" + strings.TrimPrefix(path1, "/")
|
||||||
|
}
|
||||||
|
|
||||||
|
// String pulls a string out of our io.Reader
|
||||||
|
func String(r io.Reader) string {
|
||||||
|
buf := new(bytes.Buffer)
|
||||||
|
// TODO - make String return an error as well
|
||||||
|
_, _ = buf.ReadFrom(r)
|
||||||
|
return buf.String()
|
||||||
|
}
|
||||||
|
|
||||||
|
func parseUint(s *string) uint {
|
||||||
|
if n, e := strconv.ParseUint(*s, 10, 32); e == nil {
|
||||||
|
return uint(n)
|
||||||
|
}
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
|
||||||
|
func parseInt64(s *string) int64 {
|
||||||
|
if n, e := strconv.ParseInt(*s, 10, 64); e == nil {
|
||||||
|
return n
|
||||||
|
}
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
|
||||||
|
func parseModified(s *string) time.Time {
|
||||||
|
if t, e := time.Parse(time.RFC1123, *s); e == nil {
|
||||||
|
return t
|
||||||
|
}
|
||||||
|
return time.Unix(0, 0)
|
||||||
|
}
|
||||||
|
|
||||||
|
func parseXML(data io.Reader, resp interface{}, parse func(resp interface{}) error) error {
|
||||||
|
decoder := xml.NewDecoder(data)
|
||||||
|
for t, _ := decoder.Token(); t != nil; t, _ = decoder.Token() {
|
||||||
|
switch se := t.(type) {
|
||||||
|
case xml.StartElement:
|
||||||
|
if se.Name.Local == "response" {
|
||||||
|
if e := decoder.DecodeElement(resp, &se); e == nil {
|
||||||
|
if err := parse(resp); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// limitedReadCloser wraps a io.ReadCloser and limits the number of bytes that can be read from it.
|
||||||
|
type limitedReadCloser struct {
|
||||||
|
rc io.ReadCloser
|
||||||
|
remaining int
|
||||||
|
}
|
||||||
|
|
||||||
|
func (l *limitedReadCloser) Read(buf []byte) (int, error) {
|
||||||
|
if l.remaining <= 0 {
|
||||||
|
return 0, io.EOF
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(buf) > l.remaining {
|
||||||
|
buf = buf[0:l.remaining]
|
||||||
|
}
|
||||||
|
|
||||||
|
n, err := l.rc.Read(buf)
|
||||||
|
l.remaining -= n
|
||||||
|
|
||||||
|
return n, err
|
||||||
|
}
|
||||||
|
|
||||||
|
func (l *limitedReadCloser) Close() error {
|
||||||
|
return l.rc.Close()
|
||||||
|
}
|
67
pkg/gowebdav/utils_test.go
Normal file
67
pkg/gowebdav/utils_test.go
Normal file
@ -0,0 +1,67 @@
|
|||||||
|
package gowebdav
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"net/url"
|
||||||
|
"testing"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestJoin(t *testing.T) {
|
||||||
|
eq(t, "/", "", "")
|
||||||
|
eq(t, "/", "/", "/")
|
||||||
|
eq(t, "/foo", "", "/foo")
|
||||||
|
eq(t, "foo/foo", "foo/", "/foo")
|
||||||
|
eq(t, "foo/foo", "foo/", "foo")
|
||||||
|
}
|
||||||
|
|
||||||
|
func eq(t *testing.T, expected string, s0 string, s1 string) {
|
||||||
|
s := Join(s0, s1)
|
||||||
|
if s != expected {
|
||||||
|
t.Error("For", "'"+s0+"','"+s1+"'", "expeted", "'"+expected+"'", "got", "'"+s+"'")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func ExamplePathEscape() {
|
||||||
|
fmt.Println(PathEscape(""))
|
||||||
|
fmt.Println(PathEscape("/"))
|
||||||
|
fmt.Println(PathEscape("/web"))
|
||||||
|
fmt.Println(PathEscape("/web/"))
|
||||||
|
fmt.Println(PathEscape("/w e b/d a v/s%u&c#k:s/"))
|
||||||
|
|
||||||
|
// Output:
|
||||||
|
//
|
||||||
|
// /
|
||||||
|
// /web
|
||||||
|
// /web/
|
||||||
|
// /w%20e%20b/d%20a%20v/s%25u&c%23k:s/
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestEscapeURL(t *testing.T) {
|
||||||
|
ex := "https://foo.com/w%20e%20b/d%20a%20v/s%25u&c%23k:s/"
|
||||||
|
u, _ := url.Parse("https://foo.com" + PathEscape("/w e b/d a v/s%u&c#k:s/"))
|
||||||
|
if ex != u.String() {
|
||||||
|
t.Error("expected: " + ex + " got: " + u.String())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestFixSlashes(t *testing.T) {
|
||||||
|
expected := "/"
|
||||||
|
|
||||||
|
if got := FixSlashes(""); got != expected {
|
||||||
|
t.Errorf("expected: %q, got: %q", expected, got)
|
||||||
|
}
|
||||||
|
|
||||||
|
expected = "/path/"
|
||||||
|
|
||||||
|
if got := FixSlashes("path"); got != expected {
|
||||||
|
t.Errorf("expected: %q, got: %q", expected, got)
|
||||||
|
}
|
||||||
|
|
||||||
|
if got := FixSlashes("/path"); got != expected {
|
||||||
|
t.Errorf("expected: %q, got: %q", expected, got)
|
||||||
|
}
|
||||||
|
|
||||||
|
if got := FixSlashes("path/"); got != expected {
|
||||||
|
t.Errorf("expected: %q, got: %q", expected, got)
|
||||||
|
}
|
||||||
|
}
|
@ -4,9 +4,3 @@ import "embed"
|
|||||||
|
|
||||||
//go:embed *
|
//go:embed *
|
||||||
var Public embed.FS
|
var Public embed.FS
|
||||||
|
|
||||||
////go:embed index.html
|
|
||||||
//var Index embed.FS
|
|
||||||
//
|
|
||||||
////go:embed assets/**
|
|
||||||
//var Assets embed.FS
|
|
||||||
|
@ -38,7 +38,10 @@ func ParsePath(rawPath string) (*model.Account, string, base.Driver, error) {
|
|||||||
if bIndex != -1 {
|
if bIndex != -1 {
|
||||||
name = name[:bIndex]
|
name = name[:bIndex]
|
||||||
}
|
}
|
||||||
return &account, strings.TrimPrefix(rawPath, name), driver, nil
|
//if name == "/" {
|
||||||
|
// name = ""
|
||||||
|
//}
|
||||||
|
return &account, utils.ParsePath(strings.TrimPrefix(rawPath, name)), driver, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func ErrorResp(c *gin.Context, err error, code int) {
|
func ErrorResp(c *gin.Context, err error, code int) {
|
||||||
|
@ -9,9 +9,9 @@ import (
|
|||||||
|
|
||||||
func Path(rawPath string) (*model.File, []model.File, *model.Account, base.Driver, string, error) {
|
func Path(rawPath string) (*model.File, []model.File, *model.Account, base.Driver, string, error) {
|
||||||
account, path, driver, err := ParsePath(rawPath)
|
account, path, driver, err := ParsePath(rawPath)
|
||||||
|
accountFiles := model.GetAccountFilesByPath(rawPath)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
if err.Error() == "path not found" {
|
if err.Error() == "path not found" {
|
||||||
accountFiles := model.GetAccountFilesByPath(rawPath)
|
|
||||||
if len(accountFiles) != 0 {
|
if len(accountFiles) != 0 {
|
||||||
return nil, accountFiles, nil, nil, path, nil
|
return nil, accountFiles, nil, nil, path, nil
|
||||||
}
|
}
|
||||||
@ -21,13 +21,31 @@ func Path(rawPath string) (*model.File, []model.File, *model.Account, base.Drive
|
|||||||
log.Debugln("use account: ", account.Name)
|
log.Debugln("use account: ", account.Name)
|
||||||
file, files, err := operate.Path(driver, account, path)
|
file, files, err := operate.Path(driver, account, path)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
|
if err.Error() == "path not found" {
|
||||||
|
if len(accountFiles) != 0 {
|
||||||
|
return nil, accountFiles, nil, nil, path, nil
|
||||||
|
}
|
||||||
|
}
|
||||||
return nil, nil, nil, nil, "", err
|
return nil, nil, nil, nil, "", err
|
||||||
}
|
}
|
||||||
if file != nil {
|
if file != nil {
|
||||||
return file, nil, account, driver, path, nil
|
return file, nil, account, driver, path, nil
|
||||||
} else {
|
} else {
|
||||||
accountFiles := model.GetAccountFilesByPath(rawPath)
|
accountFiles := model.GetAccountFilesByPath(rawPath)
|
||||||
files = append(files, accountFiles...)
|
for _, accountFile := range accountFiles {
|
||||||
|
if !containsByName(files, accountFile) {
|
||||||
|
files = append(files, accountFile)
|
||||||
|
}
|
||||||
|
}
|
||||||
return nil, files, account, driver, path, nil
|
return nil, files, account, driver, path, nil
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func containsByName(files []model.File, file model.File) bool {
|
||||||
|
for _, f := range files {
|
||||||
|
if f.Name == file.Name {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
@ -12,6 +12,7 @@ import (
|
|||||||
"net/url"
|
"net/url"
|
||||||
"os"
|
"os"
|
||||||
"strconv"
|
"strconv"
|
||||||
|
"strings"
|
||||||
)
|
)
|
||||||
|
|
||||||
var HttpClient = &http.Client{}
|
var HttpClient = &http.Client{}
|
||||||
@ -27,7 +28,16 @@ func Proxy(w http.ResponseWriter, r *http.Request, link *base.Link, file *model.
|
|||||||
w.Header().Set("Content-Type", "application/octet-stream")
|
w.Header().Set("Content-Type", "application/octet-stream")
|
||||||
w.Header().Set("Content-Disposition", fmt.Sprintf(`attachment; filename=%s`, url.QueryEscape(file.Name)))
|
w.Header().Set("Content-Disposition", fmt.Sprintf(`attachment; filename=%s`, url.QueryEscape(file.Name)))
|
||||||
w.Header().Set("Content-Length", strconv.FormatInt(file.Size, 10))
|
w.Header().Set("Content-Length", strconv.FormatInt(file.Size, 10))
|
||||||
w.WriteHeader(http.StatusOK)
|
if link.Header != nil {
|
||||||
|
for h, val := range link.Header {
|
||||||
|
w.Header()[h] = val
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if link.Status == 0 {
|
||||||
|
w.WriteHeader(http.StatusOK)
|
||||||
|
} else {
|
||||||
|
w.WriteHeader(link.Status)
|
||||||
|
}
|
||||||
_, err = io.Copy(w, link.Data)
|
_, err = io.Copy(w, link.Data)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
@ -56,8 +66,12 @@ func Proxy(w http.ResponseWriter, r *http.Request, link *base.Link, file *model.
|
|||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
for h, val := range r.Header {
|
for h, val := range r.Header {
|
||||||
|
if strings.ToLower(h) == "authorization" {
|
||||||
|
continue
|
||||||
|
}
|
||||||
req.Header[h] = val
|
req.Header[h] = val
|
||||||
}
|
}
|
||||||
|
log.Debugf("req headers: %+v", r.Header)
|
||||||
for _, header := range link.Headers {
|
for _, header := range link.Headers {
|
||||||
req.Header.Set(header.Name, header.Value)
|
req.Header.Set(header.Name, header.Value)
|
||||||
}
|
}
|
||||||
|
@ -60,6 +60,7 @@ type Meta struct {
|
|||||||
Driver string `json:"driver"`
|
Driver string `json:"driver"`
|
||||||
Upload bool `json:"upload"`
|
Upload bool `json:"upload"`
|
||||||
Total int `json:"total"`
|
Total int `json:"total"`
|
||||||
|
Readme string `json:"readme"`
|
||||||
//Pages int `json:"pages"`
|
//Pages int `json:"pages"`
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -75,8 +76,10 @@ func Path(c *gin.Context) {
|
|||||||
_, ok := c.Get("admin")
|
_, ok := c.Get("admin")
|
||||||
meta, _ := model.GetMetaByPath(req.Path)
|
meta, _ := model.GetMetaByPath(req.Path)
|
||||||
upload := false
|
upload := false
|
||||||
if meta != nil && meta.Upload {
|
readme := ""
|
||||||
upload = true
|
if meta != nil {
|
||||||
|
upload = meta.Upload
|
||||||
|
readme = meta.Readme
|
||||||
}
|
}
|
||||||
err := CheckPagination(&req)
|
err := CheckPagination(&req)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
@ -137,6 +140,7 @@ func Path(c *gin.Context) {
|
|||||||
Driver: driverName,
|
Driver: driverName,
|
||||||
Upload: upload,
|
Upload: upload,
|
||||||
Total: total,
|
Total: total,
|
||||||
|
Readme: readme,
|
||||||
},
|
},
|
||||||
Files: files,
|
Files: files,
|
||||||
},
|
},
|
||||||
|
@ -55,12 +55,13 @@ func Proxy(c *gin.Context) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
// 对于中转,不需要重设IP
|
// 对于中转,不需要重设IP
|
||||||
link, err := driver.Link(base.Args{Path: path}, account)
|
link, err := driver.Link(base.Args{Path: path, Header: c.Request.Header}, account)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
common.ErrorResp(c, err, 500)
|
common.ErrorResp(c, err, 500)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
err = common.Proxy(c.Writer, c.Request, link, file)
|
err = common.Proxy(c.Writer, c.Request, link, file)
|
||||||
|
log.Debugln("web proxy error:", err)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
common.ErrorResp(c, err, 500)
|
common.ErrorResp(c, err, 500)
|
||||||
}
|
}
|
||||||
|
31
server/controllers/search.go
Normal file
31
server/controllers/search.go
Normal file
@ -0,0 +1,31 @@
|
|||||||
|
package controllers
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/Xhofe/alist/conf"
|
||||||
|
"github.com/Xhofe/alist/model"
|
||||||
|
"github.com/Xhofe/alist/server/common"
|
||||||
|
"github.com/gin-gonic/gin"
|
||||||
|
)
|
||||||
|
|
||||||
|
type SearchReq struct {
|
||||||
|
Path string `json:"path"`
|
||||||
|
Keyword string `json:"keyword"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func Search(c *gin.Context) {
|
||||||
|
if !conf.GetBool("enable search") {
|
||||||
|
common.ErrorStrResp(c, "Not allowed search", 403)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
var req SearchReq
|
||||||
|
if err := c.ShouldBind(&req); err != nil {
|
||||||
|
common.ErrorResp(c, err, 400)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
files, err := model.SearchByNameAndPath(req.Path, req.Keyword)
|
||||||
|
if err != nil {
|
||||||
|
common.ErrorResp(c, err, 500)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
common.SuccessResp(c, files)
|
||||||
|
}
|
@ -25,6 +25,8 @@ func InitApiRouter(r *gin.Engine) {
|
|||||||
path.POST("/path", controllers.Path)
|
path.POST("/path", controllers.Path)
|
||||||
path.POST("/preview", controllers.Preview)
|
path.POST("/preview", controllers.Preview)
|
||||||
|
|
||||||
|
public.POST("/search", controllers.Search)
|
||||||
|
|
||||||
//path.POST("/link",middlewares.Auth, controllers.Link)
|
//path.POST("/link",middlewares.Auth, controllers.Link)
|
||||||
public.POST("/upload", file.UploadFiles)
|
public.POST("/upload", file.UploadFiles)
|
||||||
|
|
||||||
|
@ -31,21 +31,37 @@ func (fs *FileSystem) File(rawPath string) (*model.File, error) {
|
|||||||
if f, ok := upFileMap[rawPath]; ok {
|
if f, ok := upFileMap[rawPath]; ok {
|
||||||
return f, nil
|
return f, nil
|
||||||
}
|
}
|
||||||
if model.AccountsCount() > 1 && rawPath == "/" {
|
|
||||||
now := time.Now()
|
|
||||||
return &model.File{
|
|
||||||
Name: "root",
|
|
||||||
Size: 0,
|
|
||||||
Type: conf.FOLDER,
|
|
||||||
Driver: "root",
|
|
||||||
UpdatedAt: &now,
|
|
||||||
}, nil
|
|
||||||
}
|
|
||||||
account, path_, driver, err := common.ParsePath(rawPath)
|
account, path_, driver, err := common.ParsePath(rawPath)
|
||||||
|
log.Debugln(account, path_, driver, err)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
|
if err.Error() == "path not found" {
|
||||||
|
accountFiles := model.GetAccountFilesByPath(rawPath)
|
||||||
|
if len(accountFiles) != 0 {
|
||||||
|
now := time.Now()
|
||||||
|
return &model.File{
|
||||||
|
Name: "root",
|
||||||
|
Size: 0,
|
||||||
|
Type: conf.FOLDER,
|
||||||
|
UpdatedAt: &now,
|
||||||
|
}, nil
|
||||||
|
}
|
||||||
|
}
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
return operate.File(driver, account, path_)
|
file, err := operate.File(driver, account, path_)
|
||||||
|
if err != nil && err.Error() == "path not found" {
|
||||||
|
accountFiles := model.GetAccountFilesByPath(rawPath)
|
||||||
|
if len(accountFiles) != 0 {
|
||||||
|
now := time.Now()
|
||||||
|
return &model.File{
|
||||||
|
Name: "root",
|
||||||
|
Size: 0,
|
||||||
|
Type: conf.FOLDER,
|
||||||
|
UpdatedAt: &now,
|
||||||
|
}, nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return file, err
|
||||||
}
|
}
|
||||||
|
|
||||||
func (fs *FileSystem) Files(ctx context.Context, rawPath string) ([]model.File, error) {
|
func (fs *FileSystem) Files(ctx context.Context, rawPath string) ([]model.File, error) {
|
||||||
@ -117,7 +133,7 @@ func (fs *FileSystem) Link(w http.ResponseWriter, r *http.Request, rawPath strin
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
return "", err
|
return "", err
|
||||||
}
|
}
|
||||||
link_, err := driver.Link(base.Args{Path: path_}, account)
|
link_, err := driver.Link(base.Args{Path: path_, Header: r.Header}, account)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return "", err
|
return "", err
|
||||||
}
|
}
|
||||||
|
Reference in New Issue
Block a user