Compare commits
146 Commits
v3.0.0-bet
...
v3.4.0
Author | SHA1 | Date | |
---|---|---|---|
efe0e6af22 | |||
00de9bf16d | |||
1743110a70 | |||
0352a8e028 | |||
c601bb794b | |||
42865486f1 | |||
44f5cf40ef | |||
c3ab378ac5 | |||
cdcbfb24c4 | |||
e05e2fd663 | |||
6639cab1ae | |||
8241f0999a | |||
f3a5e3702d | |||
46701a176d | |||
26a29f20c3 | |||
18cd45d257 | |||
f0a533a77a | |||
619a9aeb6c | |||
7bfa5876ed | |||
f95ab6ee57 | |||
e75f19e9c0 | |||
1c212f6c30 | |||
141419056d | |||
aabfe49cb9 | |||
a3b631f9e9 | |||
18165eb50d | |||
061c462f0b | |||
5f79d665d9 | |||
f0cc0a76a9 | |||
dd4674e486 | |||
0019959eec | |||
3e9c38697d | |||
e3b7c41199 | |||
a2c808c8ce | |||
da7e17aa38 | |||
02df3759df | |||
4fef500795 | |||
07ece452b3 | |||
b8cf02ca68 | |||
3db798a82a | |||
45cc0cedbd | |||
2efade123e | |||
fc393f743f | |||
0e99e7e9b9 | |||
7a95850c1b | |||
549355bb29 | |||
55aa8ee3b1 | |||
1c22fc367e | |||
5ea8d62aa4 | |||
baebc2fbe9 | |||
8c69260972 | |||
30f992c6a8 | |||
dcaaae366b | |||
284035823f | |||
be8ff92414 | |||
a4c846a424 | |||
451e418b18 | |||
4e13b1a83c | |||
9d2e9887af | |||
dc73c2e97d | |||
a624121095 | |||
9d9c79179b | |||
b7479651e1 | |||
2fc0ccbfe0 | |||
f86ad1dce4 | |||
f0181d92cd | |||
5ac6a30c56 | |||
96d8a382e8 | |||
7c32af4649 | |||
03dbb3a403 | |||
a570e4c7a0 | |||
539c47bd3b | |||
b6d9018ebd | |||
c929888e39 | |||
af946ff13e | |||
0039dc18e1 | |||
4d6ab53336 | |||
c7f6684eed | |||
b71ecc8e89 | |||
3537153b91 | |||
9382f66f87 | |||
656f5f112c | |||
9181861f47 | |||
1ab73e0742 | |||
57686d9df1 | |||
ca177cc3b9 | |||
d8dc8d8623 | |||
5548ab62ac | |||
d6d82c3138 | |||
2185839236 | |||
24d58f278a | |||
f80be96cf9 | |||
6c89c6c8ae | |||
b74b55fa4a | |||
09564102e7 | |||
d436a6e676 | |||
bec3a327a7 | |||
d329df70f3 | |||
1af9f4061e | |||
0d012f85cb | |||
e3b213c398 | |||
d9f0603271 | |||
86a625cb40 | |||
f22232de5d | |||
7ad3748a46 | |||
66b2562d03 | |||
b197322cd8 | |||
9e5ef974a7 | |||
08a001fbd1 | |||
54ae6dce0b | |||
a90ef201c7 | |||
2de0da87fa | |||
53e08e75fe | |||
6b5236f52e | |||
78e34f0d9f | |||
6aedd0f425 | |||
5ff0d850d7 | |||
cd73e34ccc | |||
107462e42e | |||
e6c2d22700 | |||
889ddcef7e | |||
68a6a0c40e | |||
969018db37 | |||
fba1471ec4 | |||
8b72ac7f80 | |||
77a6aa487b | |||
fd99c2197b | |||
9c91f062b9 | |||
537ca030b2 | |||
b00dcdec0d | |||
57bcd376b4 | |||
8d4d8648c6 | |||
35d177b67b | |||
40882443c2 | |||
05f19cad78 | |||
7249f277b2 | |||
849124f177 | |||
f5c7a11da5 | |||
043a79189d | |||
5ed43fd17d | |||
220cd4d6b8 | |||
f692e6c011 | |||
f48365929e | |||
56219bf096 | |||
5ad3849bb6 | |||
4af9124162 |
@ -1,98 +0,0 @@
|
|||||||
{
|
|
||||||
"files": [
|
|
||||||
"CONTRIBUTORS.md"
|
|
||||||
],
|
|
||||||
"imageSize": 100,
|
|
||||||
"commit": false,
|
|
||||||
"contributors": [
|
|
||||||
{
|
|
||||||
"login": "Xhofe",
|
|
||||||
"name": "Xhofe",
|
|
||||||
"avatar_url": "https://avatars.githubusercontent.com/u/36558727?v=4",
|
|
||||||
"profile": "http://nn.ci",
|
|
||||||
"contributions": [
|
|
||||||
"code",
|
|
||||||
"ideas",
|
|
||||||
"doc"
|
|
||||||
]
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"login": "foxxorcat",
|
|
||||||
"name": "foxxorcat",
|
|
||||||
"avatar_url": "https://avatars.githubusercontent.com/u/95907542?v=4",
|
|
||||||
"profile": "https://github.com/foxxorcat",
|
|
||||||
"contributions": [
|
|
||||||
"code"
|
|
||||||
]
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"login": "DaoChen6",
|
|
||||||
"name": "道辰",
|
|
||||||
"avatar_url": "https://avatars.githubusercontent.com/u/63903027?v=4",
|
|
||||||
"profile": "https://www.iflu.cf/",
|
|
||||||
"contributions": [
|
|
||||||
"doc"
|
|
||||||
]
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"login": "vg-land",
|
|
||||||
"name": "vg-land",
|
|
||||||
"avatar_url": "https://avatars.githubusercontent.com/u/16739728?v=4",
|
|
||||||
"profile": "https://vg-land.github.io/",
|
|
||||||
"contributions": [
|
|
||||||
"code"
|
|
||||||
]
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"login": "Clansty",
|
|
||||||
"name": "凌莞~(=^▽^=)",
|
|
||||||
"avatar_url": "https://avatars.githubusercontent.com/u/18461360?v=4",
|
|
||||||
"profile": "https://c5y.moe",
|
|
||||||
"contributions": [
|
|
||||||
"doc"
|
|
||||||
]
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"login": "Windman1320",
|
|
||||||
"name": "Windman",
|
|
||||||
"avatar_url": "https://avatars.githubusercontent.com/u/9999486?v=4",
|
|
||||||
"profile": "https://github.com/Windman1320",
|
|
||||||
"contributions": [
|
|
||||||
"code"
|
|
||||||
]
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"login": "ericarena",
|
|
||||||
"name": "ericarena",
|
|
||||||
"avatar_url": "https://avatars.githubusercontent.com/u/4518927?v=4",
|
|
||||||
"profile": "https://github.com/ericarena",
|
|
||||||
"contributions": [
|
|
||||||
"code"
|
|
||||||
]
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"login": "WntFlm",
|
|
||||||
"name": "WntFlm",
|
|
||||||
"avatar_url": "https://avatars.githubusercontent.com/u/34620278?v=4",
|
|
||||||
"profile": "https://github.com/WntFlm",
|
|
||||||
"contributions": [
|
|
||||||
"code"
|
|
||||||
]
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"login": "XZB-1248",
|
|
||||||
"name": "XZB-1248",
|
|
||||||
"avatar_url": "https://avatars.githubusercontent.com/u/28593573?v=4",
|
|
||||||
"profile": "https://github.com/XZB-1248",
|
|
||||||
"contributions": [
|
|
||||||
"code"
|
|
||||||
]
|
|
||||||
}
|
|
||||||
],
|
|
||||||
"contributorsPerLine": 7,
|
|
||||||
"projectName": "alist",
|
|
||||||
"projectOwner": "alist-org",
|
|
||||||
"repoType": "github",
|
|
||||||
"repoHost": "https://github.com",
|
|
||||||
"skipCi": true
|
|
||||||
}
|
|
13
.github/FUNDING.yml
vendored
Normal file
13
.github/FUNDING.yml
vendored
Normal file
@ -0,0 +1,13 @@
|
|||||||
|
# These are supported funding model platforms
|
||||||
|
|
||||||
|
github: # Replace with up to 4 GitHub Sponsors-enabled usernames e.g., [user1, user2]
|
||||||
|
patreon: # Replace with a single Patreon username
|
||||||
|
open_collective: # Replace with a single Open Collective username
|
||||||
|
ko_fi: # Replace with a single Ko-fi username
|
||||||
|
tidelift: # Replace with a single Tidelift platform-name/package-name e.g., npm/babel
|
||||||
|
community_bridge: # Replace with a single Community Bridge project-name e.g., cloud-foundry
|
||||||
|
liberapay: # Replace with a single Liberapay username
|
||||||
|
issuehunt: # Replace with a single IssueHunt username
|
||||||
|
otechie: # Replace with a single Otechie username
|
||||||
|
lfx_crowdfunding: # Replace with a single LFX Crowdfunding project-name e.g., cloud-foundry
|
||||||
|
custom: ['https://alist.nn.ci/guide/sponsor.html']
|
3
.github/ISSUE_TEMPLATE/bug_report.yml
vendored
3
.github/ISSUE_TEMPLATE/bug_report.yml
vendored
@ -12,9 +12,10 @@ body:
|
|||||||
label: Please make sure of the following things
|
label: Please make sure of the following things
|
||||||
description: You may select more than one, even select all.
|
description: You may select more than one, even select all.
|
||||||
options:
|
options:
|
||||||
- label: I have read the [documentation](https://alist-doc.nn.ci).
|
- label: I have read the [documentation](https://alist.nn.ci).
|
||||||
- label: I'm sure there are no duplicate issues or discussions.
|
- 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`).
|
- label: I'm sure it's due to `alist` and not something else(such as `Dependencies` or `Operational`).
|
||||||
|
- label: I'm sure I'm using the latest version
|
||||||
- type: input
|
- type: input
|
||||||
id: version
|
id: version
|
||||||
attributes:
|
attributes:
|
||||||
|
2
.github/ISSUE_TEMPLATE/feature_request.yml
vendored
2
.github/ISSUE_TEMPLATE/feature_request.yml
vendored
@ -7,7 +7,7 @@ body:
|
|||||||
label: Please make sure of the following things
|
label: Please make sure of the following things
|
||||||
description: You may select more than one, even select all.
|
description: You may select more than one, even select all.
|
||||||
options:
|
options:
|
||||||
- label: I have read the [documentation](https://alist-doc.nn.ci).
|
- label: I have read the [documentation](https://alist.nn.ci).
|
||||||
- label: I'm sure there are no duplicate issues or discussions.
|
- label: I'm sure there are no duplicate issues or discussions.
|
||||||
- label: I'm sure this feature is not implemented.
|
- label: I'm sure this feature is not implemented.
|
||||||
- label: I'm sure it's a reasonable and popular requirement.
|
- label: I'm sure it's a reasonable and popular requirement.
|
||||||
|
2
.github/workflows/build.yml
vendored
2
.github/workflows/build.yml
vendored
@ -11,7 +11,7 @@ jobs:
|
|||||||
strategy:
|
strategy:
|
||||||
matrix:
|
matrix:
|
||||||
platform: [ubuntu-latest]
|
platform: [ubuntu-latest]
|
||||||
go-version: [1.18]
|
go-version: [1.19]
|
||||||
name: Build
|
name: Build
|
||||||
runs-on: ${{ matrix.platform }}
|
runs-on: ${{ matrix.platform }}
|
||||||
steps:
|
steps:
|
||||||
|
3
.github/workflows/issue_check_inactive.yml
vendored
3
.github/workflows/issue_check_inactive.yml
vendored
@ -13,4 +13,5 @@ jobs:
|
|||||||
with:
|
with:
|
||||||
actions: 'check-inactive'
|
actions: 'check-inactive'
|
||||||
token: ${{ secrets.GITHUB_TOKEN }}
|
token: ${{ secrets.GITHUB_TOKEN }}
|
||||||
inactive-day: 30
|
inactive-day: 30
|
||||||
|
body: Hello, this issue has been inactive for more than 30 days and will be closed if inactive for another 30 days.
|
21
.github/workflows/issue_close_inactive.yml
vendored
Normal file
21
.github/workflows/issue_close_inactive.yml
vendored
Normal file
@ -0,0 +1,21 @@
|
|||||||
|
name: Close inactive
|
||||||
|
|
||||||
|
on:
|
||||||
|
schedule:
|
||||||
|
- cron: "0 0 */7 * *"
|
||||||
|
workflow_dispatch:
|
||||||
|
|
||||||
|
jobs:
|
||||||
|
close-inactive:
|
||||||
|
runs-on: ubuntu-latest
|
||||||
|
steps:
|
||||||
|
- name: close-issues
|
||||||
|
uses: actions-cool/issues-helper@v3
|
||||||
|
with:
|
||||||
|
actions: 'close-issues'
|
||||||
|
token: ${{ secrets.GITHUB_TOKEN }}
|
||||||
|
labels: 'inactive'
|
||||||
|
inactive-day: 30
|
||||||
|
close-reason: 'not_planned'
|
||||||
|
body: |
|
||||||
|
Hello @${{ github.event.issue.user.login }}, this issue was closed due to inactive more than 60 days. You can reopen or recreate it if you think it should continue.
|
@ -1,11 +1,12 @@
|
|||||||
name: Check need info
|
name: Close need info
|
||||||
|
|
||||||
on:
|
on:
|
||||||
schedule:
|
schedule:
|
||||||
- cron: "0 0 */7 * *"
|
- cron: "0 0 */7 * *"
|
||||||
|
workflow_dispatch:
|
||||||
|
|
||||||
jobs:
|
jobs:
|
||||||
check-need-info:
|
close-need-info:
|
||||||
runs-on: ubuntu-latest
|
runs-on: ubuntu-latest
|
||||||
steps:
|
steps:
|
||||||
- name: close-issues
|
- name: close-issues
|
3
.github/workflows/issue_similarity.yml
vendored
3
.github/workflows/issue_similarity.yml
vendored
@ -15,4 +15,5 @@ jobs:
|
|||||||
comment-title: '### See'
|
comment-title: '### See'
|
||||||
comment-body: '${index}. ${similarity} #${number}'
|
comment-body: '${index}. ${similarity} #${number}'
|
||||||
show-footer: false
|
show-footer: false
|
||||||
show-mentioned: true
|
show-mentioned: true
|
||||||
|
since-days: 730
|
2
.github/workflows/release.yml
vendored
2
.github/workflows/release.yml
vendored
@ -22,7 +22,7 @@ jobs:
|
|||||||
strategy:
|
strategy:
|
||||||
matrix:
|
matrix:
|
||||||
platform: [ubuntu-latest]
|
platform: [ubuntu-latest]
|
||||||
go-version: [1.18]
|
go-version: [1.19]
|
||||||
name: Release
|
name: Release
|
||||||
runs-on: ${{ matrix.platform }}
|
runs-on: ${{ matrix.platform }}
|
||||||
steps:
|
steps:
|
||||||
|
1
.gitignore
vendored
1
.gitignore
vendored
@ -25,5 +25,6 @@ bin/*
|
|||||||
data/
|
data/
|
||||||
log/
|
log/
|
||||||
lang/
|
lang/
|
||||||
|
daemon/
|
||||||
public/dist/*
|
public/dist/*
|
||||||
!public/dist/README.md
|
!public/dist/README.md
|
128
CODE_OF_CONDUCT.md
Normal file
128
CODE_OF_CONDUCT.md
Normal file
@ -0,0 +1,128 @@
|
|||||||
|
# Contributor Covenant Code of Conduct
|
||||||
|
|
||||||
|
## Our Pledge
|
||||||
|
|
||||||
|
We as members, contributors, and leaders pledge to make participation in our
|
||||||
|
community a harassment-free experience for everyone, regardless of age, body
|
||||||
|
size, visible or invisible disability, ethnicity, sex characteristics, gender
|
||||||
|
identity and expression, level of experience, education, socio-economic status,
|
||||||
|
nationality, personal appearance, race, religion, or sexual identity
|
||||||
|
and orientation.
|
||||||
|
|
||||||
|
We pledge to act and interact in ways that contribute to an open, welcoming,
|
||||||
|
diverse, inclusive, and healthy community.
|
||||||
|
|
||||||
|
## Our Standards
|
||||||
|
|
||||||
|
Examples of behavior that contributes to a positive environment for our
|
||||||
|
community include:
|
||||||
|
|
||||||
|
* Demonstrating empathy and kindness toward other people
|
||||||
|
* Being respectful of differing opinions, viewpoints, and experiences
|
||||||
|
* Giving and gracefully accepting constructive feedback
|
||||||
|
* Accepting responsibility and apologizing to those affected by our mistakes,
|
||||||
|
and learning from the experience
|
||||||
|
* Focusing on what is best not just for us as individuals, but for the
|
||||||
|
overall community
|
||||||
|
|
||||||
|
Examples of unacceptable behavior include:
|
||||||
|
|
||||||
|
* The use of sexualized language or imagery, and sexual attention or
|
||||||
|
advances of any kind
|
||||||
|
* Trolling, insulting or derogatory comments, and personal or political attacks
|
||||||
|
* Public or private harassment
|
||||||
|
* Publishing others' private information, such as a physical or email
|
||||||
|
address, without their explicit permission
|
||||||
|
* Other conduct which could reasonably be considered inappropriate in a
|
||||||
|
professional setting
|
||||||
|
|
||||||
|
## Enforcement Responsibilities
|
||||||
|
|
||||||
|
Community leaders are responsible for clarifying and enforcing our standards of
|
||||||
|
acceptable behavior and will take appropriate and fair corrective action in
|
||||||
|
response to any behavior that they deem inappropriate, threatening, offensive,
|
||||||
|
or harmful.
|
||||||
|
|
||||||
|
Community leaders have the right and responsibility to remove, edit, or reject
|
||||||
|
comments, commits, code, wiki edits, issues, and other contributions that are
|
||||||
|
not aligned to this Code of Conduct, and will communicate reasons for moderation
|
||||||
|
decisions when appropriate.
|
||||||
|
|
||||||
|
## Scope
|
||||||
|
|
||||||
|
This Code of Conduct applies within all community spaces, and also applies when
|
||||||
|
an individual is officially representing the community in public spaces.
|
||||||
|
Examples of representing our community include using an official e-mail address,
|
||||||
|
posting via an official social media account, or acting as an appointed
|
||||||
|
representative at an online or offline event.
|
||||||
|
|
||||||
|
## Enforcement
|
||||||
|
|
||||||
|
Instances of abusive, harassing, or otherwise unacceptable behavior may be
|
||||||
|
reported to the community leaders responsible for enforcement at
|
||||||
|
i@nn.ci.
|
||||||
|
All complaints will be reviewed and investigated promptly and fairly.
|
||||||
|
|
||||||
|
All community leaders are obligated to respect the privacy and security of the
|
||||||
|
reporter of any incident.
|
||||||
|
|
||||||
|
## Enforcement Guidelines
|
||||||
|
|
||||||
|
Community leaders will follow these Community Impact Guidelines in determining
|
||||||
|
the consequences for any action they deem in violation of this Code of Conduct:
|
||||||
|
|
||||||
|
### 1. Correction
|
||||||
|
|
||||||
|
**Community Impact**: Use of inappropriate language or other behavior deemed
|
||||||
|
unprofessional or unwelcome in the community.
|
||||||
|
|
||||||
|
**Consequence**: A private, written warning from community leaders, providing
|
||||||
|
clarity around the nature of the violation and an explanation of why the
|
||||||
|
behavior was inappropriate. A public apology may be requested.
|
||||||
|
|
||||||
|
### 2. Warning
|
||||||
|
|
||||||
|
**Community Impact**: A violation through a single incident or series
|
||||||
|
of actions.
|
||||||
|
|
||||||
|
**Consequence**: A warning with consequences for continued behavior. No
|
||||||
|
interaction with the people involved, including unsolicited interaction with
|
||||||
|
those enforcing the Code of Conduct, for a specified period of time. This
|
||||||
|
includes avoiding interactions in community spaces as well as external channels
|
||||||
|
like social media. Violating these terms may lead to a temporary or
|
||||||
|
permanent ban.
|
||||||
|
|
||||||
|
### 3. Temporary Ban
|
||||||
|
|
||||||
|
**Community Impact**: A serious violation of community standards, including
|
||||||
|
sustained inappropriate behavior.
|
||||||
|
|
||||||
|
**Consequence**: A temporary ban from any sort of interaction or public
|
||||||
|
communication with the community for a specified period of time. No public or
|
||||||
|
private interaction with the people involved, including unsolicited interaction
|
||||||
|
with those enforcing the Code of Conduct, is allowed during this period.
|
||||||
|
Violating these terms may lead to a permanent ban.
|
||||||
|
|
||||||
|
### 4. Permanent Ban
|
||||||
|
|
||||||
|
**Community Impact**: Demonstrating a pattern of violation of community
|
||||||
|
standards, including sustained inappropriate behavior, harassment of an
|
||||||
|
individual, or aggression toward or disparagement of classes of individuals.
|
||||||
|
|
||||||
|
**Consequence**: A permanent ban from any sort of public interaction within
|
||||||
|
the community.
|
||||||
|
|
||||||
|
## Attribution
|
||||||
|
|
||||||
|
This Code of Conduct is adapted from the [Contributor Covenant][homepage],
|
||||||
|
version 2.0, available at
|
||||||
|
https://www.contributor-covenant.org/version/2/0/code_of_conduct.html.
|
||||||
|
|
||||||
|
Community Impact Guidelines were inspired by [Mozilla's code of conduct
|
||||||
|
enforcement ladder](https://github.com/mozilla/diversity).
|
||||||
|
|
||||||
|
[homepage]: https://www.contributor-covenant.org
|
||||||
|
|
||||||
|
For answers to common questions about this code of conduct, see the FAQ at
|
||||||
|
https://www.contributor-covenant.org/faq. Translations are available at
|
||||||
|
https://www.contributor-covenant.org/translations.
|
@ -7,7 +7,7 @@
|
|||||||
Prerequisites:
|
Prerequisites:
|
||||||
|
|
||||||
- [git](https://nodejs.org/zh-cn/)
|
- [git](https://nodejs.org/zh-cn/)
|
||||||
- [Go 1.18+](https://golang.org/doc/install)
|
- [Go 1.19+](https://golang.org/doc/install)
|
||||||
- [gcc](https://gcc.gnu.org/)
|
- [gcc](https://gcc.gnu.org/)
|
||||||
- [nodejs](https://nodejs.org/)
|
- [nodejs](https://nodejs.org/)
|
||||||
|
|
||||||
|
@ -1,33 +0,0 @@
|
|||||||
<!-- ALL-CONTRIBUTORS-BADGE:START - Do not remove or modify this section -->
|
|
||||||
[](#contributors-)
|
|
||||||
<!-- ALL-CONTRIBUTORS-BADGE:END -->
|
|
||||||
|
|
||||||
## Contributors ✨
|
|
||||||
|
|
||||||
Thanks goes to these wonderful people ([emoji key](https://allcontributors.org/docs/en/emoji-key)):
|
|
||||||
|
|
||||||
<!-- ALL-CONTRIBUTORS-LIST:START - Do not remove or modify this section -->
|
|
||||||
<!-- prettier-ignore-start -->
|
|
||||||
<!-- markdownlint-disable -->
|
|
||||||
<table>
|
|
||||||
<tr>
|
|
||||||
<td align="center"><a href="http://nn.ci"><img src="https://avatars.githubusercontent.com/u/36558727?v=4?s=100" width="100px;" alt=""/><br /><sub><b>Xhofe</b></sub></a><br /><a href="https://github.com/alist-org/alist/commits?author=Xhofe" title="Code">💻</a> <a href="#ideas-Xhofe" title="Ideas, Planning, & Feedback">🤔</a> <a href="https://github.com/alist-org/alist/commits?author=Xhofe" title="Documentation">📖</a></td>
|
|
||||||
<td align="center"><a href="https://github.com/foxxorcat"><img src="https://avatars.githubusercontent.com/u/95907542?v=4?s=100" width="100px;" alt=""/><br /><sub><b>foxxorcat</b></sub></a><br /><a href="https://github.com/alist-org/alist/commits?author=foxxorcat" title="Code">💻</a></td>
|
|
||||||
<td align="center"><a href="https://www.iflu.cf/"><img src="https://avatars.githubusercontent.com/u/63903027?v=4?s=100" width="100px;" alt=""/><br /><sub><b>道辰</b></sub></a><br /><a href="https://github.com/alist-org/alist/commits?author=DaoChen6" title="Documentation">📖</a></td>
|
|
||||||
<td align="center"><a href="https://vg-land.github.io/"><img src="https://avatars.githubusercontent.com/u/16739728?v=4?s=100" width="100px;" alt=""/><br /><sub><b>vg-land</b></sub></a><br /><a href="https://github.com/alist-org/alist/commits?author=vg-land" title="Code">💻</a></td>
|
|
||||||
<td align="center"><a href="https://c5y.moe"><img src="https://avatars.githubusercontent.com/u/18461360?v=4?s=100" width="100px;" alt=""/><br /><sub><b>凌莞~(=^▽^=)</b></sub></a><br /><a href="https://github.com/alist-org/alist/commits?author=Clansty" title="Documentation">📖</a></td>
|
|
||||||
<td align="center"><a href="https://github.com/Windman1320"><img src="https://avatars.githubusercontent.com/u/9999486?v=4?s=100" width="100px;" alt=""/><br /><sub><b>Windman</b></sub></a><br /><a href="https://github.com/alist-org/alist/commits?author=Windman1320" title="Code">💻</a></td>
|
|
||||||
<td align="center"><a href="https://github.com/ericarena"><img src="https://avatars.githubusercontent.com/u/4518927?v=4?s=100" width="100px;" alt=""/><br /><sub><b>ericarena</b></sub></a><br /><a href="https://github.com/alist-org/alist/commits?author=ericarena" title="Code">💻</a></td>
|
|
||||||
</tr>
|
|
||||||
<tr>
|
|
||||||
<td align="center"><a href="https://github.com/WntFlm"><img src="https://avatars.githubusercontent.com/u/34620278?v=4?s=100" width="100px;" alt=""/><br /><sub><b>WntFlm</b></sub></a><br /><a href="https://github.com/alist-org/alist/commits?author=WntFlm" title="Code">💻</a></td>
|
|
||||||
<td align="center"><a href="https://github.com/XZB-1248"><img src="https://avatars.githubusercontent.com/u/28593573?v=4?s=100" width="100px;" alt=""/><br /><sub><b>XZB-1248</b></sub></a><br /><a href="https://github.com/alist-org/alist/commits?author=XZB-1248" title="Code">💻</a></td>
|
|
||||||
</tr>
|
|
||||||
</table>
|
|
||||||
|
|
||||||
<!-- markdownlint-restore -->
|
|
||||||
<!-- prettier-ignore-end -->
|
|
||||||
|
|
||||||
<!-- ALL-CONTRIBUTORS-LIST:END -->
|
|
||||||
|
|
||||||
This project follows the [all-contributors](https://github.com/all-contributors/all-contributors) specification. Contributions of any kind welcome!
|
|
@ -10,5 +10,6 @@ LABEL MAINTAINER="i@nn.ci"
|
|||||||
VOLUME /opt/alist/data/
|
VOLUME /opt/alist/data/
|
||||||
WORKDIR /opt/alist/
|
WORKDIR /opt/alist/
|
||||||
COPY --from=builder /app/bin/alist ./
|
COPY --from=builder /app/bin/alist ./
|
||||||
|
RUN apk add ca-certificates
|
||||||
EXPOSE 5244
|
EXPOSE 5244
|
||||||
CMD [ "./alist", "server", "--no-prefix" ]
|
CMD [ "./alist", "server", "--no-prefix" ]
|
56
README.md
56
README.md
@ -1,35 +1,45 @@
|
|||||||
<div align="center">
|
<div align="center">
|
||||||
<a href="https://alist.nn.ci"><img height="100px" alt="logo" src="https://cdn.jsdelivr.net/gh/alist-org/logo@main/logo.svg"/></a>
|
<a href="https://alist.nn.ci"><img height="100px" alt="logo" src="https://cdn.jsdelivr.net/gh/alist-org/logo@main/logo.svg"/></a>
|
||||||
<p><em>🗂️A file list program that supports multiple storage, powered by Gin and Solidjs.</em></p>
|
<p><em>🗂️A file list program that supports multiple storage, powered by Gin and Solidjs.</em></p>
|
||||||
|
<div>
|
||||||
<a href="https://goreportcard.com/report/github.com/alist-org/alist/v3">
|
<a href="https://goreportcard.com/report/github.com/alist-org/alist/v3">
|
||||||
<img src="https://goreportcard.com/badge/github.com/alist-org/alist/v3" alt="latest version" />
|
<img src="https://goreportcard.com/badge/github.com/alist-org/alist/v3" alt="latest version" />
|
||||||
</a>
|
</a>
|
||||||
<a href="https://github.com/Xhofe/alist/blob/main/LICENSE">
|
<a href="https://github.com/Xhofe/alist/blob/main/LICENSE">
|
||||||
<img src="https://img.shields.io/github/license/Xhofe/alist" alt="License" />
|
<img src="https://img.shields.io/github/license/Xhofe/alist" alt="License" />
|
||||||
</a>
|
</a>
|
||||||
<a href="https://github.com/Xhofe/alist/discussions">
|
|
||||||
<img src="https://img.shields.io/github/discussions/Xhofe/alist?color=%23ED8936" alt="discussions" />
|
|
||||||
</a>
|
|
||||||
<a href="https://github.com/Xhofe/alist/actions?query=workflow%3ABuild">
|
<a href="https://github.com/Xhofe/alist/actions?query=workflow%3ABuild">
|
||||||
<img src="https://img.shields.io/github/workflow/status/Xhofe/alist/build" alt="Build status" />
|
<img src="https://img.shields.io/github/workflow/status/Xhofe/alist/build" alt="Build status" />
|
||||||
</a>
|
</a>
|
||||||
<a href="https://github.com/Xhofe/alist/releases">
|
<a href="https://github.com/Xhofe/alist/releases">
|
||||||
<img src="https://img.shields.io/github/release/Xhofe/alist" alt="latest version" />
|
<img src="https://img.shields.io/github/release/Xhofe/alist" alt="latest version" />
|
||||||
</a>
|
</a>
|
||||||
<a href="https://github.com/Xhofe/alist/releases">
|
|
||||||
<img src="https://img.shields.io/github/downloads/Xhofe/alist/total?color=%239F7AEA" alt="Downloads" />
|
|
||||||
</a>
|
|
||||||
<a title="Crowdin" target="_blank" href="https://crwd.in/alist">
|
<a title="Crowdin" target="_blank" href="https://crwd.in/alist">
|
||||||
<img src="https://badges.crowdin.net/alist/localized.svg">
|
<img src="https://badges.crowdin.net/alist/localized.svg">
|
||||||
</a>
|
</a>
|
||||||
<a href="https://pay.xhofe.top">
|
</div>
|
||||||
<img src="https://img.shields.io/badge/%24-sponsor-ff69b4.svg" alt="sponsor" />
|
<div>
|
||||||
|
<a href="https://github.com/Xhofe/alist/discussions">
|
||||||
|
<img src="https://img.shields.io/github/discussions/Xhofe/alist?color=%23ED8936" alt="discussions" />
|
||||||
</a>
|
</a>
|
||||||
|
<a href="https://discord.gg/F4ymsH4xv2">
|
||||||
|
<img src="https://img.shields.io/discord/1018870125102895134?logo=discord" alt="discussions" />
|
||||||
|
</a>
|
||||||
|
<a href="https://github.com/Xhofe/alist/releases">
|
||||||
|
<img src="https://img.shields.io/github/downloads/Xhofe/alist/total?color=%239F7AEA&logo=github" alt="Downloads" />
|
||||||
|
</a>
|
||||||
|
<a href="https://hub.docker.com/r/xhofe/alist">
|
||||||
|
<img src="https://img.shields.io/docker/pulls/xhofe/alist?color=%2348BB78&logo=docker&label=pulls" alt="Downloads" />
|
||||||
|
</a>
|
||||||
|
<a href="https://alist.nn.ci/guide/sponsor.html">
|
||||||
|
<img src="https://img.shields.io/badge/%24-sponsor-F87171.svg" alt="sponsor" />
|
||||||
|
</a>
|
||||||
|
</div>
|
||||||
</div>
|
</div>
|
||||||
|
|
||||||
---
|
---
|
||||||
|
|
||||||
English | [中文](./README_cn.md) | [Contributors](./CONTRIBUTORS.md) | [Contributing](./CONTRIBUTING.md)
|
English | [中文](./README_cn.md) | [Contributing](./CONTRIBUTING.md) | [CODE_OF_CONDUCT](./CODE_OF_CONDUCT.md)
|
||||||
|
|
||||||
## Features
|
## Features
|
||||||
|
|
||||||
@ -51,6 +61,14 @@ English | [中文](./README_cn.md) | [Contributors](./CONTRIBUTORS.md) | [Contri
|
|||||||
- [x] [YandexDisk](https://disk.yandex.com/)
|
- [x] [YandexDisk](https://disk.yandex.com/)
|
||||||
- [x] [BaiduNetdisk](http://pan.baidu.com/)
|
- [x] [BaiduNetdisk](http://pan.baidu.com/)
|
||||||
- [x] [Quark](https://pan.quark.cn)
|
- [x] [Quark](https://pan.quark.cn)
|
||||||
|
- [x] [Thunder](https://pan.xunlei.com)
|
||||||
|
- [x] [Lanzou](https://www.lanzou.com/)
|
||||||
|
- [x] [Aliyundrive share](https://www.aliyundrive.com/)
|
||||||
|
- [x] [Google photo](https://photos.google.com/)
|
||||||
|
- [x] [Mega.nz](https://mega.nz)
|
||||||
|
- [x] [Baidu photo](https://photo.baidu.com/)
|
||||||
|
- [x] SMB
|
||||||
|
- [x] [115](https://115.com/)
|
||||||
- [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
|
||||||
@ -75,17 +93,29 @@ English | [中文](./README_cn.md) | [Contributors](./CONTRIBUTORS.md) | [Contri
|
|||||||
|
|
||||||
## Demo
|
## Demo
|
||||||
|
|
||||||
<https://pan.nn.ci>
|
<https://al.nn.ci>
|
||||||
|
|
||||||
## Discussion
|
## Discussion
|
||||||
|
|
||||||
Please go to our [discussion forum](https://github.com/Xhofe/alist/discussions) for general questions, **issues are for bug reports and feature request only.**
|
Please go to our [discussion forum](https://github.com/Xhofe/alist/discussions) for general questions, **issues are for bug reports and feature request only.**
|
||||||
|
|
||||||
## Special sponsors
|
## Sponsor
|
||||||
|
|
||||||
|
AList is an open-source software, if you happen to like this project and want me to keep going, please consider sponsoring me or providing a single donation! Thanks for all the love and support:
|
||||||
|
https://alist.nn.ci/guide/sponsor.html
|
||||||
|
|
||||||
|
### Special sponsors
|
||||||
|
|
||||||
- [找资源 - 阿里云盘资源搜索引擎](https://zhaoziyuan.la/)
|
- [找资源 - 阿里云盘资源搜索引擎](https://zhaoziyuan.la/)
|
||||||
- [KinhDown 百度云盘不限速下载!永久免费!已稳定运行3年!非常可靠!](https://kinhdown.com/?Type=Tutorials)
|
- [KinhDown 百度云盘不限速下载!永久免费!已稳定运行3年!非常可靠!Q群 -> 786799372](https://kinhdown.com)
|
||||||
- [JetBrains: Essential tools for software developers and teams](https://www.jetbrains.com/)
|
- [JetBrains: Essential tools for software developers and teams](https://www.jetbrains.com/)
|
||||||
|
|
||||||
|
## Contributors
|
||||||
|
|
||||||
|
Thanks goes to these wonderful people:
|
||||||
|
|
||||||
|
[](https://github.com/alist-org/alist/graphs/contributors)
|
||||||
|
|
||||||
## License
|
## License
|
||||||
|
|
||||||
The `AList` is open-source software licensed under the AGPL-3.0 license.
|
The `AList` is open-source software licensed under the AGPL-3.0 license.
|
||||||
@ -99,4 +129,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=YJJj2Gwb)
|
> [@Blog](https://nn.ci/) · [@GitHub](https://github.com/Xhofe) · [@TelegramGroup](https://t.me/alist_chat) · [@Discord](https://discord.gg/F4ymsH4xv2)
|
||||||
|
57
README_cn.md
57
README_cn.md
@ -1,35 +1,45 @@
|
|||||||
<div align="center">
|
<div align="center">
|
||||||
<a href="https://alist.nn.ci"><img height="100px" alt="logo" src="https://cdn.jsdelivr.net/gh/alist-org/logo@main/logo.svg"/></a>
|
<a href="https://alist.nn.ci"><img height="100px" alt="logo" src="https://cdn.jsdelivr.net/gh/alist-org/logo@main/logo.svg"/></a>
|
||||||
<p><em>🗂一个支持多存储的文件列表程序,使用 Gin 和 Solidjs。</em></p>
|
<p><em>🗂一个支持多存储的文件列表程序,使用 Gin 和 Solidjs。</em></p>
|
||||||
|
<div>
|
||||||
<a href="https://goreportcard.com/report/github.com/alist-org/alist/v3">
|
<a href="https://goreportcard.com/report/github.com/alist-org/alist/v3">
|
||||||
<img src="https://goreportcard.com/badge/github.com/alist-org/alist/v3" alt="latest version" />
|
<img src="https://goreportcard.com/badge/github.com/alist-org/alist/v3" alt="latest version" />
|
||||||
</a>
|
</a>
|
||||||
<a href="https://github.com/Xhofe/alist/blob/main/LICENSE">
|
<a href="https://github.com/Xhofe/alist/blob/main/LICENSE">
|
||||||
<img src="https://img.shields.io/github/license/Xhofe/alist" alt="License" />
|
<img src="https://img.shields.io/github/license/Xhofe/alist" alt="License" />
|
||||||
</a>
|
</a>
|
||||||
<a href="https://github.com/Xhofe/alist/discussions">
|
|
||||||
<img src="https://img.shields.io/github/discussions/Xhofe/alist?color=%23ED8936" alt="discussions" />
|
|
||||||
</a>
|
|
||||||
<a href="https://github.com/Xhofe/alist/actions?query=workflow%3ABuild">
|
<a href="https://github.com/Xhofe/alist/actions?query=workflow%3ABuild">
|
||||||
<img src="https://img.shields.io/github/workflow/status/Xhofe/alist/build" alt="Build status" />
|
<img src="https://img.shields.io/github/workflow/status/Xhofe/alist/build" alt="Build status" />
|
||||||
</a>
|
</a>
|
||||||
<a href="https://github.com/Xhofe/alist/releases">
|
<a href="https://github.com/Xhofe/alist/releases">
|
||||||
<img src="https://img.shields.io/github/release/Xhofe/alist" alt="latest version" />
|
<img src="https://img.shields.io/github/release/Xhofe/alist" alt="latest version" />
|
||||||
</a>
|
</a>
|
||||||
<a href="https://github.com/Xhofe/alist/releases">
|
|
||||||
<img src="https://img.shields.io/github/downloads/Xhofe/alist/total?color=%239F7AEA" alt="Downloads" />
|
|
||||||
</a>
|
|
||||||
<a title="Crowdin" target="_blank" href="https://crwd.in/alist">
|
<a title="Crowdin" target="_blank" href="https://crwd.in/alist">
|
||||||
<img src="https://badges.crowdin.net/alist/localized.svg">
|
<img src="https://badges.crowdin.net/alist/localized.svg">
|
||||||
</a>
|
</a>
|
||||||
<a href="https://pay.xhofe.top">
|
</div>
|
||||||
<img src="https://img.shields.io/badge/%24-sponsor-ff69b4.svg" alt="sponsor" />
|
<div>
|
||||||
|
<a href="https://github.com/Xhofe/alist/discussions">
|
||||||
|
<img src="https://img.shields.io/github/discussions/Xhofe/alist?color=%23ED8936" alt="discussions" />
|
||||||
</a>
|
</a>
|
||||||
|
<a href="https://discord.gg/F4ymsH4xv2">
|
||||||
|
<img src="https://img.shields.io/discord/1018870125102895134?logo=discord" alt="discussions" />
|
||||||
|
</a>
|
||||||
|
<a href="https://github.com/Xhofe/alist/releases">
|
||||||
|
<img src="https://img.shields.io/github/downloads/Xhofe/alist/total?color=%239F7AEA&logo=github" alt="Downloads" />
|
||||||
|
</a>
|
||||||
|
<a href="https://hub.docker.com/r/xhofe/alist">
|
||||||
|
<img src="https://img.shields.io/docker/pulls/xhofe/alist?color=%2348BB78&logo=docker&label=pulls" alt="Downloads" />
|
||||||
|
</a>
|
||||||
|
<a href="https://alist.nn.ci/zh/guide/sponsor.html">
|
||||||
|
<img src="https://img.shields.io/badge/%24-sponsor-F87171.svg" alt="sponsor" />
|
||||||
|
</a>
|
||||||
|
</div>
|
||||||
</div>
|
</div>
|
||||||
|
|
||||||
---
|
---
|
||||||
|
|
||||||
[English](./README.md) | 中文 | [Contributors](./CONTRIBUTORS.md) | [Contributing](./CONTRIBUTING.md)
|
[English](./README.md) | 中文 | [Contributing](./CONTRIBUTING.md) | [CODE_OF_CONDUCT](./CODE_OF_CONDUCT.md)
|
||||||
|
|
||||||
## Features
|
## Features
|
||||||
|
|
||||||
@ -51,6 +61,14 @@
|
|||||||
- [x] [Yandex.Disk](https://disk.yandex.com/)
|
- [x] [Yandex.Disk](https://disk.yandex.com/)
|
||||||
- [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://www.lanzou.com/)
|
||||||
|
- [x] [阿里云盘分享](https://www.aliyundrive.com/)
|
||||||
|
- [x] [谷歌相册](https://photos.google.com/)
|
||||||
|
- [x] [Mega.nz](https://mega.nz)
|
||||||
|
- [x] [一刻相册](https://photo.baidu.com/)
|
||||||
|
- [x] SMB
|
||||||
|
- [x] [115](https://115.com/)
|
||||||
- [x] 部署方便,开箱即用
|
- [x] 部署方便,开箱即用
|
||||||
- [x] 文件预览(PDF、markdown、代码、纯文本……)
|
- [x] 文件预览(PDF、markdown、代码、纯文本……)
|
||||||
- [x] 画廊模式下的图像预览
|
- [x] 画廊模式下的图像预览
|
||||||
@ -75,18 +93,29 @@
|
|||||||
|
|
||||||
## Demo
|
## Demo
|
||||||
|
|
||||||
<https://pan.nn.ci>
|
<https://al.nn.ci>
|
||||||
|
|
||||||
## Discussion
|
## Discussion
|
||||||
|
|
||||||
一般问题请到[讨论论坛](https://github.com/Xhofe/alist/discussions) ,**issue仅针对错误报告和功能请求。**
|
一般问题请到[讨论论坛](https://github.com/Xhofe/alist/discussions) ,**issue仅针对错误报告和功能请求。**
|
||||||
|
|
||||||
## Special sponsors
|
## Sponsor
|
||||||
|
|
||||||
|
AList 是一个开源软件,如果你碰巧喜欢这个项目,并希望我继续下去,请考虑赞助我或提供一个单一的捐款!感谢所有的爱和支持:https://alist.nn.ci/zh/guide/sponsor.html
|
||||||
|
|
||||||
|
### Special sponsors
|
||||||
|
|
||||||
- [找资源 - 阿里云盘资源搜索引擎](https://zhaoziyuan.la/)
|
- [找资源 - 阿里云盘资源搜索引擎](https://zhaoziyuan.la/)
|
||||||
- [KinhDown 百度云盘不限速下载!永久免费!已稳定运行3年!非常可靠!](https://kinhdown.com/?Type=Tutorials)
|
- [KinhDown 百度云盘不限速下载!永久免费!已稳定运行3年!非常可靠!Q群 -> 786799372](https://kinhdown.com)
|
||||||
- [JetBrains: Essential tools for software developers and teams](https://www.jetbrains.com/)
|
- [JetBrains: Essential tools for software developers and teams](https://www.jetbrains.com/)
|
||||||
|
|
||||||
## 许可
|
## Contributors
|
||||||
|
|
||||||
|
Thanks goes to these wonderful people:
|
||||||
|
|
||||||
|
[](https://github.com/alist-org/alist/graphs/contributors)
|
||||||
|
|
||||||
|
## 许可
|
||||||
|
|
||||||
`AList` 是在 AGPL-3.0 许可下许可的开源软件。
|
`AList` 是在 AGPL-3.0 许可下许可的开源软件。
|
||||||
|
|
||||||
@ -99,4 +128,4 @@
|
|||||||
|
|
||||||
---
|
---
|
||||||
|
|
||||||
> [@博客](https://nn.ci/) · [@GitHub](https://github.com/Xhofe) · [@Telegram群](https://t.me/alist_chat) · [@QQ群](https://jq.qq.com/?_wv=1027&k=YJJj2Gwb)
|
> [@博客](https://nn.ci/) · [@GitHub](https://github.com/Xhofe) · [@Telegram群](https://t.me/alist_chat) · [@Discord](https://discord.gg/F4ymsH4xv2)
|
11
build.sh
11
build.sh
@ -12,7 +12,8 @@ else
|
|||||||
webVersion=$(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')
|
webVersion=$(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')
|
||||||
fi
|
fi
|
||||||
|
|
||||||
echo "build version: $gitTag"
|
echo "backend version: $version"
|
||||||
|
echo "frontend version: $webVersion"
|
||||||
|
|
||||||
ldflags="\
|
ldflags="\
|
||||||
-w -s \
|
-w -s \
|
||||||
@ -25,11 +26,11 @@ ldflags="\
|
|||||||
"
|
"
|
||||||
|
|
||||||
FetchWebDev() {
|
FetchWebDev() {
|
||||||
curl -L https://codeload.github.com/alist-org/web-dist/tar.gz/refs/heads/main -o web-dist-main.tar.gz
|
curl -L https://codeload.github.com/alist-org/web-dist/tar.gz/refs/heads/dev -o web-dist-dev.tar.gz
|
||||||
tar -zxvf web-dist-main.tar.gz
|
tar -zxvf web-dist-dev.tar.gz
|
||||||
rm -rf public/dist
|
rm -rf public/dist
|
||||||
mv -f web-dist-main/dist public
|
mv -f web-dist-dev/dist public
|
||||||
rm -rf web-dist-main web-dist-main.tar.gz
|
rm -rf web-dist-dev web-dist-dev.tar.gz
|
||||||
}
|
}
|
||||||
|
|
||||||
FetchWebRelease() {
|
FetchWebRelease() {
|
||||||
|
@ -11,15 +11,16 @@ import (
|
|||||||
|
|
||||||
// passwordCmd represents the password command
|
// passwordCmd represents the password command
|
||||||
var passwordCmd = &cobra.Command{
|
var passwordCmd = &cobra.Command{
|
||||||
Use: "password",
|
Use: "admin",
|
||||||
Short: "Show admin user's password",
|
Aliases: []string{"password"},
|
||||||
|
Short: "Show admin user's info",
|
||||||
Run: func(cmd *cobra.Command, args []string) {
|
Run: func(cmd *cobra.Command, args []string) {
|
||||||
Init()
|
Init()
|
||||||
admin, err := db.GetAdmin()
|
admin, err := db.GetAdmin()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
utils.Log.Errorf("failed get admin user: %+v", err)
|
utils.Log.Errorf("failed get admin user: %+v", err)
|
||||||
} else {
|
} else {
|
||||||
utils.Log.Infof("admin user's password is: %s", admin.Password)
|
utils.Log.Infof("admin user's info: \nusername: %s\npassword: %s", admin.Username, admin.Password)
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
}
|
}
|
@ -1,8 +1,14 @@
|
|||||||
package cmd
|
package cmd
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"os"
|
||||||
|
"path/filepath"
|
||||||
|
"strconv"
|
||||||
|
|
||||||
"github.com/alist-org/alist/v3/internal/bootstrap"
|
"github.com/alist-org/alist/v3/internal/bootstrap"
|
||||||
"github.com/alist-org/alist/v3/internal/bootstrap/data"
|
"github.com/alist-org/alist/v3/internal/bootstrap/data"
|
||||||
|
"github.com/alist-org/alist/v3/pkg/utils"
|
||||||
|
log "github.com/sirupsen/logrus"
|
||||||
)
|
)
|
||||||
|
|
||||||
func Init() {
|
func Init() {
|
||||||
@ -11,3 +17,27 @@ func Init() {
|
|||||||
bootstrap.InitDB()
|
bootstrap.InitDB()
|
||||||
data.InitData()
|
data.InitData()
|
||||||
}
|
}
|
||||||
|
|
||||||
|
var pid = -1
|
||||||
|
var pidFile string
|
||||||
|
|
||||||
|
func initDaemon() {
|
||||||
|
ex, err := os.Executable()
|
||||||
|
if err != nil {
|
||||||
|
log.Fatal(err)
|
||||||
|
}
|
||||||
|
exPath := filepath.Dir(ex)
|
||||||
|
_ = os.MkdirAll(filepath.Join(exPath, "daemon"), 0700)
|
||||||
|
pidFile = filepath.Join(exPath, "daemon/pid")
|
||||||
|
if utils.Exists(pidFile) {
|
||||||
|
bytes, err := os.ReadFile(pidFile)
|
||||||
|
if err != nil {
|
||||||
|
log.Fatal("failed to read pid file", err)
|
||||||
|
}
|
||||||
|
id, err := strconv.Atoi(string(bytes))
|
||||||
|
if err != nil {
|
||||||
|
log.Fatal("failed to parse pid data", err)
|
||||||
|
}
|
||||||
|
pid = id
|
||||||
|
}
|
||||||
|
}
|
||||||
|
@ -1,8 +1,9 @@
|
|||||||
package flags
|
package flags
|
||||||
|
|
||||||
var (
|
var (
|
||||||
Config string // config file
|
DataDir string
|
||||||
Debug bool
|
Debug bool
|
||||||
NoPrefix bool
|
NoPrefix bool
|
||||||
Dev bool
|
Dev bool
|
||||||
|
ForceBinDir bool
|
||||||
)
|
)
|
||||||
|
32
cmd/restart.go
Normal file
32
cmd/restart.go
Normal file
@ -0,0 +1,32 @@
|
|||||||
|
/*
|
||||||
|
Copyright © 2022 NAME HERE <EMAIL ADDRESS>
|
||||||
|
*/
|
||||||
|
package cmd
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/spf13/cobra"
|
||||||
|
)
|
||||||
|
|
||||||
|
// restartCmd represents the restart command
|
||||||
|
var restartCmd = &cobra.Command{
|
||||||
|
Use: "restart",
|
||||||
|
Short: "Restart alist server by daemon/pid file",
|
||||||
|
Run: func(cmd *cobra.Command, args []string) {
|
||||||
|
stop()
|
||||||
|
start()
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
rootCmd.AddCommand(restartCmd)
|
||||||
|
|
||||||
|
// Here you will define your flags and configuration settings.
|
||||||
|
|
||||||
|
// Cobra supports Persistent Flags which will work for this command
|
||||||
|
// and all subcommands, e.g.:
|
||||||
|
// restartCmd.PersistentFlags().String("foo", "", "A help for foo")
|
||||||
|
|
||||||
|
// Cobra supports local flags which will only run when this command
|
||||||
|
// is called directly, e.g.:
|
||||||
|
// restartCmd.Flags().BoolP("toggle", "t", false, "Help message for toggle")
|
||||||
|
}
|
@ -24,8 +24,9 @@ func Execute() {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func init() {
|
func init() {
|
||||||
rootCmd.PersistentFlags().StringVar(&flags.Config, "conf", "data/config.json", "config file")
|
rootCmd.PersistentFlags().StringVar(&flags.DataDir, "data", "data", "config file")
|
||||||
rootCmd.PersistentFlags().BoolVar(&flags.Debug, "debug", false, "start with debug mode")
|
rootCmd.PersistentFlags().BoolVar(&flags.Debug, "debug", false, "start with debug mode")
|
||||||
rootCmd.PersistentFlags().BoolVar(&flags.NoPrefix, "no-prefix", false, "disable env prefix")
|
rootCmd.PersistentFlags().BoolVar(&flags.NoPrefix, "no-prefix", false, "disable env prefix")
|
||||||
rootCmd.PersistentFlags().BoolVar(&flags.Dev, "dev", false, "start with dev mode")
|
rootCmd.PersistentFlags().BoolVar(&flags.Dev, "dev", false, "start with dev mode")
|
||||||
|
rootCmd.PersistentFlags().BoolVar(&flags.ForceBinDir, "force-bin-dir", false, "Force to use the directory where the binary file is located as data directory")
|
||||||
}
|
}
|
||||||
|
@ -61,7 +61,7 @@ the address is defined in config file`,
|
|||||||
<-quit
|
<-quit
|
||||||
utils.Log.Println("Shutdown Server ...")
|
utils.Log.Println("Shutdown Server ...")
|
||||||
|
|
||||||
ctx, cancel := context.WithTimeout(context.Background(), 3*time.Second)
|
ctx, cancel := context.WithTimeout(context.Background(), 1*time.Second)
|
||||||
defer cancel()
|
defer cancel()
|
||||||
if err := srv.Shutdown(ctx); err != nil {
|
if err := srv.Shutdown(ctx); err != nil {
|
||||||
utils.Log.Fatal("Server Shutdown:", err)
|
utils.Log.Fatal("Server Shutdown:", err)
|
||||||
@ -69,7 +69,7 @@ the address is defined in config file`,
|
|||||||
// catching ctx.Done(). timeout of 3 seconds.
|
// catching ctx.Done(). timeout of 3 seconds.
|
||||||
select {
|
select {
|
||||||
case <-ctx.Done():
|
case <-ctx.Done():
|
||||||
utils.Log.Println("timeout of 3 seconds.")
|
utils.Log.Println("timeout of 1 seconds.")
|
||||||
}
|
}
|
||||||
utils.Log.Println("Server exiting")
|
utils.Log.Println("Server exiting")
|
||||||
},
|
},
|
||||||
@ -88,3 +88,12 @@ func init() {
|
|||||||
// is called directly, e.g.:
|
// is called directly, e.g.:
|
||||||
// serverCmd.Flags().BoolP("toggle", "t", false, "Help message for toggle")
|
// serverCmd.Flags().BoolP("toggle", "t", false, "Help message for toggle")
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// OutAlistInit 暴露用于外部启动server的函数
|
||||||
|
func OutAlistInit() {
|
||||||
|
var (
|
||||||
|
cmd *cobra.Command
|
||||||
|
args []string
|
||||||
|
)
|
||||||
|
serverCmd.Run(cmd, args)
|
||||||
|
}
|
||||||
|
71
cmd/start.go
Normal file
71
cmd/start.go
Normal file
@ -0,0 +1,71 @@
|
|||||||
|
/*
|
||||||
|
Copyright © 2022 NAME HERE <EMAIL ADDRESS>
|
||||||
|
*/
|
||||||
|
package cmd
|
||||||
|
|
||||||
|
import (
|
||||||
|
"os"
|
||||||
|
"os/exec"
|
||||||
|
"path/filepath"
|
||||||
|
"strconv"
|
||||||
|
|
||||||
|
log "github.com/sirupsen/logrus"
|
||||||
|
"github.com/spf13/cobra"
|
||||||
|
)
|
||||||
|
|
||||||
|
// startCmd represents the start command
|
||||||
|
var startCmd = &cobra.Command{
|
||||||
|
Use: "start",
|
||||||
|
Short: "Silent start alist server with `--force-bin-dir`",
|
||||||
|
Run: func(cmd *cobra.Command, args []string) {
|
||||||
|
start()
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
func start() {
|
||||||
|
initDaemon()
|
||||||
|
if pid != -1 {
|
||||||
|
_, err := os.FindProcess(pid)
|
||||||
|
if err == nil {
|
||||||
|
log.Info("alist already started, pid ", pid)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
}
|
||||||
|
args := os.Args
|
||||||
|
args[1] = "server"
|
||||||
|
args = append(args, "--force-bin-dir")
|
||||||
|
cmd := &exec.Cmd{
|
||||||
|
Path: args[0],
|
||||||
|
Args: args,
|
||||||
|
Env: os.Environ(),
|
||||||
|
}
|
||||||
|
stdout, err := os.OpenFile(filepath.Join(filepath.Dir(pidFile), "start.log"), os.O_WRONLY|os.O_APPEND|os.O_CREATE, 0666)
|
||||||
|
if err != nil {
|
||||||
|
log.Fatal(os.Getpid(), ": failed to open start log file:", err)
|
||||||
|
}
|
||||||
|
cmd.Stderr = stdout
|
||||||
|
cmd.Stdout = stdout
|
||||||
|
err = cmd.Start()
|
||||||
|
if err != nil {
|
||||||
|
log.Fatal("failed to start children process: ", err)
|
||||||
|
}
|
||||||
|
log.Infof("success start pid: %d", cmd.Process.Pid)
|
||||||
|
err = os.WriteFile(pidFile, []byte(strconv.Itoa(cmd.Process.Pid)), 0666)
|
||||||
|
if err != nil {
|
||||||
|
log.Warn("failed to record pid, you may not be able to stop the program with `./alist stop`")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
rootCmd.AddCommand(startCmd)
|
||||||
|
|
||||||
|
// Here you will define your flags and configuration settings.
|
||||||
|
|
||||||
|
// Cobra supports Persistent Flags which will work for this command
|
||||||
|
// and all subcommands, e.g.:
|
||||||
|
// startCmd.PersistentFlags().String("foo", "", "A help for foo")
|
||||||
|
|
||||||
|
// Cobra supports local flags which will only run when this command
|
||||||
|
// is called directly, e.g.:
|
||||||
|
// startCmd.Flags().BoolP("toggle", "t", false, "Help message for toggle")
|
||||||
|
}
|
58
cmd/stop.go
Normal file
58
cmd/stop.go
Normal file
@ -0,0 +1,58 @@
|
|||||||
|
/*
|
||||||
|
Copyright © 2022 NAME HERE <EMAIL ADDRESS>
|
||||||
|
*/
|
||||||
|
package cmd
|
||||||
|
|
||||||
|
import (
|
||||||
|
"os"
|
||||||
|
|
||||||
|
log "github.com/sirupsen/logrus"
|
||||||
|
"github.com/spf13/cobra"
|
||||||
|
)
|
||||||
|
|
||||||
|
// stopCmd represents the stop command
|
||||||
|
var stopCmd = &cobra.Command{
|
||||||
|
Use: "stop",
|
||||||
|
Short: "Stop alist server by daemon/pid file",
|
||||||
|
Run: func(cmd *cobra.Command, args []string) {
|
||||||
|
stop()
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
func stop() {
|
||||||
|
initDaemon()
|
||||||
|
if pid == -1 {
|
||||||
|
log.Info("Seems not have been started. Try use `alist start` to start server.")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
process, err := os.FindProcess(pid)
|
||||||
|
if err != nil {
|
||||||
|
log.Errorf("failed to find process by pid: %d, reason: %v", pid, process)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
err = process.Kill()
|
||||||
|
if err != nil {
|
||||||
|
log.Errorf("failed to kill process %d: %v", pid, err)
|
||||||
|
} else {
|
||||||
|
log.Info("killed process: ", pid)
|
||||||
|
}
|
||||||
|
err = os.Remove(pidFile)
|
||||||
|
if err != nil {
|
||||||
|
log.Errorf("failed to remove pid file")
|
||||||
|
}
|
||||||
|
pid = -1
|
||||||
|
}
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
rootCmd.AddCommand(stopCmd)
|
||||||
|
|
||||||
|
// Here you will define your flags and configuration settings.
|
||||||
|
|
||||||
|
// Cobra supports Persistent Flags which will work for this command
|
||||||
|
// and all subcommands, e.g.:
|
||||||
|
// stopCmd.PersistentFlags().String("foo", "", "A help for foo")
|
||||||
|
|
||||||
|
// Cobra supports local flags which will only run when this command
|
||||||
|
// is called directly, e.g.:
|
||||||
|
// stopCmd.Flags().BoolP("toggle", "t", false, "Help message for toggle")
|
||||||
|
}
|
98
drivers/115/driver.go
Normal file
98
drivers/115/driver.go
Normal file
@ -0,0 +1,98 @@
|
|||||||
|
package _115
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"os"
|
||||||
|
|
||||||
|
driver115 "github.com/SheltonZhu/115driver/pkg/driver"
|
||||||
|
"github.com/alist-org/alist/v3/internal/driver"
|
||||||
|
"github.com/alist-org/alist/v3/internal/model"
|
||||||
|
"github.com/alist-org/alist/v3/pkg/utils"
|
||||||
|
"github.com/pkg/errors"
|
||||||
|
)
|
||||||
|
|
||||||
|
type Pan115 struct {
|
||||||
|
model.Storage
|
||||||
|
Addition
|
||||||
|
client *driver115.Pan115Client
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *Pan115) Config() driver.Config {
|
||||||
|
return config
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *Pan115) GetAddition() driver.Additional {
|
||||||
|
return d.Addition
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *Pan115) Init(ctx context.Context, storage model.Storage) error {
|
||||||
|
d.Storage = storage
|
||||||
|
err := utils.Json.UnmarshalFromString(d.Storage.Addition, &d.Addition)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
return d.login()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *Pan115) Drop(ctx context.Context) error {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *Pan115) List(ctx context.Context, dir model.Obj, args model.ListArgs) ([]model.Obj, error) {
|
||||||
|
files, err := d.getFiles(dir.GetID())
|
||||||
|
if err != nil && !errors.Is(err, driver115.ErrNotExist) {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return utils.SliceConvert(files, func(src driver115.File) (model.Obj, error) {
|
||||||
|
return src, nil
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *Pan115) Link(ctx context.Context, file model.Obj, args model.LinkArgs) (*model.Link, error) {
|
||||||
|
downloadInfo, err := d.client.Download(file.(driver115.File).PickCode)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
link := &model.Link{
|
||||||
|
URL: downloadInfo.Url.Url,
|
||||||
|
Header: downloadInfo.Header,
|
||||||
|
}
|
||||||
|
return link, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *Pan115) MakeDir(ctx context.Context, parentDir model.Obj, dirName string) error {
|
||||||
|
if _, err := d.client.Mkdir(parentDir.GetID(), dirName); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *Pan115) Move(ctx context.Context, srcObj, dstDir model.Obj) error {
|
||||||
|
return d.client.Move(dstDir.GetID(), srcObj.GetID())
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *Pan115) Rename(ctx context.Context, srcObj model.Obj, newName string) error {
|
||||||
|
return d.client.Rename(srcObj.GetID(), newName)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *Pan115) Copy(ctx context.Context, srcObj, dstDir model.Obj) error {
|
||||||
|
return d.client.Copy(dstDir.GetID(), srcObj.GetID())
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *Pan115) Remove(ctx context.Context, obj model.Obj) error {
|
||||||
|
return d.client.Delete(obj.GetID())
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *Pan115) Put(ctx context.Context, dstDir model.Obj, stream model.FileStreamer, up driver.UpdateProgress) error {
|
||||||
|
tempFile, err := utils.CreateTempFile(stream.GetReadCloser())
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
defer func() {
|
||||||
|
_ = tempFile.Close()
|
||||||
|
_ = os.Remove(tempFile.Name())
|
||||||
|
}()
|
||||||
|
return d.client.UploadFastOrByMultipart(dstDir.GetID(), stream.GetName(), stream.GetSize(), tempFile)
|
||||||
|
}
|
||||||
|
|
||||||
|
var _ driver.Driver = (*Pan115)(nil)
|
27
drivers/115/meta.go
Normal file
27
drivers/115/meta.go
Normal file
@ -0,0 +1,27 @@
|
|||||||
|
package _115
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/alist-org/alist/v3/internal/driver"
|
||||||
|
"github.com/alist-org/alist/v3/internal/op"
|
||||||
|
)
|
||||||
|
|
||||||
|
type Addition struct {
|
||||||
|
Cookie string `json:"cookie"`
|
||||||
|
QRCodeToken string `json:"qrcode_token"`
|
||||||
|
driver.RootID
|
||||||
|
}
|
||||||
|
|
||||||
|
var config = driver.Config{
|
||||||
|
Name: "115 Cloud",
|
||||||
|
DefaultRoot: "0",
|
||||||
|
OnlyProxy: true,
|
||||||
|
OnlyLocal: true,
|
||||||
|
}
|
||||||
|
|
||||||
|
func New() driver.Driver {
|
||||||
|
return &Pan115{}
|
||||||
|
}
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
op.RegisterDriver(config, New)
|
||||||
|
}
|
8
drivers/115/types.go
Normal file
8
drivers/115/types.go
Normal file
@ -0,0 +1,8 @@
|
|||||||
|
package _115
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/SheltonZhu/115driver/pkg/driver"
|
||||||
|
"github.com/alist-org/alist/v3/internal/model"
|
||||||
|
)
|
||||||
|
|
||||||
|
var _ model.Obj = (*driver.File)(nil)
|
49
drivers/115/util.go
Normal file
49
drivers/115/util.go
Normal file
@ -0,0 +1,49 @@
|
|||||||
|
package _115
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
|
||||||
|
"github.com/SheltonZhu/115driver/pkg/driver"
|
||||||
|
"github.com/pkg/errors"
|
||||||
|
)
|
||||||
|
|
||||||
|
var UserAgent = "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/106.0.0.0 Safari/537.36 115Browser/23.9.3.2 115disk/30.1.0"
|
||||||
|
|
||||||
|
func (d *Pan115) login() error {
|
||||||
|
var err error
|
||||||
|
opts := []driver.Option{
|
||||||
|
driver.UA(UserAgent),
|
||||||
|
}
|
||||||
|
d.client = driver.New(opts...)
|
||||||
|
cr := &driver.Credential{}
|
||||||
|
if d.Addition.QRCodeToken != "" {
|
||||||
|
s := &driver.QRCodeSession{
|
||||||
|
UID: d.Addition.QRCodeToken,
|
||||||
|
}
|
||||||
|
if cr, err = d.client.QRCodeLogin(s); err != nil {
|
||||||
|
return errors.Wrap(err, "failed to login by qrcode")
|
||||||
|
}
|
||||||
|
d.Addition.Cookie = fmt.Sprintf("UID=%s;CID=%s;SEID=%s", cr.UID, cr.CID, cr.SEID)
|
||||||
|
d.Addition.QRCodeToken = ""
|
||||||
|
} else if d.Addition.Cookie != "" {
|
||||||
|
if err = cr.FromCookie(d.Addition.Cookie); err != nil {
|
||||||
|
return errors.Wrap(err, "failed to login by cookies")
|
||||||
|
}
|
||||||
|
d.client.ImportCredential(cr)
|
||||||
|
} else {
|
||||||
|
return errors.New("missing cookie or qrcode account")
|
||||||
|
}
|
||||||
|
return d.client.LoginCheck()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *Pan115) getFiles(fileId string) ([]driver.File, error) {
|
||||||
|
res := make([]driver.File, 0)
|
||||||
|
files, err := d.client.List(fileId)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
for _, file := range *files {
|
||||||
|
res = append(res, file)
|
||||||
|
}
|
||||||
|
return res, nil
|
||||||
|
}
|
@ -6,13 +6,14 @@ import (
|
|||||||
"crypto/md5"
|
"crypto/md5"
|
||||||
"encoding/binary"
|
"encoding/binary"
|
||||||
"encoding/hex"
|
"encoding/hex"
|
||||||
|
"encoding/base64"
|
||||||
"fmt"
|
"fmt"
|
||||||
"io"
|
"io"
|
||||||
"net/http"
|
"net/http"
|
||||||
|
"net/url"
|
||||||
"os"
|
"os"
|
||||||
|
|
||||||
"github.com/alist-org/alist/v3/drivers/base"
|
"github.com/alist-org/alist/v3/drivers/base"
|
||||||
"github.com/alist-org/alist/v3/internal/conf"
|
|
||||||
"github.com/alist-org/alist/v3/internal/driver"
|
"github.com/alist-org/alist/v3/internal/driver"
|
||||||
"github.com/alist-org/alist/v3/internal/errs"
|
"github.com/alist-org/alist/v3/internal/errs"
|
||||||
"github.com/alist-org/alist/v3/internal/model"
|
"github.com/alist-org/alist/v3/internal/model"
|
||||||
@ -22,6 +23,7 @@ import (
|
|||||||
"github.com/aws/aws-sdk-go/aws/session"
|
"github.com/aws/aws-sdk-go/aws/session"
|
||||||
"github.com/aws/aws-sdk-go/service/s3/s3manager"
|
"github.com/aws/aws-sdk-go/service/s3/s3manager"
|
||||||
"github.com/go-resty/resty/v2"
|
"github.com/go-resty/resty/v2"
|
||||||
|
log "github.com/sirupsen/logrus"
|
||||||
)
|
)
|
||||||
|
|
||||||
type Pan123 struct {
|
type Pan123 struct {
|
||||||
@ -68,9 +70,55 @@ func (d *Pan123) List(ctx context.Context, dir model.Obj, args model.ListArgs) (
|
|||||||
|
|
||||||
func (d *Pan123) Link(ctx context.Context, file model.Obj, args model.LinkArgs) (*model.Link, error) {
|
func (d *Pan123) Link(ctx context.Context, file model.Obj, args model.LinkArgs) (*model.Link, error) {
|
||||||
if f, ok := file.(File); ok {
|
if f, ok := file.(File); ok {
|
||||||
return &model.Link{
|
var resp DownResp
|
||||||
URL: f.DownloadUrl,
|
var headers map[string]string
|
||||||
}, nil
|
if !utils.IsLocalIPAddr(args.IP) {
|
||||||
|
headers = map[string]string{
|
||||||
|
//"X-Real-IP": "1.1.1.1",
|
||||||
|
"X-Forwarded-For": args.IP,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
data := base.Json{
|
||||||
|
"driveId": 0,
|
||||||
|
"etag": f.Etag,
|
||||||
|
"fileId": f.FileId,
|
||||||
|
"fileName": f.FileName,
|
||||||
|
"s3keyFlag": f.S3KeyFlag,
|
||||||
|
"size": f.Size,
|
||||||
|
"type": f.Type,
|
||||||
|
}
|
||||||
|
_, err := d.request("https://www.123pan.com/api/file/download_info", http.MethodPost, func(req *resty.Request) {
|
||||||
|
req.SetBody(data).SetHeaders(headers)
|
||||||
|
}, &resp)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
u, err := url.Parse(resp.Data.DownloadUrl)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
nu := u.Query().Get("params")
|
||||||
|
if nu != "" {
|
||||||
|
du, _ := base64.StdEncoding.DecodeString(nu)
|
||||||
|
u, err = url.Parse(string(du))
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
u_ := fmt.Sprintf("https://%s%s", u.Host, u.Path)
|
||||||
|
res, err := base.NoRedirectClient.R().SetQueryParamsFromValues(u.Query()).Head(u_)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
log.Debug(res.String())
|
||||||
|
link := model.Link{
|
||||||
|
URL: resp.Data.DownloadUrl,
|
||||||
|
}
|
||||||
|
log.Debugln("res code: ", res.StatusCode())
|
||||||
|
if res.StatusCode() == 302 {
|
||||||
|
link.URL = res.Header().Get("location")
|
||||||
|
}
|
||||||
|
return &link, nil
|
||||||
} else {
|
} else {
|
||||||
return nil, fmt.Errorf("can't convert obj")
|
return nil, fmt.Errorf("can't convert obj")
|
||||||
}
|
}
|
||||||
@ -125,7 +173,7 @@ func (d *Pan123) Remove(ctx context.Context, obj model.Obj) error {
|
|||||||
"operation": true,
|
"operation": true,
|
||||||
"fileTrashInfoList": []File{f},
|
"fileTrashInfoList": []File{f},
|
||||||
}
|
}
|
||||||
_, err := d.request("https://www.123pan.com/a/api/file/trash", http.MethodPost, func(req *resty.Request) {
|
_, err := d.request("https://www.123pan.com/b/api/file/trash", http.MethodPost, func(req *resty.Request) {
|
||||||
req.SetBody(data)
|
req.SetBody(data)
|
||||||
}, nil)
|
}, nil)
|
||||||
return err
|
return err
|
||||||
@ -153,7 +201,7 @@ func (d *Pan123) Put(ctx context.Context, dstDir model.Obj, stream model.FileStr
|
|||||||
uploadFile = io.MultiReader(buf, stream)
|
uploadFile = io.MultiReader(buf, stream)
|
||||||
} else {
|
} else {
|
||||||
// 计算完整文件MD5
|
// 计算完整文件MD5
|
||||||
tempFile, err := os.CreateTemp(conf.Conf.TempDir, "file-*")
|
tempFile, err := utils.CreateTempFile(stream.GetReadCloser())
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
@ -161,11 +209,9 @@ func (d *Pan123) Put(ctx context.Context, dstDir model.Obj, stream model.FileStr
|
|||||||
_ = tempFile.Close()
|
_ = tempFile.Close()
|
||||||
_ = os.Remove(tempFile.Name())
|
_ = os.Remove(tempFile.Name())
|
||||||
}()
|
}()
|
||||||
|
if _, err = io.Copy(h, tempFile); err != nil {
|
||||||
if _, err = io.Copy(io.MultiWriter(tempFile, h), stream); err != nil {
|
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
_, err = tempFile.Seek(0, io.SeekStart)
|
_, err = tempFile.Seek(0, io.SeekStart)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
@ -220,8 +266,4 @@ func (d *Pan123) Put(ctx context.Context, dstDir model.Obj, stream model.FileStr
|
|||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
func (d *Pan123) Other(ctx context.Context, args model.OtherArgs) (interface{}, error) {
|
|
||||||
return nil, errs.NotSupport
|
|
||||||
}
|
|
||||||
|
|
||||||
var _ driver.Driver = (*Pan123)(nil)
|
var _ driver.Driver = (*Pan123)(nil)
|
||||||
|
@ -8,7 +8,7 @@ import (
|
|||||||
type Addition struct {
|
type Addition struct {
|
||||||
Username string `json:"username" required:"true"`
|
Username string `json:"username" required:"true"`
|
||||||
Password string `json:"password" required:"true"`
|
Password string `json:"password" required:"true"`
|
||||||
OrderBy string `json:"order_by" type:"select" options:"name,fileId,updateAt,createAt" default:"name"`
|
OrderBy string `json:"order_by" type:"select" options:"file_name,size,update_at" default:"file_name"`
|
||||||
OrderDirection string `json:"order_direction" type:"select" options:"asc,desc" default:"asc"`
|
OrderDirection string `json:"order_direction" type:"select" options:"asc,desc" default:"asc"`
|
||||||
driver.RootID
|
driver.RootID
|
||||||
// define other
|
// define other
|
||||||
|
@ -6,6 +6,7 @@ import (
|
|||||||
"net/http"
|
"net/http"
|
||||||
|
|
||||||
"github.com/alist-org/alist/v3/drivers/base"
|
"github.com/alist-org/alist/v3/drivers/base"
|
||||||
|
"github.com/alist-org/alist/v3/pkg/utils"
|
||||||
"github.com/go-resty/resty/v2"
|
"github.com/go-resty/resty/v2"
|
||||||
jsoniter "github.com/json-iterator/go"
|
jsoniter "github.com/json-iterator/go"
|
||||||
)
|
)
|
||||||
@ -13,14 +14,24 @@ import (
|
|||||||
// do others that not defined in Driver interface
|
// do others that not defined in Driver interface
|
||||||
|
|
||||||
func (d *Pan123) login() error {
|
func (d *Pan123) login() error {
|
||||||
|
var body base.Json
|
||||||
url := "https://www.123pan.com/api/user/sign_in"
|
url := "https://www.123pan.com/api/user/sign_in"
|
||||||
|
if utils.IsEmailFormat(d.Username) {
|
||||||
|
body = base.Json{
|
||||||
|
"mail": d.Username,
|
||||||
|
"password": d.Password,
|
||||||
|
"type": 2,
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
body = base.Json{
|
||||||
|
"passport": d.Username,
|
||||||
|
"password": d.Password,
|
||||||
|
}
|
||||||
|
}
|
||||||
var resp TokenResp
|
var resp TokenResp
|
||||||
_, err := base.RestyClient.R().
|
_, err := base.RestyClient.R().
|
||||||
SetResult(&resp).
|
SetResult(&resp).
|
||||||
SetBody(base.Json{
|
SetBody(body).Post(url)
|
||||||
"passport": d.Username,
|
|
||||||
"password": d.Password,
|
|
||||||
}).Post(url)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
@ -319,8 +319,4 @@ func (d *Yun139) Put(ctx context.Context, dstDir model.Obj, stream model.FileStr
|
|||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (d *Yun139) Other(ctx context.Context, args model.OtherArgs) (interface{}, error) {
|
|
||||||
return nil, errs.NotSupport
|
|
||||||
}
|
|
||||||
|
|
||||||
var _ driver.Driver = (*Yun139)(nil)
|
var _ driver.Driver = (*Yun139)(nil)
|
||||||
|
@ -7,7 +7,6 @@ import (
|
|||||||
|
|
||||||
"github.com/alist-org/alist/v3/drivers/base"
|
"github.com/alist-org/alist/v3/drivers/base"
|
||||||
"github.com/alist-org/alist/v3/internal/driver"
|
"github.com/alist-org/alist/v3/internal/driver"
|
||||||
"github.com/alist-org/alist/v3/internal/errs"
|
|
||||||
"github.com/alist-org/alist/v3/internal/model"
|
"github.com/alist-org/alist/v3/internal/model"
|
||||||
"github.com/alist-org/alist/v3/pkg/utils"
|
"github.com/alist-org/alist/v3/pkg/utils"
|
||||||
"github.com/go-resty/resty/v2"
|
"github.com/go-resty/resty/v2"
|
||||||
@ -208,8 +207,4 @@ func (d *Cloud189) Put(ctx context.Context, dstDir model.Obj, stream model.FileS
|
|||||||
return d.newUpload(dstDir, stream, up)
|
return d.newUpload(dstDir, stream, up)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (d *Cloud189) Other(ctx context.Context, args model.OtherArgs) (interface{}, error) {
|
|
||||||
return nil, errs.NotSupport
|
|
||||||
}
|
|
||||||
|
|
||||||
var _ driver.Driver = (*Cloud189)(nil)
|
var _ driver.Driver = (*Cloud189)(nil)
|
||||||
|
@ -178,7 +178,6 @@ func (d *Cloud189) request(url string, method string, callback base.ReqCallback,
|
|||||||
func (d *Cloud189) getFiles(fileId string) ([]model.Obj, error) {
|
func (d *Cloud189) getFiles(fileId string) ([]model.Obj, error) {
|
||||||
res := make([]model.Obj, 0)
|
res := make([]model.Obj, 0)
|
||||||
pageNum := 1
|
pageNum := 1
|
||||||
loc, _ := time.LoadLocation("Local")
|
|
||||||
for {
|
for {
|
||||||
var resp Files
|
var resp Files
|
||||||
_, err := d.request("https://cloud.189.cn/api/open/file/listFiles.action", http.MethodGet, func(req *resty.Request) {
|
_, err := d.request("https://cloud.189.cn/api/open/file/listFiles.action", http.MethodGet, func(req *resty.Request) {
|
||||||
@ -200,7 +199,7 @@ func (d *Cloud189) getFiles(fileId string) ([]model.Obj, error) {
|
|||||||
break
|
break
|
||||||
}
|
}
|
||||||
for _, folder := range resp.FileListAO.FolderList {
|
for _, folder := range resp.FileListAO.FolderList {
|
||||||
lastOpTime, _ := time.ParseInLocation("2006-01-02 15:04:05", folder.LastOpTime, loc)
|
lastOpTime := utils.MustParseCNTime(folder.LastOpTime)
|
||||||
res = append(res, &model.Object{
|
res = append(res, &model.Object{
|
||||||
ID: strconv.FormatInt(folder.Id, 10),
|
ID: strconv.FormatInt(folder.Id, 10),
|
||||||
Name: folder.Name,
|
Name: folder.Name,
|
||||||
@ -209,12 +208,13 @@ func (d *Cloud189) getFiles(fileId string) ([]model.Obj, error) {
|
|||||||
})
|
})
|
||||||
}
|
}
|
||||||
for _, file := range resp.FileListAO.FileList {
|
for _, file := range resp.FileListAO.FileList {
|
||||||
lastOpTime, _ := time.ParseInLocation("2006-01-02 15:04:05", file.LastOpTime, loc)
|
lastOpTime := utils.MustParseCNTime(file.LastOpTime)
|
||||||
res = append(res, &model.ObjThumb{
|
res = append(res, &model.ObjThumb{
|
||||||
Object: model.Object{
|
Object: model.Object{
|
||||||
ID: strconv.FormatInt(file.Id, 10),
|
ID: strconv.FormatInt(file.Id, 10),
|
||||||
Name: file.Name,
|
Name: file.Name,
|
||||||
Modified: lastOpTime,
|
Modified: lastOpTime,
|
||||||
|
Size: file.Size,
|
||||||
},
|
},
|
||||||
Thumbnail: model.Thumbnail{Thumbnail: file.Icon.SmallUrl},
|
Thumbnail: model.Thumbnail{Thumbnail: file.Icon.SmallUrl},
|
||||||
})
|
})
|
||||||
|
@ -3,14 +3,11 @@ package _189pc
|
|||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
"net/http"
|
"net/http"
|
||||||
"regexp"
|
|
||||||
"strconv"
|
|
||||||
"strings"
|
"strings"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/alist-org/alist/v3/drivers/base"
|
"github.com/alist-org/alist/v3/drivers/base"
|
||||||
"github.com/alist-org/alist/v3/internal/driver"
|
"github.com/alist-org/alist/v3/internal/driver"
|
||||||
"github.com/alist-org/alist/v3/internal/errs"
|
|
||||||
"github.com/alist-org/alist/v3/internal/model"
|
"github.com/alist-org/alist/v3/internal/model"
|
||||||
"github.com/alist-org/alist/v3/pkg/utils"
|
"github.com/alist-org/alist/v3/pkg/utils"
|
||||||
"github.com/go-resty/resty/v2"
|
"github.com/go-resty/resty/v2"
|
||||||
@ -134,16 +131,17 @@ func (y *Yun189PC) Link(ctx context.Context, file model.Obj, args model.LinkArgs
|
|||||||
"User-Agent": []string{base.UserAgent},
|
"User-Agent": []string{base.UserAgent},
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
|
/*
|
||||||
// 获取链接有效时常
|
// 获取链接有效时常
|
||||||
strs := regexp.MustCompile(`(?i)expire[^=]*=([0-9]*)`).FindStringSubmatch(downloadUrl.URL)
|
strs := regexp.MustCompile(`(?i)expire[^=]*=([0-9]*)`).FindStringSubmatch(downloadUrl.URL)
|
||||||
if len(strs) == 2 {
|
if len(strs) == 2 {
|
||||||
timestamp, err := strconv.ParseInt(strs[1], 10, 64)
|
timestamp, err := strconv.ParseInt(strs[1], 10, 64)
|
||||||
if err == nil {
|
if err == nil {
|
||||||
expired := time.Duration(timestamp-time.Now().Unix()) * time.Second
|
expired := time.Duration(timestamp-time.Now().Unix()) * time.Second
|
||||||
like.Expiration = &expired
|
like.Expiration = &expired
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
*/
|
||||||
return like, nil
|
return like, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -278,7 +276,3 @@ func (y *Yun189PC) Put(ctx context.Context, dstDir model.Obj, stream model.FileS
|
|||||||
}
|
}
|
||||||
return y.CommonUpload(ctx, dstDir, stream, up)
|
return y.CommonUpload(ctx, dstDir, stream, up)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (y *Yun189PC) Other(ctx context.Context, args model.OtherArgs) (interface{}, error) {
|
|
||||||
return nil, errs.NotSupport
|
|
||||||
}
|
|
||||||
|
@ -78,8 +78,7 @@ func timestamp() int64 {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func MustParseTime(str string) *time.Time {
|
func MustParseTime(str string) *time.Time {
|
||||||
loc, _ := time.LoadLocation("Asia/Shanghai")
|
lastOpTime, _ := time.ParseInLocation("2006-01-02 15:04:05 -07", str+" +08", time.Local)
|
||||||
lastOpTime, _ := time.ParseInLocation("2006-01-02 15:04:05", str, loc)
|
|
||||||
return &lastOpTime
|
return &lastOpTime
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -111,7 +110,9 @@ func ParseHttpHeader(str string) map[string]string {
|
|||||||
header := make(map[string]string)
|
header := make(map[string]string)
|
||||||
for _, value := range strings.Split(str, "&") {
|
for _, value := range strings.Split(str, "&") {
|
||||||
i := strings.Index(value, "=")
|
i := strings.Index(value, "=")
|
||||||
header[strings.TrimSpace(value[0:i])] = strings.TrimSpace(value[i+1:])
|
if i > 0 {
|
||||||
|
header[strings.TrimSpace(value[0:i])] = strings.TrimSpace(value[i+1:])
|
||||||
|
}
|
||||||
}
|
}
|
||||||
return header
|
return header
|
||||||
}
|
}
|
||||||
|
@ -6,19 +6,16 @@ import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
type Addition struct {
|
type Addition struct {
|
||||||
Username string `json:"username" required:"true"`
|
Username string `json:"username" required:"true"`
|
||||||
Password string `json:"password" required:"true"`
|
Password string `json:"password" required:"true"`
|
||||||
VCode string `json:"validate_code"`
|
VCode string `json:"validate_code"`
|
||||||
RootFolderID string `json:"root_folder_id"`
|
driver.RootID
|
||||||
OrderBy string `json:"order_by" type:"select" options:"filename,filesize,lastOpTime" default:"filename"`
|
OrderBy string `json:"order_by" type:"select" options:"filename,filesize,lastOpTime" default:"filename"`
|
||||||
OrderDirection string `json:"order_direction" type:"select" options:"asc,desc" default:"asc"`
|
OrderDirection string `json:"order_direction" type:"select" options:"asc,desc" default:"asc"`
|
||||||
Type string `json:"type" type:"select" options:"personal,family" default:"personal"`
|
Type string `json:"type" type:"select" options:"personal,family" default:"personal"`
|
||||||
FamilyID string `json:"family_id"`
|
FamilyID string `json:"family_id"`
|
||||||
RapidUpload bool `json:"rapid_upload"`
|
RapidUpload bool `json:"rapid_upload"`
|
||||||
}
|
NoUseOcr bool `json:"no_use_ocr"`
|
||||||
|
|
||||||
func (a Addition) GetRootId() string {
|
|
||||||
return a.RootFolderID
|
|
||||||
}
|
}
|
||||||
|
|
||||||
var config = driver.Config{
|
var config = driver.Config{
|
||||||
|
@ -140,13 +140,7 @@ func (y *Yun189PC) getFiles(ctx context.Context, fileId string) ([]model.Obj, er
|
|||||||
fullUrl += "/listFiles.action"
|
fullUrl += "/listFiles.action"
|
||||||
|
|
||||||
res := make([]model.Obj, 0, 130)
|
res := make([]model.Obj, 0, 130)
|
||||||
for pageNum := 1; pageNum < 100; pageNum++ {
|
for pageNum := 1; ; pageNum++ {
|
||||||
select {
|
|
||||||
case <-ctx.Done():
|
|
||||||
return nil, ctx.Err()
|
|
||||||
default:
|
|
||||||
}
|
|
||||||
|
|
||||||
var resp Cloud189FilesResp
|
var resp Cloud189FilesResp
|
||||||
_, err := y.get(fullUrl, func(r *resty.Request) {
|
_, err := y.get(fullUrl, func(r *resty.Request) {
|
||||||
r.SetContext(ctx)
|
r.SetContext(ctx)
|
||||||
@ -192,20 +186,19 @@ func (y *Yun189PC) getFiles(ctx context.Context, fileId string) ([]model.Obj, er
|
|||||||
|
|
||||||
func (y *Yun189PC) login() (err error) {
|
func (y *Yun189PC) login() (err error) {
|
||||||
// 初始化登陆所需参数
|
// 初始化登陆所需参数
|
||||||
if y.loginParam == nil {
|
if y.loginParam == nil || !y.NoUseOcr {
|
||||||
if err = y.initLoginParam(); err != nil {
|
if err = y.initLoginParam(); err != nil {
|
||||||
// 验证码也通过错误返回
|
// 验证码也通过错误返回
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
defer func() {
|
defer func() {
|
||||||
// 销毁验证码
|
// 销毁验证码
|
||||||
y.VCode = ""
|
y.VCode = ""
|
||||||
// 销毁登陆参数
|
// 销毁登陆参数
|
||||||
y.loginParam = nil
|
y.loginParam = nil
|
||||||
// 遇到错误,重新加载登陆参数
|
// 遇到错误,重新加载登陆参数
|
||||||
if err != nil {
|
if err != nil && y.NoUseOcr {
|
||||||
if err1 := y.initLoginParam(); err1 != nil {
|
if err1 := y.initLoginParam(); err1 != nil {
|
||||||
err = fmt.Errorf("err1: %s \nerr2: %s", err, err1)
|
err = fmt.Errorf("err1: %s \nerr2: %s", err, err1)
|
||||||
}
|
}
|
||||||
@ -309,44 +302,34 @@ func (y *Yun189PC) initLoginParam() error {
|
|||||||
param.jRsaKey = fmt.Sprintf("-----BEGIN PUBLIC KEY-----\n%s\n-----END PUBLIC KEY-----", encryptConf.Data.PubKey)
|
param.jRsaKey = fmt.Sprintf("-----BEGIN PUBLIC KEY-----\n%s\n-----END PUBLIC KEY-----", encryptConf.Data.PubKey)
|
||||||
param.RsaUsername = encryptConf.Data.Pre + RsaEncrypt(param.jRsaKey, y.Username)
|
param.RsaUsername = encryptConf.Data.Pre + RsaEncrypt(param.jRsaKey, y.Username)
|
||||||
param.RsaPassword = encryptConf.Data.Pre + RsaEncrypt(param.jRsaKey, y.Password)
|
param.RsaPassword = encryptConf.Data.Pre + RsaEncrypt(param.jRsaKey, y.Password)
|
||||||
|
|
||||||
// 判断是否需要验证码
|
|
||||||
res, err = y.client.R().
|
|
||||||
SetFormData(map[string]string{
|
|
||||||
"appKey": APP_ID,
|
|
||||||
"accountType": ACCOUNT_TYPE,
|
|
||||||
"userName": param.RsaUsername,
|
|
||||||
}).
|
|
||||||
Post(AUTH_URL + "/api/logbox/oauth2/needcaptcha.do")
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
y.loginParam = ¶m
|
y.loginParam = ¶m
|
||||||
if res.String() != "0" {
|
|
||||||
imgRes, err := y.client.R().
|
imgRes, err := y.client.R().
|
||||||
SetQueryParams(map[string]string{
|
SetQueryParams(map[string]string{
|
||||||
"token": param.CaptchaToken,
|
"token": param.CaptchaToken,
|
||||||
"REQID": param.ReqId,
|
"REQID": param.ReqId,
|
||||||
"rnd": fmt.Sprint(timestamp()),
|
"rnd": fmt.Sprint(timestamp()),
|
||||||
}).
|
}).
|
||||||
Get(AUTH_URL + "/api/logbox/oauth2/picCaptcha.do")
|
Get(AUTH_URL + "/api/logbox/oauth2/picCaptcha.do")
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return fmt.Errorf("failed to obtain verification code")
|
return fmt.Errorf("failed to obtain verification code")
|
||||||
|
}
|
||||||
|
if imgRes.Size() > 20 {
|
||||||
|
if setting.GetStr(conf.OcrApi) != "" && !y.NoUseOcr {
|
||||||
|
vRes, err := base.RestyClient.R().
|
||||||
|
SetMultipartField("image", "validateCode.png", "image/png", bytes.NewReader(imgRes.Body())).
|
||||||
|
Post(setting.GetStr(conf.OcrApi))
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if jsoniter.Get(vRes.Body(), "status").ToInt() == 200 {
|
||||||
|
y.VCode = jsoniter.Get(vRes.Body(), "result").ToString()
|
||||||
|
return nil
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// 尝试使用ocr
|
// 返回验证码图片给前端
|
||||||
vRes, err := base.RestyClient.R().
|
return fmt.Errorf(`need img validate code: <img src="data:image/png;base64,%s"/>`, base64.StdEncoding.EncodeToString(imgRes.Body()))
|
||||||
SetMultipartField("image", "validateCode.png", "image/png", bytes.NewReader(imgRes.Body())).
|
|
||||||
Post(setting.GetStr(conf.OcrApi))
|
|
||||||
if err == nil && jsoniter.Get(vRes.Body(), "status").ToInt() == 200 {
|
|
||||||
y.VCode = jsoniter.Get(vRes.Body(), "result").ToString()
|
|
||||||
}
|
|
||||||
|
|
||||||
// ocr无法处理,返回验证码图片给前端
|
|
||||||
if len(y.VCode) != 4 {
|
|
||||||
return fmt.Errorf("need validate code: data:image/png;base64,%s", base64.StdEncoding.EncodeToString(res.Body()))
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
@ -402,6 +385,7 @@ func (y *Yun189PC) CommonUpload(ctx context.Context, dstDir model.Obj, file mode
|
|||||||
const DEFAULT int64 = 10485760
|
const DEFAULT int64 = 10485760
|
||||||
var count = int64(math.Ceil(float64(file.GetSize()) / float64(DEFAULT)))
|
var count = int64(math.Ceil(float64(file.GetSize()) / float64(DEFAULT)))
|
||||||
|
|
||||||
|
requestID := uuid.NewString()
|
||||||
params := Params{
|
params := Params{
|
||||||
"parentFolderId": dstDir.GetID(),
|
"parentFolderId": dstDir.GetID(),
|
||||||
"fileName": url.QueryEscape(file.GetName()),
|
"fileName": url.QueryEscape(file.GetName()),
|
||||||
@ -423,6 +407,7 @@ func (y *Yun189PC) CommonUpload(ctx context.Context, dstDir model.Obj, file mode
|
|||||||
var initMultiUpload InitMultiUploadResp
|
var initMultiUpload InitMultiUploadResp
|
||||||
_, err = y.request(fullUrl+"/initMultiUpload", http.MethodGet, func(req *resty.Request) {
|
_, err = y.request(fullUrl+"/initMultiUpload", http.MethodGet, func(req *resty.Request) {
|
||||||
req.SetContext(ctx)
|
req.SetContext(ctx)
|
||||||
|
req.SetHeader("X-Request-ID", requestID)
|
||||||
}, params, &initMultiUpload)
|
}, params, &initMultiUpload)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
@ -457,6 +442,7 @@ func (y *Yun189PC) CommonUpload(ctx context.Context, dstDir model.Obj, file mode
|
|||||||
_, err = y.request(fullUrl+"/getMultiUploadUrls", http.MethodGet,
|
_, err = y.request(fullUrl+"/getMultiUploadUrls", http.MethodGet,
|
||||||
func(req *resty.Request) {
|
func(req *resty.Request) {
|
||||||
req.SetContext(ctx)
|
req.SetContext(ctx)
|
||||||
|
req.SetHeader("X-Request-ID", requestID)
|
||||||
}, Params{
|
}, Params{
|
||||||
"partInfo": fmt.Sprintf("%d-%s", i, silceMd5Base64),
|
"partInfo": fmt.Sprintf("%d-%s", i, silceMd5Base64),
|
||||||
"uploadFileId": initMultiUpload.Data.UploadFileID,
|
"uploadFileId": initMultiUpload.Data.UploadFileID,
|
||||||
@ -492,6 +478,7 @@ func (y *Yun189PC) CommonUpload(ctx context.Context, dstDir model.Obj, file mode
|
|||||||
_, err = y.request(fullUrl+"/commitMultiUploadFile", http.MethodGet,
|
_, err = y.request(fullUrl+"/commitMultiUploadFile", http.MethodGet,
|
||||||
func(req *resty.Request) {
|
func(req *resty.Request) {
|
||||||
req.SetContext(ctx)
|
req.SetContext(ctx)
|
||||||
|
req.SetHeader("X-Request-ID", requestID)
|
||||||
}, Params{
|
}, Params{
|
||||||
"uploadFileId": initMultiUpload.Data.UploadFileID,
|
"uploadFileId": initMultiUpload.Data.UploadFileID,
|
||||||
"fileMd5": fileMd5Hex,
|
"fileMd5": fileMd5Hex,
|
||||||
@ -506,14 +493,14 @@ func (y *Yun189PC) CommonUpload(ctx context.Context, dstDir model.Obj, file mode
|
|||||||
// 快传
|
// 快传
|
||||||
func (y *Yun189PC) FastUpload(ctx context.Context, dstDir model.Obj, file model.FileStreamer, up driver.UpdateProgress) (err error) {
|
func (y *Yun189PC) FastUpload(ctx context.Context, dstDir model.Obj, file model.FileStreamer, up driver.UpdateProgress) (err error) {
|
||||||
// 需要获取完整文件md5,必须支持 io.Seek
|
// 需要获取完整文件md5,必须支持 io.Seek
|
||||||
if _, ok := file.GetReadCloser().(*os.File); !ok {
|
tempFile, err := utils.CreateTempFile(file.GetReadCloser())
|
||||||
r, err := utils.CreateTempFile(file)
|
if err != nil {
|
||||||
if err != nil {
|
return err
|
||||||
return err
|
|
||||||
}
|
|
||||||
file.Close()
|
|
||||||
file.SetReadCloser(r)
|
|
||||||
}
|
}
|
||||||
|
defer func() {
|
||||||
|
_ = tempFile.Close()
|
||||||
|
_ = os.Remove(tempFile.Name())
|
||||||
|
}()
|
||||||
|
|
||||||
const DEFAULT int64 = 10485760
|
const DEFAULT int64 = 10485760
|
||||||
count := int(math.Ceil(float64(file.GetSize()) / float64(DEFAULT)))
|
count := int(math.Ceil(float64(file.GetSize()) / float64(DEFAULT)))
|
||||||
@ -531,14 +518,16 @@ func (y *Yun189PC) FastUpload(ctx context.Context, dstDir model.Obj, file model.
|
|||||||
}
|
}
|
||||||
|
|
||||||
silceMd5.Reset()
|
silceMd5.Reset()
|
||||||
if _, err := io.CopyN(io.MultiWriter(fileMd5, silceMd5), file, DEFAULT); err != nil && err != io.EOF && err != io.ErrUnexpectedEOF {
|
if _, err := io.CopyN(io.MultiWriter(fileMd5, silceMd5), tempFile, DEFAULT); err != nil && err != io.EOF && err != io.ErrUnexpectedEOF {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
md5Byte := silceMd5.Sum(nil)
|
md5Byte := silceMd5.Sum(nil)
|
||||||
silceMd5Hexs = append(silceMd5Hexs, strings.ToUpper(hex.EncodeToString(md5Byte)))
|
silceMd5Hexs = append(silceMd5Hexs, strings.ToUpper(hex.EncodeToString(md5Byte)))
|
||||||
silceMd5Base64s = append(silceMd5Base64s, fmt.Sprint(i, "-", base64.StdEncoding.EncodeToString(md5Byte)))
|
silceMd5Base64s = append(silceMd5Base64s, fmt.Sprint(i, "-", base64.StdEncoding.EncodeToString(md5Byte)))
|
||||||
}
|
}
|
||||||
file.GetReadCloser().(*os.File).Seek(0, io.SeekStart)
|
if _, err = tempFile.Seek(0, io.SeekStart); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
fileMd5Hex := strings.ToUpper(hex.EncodeToString(fileMd5.Sum(nil)))
|
fileMd5Hex := strings.ToUpper(hex.EncodeToString(fileMd5.Sum(nil)))
|
||||||
sliceMd5Hex := fileMd5Hex
|
sliceMd5Hex := fileMd5Hex
|
||||||
@ -546,6 +535,7 @@ func (y *Yun189PC) FastUpload(ctx context.Context, dstDir model.Obj, file model.
|
|||||||
sliceMd5Hex = strings.ToUpper(utils.GetMD5Encode(strings.Join(silceMd5Hexs, "\n")))
|
sliceMd5Hex = strings.ToUpper(utils.GetMD5Encode(strings.Join(silceMd5Hexs, "\n")))
|
||||||
}
|
}
|
||||||
|
|
||||||
|
requestID := uuid.NewString()
|
||||||
// 检测是否支持快传
|
// 检测是否支持快传
|
||||||
params := Params{
|
params := Params{
|
||||||
"parentFolderId": dstDir.GetID(),
|
"parentFolderId": dstDir.GetID(),
|
||||||
@ -568,6 +558,7 @@ func (y *Yun189PC) FastUpload(ctx context.Context, dstDir model.Obj, file model.
|
|||||||
var uploadInfo InitMultiUploadResp
|
var uploadInfo InitMultiUploadResp
|
||||||
_, err = y.request(fullUrl+"/initMultiUpload", http.MethodGet, func(req *resty.Request) {
|
_, err = y.request(fullUrl+"/initMultiUpload", http.MethodGet, func(req *resty.Request) {
|
||||||
req.SetContext(ctx)
|
req.SetContext(ctx)
|
||||||
|
req.SetHeader("X-Request-ID", requestID)
|
||||||
}, params, &uploadInfo)
|
}, params, &uploadInfo)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
@ -579,6 +570,7 @@ func (y *Yun189PC) FastUpload(ctx context.Context, dstDir model.Obj, file model.
|
|||||||
_, err = y.request(fullUrl+"/getMultiUploadUrls", http.MethodGet,
|
_, err = y.request(fullUrl+"/getMultiUploadUrls", http.MethodGet,
|
||||||
func(req *resty.Request) {
|
func(req *resty.Request) {
|
||||||
req.SetContext(ctx)
|
req.SetContext(ctx)
|
||||||
|
req.SetHeader("X-Request-ID", requestID)
|
||||||
}, Params{
|
}, Params{
|
||||||
"uploadFileId": uploadInfo.Data.UploadFileID,
|
"uploadFileId": uploadInfo.Data.UploadFileID,
|
||||||
"partInfo": strings.Join(silceMd5Base64s, ","),
|
"partInfo": strings.Join(silceMd5Base64s, ","),
|
||||||
@ -599,7 +591,7 @@ func (y *Yun189PC) FastUpload(ctx context.Context, dstDir model.Obj, file model.
|
|||||||
SetContext(ctx).
|
SetContext(ctx).
|
||||||
SetQueryParams(clientSuffix()).
|
SetQueryParams(clientSuffix()).
|
||||||
SetHeaders(ParseHttpHeader(uploadData.RequestHeader)).
|
SetHeaders(ParseHttpHeader(uploadData.RequestHeader)).
|
||||||
SetBody(io.LimitReader(file, DEFAULT)).
|
SetBody(io.LimitReader(tempFile, DEFAULT)).
|
||||||
Put(uploadData.RequestURL)
|
Put(uploadData.RequestURL)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
@ -615,6 +607,7 @@ func (y *Yun189PC) FastUpload(ctx context.Context, dstDir model.Obj, file model.
|
|||||||
_, err = y.request(fullUrl+"/commitMultiUploadFile", http.MethodGet,
|
_, err = y.request(fullUrl+"/commitMultiUploadFile", http.MethodGet,
|
||||||
func(req *resty.Request) {
|
func(req *resty.Request) {
|
||||||
req.SetContext(ctx)
|
req.SetContext(ctx)
|
||||||
|
req.SetHeader("X-Request-ID", requestID)
|
||||||
}, Params{
|
}, Params{
|
||||||
"uploadFileId": uploadInfo.Data.UploadFileID,
|
"uploadFileId": uploadInfo.Data.UploadFileID,
|
||||||
"isLog": "0",
|
"isLog": "0",
|
||||||
|
126
drivers/alist_v2/driver.go
Normal file
126
drivers/alist_v2/driver.go
Normal file
@ -0,0 +1,126 @@
|
|||||||
|
package alist_v2
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
|
||||||
|
"github.com/alist-org/alist/v3/drivers/base"
|
||||||
|
"github.com/alist-org/alist/v3/internal/driver"
|
||||||
|
"github.com/alist-org/alist/v3/internal/errs"
|
||||||
|
"github.com/alist-org/alist/v3/internal/model"
|
||||||
|
"github.com/alist-org/alist/v3/pkg/utils"
|
||||||
|
"github.com/alist-org/alist/v3/server/common"
|
||||||
|
)
|
||||||
|
|
||||||
|
type AListV2 struct {
|
||||||
|
model.Storage
|
||||||
|
Addition
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *AListV2) Config() driver.Config {
|
||||||
|
return config
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *AListV2) GetAddition() driver.Additional {
|
||||||
|
return d.Addition
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *AListV2) Init(ctx context.Context, storage model.Storage) error {
|
||||||
|
d.Storage = storage
|
||||||
|
err := utils.Json.UnmarshalFromString(d.Storage.Addition, &d.Addition)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
// TODO login / refresh token
|
||||||
|
//op.MustSaveDriverStorage(d)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *AListV2) Drop(ctx context.Context) error {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *AListV2) List(ctx context.Context, dir model.Obj, args model.ListArgs) ([]model.Obj, error) {
|
||||||
|
url := d.Address + "/api/public/path"
|
||||||
|
var resp common.Resp[PathResp]
|
||||||
|
_, err := base.RestyClient.R().
|
||||||
|
SetResult(&resp).
|
||||||
|
SetHeader("Authorization", d.AccessToken).
|
||||||
|
SetBody(PathReq{
|
||||||
|
PageNum: 0,
|
||||||
|
PageSize: 0,
|
||||||
|
Path: dir.GetPath(),
|
||||||
|
Password: d.Password,
|
||||||
|
}).Post(url)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
var files []model.Obj
|
||||||
|
for _, f := range resp.Data.Files {
|
||||||
|
file := model.ObjThumb{
|
||||||
|
Object: model.Object{
|
||||||
|
Name: f.Name,
|
||||||
|
Modified: *f.UpdatedAt,
|
||||||
|
Size: f.Size,
|
||||||
|
IsFolder: f.Type == 1,
|
||||||
|
},
|
||||||
|
Thumbnail: model.Thumbnail{Thumbnail: f.Thumbnail},
|
||||||
|
}
|
||||||
|
files = append(files, &file)
|
||||||
|
}
|
||||||
|
return files, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
//func (d *AList) Get(ctx context.Context, path string) (model.Obj, error) {
|
||||||
|
// // this is optional
|
||||||
|
// return nil, errs.NotImplement
|
||||||
|
//}
|
||||||
|
|
||||||
|
func (d *AListV2) Link(ctx context.Context, file model.Obj, args model.LinkArgs) (*model.Link, error) {
|
||||||
|
url := d.Address + "/api/public/path"
|
||||||
|
var resp common.Resp[PathResp]
|
||||||
|
_, err := base.RestyClient.R().
|
||||||
|
SetResult(&resp).
|
||||||
|
SetHeader("Authorization", d.AccessToken).
|
||||||
|
SetBody(PathReq{
|
||||||
|
PageNum: 0,
|
||||||
|
PageSize: 0,
|
||||||
|
Path: file.GetPath(),
|
||||||
|
Password: d.Password,
|
||||||
|
}).Post(url)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return &model.Link{
|
||||||
|
URL: resp.Data.Files[0].Url,
|
||||||
|
}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *AListV2) MakeDir(ctx context.Context, parentDir model.Obj, dirName string) error {
|
||||||
|
return errs.NotImplement
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *AListV2) Move(ctx context.Context, srcObj, dstDir model.Obj) error {
|
||||||
|
return errs.NotImplement
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *AListV2) Rename(ctx context.Context, srcObj model.Obj, newName string) error {
|
||||||
|
return errs.NotImplement
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *AListV2) Copy(ctx context.Context, srcObj, dstDir model.Obj) error {
|
||||||
|
return errs.NotImplement
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *AListV2) Remove(ctx context.Context, obj model.Obj) error {
|
||||||
|
return errs.NotImplement
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *AListV2) Put(ctx context.Context, dstDir model.Obj, stream model.FileStreamer, up driver.UpdateProgress) error {
|
||||||
|
return errs.NotImplement
|
||||||
|
}
|
||||||
|
|
||||||
|
//func (d *AList) Other(ctx context.Context, args model.OtherArgs) (interface{}, error) {
|
||||||
|
// return nil, errs.NotSupport
|
||||||
|
//}
|
||||||
|
|
||||||
|
var _ driver.Driver = (*AListV2)(nil)
|
26
drivers/alist_v2/meta.go
Normal file
26
drivers/alist_v2/meta.go
Normal file
@ -0,0 +1,26 @@
|
|||||||
|
package alist_v2
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/alist-org/alist/v3/internal/driver"
|
||||||
|
"github.com/alist-org/alist/v3/internal/op"
|
||||||
|
)
|
||||||
|
|
||||||
|
type Addition struct {
|
||||||
|
driver.RootPath
|
||||||
|
Address string `json:"url" required:"true"`
|
||||||
|
Password string `json:"password"`
|
||||||
|
AccessToken string `json:"access_token"`
|
||||||
|
}
|
||||||
|
|
||||||
|
var config = driver.Config{
|
||||||
|
Name: "AList V2",
|
||||||
|
LocalSort: true,
|
||||||
|
NoUpload: true,
|
||||||
|
DefaultRoot: "/",
|
||||||
|
}
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
op.RegisterDriver(config, func() driver.Driver {
|
||||||
|
return &AListV2{}
|
||||||
|
})
|
||||||
|
}
|
31
drivers/alist_v2/types.go
Normal file
31
drivers/alist_v2/types.go
Normal file
@ -0,0 +1,31 @@
|
|||||||
|
package alist_v2
|
||||||
|
|
||||||
|
import (
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
type File struct {
|
||||||
|
Id string `json:"-"`
|
||||||
|
Name string `json:"name"`
|
||||||
|
Size int64 `json:"size"`
|
||||||
|
Type int `json:"type"`
|
||||||
|
Driver string `json:"driver"`
|
||||||
|
UpdatedAt *time.Time `json:"updated_at"`
|
||||||
|
Thumbnail string `json:"thumbnail"`
|
||||||
|
Url string `json:"url"`
|
||||||
|
SizeStr string `json:"size_str"`
|
||||||
|
TimeStr string `json:"time_str"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type PathResp struct {
|
||||||
|
Type string `json:"type"`
|
||||||
|
//Meta Meta `json:"meta"`
|
||||||
|
Files []File `json:"files"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type PathReq struct {
|
||||||
|
PageNum int `json:"page_num"`
|
||||||
|
PageSize int `json:"page_size"`
|
||||||
|
Password string `json:"password"`
|
||||||
|
Path string `json:"path"`
|
||||||
|
}
|
1
drivers/alist_v2/util.go
Normal file
1
drivers/alist_v2/util.go
Normal file
@ -0,0 +1 @@
|
|||||||
|
package alist_v2
|
128
drivers/alist_v3/driver.go
Normal file
128
drivers/alist_v3/driver.go
Normal file
@ -0,0 +1,128 @@
|
|||||||
|
package alist_v3
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
|
||||||
|
"github.com/alist-org/alist/v3/drivers/base"
|
||||||
|
"github.com/alist-org/alist/v3/internal/driver"
|
||||||
|
"github.com/alist-org/alist/v3/internal/errs"
|
||||||
|
"github.com/alist-org/alist/v3/internal/model"
|
||||||
|
"github.com/alist-org/alist/v3/pkg/utils"
|
||||||
|
"github.com/alist-org/alist/v3/server/common"
|
||||||
|
"github.com/alist-org/alist/v3/server/handles"
|
||||||
|
)
|
||||||
|
|
||||||
|
type AListV3 struct {
|
||||||
|
model.Storage
|
||||||
|
Addition
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *AListV3) Config() driver.Config {
|
||||||
|
return config
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *AListV3) GetAddition() driver.Additional {
|
||||||
|
return d.Addition
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *AListV3) Init(ctx context.Context, storage model.Storage) error {
|
||||||
|
d.Storage = storage
|
||||||
|
err := utils.Json.UnmarshalFromString(d.Storage.Addition, &d.Addition)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
// TODO login / refresh token
|
||||||
|
//op.MustSaveDriverStorage(d)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *AListV3) Drop(ctx context.Context) error {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *AListV3) List(ctx context.Context, dir model.Obj, args model.ListArgs) ([]model.Obj, error) {
|
||||||
|
url := d.Address + "/api/fs/list"
|
||||||
|
var resp common.Resp[handles.FsListResp]
|
||||||
|
_, err := base.RestyClient.R().
|
||||||
|
SetResult(&resp).
|
||||||
|
SetHeader("Authorization", d.AccessToken).
|
||||||
|
SetBody(handles.ListReq{
|
||||||
|
PageReq: common.PageReq{
|
||||||
|
Page: 1,
|
||||||
|
PerPage: 0,
|
||||||
|
},
|
||||||
|
Path: dir.GetPath(),
|
||||||
|
Password: d.Password,
|
||||||
|
Refresh: false,
|
||||||
|
}).Post(url)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
var files []model.Obj
|
||||||
|
for _, f := range resp.Data.Content {
|
||||||
|
file := model.ObjThumb{
|
||||||
|
Object: model.Object{
|
||||||
|
Name: f.Name,
|
||||||
|
Modified: f.Modified,
|
||||||
|
Size: f.Size,
|
||||||
|
IsFolder: f.IsDir,
|
||||||
|
},
|
||||||
|
Thumbnail: model.Thumbnail{Thumbnail: f.Thumb},
|
||||||
|
}
|
||||||
|
files = append(files, &file)
|
||||||
|
}
|
||||||
|
return files, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
//func (d *AList) Get(ctx context.Context, path string) (model.Obj, error) {
|
||||||
|
// // this is optional
|
||||||
|
// return nil, errs.NotImplement
|
||||||
|
//}
|
||||||
|
|
||||||
|
func (d *AListV3) Link(ctx context.Context, file model.Obj, args model.LinkArgs) (*model.Link, error) {
|
||||||
|
url := d.Address + "/api/fs/get"
|
||||||
|
var resp common.Resp[handles.FsGetResp]
|
||||||
|
_, err := base.RestyClient.R().
|
||||||
|
SetResult(&resp).
|
||||||
|
SetHeader("Authorization", d.AccessToken).
|
||||||
|
SetBody(handles.FsGetReq{
|
||||||
|
Path: file.GetPath(),
|
||||||
|
Password: d.Password,
|
||||||
|
}).Post(url)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return &model.Link{
|
||||||
|
URL: resp.Data.RawURL,
|
||||||
|
}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *AListV3) MakeDir(ctx context.Context, parentDir model.Obj, dirName string) error {
|
||||||
|
return errs.NotImplement
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *AListV3) Move(ctx context.Context, srcObj, dstDir model.Obj) error {
|
||||||
|
return errs.NotImplement
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *AListV3) Rename(ctx context.Context, srcObj model.Obj, newName string) error {
|
||||||
|
return errs.NotImplement
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *AListV3) Copy(ctx context.Context, srcObj, dstDir model.Obj) error {
|
||||||
|
return errs.NotImplement
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *AListV3) Remove(ctx context.Context, obj model.Obj) error {
|
||||||
|
return errs.NotImplement
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *AListV3) Put(ctx context.Context, dstDir model.Obj, stream model.FileStreamer, up driver.UpdateProgress) error {
|
||||||
|
return errs.NotImplement
|
||||||
|
}
|
||||||
|
|
||||||
|
//func (d *AList) Other(ctx context.Context, args model.OtherArgs) (interface{}, error) {
|
||||||
|
// return nil, errs.NotSupport
|
||||||
|
//}
|
||||||
|
|
||||||
|
var _ driver.Driver = (*AListV3)(nil)
|
26
drivers/alist_v3/meta.go
Normal file
26
drivers/alist_v3/meta.go
Normal file
@ -0,0 +1,26 @@
|
|||||||
|
package alist_v3
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/alist-org/alist/v3/internal/driver"
|
||||||
|
"github.com/alist-org/alist/v3/internal/op"
|
||||||
|
)
|
||||||
|
|
||||||
|
type Addition struct {
|
||||||
|
driver.RootPath
|
||||||
|
Address string `json:"url" required:"true"`
|
||||||
|
Password string `json:"password"`
|
||||||
|
AccessToken string `json:"access_token"`
|
||||||
|
}
|
||||||
|
|
||||||
|
var config = driver.Config{
|
||||||
|
Name: "AList V3",
|
||||||
|
LocalSort: true,
|
||||||
|
NoUpload: true,
|
||||||
|
DefaultRoot: "/",
|
||||||
|
}
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
op.RegisterDriver(config, func() driver.Driver {
|
||||||
|
return &AListV3{}
|
||||||
|
})
|
||||||
|
}
|
1
drivers/alist_v3/types.go
Normal file
1
drivers/alist_v3/types.go
Normal file
@ -0,0 +1 @@
|
|||||||
|
package alist_v3
|
1
drivers/alist_v3/util.go
Normal file
1
drivers/alist_v3/util.go
Normal file
@ -0,0 +1 @@
|
|||||||
|
package alist_v3
|
@ -234,7 +234,10 @@ func (d *AliDrive) Put(ctx context.Context, dstDir model.Obj, stream model.FileS
|
|||||||
buf := make([]byte, 8)
|
buf := make([]byte, 8)
|
||||||
r, _ := new(big.Int).SetString(utils.GetMD5Encode(d.AccessToken)[:16], 16)
|
r, _ := new(big.Int).SetString(utils.GetMD5Encode(d.AccessToken)[:16], 16)
|
||||||
i := new(big.Int).SetInt64(file.GetSize())
|
i := new(big.Int).SetInt64(file.GetSize())
|
||||||
o := r.Mod(r, i)
|
o := new(big.Int).SetInt64(0)
|
||||||
|
if file.GetSize() > 0 {
|
||||||
|
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])
|
||||||
|
|
||||||
@ -264,7 +267,9 @@ func (d *AliDrive) Put(ctx context.Context, dstDir model.Obj, stream model.FileS
|
|||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
res.Body.Close()
|
res.Body.Close()
|
||||||
up(i * 100 / count)
|
if count > 0 {
|
||||||
|
up(i * 100 / count)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
var resp2 base.Json
|
var resp2 base.Json
|
||||||
_, err, e = d.request("https://api.aliyundrive.com/v2/file/complete", http.MethodPost, func(req *resty.Request) {
|
_, err, e = d.request("https://api.aliyundrive.com/v2/file/complete", http.MethodPost, func(req *resty.Request) {
|
||||||
@ -284,7 +289,29 @@ func (d *AliDrive) Put(ctx context.Context, dstDir model.Obj, stream model.FileS
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (d *AliDrive) Other(ctx context.Context, args model.OtherArgs) (interface{}, error) {
|
func (d *AliDrive) Other(ctx context.Context, args model.OtherArgs) (interface{}, error) {
|
||||||
return nil, errs.NotSupport
|
var resp base.Json
|
||||||
|
var url string
|
||||||
|
data := base.Json{
|
||||||
|
"drive_id": d.DriveId,
|
||||||
|
"file_id": args.Obj.GetID(),
|
||||||
|
}
|
||||||
|
switch args.Method {
|
||||||
|
case "doc_preview":
|
||||||
|
url = "https://api.aliyundrive.com/v2/file/get_office_preview_url"
|
||||||
|
data["access_token"] = d.AccessToken
|
||||||
|
case "video_preview":
|
||||||
|
url = "https://api.aliyundrive.com/v2/file/get_video_preview_play_info"
|
||||||
|
data["category"] = "live_transcoding"
|
||||||
|
default:
|
||||||
|
return nil, errs.NotSupport
|
||||||
|
}
|
||||||
|
_, err, _ := d.request(url, http.MethodPost, func(req *resty.Request) {
|
||||||
|
req.SetBody(data)
|
||||||
|
}, &resp)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return resp, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
var _ driver.Driver = (*AliDrive)(nil)
|
var _ driver.Driver = (*AliDrive)(nil)
|
||||||
|
@ -40,6 +40,7 @@ func fileToObj(f File) *model.ObjThumb {
|
|||||||
Modified: f.UpdatedAt,
|
Modified: f.UpdatedAt,
|
||||||
IsFolder: f.Type == "folder",
|
IsFolder: f.Type == "folder",
|
||||||
},
|
},
|
||||||
|
Thumbnail: model.Thumbnail{Thumbnail: f.Thumbnail},
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
178
drivers/aliyundrive_share/driver.go
Normal file
178
drivers/aliyundrive_share/driver.go
Normal file
@ -0,0 +1,178 @@
|
|||||||
|
package aliyundrive_share
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"errors"
|
||||||
|
"net/http"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/alist-org/alist/v3/drivers/base"
|
||||||
|
"github.com/alist-org/alist/v3/internal/driver"
|
||||||
|
"github.com/alist-org/alist/v3/internal/errs"
|
||||||
|
"github.com/alist-org/alist/v3/internal/model"
|
||||||
|
"github.com/alist-org/alist/v3/pkg/cron"
|
||||||
|
"github.com/alist-org/alist/v3/pkg/utils"
|
||||||
|
log "github.com/sirupsen/logrus"
|
||||||
|
)
|
||||||
|
|
||||||
|
type AliyundriveShare struct {
|
||||||
|
model.Storage
|
||||||
|
Addition
|
||||||
|
AccessToken string
|
||||||
|
ShareToken string
|
||||||
|
DriveId string
|
||||||
|
cron *cron.Cron
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *AliyundriveShare) Config() driver.Config {
|
||||||
|
return config
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *AliyundriveShare) GetAddition() driver.Additional {
|
||||||
|
return d.Addition
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *AliyundriveShare) Init(ctx context.Context, storage model.Storage) error {
|
||||||
|
d.Storage = storage
|
||||||
|
err := utils.Json.UnmarshalFromString(d.Storage.Addition, &d.Addition)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
err = d.refreshToken()
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
err = d.getShareToken()
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
d.cron = cron.NewCron(time.Hour * 2)
|
||||||
|
d.cron.Do(func() {
|
||||||
|
err := d.refreshToken()
|
||||||
|
if err != nil {
|
||||||
|
log.Errorf("%+v", err)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *AliyundriveShare) Drop(ctx context.Context) error {
|
||||||
|
if d.cron != nil {
|
||||||
|
d.cron.Stop()
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *AliyundriveShare) List(ctx context.Context, dir model.Obj, args model.ListArgs) ([]model.Obj, error) {
|
||||||
|
files, err := d.getFiles(dir.GetID())
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return utils.SliceConvert(files, func(src File) (model.Obj, error) {
|
||||||
|
return fileToObj(src), nil
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
//func (d *AliyundriveShare) Get(ctx context.Context, path string) (model.Obj, error) {
|
||||||
|
// // this is optional
|
||||||
|
// return nil, errs.NotImplement
|
||||||
|
//}
|
||||||
|
|
||||||
|
func (d *AliyundriveShare) Link(ctx context.Context, file model.Obj, args model.LinkArgs) (*model.Link, error) {
|
||||||
|
data := base.Json{
|
||||||
|
"drive_id": d.DriveId,
|
||||||
|
"file_id": file.GetID(),
|
||||||
|
"expire_sec": 14400,
|
||||||
|
}
|
||||||
|
var e ErrorResp
|
||||||
|
res, err := base.RestyClient.R().
|
||||||
|
SetError(&e).SetBody(data).
|
||||||
|
SetHeader("content-type", "application/json").
|
||||||
|
SetHeader("Authorization", "Bearer\t"+d.AccessToken).
|
||||||
|
Post("https://api.aliyundrive.com/v2/file/get_download_url")
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
var u string
|
||||||
|
if e.Code != "" {
|
||||||
|
if e.Code == "AccessTokenInvalid" {
|
||||||
|
err = d.refreshToken()
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return d.Link(ctx, file, args)
|
||||||
|
} else if e.Code == "ForbiddenNoPermission.File" {
|
||||||
|
data = utils.MergeMap(data, base.Json{
|
||||||
|
// Only ten minutes valid
|
||||||
|
"expire_sec": 600,
|
||||||
|
"share_id": d.ShareId,
|
||||||
|
})
|
||||||
|
var resp ShareLinkResp
|
||||||
|
var e2 ErrorResp
|
||||||
|
_, err = base.RestyClient.R().
|
||||||
|
SetError(&e2).SetBody(data).SetResult(&resp).
|
||||||
|
SetHeader("content-type", "application/json").
|
||||||
|
SetHeader("Authorization", "Bearer\t"+d.AccessToken).
|
||||||
|
SetHeader("x-share-token", d.ShareToken).
|
||||||
|
Post("https://api.aliyundrive.com/v2/file/get_share_link_download_url")
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if e2.Code != "" {
|
||||||
|
if e2.Code == "AccessTokenInvalid" || e2.Code == "ShareLinkTokenInvalid" {
|
||||||
|
err = d.getShareToken()
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return d.Link(ctx, file, args)
|
||||||
|
} else {
|
||||||
|
return nil, errors.New(e2.Code + ":" + e2.Message)
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
u = resp.DownloadUrl
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
return nil, errors.New(e.Code + ":" + e.Message)
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
u = utils.Json.Get(res.Body(), "url").ToString()
|
||||||
|
}
|
||||||
|
return &model.Link{
|
||||||
|
Header: http.Header{
|
||||||
|
"Referer": []string{"https://www.aliyundrive.com/"},
|
||||||
|
},
|
||||||
|
URL: u,
|
||||||
|
}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *AliyundriveShare) MakeDir(ctx context.Context, parentDir model.Obj, dirName string) error {
|
||||||
|
// TODO create folder
|
||||||
|
return errs.NotSupport
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *AliyundriveShare) Move(ctx context.Context, srcObj, dstDir model.Obj) error {
|
||||||
|
// TODO move obj
|
||||||
|
return errs.NotSupport
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *AliyundriveShare) Rename(ctx context.Context, srcObj model.Obj, newName string) error {
|
||||||
|
// TODO rename obj
|
||||||
|
return errs.NotSupport
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *AliyundriveShare) Copy(ctx context.Context, srcObj, dstDir model.Obj) error {
|
||||||
|
// TODO copy obj
|
||||||
|
return errs.NotSupport
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *AliyundriveShare) Remove(ctx context.Context, obj model.Obj) error {
|
||||||
|
// TODO remove obj
|
||||||
|
return errs.NotSupport
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *AliyundriveShare) Put(ctx context.Context, dstDir model.Obj, stream model.FileStreamer, up driver.UpdateProgress) error {
|
||||||
|
// TODO upload file
|
||||||
|
return errs.NotSupport
|
||||||
|
}
|
||||||
|
|
||||||
|
var _ driver.Driver = (*AliyundriveShare)(nil)
|
29
drivers/aliyundrive_share/meta.go
Normal file
29
drivers/aliyundrive_share/meta.go
Normal file
@ -0,0 +1,29 @@
|
|||||||
|
package aliyundrive_share
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/alist-org/alist/v3/internal/driver"
|
||||||
|
"github.com/alist-org/alist/v3/internal/op"
|
||||||
|
)
|
||||||
|
|
||||||
|
type Addition struct {
|
||||||
|
RefreshToken string `json:"refresh_token" required:"true"`
|
||||||
|
ShareId string `json:"share_id" required:"true"`
|
||||||
|
SharePwd string `json:"share_pwd"`
|
||||||
|
driver.RootID
|
||||||
|
OrderBy string `json:"order_by" type:"select" options:"name,size,updated_at,created_at"`
|
||||||
|
OrderDirection string `json:"order_direction" type:"select" options:"ASC,DESC"`
|
||||||
|
}
|
||||||
|
|
||||||
|
var config = driver.Config{
|
||||||
|
Name: "AliyundriveShare",
|
||||||
|
LocalSort: false,
|
||||||
|
OnlyProxy: false,
|
||||||
|
NoUpload: true,
|
||||||
|
DefaultRoot: "root",
|
||||||
|
}
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
op.RegisterDriver(config, func() driver.Driver {
|
||||||
|
return &AliyundriveShare{}
|
||||||
|
})
|
||||||
|
}
|
57
drivers/aliyundrive_share/types.go
Normal file
57
drivers/aliyundrive_share/types.go
Normal file
@ -0,0 +1,57 @@
|
|||||||
|
package aliyundrive_share
|
||||||
|
|
||||||
|
import (
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/alist-org/alist/v3/internal/model"
|
||||||
|
)
|
||||||
|
|
||||||
|
type ErrorResp struct {
|
||||||
|
Code string `json:"code"`
|
||||||
|
Message string `json:"message"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type ShareTokenResp struct {
|
||||||
|
ShareToken string `json:"share_token"`
|
||||||
|
ExpireTime time.Time `json:"expire_time"`
|
||||||
|
ExpiresIn int `json:"expires_in"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type ListResp struct {
|
||||||
|
Items []File `json:"items"`
|
||||||
|
NextMarker string `json:"next_marker"`
|
||||||
|
PunishedFileCount int `json:"punished_file_count"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type File struct {
|
||||||
|
DriveId string `json:"drive_id"`
|
||||||
|
DomainId string `json:"domain_id"`
|
||||||
|
FileId string `json:"file_id"`
|
||||||
|
ShareId string `json:"share_id"`
|
||||||
|
Name string `json:"name"`
|
||||||
|
Type string `json:"type"`
|
||||||
|
CreatedAt time.Time `json:"created_at"`
|
||||||
|
UpdatedAt time.Time `json:"updated_at"`
|
||||||
|
ParentFileId string `json:"parent_file_id"`
|
||||||
|
Size int64 `json:"size"`
|
||||||
|
Thumbnail string `json:"thumbnail"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func fileToObj(f File) *model.ObjThumb {
|
||||||
|
return &model.ObjThumb{
|
||||||
|
Object: model.Object{
|
||||||
|
ID: f.FileId,
|
||||||
|
Name: f.Name,
|
||||||
|
Size: f.Size,
|
||||||
|
Modified: f.UpdatedAt,
|
||||||
|
IsFolder: f.Type == "folder",
|
||||||
|
},
|
||||||
|
Thumbnail: model.Thumbnail{Thumbnail: f.Thumbnail},
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
type ShareLinkResp struct {
|
||||||
|
DownloadUrl string `json:"download_url"`
|
||||||
|
Url string `json:"url"`
|
||||||
|
Thumbnail string `json:"thumbnail"`
|
||||||
|
}
|
99
drivers/aliyundrive_share/util.go
Normal file
99
drivers/aliyundrive_share/util.go
Normal file
@ -0,0 +1,99 @@
|
|||||||
|
package aliyundrive_share
|
||||||
|
|
||||||
|
import (
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
|
||||||
|
"github.com/alist-org/alist/v3/drivers/base"
|
||||||
|
"github.com/alist-org/alist/v3/internal/op"
|
||||||
|
log "github.com/sirupsen/logrus"
|
||||||
|
)
|
||||||
|
|
||||||
|
func (d *AliyundriveShare) refreshToken() error {
|
||||||
|
url := "https://auth.aliyundrive.com/v2/account/token"
|
||||||
|
var resp base.TokenResp
|
||||||
|
var e ErrorResp
|
||||||
|
_, err := base.RestyClient.R().
|
||||||
|
SetBody(base.Json{"refresh_token": d.RefreshToken, "grant_type": "refresh_token"}).
|
||||||
|
SetResult(&resp).
|
||||||
|
SetError(&e).
|
||||||
|
Post(url)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if e.Code != "" {
|
||||||
|
return fmt.Errorf("failed to refresh token: %s", e.Message)
|
||||||
|
}
|
||||||
|
d.RefreshToken, d.AccessToken = resp.RefreshToken, resp.AccessToken
|
||||||
|
op.MustSaveDriverStorage(d)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// do others that not defined in Driver interface
|
||||||
|
func (d *AliyundriveShare) getShareToken() error {
|
||||||
|
data := base.Json{
|
||||||
|
"share_id": d.ShareId,
|
||||||
|
}
|
||||||
|
if d.SharePwd != "" {
|
||||||
|
data["share_pwd"] = d.SharePwd
|
||||||
|
}
|
||||||
|
var e ErrorResp
|
||||||
|
var resp ShareTokenResp
|
||||||
|
_, err := base.RestyClient.R().
|
||||||
|
SetResult(&resp).SetError(&e).SetBody(data).
|
||||||
|
Post("https://api.aliyundrive.com/v2/share_link/get_share_token")
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if e.Code != "" {
|
||||||
|
return errors.New(e.Message)
|
||||||
|
}
|
||||||
|
d.ShareToken = resp.ShareToken
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *AliyundriveShare) getFiles(fileId string) ([]File, error) {
|
||||||
|
files := make([]File, 0)
|
||||||
|
data := base.Json{
|
||||||
|
"image_thumbnail_process": "image/resize,w_160/format,jpeg",
|
||||||
|
"image_url_process": "image/resize,w_1920/format,jpeg",
|
||||||
|
"limit": 100,
|
||||||
|
"order_by": d.OrderBy,
|
||||||
|
"order_direction": d.OrderDirection,
|
||||||
|
"parent_file_id": fileId,
|
||||||
|
"share_id": d.ShareId,
|
||||||
|
"video_thumbnail_process": "video/snapshot,t_1000,f_jpg,ar_auto,w_300",
|
||||||
|
"marker": "first",
|
||||||
|
}
|
||||||
|
for data["marker"] != "" {
|
||||||
|
if data["marker"] == "first" {
|
||||||
|
data["marker"] = ""
|
||||||
|
}
|
||||||
|
var e ErrorResp
|
||||||
|
var resp ListResp
|
||||||
|
res, err := base.RestyClient.R().
|
||||||
|
SetHeader("x-share-token", d.ShareToken).
|
||||||
|
SetResult(&resp).SetError(&e).SetBody(data).
|
||||||
|
Post("https://api.aliyundrive.com/adrive/v3/file/list")
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
log.Debugf("aliyundrive share get files: %s", res.String())
|
||||||
|
if e.Code != "" {
|
||||||
|
if e.Code == "AccessTokenInvalid" || e.Code == "ShareLinkTokenInvalid" {
|
||||||
|
err = d.getShareToken()
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return d.getFiles(fileId)
|
||||||
|
}
|
||||||
|
return nil, errors.New(e.Message)
|
||||||
|
}
|
||||||
|
data["marker"] = resp.NextMarker
|
||||||
|
files = append(files, resp.Items...)
|
||||||
|
}
|
||||||
|
if len(files) > 0 && d.DriveId == "" {
|
||||||
|
d.DriveId = files[0].DriveId
|
||||||
|
}
|
||||||
|
return files, nil
|
||||||
|
}
|
@ -1,22 +1,32 @@
|
|||||||
package drivers
|
package drivers
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
_ "github.com/alist-org/alist/v3/drivers/115"
|
||||||
_ "github.com/alist-org/alist/v3/drivers/123"
|
_ "github.com/alist-org/alist/v3/drivers/123"
|
||||||
_ "github.com/alist-org/alist/v3/drivers/139"
|
_ "github.com/alist-org/alist/v3/drivers/139"
|
||||||
_ "github.com/alist-org/alist/v3/drivers/189"
|
_ "github.com/alist-org/alist/v3/drivers/189"
|
||||||
_ "github.com/alist-org/alist/v3/drivers/189pc"
|
_ "github.com/alist-org/alist/v3/drivers/189pc"
|
||||||
|
_ "github.com/alist-org/alist/v3/drivers/alist_v2"
|
||||||
|
_ "github.com/alist-org/alist/v3/drivers/alist_v3"
|
||||||
_ "github.com/alist-org/alist/v3/drivers/aliyundrive"
|
_ "github.com/alist-org/alist/v3/drivers/aliyundrive"
|
||||||
|
_ "github.com/alist-org/alist/v3/drivers/aliyundrive_share"
|
||||||
_ "github.com/alist-org/alist/v3/drivers/baidu_netdisk"
|
_ "github.com/alist-org/alist/v3/drivers/baidu_netdisk"
|
||||||
|
_ "github.com/alist-org/alist/v3/drivers/baidu_photo"
|
||||||
_ "github.com/alist-org/alist/v3/drivers/ftp"
|
_ "github.com/alist-org/alist/v3/drivers/ftp"
|
||||||
_ "github.com/alist-org/alist/v3/drivers/google_drive"
|
_ "github.com/alist-org/alist/v3/drivers/google_drive"
|
||||||
|
_ "github.com/alist-org/alist/v3/drivers/google_photo"
|
||||||
|
_ "github.com/alist-org/alist/v3/drivers/lanzou"
|
||||||
_ "github.com/alist-org/alist/v3/drivers/local"
|
_ "github.com/alist-org/alist/v3/drivers/local"
|
||||||
_ "github.com/alist-org/alist/v3/drivers/mediatrack"
|
_ "github.com/alist-org/alist/v3/drivers/mediatrack"
|
||||||
|
_ "github.com/alist-org/alist/v3/drivers/mega"
|
||||||
_ "github.com/alist-org/alist/v3/drivers/onedrive"
|
_ "github.com/alist-org/alist/v3/drivers/onedrive"
|
||||||
_ "github.com/alist-org/alist/v3/drivers/pikpak"
|
_ "github.com/alist-org/alist/v3/drivers/pikpak"
|
||||||
_ "github.com/alist-org/alist/v3/drivers/quark"
|
_ "github.com/alist-org/alist/v3/drivers/quark"
|
||||||
_ "github.com/alist-org/alist/v3/drivers/s3"
|
_ "github.com/alist-org/alist/v3/drivers/s3"
|
||||||
_ "github.com/alist-org/alist/v3/drivers/sftp"
|
_ "github.com/alist-org/alist/v3/drivers/sftp"
|
||||||
|
_ "github.com/alist-org/alist/v3/drivers/smb"
|
||||||
_ "github.com/alist-org/alist/v3/drivers/teambition"
|
_ "github.com/alist-org/alist/v3/drivers/teambition"
|
||||||
|
_ "github.com/alist-org/alist/v3/drivers/thunder"
|
||||||
_ "github.com/alist-org/alist/v3/drivers/uss"
|
_ "github.com/alist-org/alist/v3/drivers/uss"
|
||||||
_ "github.com/alist-org/alist/v3/drivers/virtual"
|
_ "github.com/alist-org/alist/v3/drivers/virtual"
|
||||||
_ "github.com/alist-org/alist/v3/drivers/webdav"
|
_ "github.com/alist-org/alist/v3/drivers/webdav"
|
||||||
|
@ -14,9 +14,7 @@ import (
|
|||||||
"strings"
|
"strings"
|
||||||
|
|
||||||
"github.com/alist-org/alist/v3/drivers/base"
|
"github.com/alist-org/alist/v3/drivers/base"
|
||||||
"github.com/alist-org/alist/v3/internal/conf"
|
|
||||||
"github.com/alist-org/alist/v3/internal/driver"
|
"github.com/alist-org/alist/v3/internal/driver"
|
||||||
"github.com/alist-org/alist/v3/internal/errs"
|
|
||||||
"github.com/alist-org/alist/v3/internal/model"
|
"github.com/alist-org/alist/v3/internal/model"
|
||||||
"github.com/alist-org/alist/v3/pkg/utils"
|
"github.com/alist-org/alist/v3/pkg/utils"
|
||||||
log "github.com/sirupsen/logrus"
|
log "github.com/sirupsen/logrus"
|
||||||
@ -119,28 +117,14 @@ func (d *BaiduNetdisk) Remove(ctx context.Context, obj model.Obj) error {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (d *BaiduNetdisk) Put(ctx context.Context, dstDir model.Obj, stream model.FileStreamer, up driver.UpdateProgress) error {
|
func (d *BaiduNetdisk) Put(ctx context.Context, dstDir model.Obj, stream model.FileStreamer, up driver.UpdateProgress) error {
|
||||||
var tempFile *os.File
|
tempFile, err := utils.CreateTempFile(stream.GetReadCloser())
|
||||||
var err error
|
if err != nil {
|
||||||
if f, ok := stream.GetReadCloser().(*os.File); ok {
|
return err
|
||||||
tempFile = f
|
|
||||||
} else {
|
|
||||||
tempFile, err = os.CreateTemp(conf.Conf.TempDir, "file-*")
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
defer func() {
|
|
||||||
_ = tempFile.Close()
|
|
||||||
_ = os.Remove(tempFile.Name())
|
|
||||||
}()
|
|
||||||
_, err = io.Copy(tempFile, stream)
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
_, err = tempFile.Seek(0, io.SeekStart)
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
defer func() {
|
||||||
|
_ = tempFile.Close()
|
||||||
|
_ = os.Remove(tempFile.Name())
|
||||||
|
}()
|
||||||
var Default int64 = 4 * 1024 * 1024
|
var Default int64 = 4 * 1024 * 1024
|
||||||
defaultByteData := make([]byte, Default)
|
defaultByteData := make([]byte, Default)
|
||||||
count := int(math.Ceil(float64(stream.GetSize()) / float64(Default)))
|
count := int(math.Ceil(float64(stream.GetSize()) / float64(Default)))
|
||||||
@ -217,7 +201,7 @@ func (d *BaiduNetdisk) Put(ctx context.Context, dstDir model.Obj, stream model.F
|
|||||||
"uploadid": precreateResp.Uploadid,
|
"uploadid": precreateResp.Uploadid,
|
||||||
}
|
}
|
||||||
left = stream.GetSize()
|
left = stream.GetSize()
|
||||||
for _, partseq := range precreateResp.BlockList {
|
for i, partseq := range precreateResp.BlockList {
|
||||||
byteSize := Default
|
byteSize := Default
|
||||||
var byteData []byte
|
var byteData []byte
|
||||||
if left < Default {
|
if left < Default {
|
||||||
@ -238,13 +222,12 @@ func (d *BaiduNetdisk) Put(ctx context.Context, dstDir model.Obj, stream model.F
|
|||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
log.Debugln(res.String())
|
log.Debugln(res.String())
|
||||||
|
if len(precreateResp.BlockList) > 0 {
|
||||||
|
up(i * 100 / len(precreateResp.BlockList))
|
||||||
|
}
|
||||||
}
|
}
|
||||||
_, err = d.create(path, stream.GetSize(), 0, precreateResp.Uploadid, block_list_str)
|
_, err = d.create(path, stream.GetSize(), 0, precreateResp.Uploadid, block_list_str)
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
func (d *BaiduNetdisk) Other(ctx context.Context, args model.OtherArgs) (interface{}, error) {
|
|
||||||
return nil, errs.NotSupport
|
|
||||||
}
|
|
||||||
|
|
||||||
var _ driver.Driver = (*BaiduNetdisk)(nil)
|
var _ driver.Driver = (*BaiduNetdisk)(nil)
|
||||||
|
282
drivers/baidu_photo/driver.go
Normal file
282
drivers/baidu_photo/driver.go
Normal file
@ -0,0 +1,282 @@
|
|||||||
|
package baiduphoto
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"crypto/md5"
|
||||||
|
"encoding/hex"
|
||||||
|
"fmt"
|
||||||
|
"io"
|
||||||
|
"math"
|
||||||
|
"os"
|
||||||
|
"regexp"
|
||||||
|
|
||||||
|
"github.com/alist-org/alist/v3/internal/driver"
|
||||||
|
"github.com/alist-org/alist/v3/internal/errs"
|
||||||
|
"github.com/alist-org/alist/v3/internal/model"
|
||||||
|
"github.com/alist-org/alist/v3/pkg/utils"
|
||||||
|
"github.com/go-resty/resty/v2"
|
||||||
|
)
|
||||||
|
|
||||||
|
type BaiduPhoto struct {
|
||||||
|
model.Storage
|
||||||
|
Addition
|
||||||
|
|
||||||
|
AccessToken string
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *BaiduPhoto) Config() driver.Config {
|
||||||
|
return config
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *BaiduPhoto) GetAddition() driver.Additional {
|
||||||
|
return d.Addition
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *BaiduPhoto) Init(ctx context.Context, storage model.Storage) error {
|
||||||
|
d.Storage = storage
|
||||||
|
err := utils.Json.UnmarshalFromString(d.Storage.Addition, &d.Addition)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
return d.refreshToken()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *BaiduPhoto) Drop(ctx context.Context) error {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *BaiduPhoto) List(ctx context.Context, dir model.Obj, args model.ListArgs) ([]model.Obj, error) {
|
||||||
|
var objs []model.Obj
|
||||||
|
var err error
|
||||||
|
if IsRoot(dir) {
|
||||||
|
var albums []Album
|
||||||
|
if d.ShowType != "root_only_file" {
|
||||||
|
albums, err = d.GetAllAlbum(ctx)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
var files []File
|
||||||
|
if d.ShowType != "root_only_album" {
|
||||||
|
files, err = d.GetAllFile(ctx)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
alubmName := make(map[string]int)
|
||||||
|
objs, _ = utils.SliceConvert(albums, func(album Album) (model.Obj, error) {
|
||||||
|
i := alubmName[album.GetName()]
|
||||||
|
if i != 0 {
|
||||||
|
alubmName[album.GetName()]++
|
||||||
|
album.Title = fmt.Sprintf("%s(%d)", album.Title, i)
|
||||||
|
}
|
||||||
|
alubmName[album.GetName()]++
|
||||||
|
return &album, nil
|
||||||
|
})
|
||||||
|
for i := 0; i < len(files); i++ {
|
||||||
|
objs = append(objs, &files[i])
|
||||||
|
}
|
||||||
|
} else if IsAlbum(dir) || IsAlbumRoot(dir) {
|
||||||
|
var files []AlbumFile
|
||||||
|
files, err = d.GetAllAlbumFile(ctx, splitID(dir.GetID())[0], "")
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
objs = make([]model.Obj, 0, len(files))
|
||||||
|
for i := 0; i < len(files); i++ {
|
||||||
|
objs = append(objs, &files[i])
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return objs, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *BaiduPhoto) Link(ctx context.Context, file model.Obj, args model.LinkArgs) (*model.Link, error) {
|
||||||
|
if IsAlbumFile(file) {
|
||||||
|
return d.linkAlbum(ctx, file, args)
|
||||||
|
} else if IsFile(file) {
|
||||||
|
return d.linkFile(ctx, file, args)
|
||||||
|
}
|
||||||
|
return nil, errs.NotFile
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *BaiduPhoto) MakeDir(ctx context.Context, parentDir model.Obj, dirName string) error {
|
||||||
|
if IsRoot(parentDir) {
|
||||||
|
code := regexp.MustCompile(`(?i)join:([\S]*)`).FindStringSubmatch(dirName)
|
||||||
|
if len(code) > 1 {
|
||||||
|
return d.JoinAlbum(ctx, code[1])
|
||||||
|
}
|
||||||
|
return d.CreateAlbum(ctx, dirName)
|
||||||
|
}
|
||||||
|
return errs.NotSupport
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *BaiduPhoto) Copy(ctx context.Context, srcObj, dstDir model.Obj) error {
|
||||||
|
if IsFile(srcObj) {
|
||||||
|
if IsAlbum(dstDir) {
|
||||||
|
//rootfile -> album
|
||||||
|
e := splitID(dstDir.GetID())
|
||||||
|
return d.AddAlbumFile(ctx, e[0], e[1], srcObj.GetID())
|
||||||
|
}
|
||||||
|
} else if IsAlbumFile(srcObj) {
|
||||||
|
if IsRoot(dstDir) {
|
||||||
|
//albumfile -> root
|
||||||
|
e := splitID(srcObj.GetID())
|
||||||
|
_, err := d.CopyAlbumFile(ctx, e[1], e[2], e[3], srcObj.GetID())
|
||||||
|
return err
|
||||||
|
} else if IsAlbum(dstDir) {
|
||||||
|
// albumfile -> root -> album
|
||||||
|
e := splitID(srcObj.GetID())
|
||||||
|
file, err := d.CopyAlbumFile(ctx, e[1], e[2], e[3], srcObj.GetID())
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
e = splitID(dstDir.GetID())
|
||||||
|
return d.AddAlbumFile(ctx, e[0], e[1], fmt.Sprint(file.Fsid))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return errs.NotSupport
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *BaiduPhoto) Move(ctx context.Context, srcObj, dstDir model.Obj) error {
|
||||||
|
// 仅支持相册之间移动
|
||||||
|
if IsAlbumFile(srcObj) && IsAlbum(dstDir) {
|
||||||
|
err := d.Copy(ctx, srcObj, dstDir)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
e := splitID(srcObj.GetID())
|
||||||
|
return d.DeleteAlbumFile(ctx, e[1], e[2], srcObj.GetID())
|
||||||
|
}
|
||||||
|
return errs.NotSupport
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *BaiduPhoto) Rename(ctx context.Context, srcObj model.Obj, newName string) error {
|
||||||
|
// 仅支持相册改名
|
||||||
|
if IsAlbum(srcObj) {
|
||||||
|
e := splitID(srcObj.GetID())
|
||||||
|
return d.SetAlbumName(ctx, e[0], e[1], newName)
|
||||||
|
}
|
||||||
|
return errs.NotSupport
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *BaiduPhoto) Remove(ctx context.Context, obj model.Obj) error {
|
||||||
|
e := splitID(obj.GetID())
|
||||||
|
if IsFile(obj) {
|
||||||
|
return d.DeleteFile(ctx, e[0])
|
||||||
|
} else if IsAlbum(obj) {
|
||||||
|
return d.DeleteAlbum(ctx, e[0], e[1])
|
||||||
|
} else if IsAlbumFile(obj) {
|
||||||
|
return d.DeleteAlbumFile(ctx, e[1], e[2], obj.GetID())
|
||||||
|
}
|
||||||
|
return errs.NotSupport
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *BaiduPhoto) Put(ctx context.Context, dstDir model.Obj, stream model.FileStreamer, up driver.UpdateProgress) error {
|
||||||
|
// 需要获取完整文件md5,必须支持 io.Seek
|
||||||
|
tempFile, err := utils.CreateTempFile(stream.GetReadCloser())
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
defer func() {
|
||||||
|
_ = tempFile.Close()
|
||||||
|
_ = os.Remove(tempFile.Name())
|
||||||
|
}()
|
||||||
|
|
||||||
|
// 计算需要的数据
|
||||||
|
const DEFAULT = 1 << 22
|
||||||
|
const SliceSize = 1 << 18
|
||||||
|
count := int(math.Ceil(float64(stream.GetSize()) / float64(DEFAULT)))
|
||||||
|
|
||||||
|
sliceMD5List := make([]string, 0, count)
|
||||||
|
fileMd5 := md5.New()
|
||||||
|
sliceMd5 := md5.New()
|
||||||
|
sliceMd52 := md5.New()
|
||||||
|
slicemd52Write := utils.LimitWriter(sliceMd52, SliceSize)
|
||||||
|
for i := 1; i <= count; i++ {
|
||||||
|
select {
|
||||||
|
case <-ctx.Done():
|
||||||
|
return ctx.Err()
|
||||||
|
default:
|
||||||
|
}
|
||||||
|
_, err := io.CopyN(io.MultiWriter(fileMd5, sliceMd5, slicemd52Write), tempFile, DEFAULT)
|
||||||
|
if err != nil && err != io.EOF && err != io.ErrUnexpectedEOF {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
sliceMD5List = append(sliceMD5List, hex.EncodeToString(sliceMd5.Sum(nil)))
|
||||||
|
sliceMd5.Reset()
|
||||||
|
}
|
||||||
|
if _, err = tempFile.Seek(0, io.SeekStart); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
content_md5 := hex.EncodeToString(fileMd5.Sum(nil))
|
||||||
|
slice_md5 := hex.EncodeToString(sliceMd52.Sum(nil))
|
||||||
|
|
||||||
|
// 开始执行上传
|
||||||
|
params := map[string]string{
|
||||||
|
"autoinit": "1",
|
||||||
|
"isdir": "0",
|
||||||
|
"rtype": "1",
|
||||||
|
"ctype": "11",
|
||||||
|
"path": stream.GetName(),
|
||||||
|
"size": fmt.Sprint(stream.GetSize()),
|
||||||
|
"slice-md5": slice_md5,
|
||||||
|
"content-md5": content_md5,
|
||||||
|
"block_list": MustString(utils.Json.MarshalToString(sliceMD5List)),
|
||||||
|
}
|
||||||
|
|
||||||
|
// 预上传
|
||||||
|
var precreateResp PrecreateResp
|
||||||
|
_, err = d.Post(FILE_API_URL_V1+"/precreate", func(r *resty.Request) {
|
||||||
|
r.SetContext(ctx)
|
||||||
|
r.SetFormData(params)
|
||||||
|
}, &precreateResp)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
switch precreateResp.ReturnType {
|
||||||
|
case 1: // 上传文件
|
||||||
|
uploadParams := map[string]string{
|
||||||
|
"method": "upload",
|
||||||
|
"path": params["path"],
|
||||||
|
"uploadid": precreateResp.UploadID,
|
||||||
|
}
|
||||||
|
|
||||||
|
for i := 0; i < count; i++ {
|
||||||
|
uploadParams["partseq"] = fmt.Sprint(i)
|
||||||
|
_, err = d.Post("https://c3.pcs.baidu.com/rest/2.0/pcs/superfile2", func(r *resty.Request) {
|
||||||
|
r.SetContext(ctx)
|
||||||
|
r.SetQueryParams(uploadParams)
|
||||||
|
r.SetFileReader("file", stream.GetName(), io.LimitReader(tempFile, DEFAULT))
|
||||||
|
}, nil)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
up(i * 100 / count)
|
||||||
|
}
|
||||||
|
fallthrough
|
||||||
|
case 2: // 创建文件
|
||||||
|
params["uploadid"] = precreateResp.UploadID
|
||||||
|
_, err = d.Post(FILE_API_URL_V1+"/create", func(r *resty.Request) {
|
||||||
|
r.SetContext(ctx)
|
||||||
|
r.SetFormData(params)
|
||||||
|
}, &precreateResp)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
fallthrough
|
||||||
|
case 3: // 增加到相册
|
||||||
|
if IsAlbum(dstDir) || IsAlbumRoot(dstDir) {
|
||||||
|
e := splitID(dstDir.GetID())
|
||||||
|
err = d.AddAlbumFile(ctx, e[0], e[1], fmt.Sprint(precreateResp.Data.FsID))
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
var _ driver.Driver = (*BaiduPhoto)(nil)
|
107
drivers/baidu_photo/help.go
Normal file
107
drivers/baidu_photo/help.go
Normal file
@ -0,0 +1,107 @@
|
|||||||
|
package baiduphoto
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"math"
|
||||||
|
"math/rand"
|
||||||
|
"regexp"
|
||||||
|
"strings"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/alist-org/alist/v3/internal/model"
|
||||||
|
)
|
||||||
|
|
||||||
|
//Tid生成
|
||||||
|
func getTid() string {
|
||||||
|
return fmt.Sprintf("3%d%.0f", time.Now().Unix(), math.Floor(9000000*rand.Float64()+1000000))
|
||||||
|
}
|
||||||
|
|
||||||
|
// 检查名称
|
||||||
|
func checkName(name string) bool {
|
||||||
|
return len(name) <= 20 && regexp.MustCompile("[\u4e00-\u9fa5A-Za-z0-9_-]").MatchString(name)
|
||||||
|
}
|
||||||
|
|
||||||
|
func toTime(t int64) *time.Time {
|
||||||
|
tm := time.Unix(t, 0)
|
||||||
|
return &tm
|
||||||
|
}
|
||||||
|
|
||||||
|
func fsidsFormat(ids ...string) string {
|
||||||
|
var buf []string
|
||||||
|
for _, id := range ids {
|
||||||
|
e := splitID(id)
|
||||||
|
buf = append(buf, fmt.Sprintf(`{"fsid":%s,"uk":%s}`, e[0], e[3]))
|
||||||
|
}
|
||||||
|
return fmt.Sprintf("[%s]", strings.Join(buf, ","))
|
||||||
|
}
|
||||||
|
|
||||||
|
func fsidsFormatNotUk(ids ...string) string {
|
||||||
|
var buf []string
|
||||||
|
for _, id := range ids {
|
||||||
|
buf = append(buf, fmt.Sprintf(`{"fsid":%s}`, splitID(id)[0]))
|
||||||
|
}
|
||||||
|
return fmt.Sprintf("[%s]", strings.Join(buf, ","))
|
||||||
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
结构
|
||||||
|
|
||||||
|
{fsid} 文件
|
||||||
|
|
||||||
|
{album_id}|{tid} 相册
|
||||||
|
|
||||||
|
{fsid}|{album_id}|{tid}|{uk} 相册文件
|
||||||
|
*/
|
||||||
|
func splitID(id string) []string {
|
||||||
|
return strings.SplitN(id, "|", 4)[:4]
|
||||||
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
结构
|
||||||
|
|
||||||
|
{fsid} 文件
|
||||||
|
|
||||||
|
{album_id}|{tid} 相册
|
||||||
|
|
||||||
|
{fsid}|{album_id}|{tid}|{uk} 相册文件
|
||||||
|
*/
|
||||||
|
func joinID(ids ...interface{}) string {
|
||||||
|
idsStr := make([]string, 0, len(ids))
|
||||||
|
for _, id := range ids {
|
||||||
|
idsStr = append(idsStr, fmt.Sprint(id))
|
||||||
|
}
|
||||||
|
return strings.Join(idsStr, "|")
|
||||||
|
}
|
||||||
|
|
||||||
|
func getFileName(path string) string {
|
||||||
|
return path[strings.LastIndex(path, "/")+1:]
|
||||||
|
}
|
||||||
|
|
||||||
|
// 相册
|
||||||
|
func IsAlbum(obj model.Obj) bool {
|
||||||
|
return obj.IsDir() && obj.GetPath() == "album"
|
||||||
|
}
|
||||||
|
|
||||||
|
// 根目录
|
||||||
|
func IsRoot(obj model.Obj) bool {
|
||||||
|
return obj.IsDir() && obj.GetPath() == "" && obj.GetID() == ""
|
||||||
|
}
|
||||||
|
|
||||||
|
// 以相册为根目录
|
||||||
|
func IsAlbumRoot(obj model.Obj) bool {
|
||||||
|
return obj.IsDir() && obj.GetPath() == "" && obj.GetID() != ""
|
||||||
|
}
|
||||||
|
|
||||||
|
// 根文件
|
||||||
|
func IsFile(obj model.Obj) bool {
|
||||||
|
return !obj.IsDir() && obj.GetPath() == "file"
|
||||||
|
}
|
||||||
|
|
||||||
|
// 相册文件
|
||||||
|
func IsAlbumFile(obj model.Obj) bool {
|
||||||
|
return !obj.IsDir() && obj.GetPath() == "albumfile"
|
||||||
|
}
|
||||||
|
|
||||||
|
func MustString(str string, err error) string {
|
||||||
|
return str
|
||||||
|
}
|
30
drivers/baidu_photo/meta.go
Normal file
30
drivers/baidu_photo/meta.go
Normal file
@ -0,0 +1,30 @@
|
|||||||
|
package baiduphoto
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/alist-org/alist/v3/internal/driver"
|
||||||
|
"github.com/alist-org/alist/v3/internal/op"
|
||||||
|
)
|
||||||
|
|
||||||
|
type Addition struct {
|
||||||
|
RefreshToken string `json:"refresh_token" required:"true"`
|
||||||
|
ShowType string `json:"show_type" type:"select" options:"root,root_only_album,root_only_file" default:"root"`
|
||||||
|
AlbumID string `json:"album_id"`
|
||||||
|
//AlbumPassword string `json:"album_password"`
|
||||||
|
ClientID string `json:"client_id" required:"true" default:"iYCeC9g08h5vuP9UqvPHKKSVrKFXGa1v"`
|
||||||
|
ClientSecret string `json:"client_secret" required:"true" default:"jXiFMOPVPCWlO2M5CwWQzffpNPaGTRBG"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a Addition) GetRootId() string {
|
||||||
|
return a.AlbumID
|
||||||
|
}
|
||||||
|
|
||||||
|
var config = driver.Config{
|
||||||
|
Name: "BaiduPhoto",
|
||||||
|
LocalSort: true,
|
||||||
|
}
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
op.RegisterDriver(config, func() driver.Driver {
|
||||||
|
return &BaiduPhoto{}
|
||||||
|
})
|
||||||
|
}
|
169
drivers/baidu_photo/types.go
Normal file
169
drivers/baidu_photo/types.go
Normal file
@ -0,0 +1,169 @@
|
|||||||
|
package baiduphoto
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
type TokenErrResp struct {
|
||||||
|
ErrorDescription string `json:"error_description"`
|
||||||
|
ErrorMsg string `json:"error"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (e *TokenErrResp) Error() string {
|
||||||
|
return fmt.Sprint(e.ErrorMsg, " : ", e.ErrorDescription)
|
||||||
|
}
|
||||||
|
|
||||||
|
type Erron struct {
|
||||||
|
Errno int `json:"errno"`
|
||||||
|
RequestID int `json:"request_id"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type Page struct {
|
||||||
|
HasMore int `json:"has_more"`
|
||||||
|
Cursor string `json:"cursor"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p Page) HasNextPage() bool {
|
||||||
|
return p.HasMore == 1
|
||||||
|
}
|
||||||
|
|
||||||
|
type (
|
||||||
|
FileListResp struct {
|
||||||
|
Page
|
||||||
|
List []File `json:"list"`
|
||||||
|
}
|
||||||
|
|
||||||
|
File struct {
|
||||||
|
Fsid int64 `json:"fsid"` // 文件ID
|
||||||
|
Path string `json:"path"` // 文件路径
|
||||||
|
Size int64 `json:"size"`
|
||||||
|
Ctime int64 `json:"ctime"` // 创建时间 s
|
||||||
|
Mtime int64 `json:"mtime"` // 修改时间 s
|
||||||
|
Thumburl []string `json:"thumburl"`
|
||||||
|
|
||||||
|
parseTime *time.Time
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
func (c *File) GetSize() int64 { return c.Size }
|
||||||
|
func (c *File) GetName() string { return getFileName(c.Path) }
|
||||||
|
func (c *File) ModTime() time.Time {
|
||||||
|
if c.parseTime == nil {
|
||||||
|
c.parseTime = toTime(c.Mtime)
|
||||||
|
}
|
||||||
|
return *c.parseTime
|
||||||
|
}
|
||||||
|
func (c *File) IsDir() bool { return false }
|
||||||
|
func (c *File) GetID() string { return joinID(c.Fsid) }
|
||||||
|
func (c *File) GetPath() string { return "file" }
|
||||||
|
func (c *File) Thumb() string {
|
||||||
|
if len(c.Thumburl) > 0 {
|
||||||
|
return c.Thumburl[0]
|
||||||
|
}
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
|
||||||
|
/*相册部分*/
|
||||||
|
type (
|
||||||
|
AlbumListResp struct {
|
||||||
|
Page
|
||||||
|
List []Album `json:"list"`
|
||||||
|
Reset int64 `json:"reset"`
|
||||||
|
TotalCount int64 `json:"total_count"`
|
||||||
|
}
|
||||||
|
|
||||||
|
Album struct {
|
||||||
|
AlbumID string `json:"album_id"`
|
||||||
|
Tid int64 `json:"tid"`
|
||||||
|
Title string `json:"title"`
|
||||||
|
JoinTime int64 `json:"join_time"`
|
||||||
|
CreateTime int64 `json:"create_time"`
|
||||||
|
Mtime int64 `json:"mtime"`
|
||||||
|
|
||||||
|
parseTime *time.Time
|
||||||
|
}
|
||||||
|
|
||||||
|
AlbumFileListResp struct {
|
||||||
|
Page
|
||||||
|
List []AlbumFile `json:"list"`
|
||||||
|
Reset int64 `json:"reset"`
|
||||||
|
TotalCount int64 `json:"total_count"`
|
||||||
|
}
|
||||||
|
|
||||||
|
AlbumFile struct {
|
||||||
|
File
|
||||||
|
AlbumID string `json:"album_id"`
|
||||||
|
Tid int64 `json:"tid"`
|
||||||
|
Uk int64 `json:"uk"`
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
func (a *Album) GetSize() int64 { return 0 }
|
||||||
|
func (a *Album) GetName() string { return fmt.Sprint(a.Title) }
|
||||||
|
func (a *Album) ModTime() time.Time {
|
||||||
|
if a.parseTime == nil {
|
||||||
|
a.parseTime = toTime(a.Mtime)
|
||||||
|
}
|
||||||
|
return *a.parseTime
|
||||||
|
}
|
||||||
|
func (a *Album) IsDir() bool { return true }
|
||||||
|
func (a *Album) GetID() string { return joinID(a.AlbumID, a.Tid) }
|
||||||
|
func (a *Album) GetPath() string { return "album" }
|
||||||
|
|
||||||
|
func (af *AlbumFile) GetID() string { return joinID(af.Fsid, af.AlbumID, af.Tid, af.Uk) }
|
||||||
|
func (c *AlbumFile) GetPath() string { return "albumfile" }
|
||||||
|
|
||||||
|
type (
|
||||||
|
CopyFileResp struct {
|
||||||
|
List []CopyFile `json:"list"`
|
||||||
|
}
|
||||||
|
CopyFile struct {
|
||||||
|
FromFsid int64 `json:"from_fsid"` // 源ID
|
||||||
|
Fsid int64 `json:"fsid"` // 目标ID
|
||||||
|
Path string `json:"path"`
|
||||||
|
ShootTime int `json:"shoot_time"`
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
/*上传部分*/
|
||||||
|
type (
|
||||||
|
UploadFile struct {
|
||||||
|
FsID int64 `json:"fs_id"`
|
||||||
|
Size int64 `json:"size"`
|
||||||
|
Md5 string `json:"md5"`
|
||||||
|
ServerFilename string `json:"server_filename"`
|
||||||
|
Path string `json:"path"`
|
||||||
|
Ctime int `json:"ctime"`
|
||||||
|
Mtime int `json:"mtime"`
|
||||||
|
Isdir int `json:"isdir"`
|
||||||
|
Category int `json:"category"`
|
||||||
|
ServerMd5 string `json:"server_md5"`
|
||||||
|
ShootTime int `json:"shoot_time"`
|
||||||
|
}
|
||||||
|
|
||||||
|
CreateFileResp struct {
|
||||||
|
Data UploadFile `json:"data"`
|
||||||
|
}
|
||||||
|
|
||||||
|
PrecreateResp struct {
|
||||||
|
ReturnType int `json:"return_type"` //存在返回2 不存在返回1 已经保存3
|
||||||
|
//存在返回
|
||||||
|
CreateFileResp
|
||||||
|
|
||||||
|
//不存在返回
|
||||||
|
Path string `json:"path"`
|
||||||
|
UploadID string `json:"uploadid"`
|
||||||
|
Blocklist []int64 `json:"block_list"`
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
type InviteResp struct {
|
||||||
|
Pdata struct {
|
||||||
|
// 邀请码
|
||||||
|
InviteCode string `json:"invite_code"`
|
||||||
|
// 有效时间
|
||||||
|
ExpireTime int `json:"expire_time"`
|
||||||
|
ShareID string `json:"share_id"`
|
||||||
|
} `json:"pdata"`
|
||||||
|
}
|
378
drivers/baidu_photo/utils.go
Normal file
378
drivers/baidu_photo/utils.go
Normal file
@ -0,0 +1,378 @@
|
|||||||
|
package baiduphoto
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
"net/http"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
"github.com/alist-org/alist/v3/drivers/base"
|
||||||
|
"github.com/alist-org/alist/v3/internal/errs"
|
||||||
|
"github.com/alist-org/alist/v3/internal/model"
|
||||||
|
"github.com/alist-org/alist/v3/internal/op"
|
||||||
|
"github.com/alist-org/alist/v3/pkg/utils"
|
||||||
|
"github.com/go-resty/resty/v2"
|
||||||
|
)
|
||||||
|
|
||||||
|
const (
|
||||||
|
API_URL = "https://photo.baidu.com/youai"
|
||||||
|
ALBUM_API_URL = API_URL + "/album/v1"
|
||||||
|
FILE_API_URL_V1 = API_URL + "/file/v1"
|
||||||
|
FILE_API_URL_V2 = API_URL + "/file/v2"
|
||||||
|
)
|
||||||
|
|
||||||
|
var (
|
||||||
|
ErrNotSupportName = errors.New("only chinese and english, numbers and underscores are supported, and the length is no more than 20")
|
||||||
|
)
|
||||||
|
|
||||||
|
func (p *BaiduPhoto) Request(furl string, method string, callback base.ReqCallback, resp interface{}) ([]byte, error) {
|
||||||
|
req := base.RestyClient.R().
|
||||||
|
SetQueryParam("access_token", p.AccessToken)
|
||||||
|
if callback != nil {
|
||||||
|
callback(req)
|
||||||
|
}
|
||||||
|
if resp != nil {
|
||||||
|
req.SetResult(resp)
|
||||||
|
}
|
||||||
|
res, err := req.Execute(method, furl)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
erron := utils.Json.Get(res.Body(), "errno").ToInt()
|
||||||
|
switch erron {
|
||||||
|
case 0:
|
||||||
|
break
|
||||||
|
case 50805:
|
||||||
|
return nil, fmt.Errorf("you have joined album")
|
||||||
|
case 50820:
|
||||||
|
return nil, fmt.Errorf("no shared albums found")
|
||||||
|
case -6:
|
||||||
|
if err = p.refreshToken(); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
default:
|
||||||
|
return nil, fmt.Errorf("errno: %d, refer to https://photo.baidu.com/union/doc", erron)
|
||||||
|
}
|
||||||
|
return res.Body(), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *BaiduPhoto) refreshToken() error {
|
||||||
|
u := "https://openapi.baidu.com/oauth/2.0/token"
|
||||||
|
var resp base.TokenResp
|
||||||
|
var e TokenErrResp
|
||||||
|
_, err := base.RestyClient.R().SetResult(&resp).SetError(&e).SetQueryParams(map[string]string{
|
||||||
|
"grant_type": "refresh_token",
|
||||||
|
"refresh_token": p.RefreshToken,
|
||||||
|
"client_id": p.ClientID,
|
||||||
|
"client_secret": p.ClientSecret,
|
||||||
|
}).Get(u)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if e.ErrorMsg != "" {
|
||||||
|
return &e
|
||||||
|
}
|
||||||
|
if resp.RefreshToken == "" {
|
||||||
|
return errs.EmptyToken
|
||||||
|
}
|
||||||
|
p.AccessToken, p.RefreshToken = resp.AccessToken, resp.RefreshToken
|
||||||
|
op.MustSaveDriverStorage(p)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *BaiduPhoto) Get(furl string, callback base.ReqCallback, resp interface{}) ([]byte, error) {
|
||||||
|
return p.Request(furl, http.MethodGet, callback, resp)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *BaiduPhoto) Post(furl string, callback base.ReqCallback, resp interface{}) ([]byte, error) {
|
||||||
|
return p.Request(furl, http.MethodPost, callback, resp)
|
||||||
|
}
|
||||||
|
|
||||||
|
// 获取所有文件
|
||||||
|
func (p *BaiduPhoto) GetAllFile(ctx context.Context) (files []File, err error) {
|
||||||
|
var cursor string
|
||||||
|
for {
|
||||||
|
var resp FileListResp
|
||||||
|
_, err = p.Get(FILE_API_URL_V1+"/list", func(r *resty.Request) {
|
||||||
|
r.SetContext(ctx)
|
||||||
|
r.SetQueryParams(map[string]string{
|
||||||
|
"need_thumbnail": "1",
|
||||||
|
"need_filter_hidden": "0",
|
||||||
|
"cursor": cursor,
|
||||||
|
})
|
||||||
|
}, &resp)
|
||||||
|
if err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
files = append(files, resp.List...)
|
||||||
|
if !resp.HasNextPage() {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
cursor = resp.Cursor
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 删除根文件
|
||||||
|
func (p *BaiduPhoto) DeleteFile(ctx context.Context, fileIDs ...string) error {
|
||||||
|
_, err := p.Get(FILE_API_URL_V1+"/delete", func(req *resty.Request) {
|
||||||
|
req.SetContext(ctx)
|
||||||
|
req.SetQueryParams(map[string]string{
|
||||||
|
"fsid_list": fmt.Sprintf("[%s]", strings.Join(fileIDs, ",")),
|
||||||
|
})
|
||||||
|
}, nil)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
// 获取所有相册
|
||||||
|
func (p *BaiduPhoto) GetAllAlbum(ctx context.Context) (albums []Album, err error) {
|
||||||
|
var cursor string
|
||||||
|
for {
|
||||||
|
var resp AlbumListResp
|
||||||
|
_, err = p.Get(ALBUM_API_URL+"/list", func(r *resty.Request) {
|
||||||
|
r.SetContext(ctx)
|
||||||
|
r.SetQueryParams(map[string]string{
|
||||||
|
"need_amount": "1",
|
||||||
|
"limit": "100",
|
||||||
|
"cursor": cursor,
|
||||||
|
})
|
||||||
|
}, &resp)
|
||||||
|
if err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if albums == nil {
|
||||||
|
albums = make([]Album, 0, resp.TotalCount)
|
||||||
|
}
|
||||||
|
|
||||||
|
cursor = resp.Cursor
|
||||||
|
albums = append(albums, resp.List...)
|
||||||
|
|
||||||
|
if !resp.HasNextPage() {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 获取相册中所有文件
|
||||||
|
func (p *BaiduPhoto) GetAllAlbumFile(ctx context.Context, albumID, passwd string) (files []AlbumFile, err error) {
|
||||||
|
var cursor string
|
||||||
|
for {
|
||||||
|
var resp AlbumFileListResp
|
||||||
|
_, err = p.Get(ALBUM_API_URL+"/listfile", func(r *resty.Request) {
|
||||||
|
r.SetContext(ctx)
|
||||||
|
r.SetQueryParams(map[string]string{
|
||||||
|
"album_id": albumID,
|
||||||
|
"need_amount": "1",
|
||||||
|
"limit": "1000",
|
||||||
|
"passwd": passwd,
|
||||||
|
"cursor": cursor,
|
||||||
|
})
|
||||||
|
}, &resp)
|
||||||
|
if err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if files == nil {
|
||||||
|
files = make([]AlbumFile, 0, resp.TotalCount)
|
||||||
|
}
|
||||||
|
|
||||||
|
cursor = resp.Cursor
|
||||||
|
files = append(files, resp.List...)
|
||||||
|
|
||||||
|
if !resp.HasNextPage() {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 创建相册
|
||||||
|
func (p *BaiduPhoto) CreateAlbum(ctx context.Context, name string) error {
|
||||||
|
if !checkName(name) {
|
||||||
|
return ErrNotSupportName
|
||||||
|
}
|
||||||
|
_, err := p.Post(ALBUM_API_URL+"/create", func(r *resty.Request) {
|
||||||
|
r.SetContext(ctx)
|
||||||
|
r.SetQueryParams(map[string]string{
|
||||||
|
"title": name,
|
||||||
|
"tid": getTid(),
|
||||||
|
"source": "0",
|
||||||
|
})
|
||||||
|
}, nil)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
// 相册改名
|
||||||
|
func (p *BaiduPhoto) SetAlbumName(ctx context.Context, albumID, tID, name string) error {
|
||||||
|
if !checkName(name) {
|
||||||
|
return ErrNotSupportName
|
||||||
|
}
|
||||||
|
|
||||||
|
_, err := p.Post(ALBUM_API_URL+"/settitle", func(r *resty.Request) {
|
||||||
|
r.SetContext(ctx)
|
||||||
|
r.SetFormData(map[string]string{
|
||||||
|
"title": name,
|
||||||
|
"album_id": albumID,
|
||||||
|
"tid": tID,
|
||||||
|
})
|
||||||
|
}, nil)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
// 删除相册
|
||||||
|
func (p *BaiduPhoto) DeleteAlbum(ctx context.Context, albumID, tID string) error {
|
||||||
|
_, err := p.Post(ALBUM_API_URL+"/delete", func(r *resty.Request) {
|
||||||
|
r.SetContext(ctx)
|
||||||
|
r.SetFormData(map[string]string{
|
||||||
|
"album_id": albumID,
|
||||||
|
"tid": tID,
|
||||||
|
"delete_origin_image": "0", // 是否删除原图 0 不删除 1 删除
|
||||||
|
})
|
||||||
|
}, nil)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
// 删除相册文件
|
||||||
|
func (p *BaiduPhoto) DeleteAlbumFile(ctx context.Context, albumID, tID string, fileIDs ...string) error {
|
||||||
|
_, err := p.Post(ALBUM_API_URL+"/delfile", func(r *resty.Request) {
|
||||||
|
r.SetContext(ctx)
|
||||||
|
r.SetFormData(map[string]string{
|
||||||
|
"album_id": albumID,
|
||||||
|
"tid": tID,
|
||||||
|
"list": fsidsFormat(fileIDs...),
|
||||||
|
"del_origin": "0", // 是否删除原图 0 不删除 1 删除
|
||||||
|
})
|
||||||
|
}, nil)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
// 增加相册文件
|
||||||
|
func (p *BaiduPhoto) AddAlbumFile(ctx context.Context, albumID, tID string, fileIDs ...string) error {
|
||||||
|
_, err := p.Get(ALBUM_API_URL+"/addfile", func(r *resty.Request) {
|
||||||
|
r.SetContext(ctx)
|
||||||
|
r.SetQueryParams(map[string]string{
|
||||||
|
"album_id": albumID,
|
||||||
|
"tid": tID,
|
||||||
|
"list": fsidsFormatNotUk(fileIDs...),
|
||||||
|
})
|
||||||
|
}, nil)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
// 保存相册文件为根文件
|
||||||
|
func (p *BaiduPhoto) CopyAlbumFile(ctx context.Context, albumID, tID, uk string, fileID ...string) (*CopyFile, error) {
|
||||||
|
var resp CopyFileResp
|
||||||
|
_, err := p.Post(ALBUM_API_URL+"/copyfile", func(r *resty.Request) {
|
||||||
|
r.SetContext(ctx)
|
||||||
|
r.SetFormData(map[string]string{
|
||||||
|
"album_id": albumID,
|
||||||
|
"tid": tID,
|
||||||
|
"uk": uk,
|
||||||
|
"list": fsidsFormatNotUk(fileID...),
|
||||||
|
})
|
||||||
|
r.SetResult(&resp)
|
||||||
|
}, nil)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return &resp.List[0], nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// 加入相册
|
||||||
|
func (p *BaiduPhoto) JoinAlbum(ctx context.Context, code string) error {
|
||||||
|
var resp InviteResp
|
||||||
|
_, err := p.Get(ALBUM_API_URL+"/querypcode", func(req *resty.Request) {
|
||||||
|
req.SetContext(ctx)
|
||||||
|
req.SetQueryParams(map[string]string{
|
||||||
|
"pcode": code,
|
||||||
|
"web": "1",
|
||||||
|
})
|
||||||
|
}, &resp)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
_, err = p.Get(ALBUM_API_URL+"/join", func(req *resty.Request) {
|
||||||
|
req.SetContext(ctx)
|
||||||
|
req.SetQueryParams(map[string]string{
|
||||||
|
"invite_code": resp.Pdata.InviteCode,
|
||||||
|
})
|
||||||
|
}, nil)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *BaiduPhoto) linkAlbum(ctx context.Context, file model.Obj, args model.LinkArgs) (*model.Link, error) {
|
||||||
|
headers := map[string]string{
|
||||||
|
"User-Agent": base.UserAgent,
|
||||||
|
}
|
||||||
|
if args.Header.Get("User-Agent") != "" {
|
||||||
|
headers["User-Agent"] = args.Header.Get("User-Agent")
|
||||||
|
}
|
||||||
|
if !utils.IsLocalIPAddr(args.IP) {
|
||||||
|
headers["X-Forwarded-For"] = args.IP
|
||||||
|
}
|
||||||
|
|
||||||
|
e := splitID(file.GetID())
|
||||||
|
res, err := base.NoRedirectClient.R().
|
||||||
|
SetContext(ctx).
|
||||||
|
SetHeaders(headers).
|
||||||
|
SetQueryParams(map[string]string{
|
||||||
|
"access_token": d.AccessToken,
|
||||||
|
"fsid": e[0],
|
||||||
|
"album_id": e[1],
|
||||||
|
"tid": e[2],
|
||||||
|
"uk": e[3],
|
||||||
|
}).
|
||||||
|
Head(ALBUM_API_URL + "/download")
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
//exp := 8 * time.Hour
|
||||||
|
link := &model.Link{
|
||||||
|
URL: res.Header().Get("location"),
|
||||||
|
Header: http.Header{
|
||||||
|
"User-Agent": []string{headers["User-Agent"]},
|
||||||
|
"Referer": []string{"https://photo.baidu.com/"},
|
||||||
|
},
|
||||||
|
//Expiration: &exp,
|
||||||
|
}
|
||||||
|
return link, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *BaiduPhoto) linkFile(ctx context.Context, file model.Obj, args model.LinkArgs) (*model.Link, error) {
|
||||||
|
headers := map[string]string{
|
||||||
|
"User-Agent": base.UserAgent,
|
||||||
|
}
|
||||||
|
if args.Header.Get("User-Agent") != "" {
|
||||||
|
headers["User-Agent"] = args.Header.Get("User-Agent")
|
||||||
|
}
|
||||||
|
if !utils.IsLocalIPAddr(args.IP) {
|
||||||
|
headers["X-Forwarded-For"] = args.IP
|
||||||
|
}
|
||||||
|
|
||||||
|
var downloadUrl struct {
|
||||||
|
Dlink string `json:"dlink"`
|
||||||
|
}
|
||||||
|
_, err := d.Get(FILE_API_URL_V2+"/download", func(r *resty.Request) {
|
||||||
|
r.SetContext(ctx)
|
||||||
|
r.SetHeaders(headers)
|
||||||
|
r.SetQueryParams(map[string]string{
|
||||||
|
"fsid": splitID(file.GetID())[0],
|
||||||
|
})
|
||||||
|
}, &downloadUrl)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
//exp := 8 * time.Hour
|
||||||
|
link := &model.Link{
|
||||||
|
URL: downloadUrl.Dlink,
|
||||||
|
Header: http.Header{
|
||||||
|
"User-Agent": []string{headers["User-Agent"]},
|
||||||
|
"Referer": []string{"https://photo.baidu.com/"},
|
||||||
|
},
|
||||||
|
//Expiration: &exp,
|
||||||
|
}
|
||||||
|
return link, nil
|
||||||
|
}
|
@ -11,7 +11,7 @@ var NoRedirectClient *resty.Client
|
|||||||
var RestyClient = NewRestyClient()
|
var RestyClient = NewRestyClient()
|
||||||
var HttpClient = &http.Client{}
|
var HttpClient = &http.Client{}
|
||||||
var UserAgent = "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/87.0.4280.88 Safari/537.36"
|
var UserAgent = "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/87.0.4280.88 Safari/537.36"
|
||||||
var DefaultTimeout = time.Second * 10
|
var DefaultTimeout = time.Second * 30
|
||||||
|
|
||||||
func init() {
|
func init() {
|
||||||
NoRedirectClient = resty.New().SetRedirectPolicy(
|
NoRedirectClient = resty.New().SetRedirectPolicy(
|
||||||
|
@ -127,8 +127,4 @@ func (d *FTP) Put(ctx context.Context, dstDir model.Obj, stream model.FileStream
|
|||||||
return d.conn.Stor(stdpath.Join(dstDir.GetPath(), stream.GetName()), stream)
|
return d.conn.Stor(stdpath.Join(dstDir.GetPath(), stream.GetName()), stream)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (d *FTP) Other(ctx context.Context, args model.OtherArgs) (interface{}, error) {
|
|
||||||
return nil, errs.NotSupport
|
|
||||||
}
|
|
||||||
|
|
||||||
var _ driver.Driver = (*FTP)(nil)
|
var _ driver.Driver = (*FTP)(nil)
|
||||||
|
@ -19,5 +19,6 @@ func (d *FTP) login() error {
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
d.conn = conn
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
@ -4,11 +4,14 @@ import (
|
|||||||
"context"
|
"context"
|
||||||
"fmt"
|
"fmt"
|
||||||
"net/http"
|
"net/http"
|
||||||
|
stdpath "path"
|
||||||
|
"strconv"
|
||||||
|
|
||||||
"github.com/alist-org/alist/v3/drivers/base"
|
"github.com/alist-org/alist/v3/drivers/base"
|
||||||
"github.com/alist-org/alist/v3/internal/driver"
|
"github.com/alist-org/alist/v3/internal/driver"
|
||||||
"github.com/alist-org/alist/v3/internal/errs"
|
"github.com/alist-org/alist/v3/internal/errs"
|
||||||
"github.com/alist-org/alist/v3/internal/model"
|
"github.com/alist-org/alist/v3/internal/model"
|
||||||
|
"github.com/alist-org/alist/v3/internal/op"
|
||||||
"github.com/alist-org/alist/v3/pkg/utils"
|
"github.com/alist-org/alist/v3/pkg/utils"
|
||||||
"github.com/go-resty/resty/v2"
|
"github.com/go-resty/resty/v2"
|
||||||
)
|
)
|
||||||
@ -33,6 +36,9 @@ func (d *GoogleDrive) Init(ctx context.Context, storage model.Storage) error {
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
if d.ChunkSize == 0 {
|
||||||
|
d.ChunkSize = 5
|
||||||
|
}
|
||||||
return d.refreshToken()
|
return d.refreshToken()
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -112,15 +118,37 @@ func (d *GoogleDrive) Remove(ctx context.Context, obj model.Obj) error {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (d *GoogleDrive) Put(ctx context.Context, dstDir model.Obj, stream model.FileStreamer, up driver.UpdateProgress) error {
|
func (d *GoogleDrive) Put(ctx context.Context, dstDir model.Obj, stream model.FileStreamer, up driver.UpdateProgress) error {
|
||||||
data := base.Json{
|
obj, _ := op.Get(ctx, d, stdpath.Join(dstDir.GetPath(), stream.GetName()))
|
||||||
"name": stream.GetName(),
|
|
||||||
"parents": []string{dstDir.GetID()},
|
var (
|
||||||
|
e Error
|
||||||
|
url string
|
||||||
|
data base.Json
|
||||||
|
res *resty.Response
|
||||||
|
err error
|
||||||
|
)
|
||||||
|
if obj != nil {
|
||||||
|
url = fmt.Sprintf("https://www.googleapis.com/upload/drive/v3/files/%s?uploadType=resumable&supportsAllDrives=true", obj.GetID())
|
||||||
|
data = base.Json{}
|
||||||
|
} else {
|
||||||
|
data = base.Json{
|
||||||
|
"name": stream.GetName(),
|
||||||
|
"parents": []string{dstDir.GetID()},
|
||||||
|
}
|
||||||
|
url = "https://www.googleapis.com/upload/drive/v3/files?uploadType=resumable&supportsAllDrives=true"
|
||||||
|
}
|
||||||
|
req := base.NoRedirectClient.R().
|
||||||
|
SetHeaders(map[string]string{
|
||||||
|
"Authorization": "Bearer " + d.AccessToken,
|
||||||
|
"X-Upload-Content-Type": stream.GetMimetype(),
|
||||||
|
"X-Upload-Content-Length": strconv.FormatInt(stream.GetSize(), 10),
|
||||||
|
}).
|
||||||
|
SetError(&e).SetBody(data)
|
||||||
|
if obj != nil {
|
||||||
|
res, err = req.Patch(url)
|
||||||
|
} else {
|
||||||
|
res, err = req.Post(url)
|
||||||
}
|
}
|
||||||
var e Error
|
|
||||||
url := "https://www.googleapis.com/upload/drive/v3/files?uploadType=resumable&supportsAllDrives=true"
|
|
||||||
res, err := base.NoRedirectClient.R().SetHeader("Authorization", "Bearer "+d.AccessToken).
|
|
||||||
SetError(&e).SetBody(data).
|
|
||||||
Post(url)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
@ -135,14 +163,14 @@ func (d *GoogleDrive) Put(ctx context.Context, dstDir model.Obj, stream model.Fi
|
|||||||
return fmt.Errorf("%s: %v", e.Error.Message, e.Error.Errors)
|
return fmt.Errorf("%s: %v", e.Error.Message, e.Error.Errors)
|
||||||
}
|
}
|
||||||
putUrl := res.Header().Get("location")
|
putUrl := res.Header().Get("location")
|
||||||
_, err = d.request(putUrl, http.MethodPut, func(req *resty.Request) {
|
if stream.GetSize() < d.ChunkSize*1024*1024 {
|
||||||
req.SetBody(stream.GetReadCloser())
|
_, err = d.request(putUrl, http.MethodPut, func(req *resty.Request) {
|
||||||
}, nil)
|
req.SetHeader("Content-Length", strconv.FormatInt(stream.GetSize(), 10)).SetBody(stream.GetReadCloser())
|
||||||
|
}, nil)
|
||||||
|
} else {
|
||||||
|
err = d.chunkUpload(ctx, stream, putUrl)
|
||||||
|
}
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
func (d *GoogleDrive) Other(ctx context.Context, args model.OtherArgs) (interface{}, error) {
|
|
||||||
return nil, errs.NotSupport
|
|
||||||
}
|
|
||||||
|
|
||||||
var _ driver.Driver = (*GoogleDrive)(nil)
|
var _ driver.Driver = (*GoogleDrive)(nil)
|
||||||
|
@ -12,6 +12,7 @@ type Addition struct {
|
|||||||
OrderDirection string `json:"order_direction" type:"select" options:"asc,desc"`
|
OrderDirection string `json:"order_direction" type:"select" options:"asc,desc"`
|
||||||
ClientID string `json:"client_id" required:"true" default:"202264815644.apps.googleusercontent.com"`
|
ClientID string `json:"client_id" required:"true" default:"202264815644.apps.googleusercontent.com"`
|
||||||
ClientSecret string `json:"client_secret" required:"true" default:"X4Z3ca8xfWDb1Voo-F9a7ZxJ"`
|
ClientSecret string `json:"client_secret" required:"true" default:"X4Z3ca8xfWDb1Voo-F9a7ZxJ"`
|
||||||
|
ChunkSize int64 `json:"chunk_size" default:"5" help:"chunk size while uploading (unit: MB)"`
|
||||||
}
|
}
|
||||||
|
|
||||||
var config = driver.Config{
|
var config = driver.Config{
|
||||||
|
@ -5,6 +5,7 @@ import (
|
|||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/alist-org/alist/v3/internal/model"
|
"github.com/alist-org/alist/v3/internal/model"
|
||||||
|
log "github.com/sirupsen/logrus"
|
||||||
)
|
)
|
||||||
|
|
||||||
type TokenError struct {
|
type TokenError struct {
|
||||||
@ -18,26 +19,36 @@ type Files struct {
|
|||||||
}
|
}
|
||||||
|
|
||||||
type File struct {
|
type File struct {
|
||||||
Id string `json:"id"`
|
Id string `json:"id"`
|
||||||
Name string `json:"name"`
|
Name string `json:"name"`
|
||||||
MimeType string `json:"mimeType"`
|
MimeType string `json:"mimeType"`
|
||||||
ModifiedTime time.Time `json:"modifiedTime"`
|
ModifiedTime time.Time `json:"modifiedTime"`
|
||||||
Size string `json:"size"`
|
Size string `json:"size"`
|
||||||
ThumbnailLink string `json:"thumbnailLink"`
|
ThumbnailLink string `json:"thumbnailLink"`
|
||||||
|
ShortcutDetails struct {
|
||||||
|
TargetId string `json:"targetId"`
|
||||||
|
TargetMimeType string `json:"targetMimeType"`
|
||||||
|
} `json:"shortcutDetails"`
|
||||||
}
|
}
|
||||||
|
|
||||||
func fileToObj(f File) *model.ObjThumb {
|
func fileToObj(f File) *model.ObjThumb {
|
||||||
|
log.Debugf("google file: %+v", f)
|
||||||
size, _ := strconv.ParseInt(f.Size, 10, 64)
|
size, _ := strconv.ParseInt(f.Size, 10, 64)
|
||||||
return &model.ObjThumb{
|
obj := &model.ObjThumb{
|
||||||
Object: model.Object{
|
Object: model.Object{
|
||||||
ID: f.Id,
|
ID: f.Id,
|
||||||
Name: f.Name,
|
Name: f.Name,
|
||||||
Size: size,
|
Size: size,
|
||||||
Modified: time.Time{},
|
Modified: f.ModifiedTime,
|
||||||
IsFolder: false,
|
IsFolder: f.MimeType == "application/vnd.google-apps.folder",
|
||||||
},
|
},
|
||||||
Thumbnail: model.Thumbnail{},
|
Thumbnail: model.Thumbnail{},
|
||||||
}
|
}
|
||||||
|
if f.MimeType == "application/vnd.google-apps.shortcut" {
|
||||||
|
obj.ID = f.ShortcutDetails.TargetId
|
||||||
|
obj.IsFolder = f.ShortcutDetails.TargetMimeType == "application/vnd.google-apps.folder"
|
||||||
|
}
|
||||||
|
return obj
|
||||||
}
|
}
|
||||||
|
|
||||||
type Error struct {
|
type Error struct {
|
||||||
|
@ -1,10 +1,14 @@
|
|||||||
package google_drive
|
package google_drive
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"context"
|
||||||
"fmt"
|
"fmt"
|
||||||
|
"io"
|
||||||
"net/http"
|
"net/http"
|
||||||
|
"strconv"
|
||||||
|
|
||||||
"github.com/alist-org/alist/v3/drivers/base"
|
"github.com/alist-org/alist/v3/drivers/base"
|
||||||
|
"github.com/alist-org/alist/v3/internal/model"
|
||||||
"github.com/go-resty/resty/v2"
|
"github.com/go-resty/resty/v2"
|
||||||
log "github.com/sirupsen/logrus"
|
log "github.com/sirupsen/logrus"
|
||||||
)
|
)
|
||||||
@ -77,7 +81,7 @@ func (d *GoogleDrive) getFiles(id string) ([]File, error) {
|
|||||||
}
|
}
|
||||||
query := map[string]string{
|
query := map[string]string{
|
||||||
"orderBy": orderBy,
|
"orderBy": orderBy,
|
||||||
"fields": "files(id,name,mimeType,size,modifiedTime,thumbnailLink),nextPageToken",
|
"fields": "files(id,name,mimeType,size,modifiedTime,thumbnailLink,shortcutDetails),nextPageToken",
|
||||||
"pageSize": "1000",
|
"pageSize": "1000",
|
||||||
"q": fmt.Sprintf("'%s' in parents and trashed = false", id),
|
"q": fmt.Sprintf("'%s' in parents and trashed = false", id),
|
||||||
//"includeItemsFromAllDrives": "true",
|
//"includeItemsFromAllDrives": "true",
|
||||||
@ -95,3 +99,25 @@ func (d *GoogleDrive) getFiles(id string) ([]File, error) {
|
|||||||
}
|
}
|
||||||
return res, nil
|
return res, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (d *GoogleDrive) chunkUpload(ctx context.Context, stream model.FileStreamer, url string) error {
|
||||||
|
var defaultChunkSize = d.ChunkSize * 1024 * 1024
|
||||||
|
var finish int64 = 0
|
||||||
|
for finish < stream.GetSize() {
|
||||||
|
chunkSize := stream.GetSize() - finish
|
||||||
|
if chunkSize > defaultChunkSize {
|
||||||
|
chunkSize = defaultChunkSize
|
||||||
|
}
|
||||||
|
_, err := d.request(url, http.MethodPut, func(req *resty.Request) {
|
||||||
|
req.SetHeaders(map[string]string{
|
||||||
|
"Content-Length": strconv.FormatInt(chunkSize, 10),
|
||||||
|
"Content-Range": fmt.Sprintf("bytes %d-%d/%d", finish, finish+chunkSize-1, stream.GetSize()),
|
||||||
|
}).SetBody(io.LimitReader(stream.GetReadCloser(), chunkSize))
|
||||||
|
}, nil)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
finish += chunkSize
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
170
drivers/google_photo/driver.go
Normal file
170
drivers/google_photo/driver.go
Normal file
@ -0,0 +1,170 @@
|
|||||||
|
package google_photo
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"fmt"
|
||||||
|
"net/http"
|
||||||
|
"strconv"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
"github.com/alist-org/alist/v3/drivers/base"
|
||||||
|
"github.com/alist-org/alist/v3/internal/driver"
|
||||||
|
"github.com/alist-org/alist/v3/internal/errs"
|
||||||
|
"github.com/alist-org/alist/v3/internal/model"
|
||||||
|
"github.com/alist-org/alist/v3/pkg/utils"
|
||||||
|
"github.com/go-resty/resty/v2"
|
||||||
|
)
|
||||||
|
|
||||||
|
type GooglePhoto struct {
|
||||||
|
model.Storage
|
||||||
|
Addition
|
||||||
|
AccessToken string
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *GooglePhoto) Config() driver.Config {
|
||||||
|
return config
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *GooglePhoto) GetAddition() driver.Additional {
|
||||||
|
return d.Addition
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *GooglePhoto) Init(ctx context.Context, storage model.Storage) error {
|
||||||
|
d.Storage = storage
|
||||||
|
err := utils.Json.UnmarshalFromString(d.Storage.Addition, &d.Addition)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
return d.refreshToken()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *GooglePhoto) Drop(ctx context.Context) error {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *GooglePhoto) List(ctx context.Context, dir model.Obj, args model.ListArgs) ([]model.Obj, error) {
|
||||||
|
files, err := d.getFiles(dir.GetID())
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return utils.SliceConvert(files, func(src MediaItem) (model.Obj, error) {
|
||||||
|
return fileToObj(src), nil
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
//func (d *GooglePhoto) Get(ctx context.Context, path string) (model.Obj, error) {
|
||||||
|
// // this is optional
|
||||||
|
// return nil, errs.NotImplement
|
||||||
|
//}
|
||||||
|
|
||||||
|
func (d *GooglePhoto) Link(ctx context.Context, file model.Obj, args model.LinkArgs) (*model.Link, error) {
|
||||||
|
f, err := d.getMedia(file.GetID())
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
if strings.Contains(f.MimeType, "image/") {
|
||||||
|
return &model.Link{
|
||||||
|
URL: f.BaseURL + "=d",
|
||||||
|
}, nil
|
||||||
|
} else if strings.Contains(f.MimeType, "video/") {
|
||||||
|
return &model.Link{
|
||||||
|
URL: f.BaseURL + "=dv",
|
||||||
|
}, nil
|
||||||
|
}
|
||||||
|
return &model.Link{}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *GooglePhoto) MakeDir(ctx context.Context, parentDir model.Obj, dirName string) error {
|
||||||
|
return errs.NotSupport
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *GooglePhoto) Move(ctx context.Context, srcObj, dstDir model.Obj) error {
|
||||||
|
return errs.NotSupport
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *GooglePhoto) Rename(ctx context.Context, srcObj model.Obj, newName string) error {
|
||||||
|
return errs.NotSupport
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *GooglePhoto) Copy(ctx context.Context, srcObj, dstDir model.Obj) error {
|
||||||
|
return errs.NotSupport
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *GooglePhoto) Remove(ctx context.Context, obj model.Obj) error {
|
||||||
|
return errs.NotSupport
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *GooglePhoto) Put(ctx context.Context, dstDir model.Obj, stream model.FileStreamer, up driver.UpdateProgress) error {
|
||||||
|
var e Error
|
||||||
|
// Create resumable upload url
|
||||||
|
postHeaders := map[string]string{
|
||||||
|
"Authorization": "Bearer " + d.AccessToken,
|
||||||
|
"Content-type": "application/octet-stream",
|
||||||
|
"X-Goog-Upload-Command": "start",
|
||||||
|
"X-Goog-Upload-Content-Type": stream.GetMimetype(),
|
||||||
|
"X-Goog-Upload-Protocol": "resumable",
|
||||||
|
"X-Goog-Upload-Raw-Size": strconv.FormatInt(stream.GetSize(), 10),
|
||||||
|
}
|
||||||
|
url := "https://photoslibrary.googleapis.com/v1/uploads"
|
||||||
|
res, err := base.NoRedirectClient.R().SetHeaders(postHeaders).
|
||||||
|
SetError(&e).
|
||||||
|
Post(url)
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if e.Error.Code != 0 {
|
||||||
|
if e.Error.Code == 401 {
|
||||||
|
err = d.refreshToken()
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
return d.Put(ctx, dstDir, stream, up)
|
||||||
|
}
|
||||||
|
return fmt.Errorf("%s: %v", e.Error.Message, e.Error.Errors)
|
||||||
|
}
|
||||||
|
|
||||||
|
//Upload to the Google Photo
|
||||||
|
postUrl := res.Header().Get("X-Goog-Upload-URL")
|
||||||
|
//chunkSize := res.Header().Get("X-Goog-Upload-Chunk-Granularity")
|
||||||
|
postHeaders = map[string]string{
|
||||||
|
"X-Goog-Upload-Command": "upload, finalize",
|
||||||
|
"X-Goog-Upload-Offset": "0",
|
||||||
|
}
|
||||||
|
|
||||||
|
resp, err := d.request(postUrl, http.MethodPost, func(req *resty.Request) {
|
||||||
|
req.SetBody(stream.GetReadCloser())
|
||||||
|
}, nil, postHeaders)
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
//Create MediaItem
|
||||||
|
createItemUrl := "https://photoslibrary.googleapis.com/v1/mediaItems:batchCreate"
|
||||||
|
|
||||||
|
postHeaders = map[string]string{
|
||||||
|
"X-Goog-Upload-Command": "upload, finalize",
|
||||||
|
"X-Goog-Upload-Offset": "0",
|
||||||
|
}
|
||||||
|
|
||||||
|
data := base.Json{
|
||||||
|
"newMediaItems": []base.Json{
|
||||||
|
{
|
||||||
|
"description": "item-description",
|
||||||
|
"simpleMediaItem": base.Json{
|
||||||
|
"fileName": stream.GetName(),
|
||||||
|
"uploadToken": string(resp),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
_, err = d.request(createItemUrl, http.MethodPost, func(req *resty.Request) {
|
||||||
|
req.SetBody(data)
|
||||||
|
}, nil, postHeaders)
|
||||||
|
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
var _ driver.Driver = (*GooglePhoto)(nil)
|
30
drivers/google_photo/meta.go
Normal file
30
drivers/google_photo/meta.go
Normal file
@ -0,0 +1,30 @@
|
|||||||
|
package google_photo
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/alist-org/alist/v3/internal/driver"
|
||||||
|
"github.com/alist-org/alist/v3/internal/op"
|
||||||
|
)
|
||||||
|
|
||||||
|
type Addition struct {
|
||||||
|
driver.RootID
|
||||||
|
RefreshToken string `json:"refresh_token" required:"true"`
|
||||||
|
ClientID string `json:"client_id" required:"true" default:"202264815644.apps.googleusercontent.com"`
|
||||||
|
ClientSecret string `json:"client_secret" required:"true" default:"X4Z3ca8xfWDb1Voo-F9a7ZxJ"`
|
||||||
|
ShowArchive bool `json:"show_archive"`
|
||||||
|
}
|
||||||
|
|
||||||
|
var config = driver.Config{
|
||||||
|
Name: "GooglePhoto",
|
||||||
|
OnlyProxy: true,
|
||||||
|
DefaultRoot: "root",
|
||||||
|
NoUpload: true,
|
||||||
|
LocalSort: true,
|
||||||
|
}
|
||||||
|
|
||||||
|
func New() driver.Driver {
|
||||||
|
return &GooglePhoto{}
|
||||||
|
}
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
op.RegisterDriver(config, New)
|
||||||
|
}
|
85
drivers/google_photo/types.go
Normal file
85
drivers/google_photo/types.go
Normal file
@ -0,0 +1,85 @@
|
|||||||
|
package google_photo
|
||||||
|
|
||||||
|
import (
|
||||||
|
"reflect"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/alist-org/alist/v3/internal/model"
|
||||||
|
)
|
||||||
|
|
||||||
|
type TokenError struct {
|
||||||
|
Error string `json:"error"`
|
||||||
|
ErrorDescription string `json:"error_description"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type Items struct {
|
||||||
|
NextPageToken string `json:"nextPageToken"`
|
||||||
|
MediaItems []MediaItem `json:"mediaItems,omitempty"`
|
||||||
|
Albums []MediaItem `json:"albums,omitempty"`
|
||||||
|
SharedAlbums []MediaItem `json:"sharedAlbums,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type MediaItem struct {
|
||||||
|
Id string `json:"id"`
|
||||||
|
Title string `json:"title,omitempty"`
|
||||||
|
BaseURL string `json:"baseUrl,omitempty"`
|
||||||
|
CoverPhotoBaseUrl string `json:"coverPhotoBaseUrl,omitempty"`
|
||||||
|
MimeType string `json:"mimeType,omitempty"`
|
||||||
|
FileName string `json:"filename,omitempty"`
|
||||||
|
MediaMetadata MediaMetadata `json:"mediaMetadata,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type MediaMetadata struct {
|
||||||
|
CreationTime time.Time `json:"creationTime"`
|
||||||
|
Width string `json:"width"`
|
||||||
|
Height string `json:"height"`
|
||||||
|
Photo Photo `json:"photo,omitempty"`
|
||||||
|
Video Video `json:"video,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type Photo struct {
|
||||||
|
}
|
||||||
|
|
||||||
|
type Video struct {
|
||||||
|
}
|
||||||
|
|
||||||
|
func fileToObj(f MediaItem) *model.ObjThumb {
|
||||||
|
if !reflect.DeepEqual(f.MediaMetadata, MediaMetadata{}){
|
||||||
|
return &model.ObjThumb{
|
||||||
|
Object: model.Object{
|
||||||
|
ID: f.Id,
|
||||||
|
Name: f.FileName,
|
||||||
|
Size: 0,
|
||||||
|
Modified: f.MediaMetadata.CreationTime,
|
||||||
|
IsFolder: false,
|
||||||
|
},
|
||||||
|
Thumbnail: model.Thumbnail{
|
||||||
|
Thumbnail: f.BaseURL + "=w100-h100-c",
|
||||||
|
},
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return &model.ObjThumb{
|
||||||
|
Object: model.Object{
|
||||||
|
ID: f.Id,
|
||||||
|
Name: f.Title,
|
||||||
|
Size: 0,
|
||||||
|
Modified: time.Time{},
|
||||||
|
IsFolder: true,
|
||||||
|
},
|
||||||
|
Thumbnail: model.Thumbnail{},
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
type Error struct {
|
||||||
|
Error struct {
|
||||||
|
Errors []struct {
|
||||||
|
Domain string `json:"domain"`
|
||||||
|
Reason string `json:"reason"`
|
||||||
|
Message string `json:"message"`
|
||||||
|
LocationType string `json:"location_type"`
|
||||||
|
Location string `json:"location"`
|
||||||
|
}
|
||||||
|
Code int `json:"code"`
|
||||||
|
Message string `json:"message"`
|
||||||
|
} `json:"error"`
|
||||||
|
}
|
186
drivers/google_photo/util.go
Normal file
186
drivers/google_photo/util.go
Normal file
@ -0,0 +1,186 @@
|
|||||||
|
package google_photo
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"net/http"
|
||||||
|
|
||||||
|
"github.com/alist-org/alist/v3/drivers/base"
|
||||||
|
"github.com/go-resty/resty/v2"
|
||||||
|
)
|
||||||
|
|
||||||
|
// do others that not defined in Driver interface
|
||||||
|
|
||||||
|
const (
|
||||||
|
FETCH_ALL = "all"
|
||||||
|
FETCH_ALBUMS = "albums"
|
||||||
|
FETCH_ROOT = "root"
|
||||||
|
FETCH_SHARE_ALBUMS = "share_albums"
|
||||||
|
)
|
||||||
|
|
||||||
|
func (d *GooglePhoto) refreshToken() error {
|
||||||
|
url := "https://www.googleapis.com/oauth2/v4/token"
|
||||||
|
var resp base.TokenResp
|
||||||
|
var e TokenError
|
||||||
|
_, err := base.RestyClient.R().SetResult(&resp).SetError(&e).
|
||||||
|
SetFormData(map[string]string{
|
||||||
|
"client_id": d.ClientID,
|
||||||
|
"client_secret": d.ClientSecret,
|
||||||
|
"refresh_token": d.RefreshToken,
|
||||||
|
"grant_type": "refresh_token",
|
||||||
|
}).Post(url)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if e.Error != "" {
|
||||||
|
return fmt.Errorf(e.Error)
|
||||||
|
}
|
||||||
|
d.AccessToken = resp.AccessToken
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *GooglePhoto) request(url string, method string, callback base.ReqCallback, resp interface{}, headers map[string]string) ([]byte, error) {
|
||||||
|
req := base.RestyClient.R()
|
||||||
|
req.SetHeader("Authorization", "Bearer "+d.AccessToken)
|
||||||
|
req.SetHeader("Accept-Encoding", "gzip")
|
||||||
|
if headers != nil {
|
||||||
|
req.SetHeaders(headers)
|
||||||
|
}
|
||||||
|
|
||||||
|
if callback != nil {
|
||||||
|
callback(req)
|
||||||
|
}
|
||||||
|
if resp != nil {
|
||||||
|
req.SetResult(resp)
|
||||||
|
}
|
||||||
|
var e Error
|
||||||
|
req.SetError(&e)
|
||||||
|
res, err := req.Execute(method, url)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if e.Error.Code != 0 {
|
||||||
|
if e.Error.Code == 401 {
|
||||||
|
err = d.refreshToken()
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return d.request(url, method, callback, resp, headers)
|
||||||
|
}
|
||||||
|
return nil, fmt.Errorf("%s: %v", e.Error.Message, e.Error.Errors)
|
||||||
|
}
|
||||||
|
return res.Body(), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *GooglePhoto) getFiles(id string) ([]MediaItem, error) {
|
||||||
|
switch id {
|
||||||
|
case FETCH_ALL:
|
||||||
|
return d.getAllMedias()
|
||||||
|
case FETCH_ALBUMS:
|
||||||
|
return d.getAlbums()
|
||||||
|
case FETCH_SHARE_ALBUMS:
|
||||||
|
return d.getShareAlbums()
|
||||||
|
case FETCH_ROOT:
|
||||||
|
return d.getFakeRoot()
|
||||||
|
default:
|
||||||
|
return d.getMedias(id)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *GooglePhoto) getFakeRoot() ([]MediaItem, error) {
|
||||||
|
return []MediaItem{
|
||||||
|
{
|
||||||
|
Id: FETCH_ALL,
|
||||||
|
Title: "全部媒体",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
Id: FETCH_ALBUMS,
|
||||||
|
Title: "全部影集",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
Id: FETCH_SHARE_ALBUMS,
|
||||||
|
Title: "共享影集",
|
||||||
|
},
|
||||||
|
}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *GooglePhoto) getAlbums() ([]MediaItem, error) {
|
||||||
|
return d.fetchItems(
|
||||||
|
"https://photoslibrary.googleapis.com/v1/albums",
|
||||||
|
map[string]string{
|
||||||
|
"fields": "albums(id,title,coverPhotoBaseUrl),nextPageToken",
|
||||||
|
"pageSize": "50",
|
||||||
|
"pageToken": "first",
|
||||||
|
},
|
||||||
|
http.MethodGet)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *GooglePhoto) getShareAlbums() ([]MediaItem, error) {
|
||||||
|
return d.fetchItems(
|
||||||
|
"https://photoslibrary.googleapis.com/v1/sharedAlbums",
|
||||||
|
map[string]string{
|
||||||
|
"fields": "sharedAlbums(id,title,coverPhotoBaseUrl),nextPageToken",
|
||||||
|
"pageSize": "50",
|
||||||
|
"pageToken": "first",
|
||||||
|
},
|
||||||
|
http.MethodGet)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *GooglePhoto) getMedias(albumId string) ([]MediaItem, error) {
|
||||||
|
return d.fetchItems(
|
||||||
|
"https://photoslibrary.googleapis.com/v1/mediaItems:search",
|
||||||
|
map[string]string{
|
||||||
|
"fields": "mediaItems(id,baseUrl,mimeType,mediaMetadata,filename),nextPageToken",
|
||||||
|
"pageSize": "100",
|
||||||
|
"albumId": albumId,
|
||||||
|
"pageToken": "first",
|
||||||
|
}, http.MethodPost)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *GooglePhoto) getAllMedias() ([]MediaItem, error) {
|
||||||
|
return d.fetchItems(
|
||||||
|
"https://photoslibrary.googleapis.com/v1/mediaItems",
|
||||||
|
map[string]string{
|
||||||
|
"fields": "mediaItems(id,baseUrl,mimeType,mediaMetadata,filename),nextPageToken",
|
||||||
|
"pageSize": "100",
|
||||||
|
"pageToken": "first",
|
||||||
|
},
|
||||||
|
http.MethodGet)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *GooglePhoto) getMedia(id string) (MediaItem, error) {
|
||||||
|
var resp MediaItem
|
||||||
|
|
||||||
|
query := map[string]string{
|
||||||
|
"fields": "baseUrl,mimeType",
|
||||||
|
}
|
||||||
|
_, err := d.request(fmt.Sprintf("https://photoslibrary.googleapis.com/v1/mediaItems/%s", id), http.MethodGet, func(req *resty.Request) {
|
||||||
|
req.SetQueryParams(query)
|
||||||
|
}, &resp, nil)
|
||||||
|
if err != nil {
|
||||||
|
return resp, err
|
||||||
|
}
|
||||||
|
|
||||||
|
return resp, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *GooglePhoto) fetchItems(url string, query map[string]string, method string) ([]MediaItem, error){
|
||||||
|
res := make([]MediaItem, 0)
|
||||||
|
for query["pageToken"] != "" {
|
||||||
|
if query["pageToken"] == "first" {
|
||||||
|
query["pageToken"] = ""
|
||||||
|
}
|
||||||
|
var resp Items
|
||||||
|
|
||||||
|
_, err := d.request(url, method, func(req *resty.Request) {
|
||||||
|
req.SetQueryParams(query)
|
||||||
|
}, &resp, nil)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
query["pageToken"] = resp.NextPageToken
|
||||||
|
res = append(res, resp.MediaItems...)
|
||||||
|
res = append(res, resp.Albums...)
|
||||||
|
res = append(res, resp.SharedAlbums...)
|
||||||
|
}
|
||||||
|
return res, nil
|
||||||
|
}
|
171
drivers/lanzou/driver.go
Normal file
171
drivers/lanzou/driver.go
Normal file
@ -0,0 +1,171 @@
|
|||||||
|
package lanzou
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"net/http"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/alist-org/alist/v3/drivers/base"
|
||||||
|
"github.com/alist-org/alist/v3/internal/driver"
|
||||||
|
"github.com/alist-org/alist/v3/internal/errs"
|
||||||
|
"github.com/alist-org/alist/v3/internal/model"
|
||||||
|
"github.com/alist-org/alist/v3/pkg/utils"
|
||||||
|
"github.com/go-resty/resty/v2"
|
||||||
|
)
|
||||||
|
|
||||||
|
var upClient = base.NewRestyClient().SetTimeout(120 * time.Second)
|
||||||
|
|
||||||
|
type LanZou struct {
|
||||||
|
Addition
|
||||||
|
model.Storage
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *LanZou) Config() driver.Config {
|
||||||
|
return config
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *LanZou) GetAddition() driver.Additional {
|
||||||
|
return d.Addition
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *LanZou) Init(ctx context.Context, storage model.Storage) error {
|
||||||
|
d.Storage = storage
|
||||||
|
err := utils.Json.UnmarshalFromString(d.Storage.Addition, &d.Addition)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if d.IsCookie() {
|
||||||
|
if d.RootFolderID == "" {
|
||||||
|
d.RootFolderID = "-1"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *LanZou) Drop(ctx context.Context) error {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// 获取的大小和时间不准确
|
||||||
|
func (d *LanZou) List(ctx context.Context, dir model.Obj, args model.ListArgs) ([]model.Obj, error) {
|
||||||
|
if d.IsCookie() {
|
||||||
|
return d.GetFiles(ctx, dir.GetID())
|
||||||
|
} else {
|
||||||
|
return d.GetFileOrFolderByShareUrl(ctx, dir.GetID(), d.SharePassword)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *LanZou) Link(ctx context.Context, file model.Obj, args model.LinkArgs) (*model.Link, error) {
|
||||||
|
downID := file.GetID()
|
||||||
|
pwd := d.SharePassword
|
||||||
|
if d.IsCookie() {
|
||||||
|
share, err := d.getFileShareUrlByID(ctx, file.GetID())
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
downID = share.FID
|
||||||
|
pwd = share.Pwd
|
||||||
|
}
|
||||||
|
fileInfo, err := d.getFilesByShareUrl(ctx, downID, pwd, nil)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
return &model.Link{
|
||||||
|
URL: fileInfo.Url,
|
||||||
|
Header: http.Header{
|
||||||
|
"User-Agent": []string{base.UserAgent},
|
||||||
|
},
|
||||||
|
}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *LanZou) MakeDir(ctx context.Context, parentDir model.Obj, dirName string) error {
|
||||||
|
if d.IsCookie() {
|
||||||
|
_, err := d.post(d.BaseUrl+"/doupload.php", func(req *resty.Request) {
|
||||||
|
req.SetContext(ctx)
|
||||||
|
req.SetFormData(map[string]string{
|
||||||
|
"task": "2",
|
||||||
|
"parent_id": parentDir.GetID(),
|
||||||
|
"folder_name": dirName,
|
||||||
|
"folder_description": "",
|
||||||
|
})
|
||||||
|
}, nil)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
return errs.NotImplement
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *LanZou) Move(ctx context.Context, srcObj, dstDir model.Obj) error {
|
||||||
|
if d.IsCookie() {
|
||||||
|
if !srcObj.IsDir() {
|
||||||
|
_, err := d.post(d.BaseUrl+"/doupload.php", func(req *resty.Request) {
|
||||||
|
req.SetContext(ctx)
|
||||||
|
req.SetFormData(map[string]string{
|
||||||
|
"task": "20",
|
||||||
|
"folder_id": dstDir.GetID(),
|
||||||
|
"file_id": srcObj.GetID(),
|
||||||
|
})
|
||||||
|
}, nil)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return errs.NotImplement
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *LanZou) Rename(ctx context.Context, srcObj model.Obj, newName string) error {
|
||||||
|
if d.IsCookie() {
|
||||||
|
if !srcObj.IsDir() {
|
||||||
|
_, err := d.post(d.BaseUrl+"/doupload.php", func(req *resty.Request) {
|
||||||
|
req.SetContext(ctx)
|
||||||
|
req.SetFormData(map[string]string{
|
||||||
|
"task": "46",
|
||||||
|
"file_id": srcObj.GetID(),
|
||||||
|
"file_name": newName,
|
||||||
|
"type": "2",
|
||||||
|
})
|
||||||
|
}, nil)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return errs.NotImplement
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *LanZou) Copy(ctx context.Context, srcObj, dstDir model.Obj) error {
|
||||||
|
return errs.NotImplement
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *LanZou) Remove(ctx context.Context, obj model.Obj) error {
|
||||||
|
if d.IsCookie() {
|
||||||
|
_, err := d.post(d.BaseUrl+"/doupload.php", func(req *resty.Request) {
|
||||||
|
req.SetContext(ctx)
|
||||||
|
if obj.IsDir() {
|
||||||
|
req.SetFormData(map[string]string{
|
||||||
|
"task": "3",
|
||||||
|
"folder_id": obj.GetID(),
|
||||||
|
})
|
||||||
|
} else {
|
||||||
|
req.SetFormData(map[string]string{
|
||||||
|
"task": "6",
|
||||||
|
"file_id": obj.GetID(),
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}, nil)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
return errs.NotImplement
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *LanZou) Put(ctx context.Context, dstDir model.Obj, stream model.FileStreamer, up driver.UpdateProgress) error {
|
||||||
|
if d.IsCookie() {
|
||||||
|
_, err := d._post(d.BaseUrl+"/fileup.php", func(req *resty.Request) {
|
||||||
|
req.SetFormData(map[string]string{
|
||||||
|
"task": "1",
|
||||||
|
"id": "WU_FILE_0",
|
||||||
|
"name": stream.GetName(),
|
||||||
|
"folder_id": dstDir.GetID(),
|
||||||
|
}).SetFileReader("upload_file", stream.GetName(), stream)
|
||||||
|
}, nil, true)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
return errs.NotImplement
|
||||||
|
}
|
165
drivers/lanzou/help.go
Normal file
165
drivers/lanzou/help.go
Normal file
@ -0,0 +1,165 @@
|
|||||||
|
package lanzou
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
"fmt"
|
||||||
|
"regexp"
|
||||||
|
"strconv"
|
||||||
|
"strings"
|
||||||
|
"time"
|
||||||
|
"unicode"
|
||||||
|
)
|
||||||
|
|
||||||
|
const DAY time.Duration = 84600000000000
|
||||||
|
|
||||||
|
var timeSplitReg = regexp.MustCompile("([0-9.]*)\\s*([\u4e00-\u9fa5]+)")
|
||||||
|
|
||||||
|
func MustParseTime(str string) time.Time {
|
||||||
|
lastOpTime, err := time.ParseInLocation("2006-01-02 -07", str+" +08", time.Local)
|
||||||
|
if err != nil {
|
||||||
|
strs := timeSplitReg.FindStringSubmatch(str)
|
||||||
|
lastOpTime = time.Now()
|
||||||
|
if len(strs) == 3 {
|
||||||
|
i, _ := strconv.ParseInt(strs[1], 10, 64)
|
||||||
|
ti := time.Duration(-i)
|
||||||
|
switch strs[2] {
|
||||||
|
case "秒前":
|
||||||
|
lastOpTime = lastOpTime.Add(time.Second * ti)
|
||||||
|
case "分钟前":
|
||||||
|
lastOpTime = lastOpTime.Add(time.Minute * ti)
|
||||||
|
case "小时前":
|
||||||
|
lastOpTime = lastOpTime.Add(time.Hour * ti)
|
||||||
|
case "天前":
|
||||||
|
lastOpTime = lastOpTime.Add(DAY * ti)
|
||||||
|
case "昨天":
|
||||||
|
lastOpTime = lastOpTime.Add(-DAY)
|
||||||
|
case "前天":
|
||||||
|
lastOpTime = lastOpTime.Add(-DAY * 2)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return lastOpTime
|
||||||
|
}
|
||||||
|
|
||||||
|
var sizeSplitReg = regexp.MustCompile(`(?i)([0-9.]+)\s*([bkm]+)`)
|
||||||
|
|
||||||
|
func SizeStrToInt64(size string) int64 {
|
||||||
|
strs := sizeSplitReg.FindStringSubmatch(size)
|
||||||
|
if len(strs) < 3 {
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
|
||||||
|
s, _ := strconv.ParseFloat(strs[1], 64)
|
||||||
|
switch strings.ToUpper(strs[2]) {
|
||||||
|
case "B":
|
||||||
|
return int64(s)
|
||||||
|
case "K":
|
||||||
|
return int64(s * (1 << 10))
|
||||||
|
case "M":
|
||||||
|
return int64(s * (1 << 20))
|
||||||
|
}
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
|
||||||
|
// 移除注释
|
||||||
|
func RemoveNotes(html []byte) []byte {
|
||||||
|
return regexp.MustCompile(`<!--.*?-->|//.*|/\*.*?\*/`).ReplaceAll(html, []byte{})
|
||||||
|
}
|
||||||
|
|
||||||
|
var findAcwScV2Reg = regexp.MustCompile(`arg1='([0-9A-Z]+)'`)
|
||||||
|
|
||||||
|
// 在页面被过多访问或其他情况下,有时候会先返回一个加密的页面,其执行计算出一个acw_sc__v2后放入页面后再重新访问页面才能获得正常页面
|
||||||
|
// 若该页面进行了js加密,则进行解密,计算acw_sc__v2,并加入cookie
|
||||||
|
func CalcAcwScV2(html string) (string, error) {
|
||||||
|
acwScV2s := findAcwScV2Reg.FindStringSubmatch(html)
|
||||||
|
if len(acwScV2s) != 2 {
|
||||||
|
return "", fmt.Errorf("无法匹配acw_sc__v2")
|
||||||
|
}
|
||||||
|
return HexXor(Unbox(acwScV2s[1]), "3000176000856006061501533003690027800375"), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func Unbox(hex string) string {
|
||||||
|
var box = []int{6, 28, 34, 31, 33, 18, 30, 23, 9, 8, 19, 38, 17, 24, 0, 5, 32, 21, 10, 22, 25, 14, 15, 3, 16, 27, 13, 35, 2, 29, 11, 26, 4, 36, 1, 39, 37, 7, 20, 12}
|
||||||
|
var newBox = make([]byte, len(hex))
|
||||||
|
for i := 0; i < len(box); i++ {
|
||||||
|
j := box[i]
|
||||||
|
if len(newBox) > j {
|
||||||
|
newBox[j] = hex[i]
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return string(newBox)
|
||||||
|
}
|
||||||
|
|
||||||
|
func HexXor(hex1, hex2 string) string {
|
||||||
|
out := bytes.NewBuffer(make([]byte, len(hex1)))
|
||||||
|
for i := 0; i < len(hex1) && i < len(hex2); i += 2 {
|
||||||
|
v1, _ := strconv.ParseInt(hex1[i:i+2], 16, 64)
|
||||||
|
v2, _ := strconv.ParseInt(hex2[i:i+2], 16, 64)
|
||||||
|
out.WriteString(strconv.FormatInt(v1^v2, 16))
|
||||||
|
}
|
||||||
|
return out.String()
|
||||||
|
}
|
||||||
|
|
||||||
|
var findDataReg = regexp.MustCompile(`data[:\s]+({[^}]+})`) // 查找json
|
||||||
|
var findKVReg = regexp.MustCompile(`'(.+?)':('?([^' },]*)'?)`) // 拆分kv
|
||||||
|
|
||||||
|
// 根据key查询js变量
|
||||||
|
func findJSVarFunc(key, data string) string {
|
||||||
|
values := regexp.MustCompile(`var ` + key + ` = '(.+?)';`).FindStringSubmatch(data)
|
||||||
|
if len(values) == 0 {
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
return values[1]
|
||||||
|
}
|
||||||
|
|
||||||
|
// 解析html中的JSON
|
||||||
|
func htmlJsonToMap(html string) (map[string]string, error) {
|
||||||
|
datas := findDataReg.FindStringSubmatch(html)
|
||||||
|
if len(datas) != 2 {
|
||||||
|
return nil, fmt.Errorf("not find data")
|
||||||
|
}
|
||||||
|
return jsonToMap(datas[1], html), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func jsonToMap(data, html string) map[string]string {
|
||||||
|
var param = make(map[string]string)
|
||||||
|
kvs := findKVReg.FindAllStringSubmatch(data, -1)
|
||||||
|
for _, kv := range kvs {
|
||||||
|
k, v := kv[1], kv[3]
|
||||||
|
if v == "" || strings.Contains(kv[2], "'") || IsNumber(kv[2]) {
|
||||||
|
param[k] = v
|
||||||
|
} else {
|
||||||
|
param[k] = findJSVarFunc(v, html)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return param
|
||||||
|
}
|
||||||
|
|
||||||
|
func IsNumber(str string) bool {
|
||||||
|
for _, s := range str {
|
||||||
|
if !unicode.IsDigit(s) {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
var findFromReg = regexp.MustCompile(`data : '(.+?)'`) // 查找from字符串
|
||||||
|
|
||||||
|
// 解析html中的form
|
||||||
|
func htmlFormToMap(html string) (map[string]string, error) {
|
||||||
|
forms := findFromReg.FindStringSubmatch(html)
|
||||||
|
if len(forms) != 2 {
|
||||||
|
return nil, fmt.Errorf("not find file sgin")
|
||||||
|
}
|
||||||
|
return formToMap(forms[1]), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func formToMap(from string) map[string]string {
|
||||||
|
var param = make(map[string]string)
|
||||||
|
for _, kv := range strings.Split(from, "&") {
|
||||||
|
kv := strings.SplitN(kv, "=", 2)[:2]
|
||||||
|
param[kv[0]] = kv[1]
|
||||||
|
}
|
||||||
|
return param
|
||||||
|
}
|
31
drivers/lanzou/meta.go
Normal file
31
drivers/lanzou/meta.go
Normal file
@ -0,0 +1,31 @@
|
|||||||
|
package lanzou
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/alist-org/alist/v3/internal/driver"
|
||||||
|
"github.com/alist-org/alist/v3/internal/op"
|
||||||
|
)
|
||||||
|
|
||||||
|
type Addition struct {
|
||||||
|
Type string `json:"type" type:"select" options:"cookie,url" default:"cookie"`
|
||||||
|
Cookie string `json:"cookie" required:"true" help:"about 15 days valid"`
|
||||||
|
driver.RootID
|
||||||
|
SharePassword string `json:"share_password"`
|
||||||
|
BaseUrl string `json:"baseUrl" required:"true" default:"https://pc.woozooo.com"`
|
||||||
|
ShareUrl string `json:"shareUrl" required:"true" default:"https://pan.lanzouo.com"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a *Addition) IsCookie() bool {
|
||||||
|
return a.Type == "cookie"
|
||||||
|
}
|
||||||
|
|
||||||
|
var config = driver.Config{
|
||||||
|
Name: "Lanzou",
|
||||||
|
LocalSort: true,
|
||||||
|
DefaultRoot: "-1",
|
||||||
|
}
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
op.RegisterDriver(config, func() driver.Driver {
|
||||||
|
return &LanZou{}
|
||||||
|
})
|
||||||
|
}
|
131
drivers/lanzou/types.go
Normal file
131
drivers/lanzou/types.go
Normal file
@ -0,0 +1,131 @@
|
|||||||
|
package lanzou
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/alist-org/alist/v3/internal/model"
|
||||||
|
)
|
||||||
|
|
||||||
|
type FilesOrFoldersResp struct {
|
||||||
|
Text []FileOrFolder `json:"text"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type FileOrFolder struct {
|
||||||
|
Name string `json:"name"`
|
||||||
|
//Onof string `json:"onof"` // 是否存在提取码
|
||||||
|
//IsLock string `json:"is_lock"`
|
||||||
|
//IsCopyright int `json:"is_copyright"`
|
||||||
|
|
||||||
|
// 文件通用
|
||||||
|
ID string `json:"id"`
|
||||||
|
NameAll string `json:"name_all"`
|
||||||
|
Size string `json:"size"`
|
||||||
|
Time string `json:"time"`
|
||||||
|
//Icon string `json:"icon"`
|
||||||
|
//Downs string `json:"downs"`
|
||||||
|
//Filelock string `json:"filelock"`
|
||||||
|
//IsBakdownload int `json:"is_bakdownload"`
|
||||||
|
//Bakdownload string `json:"bakdownload"`
|
||||||
|
//IsDes int `json:"is_des"` // 是否存在描述
|
||||||
|
//IsIco int `json:"is_ico"`
|
||||||
|
|
||||||
|
// 文件夹
|
||||||
|
FolID string `json:"fol_id"`
|
||||||
|
//Folderlock string `json:"folderlock"`
|
||||||
|
//FolderDes string `json:"folder_des"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (f *FileOrFolder) isFloder() bool {
|
||||||
|
return f.FolID != ""
|
||||||
|
}
|
||||||
|
func (f *FileOrFolder) ToObj() model.Obj {
|
||||||
|
obj := &model.Object{}
|
||||||
|
if f.isFloder() {
|
||||||
|
obj.ID = f.FolID
|
||||||
|
obj.Name = f.Name
|
||||||
|
obj.Modified = time.Now()
|
||||||
|
obj.IsFolder = true
|
||||||
|
} else {
|
||||||
|
obj.ID = f.ID
|
||||||
|
obj.Name = f.NameAll
|
||||||
|
obj.Modified = MustParseTime(f.Time)
|
||||||
|
obj.Size = SizeStrToInt64(f.Size)
|
||||||
|
}
|
||||||
|
return obj
|
||||||
|
}
|
||||||
|
|
||||||
|
type FileShareResp struct {
|
||||||
|
Info FileShare `json:"info"`
|
||||||
|
}
|
||||||
|
type FileShare struct {
|
||||||
|
Pwd string `json:"pwd"`
|
||||||
|
Onof string `json:"onof"`
|
||||||
|
Taoc string `json:"taoc"`
|
||||||
|
IsNewd string `json:"is_newd"`
|
||||||
|
|
||||||
|
// 文件
|
||||||
|
FID string `json:"f_id"`
|
||||||
|
|
||||||
|
// 文件夹
|
||||||
|
NewUrl string `json:"new_url"`
|
||||||
|
Name string `json:"name"`
|
||||||
|
Des string `json:"des"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type FileOrFolderByShareUrlResp struct {
|
||||||
|
Text []FileOrFolderByShareUrl `json:"text"`
|
||||||
|
}
|
||||||
|
type FileOrFolderByShareUrl struct {
|
||||||
|
ID string `json:"id"`
|
||||||
|
NameAll string `json:"name_all"`
|
||||||
|
Size string `json:"size"`
|
||||||
|
Time string `json:"time"`
|
||||||
|
Duan string `json:"duan"`
|
||||||
|
//Icon string `json:"icon"`
|
||||||
|
//PIco int `json:"p_ico"`
|
||||||
|
//T int `json:"t"`
|
||||||
|
IsFloder bool
|
||||||
|
}
|
||||||
|
|
||||||
|
func (f *FileOrFolderByShareUrl) ToObj() model.Obj {
|
||||||
|
return &model.Object{
|
||||||
|
ID: f.ID,
|
||||||
|
Name: f.NameAll,
|
||||||
|
Size: SizeStrToInt64(f.Size),
|
||||||
|
Modified: MustParseTime(f.Time),
|
||||||
|
IsFolder: f.IsFloder,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
type FileShareInfoAndUrlResp[T string | int] struct {
|
||||||
|
Dom string `json:"dom"`
|
||||||
|
URL string `json:"url"`
|
||||||
|
Inf T `json:"inf"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (u *FileShareInfoAndUrlResp[T]) GetBaseUrl() string {
|
||||||
|
return fmt.Sprint(u.Dom, "/file")
|
||||||
|
}
|
||||||
|
|
||||||
|
func (u *FileShareInfoAndUrlResp[T]) GetDownloadUrl() string {
|
||||||
|
return fmt.Sprint(u.GetBaseUrl(), "/", u.URL)
|
||||||
|
}
|
||||||
|
|
||||||
|
// 通过分享链接获取文件信息和下载链接
|
||||||
|
type FileInfoAndUrlByShareUrl struct {
|
||||||
|
ID string
|
||||||
|
Name string
|
||||||
|
Size string
|
||||||
|
Time string
|
||||||
|
Url string
|
||||||
|
}
|
||||||
|
|
||||||
|
func (f *FileInfoAndUrlByShareUrl) ToObj() model.Obj {
|
||||||
|
return &model.Object{
|
||||||
|
ID: f.ID,
|
||||||
|
Name: f.Name,
|
||||||
|
Size: SizeStrToInt64(f.Size),
|
||||||
|
Modified: MustParseTime(f.Time),
|
||||||
|
}
|
||||||
|
}
|
416
drivers/lanzou/util.go
Normal file
416
drivers/lanzou/util.go
Normal file
@ -0,0 +1,416 @@
|
|||||||
|
package lanzou
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"fmt"
|
||||||
|
"net/http"
|
||||||
|
"regexp"
|
||||||
|
"strconv"
|
||||||
|
"strings"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/alist-org/alist/v3/drivers/base"
|
||||||
|
"github.com/alist-org/alist/v3/internal/model"
|
||||||
|
"github.com/alist-org/alist/v3/pkg/utils"
|
||||||
|
"github.com/go-resty/resty/v2"
|
||||||
|
)
|
||||||
|
|
||||||
|
func (d *LanZou) get(url string, callback base.ReqCallback, resp interface{}) ([]byte, error) {
|
||||||
|
return d.request(url, http.MethodGet, callback, false)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *LanZou) post(url string, callback base.ReqCallback, resp interface{}) ([]byte, error) {
|
||||||
|
return d._post(url, callback, resp, false)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *LanZou) _post(url string, callback base.ReqCallback, resp interface{}, up bool) ([]byte, error) {
|
||||||
|
data, err := d.request(url, http.MethodPost, callback, up)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
switch utils.Json.Get(data, "zt").ToInt() {
|
||||||
|
case 1, 2, 4:
|
||||||
|
if resp != nil {
|
||||||
|
// 返回类型不统一,忽略错误
|
||||||
|
utils.Json.Unmarshal(data, resp)
|
||||||
|
}
|
||||||
|
return data, nil
|
||||||
|
default:
|
||||||
|
info := utils.Json.Get(data, "inf").ToString()
|
||||||
|
if info == "" {
|
||||||
|
info = utils.Json.Get(data, "info").ToString()
|
||||||
|
}
|
||||||
|
return nil, fmt.Errorf(info)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *LanZou) request(url string, method string, callback base.ReqCallback, up bool) ([]byte, error) {
|
||||||
|
var req *resty.Request
|
||||||
|
if up {
|
||||||
|
req = upClient.R()
|
||||||
|
} else {
|
||||||
|
req = base.RestyClient.R()
|
||||||
|
}
|
||||||
|
|
||||||
|
req.SetHeaders(map[string]string{
|
||||||
|
"Referer": "https://pc.woozooo.com",
|
||||||
|
})
|
||||||
|
|
||||||
|
if d.Cookie != "" {
|
||||||
|
req.SetHeader("cookie", d.Cookie)
|
||||||
|
}
|
||||||
|
|
||||||
|
if callback != nil {
|
||||||
|
callback(req)
|
||||||
|
}
|
||||||
|
|
||||||
|
res, err := req.Execute(method, url)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
return res.Body(), err
|
||||||
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
通过cookie获取数据
|
||||||
|
*/
|
||||||
|
|
||||||
|
// 获取文件和文件夹,获取到的文件大小、更改时间不可信
|
||||||
|
func (d *LanZou) GetFiles(ctx context.Context, folderID string) ([]model.Obj, error) {
|
||||||
|
folders, err := d.getFolders(ctx, folderID)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
files, err := d.getFiles(ctx, folderID)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
objs := make([]model.Obj, 0, len(folders)+len(files))
|
||||||
|
for _, folder := range folders {
|
||||||
|
objs = append(objs, folder.ToObj())
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, file := range files {
|
||||||
|
objs = append(objs, file.ToObj())
|
||||||
|
}
|
||||||
|
return objs, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// 通过ID获取文件夹
|
||||||
|
func (d *LanZou) getFolders(ctx context.Context, folderID string) ([]FileOrFolder, error) {
|
||||||
|
var resp FilesOrFoldersResp
|
||||||
|
_, err := d.post(d.BaseUrl+"/doupload.php", func(req *resty.Request) {
|
||||||
|
req.SetContext(ctx)
|
||||||
|
req.SetFormData(map[string]string{
|
||||||
|
"task": "47",
|
||||||
|
"folder_id": folderID,
|
||||||
|
})
|
||||||
|
}, &resp)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return resp.Text, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// 通过ID获取文件
|
||||||
|
func (d *LanZou) getFiles(ctx context.Context, folderID string) ([]FileOrFolder, error) {
|
||||||
|
files := make([]FileOrFolder, 0)
|
||||||
|
for pg := 1; ; pg++ {
|
||||||
|
var resp FilesOrFoldersResp
|
||||||
|
_, err := d.post(d.BaseUrl+"/doupload.php", func(req *resty.Request) {
|
||||||
|
req.SetContext(ctx)
|
||||||
|
req.SetFormData(map[string]string{
|
||||||
|
"task": "5",
|
||||||
|
"folder_id": folderID,
|
||||||
|
"pg": strconv.Itoa(pg),
|
||||||
|
})
|
||||||
|
}, &resp)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if len(resp.Text) == 0 {
|
||||||
|
break
|
||||||
|
}
|
||||||
|
files = append(files, resp.Text...)
|
||||||
|
}
|
||||||
|
return files, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// 通过ID获取文件夹分享地址
|
||||||
|
func (d *LanZou) getFolderShareUrlByID(ctx context.Context, fileID string) (share FileShare, err error) {
|
||||||
|
var resp FileShareResp
|
||||||
|
_, err = d.post(d.BaseUrl+"/doupload.php", func(req *resty.Request) {
|
||||||
|
req.SetContext(ctx)
|
||||||
|
req.SetFormData(map[string]string{
|
||||||
|
"task": "18",
|
||||||
|
"file_id": fileID,
|
||||||
|
})
|
||||||
|
}, &resp)
|
||||||
|
if err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
share = resp.Info
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// 通过ID获取文件分享地址
|
||||||
|
func (d *LanZou) getFileShareUrlByID(ctx context.Context, fileID string) (share FileShare, err error) {
|
||||||
|
var resp FileShareResp
|
||||||
|
_, err = d.post(d.BaseUrl+"/doupload.php", func(req *resty.Request) {
|
||||||
|
req.SetContext(ctx)
|
||||||
|
req.SetFormData(map[string]string{
|
||||||
|
"task": "22",
|
||||||
|
"file_id": fileID,
|
||||||
|
})
|
||||||
|
}, &resp)
|
||||||
|
if err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
share = resp.Info
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
通过分享链接获取数据
|
||||||
|
*/
|
||||||
|
|
||||||
|
// 判断类容
|
||||||
|
var isFileReg = regexp.MustCompile(`class="fileinfo"|id="file"|文件描述`)
|
||||||
|
var isFolderReg = regexp.MustCompile(`id="infos"`)
|
||||||
|
|
||||||
|
// 获取文件文件夹基础信息
|
||||||
|
var nameFindReg = regexp.MustCompile(`<title>(.+?) - 蓝奏云</title>|id="filenajax">(.+?)</div>|var filename = '(.+?)';|<div style="font-size.+?>([^<>].+?)</div>|<div class="filethetext".+?>([^<>]+?)</div>`)
|
||||||
|
var sizeFindReg = regexp.MustCompile(`(?i)大小\W*([0-9.]+\s*[bkm]+)`)
|
||||||
|
var timeFindReg = regexp.MustCompile(`\d+\s*[秒天分小][钟时]?前|[昨前]天|\d{4}-\d{2}-\d{2}`)
|
||||||
|
|
||||||
|
var findSubFolaerReg = regexp.MustCompile(`(folderlink|mbxfolder).+href="/(.+?)"(.+filename")?>(.+?)<`) // 查找分享文件夹子文件夹ID和名称
|
||||||
|
|
||||||
|
// 获取关键数据
|
||||||
|
var findDownPageParamReg = regexp.MustCompile(`<iframe.*?src="(.+?)"`)
|
||||||
|
|
||||||
|
// 通过分享链接获取文件或文件夹,如果是文件则会返回下载链接
|
||||||
|
func (d *LanZou) GetFileOrFolderByShareUrl(ctx context.Context, downID, pwd string) ([]model.Obj, error) {
|
||||||
|
pageData, err := d.get(fmt.Sprint(d.ShareUrl, "/", downID), func(req *resty.Request) { req.SetContext(ctx) }, nil)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
pageData = RemoveNotes(pageData)
|
||||||
|
|
||||||
|
var objs []model.Obj
|
||||||
|
if !isFileReg.Match(pageData) {
|
||||||
|
files, err := d.getFolderByShareUrl(ctx, downID, pwd, pageData)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
objs = make([]model.Obj, 0, len(files))
|
||||||
|
for _, file := range files {
|
||||||
|
objs = append(objs, file.ToObj())
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
file, err := d.getFilesByShareUrl(ctx, downID, pwd, pageData)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
objs = []model.Obj{file.ToObj()}
|
||||||
|
}
|
||||||
|
return objs, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// 通过分享链接获取文件(下载链接也使用此方法)
|
||||||
|
// 参考 https://github.com/zaxtyson/LanZouCloud-API/blob/ab2e9ec715d1919bf432210fc16b91c6775fbb99/lanzou/api/core.py#L440
|
||||||
|
func (d *LanZou) getFilesByShareUrl(ctx context.Context, downID, pwd string, firstPageData []byte) (file FileInfoAndUrlByShareUrl, err error) {
|
||||||
|
if firstPageData == nil {
|
||||||
|
firstPageData, err = d.get(fmt.Sprint(d.ShareUrl, "/", downID), func(req *resty.Request) { req.SetContext(ctx) }, nil)
|
||||||
|
if err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
firstPageData = RemoveNotes(firstPageData)
|
||||||
|
}
|
||||||
|
firstPageDataStr := string(firstPageData)
|
||||||
|
|
||||||
|
if strings.Contains(firstPageDataStr, "acw_sc__v2") {
|
||||||
|
var vs string
|
||||||
|
if vs, err = CalcAcwScV2(firstPageDataStr); err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
firstPageData, err = d.get(fmt.Sprint(d.ShareUrl, "/", downID), func(req *resty.Request) {
|
||||||
|
req.SetCookie(&http.Cookie{
|
||||||
|
Name: "acw_sc__v2",
|
||||||
|
Value: vs,
|
||||||
|
})
|
||||||
|
req.SetContext(ctx)
|
||||||
|
}, nil)
|
||||||
|
if err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
firstPageData = RemoveNotes(firstPageData)
|
||||||
|
firstPageDataStr = string(firstPageData)
|
||||||
|
}
|
||||||
|
|
||||||
|
var (
|
||||||
|
param map[string]string
|
||||||
|
downloadUrl string
|
||||||
|
baseUrl string
|
||||||
|
)
|
||||||
|
|
||||||
|
// 需要密码
|
||||||
|
if strings.Contains(firstPageDataStr, "pwdload") || strings.Contains(firstPageDataStr, "passwddiv") {
|
||||||
|
param, err = htmlFormToMap(firstPageDataStr)
|
||||||
|
if err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
param["p"] = pwd
|
||||||
|
var resp FileShareInfoAndUrlResp[string]
|
||||||
|
_, err = d.post(d.ShareUrl+"/ajaxm.php", func(req *resty.Request) { req.SetFormData(param).SetContext(ctx) }, &resp)
|
||||||
|
if err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
file.Name = resp.Inf
|
||||||
|
baseUrl = resp.GetBaseUrl()
|
||||||
|
downloadUrl = resp.GetDownloadUrl()
|
||||||
|
} else {
|
||||||
|
urlpaths := findDownPageParamReg.FindStringSubmatch(firstPageDataStr)
|
||||||
|
if len(urlpaths) != 2 {
|
||||||
|
err = fmt.Errorf("not find file page param")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
var nextPageData []byte
|
||||||
|
nextPageData, err = d.get(fmt.Sprint(d.ShareUrl, urlpaths[1]), func(req *resty.Request) { req.SetContext(ctx) }, nil)
|
||||||
|
if err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
nextPageData = RemoveNotes(nextPageData)
|
||||||
|
nextPageDataStr := string(nextPageData)
|
||||||
|
|
||||||
|
param, err = htmlJsonToMap(nextPageDataStr)
|
||||||
|
if err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
var resp FileShareInfoAndUrlResp[int]
|
||||||
|
_, err = d.post(d.ShareUrl+"/ajaxm.php", func(req *resty.Request) { req.SetFormData(param).SetContext(ctx) }, &resp)
|
||||||
|
if err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
baseUrl = resp.GetBaseUrl()
|
||||||
|
downloadUrl = resp.GetDownloadUrl()
|
||||||
|
|
||||||
|
names := nameFindReg.FindStringSubmatch(firstPageDataStr)
|
||||||
|
if len(names) > 1 {
|
||||||
|
for _, name := range names[1:] {
|
||||||
|
if name != "" {
|
||||||
|
file.Name = name
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
sizes := sizeFindReg.FindStringSubmatch(firstPageDataStr)
|
||||||
|
if len(sizes) == 2 {
|
||||||
|
file.Size = sizes[1]
|
||||||
|
}
|
||||||
|
file.ID = downID
|
||||||
|
file.Time = timeFindReg.FindString(firstPageDataStr)
|
||||||
|
|
||||||
|
// 重定向获取真实链接
|
||||||
|
res, err := base.NoRedirectClient.R().SetHeaders(map[string]string{
|
||||||
|
"accept-language": "zh-CN,zh;q=0.9,en;q=0.8,en-GB;q=0.7,en-US;q=0.6",
|
||||||
|
}).SetContext(ctx).Get(downloadUrl)
|
||||||
|
if err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
file.Url = res.Header().Get("location")
|
||||||
|
|
||||||
|
// 触发验证
|
||||||
|
rPageDataStr := res.String()
|
||||||
|
if res.StatusCode() != 302 && strings.Contains(rPageDataStr, "网络异常") {
|
||||||
|
param, err = htmlJsonToMap(rPageDataStr)
|
||||||
|
if err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
param["el"] = "2"
|
||||||
|
time.Sleep(time.Second * 2)
|
||||||
|
|
||||||
|
// 通过验证获取直连
|
||||||
|
var rUrl struct {
|
||||||
|
Url string `json:"url"`
|
||||||
|
}
|
||||||
|
_, err = d.post(fmt.Sprint(baseUrl, "/ajax.php"), func(req *resty.Request) { req.SetContext(ctx).SetFormData(param) }, &rUrl)
|
||||||
|
if err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
file.Url = rUrl.Url
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// 通过分享链接获取文件夹
|
||||||
|
// 参考 https://github.com/zaxtyson/LanZouCloud-API/blob/ab2e9ec715d1919bf432210fc16b91c6775fbb99/lanzou/api/core.py#L1089
|
||||||
|
func (d *LanZou) getFolderByShareUrl(ctx context.Context, downID, pwd string, firstPageData []byte) ([]FileOrFolderByShareUrl, error) {
|
||||||
|
if firstPageData == nil {
|
||||||
|
var err error
|
||||||
|
firstPageData, err = d.get(fmt.Sprint(d.ShareUrl, "/", downID), func(req *resty.Request) { req.SetContext(ctx) }, nil)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
firstPageData = RemoveNotes(firstPageData)
|
||||||
|
}
|
||||||
|
firstPageDataStr := string(firstPageData)
|
||||||
|
|
||||||
|
//
|
||||||
|
if strings.Contains(firstPageDataStr, "acw_sc__v2") {
|
||||||
|
vs, err := CalcAcwScV2(firstPageDataStr)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
firstPageData, err = d.get(fmt.Sprint(d.ShareUrl, "/", downID), func(req *resty.Request) {
|
||||||
|
req.SetCookie(&http.Cookie{
|
||||||
|
Name: "acw_sc__v2",
|
||||||
|
Value: vs,
|
||||||
|
})
|
||||||
|
req.SetContext(ctx)
|
||||||
|
}, nil)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
firstPageData = RemoveNotes(firstPageData)
|
||||||
|
firstPageDataStr = string(firstPageData)
|
||||||
|
}
|
||||||
|
|
||||||
|
from, err := htmlJsonToMap(firstPageDataStr)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
from["pwd"] = pwd
|
||||||
|
|
||||||
|
files := make([]FileOrFolderByShareUrl, 0)
|
||||||
|
// vip获取文件夹
|
||||||
|
floders := findSubFolaerReg.FindAllStringSubmatch(firstPageDataStr, -1)
|
||||||
|
for _, floder := range floders {
|
||||||
|
if len(floder) == 5 {
|
||||||
|
files = append(files, FileOrFolderByShareUrl{
|
||||||
|
ID: floder[2],
|
||||||
|
NameAll: floder[4],
|
||||||
|
IsFloder: true,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
for page := 1; ; page++ {
|
||||||
|
from["pg"] = strconv.Itoa(page)
|
||||||
|
var resp FileOrFolderByShareUrlResp
|
||||||
|
_, err := d.post(d.ShareUrl+"/filemoreajax.php", func(req *resty.Request) { req.SetFormData(from).SetContext(ctx) }, &resp)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
files = append(files, resp.Text...)
|
||||||
|
if len(resp.Text) == 0 {
|
||||||
|
break
|
||||||
|
}
|
||||||
|
time.Sleep(time.Millisecond * 600)
|
||||||
|
}
|
||||||
|
return files, nil
|
||||||
|
}
|
@ -68,7 +68,7 @@ func (d *Local) List(ctx context.Context, dir model.Obj, args model.ListArgs) ([
|
|||||||
}
|
}
|
||||||
var files []model.Obj
|
var files []model.Obj
|
||||||
for _, f := range rawFiles {
|
for _, f := range rawFiles {
|
||||||
if strings.HasPrefix(f.Name(), ".") {
|
if !d.ShowHidden && strings.HasPrefix(f.Name(), ".") {
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
thumb := ""
|
thumb := ""
|
||||||
@ -77,12 +77,17 @@ func (d *Local) List(ctx context.Context, dir model.Obj, args model.ListArgs) ([
|
|||||||
thumb = utils.EncodePath(thumb, true)
|
thumb = utils.EncodePath(thumb, true)
|
||||||
thumb += "?type=thumb"
|
thumb += "?type=thumb"
|
||||||
}
|
}
|
||||||
|
isFolder := f.IsDir() || isSymlinkDir(f, fullPath)
|
||||||
|
size := f.Size()
|
||||||
|
if isFolder {
|
||||||
|
size = 0
|
||||||
|
}
|
||||||
file := model.ObjThumb{
|
file := model.ObjThumb{
|
||||||
Object: model.Object{
|
Object: model.Object{
|
||||||
Name: f.Name(),
|
Name: f.Name(),
|
||||||
Modified: f.ModTime(),
|
Modified: f.ModTime(),
|
||||||
Size: f.Size(),
|
Size: size,
|
||||||
IsFolder: f.IsDir(),
|
IsFolder: isFolder,
|
||||||
},
|
},
|
||||||
Thumbnail: model.Thumbnail{
|
Thumbnail: model.Thumbnail{
|
||||||
Thumbnail: thumb,
|
Thumbnail: thumb,
|
||||||
@ -101,12 +106,17 @@ func (d *Local) Get(ctx context.Context, path string) (model.Obj, error) {
|
|||||||
}
|
}
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
isFolder := f.IsDir() || isSymlinkDir(f, path)
|
||||||
|
size := f.Size()
|
||||||
|
if isFolder {
|
||||||
|
size = 0
|
||||||
|
}
|
||||||
file := model.Object{
|
file := model.Object{
|
||||||
Path: path,
|
Path: path,
|
||||||
Name: f.Name(),
|
Name: f.Name(),
|
||||||
Modified: f.ModTime(),
|
Modified: f.ModTime(),
|
||||||
Size: f.Size(),
|
Size: size,
|
||||||
IsFolder: f.IsDir(),
|
IsFolder: isFolder,
|
||||||
}
|
}
|
||||||
return &file, nil
|
return &file, nil
|
||||||
}
|
}
|
||||||
@ -175,9 +185,9 @@ func (d *Local) Copy(ctx context.Context, srcObj, dstDir model.Obj) error {
|
|||||||
dstPath := filepath.Join(dstDir.GetPath(), srcObj.GetName())
|
dstPath := filepath.Join(dstDir.GetPath(), srcObj.GetName())
|
||||||
var err error
|
var err error
|
||||||
if srcObj.IsDir() {
|
if srcObj.IsDir() {
|
||||||
err = copyDir(srcPath, dstPath)
|
err = utils.CopyDir(srcPath, dstPath)
|
||||||
} else {
|
} else {
|
||||||
err = copyFile(srcPath, dstPath)
|
err = utils.CopyFile(srcPath, dstPath)
|
||||||
}
|
}
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
@ -217,8 +227,4 @@ func (d *Local) Put(ctx context.Context, dstDir model.Obj, stream model.FileStre
|
|||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (d *Local) Other(ctx context.Context, args model.OtherArgs) (interface{}, error) {
|
|
||||||
return nil, errs.NotSupport
|
|
||||||
}
|
|
||||||
|
|
||||||
var _ driver.Driver = (*Local)(nil)
|
var _ driver.Driver = (*Local)(nil)
|
||||||
|
@ -7,7 +7,8 @@ import (
|
|||||||
|
|
||||||
type Addition struct {
|
type Addition struct {
|
||||||
driver.RootPath
|
driver.RootPath
|
||||||
Thumbnail bool `json:"thumbnail" required:"true" help:"enable thumbnail"`
|
Thumbnail bool `json:"thumbnail" required:"true" help:"enable thumbnail"`
|
||||||
|
ShowHidden bool `json:"show_hidden" default:"true" required:"false" help:"show hidden directories and files"`
|
||||||
}
|
}
|
||||||
|
|
||||||
var config = driver.Config{
|
var config = driver.Config{
|
||||||
|
@ -1,67 +1,25 @@
|
|||||||
package local
|
package local
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"io/fs"
|
||||||
"io"
|
|
||||||
"io/ioutil"
|
|
||||||
"os"
|
"os"
|
||||||
"path"
|
"path/filepath"
|
||||||
)
|
)
|
||||||
|
|
||||||
// copyFile File copies a single file from src to dst
|
func isSymlinkDir(f fs.FileInfo, path string) bool {
|
||||||
func copyFile(src, dst string) error {
|
if f.Mode()&os.ModeSymlink == os.ModeSymlink {
|
||||||
var err error
|
dst, err := os.Readlink(filepath.Join(path, f.Name()))
|
||||||
var srcfd *os.File
|
if err != nil {
|
||||||
var dstfd *os.File
|
return false
|
||||||
var srcinfo os.FileInfo
|
|
||||||
|
|
||||||
if srcfd, err = os.Open(src); err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
defer srcfd.Close()
|
|
||||||
|
|
||||||
if dstfd, err = os.Create(dst); err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
defer dstfd.Close()
|
|
||||||
|
|
||||||
if _, err = io.Copy(dstfd, srcfd); err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
if srcinfo, err = os.Stat(src); err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
return os.Chmod(dst, srcinfo.Mode())
|
|
||||||
}
|
|
||||||
|
|
||||||
// copyDir Dir copies a whole directory recursively
|
|
||||||
func copyDir(src string, dst string) error {
|
|
||||||
var err error
|
|
||||||
var fds []os.FileInfo
|
|
||||||
var srcinfo os.FileInfo
|
|
||||||
|
|
||||||
if srcinfo, err = os.Stat(src); err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
if err = os.MkdirAll(dst, srcinfo.Mode()); err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
if fds, err = ioutil.ReadDir(src); err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
for _, fd := range fds {
|
|
||||||
srcfp := path.Join(src, fd.Name())
|
|
||||||
dstfp := path.Join(dst, fd.Name())
|
|
||||||
|
|
||||||
if fd.IsDir() {
|
|
||||||
if err = copyDir(srcfp, dstfp); err != nil {
|
|
||||||
fmt.Println(err)
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
if err = copyFile(srcfp, dstfp); err != nil {
|
|
||||||
fmt.Println(err)
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
if !filepath.IsAbs(dst) {
|
||||||
|
dst = filepath.Join(path, dst)
|
||||||
|
}
|
||||||
|
stat, err := os.Stat(dst)
|
||||||
|
if err != nil {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
return stat.IsDir()
|
||||||
}
|
}
|
||||||
return nil
|
return false
|
||||||
}
|
}
|
||||||
|
@ -10,11 +10,10 @@ import (
|
|||||||
"os"
|
"os"
|
||||||
"path"
|
"path"
|
||||||
"strconv"
|
"strconv"
|
||||||
|
"time"
|
||||||
|
|
||||||
"github.com/alist-org/alist/v3/drivers/base"
|
"github.com/alist-org/alist/v3/drivers/base"
|
||||||
"github.com/alist-org/alist/v3/internal/conf"
|
|
||||||
"github.com/alist-org/alist/v3/internal/driver"
|
"github.com/alist-org/alist/v3/internal/driver"
|
||||||
"github.com/alist-org/alist/v3/internal/errs"
|
|
||||||
"github.com/alist-org/alist/v3/internal/model"
|
"github.com/alist-org/alist/v3/internal/model"
|
||||||
"github.com/alist-org/alist/v3/internal/op"
|
"github.com/alist-org/alist/v3/internal/op"
|
||||||
"github.com/alist-org/alist/v3/pkg/utils"
|
"github.com/alist-org/alist/v3/pkg/utils"
|
||||||
@ -93,7 +92,21 @@ func (d *MediaTrack) Link(ctx context.Context, file model.Obj, args model.LinkAr
|
|||||||
}
|
}
|
||||||
token := utils.Json.Get(body, "data", "token").ToString()
|
token := utils.Json.Get(body, "data", "token").ToString()
|
||||||
url = "https://kayn.api.mediatrack.cn/v1/download/redirect?token=" + token
|
url = "https://kayn.api.mediatrack.cn/v1/download/redirect?token=" + token
|
||||||
return &model.Link{URL: url}, nil
|
res, err := base.NoRedirectClient.R().Get(url)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
log.Debug(res.String())
|
||||||
|
link := model.Link{
|
||||||
|
URL: url,
|
||||||
|
}
|
||||||
|
log.Debugln("res code: ", res.StatusCode())
|
||||||
|
if res.StatusCode() == 302 {
|
||||||
|
link.URL = res.Header().Get("location")
|
||||||
|
expired := time.Duration(60) * time.Second
|
||||||
|
link.Expiration = &expired
|
||||||
|
}
|
||||||
|
return &link, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (d *MediaTrack) MakeDir(ctx context.Context, parentDir model.Obj, dirName string) error {
|
func (d *MediaTrack) MakeDir(ctx context.Context, parentDir model.Obj, dirName string) error {
|
||||||
@ -177,7 +190,7 @@ func (d *MediaTrack) Put(ctx context.Context, dstDir model.Obj, stream model.Fil
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
tempFile, err := os.CreateTemp(conf.Conf.TempDir, "file-*")
|
tempFile, err := utils.CreateTempFile(stream.GetReadCloser())
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
@ -185,14 +198,6 @@ func (d *MediaTrack) Put(ctx context.Context, dstDir model.Obj, stream model.Fil
|
|||||||
_ = tempFile.Close()
|
_ = tempFile.Close()
|
||||||
_ = os.Remove(tempFile.Name())
|
_ = os.Remove(tempFile.Name())
|
||||||
}()
|
}()
|
||||||
_, err = io.Copy(tempFile, stream)
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
_, err = tempFile.Seek(0, io.SeekStart)
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
uploader := s3manager.NewUploader(s)
|
uploader := s3manager.NewUploader(s)
|
||||||
input := &s3manager.UploadInput{
|
input := &s3manager.UploadInput{
|
||||||
Bucket: &resp.Data.Bucket,
|
Bucket: &resp.Data.Bucket,
|
||||||
@ -230,8 +235,4 @@ func (d *MediaTrack) Put(ctx context.Context, dstDir model.Obj, stream model.Fil
|
|||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
func (d *MediaTrack) Other(ctx context.Context, args model.OtherArgs) (interface{}, error) {
|
|
||||||
return nil, errs.NotSupport
|
|
||||||
}
|
|
||||||
|
|
||||||
var _ driver.Driver = (*MediaTrack)(nil)
|
var _ driver.Driver = (*MediaTrack)(nil)
|
||||||
|
192
drivers/mega/driver.go
Normal file
192
drivers/mega/driver.go
Normal file
@ -0,0 +1,192 @@
|
|||||||
|
package mega
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
"io"
|
||||||
|
|
||||||
|
"github.com/alist-org/alist/v3/internal/driver"
|
||||||
|
"github.com/alist-org/alist/v3/internal/errs"
|
||||||
|
"github.com/alist-org/alist/v3/internal/model"
|
||||||
|
"github.com/alist-org/alist/v3/pkg/chanio"
|
||||||
|
"github.com/alist-org/alist/v3/pkg/utils"
|
||||||
|
log "github.com/sirupsen/logrus"
|
||||||
|
"github.com/t3rm1n4l/go-mega"
|
||||||
|
)
|
||||||
|
|
||||||
|
type Mega struct {
|
||||||
|
model.Storage
|
||||||
|
Addition
|
||||||
|
c *mega.Mega
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *Mega) Config() driver.Config {
|
||||||
|
return config
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *Mega) GetAddition() driver.Additional {
|
||||||
|
return d.Addition
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *Mega) Init(ctx context.Context, storage model.Storage) error {
|
||||||
|
d.Storage = storage
|
||||||
|
err := utils.Json.UnmarshalFromString(d.Storage.Addition, &d.Addition)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
d.c = mega.New()
|
||||||
|
return d.c.Login(d.Email, d.Password)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *Mega) Drop(ctx context.Context) error {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *Mega) List(ctx context.Context, dir model.Obj, args model.ListArgs) ([]model.Obj, error) {
|
||||||
|
if node, ok := dir.(*MegaNode); ok {
|
||||||
|
nodes, err := d.c.FS.GetChildren(node.Node)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
res := make([]model.Obj, 0)
|
||||||
|
for i := range nodes {
|
||||||
|
n := nodes[i]
|
||||||
|
if n.GetType() == mega.FILE || n.GetType() == mega.FOLDER {
|
||||||
|
res = append(res, &MegaNode{n})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return res, nil
|
||||||
|
}
|
||||||
|
log.Errorf("can't convert: %+v", dir)
|
||||||
|
return nil, fmt.Errorf("unable to convert dir to mega node")
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *Mega) Get(ctx context.Context, path string) (model.Obj, error) {
|
||||||
|
if path == "/" {
|
||||||
|
n := d.c.FS.GetRoot()
|
||||||
|
log.Debugf("mega root: %+v", *n)
|
||||||
|
return &MegaNode{n}, nil
|
||||||
|
}
|
||||||
|
return nil, errs.NotSupport
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *Mega) Link(ctx context.Context, file model.Obj, args model.LinkArgs) (*model.Link, error) {
|
||||||
|
if node, ok := file.(*MegaNode); ok {
|
||||||
|
//link, err := d.c.Link(node.Node, true)
|
||||||
|
//if err != nil {
|
||||||
|
// return nil, err
|
||||||
|
//}
|
||||||
|
//return &model.Link{URL: link}, nil
|
||||||
|
down, err := d.c.NewDownload(node.Node)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
//u := down.GetResourceUrl()
|
||||||
|
//u = strings.Replace(u, "http", "https", 1)
|
||||||
|
//return &model.Link{URL: u}, nil
|
||||||
|
c := chanio.New()
|
||||||
|
go func() {
|
||||||
|
defer func() {
|
||||||
|
_ = recover()
|
||||||
|
}()
|
||||||
|
log.Debugf("chunk size: %d", down.Chunks())
|
||||||
|
for id := 0; id < down.Chunks(); id++ {
|
||||||
|
chunk, err := down.DownloadChunk(id)
|
||||||
|
if err != nil {
|
||||||
|
log.Errorf("mega down: %+v", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
log.Debugf("id: %d,len: %d", id, len(chunk))
|
||||||
|
//_, _, err = down.ChunkLocation(id)
|
||||||
|
//if err != nil {
|
||||||
|
// log.Errorf("mega down: %+v", err)
|
||||||
|
// return
|
||||||
|
//}
|
||||||
|
//_, err = c.Write(chunk)
|
||||||
|
_, err = c.Write(chunk)
|
||||||
|
}
|
||||||
|
err := c.Close()
|
||||||
|
if err != nil {
|
||||||
|
log.Errorf("mega down: %+v", err)
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
return &model.Link{Data: c}, nil
|
||||||
|
}
|
||||||
|
return nil, fmt.Errorf("unable to convert dir to mega node")
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *Mega) MakeDir(ctx context.Context, parentDir model.Obj, dirName string) error {
|
||||||
|
if parentNode, ok := parentDir.(*MegaNode); ok {
|
||||||
|
_, err := d.c.CreateDir(dirName, parentNode.Node)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
return fmt.Errorf("unable to convert dir to mega node")
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *Mega) Move(ctx context.Context, srcObj, dstDir model.Obj) error {
|
||||||
|
if srcNode, ok := srcObj.(*MegaNode); ok {
|
||||||
|
if dstNode, ok := dstDir.(*MegaNode); ok {
|
||||||
|
return d.c.Move(srcNode.Node, dstNode.Node)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return fmt.Errorf("unable to convert dir to mega node")
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *Mega) Rename(ctx context.Context, srcObj model.Obj, newName string) error {
|
||||||
|
if srcNode, ok := srcObj.(*MegaNode); ok {
|
||||||
|
return d.c.Rename(srcNode.Node, newName)
|
||||||
|
}
|
||||||
|
return fmt.Errorf("unable to convert dir to mega node")
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *Mega) Copy(ctx context.Context, srcObj, dstDir model.Obj) error {
|
||||||
|
return errs.NotImplement
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *Mega) Remove(ctx context.Context, obj model.Obj) error {
|
||||||
|
if node, ok := obj.(*MegaNode); ok {
|
||||||
|
return d.c.Delete(node.Node, false)
|
||||||
|
}
|
||||||
|
return fmt.Errorf("unable to convert dir to mega node")
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *Mega) Put(ctx context.Context, dstDir model.Obj, stream model.FileStreamer, up driver.UpdateProgress) error {
|
||||||
|
if dstNode, ok := dstDir.(*MegaNode); ok {
|
||||||
|
u, err := d.c.NewUpload(dstNode.Node, stream.GetName(), stream.GetSize())
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
for id := 0; id < u.Chunks(); id++ {
|
||||||
|
_, chkSize, err := u.ChunkLocation(id)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
chunk := make([]byte, chkSize)
|
||||||
|
n, err := io.ReadFull(stream, chunk)
|
||||||
|
if err != nil && err != io.EOF {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if n != len(chunk) {
|
||||||
|
return errors.New("chunk too short")
|
||||||
|
}
|
||||||
|
|
||||||
|
err = u.UploadChunk(id, chunk)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
up(id * 100 / u.Chunks())
|
||||||
|
}
|
||||||
|
|
||||||
|
_, err = u.Finish()
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
return fmt.Errorf("unable to convert dir to mega node")
|
||||||
|
}
|
||||||
|
|
||||||
|
//func (d *Mega) Other(ctx context.Context, args model.OtherArgs) (interface{}, error) {
|
||||||
|
// return nil, errs.NotSupport
|
||||||
|
//}
|
||||||
|
|
||||||
|
var _ driver.Driver = (*Mega)(nil)
|
26
drivers/mega/meta.go
Normal file
26
drivers/mega/meta.go
Normal file
@ -0,0 +1,26 @@
|
|||||||
|
package mega
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/alist-org/alist/v3/internal/driver"
|
||||||
|
"github.com/alist-org/alist/v3/internal/op"
|
||||||
|
)
|
||||||
|
|
||||||
|
type Addition struct {
|
||||||
|
// Usually one of two
|
||||||
|
//driver.RootPath
|
||||||
|
//driver.RootID
|
||||||
|
Email string `json:"email" required:"true"`
|
||||||
|
Password string `json:"password" required:"true"`
|
||||||
|
}
|
||||||
|
|
||||||
|
var config = driver.Config{
|
||||||
|
Name: "Mega_nz",
|
||||||
|
LocalSort: true,
|
||||||
|
OnlyLocal: true,
|
||||||
|
}
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
op.RegisterDriver(config, func() driver.Driver {
|
||||||
|
return &Mega{}
|
||||||
|
})
|
||||||
|
}
|
40
drivers/mega/types.go
Normal file
40
drivers/mega/types.go
Normal file
@ -0,0 +1,40 @@
|
|||||||
|
package mega
|
||||||
|
|
||||||
|
import (
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/alist-org/alist/v3/internal/model"
|
||||||
|
"github.com/t3rm1n4l/go-mega"
|
||||||
|
)
|
||||||
|
|
||||||
|
type MegaNode struct {
|
||||||
|
*mega.Node
|
||||||
|
}
|
||||||
|
|
||||||
|
//func (m *MegaNode) GetSize() int64 {
|
||||||
|
// //TODO implement me
|
||||||
|
// panic("implement me")
|
||||||
|
//}
|
||||||
|
//
|
||||||
|
//func (m *MegaNode) GetName() string {
|
||||||
|
// //TODO implement me
|
||||||
|
// panic("implement me")
|
||||||
|
//}
|
||||||
|
|
||||||
|
func (m *MegaNode) ModTime() time.Time {
|
||||||
|
return m.GetTimeStamp()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *MegaNode) IsDir() bool {
|
||||||
|
return m.GetType() == mega.FOLDER || m.GetType() == mega.ROOT
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *MegaNode) GetID() string {
|
||||||
|
return m.GetHash()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *MegaNode) GetPath() string {
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
|
||||||
|
var _ model.Obj = (*MegaNode)(nil)
|
3
drivers/mega/util.go
Normal file
3
drivers/mega/util.go
Normal file
@ -0,0 +1,3 @@
|
|||||||
|
package mega
|
||||||
|
|
||||||
|
// do others that not defined in Driver interface
|
@ -34,6 +34,9 @@ func (d *Onedrive) Init(ctx context.Context, storage model.Storage) error {
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
if d.ChunkSize < 1 {
|
||||||
|
d.ChunkSize = 5
|
||||||
|
}
|
||||||
return d.refreshToken()
|
return d.refreshToken()
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -144,8 +147,4 @@ func (d *Onedrive) Put(ctx context.Context, dstDir model.Obj, stream model.FileS
|
|||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
func (d *Onedrive) Other(ctx context.Context, args model.OtherArgs) (interface{}, error) {
|
|
||||||
return nil, errs.NotSupport
|
|
||||||
}
|
|
||||||
|
|
||||||
var _ driver.Driver = (*Onedrive)(nil)
|
var _ driver.Driver = (*Onedrive)(nil)
|
||||||
|
@ -7,13 +7,14 @@ import (
|
|||||||
|
|
||||||
type Addition struct {
|
type Addition struct {
|
||||||
driver.RootPath
|
driver.RootPath
|
||||||
Region string `json:"region" type:"select" required:"true" options:"global,cn,us,de"`
|
Region string `json:"region" type:"select" required:"true" options:"global,cn,us,de" default:"global"`
|
||||||
IsSharepoint bool `json:"is_sharepoint"`
|
IsSharepoint bool `json:"is_sharepoint"`
|
||||||
ClientID string `json:"client_id" required:"true"`
|
ClientID string `json:"client_id" required:"true"`
|
||||||
ClientSecret string `json:"client_secret" required:"true"`
|
ClientSecret string `json:"client_secret" required:"true"`
|
||||||
RedirectUri string `json:"redirect_uri" required:"true" default:"https://tool.nn.ci/onedrive/callback"`
|
RedirectUri string `json:"redirect_uri" required:"true" default:"https://tool.nn.ci/onedrive/callback"`
|
||||||
RefreshToken string `json:"refresh_token" required:"true"`
|
RefreshToken string `json:"refresh_token" required:"true"`
|
||||||
SiteId string `json:"site_id"`
|
SiteId string `json:"site_id"`
|
||||||
|
ChunkSize int64 `json:"chunk_size" type:"number" default:"5"`
|
||||||
}
|
}
|
||||||
|
|
||||||
var config = driver.Config{
|
var config = driver.Config{
|
||||||
|
@ -42,6 +42,7 @@ var onedriveHostMap = map[string]Host{
|
|||||||
|
|
||||||
func (d *Onedrive) GetMetaUrl(auth bool, path string) string {
|
func (d *Onedrive) GetMetaUrl(auth bool, path string) string {
|
||||||
host, _ := onedriveHostMap[d.Region]
|
host, _ := onedriveHostMap[d.Region]
|
||||||
|
path = utils.EncodePath(path, true)
|
||||||
if auth {
|
if auth {
|
||||||
return host.Oauth
|
return host.Oauth
|
||||||
}
|
}
|
||||||
@ -166,7 +167,7 @@ func (d *Onedrive) upBig(ctx context.Context, dstDir model.Obj, stream model.Fil
|
|||||||
}
|
}
|
||||||
uploadUrl := jsoniter.Get(res, "uploadUrl").ToString()
|
uploadUrl := jsoniter.Get(res, "uploadUrl").ToString()
|
||||||
var finish int64 = 0
|
var finish int64 = 0
|
||||||
const DEFAULT = 4 * 1024 * 1024
|
DEFAULT := d.ChunkSize * 1024 * 1024
|
||||||
for finish < stream.GetSize() {
|
for finish < stream.GetSize() {
|
||||||
if utils.IsCanceled(ctx) {
|
if utils.IsCanceled(ctx) {
|
||||||
return ctx.Err()
|
return ctx.Err()
|
||||||
|
@ -11,9 +11,7 @@ import (
|
|||||||
"strings"
|
"strings"
|
||||||
|
|
||||||
"github.com/alist-org/alist/v3/drivers/base"
|
"github.com/alist-org/alist/v3/drivers/base"
|
||||||
"github.com/alist-org/alist/v3/internal/conf"
|
|
||||||
"github.com/alist-org/alist/v3/internal/driver"
|
"github.com/alist-org/alist/v3/internal/driver"
|
||||||
"github.com/alist-org/alist/v3/internal/errs"
|
|
||||||
"github.com/alist-org/alist/v3/internal/model"
|
"github.com/alist-org/alist/v3/internal/model"
|
||||||
"github.com/alist-org/alist/v3/pkg/utils"
|
"github.com/alist-org/alist/v3/pkg/utils"
|
||||||
"github.com/aws/aws-sdk-go/aws"
|
"github.com/aws/aws-sdk-go/aws"
|
||||||
@ -134,28 +132,14 @@ func (d *PikPak) Remove(ctx context.Context, obj model.Obj) error {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (d *PikPak) Put(ctx context.Context, dstDir model.Obj, stream model.FileStreamer, up driver.UpdateProgress) error {
|
func (d *PikPak) Put(ctx context.Context, dstDir model.Obj, stream model.FileStreamer, up driver.UpdateProgress) error {
|
||||||
var tempFile *os.File
|
tempFile, err := utils.CreateTempFile(stream.GetReadCloser())
|
||||||
var err error
|
if err != nil {
|
||||||
if f, ok := stream.GetReadCloser().(*os.File); ok {
|
return err
|
||||||
tempFile = f
|
|
||||||
} else {
|
|
||||||
tempFile, err = os.CreateTemp(conf.Conf.TempDir, "file-*")
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
defer func() {
|
|
||||||
_ = tempFile.Close()
|
|
||||||
_ = os.Remove(tempFile.Name())
|
|
||||||
}()
|
|
||||||
_, err = io.Copy(tempFile, stream)
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
_, err = tempFile.Seek(0, io.SeekStart)
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
defer func() {
|
||||||
|
_ = tempFile.Close()
|
||||||
|
_ = os.Remove(tempFile.Name())
|
||||||
|
}()
|
||||||
// cal sha1
|
// cal sha1
|
||||||
s := sha1.New()
|
s := sha1.New()
|
||||||
_, err = io.Copy(s, tempFile)
|
_, err = io.Copy(s, tempFile)
|
||||||
@ -182,6 +166,10 @@ func (d *PikPak) Put(ctx context.Context, dstDir model.Obj, stream model.FileStr
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
if stream.GetSize() == 0 {
|
||||||
|
log.Debugln(string(res))
|
||||||
|
return nil
|
||||||
|
}
|
||||||
params := jsoniter.Get(res, "resumable").Get("params")
|
params := jsoniter.Get(res, "resumable").Get("params")
|
||||||
endpoint := params.Get("endpoint").ToString()
|
endpoint := params.Get("endpoint").ToString()
|
||||||
endpointS := strings.Split(endpoint, ".")
|
endpointS := strings.Split(endpoint, ".")
|
||||||
@ -210,8 +198,4 @@ func (d *PikPak) Put(ctx context.Context, dstDir model.Obj, stream model.FileStr
|
|||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
func (d *PikPak) Other(ctx context.Context, args model.OtherArgs) (interface{}, error) {
|
|
||||||
return nil, errs.NotSupport
|
|
||||||
}
|
|
||||||
|
|
||||||
var _ driver.Driver = (*PikPak)(nil)
|
var _ driver.Driver = (*PikPak)(nil)
|
||||||
|
@ -8,9 +8,9 @@ import (
|
|||||||
"io"
|
"io"
|
||||||
"net/http"
|
"net/http"
|
||||||
"os"
|
"os"
|
||||||
|
"time"
|
||||||
|
|
||||||
"github.com/alist-org/alist/v3/drivers/base"
|
"github.com/alist-org/alist/v3/drivers/base"
|
||||||
"github.com/alist-org/alist/v3/internal/conf"
|
|
||||||
"github.com/alist-org/alist/v3/internal/driver"
|
"github.com/alist-org/alist/v3/internal/driver"
|
||||||
"github.com/alist-org/alist/v3/internal/errs"
|
"github.com/alist-org/alist/v3/internal/errs"
|
||||||
"github.com/alist-org/alist/v3/internal/model"
|
"github.com/alist-org/alist/v3/internal/model"
|
||||||
@ -91,6 +91,9 @@ func (d *Quark) MakeDir(ctx context.Context, parentDir model.Obj, dirName string
|
|||||||
_, err := d.request("/file", http.MethodPost, func(req *resty.Request) {
|
_, err := d.request("/file", http.MethodPost, func(req *resty.Request) {
|
||||||
req.SetBody(data)
|
req.SetBody(data)
|
||||||
}, nil)
|
}, nil)
|
||||||
|
if err == nil {
|
||||||
|
time.Sleep(time.Second)
|
||||||
|
}
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -135,28 +138,14 @@ func (d *Quark) Remove(ctx context.Context, obj model.Obj) error {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (d *Quark) Put(ctx context.Context, dstDir model.Obj, stream model.FileStreamer, up driver.UpdateProgress) error {
|
func (d *Quark) Put(ctx context.Context, dstDir model.Obj, stream model.FileStreamer, up driver.UpdateProgress) error {
|
||||||
var tempFile *os.File
|
tempFile, err := utils.CreateTempFile(stream.GetReadCloser())
|
||||||
var err error
|
if err != nil {
|
||||||
if f, ok := stream.GetReadCloser().(*os.File); ok {
|
return err
|
||||||
tempFile = f
|
|
||||||
} else {
|
|
||||||
tempFile, err = os.CreateTemp(conf.Conf.TempDir, "file-*")
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
defer func() {
|
|
||||||
_ = tempFile.Close()
|
|
||||||
_ = os.Remove(tempFile.Name())
|
|
||||||
}()
|
|
||||||
_, err = io.Copy(tempFile, stream)
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
_, err = tempFile.Seek(0, io.SeekStart)
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
defer func() {
|
||||||
|
_ = tempFile.Close()
|
||||||
|
_ = os.Remove(tempFile.Name())
|
||||||
|
}()
|
||||||
m := md5.New()
|
m := md5.New()
|
||||||
_, err = io.Copy(m, tempFile)
|
_, err = io.Copy(m, tempFile)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
@ -230,8 +219,4 @@ func (d *Quark) Put(ctx context.Context, dstDir model.Obj, stream model.FileStre
|
|||||||
return d.upFinish(pre)
|
return d.upFinish(pre)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (d *Quark) Other(ctx context.Context, args model.OtherArgs) (interface{}, error) {
|
|
||||||
return nil, errs.NotSupport
|
|
||||||
}
|
|
||||||
|
|
||||||
var _ driver.Driver = (*Quark)(nil)
|
var _ driver.Driver = (*Quark)(nil)
|
||||||
|
@ -8,13 +8,14 @@ import (
|
|||||||
type Addition struct {
|
type Addition struct {
|
||||||
Cookie string `json:"cookie" required:"true"`
|
Cookie string `json:"cookie" required:"true"`
|
||||||
driver.RootID
|
driver.RootID
|
||||||
OrderBy string `json:"order_by" type:"select" options:"file_type,file_name,updated_at" default:"file_name"`
|
OrderBy string `json:"order_by" type:"select" options:"none,file_type,file_name,updated_at" default:"none"`
|
||||||
OrderDirection string `json:"order_direction" type:"select" options:"asc,desc" default:"asc"`
|
OrderDirection string `json:"order_direction" type:"select" options:"asc,desc" default:"asc"`
|
||||||
}
|
}
|
||||||
|
|
||||||
var config = driver.Config{
|
var config = driver.Config{
|
||||||
Name: "Quark",
|
Name: "Quark",
|
||||||
OnlyProxy: true,
|
OnlyProxy: true,
|
||||||
|
DefaultRoot: "0",
|
||||||
}
|
}
|
||||||
|
|
||||||
func New() driver.Driver {
|
func New() driver.Driver {
|
||||||
|
@ -62,7 +62,9 @@ func (d *Quark) GetFiles(parent string) ([]File, error) {
|
|||||||
"pdir_fid": parent,
|
"pdir_fid": parent,
|
||||||
"_size": strconv.Itoa(size),
|
"_size": strconv.Itoa(size),
|
||||||
"_fetch_total": "1",
|
"_fetch_total": "1",
|
||||||
"_sort": "file_type:asc," + d.OrderBy + ":" + d.OrderDirection,
|
}
|
||||||
|
if d.OrderBy != "none" {
|
||||||
|
query["_sort"] = "file_type:asc," + d.OrderBy + ":" + d.OrderDirection
|
||||||
}
|
}
|
||||||
for {
|
for {
|
||||||
query["_page"] = strconv.Itoa(page)
|
query["_page"] = strconv.Itoa(page)
|
||||||
|
@ -10,7 +10,6 @@ import (
|
|||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/alist-org/alist/v3/internal/driver"
|
"github.com/alist-org/alist/v3/internal/driver"
|
||||||
"github.com/alist-org/alist/v3/internal/errs"
|
|
||||||
"github.com/alist-org/alist/v3/internal/model"
|
"github.com/alist-org/alist/v3/internal/model"
|
||||||
"github.com/alist-org/alist/v3/pkg/utils"
|
"github.com/alist-org/alist/v3/pkg/utils"
|
||||||
"github.com/aws/aws-sdk-go/aws/session"
|
"github.com/aws/aws-sdk-go/aws/session"
|
||||||
@ -119,7 +118,7 @@ func (d *S3) Move(ctx context.Context, srcObj, dstDir model.Obj) error {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (d *S3) Rename(ctx context.Context, srcObj model.Obj, newName string) error {
|
func (d *S3) Rename(ctx context.Context, srcObj model.Obj, newName string) error {
|
||||||
err := d.copy(srcObj.GetPath(), stdpath.Join(stdpath.Dir(srcObj.GetPath()), newName), srcObj.IsDir())
|
err := d.copy(ctx, srcObj.GetPath(), stdpath.Join(stdpath.Dir(srcObj.GetPath()), newName), srcObj.IsDir())
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
@ -127,17 +126,14 @@ func (d *S3) Rename(ctx context.Context, srcObj model.Obj, newName string) error
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (d *S3) Copy(ctx context.Context, srcObj, dstDir model.Obj) error {
|
func (d *S3) Copy(ctx context.Context, srcObj, dstDir model.Obj) error {
|
||||||
return d.copy(srcObj.GetPath(), stdpath.Join(dstDir.GetPath(), srcObj.GetName()), srcObj.IsDir())
|
return d.copy(ctx, srcObj.GetPath(), stdpath.Join(dstDir.GetPath(), srcObj.GetName()), srcObj.IsDir())
|
||||||
}
|
}
|
||||||
|
|
||||||
func (d *S3) Remove(ctx context.Context, obj model.Obj) error {
|
func (d *S3) Remove(ctx context.Context, obj model.Obj) error {
|
||||||
key := getKey(obj.GetPath(), obj.IsDir())
|
if obj.IsDir() {
|
||||||
input := &s3.DeleteObjectInput{
|
return d.removeDir(ctx, obj.GetPath())
|
||||||
Bucket: &d.Bucket,
|
|
||||||
Key: &key,
|
|
||||||
}
|
}
|
||||||
_, err := d.client.DeleteObject(input)
|
return d.removeFile(obj.GetPath())
|
||||||
return err
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (d *S3) Put(ctx context.Context, dstDir model.Obj, stream model.FileStreamer, up driver.UpdateProgress) error {
|
func (d *S3) Put(ctx context.Context, dstDir model.Obj, stream model.FileStreamer, up driver.UpdateProgress) error {
|
||||||
@ -153,8 +149,4 @@ func (d *S3) Put(ctx context.Context, dstDir model.Obj, stream model.FileStreame
|
|||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
func (d *S3) Other(ctx context.Context, args model.OtherArgs) (interface{}, error) {
|
|
||||||
return nil, errs.NotSupport
|
|
||||||
}
|
|
||||||
|
|
||||||
var _ driver.Driver = (*S3)(nil)
|
var _ driver.Driver = (*S3)(nil)
|
||||||
|
@ -1,12 +1,14 @@
|
|||||||
package s3
|
package s3
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"context"
|
||||||
"errors"
|
"errors"
|
||||||
"net/http"
|
"net/http"
|
||||||
"path"
|
"path"
|
||||||
"strings"
|
"strings"
|
||||||
|
|
||||||
"github.com/alist-org/alist/v3/internal/model"
|
"github.com/alist-org/alist/v3/internal/model"
|
||||||
|
"github.com/alist-org/alist/v3/internal/op"
|
||||||
"github.com/aws/aws-sdk-go/aws"
|
"github.com/aws/aws-sdk-go/aws"
|
||||||
"github.com/aws/aws-sdk-go/aws/credentials"
|
"github.com/aws/aws-sdk-go/aws/credentials"
|
||||||
"github.com/aws/aws-sdk-go/aws/request"
|
"github.com/aws/aws-sdk-go/aws/request"
|
||||||
@ -50,7 +52,7 @@ func getKey(path string, dir bool) string {
|
|||||||
return path
|
return path
|
||||||
}
|
}
|
||||||
|
|
||||||
var defaultPlaceholderName = ".placeholder"
|
var defaultPlaceholderName = ".alist"
|
||||||
|
|
||||||
func getPlaceholderName(placeholder string) string {
|
func getPlaceholderName(placeholder string) string {
|
||||||
if placeholder == "" {
|
if placeholder == "" {
|
||||||
@ -87,7 +89,7 @@ func (d *S3) listV1(prefix string) ([]model.Obj, error) {
|
|||||||
}
|
}
|
||||||
for _, object := range listObjectsResult.Contents {
|
for _, object := range listObjectsResult.Contents {
|
||||||
name := path.Base(*object.Key)
|
name := path.Base(*object.Key)
|
||||||
if name == getPlaceholderName(d.Placeholder) {
|
if name == getPlaceholderName(d.Placeholder) || name == d.Placeholder {
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
file := model.Object{
|
file := model.Object{
|
||||||
@ -139,7 +141,7 @@ func (d *S3) listV2(prefix string) ([]model.Obj, error) {
|
|||||||
}
|
}
|
||||||
for _, object := range listObjectsResult.Contents {
|
for _, object := range listObjectsResult.Contents {
|
||||||
name := path.Base(*object.Key)
|
name := path.Base(*object.Key)
|
||||||
if name == getPlaceholderName(d.Placeholder) {
|
if name == getPlaceholderName(d.Placeholder) || name == d.Placeholder {
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
file := model.Object{
|
file := model.Object{
|
||||||
@ -165,14 +167,72 @@ func (d *S3) listV2(prefix string) ([]model.Obj, error) {
|
|||||||
return files, nil
|
return files, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (d *S3) copy(src string, dst string, isDir bool) error {
|
func (d *S3) copy(ctx context.Context, src string, dst string, isDir bool) error {
|
||||||
srcKey := getKey(src, isDir)
|
if isDir {
|
||||||
dstKey := getKey(dst, isDir)
|
return d.copyDir(ctx, src, dst)
|
||||||
|
}
|
||||||
|
return d.copyFile(ctx, src, dst)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *S3) copyFile(ctx context.Context, src string, dst string) error {
|
||||||
|
srcKey := getKey(src, false)
|
||||||
|
dstKey := getKey(dst, false)
|
||||||
input := &s3.CopyObjectInput{
|
input := &s3.CopyObjectInput{
|
||||||
Bucket: &d.Bucket,
|
Bucket: &d.Bucket,
|
||||||
CopySource: &srcKey,
|
CopySource: aws.String("/" + d.Bucket + "/" + srcKey),
|
||||||
Key: &dstKey,
|
Key: &dstKey,
|
||||||
}
|
}
|
||||||
_, err := d.client.CopyObject(input)
|
_, err := d.client.CopyObject(input)
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (d *S3) copyDir(ctx context.Context, src string, dst string) error {
|
||||||
|
objs, err := op.List(ctx, d, src, model.ListArgs{})
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
for _, obj := range objs {
|
||||||
|
cSrc := path.Join(src, obj.GetName())
|
||||||
|
cDst := path.Join(dst, obj.GetName())
|
||||||
|
if obj.IsDir() {
|
||||||
|
err = d.copyDir(ctx, cSrc, cDst)
|
||||||
|
} else {
|
||||||
|
err = d.copyFile(ctx, cSrc, cDst)
|
||||||
|
}
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *S3) removeDir(ctx context.Context, src string) error {
|
||||||
|
objs, err := op.List(ctx, d, src, model.ListArgs{})
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
for _, obj := range objs {
|
||||||
|
cSrc := path.Join(src, obj.GetName())
|
||||||
|
if obj.IsDir() {
|
||||||
|
err = d.removeDir(ctx, cSrc)
|
||||||
|
} else {
|
||||||
|
err = d.removeFile(cSrc)
|
||||||
|
}
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
_ = d.removeFile(path.Join(src, getPlaceholderName(d.Placeholder)))
|
||||||
|
_ = d.removeFile(path.Join(src, d.Placeholder))
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *S3) removeFile(src string) error {
|
||||||
|
key := getKey(src, false)
|
||||||
|
input := &s3.DeleteObjectInput{
|
||||||
|
Bucket: &d.Bucket,
|
||||||
|
Key: &key,
|
||||||
|
}
|
||||||
|
_, err := d.client.DeleteObject(input)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
@ -99,8 +99,4 @@ func (d *SFTP) Put(ctx context.Context, dstDir model.Obj, stream model.FileStrea
|
|||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
func (d *SFTP) Other(ctx context.Context, args model.OtherArgs) (interface{}, error) {
|
|
||||||
return nil, errs.NotSupport
|
|
||||||
}
|
|
||||||
|
|
||||||
var _ driver.Driver = (*SFTP)(nil)
|
var _ driver.Driver = (*SFTP)(nil)
|
||||||
|
205
drivers/smb/driver.go
Normal file
205
drivers/smb/driver.go
Normal file
@ -0,0 +1,205 @@
|
|||||||
|
package smb
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"errors"
|
||||||
|
"path/filepath"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/alist-org/alist/v3/internal/driver"
|
||||||
|
"github.com/alist-org/alist/v3/internal/model"
|
||||||
|
"github.com/alist-org/alist/v3/pkg/utils"
|
||||||
|
|
||||||
|
"github.com/hirochachacha/go-smb2"
|
||||||
|
)
|
||||||
|
|
||||||
|
type SMB struct {
|
||||||
|
model.Storage
|
||||||
|
Addition
|
||||||
|
fs *smb2.Share
|
||||||
|
lastConnTime time.Time
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *SMB) Config() driver.Config {
|
||||||
|
return config
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *SMB) GetAddition() driver.Additional {
|
||||||
|
return d.Addition
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *SMB) Init(ctx context.Context, storage model.Storage) error {
|
||||||
|
d.Storage = storage
|
||||||
|
err := utils.Json.UnmarshalFromString(d.Storage.Addition, &d.Addition)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
return d.initFS()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *SMB) Drop(ctx context.Context) error {
|
||||||
|
if d.fs != nil {
|
||||||
|
_ = d.fs.Umount()
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *SMB) List(ctx context.Context, dir model.Obj, args model.ListArgs) ([]model.Obj, error) {
|
||||||
|
if err := d.checkConn(); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
fullPath := d.getSMBPath(dir)
|
||||||
|
rawFiles, err := d.fs.ReadDir(fullPath)
|
||||||
|
if err != nil {
|
||||||
|
d.cleanLastConnTime()
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
d.updateLastConnTime()
|
||||||
|
var files []model.Obj
|
||||||
|
for _, f := range rawFiles {
|
||||||
|
file := model.ObjThumb{
|
||||||
|
Object: model.Object{
|
||||||
|
Name: f.Name(),
|
||||||
|
Modified: f.ModTime(),
|
||||||
|
Size: f.Size(),
|
||||||
|
IsFolder: f.IsDir(),
|
||||||
|
},
|
||||||
|
}
|
||||||
|
files = append(files, &file)
|
||||||
|
}
|
||||||
|
return files, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
//func (d *SMB) Get(ctx context.Context, path string) (model.Obj, error) {
|
||||||
|
// // this is optional
|
||||||
|
// return nil, errs.NotImplement
|
||||||
|
//}
|
||||||
|
|
||||||
|
func (d *SMB) Link(ctx context.Context, file model.Obj, args model.LinkArgs) (*model.Link, error) {
|
||||||
|
if err := d.checkConn(); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
fullPath := d.getSMBPath(file)
|
||||||
|
remoteFile, err := d.fs.Open(fullPath)
|
||||||
|
if err != nil {
|
||||||
|
d.cleanLastConnTime()
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
d.updateLastConnTime()
|
||||||
|
return &model.Link{
|
||||||
|
Data: remoteFile,
|
||||||
|
}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *SMB) MakeDir(ctx context.Context, parentDir model.Obj, dirName string) error {
|
||||||
|
if err := d.checkConn(); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
fullPath := filepath.Join(d.getSMBPath(parentDir), dirName)
|
||||||
|
err := d.fs.MkdirAll(fullPath, 0700)
|
||||||
|
if err != nil {
|
||||||
|
d.cleanLastConnTime()
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
d.updateLastConnTime()
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *SMB) Move(ctx context.Context, srcObj, dstDir model.Obj) error {
|
||||||
|
if err := d.checkConn(); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
srcPath := d.getSMBPath(srcObj)
|
||||||
|
dstPath := filepath.Join(d.getSMBPath(dstDir), srcObj.GetName())
|
||||||
|
err := d.fs.Rename(srcPath, dstPath)
|
||||||
|
if err != nil {
|
||||||
|
d.cleanLastConnTime()
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
d.updateLastConnTime()
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *SMB) Rename(ctx context.Context, srcObj model.Obj, newName string) error {
|
||||||
|
if err := d.checkConn(); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
srcPath := d.getSMBPath(srcObj)
|
||||||
|
dstPath := filepath.Join(filepath.Dir(srcPath), newName)
|
||||||
|
err := d.fs.Rename(srcPath, dstPath)
|
||||||
|
if err != nil {
|
||||||
|
d.cleanLastConnTime()
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
d.updateLastConnTime()
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *SMB) Copy(ctx context.Context, srcObj, dstDir model.Obj) error {
|
||||||
|
if err := d.checkConn(); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
srcPath := d.getSMBPath(srcObj)
|
||||||
|
dstPath := filepath.Join(d.getSMBPath(dstDir), srcObj.GetName())
|
||||||
|
var err error
|
||||||
|
if srcObj.IsDir() {
|
||||||
|
err = d.CopyDir(srcPath, dstPath)
|
||||||
|
} else {
|
||||||
|
err = d.CopyFile(srcPath, dstPath)
|
||||||
|
}
|
||||||
|
if err != nil {
|
||||||
|
d.cleanLastConnTime()
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
d.updateLastConnTime()
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *SMB) Remove(ctx context.Context, obj model.Obj) error {
|
||||||
|
if err := d.checkConn(); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
var err error
|
||||||
|
fullPath := d.getSMBPath(obj)
|
||||||
|
if obj.IsDir() {
|
||||||
|
err = d.fs.RemoveAll(fullPath)
|
||||||
|
} else {
|
||||||
|
err = d.fs.Remove(fullPath)
|
||||||
|
}
|
||||||
|
if err != nil {
|
||||||
|
d.cleanLastConnTime()
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
d.updateLastConnTime()
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *SMB) Put(ctx context.Context, dstDir model.Obj, stream model.FileStreamer, up driver.UpdateProgress) error {
|
||||||
|
if err := d.checkConn(); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
fullPath := filepath.Join(d.getSMBPath(dstDir), stream.GetName())
|
||||||
|
out, err := d.fs.Create(fullPath)
|
||||||
|
if err != nil {
|
||||||
|
d.cleanLastConnTime()
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
d.updateLastConnTime()
|
||||||
|
defer func() {
|
||||||
|
_ = out.Close()
|
||||||
|
if errors.Is(err, context.Canceled) {
|
||||||
|
_ = d.fs.Remove(fullPath)
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
err = utils.CopyWithCtx(ctx, out, stream, stream.GetSize(), up)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
//func (d *SMB) Other(ctx context.Context, args model.OtherArgs) (interface{}, error) {
|
||||||
|
// return nil, errs.NotSupport
|
||||||
|
//}
|
||||||
|
|
||||||
|
var _ driver.Driver = (*SMB)(nil)
|
28
drivers/smb/meta.go
Normal file
28
drivers/smb/meta.go
Normal file
@ -0,0 +1,28 @@
|
|||||||
|
package smb
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/alist-org/alist/v3/internal/driver"
|
||||||
|
"github.com/alist-org/alist/v3/internal/op"
|
||||||
|
)
|
||||||
|
|
||||||
|
type Addition struct {
|
||||||
|
driver.RootPath
|
||||||
|
Address string `json:"address" required:"true"`
|
||||||
|
Username string `json:"username" required:"true"`
|
||||||
|
Password string `json:"password"`
|
||||||
|
ShareName string `json:"share_name" required:"true"`
|
||||||
|
}
|
||||||
|
|
||||||
|
var config = driver.Config{
|
||||||
|
Name: "SMB",
|
||||||
|
LocalSort: true,
|
||||||
|
OnlyLocal: true,
|
||||||
|
DefaultRoot: ".",
|
||||||
|
NoCache: true,
|
||||||
|
}
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
op.RegisterDriver(config, func() driver.Driver {
|
||||||
|
return &SMB{}
|
||||||
|
})
|
||||||
|
}
|
1
drivers/smb/types.go
Normal file
1
drivers/smb/types.go
Normal file
@ -0,0 +1 @@
|
|||||||
|
package smb
|
142
drivers/smb/util.go
Normal file
142
drivers/smb/util.go
Normal file
@ -0,0 +1,142 @@
|
|||||||
|
package smb
|
||||||
|
|
||||||
|
import (
|
||||||
|
"io"
|
||||||
|
"io/fs"
|
||||||
|
"net"
|
||||||
|
"os"
|
||||||
|
"path/filepath"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/alist-org/alist/v3/internal/model"
|
||||||
|
"github.com/hirochachacha/go-smb2"
|
||||||
|
)
|
||||||
|
|
||||||
|
func (d *SMB) updateLastConnTime() {
|
||||||
|
d.lastConnTime = time.Now()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *SMB) cleanLastConnTime() {
|
||||||
|
d.lastConnTime = time.Now().AddDate(0, 0, -1)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *SMB) initFS() error {
|
||||||
|
conn, err := net.Dial("tcp", d.Address)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
dialer := &smb2.Dialer{
|
||||||
|
Initiator: &smb2.NTLMInitiator{
|
||||||
|
User: d.Username,
|
||||||
|
Password: d.Password,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
s, err := dialer.Dial(conn)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
d.fs, err = s.Mount(d.ShareName)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
d.updateLastConnTime()
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *SMB) checkConn() error {
|
||||||
|
if time.Since(d.lastConnTime) < 5*time.Minute {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
if d.fs != nil {
|
||||||
|
_ = d.fs.Umount()
|
||||||
|
}
|
||||||
|
return d.initFS()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *SMB) getSMBPath(dir model.Obj) string {
|
||||||
|
fullPath := dir.GetPath()
|
||||||
|
if fullPath[0:1] != "." {
|
||||||
|
fullPath = "." + fullPath
|
||||||
|
}
|
||||||
|
return fullPath
|
||||||
|
}
|
||||||
|
|
||||||
|
// CopyFile File copies a single file from src to dst
|
||||||
|
func (d *SMB) CopyFile(src, dst string) error {
|
||||||
|
var err error
|
||||||
|
var srcfd *smb2.File
|
||||||
|
var dstfd *smb2.File
|
||||||
|
var srcinfo fs.FileInfo
|
||||||
|
|
||||||
|
if srcfd, err = d.fs.Open(src); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
defer srcfd.Close()
|
||||||
|
|
||||||
|
if dstfd, err = d.CreateNestedFile(dst); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
defer dstfd.Close()
|
||||||
|
|
||||||
|
if _, err = io.Copy(dstfd, srcfd); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if srcinfo, err = d.fs.Stat(src); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
return d.fs.Chmod(dst, srcinfo.Mode())
|
||||||
|
}
|
||||||
|
|
||||||
|
// CopyDir Dir copies a whole directory recursively
|
||||||
|
func (d *SMB) CopyDir(src string, dst string) error {
|
||||||
|
var err error
|
||||||
|
var fds []fs.FileInfo
|
||||||
|
var srcinfo fs.FileInfo
|
||||||
|
|
||||||
|
if srcinfo, err = d.fs.Stat(src); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if err = d.fs.MkdirAll(dst, srcinfo.Mode()); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if fds, err = d.fs.ReadDir(src); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
for _, fd := range fds {
|
||||||
|
srcfp := filepath.Join(src, fd.Name())
|
||||||
|
dstfp := filepath.Join(dst, fd.Name())
|
||||||
|
|
||||||
|
if fd.IsDir() {
|
||||||
|
if err = d.CopyDir(srcfp, dstfp); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
if err = d.CopyFile(srcfp, dstfp); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// Exists determine whether the file exists
|
||||||
|
func (d *SMB) Exists(name string) bool {
|
||||||
|
if _, err := d.fs.Stat(name); err != nil {
|
||||||
|
if os.IsNotExist(err) {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
// CreateNestedFile create nested file
|
||||||
|
func (d *SMB) CreateNestedFile(path string) (*smb2.File, error) {
|
||||||
|
basePath := filepath.Dir(path)
|
||||||
|
if !d.Exists(basePath) {
|
||||||
|
err := d.fs.MkdirAll(basePath, 0700)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return d.fs.Create(path)
|
||||||
|
}
|
@ -7,7 +7,6 @@ import (
|
|||||||
|
|
||||||
"github.com/alist-org/alist/v3/drivers/base"
|
"github.com/alist-org/alist/v3/drivers/base"
|
||||||
"github.com/alist-org/alist/v3/internal/driver"
|
"github.com/alist-org/alist/v3/internal/driver"
|
||||||
"github.com/alist-org/alist/v3/internal/errs"
|
|
||||||
"github.com/alist-org/alist/v3/internal/model"
|
"github.com/alist-org/alist/v3/internal/model"
|
||||||
"github.com/alist-org/alist/v3/pkg/utils"
|
"github.com/alist-org/alist/v3/pkg/utils"
|
||||||
"github.com/go-resty/resty/v2"
|
"github.com/go-resty/resty/v2"
|
||||||
@ -156,8 +155,4 @@ func (d *Teambition) Put(ctx context.Context, dstDir model.Obj, stream model.Fil
|
|||||||
return d.finishUpload(newFile, dstDir.GetID())
|
return d.finishUpload(newFile, dstDir.GetID())
|
||||||
}
|
}
|
||||||
|
|
||||||
func (d *Teambition) Other(ctx context.Context, args model.OtherArgs) (interface{}, error) {
|
|
||||||
return nil, errs.NotSupport
|
|
||||||
}
|
|
||||||
|
|
||||||
var _ driver.Driver = (*Teambition)(nil)
|
var _ driver.Driver = (*Teambition)(nil)
|
||||||
|
Some files were not shown because too many files have changed in this diff Show More
Reference in New Issue
Block a user