Compare commits
152 Commits
v0.29.2
...
copilot/op
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
ce88dc70f4 | ||
|
|
154ea47e6b | ||
|
|
1b9c8cd2ad | ||
|
|
9ee9972dec | ||
|
|
bd70160555 | ||
|
|
3ce00884a0 | ||
|
|
cd7cf4964d | ||
|
|
bc3c841d1d | ||
|
|
743c15f1a5 | ||
|
|
b05d86509c | ||
|
|
f17a380579 | ||
|
|
cabe1189e2 | ||
|
|
a988d05e24 | ||
|
|
1af2c1f7c7 | ||
|
|
7b36fb45f5 | ||
|
|
62cceee592 | ||
|
|
6d315f7af2 | ||
|
|
5352491c76 | ||
|
|
3f914f7a64 | ||
|
|
8972d8a169 | ||
|
|
1339c69dbf | ||
|
|
63aeabb39b | ||
|
|
e60e983229 | ||
|
|
75e5fd10ea | ||
|
|
c8d8a2e0eb | ||
|
|
044e732084 | ||
|
|
0e951f641c | ||
|
|
8dd6265d55 | ||
|
|
ad7e01f052 | ||
|
|
c069d1edc7 | ||
|
|
fd3b17a360 | ||
|
|
a3162eff89 | ||
|
|
cebadd39f7 | ||
|
|
95f7d5abb5 | ||
|
|
5b8249060d | ||
|
|
0b97d15b2a | ||
|
|
6db5633ca6 | ||
|
|
4ccbf8177e | ||
|
|
6e16dc6051 | ||
|
|
9ed2c425be | ||
|
|
bc892d9370 | ||
|
|
29d523bd4f | ||
|
|
3f9c3f2a73 | ||
|
|
b796b045a8 | ||
|
|
5f7b936270 | ||
|
|
df64ec3069 | ||
|
|
d3cc56c8e6 | ||
|
|
018ed47949 | ||
|
|
09f4dd4ce7 | ||
|
|
adc64ad510 | ||
|
|
da6cf42355 | ||
|
|
8c76516953 | ||
|
|
15c4fffb98 | ||
|
|
b5cdf1e880 | ||
|
|
264dd9f9ed | ||
|
|
b25df2e214 | ||
|
|
5999ddbe1d | ||
|
|
7424190ee5 | ||
|
|
87e8836c78 | ||
|
|
1a7747c2d2 | ||
|
|
ca0fd67fba | ||
|
|
4d736b925b | ||
|
|
ead2b20f4e | ||
|
|
080d474714 | ||
|
|
f453205fde | ||
|
|
407677f270 | ||
|
|
958bfd1dbe | ||
|
|
debe33d84d | ||
|
|
52eead3bf5 | ||
|
|
0af049a507 | ||
|
|
8752dd865c | ||
|
|
c0b4580e34 | ||
|
|
280fd6ead8 | ||
|
|
0ca3d97711 | ||
|
|
51198a1e3d | ||
|
|
651835c467 | ||
|
|
45c978980c | ||
|
|
c21ff7e499 | ||
|
|
32cc1e4b5a | ||
|
|
c974791dc0 | ||
|
|
91814a83c7 | ||
|
|
685047e463 | ||
|
|
37e9c79ceb | ||
|
|
494d1bf51c | ||
|
|
a6f194aedd | ||
|
|
acd16a91a3 | ||
|
|
75f79e8abc | ||
|
|
1065acfdb8 | ||
|
|
fef7d37a7e | ||
|
|
b5e9cf987a | ||
|
|
c58fa454bb | ||
|
|
2c5d6f0e57 | ||
|
|
7d57ad30a9 | ||
|
|
4f314bd37f | ||
|
|
131dfeb4cd | ||
|
|
3f40acff55 | ||
|
|
fe47ee3b51 | ||
|
|
4a6f63e58f | ||
|
|
16c71e6384 | ||
|
|
0c2d116708 | ||
|
|
450d32b2b7 | ||
|
|
f80ecae3cc | ||
|
|
f0853536d9 | ||
|
|
15cf81e1bd | ||
|
|
ae48bd52bf | ||
|
|
44de871f63 | ||
|
|
7a2a530e49 | ||
|
|
3aa84e89bf | ||
|
|
257c292679 | ||
|
|
0e2a9cacf2 | ||
|
|
a7854afb2a | ||
|
|
0e989cc1a6 | ||
|
|
76a82a38ee | ||
|
|
c7a0076c15 | ||
|
|
ea07ff7eca | ||
|
|
4d837e946c | ||
|
|
f947ee6fc7 | ||
|
|
40ad12a892 | ||
|
|
697e419643 | ||
|
|
eef051de3b | ||
|
|
6e29442c05 | ||
|
|
a3f1f75caf | ||
|
|
f05dd883e3 | ||
|
|
9cb866de8c | ||
|
|
980455fd24 | ||
|
|
24978470cd | ||
|
|
215e082028 | ||
|
|
a7b93e57fc | ||
|
|
a4b3b459a9 | ||
|
|
06f326088a | ||
|
|
b7d3ec6230 | ||
|
|
f812990e1c | ||
|
|
492900bbef | ||
|
|
764be2a083 | ||
|
|
46c21b77e9 | ||
|
|
8b389a58d5 | ||
|
|
25ad9befa0 | ||
|
|
e824b210d1 | ||
|
|
ae0aa7db3f | ||
|
|
226c15ef08 | ||
|
|
9b3f955e48 | ||
|
|
4997ec408f | ||
|
|
0756cc9eb1 | ||
|
|
37c32a23d4 | ||
|
|
3aa1e2eaed | ||
|
|
b87dd68880 | ||
|
|
68e5a51300 | ||
|
|
7300e54c40 | ||
|
|
94f796d0e8 | ||
|
|
c023fd869d | ||
|
|
e5d1e143e0 | ||
|
|
03eb4f8a18 |
@@ -9,3 +9,4 @@ cache/
|
|||||||
docs/
|
docs/
|
||||||
config.example.toml
|
config.example.toml
|
||||||
docker-compose.*
|
docker-compose.*
|
||||||
|
playwright/
|
||||||
22
.github/ISSUE_TEMPLATE/bug.yml
vendored
22
.github/ISSUE_TEMPLATE/bug.yml
vendored
@@ -1,35 +1,41 @@
|
|||||||
name: "👾 报告 bug"
|
name: "👾 Bug Report"
|
||||||
description: "报告 bug"
|
description: "Report a bug or unexpected behavior"
|
||||||
labels:
|
labels:
|
||||||
- "bug"
|
- "bug"
|
||||||
assignees:
|
assignees:
|
||||||
- krau
|
- krau
|
||||||
body:
|
body:
|
||||||
|
- type: markdown
|
||||||
|
attributes:
|
||||||
|
value: |
|
||||||
|
# Please Search Before Submitting / 提交前请搜索
|
||||||
|
Please make sure to search existing issues before submitting a new bug report.
|
||||||
|
提交新的 Bug 报告前请务必搜索已有的 issue,避免重复
|
||||||
- type: textarea
|
- type: textarea
|
||||||
attributes:
|
attributes:
|
||||||
label: "👾 问题描述"
|
label: "👾 Description"
|
||||||
description: "What happened?"
|
description: "What happened?"
|
||||||
placeholder: "When called ... happens ..."
|
placeholder: "When called ... happens ..."
|
||||||
validations:
|
validations:
|
||||||
required: true
|
required: true
|
||||||
- type: textarea
|
- type: textarea
|
||||||
attributes:
|
attributes:
|
||||||
label: "⚡️ 预期行为"
|
label: "⚡️ Expected Behavior"
|
||||||
description: "What was expected?"
|
description: "What was expected?"
|
||||||
placeholder: "It should be ..."
|
placeholder: "It should be ..."
|
||||||
- type: textarea
|
- type: textarea
|
||||||
attributes:
|
attributes:
|
||||||
label: "📄 配置文件"
|
label: "📄 Configuration File"
|
||||||
description: "Please provide your config file"
|
description: "Please provide your config file"
|
||||||
placeholder: "请自行隐去密钥信息"
|
placeholder: "Please remove sensitive information"
|
||||||
render: toml
|
render: toml
|
||||||
validations:
|
validations:
|
||||||
required: true
|
required: true
|
||||||
- type: textarea
|
- type: textarea
|
||||||
attributes:
|
attributes:
|
||||||
label: "🔍 日志"
|
label: "🔍 Logs"
|
||||||
description: "Please provide logs"
|
description: "Please provide logs"
|
||||||
placeholder: "可删除隐私信息"
|
placeholder: "Please remove sensitive information"
|
||||||
render: shell
|
render: shell
|
||||||
validations:
|
validations:
|
||||||
required: true
|
required: true
|
||||||
|
|||||||
8
.github/ISSUE_TEMPLATE/config.yml
vendored
8
.github/ISSUE_TEMPLATE/config.yml
vendored
@@ -1,8 +1,8 @@
|
|||||||
blank_issues_enabled: true
|
blank_issues_enabled: true
|
||||||
contact_links:
|
contact_links:
|
||||||
- name: 💬 不知道如何正确使用?
|
- name: 💬 Don't know how to use it correctly?
|
||||||
url: https://github.com/krau/SaveAny-Bot/discussions
|
url: https://github.com/krau/SaveAny-Bot/discussions
|
||||||
about: "前往讨论区提问"
|
about: "Go to the discussion area to ask questions"
|
||||||
- name: 📄 文档
|
- name: 📄 Documentation
|
||||||
url: https://sabot.unv.app
|
url: https://sabot.unv.app
|
||||||
about: "查看文档"
|
about: "View the documentation"
|
||||||
|
|||||||
70
.github/ISSUE_TEMPLATE/feature.yml
vendored
70
.github/ISSUE_TEMPLATE/feature.yml
vendored
@@ -1,5 +1,5 @@
|
|||||||
name: "⭐️ 功能请求"
|
name: "⭐️ Feature Request"
|
||||||
description: "功能请求"
|
description: "Feature request"
|
||||||
labels:
|
labels:
|
||||||
- "enhancement"
|
- "enhancement"
|
||||||
assignees:
|
assignees:
|
||||||
@@ -8,7 +8,69 @@ body:
|
|||||||
- type: markdown
|
- type: markdown
|
||||||
attributes:
|
attributes:
|
||||||
value: |
|
value: |
|
||||||
# 请详细描述你想要的功能
|
Please describe the feature you want in detail.
|
||||||
|
请详细描述你想要的功能。
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## ⚠️ IMPORTANT NOTICE / 说明
|
||||||
|
|
||||||
|
Save Any Bot supports multiple storage backends, **including Telegram**.
|
||||||
|
However, **all backends are treated equally**, keep this in mind when submitting feature requests.
|
||||||
|
|
||||||
|
Save Any Bot 支持多种存储后端,**包括 Telegram**。
|
||||||
|
但**所有后端在设计上是平等的**,请在提出功能请求前务必理解这一点。
|
||||||
|
|
||||||
|
### ❌ Out of scope requests / 不在项目范围内的请求
|
||||||
|
The following requests are **out of scope** and will be closed without discussion:
|
||||||
|
|
||||||
|
以下请求**不属于本项目设计范围**,将被直接关闭,不再讨论:
|
||||||
|
|
||||||
|
- Adding **Telegram-specific behaviors or exceptions**
|
||||||
|
添加 **仅针对 Telegram 的特殊行为或例外逻辑**
|
||||||
|
- Treating Telegram as anything other than a **generic file storage backend**
|
||||||
|
将 Telegram 视为非“通用文件存储后端”的特殊存在
|
||||||
|
- Saving or syncing **non-file content** (text messages, chat history, etc.)
|
||||||
|
保存或同步 **非文件内容**(文本消息、聊天记录等)
|
||||||
|
- Preserving or reconstructing original messages (e.g. 1:1 forwarding)
|
||||||
|
保留或还原原始消息形态(例如 1:1 转发)
|
||||||
|
- Perform special reprocessing on files to adapt to specific storage backends
|
||||||
|
(e.g. splitting, re-encoding, transforming, etc.)
|
||||||
|
为适配特定存储后端而对文件进行特殊处理
|
||||||
|
(如分割、转码、重编码、转换格式等)
|
||||||
|
- Any request that requires different logic *only because the backend is Telegram*
|
||||||
|
任何**仅因后端是 Telegram 而需要不同逻辑**的请求
|
||||||
|
|
||||||
|
### ❌ Abuse-leaning or high-risk requests / 滥用倾向的请求
|
||||||
|
Requests that may **enable or encourage** the following will NOT be accepted:
|
||||||
|
|
||||||
|
可能**促成或鼓励**以下行为的请求将不会被接受:
|
||||||
|
|
||||||
|
- Violating Telegram Terms of Service
|
||||||
|
违反 Telegram 服务条款
|
||||||
|
- Building traffic, mirror, or profit-oriented channels using third-party content
|
||||||
|
利用第三方内容构建引流、镜像或牟利用途的频道
|
||||||
|
|
||||||
|
### ⚖️ Design principle / 设计原则
|
||||||
|
Save Any Bot follows a **backend-agnostic design**:
|
||||||
|
|
||||||
|
Save Any Bot 遵循 **后端无关(backend-agnostic)** 的设计原则:
|
||||||
|
|
||||||
|
- If a feature cannot be implemented **uniformly across all backends**, it will not be added.
|
||||||
|
如果某个功能无法在 **所有后端** 中统一实现,则不会被添加。
|
||||||
|
- No backend-specific hacks or special cases will be introduced.
|
||||||
|
不会引入任何后端特有的 hack 或特殊处理逻辑。
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
If your request falls into any of the categories above, please do not open an issue.
|
||||||
|
Such issues will be closed.
|
||||||
|
|
||||||
|
如果你的请求符合以上任一情况,请不要提交 issue,
|
||||||
|
相关 issue 将被直接关闭。
|
||||||
|
|
||||||
|
Thank you for respecting the scope and design principles of this project.
|
||||||
|
感谢你的理解与支持。
|
||||||
- type: textarea
|
- type: textarea
|
||||||
attributes:
|
attributes:
|
||||||
label: "⭐️ Feature description"
|
label: "⭐️ Feature description"
|
||||||
@@ -30,4 +92,4 @@ body:
|
|||||||
- type: markdown
|
- type: markdown
|
||||||
attributes:
|
attributes:
|
||||||
value: |
|
value: |
|
||||||
## Thank you for contributing to the project :slightly_smiling_face:
|
## Thank you for contributing to the project :slightly_smiling_face:
|
||||||
|
|||||||
166
.github/workflows/build-docker.yml
vendored
166
.github/workflows/build-docker.yml
vendored
@@ -7,32 +7,53 @@ on:
|
|||||||
|
|
||||||
env:
|
env:
|
||||||
REGISTRY: ghcr.io
|
REGISTRY: ghcr.io
|
||||||
IMAGE_NAME: ${{ github.repository }}
|
IMAGE_NAME: krau/saveany-bot
|
||||||
|
|
||||||
|
concurrency:
|
||||||
|
group: docker-build-${{ github.repository }}
|
||||||
|
cancel-in-progress: true
|
||||||
|
|
||||||
jobs:
|
jobs:
|
||||||
build-and-push:
|
prepare:
|
||||||
runs-on: ubuntu-latest
|
runs-on: ubuntu-latest
|
||||||
|
permissions:
|
||||||
|
contents: read
|
||||||
|
outputs:
|
||||||
|
version: ${{ steps.vars.outputs.version }}
|
||||||
|
major_minor: ${{ steps.vars.outputs.major_minor }}
|
||||||
|
short_sha: ${{ steps.vars.outputs.short_sha }}
|
||||||
|
build_time: ${{ steps.vars.outputs.build_time }}
|
||||||
|
steps:
|
||||||
|
- name: Checkout repository
|
||||||
|
uses: actions/checkout@v6
|
||||||
|
|
||||||
|
- name: Extract Version Components
|
||||||
|
id: vars
|
||||||
|
run: |
|
||||||
|
VERSION=${GITHUB_REF#refs/tags/v}
|
||||||
|
MAJOR_MINOR=$(echo "$VERSION" | cut -d. -f1,2)
|
||||||
|
SHORT_SHA=$(git rev-parse --short HEAD)
|
||||||
|
|
||||||
|
echo "version=$VERSION" >> "$GITHUB_OUTPUT"
|
||||||
|
echo "major_minor=$MAJOR_MINOR" >> "$GITHUB_OUTPUT"
|
||||||
|
echo "short_sha=$SHORT_SHA" >> "$GITHUB_OUTPUT"
|
||||||
|
echo "build_time=$(date -u +'%Y-%m-%dT%H:%M:%SZ')" >> "$GITHUB_OUTPUT"
|
||||||
|
|
||||||
|
build:
|
||||||
|
needs: prepare
|
||||||
permissions:
|
permissions:
|
||||||
contents: read
|
contents: read
|
||||||
packages: write
|
packages: write
|
||||||
|
strategy:
|
||||||
|
matrix:
|
||||||
|
arch: [amd64, arm64]
|
||||||
|
type: [default, micro, pico]
|
||||||
|
fail-fast: false
|
||||||
|
runs-on: ${{ matrix.arch == 'amd64' && 'ubuntu-latest' || 'ubuntu-24.04-arm' }}
|
||||||
|
|
||||||
steps:
|
steps:
|
||||||
- name: Checkout repository
|
- name: Checkout repository
|
||||||
uses: actions/checkout@v4
|
uses: actions/checkout@v6
|
||||||
|
|
||||||
- name: Extract metadata for Docker
|
|
||||||
id: meta
|
|
||||||
uses: docker/metadata-action@v5
|
|
||||||
with:
|
|
||||||
images: ${{ env.REGISTRY }}/${{ env.IMAGE_NAME }}
|
|
||||||
tags: |
|
|
||||||
type=semver,pattern={{version}}
|
|
||||||
type=semver,pattern={{major}}.{{minor}}
|
|
||||||
type=sha
|
|
||||||
type=raw,value=latest,enable={{is_default_branch}}
|
|
||||||
|
|
||||||
- name: Set up QEMU
|
|
||||||
uses: docker/setup-qemu-action@v3
|
|
||||||
|
|
||||||
- name: Set up Docker Buildx
|
- name: Set up Docker Buildx
|
||||||
uses: docker/setup-buildx-action@v3
|
uses: docker/setup-buildx-action@v3
|
||||||
@@ -44,26 +65,99 @@ jobs:
|
|||||||
username: ${{ github.actor }}
|
username: ${{ github.actor }}
|
||||||
password: ${{ secrets.GITHUB_TOKEN }}
|
password: ${{ secrets.GITHUB_TOKEN }}
|
||||||
|
|
||||||
- name: Extract Dockerfile args
|
- name: Build and push by digest
|
||||||
id: args
|
id: build
|
||||||
run: |
|
|
||||||
echo "git_commit=$(git rev-parse --short HEAD)" >> "$GITHUB_OUTPUT"
|
|
||||||
echo "build_time=$(git show -s --format=%cI)" >> "$GITHUB_OUTPUT"
|
|
||||||
|
|
||||||
- name: Build and push Docker image
|
|
||||||
id: build-and-push
|
|
||||||
uses: docker/build-push-action@v6
|
uses: docker/build-push-action@v6
|
||||||
with:
|
with:
|
||||||
context: .
|
context: .
|
||||||
platforms: linux/amd64,linux/arm64
|
file: ${{ matrix.type == 'default' && './Dockerfile' || format('./Dockerfile.{0}', matrix.type) }}
|
||||||
push: ${{ github.event_name != 'pull_request' }}
|
platforms: ${{ matrix.arch == 'amd64' && 'linux/amd64' || 'linux/arm64' }}
|
||||||
tags: ${{ steps.meta.outputs.tags }}
|
# 关键修改:不再使用 tags,而是通过 image output 按摘要推送
|
||||||
labels: ${{ steps.meta.outputs.labels }}
|
outputs: type=image,name=${{ env.REGISTRY }}/${{ env.IMAGE_NAME }},push-by-digest=true,name-canonical=true,push=true
|
||||||
cache-from: |
|
|
||||||
type=registry,ref=${{ env.REGISTRY }}/${{ env.IMAGE_NAME }}:latest
|
|
||||||
type=gha
|
|
||||||
cache-to: type=gha,mode=max
|
|
||||||
build-args: |
|
build-args: |
|
||||||
VERSION=${{ steps.meta.outputs.version }}
|
VERSION=${{ needs.prepare.outputs.version }}
|
||||||
GitCommit=${{ steps.args.outputs.git_commit }}
|
GitCommit=${{ needs.prepare.outputs.short_sha }}
|
||||||
BuildTime=${{ steps.args.outputs.build_time }}
|
BuildTime=${{ needs.prepare.outputs.build_time }}
|
||||||
|
cache-from: type=gha
|
||||||
|
cache-to: type=gha,mode=max
|
||||||
|
|
||||||
|
- name:
|
||||||
|
Export digest
|
||||||
|
# 将 digest 写入文件,供后续步骤读取
|
||||||
|
run: |
|
||||||
|
mkdir -p /tmp/digests
|
||||||
|
digest="${{ steps.build.outputs.digest }}"
|
||||||
|
touch "/tmp/digests/${digest#sha256:}"
|
||||||
|
|
||||||
|
echo "$digest" > /tmp/digests/digest
|
||||||
|
|
||||||
|
- name: Upload digest
|
||||||
|
uses: actions/upload-artifact@v6
|
||||||
|
with:
|
||||||
|
name: digest-${{ matrix.type }}-${{ matrix.arch }}
|
||||||
|
path: /tmp/digests/digest
|
||||||
|
if-no-files-found: error
|
||||||
|
retention-days: 1
|
||||||
|
|
||||||
|
create-manifest:
|
||||||
|
needs: [prepare, build]
|
||||||
|
runs-on: ubuntu-latest
|
||||||
|
permissions:
|
||||||
|
packages: write
|
||||||
|
strategy:
|
||||||
|
matrix:
|
||||||
|
type: [default, micro, pico]
|
||||||
|
steps:
|
||||||
|
- name: Login to GitHub Container Registry
|
||||||
|
uses: docker/login-action@v3
|
||||||
|
with:
|
||||||
|
registry: ${{ env.REGISTRY }}
|
||||||
|
username: ${{ github.actor }}
|
||||||
|
password: ${{ secrets.GITHUB_TOKEN }}
|
||||||
|
|
||||||
|
- name: Download digests
|
||||||
|
uses: actions/download-artifact@v7
|
||||||
|
with:
|
||||||
|
path: /tmp/digests
|
||||||
|
pattern: digest-${{ matrix.type }}-*
|
||||||
|
merge-multiple: false
|
||||||
|
|
||||||
|
- name: Create and push manifest lists
|
||||||
|
run: |
|
||||||
|
REPO="${{ env.REGISTRY }}/${{ env.IMAGE_NAME }}"
|
||||||
|
VERSION="${{ needs.prepare.outputs.version }}"
|
||||||
|
MAJOR_MINOR="${{ needs.prepare.outputs.major_minor }}"
|
||||||
|
SHA="${{ needs.prepare.outputs.short_sha }}"
|
||||||
|
TYPE="${{ matrix.type }}"
|
||||||
|
|
||||||
|
DIGEST_AMD64=$(cat /tmp/digests/digest-${TYPE}-amd64/digest)
|
||||||
|
DIGEST_ARM64=$(cat /tmp/digests/digest-${TYPE}-arm64/digest)
|
||||||
|
|
||||||
|
echo "Found digests for $TYPE:"
|
||||||
|
echo "AMD64: $DIGEST_AMD64"
|
||||||
|
echo "ARM64: $DIGEST_ARM64"
|
||||||
|
|
||||||
|
TAGS=()
|
||||||
|
|
||||||
|
if [ "$TYPE" == "default" ]; then
|
||||||
|
TAGS+=("$REPO:latest")
|
||||||
|
TAGS+=("$REPO:$VERSION")
|
||||||
|
TAGS+=("$REPO:$MAJOR_MINOR")
|
||||||
|
TAGS+=("$REPO:sha-$SHA")
|
||||||
|
else
|
||||||
|
TAGS+=("$REPO:$TYPE")
|
||||||
|
TAGS+=("$REPO:$TYPE-latest")
|
||||||
|
TAGS+=("$REPO:$TYPE-$VERSION")
|
||||||
|
fi
|
||||||
|
|
||||||
|
SRC_AMD64="${REPO}@${DIGEST_AMD64}"
|
||||||
|
SRC_ARM64="${REPO}@${DIGEST_ARM64}"
|
||||||
|
|
||||||
|
echo "Creating manifest list with sources:"
|
||||||
|
echo " $SRC_AMD64"
|
||||||
|
echo " $SRC_ARM64"
|
||||||
|
|
||||||
|
for TAG in "${TAGS[@]}"; do
|
||||||
|
echo "Pushing tag: $TAG"
|
||||||
|
docker buildx imagetools create -t "$TAG" "$SRC_AMD64" "$SRC_ARM64"
|
||||||
|
done
|
||||||
|
|||||||
10
.github/workflows/build-release.yml
vendored
10
.github/workflows/build-release.yml
vendored
@@ -15,7 +15,7 @@ jobs:
|
|||||||
runs-on: ubuntu-latest
|
runs-on: ubuntu-latest
|
||||||
steps:
|
steps:
|
||||||
- name: Checkout
|
- name: Checkout
|
||||||
uses: actions/checkout@v4
|
uses: actions/checkout@v6
|
||||||
with:
|
with:
|
||||||
fetch-depth: 0
|
fetch-depth: 0
|
||||||
|
|
||||||
@@ -23,7 +23,7 @@ jobs:
|
|||||||
uses: softprops/action-gh-release@v2
|
uses: softprops/action-gh-release@v2
|
||||||
|
|
||||||
- name: Setup node
|
- name: Setup node
|
||||||
uses: actions/setup-node@v4
|
uses: actions/setup-node@v6
|
||||||
with:
|
with:
|
||||||
node-version: 22
|
node-version: 22
|
||||||
|
|
||||||
@@ -63,9 +63,9 @@ jobs:
|
|||||||
README.md
|
README.md
|
||||||
ldflags: >-
|
ldflags: >-
|
||||||
-s -w
|
-s -w
|
||||||
-X "github.com/krau/SaveAny-Bot/pkg/consts.Version=${{ env.VERSION }}"
|
-X "github.com/krau/SaveAny-Bot/config.Version=${{ env.VERSION }}"
|
||||||
-X "github.com/krau/SaveAny-Bot/pkg/consts.BuildTime=${{ format(github.event.repository.updated_at, 'yyyy-MM-dd HH:mm:ss') }}"
|
-X "github.com/krau/SaveAny-Bot/config.BuildTime=${{ format(github.event.repository.updated_at, 'yyyy-MM-dd HH:mm:ss') }}"
|
||||||
-X "github.com/krau/SaveAny-Bot/pkg/consts.GitCommit=${{ github.sha }}"
|
-X "github.com/krau/SaveAny-Bot/config.GitCommit=${{ github.sha }}"
|
||||||
binary_name: saveany-bot
|
binary_name: saveany-bot
|
||||||
env:
|
env:
|
||||||
VERSION: ${{ env.VERSION }}
|
VERSION: ${{ env.VERSION }}
|
||||||
|
|||||||
4
.github/workflows/docs.yml
vendored
4
.github/workflows/docs.yml
vendored
@@ -9,11 +9,11 @@ on:
|
|||||||
|
|
||||||
jobs:
|
jobs:
|
||||||
deploy:
|
deploy:
|
||||||
runs-on: ubuntu-22.04
|
runs-on: ubuntu-latest
|
||||||
concurrency:
|
concurrency:
|
||||||
group: ${{ github.workflow }}-${{ github.ref }}
|
group: ${{ github.workflow }}-${{ github.ref }}
|
||||||
steps:
|
steps:
|
||||||
- uses: actions/checkout@v4
|
- uses: actions/checkout@v6
|
||||||
with:
|
with:
|
||||||
submodules: true # Fetch Hugo themes (true OR recursive)
|
submodules: true # Fetch Hugo themes (true OR recursive)
|
||||||
fetch-depth: 0 # Fetch all history for .GitInfo and .Lastmod
|
fetch-depth: 0 # Fetch all history for .GitInfo and .Lastmod
|
||||||
|
|||||||
17
.github/workflows/update-contributors.yml
vendored
17
.github/workflows/update-contributors.yml
vendored
@@ -1,17 +0,0 @@
|
|||||||
name: Update Contributors
|
|
||||||
|
|
||||||
on:
|
|
||||||
workflow_dispatch:
|
|
||||||
|
|
||||||
jobs:
|
|
||||||
contrib-readme-job:
|
|
||||||
runs-on: ubuntu-latest
|
|
||||||
name: A job to automate contrib in readme
|
|
||||||
permissions:
|
|
||||||
contents: write
|
|
||||||
pull-requests: write
|
|
||||||
steps:
|
|
||||||
- name: Contribute List
|
|
||||||
uses: akhilmhdh/contributors-readme-action@v2.3.10
|
|
||||||
env:
|
|
||||||
GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
|
|
||||||
4
.gitignore
vendored
4
.gitignore
vendored
@@ -8,3 +8,7 @@ cache.db
|
|||||||
.vscode/
|
.vscode/
|
||||||
temp/
|
temp/
|
||||||
.hugo_build.lock
|
.hugo_build.lock
|
||||||
|
playwright/
|
||||||
|
testplugins/
|
||||||
|
*.exe
|
||||||
|
tmp-*
|
||||||
301
AGENTS.md
Normal file
301
AGENTS.md
Normal file
@@ -0,0 +1,301 @@
|
|||||||
|
# SaveAny-Bot Agent Guidelines
|
||||||
|
|
||||||
|
This document provides essential information for AI coding agents working on the SaveAny-Bot project.
|
||||||
|
|
||||||
|
## Project Overview
|
||||||
|
|
||||||
|
SaveAny-Bot is a Telegram bot written in Go that saves files/messages from Telegram and various websites to multiple storage backends (local, S3, MinIO, WebDAV, AList, Telegram). It features a plugin system for parsing web content and extensible storage backends.
|
||||||
|
|
||||||
|
**Tech Stack**: Go 1.24.2, gotd/td (Telegram MTProto), Cobra (CLI), Viper (config), GORM (ORM), SQLite, Goja (JS runtime), Playwright (browser automation)
|
||||||
|
|
||||||
|
## Build & Test Commands
|
||||||
|
|
||||||
|
### Build
|
||||||
|
```bash
|
||||||
|
# Standard build
|
||||||
|
go build -o saveany-bot .
|
||||||
|
|
||||||
|
# Run directly
|
||||||
|
go run ./cmd
|
||||||
|
|
||||||
|
# Docker build (multi-stage, Alpine-based)
|
||||||
|
docker build -t saveany-bot .
|
||||||
|
docker compose up -d
|
||||||
|
```
|
||||||
|
|
||||||
|
### Test
|
||||||
|
```bash
|
||||||
|
# Run all tests
|
||||||
|
go test ./...
|
||||||
|
|
||||||
|
# Run tests in specific package
|
||||||
|
go test ./pkg/queue
|
||||||
|
go test ./storage/telegram
|
||||||
|
|
||||||
|
# Run tests with verbose output
|
||||||
|
go test -v ./...
|
||||||
|
|
||||||
|
# Run a single test
|
||||||
|
go test -run TestQueueBasic ./pkg/queue
|
||||||
|
|
||||||
|
# Run with coverage
|
||||||
|
go test -cover ./...
|
||||||
|
```
|
||||||
|
|
||||||
|
### Lint & Format
|
||||||
|
```bash
|
||||||
|
# Format code (standard Go formatting)
|
||||||
|
go fmt ./...
|
||||||
|
|
||||||
|
# Vet code for common issues
|
||||||
|
go vet ./...
|
||||||
|
|
||||||
|
# Generate code (i18n keys)
|
||||||
|
go generate ./...
|
||||||
|
```
|
||||||
|
|
||||||
|
### Other Commands
|
||||||
|
```bash
|
||||||
|
# Update dependencies
|
||||||
|
go mod tidy
|
||||||
|
|
||||||
|
# View documentation
|
||||||
|
cd docs && hugo server -D
|
||||||
|
```
|
||||||
|
|
||||||
|
## Code Style Guidelines
|
||||||
|
|
||||||
|
### Imports
|
||||||
|
- Standard library first, then third-party, then project-internal
|
||||||
|
- Group imports with blank lines between groups
|
||||||
|
- Use explicit import aliases for clarity when needed (e.g., `storconfig`, `storenum`)
|
||||||
|
|
||||||
|
```go
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"fmt"
|
||||||
|
|
||||||
|
"github.com/charmbracelet/log"
|
||||||
|
|
||||||
|
"github.com/krau/SaveAny-Bot/config"
|
||||||
|
"github.com/krau/SaveAny-Bot/pkg/enums/storage"
|
||||||
|
)
|
||||||
|
```
|
||||||
|
|
||||||
|
### Formatting
|
||||||
|
- Line length: reasonable (no hard limit, but be sensible)
|
||||||
|
- Organize code with blank lines between logical sections
|
||||||
|
- Follow standard Go conventions for braces, spacing, etc.
|
||||||
|
|
||||||
|
### Types & Interfaces
|
||||||
|
- Use clear, descriptive type names (PascalCase for exported, camelCase for unexported)
|
||||||
|
- Define interfaces where abstraction is needed (e.g., `Executable`, `StorageConfig`)
|
||||||
|
- Embed context in method signatures, not structs: `func (s *Service) Do(ctx context.Context) error`
|
||||||
|
- Prefer composition over inheritance
|
||||||
|
|
||||||
|
```go
|
||||||
|
// Interfaces define behavior
|
||||||
|
type Executable interface {
|
||||||
|
Type() tasktype.TaskType
|
||||||
|
Title() string
|
||||||
|
TaskID() string
|
||||||
|
Execute(ctx context.Context) error
|
||||||
|
}
|
||||||
|
|
||||||
|
// Structs compose behavior
|
||||||
|
type Local struct {
|
||||||
|
config config.LocalStorageConfig
|
||||||
|
logger *log.Logger
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
### Naming Conventions
|
||||||
|
- **Packages**: lowercase, single word when possible (avoid underscores)
|
||||||
|
- **Files**: lowercase with underscores for multiword (e.g., `auth_terminal.go`, `progress_reader.go`)
|
||||||
|
- **Variables**: camelCase for unexported, PascalCase for exported
|
||||||
|
- **Constants**: PascalCase for exported, camelCase for unexported (not ALL_CAPS)
|
||||||
|
- **Functions/Methods**: PascalCase for exported, camelCase for unexported
|
||||||
|
- **Test files**: `*_test.go` pattern
|
||||||
|
|
||||||
|
### Error Handling
|
||||||
|
- Always handle errors explicitly; never ignore them
|
||||||
|
- Wrap errors with context using `fmt.Errorf("context: %w", err)`
|
||||||
|
- Use `errors.Is()` and `errors.As()` for error checking
|
||||||
|
- Log errors with appropriate level (Error, Warn, Info)
|
||||||
|
- Return errors from functions rather than panicking (except for truly unrecoverable situations)
|
||||||
|
|
||||||
|
```go
|
||||||
|
// Good error handling
|
||||||
|
if err := db.Save(user).Error; err != nil {
|
||||||
|
return fmt.Errorf("failed to save user %d: %w", user.ChatID, err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Check specific errors
|
||||||
|
if errors.Is(err, context.Canceled) {
|
||||||
|
logger.Info("Operation was canceled")
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
### Logging
|
||||||
|
- Use `github.com/charmbracelet/log` package
|
||||||
|
- Get logger from context: `log.FromContext(ctx)`
|
||||||
|
- Create prefixed loggers for components: `logger.WithPrefix("component")`
|
||||||
|
- Use appropriate levels: Debug, Info, Warn, Error
|
||||||
|
- Include context in log messages (e.g., task IDs, file names)
|
||||||
|
|
||||||
|
```go
|
||||||
|
logger := log.FromContext(ctx)
|
||||||
|
logger.Infof("Processing task: %s", task.ID)
|
||||||
|
logger.Errorf("Failed to save file %s: %v", filename, err)
|
||||||
|
```
|
||||||
|
|
||||||
|
### Concurrency
|
||||||
|
- Use channels for communication between goroutines
|
||||||
|
- Protect shared state with `sync.Mutex` or `sync.RWMutex`
|
||||||
|
- Use `sync.WaitGroup` for coordinating goroutine completion
|
||||||
|
- Always pass `context.Context` for cancellation support
|
||||||
|
- Use `context.WithCancel/WithTimeout` for managing goroutine lifetimes
|
||||||
|
|
||||||
|
```go
|
||||||
|
// Example from queue implementation
|
||||||
|
func (tq *TaskQueue[T]) Add(task *Task[T]) error {
|
||||||
|
tq.mu.Lock()
|
||||||
|
defer tq.mu.Unlock()
|
||||||
|
// ... critical section
|
||||||
|
tq.cond.Signal()
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
### Comments
|
||||||
|
- Document exported types, functions, and packages with doc comments
|
||||||
|
- Start doc comments with the name being documented
|
||||||
|
- Use `//` for single-line comments
|
||||||
|
- Explain *why*, not *what* (code should be self-explanatory for "what")
|
||||||
|
- Add `[NOTE]`, `[WARN]`, `[IMPORTANT]` tags for important clarifications
|
||||||
|
|
||||||
|
```go
|
||||||
|
// GetUserByChatID retrieves a user by their Telegram chat ID.
|
||||||
|
// Returns an error if the user is not found.
|
||||||
|
func GetUserByChatID(ctx context.Context, chatID int64) (*User, error) {
|
||||||
|
```
|
||||||
|
|
||||||
|
## Architecture & Conventions
|
||||||
|
|
||||||
|
### Application Structure
|
||||||
|
- **Entry point**: `main.go` → `cmd.Execute(ctx)`
|
||||||
|
- **CLI root**: `cmd/root.go` (Cobra), implementation in `cmd/run.go`
|
||||||
|
- **Startup sequence**: Config → Cache → i18n → Database → Storage → Parsers → Userbot → Bot → Queue
|
||||||
|
- Follow this order when adding new initialization steps in `cmd/run.go::initAll`
|
||||||
|
|
||||||
|
### Configuration (Viper)
|
||||||
|
- Config defined in `config/viper.go::Config`
|
||||||
|
- Read from `config.toml` (see `config.example.toml`)
|
||||||
|
- Environment variables: `SAVEANY_*` prefix (e.g., `SAVEANY_TELEGRAM_TOKEN`)
|
||||||
|
- Access via `config.C()` (returns a copy, don't modify the return value)
|
||||||
|
- Storage configs validated via `config/storage/factory.go::LoadStorageConfigs`
|
||||||
|
|
||||||
|
### Telegram Client
|
||||||
|
- **Bot client**: `client/bot/bot.go::Init` (uses gotgproto)
|
||||||
|
- **Handlers**: Centralized in `client/bot/handlers/` directory
|
||||||
|
- **Registration**: All handlers registered in `handlers.Register`
|
||||||
|
- **Commands**: Add to `CommandHandlers` slice for automatic `/help` and bot command list updates
|
||||||
|
- **Middleware**: Common middleware in `client/middleware/` (floodwait, retry, etc.)
|
||||||
|
|
||||||
|
### Tasks & Queue
|
||||||
|
- **Task interface**: `core/core.go::Executable` (Type, Title, TaskID, Execute methods)
|
||||||
|
- **Queue**: `pkg/queue.TaskQueue[Executable]` (generic, thread-safe)
|
||||||
|
- **Workers**: Count from `config.C().Workers`
|
||||||
|
- **Task types**: Implementations in `core/tasks/**` (tfile, parsed, telegraph, directlinks, batchtfile)
|
||||||
|
- **Lifecycle hooks**: `TaskBeforeStart`, `TaskSuccess`, `TaskFail`, `TaskCancel` (defined in config)
|
||||||
|
- **Adding tasks**: Use `core.AddTask(ctx, task)`
|
||||||
|
|
||||||
|
### Database (GORM + SQLite)
|
||||||
|
- **Init**: `database.Init` using `config.C().DB.Path`
|
||||||
|
- **Models**: User, Dir, Rule, WatchChat (in `database/*.go`)
|
||||||
|
- **Migrations**: Automatic via `db.AutoMigrate`
|
||||||
|
- **User sync**: `database.syncUsers` syncs DB with `config.C().Users` (don't manually create/delete users)
|
||||||
|
- **Context**: Always use `db.WithContext(ctx)` for operations
|
||||||
|
|
||||||
|
### Storage Backends
|
||||||
|
- **Interface**: Defined in `config/storage/types.go` and `storage/`
|
||||||
|
- **Implementations**: local, alist, s3/minio, webdav, telegram (each in subdirectory)
|
||||||
|
- **Adding new storage**:
|
||||||
|
1. Add enum to `pkg/enums/storage`
|
||||||
|
2. Create config struct in `config/storage/` with `Validate()` method
|
||||||
|
3. Implement storage in `storage/<name>/`
|
||||||
|
4. Register in `storageFactories` mapping
|
||||||
|
5. Update `config.example.toml` with example
|
||||||
|
|
||||||
|
### Parser Plugins (JavaScript)
|
||||||
|
- **Runtime**: Goja (JS runtime) + Playwright (browser automation)
|
||||||
|
- **Plugin API**: `registerParser({ metadata, canHandle, parse })` in JS
|
||||||
|
- **Integration**: Defined in `parsers/` directory
|
||||||
|
- **Documentation**: See `plugins/README.md`
|
||||||
|
- Plugin `parse` returns `Item`/`Resource` which becomes download/transfer task
|
||||||
|
|
||||||
|
### Internationalization (i18n)
|
||||||
|
- **Usage**: `i18n.T(i18nk.SomeKey, map[string]any{"Name": value})`
|
||||||
|
- **Locale files**: `common/i18n/locale/*.yaml`
|
||||||
|
- **Key generation**: Run `go generate ./...` to generate `common/i18n/i18nk/keys.go`
|
||||||
|
- **Adding new strings**: Add to YAML → run `go generate` → use in code
|
||||||
|
- All user-facing strings should be internationalized
|
||||||
|
|
||||||
|
### Context Usage
|
||||||
|
- Always pass `context.Context` as first parameter
|
||||||
|
- Use `log.FromContext(ctx)` to get contextual logger
|
||||||
|
- Respect context cancellation in long-running operations
|
||||||
|
- Store request-scoped data in context (e.g., `ctxkey.ContentLength`)
|
||||||
|
|
||||||
|
## Special Rules from .github/copilot-instructions.md
|
||||||
|
|
||||||
|
1. **Never modify `config.C()` return values** - it returns a copy. Modify config in `config.Init` or via Viper.
|
||||||
|
2. **Handlers must update `CommandHandlers` slice** - ensures `/help` and bot commands stay in sync.
|
||||||
|
3. **Task execution must preserve hooks** - don't remove `TaskBeforeStart`, `TaskSuccess`, `TaskFail`, `TaskCancel` hook calls.
|
||||||
|
4. **User sync is automatic** - don't manually create/delete users in DB; use config-based sync.
|
||||||
|
5. **Prefer context logger** - use `log.FromContext(ctx)` over global logger when context is available.
|
||||||
|
6. **Storage factory pattern** - new storage types must register in `storageFactories` mapping.
|
||||||
|
7. **Plugin API compatibility** - changes to `Item`/`Resource` structures require updating `plugins/README.md`.
|
||||||
|
|
||||||
|
## Common Patterns
|
||||||
|
|
||||||
|
### Adding a New Command
|
||||||
|
1. Create handler function in `client/bot/handlers/<name>.go`
|
||||||
|
2. Add to `CommandHandlers` slice in `register.go`
|
||||||
|
3. Add i18n key to `common/i18n/locale/*.yaml`
|
||||||
|
4. Run `go generate ./...`
|
||||||
|
5. Test with Telegram bot
|
||||||
|
|
||||||
|
### Adding a New Task Type
|
||||||
|
1. Create struct implementing `core.Executable` in `core/tasks/<type>/`
|
||||||
|
2. Implement `Type()`, `Title()`, `TaskID()`, `Execute(ctx)` methods
|
||||||
|
3. Add task type enum to `pkg/enums/tasktype`
|
||||||
|
4. Use `core.AddTask(ctx, task)` to enqueue
|
||||||
|
|
||||||
|
### Adding a New Storage Backend
|
||||||
|
1. Define config struct in `config/storage/<name>.go` with `Validate()` method
|
||||||
|
2. Implement storage interface in `storage/<name>/<name>.go`
|
||||||
|
3. Add storage type enum to `pkg/enums/storage`
|
||||||
|
4. Register factory in `config/storage/factory.go::storageFactories`
|
||||||
|
5. Update `config.example.toml` with configuration example
|
||||||
|
|
||||||
|
## File References
|
||||||
|
|
||||||
|
When referencing code locations, use `path/to/file.go:line` format (e.g., `core/core.go:23` for the worker function).
|
||||||
|
|
||||||
|
## Testing Guidelines
|
||||||
|
|
||||||
|
- Write tests for new functionality (place in `*_test.go` files)
|
||||||
|
- Test files should be in same package as code being tested
|
||||||
|
- Use table-driven tests for multiple test cases
|
||||||
|
- Mock external dependencies (databases, network calls)
|
||||||
|
- Aim for meaningful tests, not just coverage numbers
|
||||||
|
|
||||||
|
## Notes
|
||||||
|
|
||||||
|
- Binary size matters: use `CGO_ENABLED=0` for static binaries
|
||||||
|
- FFmpeg is included in Docker images for media processing
|
||||||
|
- Build process supports cross-compilation (amd64/arm64, Linux/macOS/Windows)
|
||||||
|
- Documentation site uses Hugo; edit files in `docs/` directory
|
||||||
|
- Session data stored in SQLite; delete `data/session.db` if changing bot token
|
||||||
@@ -17,15 +17,16 @@ RUN --mount=type=cache,target=/root/.cache/go-build \
|
|||||||
go build -trimpath \
|
go build -trimpath \
|
||||||
-ldflags=" \
|
-ldflags=" \
|
||||||
-s -w \
|
-s -w \
|
||||||
-X 'github.com/krau/SaveAny-Bot/common.Version=${VERSION}' \
|
-X 'github.com/krau/SaveAny-Bot/config.Version=${VERSION}' \
|
||||||
-X 'github.com/krau/SaveAny-Bot/common.GitCommit=${GitCommit}' \
|
-X 'github.com/krau/SaveAny-Bot/config.GitCommit=${GitCommit}' \
|
||||||
-X 'github.com/krau/SaveAny-Bot/common.BuildTime=${BuildTime}' \
|
-X 'github.com/krau/SaveAny-Bot/config.BuildTime=${BuildTime}' \
|
||||||
|
-X 'github.com/krau/SaveAny-Bot/config.Docker=true' \
|
||||||
" \
|
" \
|
||||||
-o saveany-bot .
|
-o saveany-bot .
|
||||||
|
|
||||||
FROM alpine:latest
|
FROM alpine:latest
|
||||||
|
|
||||||
RUN apk add --no-cache curl
|
RUN apk add --no-cache curl ffmpeg
|
||||||
|
|
||||||
WORKDIR /app
|
WORKDIR /app
|
||||||
|
|
||||||
|
|||||||
41
Dockerfile.micro
Normal file
41
Dockerfile.micro
Normal file
@@ -0,0 +1,41 @@
|
|||||||
|
FROM golang:alpine AS builder
|
||||||
|
|
||||||
|
ARG VERSION="dev"
|
||||||
|
ARG GitCommit="Unknown"
|
||||||
|
ARG BuildTime="Unknown"
|
||||||
|
|
||||||
|
WORKDIR /app
|
||||||
|
|
||||||
|
COPY go.mod go.sum ./
|
||||||
|
RUN --mount=type=cache,target=/go/pkg/mod \
|
||||||
|
go mod download
|
||||||
|
|
||||||
|
COPY . .
|
||||||
|
RUN --mount=type=cache,target=/root/.cache/go-build \
|
||||||
|
--mount=type=cache,target=/go/pkg \
|
||||||
|
CGO_ENABLED=0 \
|
||||||
|
go build -trimpath \
|
||||||
|
-tags=no_jsparser,no_minio,no_bubbletea \
|
||||||
|
-ldflags=" \
|
||||||
|
-s -w \
|
||||||
|
-X 'github.com/krau/SaveAny-Bot/config.Version=${VERSION}' \
|
||||||
|
-X 'github.com/krau/SaveAny-Bot/config.GitCommit=${GitCommit}' \
|
||||||
|
-X 'github.com/krau/SaveAny-Bot/config.BuildTime=${BuildTime}' \
|
||||||
|
-X 'github.com/krau/SaveAny-Bot/config.Docker=true' \
|
||||||
|
" \
|
||||||
|
-o saveany-bot .
|
||||||
|
|
||||||
|
FROM alpine:latest
|
||||||
|
|
||||||
|
RUN apk add --no-cache curl
|
||||||
|
|
||||||
|
WORKDIR /app
|
||||||
|
|
||||||
|
COPY --from=builder /app/saveany-bot .
|
||||||
|
COPY entrypoint.sh .
|
||||||
|
|
||||||
|
RUN chmod +x /app/saveany-bot && \
|
||||||
|
chmod +x /app/entrypoint.sh
|
||||||
|
|
||||||
|
ENTRYPOINT ["/app/entrypoint.sh"]
|
||||||
|
|
||||||
35
Dockerfile.pico
Normal file
35
Dockerfile.pico
Normal file
@@ -0,0 +1,35 @@
|
|||||||
|
# pico is the minimum build of SaveAnyBot, which disables all the optional features like JS parsing and MinIO support.
|
||||||
|
FROM golang:alpine AS builder
|
||||||
|
|
||||||
|
ARG VERSION="dev"
|
||||||
|
ARG GitCommit="Unknown"
|
||||||
|
ARG BuildTime="Unknown"
|
||||||
|
|
||||||
|
WORKDIR /app
|
||||||
|
|
||||||
|
COPY go.mod go.sum ./
|
||||||
|
RUN --mount=type=cache,target=/go/pkg/mod \
|
||||||
|
go mod download
|
||||||
|
|
||||||
|
COPY . .
|
||||||
|
RUN --mount=type=cache,target=/root/.cache/go-build \
|
||||||
|
--mount=type=cache,target=/go/pkg \
|
||||||
|
CGO_ENABLED=0 \
|
||||||
|
go build -trimpath \
|
||||||
|
-tags=no_jsparser,no_minio,sqlite_glebarez,no_bubbletea \
|
||||||
|
-ldflags=" \
|
||||||
|
-s -w \
|
||||||
|
-X 'github.com/krau/SaveAny-Bot/config.Version=${VERSION}' \
|
||||||
|
-X 'github.com/krau/SaveAny-Bot/config.GitCommit=${GitCommit}' \
|
||||||
|
-X 'github.com/krau/SaveAny-Bot/config.BuildTime=${BuildTime}' \
|
||||||
|
-X 'github.com/krau/SaveAny-Bot/config.Docker=true' \
|
||||||
|
" \
|
||||||
|
-o saveany-bot . && chmod +x saveany-bot
|
||||||
|
|
||||||
|
FROM scratch
|
||||||
|
|
||||||
|
WORKDIR /app
|
||||||
|
|
||||||
|
COPY --from=builder /app/saveany-bot .
|
||||||
|
|
||||||
|
ENTRYPOINT ["/app/saveany-bot"]
|
||||||
99
README.md
99
README.md
@@ -2,9 +2,9 @@
|
|||||||
|
|
||||||
# <img src="docs/static/logo.png" width="45" align="center"> Save Any Bot
|
# <img src="docs/static/logo.png" width="45" align="center"> Save Any Bot
|
||||||
|
|
||||||
**简体中文** | [English](https://sabot.unv.app/en/)
|
**English** | [简体中文](./README_zh.md)
|
||||||
|
|
||||||
> **把 Telegram 上的文件转存到多种存储端.**
|
> **Save Any Telegram File to Anywhere 📂. Support restrict saving content and beyond telegram.**
|
||||||
|
|
||||||
[](https://github.com/krau/saveany-bot/releases)
|
[](https://github.com/krau/saveany-bot/releases)
|
||||||
[](https://github.com/krau/saveany-bot/releases)
|
[](https://github.com/krau/saveany-bot/releases)
|
||||||
@@ -19,46 +19,47 @@
|
|||||||
|
|
||||||
## 🎯 Features
|
## 🎯 Features
|
||||||
|
|
||||||
- 支持文档/视频/图片/贴纸…甚至还有 [Telegraph](https://telegra.ph/)
|
- Support documents / videos / photos / stickers… and even [Telegraph](https://telegra.ph/)
|
||||||
- 破解禁止保存的文件
|
- Bypass "restrict saving content" media
|
||||||
- 批量下载
|
- Batch download
|
||||||
- 流式传输
|
- Streaming transfer
|
||||||
- 多用户使用
|
- Multi-user support
|
||||||
- 基于存储规则的自动整理
|
- Auto organize files based on storage rules
|
||||||
- 监听并自动转存指定聊天的消息, 支持过滤
|
- Watch specified chats and auto-save messages, with filters
|
||||||
- 使用 js 编写解析器插件以转存任意网站的文件
|
- Write JS parser plugins to save files from almost any website
|
||||||
- 存储端支持:
|
- Storage backends:
|
||||||
- Alist
|
- Alist
|
||||||
- S3 (MinioSDK)
|
- S3
|
||||||
- WebDAV
|
- WebDAV
|
||||||
- 本地磁盘
|
- Local filesystem
|
||||||
- Telegram (重传回指定聊天)
|
- Telegram (re-upload to specified chats)
|
||||||
|
|
||||||
## 📦 Quick Start
|
## 📦 Quick Start
|
||||||
|
|
||||||
创建文件 `config.toml` 并填入以下内容:
|
Create a `config.toml` file with the following content:
|
||||||
|
|
||||||
```toml
|
```toml
|
||||||
|
lang = "en" # Language setting, "en" for English
|
||||||
[telegram]
|
[telegram]
|
||||||
token = "" # 你的 Bot Token, 在 @BotFather 获取
|
token = "" # Your bot token, obtained from @BotFather
|
||||||
[telegram.proxy]
|
[telegram.proxy]
|
||||||
# 启用代理连接 telegram, 当前只支持 socks5
|
# Enable proxy for Telegram
|
||||||
enable = false
|
enable = false
|
||||||
url = "socks5://127.0.0.1:7890"
|
url = "socks5://127.0.0.1:7890"
|
||||||
|
|
||||||
[[storages]]
|
[[storages]]
|
||||||
name = "本地磁盘"
|
name = "Local Disk"
|
||||||
type = "local"
|
type = "local"
|
||||||
enable = true
|
enable = true
|
||||||
base_path = "./downloads"
|
base_path = "./downloads"
|
||||||
|
|
||||||
[[users]]
|
[[users]]
|
||||||
id = 114514 # 你的 Telegram 账号 id
|
id = 114514 # Your Telegram account id
|
||||||
storages = []
|
storages = []
|
||||||
blacklist = true
|
blacklist = true
|
||||||
```
|
```
|
||||||
|
|
||||||
使用 Docker 运行 Save Any Bot:
|
Run Save Any Bot with Docker:
|
||||||
|
|
||||||
```bash
|
```bash
|
||||||
docker run -d --name saveany-bot \
|
docker run -d --name saveany-bot \
|
||||||
@@ -67,69 +68,23 @@ docker run -d --name saveany-bot \
|
|||||||
ghcr.io/krau/saveany-bot:latest
|
ghcr.io/krau/saveany-bot:latest
|
||||||
```
|
```
|
||||||
|
|
||||||
请 [**查看文档**](https://sabot.unv.app/) 以获取更多配置选项和使用方法.
|
Please [**read the docs**](https://sabot.unv.app/en/) for more configuration options and usage.
|
||||||
|
|
||||||
## Sponsors
|
## Sponsors
|
||||||
|
|
||||||
本项目受到 [YxVM](https://yxvm.com/) 与 [NodeSupport](https://github.com/NodeSeekDev/NodeSupport) 的支持.
|
This project is supported by [YxVM](https://yxvm.com/) and [NodeSupport](https://github.com/NodeSeekDev/NodeSupport).
|
||||||
|
|
||||||
如果这个项目对你有帮助, 你可以考虑通过以下方式赞助我:
|
If this project is helpful to you, consider sponsoring me via:
|
||||||
|
|
||||||
- [爱发电](https://afdian.com/a/unvapp)
|
- [Afdian](https://afdian.com/a/unvapp)
|
||||||
|
|
||||||
## Contributors
|
## Thanks To
|
||||||
|
|
||||||
<!-- readme: contributors -start -->
|
|
||||||
<table>
|
|
||||||
<tbody>
|
|
||||||
<tr>
|
|
||||||
<td align="center">
|
|
||||||
<a href="https://github.com/krau">
|
|
||||||
<img src="https://avatars.githubusercontent.com/u/71133316?v=4" width="100;" alt="krau"/>
|
|
||||||
<br />
|
|
||||||
<sub><b>Krau</b></sub>
|
|
||||||
</a>
|
|
||||||
</td>
|
|
||||||
<td align="center">
|
|
||||||
<a href="https://github.com/Silentely">
|
|
||||||
<img src="https://avatars.githubusercontent.com/u/22141172?v=4" width="100;" alt="Silentely"/>
|
|
||||||
<br />
|
|
||||||
<sub><b>Abner</b></sub>
|
|
||||||
</a>
|
|
||||||
</td>
|
|
||||||
<td align="center">
|
|
||||||
<a href="https://github.com/TG-Twilight">
|
|
||||||
<img src="https://avatars.githubusercontent.com/u/121682528?v=4" width="100;" alt="TG-Twilight"/>
|
|
||||||
<br />
|
|
||||||
<sub><b>Simon Twilight</b></sub>
|
|
||||||
</a>
|
|
||||||
</td>
|
|
||||||
<td align="center">
|
|
||||||
<a href="https://github.com/ysicing">
|
|
||||||
<img src="https://avatars.githubusercontent.com/u/8605565?v=4" width="100;" alt="ysicing"/>
|
|
||||||
<br />
|
|
||||||
<sub><b>缘生</b></sub>
|
|
||||||
</a>
|
|
||||||
</td>
|
|
||||||
<td align="center">
|
|
||||||
<a href="https://github.com/AHCorn">
|
|
||||||
<img src="https://avatars.githubusercontent.com/u/42889600?v=4" width="100;" alt="AHCorn"/>
|
|
||||||
<br />
|
|
||||||
<sub><b>安和</b></sub>
|
|
||||||
</a>
|
|
||||||
</td>
|
|
||||||
</tr>
|
|
||||||
<tbody>
|
|
||||||
</table>
|
|
||||||
<!-- readme: contributors -end -->
|
|
||||||
|
|
||||||
## Thanks
|
|
||||||
|
|
||||||
- [gotd](https://github.com/gotd/td)
|
- [gotd](https://github.com/gotd/td)
|
||||||
- [TG-FileStreamBot](https://github.com/EverythingSuckz/TG-FileStreamBot)
|
- [TG-FileStreamBot](https://github.com/EverythingSuckz/TG-FileStreamBot)
|
||||||
- [gotgproto](https://github.com/celestix/gotgproto)
|
- [gotgproto](https://github.com/celestix/gotgproto)
|
||||||
- [tdl](https://github.com/iyear/tdl)
|
- [tdl](https://github.com/iyear/tdl)
|
||||||
- All the dependencies
|
- All the dependencies, contributors, sponsors and users.
|
||||||
|
|
||||||
## Contact
|
## Contact
|
||||||
|
|
||||||
|
|||||||
90
README_zh.md
Normal file
90
README_zh.md
Normal file
@@ -0,0 +1,90 @@
|
|||||||
|
<div align="center">
|
||||||
|
|
||||||
|
# <img src="docs/static/logo.png" width="45" align="center"> Save Any Bot
|
||||||
|
|
||||||
|
> **把 Telegram 上的文件转存到多种存储端**
|
||||||
|
|
||||||
|
[](https://github.com/krau/saveany-bot/releases)
|
||||||
|
[](https://github.com/krau/saveany-bot/releases)
|
||||||
|
[](https://github.com/krau/saveany-bot/actions/workflows/build-release.yml)
|
||||||
|
[](https://github.com/krau/saveany-bot/stargazers)
|
||||||
|
[](https://github.com/krau/saveany-bot/releases)
|
||||||
|
[](https://github.com/krau/saveany-bot/issues)
|
||||||
|
[](https://github.com/krau/saveany-bot/pulls)
|
||||||
|
[](./LICENSE)
|
||||||
|
|
||||||
|
</div>
|
||||||
|
|
||||||
|
## 🎯 特性
|
||||||
|
|
||||||
|
- 支持文档/视频/图片/贴纸…甚至还有 [Telegraph](https://telegra.ph/)
|
||||||
|
- 破解禁止保存的文件
|
||||||
|
- 批量下载
|
||||||
|
- 流式传输
|
||||||
|
- 多用户使用
|
||||||
|
- 基于存储规则的自动整理
|
||||||
|
- 监听并自动转存指定聊天的消息, 支持过滤
|
||||||
|
- 使用 js 编写解析器插件以转存任意网站的文件
|
||||||
|
- 存储端支持:
|
||||||
|
- Alist
|
||||||
|
- S3
|
||||||
|
- WebDAV
|
||||||
|
- 本地磁盘
|
||||||
|
- Telegram (重传回指定聊天)
|
||||||
|
|
||||||
|
## 快速开始
|
||||||
|
|
||||||
|
创建文件 `config.toml` 并填入以下内容:
|
||||||
|
|
||||||
|
```toml
|
||||||
|
[telegram]
|
||||||
|
token = "" # 你的 Bot Token, 在 @BotFather 获取
|
||||||
|
[telegram.proxy]
|
||||||
|
# 启用代理连接 telegram
|
||||||
|
enable = false
|
||||||
|
url = "socks5://127.0.0.1:7890"
|
||||||
|
|
||||||
|
[[storages]]
|
||||||
|
name = "本地磁盘"
|
||||||
|
type = "local"
|
||||||
|
enable = true
|
||||||
|
base_path = "./downloads"
|
||||||
|
|
||||||
|
[[users]]
|
||||||
|
id = 114514 # 你的 Telegram 账号 id
|
||||||
|
storages = []
|
||||||
|
blacklist = true
|
||||||
|
```
|
||||||
|
|
||||||
|
使用 Docker 运行 Save Any Bot:
|
||||||
|
|
||||||
|
```bash
|
||||||
|
docker run -d --name saveany-bot \
|
||||||
|
-v ./config.toml:/app/config.toml \
|
||||||
|
-v ./downloads:/app/downloads \
|
||||||
|
ghcr.io/krau/saveany-bot:latest
|
||||||
|
```
|
||||||
|
|
||||||
|
请 [**查看文档**](https://sabot.unv.app/) 以获取更多配置选项和使用方法.
|
||||||
|
|
||||||
|
## 赞助
|
||||||
|
|
||||||
|
本项目受到 [YxVM](https://yxvm.com/) 与 [NodeSupport](https://github.com/NodeSeekDev/NodeSupport) 的支持.
|
||||||
|
|
||||||
|
如果这个项目对你有帮助, 你可以考虑通过以下方式赞助我:
|
||||||
|
|
||||||
|
- [爱发电](https://afdian.com/a/unvapp)
|
||||||
|
|
||||||
|
## 鸣谢
|
||||||
|
|
||||||
|
- [gotd](https://github.com/gotd/td)
|
||||||
|
- [TG-FileStreamBot](https://github.com/EverythingSuckz/TG-FileStreamBot)
|
||||||
|
- [gotgproto](https://github.com/celestix/gotgproto)
|
||||||
|
- [tdl](https://github.com/iyear/tdl)
|
||||||
|
- All the dependencies, contributors, sponsors and users.
|
||||||
|
|
||||||
|
## 社区和关于作者
|
||||||
|
|
||||||
|
- [](https://t.me/ProjectSaveAny)
|
||||||
|
- [](https://github.com/krau/saveany-bot/discussions)
|
||||||
|
- [](https://t.me/acherkrau)
|
||||||
@@ -9,53 +9,56 @@ import (
|
|||||||
"github.com/celestix/gotgproto/ext"
|
"github.com/celestix/gotgproto/ext"
|
||||||
"github.com/celestix/gotgproto/sessionMaker"
|
"github.com/celestix/gotgproto/sessionMaker"
|
||||||
"github.com/charmbracelet/log"
|
"github.com/charmbracelet/log"
|
||||||
"github.com/gotd/td/telegram/dcs"
|
|
||||||
"github.com/gotd/td/tg"
|
"github.com/gotd/td/tg"
|
||||||
"github.com/krau/SaveAny-Bot/client/bot/handlers"
|
"github.com/krau/SaveAny-Bot/client/bot/handlers"
|
||||||
"github.com/krau/SaveAny-Bot/client/middleware"
|
"github.com/krau/SaveAny-Bot/client/middleware"
|
||||||
"github.com/krau/SaveAny-Bot/common/utils/netutil"
|
"github.com/krau/SaveAny-Bot/common/i18n"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/utils/tgutil"
|
||||||
"github.com/krau/SaveAny-Bot/config"
|
"github.com/krau/SaveAny-Bot/config"
|
||||||
"github.com/ncruces/go-sqlite3/gormlite"
|
"github.com/krau/SaveAny-Bot/database"
|
||||||
"golang.org/x/net/proxy"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
func Init(ctx context.Context) {
|
var ectx *ext.Context
|
||||||
log.FromContext(ctx).Info("初始化 Bot...")
|
|
||||||
|
func ExtContext() *ext.Context {
|
||||||
|
return ectx
|
||||||
|
}
|
||||||
|
|
||||||
|
func Init(ctx context.Context) <-chan struct{} {
|
||||||
|
log.FromContext(ctx).Info("Initializing Bot...")
|
||||||
resultChan := make(chan struct {
|
resultChan := make(chan struct {
|
||||||
client *gotgproto.Client
|
client *gotgproto.Client
|
||||||
err error
|
err error
|
||||||
})
|
})
|
||||||
|
shouldRestart := make(chan struct{})
|
||||||
|
|
||||||
go func() {
|
go func() {
|
||||||
var resolver dcs.Resolver
|
resolver, err := tgutil.NewConfigProxyResolver()
|
||||||
if config.Cfg.Telegram.Proxy.Enable && config.Cfg.Telegram.Proxy.URL != "" {
|
if err != nil {
|
||||||
dialer, err := netutil.NewProxyDialer(config.Cfg.Telegram.Proxy.URL)
|
resultChan <- struct {
|
||||||
if err != nil {
|
client *gotgproto.Client
|
||||||
resultChan <- struct {
|
err error
|
||||||
client *gotgproto.Client
|
}{nil, err}
|
||||||
err error
|
return
|
||||||
}{nil, err}
|
|
||||||
return
|
|
||||||
}
|
|
||||||
resolver = dcs.Plain(dcs.PlainOptions{
|
|
||||||
Dial: dialer.(proxy.ContextDialer).DialContext,
|
|
||||||
})
|
|
||||||
} else {
|
|
||||||
resolver = dcs.DefaultResolver()
|
|
||||||
}
|
}
|
||||||
client, err := gotgproto.NewClient(
|
client, err := gotgproto.NewClient(
|
||||||
config.Cfg.Telegram.AppID,
|
config.C().Telegram.AppID,
|
||||||
config.Cfg.Telegram.AppHash,
|
config.C().Telegram.AppHash,
|
||||||
gotgproto.ClientTypeBot(config.Cfg.Telegram.Token),
|
gotgproto.ClientTypeBot(config.C().Telegram.Token),
|
||||||
&gotgproto.ClientOpts{
|
&gotgproto.ClientOpts{
|
||||||
Session: sessionMaker.SqlSession(gormlite.Open(config.Cfg.DB.Session)),
|
Session: sessionMaker.SqlSession(database.GetDialect(config.C().DB.Session)),
|
||||||
DisableCopyright: true,
|
DisableCopyright: true,
|
||||||
Middlewares: middleware.NewDefaultMiddlewares(ctx, 5*time.Minute),
|
Middlewares: middleware.NewDefaultMiddlewares(ctx, 5*time.Minute),
|
||||||
Resolver: resolver,
|
Resolver: resolver,
|
||||||
Context: ctx,
|
Context: ctx,
|
||||||
MaxRetries: config.Cfg.Telegram.RpcRetry,
|
MaxRetries: config.C().Telegram.RpcRetry,
|
||||||
AutoFetchReply: true,
|
AutoFetchReply: true,
|
||||||
ErrorHandler: func(ctx *ext.Context, u *ext.Update, s string) error {
|
ErrorHandler: func(ctx *ext.Context, u *ext.Update, s string) error {
|
||||||
log.FromContext(ctx).Errorf("Unhandled error: %s", s)
|
if s == "SAVEANTBOT-RESTART" {
|
||||||
|
shouldRestart <- struct{}{}
|
||||||
|
return dispatcher.EndGroups
|
||||||
|
}
|
||||||
|
log.FromContext(ctx).Errorf("unhandled error: %s", s)
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
@@ -70,18 +73,9 @@ func Init(ctx context.Context) {
|
|||||||
client.API().BotsSetBotCommands(ctx, &tg.BotsSetBotCommandsRequest{
|
client.API().BotsSetBotCommands(ctx, &tg.BotsSetBotCommandsRequest{
|
||||||
Scope: &tg.BotCommandScopeDefault{},
|
Scope: &tg.BotCommandScopeDefault{},
|
||||||
})
|
})
|
||||||
commands := []tg.BotCommand{
|
commands := make([]tg.BotCommand, 0, len(handlers.CommandHandlers))
|
||||||
{Command: "start", Description: "开始使用"},
|
for _, info := range handlers.CommandHandlers {
|
||||||
{Command: "help", Description: "显示帮助"},
|
commands = append(commands, tg.BotCommand{Command: info.Cmd, Description: i18n.T(info.Desc)})
|
||||||
{Command: "silent", Description: "开启/关闭静默模式"},
|
|
||||||
{Command: "storage", Description: "设置默认存储端"},
|
|
||||||
{Command: "save", Description: "保存文件"},
|
|
||||||
{Command: "dir", Description: "管理存储文件夹"},
|
|
||||||
{Command: "rule", Description: "管理规则"},
|
|
||||||
}
|
|
||||||
if config.Cfg.Telegram.Userbot.Enable {
|
|
||||||
commands = append(commands, tg.BotCommand{Command: "watch", Description: "监听聊天"})
|
|
||||||
commands = append(commands, tg.BotCommand{Command: "unwatch", Description: "取消监听聊天"})
|
|
||||||
}
|
}
|
||||||
_, err = client.API().BotsSetBotCommands(ctx, &tg.BotsSetBotCommandsRequest{
|
_, err = client.API().BotsSetBotCommands(ctx, &tg.BotsSetBotCommandsRequest{
|
||||||
Scope: &tg.BotCommandScopeDefault{},
|
Scope: &tg.BotCommandScopeDefault{},
|
||||||
@@ -95,12 +89,14 @@ func Init(ctx context.Context) {
|
|||||||
|
|
||||||
select {
|
select {
|
||||||
case <-ctx.Done():
|
case <-ctx.Done():
|
||||||
log.FromContext(ctx).Errorf("已取消 Bot 初始化: %s", ctx.Err())
|
log.FromContext(ctx).Errorf("Bot initialization cancelled: %s", ctx.Err())
|
||||||
case result := <-resultChan:
|
case result := <-resultChan:
|
||||||
if result.err != nil {
|
if result.err != nil {
|
||||||
log.FromContext(ctx).Fatalf("初始化 Bot 失败: %s", result.err)
|
log.FromContext(ctx).Fatalf("Failed to initialize Bot: %s", result.err)
|
||||||
}
|
}
|
||||||
handlers.Register(result.client.Dispatcher)
|
handlers.Register(result.client.Dispatcher)
|
||||||
log.FromContext(ctx).Info("Bot 初始化完成")
|
ectx = result.client.CreateContext()
|
||||||
|
log.FromContext(ctx).Info("Bot initialization completed.")
|
||||||
}
|
}
|
||||||
|
return shouldRestart
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -3,6 +3,7 @@ package handlers
|
|||||||
import (
|
import (
|
||||||
"errors"
|
"errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
|
"path"
|
||||||
"strings"
|
"strings"
|
||||||
|
|
||||||
"github.com/celestix/gotgproto/dispatcher"
|
"github.com/celestix/gotgproto/dispatcher"
|
||||||
@@ -11,6 +12,9 @@ import (
|
|||||||
"github.com/gotd/td/tg"
|
"github.com/gotd/td/tg"
|
||||||
"github.com/krau/SaveAny-Bot/client/bot/handlers/utils/msgelem"
|
"github.com/krau/SaveAny-Bot/client/bot/handlers/utils/msgelem"
|
||||||
"github.com/krau/SaveAny-Bot/client/bot/handlers/utils/shortcut"
|
"github.com/krau/SaveAny-Bot/client/bot/handlers/utils/shortcut"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/i18n"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/i18n/i18nk"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/utils/fsutil"
|
||||||
"github.com/krau/SaveAny-Bot/database"
|
"github.com/krau/SaveAny-Bot/database"
|
||||||
"github.com/krau/SaveAny-Bot/pkg/enums/tasktype"
|
"github.com/krau/SaveAny-Bot/pkg/enums/tasktype"
|
||||||
"github.com/krau/SaveAny-Bot/pkg/tcbdata"
|
"github.com/krau/SaveAny-Bot/pkg/tcbdata"
|
||||||
@@ -31,25 +35,29 @@ func handleAddCallback(ctx *ext.Context, update *ext.Update) error {
|
|||||||
selectedStorage, err := storage.GetStorageByUserIDAndName(ctx, userID, data.SelectedStorName)
|
selectedStorage, err := storage.GetStorageByUserIDAndName(ctx, userID, data.SelectedStorName)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.FromContext(ctx).Errorf("Failed to get storage: %s", err)
|
log.FromContext(ctx).Errorf("Failed to get storage: %s", err)
|
||||||
ctx.AnswerCallback(msgelem.AlertCallbackAnswer(queryID, "存储获取失败: "+err.Error()))
|
ctx.AnswerCallback(msgelem.AlertCallbackAnswer(queryID, i18n.T(i18nk.BotMsgCommonErrorGetStorageFailed, map[string]any{
|
||||||
|
"Error": err.Error(),
|
||||||
|
})))
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
dirs, err := database.GetDirsByUserChatIDAndStorageName(ctx, userID, data.SelectedStorName)
|
dirs, err := database.GetDirsByUserChatIDAndStorageName(ctx, userID, data.SelectedStorName)
|
||||||
if err != nil && !errors.Is(err, gorm.ErrRecordNotFound) {
|
if err != nil && !errors.Is(err, gorm.ErrRecordNotFound) {
|
||||||
return fmt.Errorf("获取用户目录失败: %w", err)
|
return fmt.Errorf("failed to get user directories: %w", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
if !data.SettedDir && len(dirs) != 0 {
|
if !data.SettedDir && len(dirs) != 0 {
|
||||||
// ask for directory selection
|
// ask for directory selection
|
||||||
markup, err := msgelem.BuildSetDirKeyboard(dirs, dataid)
|
markup, err := msgelem.BuildSetDirMarkupForAdd(dirs, dataid)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.FromContext(ctx).Errorf("Failed to build directory keyboard: %s", err)
|
log.FromContext(ctx).Errorf("Failed to build directory keyboard: %s", err)
|
||||||
ctx.AnswerCallback(msgelem.AlertCallbackAnswer(queryID, "目录键盘构建失败: "+err.Error()))
|
ctx.AnswerCallback(msgelem.AlertCallbackAnswer(queryID, i18n.T(i18nk.BotMsgCommonErrorBuildStorageSelectKeyboardFailed, map[string]any{
|
||||||
|
"Error": err.Error(),
|
||||||
|
})))
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
ctx.EditMessage(userID, &tg.MessagesEditMessageRequest{
|
ctx.EditMessage(userID, &tg.MessagesEditMessageRequest{
|
||||||
ID: update.CallbackQuery.GetMsgID(),
|
ID: update.CallbackQuery.GetMsgID(),
|
||||||
Message: "请选择要存储到的目录",
|
Message: i18n.T(i18nk.BotMsgCommonPromptSelectDir, nil),
|
||||||
ReplyMarkup: markup,
|
ReplyMarkup: markup,
|
||||||
})
|
})
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
@@ -59,7 +67,9 @@ func handleAddCallback(ctx *ext.Context, update *ext.Update) error {
|
|||||||
if data.DirID != 0 {
|
if data.DirID != 0 {
|
||||||
dir, err := database.GetDirByID(ctx, data.DirID)
|
dir, err := database.GetDirByID(ctx, data.DirID)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
ctx.AnswerCallback(msgelem.AlertCallbackAnswer(queryID, "获取目录失败: "+err.Error()))
|
ctx.AnswerCallback(msgelem.AlertCallbackAnswer(queryID, i18n.T(i18nk.BotMsgCommonErrorGetDirFailed, map[string]any{
|
||||||
|
"Error": err.Error(),
|
||||||
|
})))
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
dirPath = dir.Path
|
dirPath = dir.Path
|
||||||
@@ -74,9 +84,25 @@ func handleAddCallback(ctx *ext.Context, update *ext.Update) error {
|
|||||||
case tasktype.TaskTypeTphpics:
|
case tasktype.TaskTypeTphpics:
|
||||||
return shortcut.CreateAndAddtelegraphWithEdit(ctx, userID, data.TphPageNode, data.TphDirPath, data.TphPics, selectedStorage, msgID)
|
return shortcut.CreateAndAddtelegraphWithEdit(ctx, userID, data.TphPageNode, data.TphDirPath, data.TphPics, selectedStorage, msgID)
|
||||||
case tasktype.TaskTypeParseditem:
|
case tasktype.TaskTypeParseditem:
|
||||||
|
if len(data.ParsedItem.Resources) > 1 {
|
||||||
|
dirPath = path.Join(dirPath, fsutil.NormalizePathname(data.ParsedItem.Title))
|
||||||
|
}
|
||||||
shortcut.CreateAndAddParsedTaskWithEdit(ctx, selectedStorage, dirPath, data.ParsedItem, msgID, userID)
|
shortcut.CreateAndAddParsedTaskWithEdit(ctx, selectedStorage, dirPath, data.ParsedItem, msgID, userID)
|
||||||
|
case tasktype.TaskTypeDirectlinks:
|
||||||
|
shortcut.CreateAndAddDirectTaskWithEdit(ctx, selectedStorage, dirPath, data.DirectLinks, msgID, userID)
|
||||||
|
case tasktype.TaskTypeAria2:
|
||||||
|
client := GetAria2Client()
|
||||||
|
if client == nil {
|
||||||
|
ctx.AnswerCallback(msgelem.AlertCallbackAnswer(queryID, i18n.T(i18nk.BotMsgAria2ErrorAria2ClientInitFailed, map[string]any{
|
||||||
|
"Error": "aria2 client not initialized",
|
||||||
|
})))
|
||||||
|
return dispatcher.EndGroups
|
||||||
|
}
|
||||||
|
shortcut.CreateAndAddAria2TaskWithEdit(ctx, selectedStorage, dirPath, data.Aria2URIs, client, msgID, userID)
|
||||||
|
case tasktype.TaskTypeYtdlp:
|
||||||
|
shortcut.CreateAndAddYtdlpTaskWithEdit(ctx, selectedStorage, dirPath, data.YtdlpURLs, data.YtdlpFlags, msgID, userID)
|
||||||
default:
|
default:
|
||||||
log.FromContext(ctx).Errorf("Unsupported task type: %s", data.TaskType)
|
return fmt.Errorf("unexcept task type: %s", data.TaskType)
|
||||||
}
|
}
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -8,21 +8,46 @@ import (
|
|||||||
"github.com/charmbracelet/log"
|
"github.com/charmbracelet/log"
|
||||||
"github.com/gotd/td/tg"
|
"github.com/gotd/td/tg"
|
||||||
"github.com/krau/SaveAny-Bot/client/bot/handlers/utils/msgelem"
|
"github.com/krau/SaveAny-Bot/client/bot/handlers/utils/msgelem"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/i18n"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/i18n/i18nk"
|
||||||
"github.com/krau/SaveAny-Bot/core"
|
"github.com/krau/SaveAny-Bot/core"
|
||||||
)
|
)
|
||||||
|
|
||||||
func handleCancelCallback(ctx *ext.Context, update *ext.Update) error {
|
func handleCancelCallback(ctx *ext.Context, update *ext.Update) error {
|
||||||
taskid := strings.Split(string(update.CallbackQuery.Data), " ")[1]
|
taskid := strings.Split(string(update.CallbackQuery.Data), " ")[1]
|
||||||
if err := core.CancelTask(ctx, taskid); err != nil {
|
if err := core.CancelTask(ctx, taskid); err != nil {
|
||||||
log.FromContext(ctx).Errorf("error cancelling task %s: %v", taskid, err)
|
log.FromContext(ctx).Errorf("Failed to cancel task %s: %v", taskid, err)
|
||||||
ctx.AnswerCallback(msgelem.AlertCallbackAnswer(update.CallbackQuery.GetQueryID(), "取消任务失败: "+err.Error()))
|
ctx.AnswerCallback(msgelem.AlertCallbackAnswer(update.CallbackQuery.GetQueryID(), i18n.T(i18nk.BotMsgCancelErrorCancelFailed, map[string]any{
|
||||||
|
"Error": err.Error(),
|
||||||
|
})))
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
|
|
||||||
ctx.EditMessage(update.CallbackQuery.GetUserID(), &tg.MessagesEditMessageRequest{
|
ctx.EditMessage(update.CallbackQuery.GetUserID(), &tg.MessagesEditMessageRequest{
|
||||||
ID: update.CallbackQuery.GetMsgID(),
|
ID: update.CallbackQuery.GetMsgID(),
|
||||||
Message: "正在取消任务...",
|
Message: i18n.T(i18nk.BotMsgCancelInfoCancellingTask, nil),
|
||||||
})
|
})
|
||||||
|
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func handleCancelCmd(ctx *ext.Context, update *ext.Update) error {
|
||||||
|
logger := log.FromContext(ctx)
|
||||||
|
args := strings.Fields(update.EffectiveMessage.Text)
|
||||||
|
if len(args) < 2 {
|
||||||
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgCancelUsage, nil)), nil)
|
||||||
|
return dispatcher.EndGroups
|
||||||
|
}
|
||||||
|
taskID := args[1]
|
||||||
|
if err := core.CancelTask(ctx, taskID); err != nil {
|
||||||
|
logger.Errorf("failed to cancel task %s: %v", taskID, err)
|
||||||
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgCancelErrorCancelFailed, map[string]any{
|
||||||
|
"Error": err.Error(),
|
||||||
|
})), nil)
|
||||||
|
return dispatcher.EndGroups
|
||||||
|
}
|
||||||
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgCancelInfoCancelRequested, map[string]any{
|
||||||
|
"TaskID": taskID,
|
||||||
|
})), nil)
|
||||||
|
return dispatcher.EndGroups
|
||||||
|
}
|
||||||
|
|||||||
144
client/bot/handlers/config.go
Normal file
144
client/bot/handlers/config.go
Normal file
@@ -0,0 +1,144 @@
|
|||||||
|
package handlers
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"strings"
|
||||||
|
"text/template"
|
||||||
|
|
||||||
|
"github.com/celestix/gotgproto/dispatcher"
|
||||||
|
"github.com/celestix/gotgproto/ext"
|
||||||
|
"github.com/gotd/td/tg"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/i18n"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/i18n/i18nk"
|
||||||
|
"github.com/krau/SaveAny-Bot/config"
|
||||||
|
"github.com/krau/SaveAny-Bot/database"
|
||||||
|
"github.com/krau/SaveAny-Bot/pkg/enums/fnamest"
|
||||||
|
"github.com/krau/SaveAny-Bot/pkg/tcbdata"
|
||||||
|
)
|
||||||
|
|
||||||
|
func handleConfigCmd(ctx *ext.Context, update *ext.Update) error {
|
||||||
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgConfigPromptSelectOption)), &ext.ReplyOpts{
|
||||||
|
Markup: &tg.ReplyInlineMarkup{
|
||||||
|
Rows: []tg.KeyboardButtonRow{
|
||||||
|
{
|
||||||
|
Buttons: []tg.KeyboardButtonClass{
|
||||||
|
&tg.KeyboardButtonCallback{
|
||||||
|
Text: i18n.T(i18nk.BotMsgConfigButtonFilenameStrategy),
|
||||||
|
Data: fmt.Appendf(nil, "%s %s", tcbdata.TypeConfig, "fnamest"),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
})
|
||||||
|
return dispatcher.EndGroups
|
||||||
|
}
|
||||||
|
|
||||||
|
func handleConfigCallback(ctx *ext.Context, update *ext.Update) error {
|
||||||
|
args := strings.Fields(string(update.CallbackQuery.Data))
|
||||||
|
invaildDataAnswer := func() error {
|
||||||
|
ctx.AnswerCallback(&tg.MessagesSetBotCallbackAnswerRequest{
|
||||||
|
QueryID: update.CallbackQuery.GetQueryID(),
|
||||||
|
Alert: true,
|
||||||
|
Message: i18n.T(i18nk.BotMsgConfigErrorInvalidCallbackData),
|
||||||
|
CacheTime: 5,
|
||||||
|
})
|
||||||
|
return dispatcher.EndGroups
|
||||||
|
}
|
||||||
|
if len(args) < 2 {
|
||||||
|
return invaildDataAnswer()
|
||||||
|
}
|
||||||
|
switch args[1] {
|
||||||
|
case "fnamest":
|
||||||
|
return handleConfigFnameSTCallback(ctx, update)
|
||||||
|
default:
|
||||||
|
return invaildDataAnswer()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func handleConfigFnameSTCallback(ctx *ext.Context, update *ext.Update) error {
|
||||||
|
userID := update.CallbackQuery.GetUserID()
|
||||||
|
user, err := database.GetUserByChatID(ctx, userID)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
args := strings.Fields(string(update.CallbackQuery.Data))
|
||||||
|
if len(args) == 3 {
|
||||||
|
selected := args[2]
|
||||||
|
st, err := fnamest.ParseFnameST(selected)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
user.FilenameStrategy = st.String()
|
||||||
|
if err := database.UpdateUser(ctx, user); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
ctx.EditMessage(userID, &tg.MessagesEditMessageRequest{
|
||||||
|
ID: update.CallbackQuery.GetMsgID(),
|
||||||
|
Message: i18n.T(i18nk.BotMsgConfigInfoFilenameStrategySet, map[string]any{
|
||||||
|
"Strategy": fnamest.GetDisplay(st, config.C().Lang),
|
||||||
|
}),
|
||||||
|
})
|
||||||
|
return dispatcher.EndGroups
|
||||||
|
}
|
||||||
|
opts := fnamest.FnameSTValues()
|
||||||
|
buttons := make([]tg.KeyboardButtonClass, 0, len(opts))
|
||||||
|
for _, opt := range opts {
|
||||||
|
buttons = append(buttons, &tg.KeyboardButtonCallback{
|
||||||
|
Text: fnamest.GetDisplay(opt, config.C().Lang),
|
||||||
|
Data: fmt.Appendf(nil, "%s %s %s", tcbdata.TypeConfig, "fnamest", opt),
|
||||||
|
})
|
||||||
|
}
|
||||||
|
markup := &tg.ReplyInlineMarkup{Rows: []tg.KeyboardButtonRow{
|
||||||
|
{Buttons: buttons},
|
||||||
|
}}
|
||||||
|
currentStStr := user.FilenameStrategy
|
||||||
|
if currentStStr == "" {
|
||||||
|
currentStStr = fnamest.Default.String()
|
||||||
|
}
|
||||||
|
currentSt, err := fnamest.ParseFnameST(currentStStr)
|
||||||
|
if err != nil {
|
||||||
|
currentSt = fnamest.Default
|
||||||
|
}
|
||||||
|
ctx.EditMessage(userID, &tg.MessagesEditMessageRequest{
|
||||||
|
ID: update.CallbackQuery.GetMsgID(),
|
||||||
|
Message: i18n.T(i18nk.BotMsgConfigPromptSelectFilenameStrategy, map[string]any{
|
||||||
|
"Strategy": fnamest.GetDisplay(currentSt, config.C().Lang),
|
||||||
|
}),
|
||||||
|
ReplyMarkup: markup,
|
||||||
|
})
|
||||||
|
return dispatcher.EndGroups
|
||||||
|
}
|
||||||
|
|
||||||
|
func handleConfigFnameTmpl(ctx *ext.Context, update *ext.Update) error {
|
||||||
|
userID := update.GetUserChat().GetID()
|
||||||
|
user, err := database.GetUserByChatID(ctx, userID)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
args := strings.Fields(string(update.EffectiveMessage.Text))
|
||||||
|
if len(args) <= 1 {
|
||||||
|
text := i18n.T(i18nk.BotMsgConfigFnametmplHelp, nil)
|
||||||
|
if user.FilenameTemplate != "" {
|
||||||
|
text += "\n\n" + i18n.T(i18nk.BotMsgConfigInfoCurrentTemplatePrefix, map[string]any{
|
||||||
|
"Template": user.FilenameTemplate,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
ctx.Reply(update, ext.ReplyTextString(text), nil)
|
||||||
|
return dispatcher.EndGroups
|
||||||
|
}
|
||||||
|
newTmpl := strings.Join(args[1:], " ")
|
||||||
|
_, err = template.New("filename").Parse(newTmpl)
|
||||||
|
if err != nil {
|
||||||
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgConfigErrorInvalidTemplate, map[string]any{
|
||||||
|
"Error": err.Error(),
|
||||||
|
})), nil)
|
||||||
|
return dispatcher.EndGroups
|
||||||
|
}
|
||||||
|
user.FilenameTemplate = newTmpl
|
||||||
|
if err := database.UpdateUser(ctx, user); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgConfigInfoTemplateUpdated, nil)), nil)
|
||||||
|
return dispatcher.EndGroups
|
||||||
|
}
|
||||||
@@ -8,6 +8,8 @@ import (
|
|||||||
"github.com/celestix/gotgproto/ext"
|
"github.com/celestix/gotgproto/ext"
|
||||||
"github.com/charmbracelet/log"
|
"github.com/charmbracelet/log"
|
||||||
"github.com/krau/SaveAny-Bot/client/bot/handlers/utils/msgelem"
|
"github.com/krau/SaveAny-Bot/client/bot/handlers/utils/msgelem"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/i18n"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/i18n/i18nk"
|
||||||
"github.com/krau/SaveAny-Bot/database"
|
"github.com/krau/SaveAny-Bot/database"
|
||||||
"github.com/krau/SaveAny-Bot/storage"
|
"github.com/krau/SaveAny-Bot/storage"
|
||||||
)
|
)
|
||||||
@@ -18,8 +20,8 @@ func handleDirCmd(ctx *ext.Context, update *ext.Update) error {
|
|||||||
userChatID := update.GetUserChat().GetID()
|
userChatID := update.GetUserChat().GetID()
|
||||||
dirs, err := database.GetUserDirsByChatID(ctx, userChatID)
|
dirs, err := database.GetUserDirsByChatID(ctx, userChatID)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Errorf("获取用户文件夹失败: %s", err)
|
logger.Errorf("Failed to get user directories: %s", err)
|
||||||
ctx.Reply(update, ext.ReplyTextString("获取用户文件夹失败"), nil)
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgDirErrorGetUserDirsFailed)), nil)
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
if len(args) < 2 {
|
if len(args) < 2 {
|
||||||
@@ -28,8 +30,8 @@ func handleDirCmd(ctx *ext.Context, update *ext.Update) error {
|
|||||||
}
|
}
|
||||||
user, err := database.GetUserByChatID(ctx, update.GetUserChat().GetID())
|
user, err := database.GetUserByChatID(ctx, update.GetUserChat().GetID())
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Errorf("获取用户失败: %s", err)
|
logger.Errorf("Failed to get user: %s", err)
|
||||||
ctx.Reply(update, ext.ReplyTextString("获取用户失败"), nil)
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgDirErrorGetUserFailed)), nil)
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
switch args[1] {
|
switch args[1] {
|
||||||
@@ -45,11 +47,11 @@ func handleDirCmd(ctx *ext.Context, update *ext.Update) error {
|
|||||||
}
|
}
|
||||||
|
|
||||||
if err := database.CreateDirForUser(ctx, user.ID, args[2], args[3]); err != nil {
|
if err := database.CreateDirForUser(ctx, user.ID, args[2], args[3]); err != nil {
|
||||||
logger.Errorf("创建文件夹失败: %s", err)
|
logger.Errorf("Failed to create directory: %s", err)
|
||||||
ctx.Reply(update, ext.ReplyTextString("创建文件夹失败"), nil)
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgDirErrorCreateDirFailed)), nil)
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
ctx.Reply(update, ext.ReplyTextString("文件夹添加成功"), nil)
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgDirInfoCreateDirSuccess)), nil)
|
||||||
case "del":
|
case "del":
|
||||||
// /dir del 3
|
// /dir del 3
|
||||||
if len(args) < 3 {
|
if len(args) < 3 {
|
||||||
@@ -58,17 +60,17 @@ func handleDirCmd(ctx *ext.Context, update *ext.Update) error {
|
|||||||
}
|
}
|
||||||
dirID, err := strconv.Atoi(args[2])
|
dirID, err := strconv.Atoi(args[2])
|
||||||
if err != nil {
|
if err != nil {
|
||||||
ctx.Reply(update, ext.ReplyTextString("文件夹ID无效"), nil)
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgDirErrorInvalidDirId)), nil)
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
if err := database.DeleteDirByID(ctx, uint(dirID)); err != nil {
|
if err := database.DeleteDirByID(ctx, uint(dirID)); err != nil {
|
||||||
logger.Errorf("删除文件夹失败: %s", err)
|
logger.Errorf("Failed to delete directory: %s", err)
|
||||||
ctx.Reply(update, ext.ReplyTextString("删除文件夹失败"), nil)
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgDirErrorDeleteDirFailed)), nil)
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
ctx.Reply(update, ext.ReplyTextString("文件夹删除成功"), nil)
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgDirInfoDeleteDirSuccess)), nil)
|
||||||
default:
|
default:
|
||||||
ctx.Reply(update, ext.ReplyTextString("未知操作"), nil)
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgDirErrorUnknownOperation)), nil)
|
||||||
}
|
}
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
|
|||||||
113
client/bot/handlers/dl.go
Normal file
113
client/bot/handlers/dl.go
Normal file
@@ -0,0 +1,113 @@
|
|||||||
|
package handlers
|
||||||
|
|
||||||
|
import (
|
||||||
|
"net/url"
|
||||||
|
"strings"
|
||||||
|
"sync"
|
||||||
|
|
||||||
|
"github.com/celestix/gotgproto/ext"
|
||||||
|
"github.com/charmbracelet/log"
|
||||||
|
"github.com/duke-git/lancet/v2/slice"
|
||||||
|
"github.com/krau/SaveAny-Bot/client/bot/handlers/utils/msgelem"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/i18n"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/i18n/i18nk"
|
||||||
|
"github.com/krau/SaveAny-Bot/config"
|
||||||
|
"github.com/krau/SaveAny-Bot/pkg/aria2"
|
||||||
|
"github.com/krau/SaveAny-Bot/pkg/enums/tasktype"
|
||||||
|
"github.com/krau/SaveAny-Bot/pkg/tcbdata"
|
||||||
|
"github.com/krau/SaveAny-Bot/storage"
|
||||||
|
)
|
||||||
|
|
||||||
|
func handleDlCmd(ctx *ext.Context, update *ext.Update) error {
|
||||||
|
logger := log.FromContext(ctx)
|
||||||
|
args := strings.Split(update.EffectiveMessage.Text, " ")
|
||||||
|
if len(args) < 2 {
|
||||||
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgDlUsage)), nil)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
links := args[1:]
|
||||||
|
for i, link := range links {
|
||||||
|
links[i] = strings.TrimSpace(link)
|
||||||
|
u, err := url.Parse(link)
|
||||||
|
if err != nil || u.Scheme == "" || u.Host == "" {
|
||||||
|
logger.Warn("invaild link", link)
|
||||||
|
links[i] = ""
|
||||||
|
}
|
||||||
|
}
|
||||||
|
links = slice.Compact(links)
|
||||||
|
if len(links) == 0 {
|
||||||
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgDlErrorNoValidLinks)), nil)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
markup, err := msgelem.BuildAddSelectStorageKeyboard(storage.GetUserStorages(ctx, update.GetUserChat().GetID()), tcbdata.Add{
|
||||||
|
TaskType: tasktype.TaskTypeDirectlinks,
|
||||||
|
DirectLinks: links,
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgDlInfoFilesSelectStorage, map[string]any{
|
||||||
|
"Count": len(links),
|
||||||
|
})), &ext.ReplyOpts{
|
||||||
|
Markup: markup,
|
||||||
|
})
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
var aria2ClientInitOnce sync.Once
|
||||||
|
var aria2ClientInitErr error
|
||||||
|
var aria2Client *aria2.Client
|
||||||
|
|
||||||
|
// GetAria2Client returns the shared aria2 client instance
|
||||||
|
func GetAria2Client() *aria2.Client {
|
||||||
|
return aria2Client
|
||||||
|
}
|
||||||
|
|
||||||
|
func handleAria2DlCmd(ctx *ext.Context, update *ext.Update) error {
|
||||||
|
if !config.C().Aria2.Enable {
|
||||||
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgAria2ErrorAria2NotEnabled)), nil)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
logger := log.FromContext(ctx)
|
||||||
|
args := strings.Split(update.EffectiveMessage.Text, " ")
|
||||||
|
if len(args) < 2 {
|
||||||
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgDlUsage)), nil)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
links := args[1:]
|
||||||
|
for i, link := range links {
|
||||||
|
links[i] = strings.TrimSpace(link)
|
||||||
|
}
|
||||||
|
links = slice.Compact(links)
|
||||||
|
if len(links) == 0 {
|
||||||
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgDlErrorNoValidLinks)), nil)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
logger.Debug("Preparing aria2 download", "links", links)
|
||||||
|
|
||||||
|
// Initialize aria2 client to check connection
|
||||||
|
aria2ClientInitOnce.Do(func() {
|
||||||
|
aria2Client, aria2ClientInitErr = aria2.NewClient(config.C().Aria2.Url, config.C().Aria2.Secret)
|
||||||
|
})
|
||||||
|
if aria2ClientInitErr != nil {
|
||||||
|
logger.Error("Failed to initialize aria2 client", "error", aria2ClientInitErr)
|
||||||
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgAria2ErrorAria2ClientInitFailed, map[string]any{
|
||||||
|
"Error": aria2ClientInitErr.Error(),
|
||||||
|
})), nil)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// Build storage selection keyboard (don't add to aria2 yet)
|
||||||
|
markup, err := msgelem.BuildAddSelectStorageKeyboard(storage.GetUserStorages(ctx, update.GetUserChat().GetID()), tcbdata.Add{
|
||||||
|
TaskType: tasktype.TaskTypeAria2,
|
||||||
|
Aria2URIs: links,
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgAria2InfoSelectStorage)), &ext.ReplyOpts{
|
||||||
|
Markup: markup,
|
||||||
|
})
|
||||||
|
return nil
|
||||||
|
}
|
||||||
@@ -5,29 +5,16 @@ import (
|
|||||||
|
|
||||||
"github.com/celestix/gotgproto/dispatcher"
|
"github.com/celestix/gotgproto/dispatcher"
|
||||||
"github.com/celestix/gotgproto/ext"
|
"github.com/celestix/gotgproto/ext"
|
||||||
"github.com/krau/SaveAny-Bot/pkg/consts"
|
"github.com/krau/SaveAny-Bot/common/i18n"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/i18n/i18nk"
|
||||||
|
"github.com/krau/SaveAny-Bot/config"
|
||||||
)
|
)
|
||||||
|
|
||||||
func handleHelpCmd(ctx *ext.Context, update *ext.Update) error {
|
func handleHelpCmd(ctx *ext.Context, update *ext.Update) error {
|
||||||
const helpText string = `
|
shortHash := config.GitCommit
|
||||||
Save Any Bot - 转存你的 Telegram 文件
|
|
||||||
版本: %s , 提交: %s
|
|
||||||
|
|
||||||
命令:
|
|
||||||
/start - 开始使用
|
|
||||||
/help - 显示帮助
|
|
||||||
/silent - 开关静默模式
|
|
||||||
/storage - 设置默认存储位置
|
|
||||||
/save [自定义文件名] - 保存文件
|
|
||||||
/dir - 管理存储目录
|
|
||||||
/rule - 管理规则
|
|
||||||
|
|
||||||
使用帮助: https://sabot.unv.app/usage/
|
|
||||||
`
|
|
||||||
shortHash := consts.GitCommit
|
|
||||||
if len(shortHash) > 7 {
|
if len(shortHash) > 7 {
|
||||||
shortHash = shortHash[:7]
|
shortHash = shortHash[:7]
|
||||||
}
|
}
|
||||||
ctx.Reply(update, ext.ReplyTextString(fmt.Sprintf(helpText, consts.Version, shortHash)), nil)
|
ctx.Reply(update, ext.ReplyTextString(fmt.Sprintf(i18n.T(i18nk.BotMsgHelpTextFmt), config.Version, shortHash)), nil)
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,13 +1,14 @@
|
|||||||
package handlers
|
package handlers
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
|
||||||
|
|
||||||
"github.com/celestix/gotgproto/dispatcher"
|
"github.com/celestix/gotgproto/dispatcher"
|
||||||
"github.com/celestix/gotgproto/ext"
|
"github.com/celestix/gotgproto/ext"
|
||||||
"github.com/charmbracelet/log"
|
"github.com/charmbracelet/log"
|
||||||
|
"github.com/krau/SaveAny-Bot/client/bot/handlers/utils/dirutil"
|
||||||
"github.com/krau/SaveAny-Bot/client/bot/handlers/utils/msgelem"
|
"github.com/krau/SaveAny-Bot/client/bot/handlers/utils/msgelem"
|
||||||
"github.com/krau/SaveAny-Bot/client/bot/handlers/utils/shortcut"
|
"github.com/krau/SaveAny-Bot/client/bot/handlers/utils/shortcut"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/i18n"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/i18n/i18nk"
|
||||||
"github.com/krau/SaveAny-Bot/pkg/tcbdata"
|
"github.com/krau/SaveAny-Bot/pkg/tcbdata"
|
||||||
"github.com/krau/SaveAny-Bot/storage"
|
"github.com/krau/SaveAny-Bot/storage"
|
||||||
)
|
)
|
||||||
@@ -23,8 +24,10 @@ func handleMessageLink(ctx *ext.Context, update *ext.Update) error {
|
|||||||
if len(files) == 1 {
|
if len(files) == 1 {
|
||||||
req, err := msgelem.BuildAddOneSelectStorageMessage(ctx, stors, files[0], replied.ID)
|
req, err := msgelem.BuildAddOneSelectStorageMessage(ctx, stors, files[0], replied.ID)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Errorf("构建存储选择消息失败: %s", err)
|
logger.Errorf("Failed to build storage selection message: %s", err)
|
||||||
editReplied("构建存储选择消息失败: "+err.Error(), nil)
|
editReplied(i18n.T(i18nk.BotMsgCommonErrorBuildStorageSelectMessageFailed, map[string]any{
|
||||||
|
"Error": err.Error(),
|
||||||
|
}), nil)
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
ctx.EditMessage(update.EffectiveChat().GetID(), req)
|
ctx.EditMessage(update.EffectiveChat().GetID(), req)
|
||||||
@@ -34,29 +37,27 @@ func handleMessageLink(ctx *ext.Context, update *ext.Update) error {
|
|||||||
Files: files,
|
Files: files,
|
||||||
})
|
})
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Errorf("构建存储选择键盘失败: %s", err)
|
logger.Errorf("Failed to build storage selection keyboard: %s", err)
|
||||||
editReplied("构建存储选择键盘失败: "+err.Error(), nil)
|
editReplied(i18n.T(i18nk.BotMsgCommonErrorBuildStorageSelectKeyboardFailed, map[string]any{
|
||||||
|
"Error": err.Error(),
|
||||||
|
}), nil)
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
editReplied(fmt.Sprintf("找到 %d 个文件, 请选择存储位置", len(files)), markup)
|
editReplied(i18n.T(i18nk.BotMsgCommonInfoFoundFilesSelectStorage, map[string]any{
|
||||||
|
"Count": len(files),
|
||||||
|
}), markup)
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
|
|
||||||
func handleSilentSaveLink(ctx *ext.Context, update *ext.Update) error {
|
func handleSilentSaveLink(ctx *ext.Context, update *ext.Update) error {
|
||||||
logger := log.FromContext(ctx)
|
|
||||||
stor := storage.FromContext(ctx)
|
stor := storage.FromContext(ctx)
|
||||||
if stor == nil {
|
|
||||||
logger.Warn("Context storage is nil")
|
|
||||||
ctx.Reply(update, ext.ReplyTextString("未找到存储"), nil)
|
|
||||||
return dispatcher.EndGroups
|
|
||||||
}
|
|
||||||
replied, files, _, err := shortcut.GetFilesFromUpdateLinkMessageWithReplyEdit(ctx, update)
|
replied, files, _, err := shortcut.GetFilesFromUpdateLinkMessageWithReplyEdit(ctx, update)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
userId := update.GetUserChat().GetID()
|
userId := update.GetUserChat().GetID()
|
||||||
if len(files) == 1 {
|
if len(files) == 1 {
|
||||||
return shortcut.CreateAndAddTGFileTaskWithEdit(ctx, userId, stor, "", files[0], replied.ID)
|
return shortcut.CreateAndAddTGFileTaskWithEdit(ctx, userId, stor, dirutil.PathFromContext(ctx), files[0], replied.ID)
|
||||||
}
|
}
|
||||||
return shortcut.CreateAndAddBatchTGFileTaskWithEdit(ctx, userId, stor, "", files, replied.ID)
|
return shortcut.CreateAndAddBatchTGFileTaskWithEdit(ctx, userId, stor, dirutil.PathFromContext(ctx), files, replied.ID)
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,20 +1,16 @@
|
|||||||
package handlers
|
package handlers
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
|
||||||
"sync"
|
|
||||||
"time"
|
|
||||||
|
|
||||||
"github.com/celestix/gotgproto/dispatcher"
|
"github.com/celestix/gotgproto/dispatcher"
|
||||||
"github.com/celestix/gotgproto/ext"
|
"github.com/celestix/gotgproto/ext"
|
||||||
"github.com/charmbracelet/log"
|
"github.com/charmbracelet/log"
|
||||||
"github.com/gotd/td/tg"
|
"github.com/krau/SaveAny-Bot/client/bot/handlers/utils/dirutil"
|
||||||
"github.com/krau/SaveAny-Bot/client/bot/handlers/utils/mediautil"
|
"github.com/krau/SaveAny-Bot/client/bot/handlers/utils/mediautil"
|
||||||
"github.com/krau/SaveAny-Bot/client/bot/handlers/utils/msgelem"
|
"github.com/krau/SaveAny-Bot/client/bot/handlers/utils/msgelem"
|
||||||
"github.com/krau/SaveAny-Bot/client/bot/handlers/utils/shortcut"
|
"github.com/krau/SaveAny-Bot/client/bot/handlers/utils/shortcut"
|
||||||
"github.com/krau/SaveAny-Bot/common/utils/tgutil"
|
"github.com/krau/SaveAny-Bot/common/i18n"
|
||||||
"github.com/krau/SaveAny-Bot/pkg/tcbdata"
|
"github.com/krau/SaveAny-Bot/common/i18n/i18nk"
|
||||||
"github.com/krau/SaveAny-Bot/pkg/tfile"
|
"github.com/krau/SaveAny-Bot/database"
|
||||||
"github.com/krau/SaveAny-Bot/storage"
|
"github.com/krau/SaveAny-Bot/storage"
|
||||||
)
|
)
|
||||||
|
|
||||||
@@ -26,17 +22,24 @@ func handleMediaMessage(ctx *ext.Context, update *ext.Update) error {
|
|||||||
return handleGroupMediaMessage(ctx, update, message, groupID)
|
return handleGroupMediaMessage(ctx, update, message, groupID)
|
||||||
}
|
}
|
||||||
logger.Debugf("Got media: %s", message.Media.TypeName())
|
logger.Debugf("Got media: %s", message.Media.TypeName())
|
||||||
|
userId := update.GetUserChat().GetID()
|
||||||
msg, file, err := shortcut.GetFileFromMessageWithReply(ctx, update, message)
|
userDB, err := database.GetUserByChatID(ctx, userId)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
userId := update.GetUserChat().GetID()
|
tfOpts := mediautil.TfileOptions(ctx, userDB, message)
|
||||||
|
msg, file, err := shortcut.GetFileFromMessageWithReply(ctx, update, message, tfOpts...)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
stors := storage.GetUserStorages(ctx, userId)
|
stors := storage.GetUserStorages(ctx, userId)
|
||||||
req, err := msgelem.BuildAddOneSelectStorageMessage(ctx, stors, file, msg.ID)
|
req, err := msgelem.BuildAddOneSelectStorageMessage(ctx, stors, file, msg.ID)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Errorf("构建存储选择消息失败: %s", err)
|
logger.Errorf("Failed to build storage selection message: %s", err)
|
||||||
ctx.Reply(update, ext.ReplyTextString("构建存储选择消息失败: "+err.Error()), nil)
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgCommonErrorBuildStorageSelectMessageFailed, map[string]any{
|
||||||
|
"Error": err.Error(),
|
||||||
|
})), nil)
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
ctx.EditMessage(update.EffectiveChat().GetID(), req)
|
ctx.EditMessage(update.EffectiveChat().GetID(), req)
|
||||||
@@ -46,11 +49,6 @@ func handleMediaMessage(ctx *ext.Context, update *ext.Update) error {
|
|||||||
func handleSilentSaveMedia(ctx *ext.Context, update *ext.Update) error {
|
func handleSilentSaveMedia(ctx *ext.Context, update *ext.Update) error {
|
||||||
logger := log.FromContext(ctx)
|
logger := log.FromContext(ctx)
|
||||||
stor := storage.FromContext(ctx)
|
stor := storage.FromContext(ctx)
|
||||||
if stor == nil {
|
|
||||||
logger.Warn("Context storage is nil")
|
|
||||||
ctx.Reply(update, ext.ReplyTextString("未找到存储"), nil)
|
|
||||||
return dispatcher.EndGroups
|
|
||||||
}
|
|
||||||
message := update.EffectiveMessage.Message
|
message := update.EffectiveMessage.Message
|
||||||
groupID, isGroup := message.GetGroupedID()
|
groupID, isGroup := message.GetGroupedID()
|
||||||
if isGroup && groupID != 0 {
|
if isGroup && groupID != 0 {
|
||||||
@@ -58,102 +56,14 @@ func handleSilentSaveMedia(ctx *ext.Context, update *ext.Update) error {
|
|||||||
}
|
}
|
||||||
logger.Debugf("Got media: %s", message.Media.TypeName())
|
logger.Debugf("Got media: %s", message.Media.TypeName())
|
||||||
userID := update.GetUserChat().GetID()
|
userID := update.GetUserChat().GetID()
|
||||||
msg, file, err := shortcut.GetFileFromMessageWithReply(ctx, update, message)
|
userDB, err := database.GetUserByChatID(ctx, userID)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
return shortcut.CreateAndAddTGFileTaskWithEdit(ctx, userID, stor, "", file, msg.ID)
|
tfOpts := mediautil.TfileOptions(ctx, userDB, message)
|
||||||
}
|
msg, file, err := shortcut.GetFileFromMessageWithReply(ctx, update, message, tfOpts...)
|
||||||
|
|
||||||
type MediaGroupHandler struct {
|
|
||||||
groups map[int64][]tfile.TGFileMessage
|
|
||||||
timers map[int64]*time.Timer
|
|
||||||
mu sync.Mutex
|
|
||||||
timeout time.Duration
|
|
||||||
}
|
|
||||||
|
|
||||||
var mediaGroupHandler = &MediaGroupHandler{
|
|
||||||
groups: make(map[int64][]tfile.TGFileMessage),
|
|
||||||
timers: make(map[int64]*time.Timer),
|
|
||||||
timeout: 1 * time.Second,
|
|
||||||
}
|
|
||||||
|
|
||||||
func handleGroupMediaMessage(ctx *ext.Context, update *ext.Update, message *tg.Message, groupID int64) error {
|
|
||||||
logger := log.FromContext(ctx)
|
|
||||||
media := message.Media
|
|
||||||
supported := mediautil.IsSupported(media)
|
|
||||||
if !supported {
|
|
||||||
return dispatcher.EndGroups
|
|
||||||
}
|
|
||||||
file, err := tfile.FromMediaMessage(media, ctx.Raw, message, tfile.WithNameIfEmpty(
|
|
||||||
tgutil.GenFileNameFromMessage(*message),
|
|
||||||
))
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Errorf("Failed to get file from media: %s", err)
|
return err
|
||||||
return dispatcher.EndGroups
|
|
||||||
}
|
}
|
||||||
mediaGroupHandler.mu.Lock()
|
return shortcut.CreateAndAddTGFileTaskWithEdit(ctx, userID, stor, dirutil.PathFromContext(ctx), file, msg.ID)
|
||||||
defer mediaGroupHandler.mu.Unlock()
|
|
||||||
if mediaGroupHandler.groups[groupID] == nil {
|
|
||||||
mediaGroupHandler.groups[groupID] = make([]tfile.TGFileMessage, 0)
|
|
||||||
}
|
|
||||||
mediaGroupHandler.groups[groupID] = append(mediaGroupHandler.groups[groupID], file)
|
|
||||||
|
|
||||||
if timer, exists := mediaGroupHandler.timers[groupID]; exists {
|
|
||||||
timer.Stop()
|
|
||||||
}
|
|
||||||
mediaGroupHandler.timers[groupID] = time.AfterFunc(mediaGroupHandler.timeout, func() {
|
|
||||||
processMediaGroup(ctx, update, groupID)
|
|
||||||
})
|
|
||||||
return dispatcher.EndGroups
|
|
||||||
}
|
|
||||||
|
|
||||||
func processMediaGroup(ctx *ext.Context, update *ext.Update, groupID int64) {
|
|
||||||
logger := log.FromContext(ctx)
|
|
||||||
mediaGroupHandler.mu.Lock()
|
|
||||||
items := mediaGroupHandler.groups[groupID]
|
|
||||||
delete(mediaGroupHandler.groups, groupID)
|
|
||||||
delete(mediaGroupHandler.timers, groupID)
|
|
||||||
mediaGroupHandler.mu.Unlock()
|
|
||||||
if len(items) == 0 {
|
|
||||||
logger.Warn("No media items to process for group", "groupID", groupID)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
logger.Debugf("Processing media group %d with %d items", groupID, len(items))
|
|
||||||
|
|
||||||
userId := update.GetUserChat().GetID()
|
|
||||||
msg, err := ctx.Reply(update, ext.ReplyTextString("正在保存文件..."), nil)
|
|
||||||
if err != nil {
|
|
||||||
logger.Errorf("Failed to reply: %s", err)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
stor := storage.FromContext(ctx)
|
|
||||||
if stor != nil {
|
|
||||||
// In silent mode
|
|
||||||
if len(items) == 1 {
|
|
||||||
shortcut.CreateAndAddTGFileTaskWithEdit(ctx, userId, stor, "", items[0], msg.ID)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
shortcut.CreateAndAddBatchTGFileTaskWithEdit(ctx, userId, stor, "", items, msg.ID)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
stors := storage.GetUserStorages(ctx, userId)
|
|
||||||
markup, err := msgelem.BuildAddSelectStorageKeyboard(stors, tcbdata.Add{
|
|
||||||
Files: items,
|
|
||||||
AsBatch: len(items) > 1,
|
|
||||||
})
|
|
||||||
if err != nil {
|
|
||||||
logger.Errorf("构建存储选择键盘失败: %s", err)
|
|
||||||
ctx.EditMessage(userId, &tg.MessagesEditMessageRequest{
|
|
||||||
ID: msg.ID,
|
|
||||||
Message: "构建存储选择键盘失败: " + err.Error(),
|
|
||||||
})
|
|
||||||
return
|
|
||||||
}
|
|
||||||
ctx.EditMessage(userId, &tg.MessagesEditMessageRequest{
|
|
||||||
ID: msg.ID,
|
|
||||||
Message: fmt.Sprintf("共 %d 个文件, 请选择存储位置", len(items)),
|
|
||||||
ReplyMarkup: markup,
|
|
||||||
})
|
|
||||||
}
|
}
|
||||||
|
|||||||
131
client/bot/handlers/media_group.go
Normal file
131
client/bot/handlers/media_group.go
Normal file
@@ -0,0 +1,131 @@
|
|||||||
|
package handlers
|
||||||
|
|
||||||
|
import (
|
||||||
|
"sync"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/celestix/gotgproto/dispatcher"
|
||||||
|
"github.com/celestix/gotgproto/ext"
|
||||||
|
"github.com/charmbracelet/log"
|
||||||
|
"github.com/gotd/td/tg"
|
||||||
|
"github.com/krau/SaveAny-Bot/client/bot/handlers/utils/mediautil"
|
||||||
|
"github.com/krau/SaveAny-Bot/client/bot/handlers/utils/msgelem"
|
||||||
|
"github.com/krau/SaveAny-Bot/client/bot/handlers/utils/shortcut"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/i18n"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/i18n/i18nk"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/utils/tgutil"
|
||||||
|
"github.com/krau/SaveAny-Bot/config"
|
||||||
|
"github.com/krau/SaveAny-Bot/pkg/tcbdata"
|
||||||
|
"github.com/krau/SaveAny-Bot/pkg/tfile"
|
||||||
|
"github.com/krau/SaveAny-Bot/storage"
|
||||||
|
)
|
||||||
|
|
||||||
|
type MediaGroupHandler struct {
|
||||||
|
groups map[int64][]tfile.TGFileMessage
|
||||||
|
timers map[int64]*time.Timer
|
||||||
|
mu sync.Mutex
|
||||||
|
timeout time.Duration
|
||||||
|
setupOnce sync.Once
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *MediaGroupHandler) SetupTimeout(timeoutSec int) {
|
||||||
|
m.setupOnce.Do(func() {
|
||||||
|
if timeoutSec < 1 {
|
||||||
|
timeoutSec = 1
|
||||||
|
}
|
||||||
|
m.timeout = time.Duration(timeoutSec) * time.Second
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
var (
|
||||||
|
mediaGroupHandler = &MediaGroupHandler{
|
||||||
|
groups: make(map[int64][]tfile.TGFileMessage),
|
||||||
|
timers: make(map[int64]*time.Timer),
|
||||||
|
mu: sync.Mutex{},
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
func handleGroupMediaMessage(ctx *ext.Context, update *ext.Update, message *tg.Message, groupID int64) error {
|
||||||
|
mediaGroupHandler.SetupTimeout(max(config.C().Telegram.MediaGroupTimeout, 1))
|
||||||
|
logger := log.FromContext(ctx)
|
||||||
|
media := message.Media
|
||||||
|
supported := mediautil.IsSupported(media)
|
||||||
|
if !supported {
|
||||||
|
return dispatcher.EndGroups
|
||||||
|
}
|
||||||
|
file, err := tfile.FromMediaMessage(media, ctx.Raw, message, tfile.WithNameIfEmpty(
|
||||||
|
tgutil.GenFileNameFromMessage(*message),
|
||||||
|
))
|
||||||
|
if err != nil {
|
||||||
|
logger.Errorf("Failed to get file from media: %s", err)
|
||||||
|
return dispatcher.EndGroups
|
||||||
|
}
|
||||||
|
mediaGroupHandler.mu.Lock()
|
||||||
|
defer mediaGroupHandler.mu.Unlock()
|
||||||
|
if mediaGroupHandler.groups[groupID] == nil {
|
||||||
|
mediaGroupHandler.groups[groupID] = make([]tfile.TGFileMessage, 0)
|
||||||
|
}
|
||||||
|
mediaGroupHandler.groups[groupID] = append(mediaGroupHandler.groups[groupID], file)
|
||||||
|
|
||||||
|
if timer, exists := mediaGroupHandler.timers[groupID]; exists {
|
||||||
|
timer.Stop()
|
||||||
|
}
|
||||||
|
mediaGroupHandler.timers[groupID] = time.AfterFunc(mediaGroupHandler.timeout, func() {
|
||||||
|
processMediaGroup(ctx, update, groupID)
|
||||||
|
})
|
||||||
|
return dispatcher.EndGroups
|
||||||
|
}
|
||||||
|
|
||||||
|
func processMediaGroup(ctx *ext.Context, update *ext.Update, groupID int64) {
|
||||||
|
logger := log.FromContext(ctx)
|
||||||
|
mediaGroupHandler.mu.Lock()
|
||||||
|
items := mediaGroupHandler.groups[groupID]
|
||||||
|
delete(mediaGroupHandler.groups, groupID)
|
||||||
|
delete(mediaGroupHandler.timers, groupID)
|
||||||
|
mediaGroupHandler.mu.Unlock()
|
||||||
|
if len(items) == 0 {
|
||||||
|
logger.Warn("No media items to process for group", "groupID", groupID)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
logger.Debugf("Processing media group %d with %d items", groupID, len(items))
|
||||||
|
|
||||||
|
userId := update.GetUserChat().GetID()
|
||||||
|
msg, err := ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgMediaGroupInfoSavingFiles, nil)), nil)
|
||||||
|
if err != nil {
|
||||||
|
logger.Errorf("Failed to reply: %s", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
stor := storage.FromContext(ctx)
|
||||||
|
if stor != nil {
|
||||||
|
// In silent mode
|
||||||
|
if len(items) == 1 {
|
||||||
|
shortcut.CreateAndAddTGFileTaskWithEdit(ctx, userId, stor, "", items[0], msg.ID)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
shortcut.CreateAndAddBatchTGFileTaskWithEdit(ctx, userId, stor, "", items, msg.ID)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
stors := storage.GetUserStorages(ctx, userId)
|
||||||
|
markup, err := msgelem.BuildAddSelectStorageKeyboard(stors, tcbdata.Add{
|
||||||
|
Files: items,
|
||||||
|
AsBatch: len(items) > 1,
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
logger.Errorf("Failed to build storage selection keyboard: %s", err)
|
||||||
|
ctx.EditMessage(userId, &tg.MessagesEditMessageRequest{
|
||||||
|
ID: msg.ID,
|
||||||
|
Message: i18n.T(i18nk.BotMsgMediaGroupErrorBuildStorageSelectKeyboardFailed, map[string]any{
|
||||||
|
"Error": err.Error(),
|
||||||
|
}),
|
||||||
|
})
|
||||||
|
return
|
||||||
|
}
|
||||||
|
ctx.EditMessage(userId, &tg.MessagesEditMessageRequest{
|
||||||
|
ID: msg.ID,
|
||||||
|
Message: i18n.T(i18nk.BotMsgMediaGroupInfoGroupFoundFilesSelectStorage, map[string]any{
|
||||||
|
"Count": len(items),
|
||||||
|
}),
|
||||||
|
ReplyMarkup: markup,
|
||||||
|
})
|
||||||
|
}
|
||||||
@@ -4,6 +4,9 @@ import (
|
|||||||
"github.com/celestix/gotgproto/dispatcher"
|
"github.com/celestix/gotgproto/dispatcher"
|
||||||
"github.com/celestix/gotgproto/ext"
|
"github.com/celestix/gotgproto/ext"
|
||||||
"github.com/duke-git/lancet/v2/slice"
|
"github.com/duke-git/lancet/v2/slice"
|
||||||
|
"github.com/krau/SaveAny-Bot/client/bot/handlers/utils/dirutil"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/i18n"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/i18n/i18nk"
|
||||||
"github.com/krau/SaveAny-Bot/config"
|
"github.com/krau/SaveAny-Bot/config"
|
||||||
"github.com/krau/SaveAny-Bot/database"
|
"github.com/krau/SaveAny-Bot/database"
|
||||||
"github.com/krau/SaveAny-Bot/storage"
|
"github.com/krau/SaveAny-Bot/storage"
|
||||||
@@ -11,12 +14,8 @@ import (
|
|||||||
|
|
||||||
func checkPermission(ctx *ext.Context, update *ext.Update) error {
|
func checkPermission(ctx *ext.Context, update *ext.Update) error {
|
||||||
userID := update.GetUserChat().GetID()
|
userID := update.GetUserChat().GetID()
|
||||||
if !slice.Contain(config.Cfg.GetUsersID(), userID) {
|
if !slice.Contain(config.C().GetUsersID(), userID) {
|
||||||
const noPermissionText string = `
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgCommonErrorNoPermission, nil)), nil)
|
||||||
您不在白名单中, 无法使用此 Bot.
|
|
||||||
您可以部署自己的实例: https://github.com/krau/SaveAny-Bot
|
|
||||||
`
|
|
||||||
ctx.Reply(update, ext.ReplyTextString(noPermissionText), nil)
|
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -28,21 +27,35 @@ func handleSilentMode(next func(*ext.Context, *ext.Update) error, handler func(*
|
|||||||
userID := update.GetUserChat().GetID()
|
userID := update.GetUserChat().GetID()
|
||||||
user, err := database.GetUserByChatID(ctx, userID)
|
user, err := database.GetUserByChatID(ctx, userID)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
ctx.Reply(update, ext.ReplyTextString("获取用户信息失败: "+err.Error()), nil)
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgCommonErrorGetUserInfoFailed, map[string]any{
|
||||||
|
"Error": err.Error(),
|
||||||
|
})), nil)
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
if !user.Silent {
|
if !user.Silent {
|
||||||
return next(ctx, update)
|
return next(ctx, update)
|
||||||
}
|
}
|
||||||
if user.DefaultStorage == "" {
|
if user.DefaultStorage == "" {
|
||||||
ctx.Reply(update, ext.ReplyTextString("您已开启静默模式, 但未设置默认存储端, 请先使用 /storage 设置"), nil)
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgCommonErrorDefaultStorageNotSet, nil)), nil)
|
||||||
return next(ctx, update)
|
return next(ctx, update)
|
||||||
}
|
}
|
||||||
stor, err := storage.GetStorageByUserIDAndName(ctx, userID, user.DefaultStorage)
|
stor, err := storage.GetStorageByUserIDAndName(ctx, userID, user.DefaultStorage)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
ctx.Reply(update, ext.ReplyTextString("获取默认存储失败: "+err.Error()), nil)
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgCommonErrorGetStorageFailed, map[string]any{
|
||||||
|
"Error": err.Error(),
|
||||||
|
})), nil)
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
|
if user.DefaultDir != 0 {
|
||||||
|
dir, err := database.GetDirByID(ctx, user.DefaultDir)
|
||||||
|
if err != nil {
|
||||||
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgCommonErrorGetDirFailed, map[string]any{
|
||||||
|
"Error": err.Error(),
|
||||||
|
})), nil)
|
||||||
|
return next(ctx, update)
|
||||||
|
}
|
||||||
|
ctx.Context = dirutil.WithContext(ctx.Context, dir)
|
||||||
|
}
|
||||||
ctx.Context = storage.WithContext(ctx.Context, stor)
|
ctx.Context = storage.WithContext(ctx.Context, stor)
|
||||||
return handler(ctx, update)
|
return handler(ctx, update)
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -4,13 +4,20 @@ package handlers
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"errors"
|
"errors"
|
||||||
|
"path"
|
||||||
|
"strings"
|
||||||
|
|
||||||
"github.com/celestix/gotgproto/dispatcher"
|
"github.com/celestix/gotgproto/dispatcher"
|
||||||
"github.com/celestix/gotgproto/ext"
|
"github.com/celestix/gotgproto/ext"
|
||||||
"github.com/charmbracelet/log"
|
"github.com/charmbracelet/log"
|
||||||
"github.com/gotd/td/tg"
|
"github.com/gotd/td/tg"
|
||||||
|
"github.com/krau/SaveAny-Bot/client/bot/handlers/utils/dirutil"
|
||||||
"github.com/krau/SaveAny-Bot/client/bot/handlers/utils/msgelem"
|
"github.com/krau/SaveAny-Bot/client/bot/handlers/utils/msgelem"
|
||||||
"github.com/krau/SaveAny-Bot/client/bot/handlers/utils/shortcut"
|
"github.com/krau/SaveAny-Bot/client/bot/handlers/utils/shortcut"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/i18n"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/i18n/i18nk"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/utils/fsutil"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/utils/tgutil"
|
||||||
"github.com/krau/SaveAny-Bot/parsers"
|
"github.com/krau/SaveAny-Bot/parsers"
|
||||||
"github.com/krau/SaveAny-Bot/pkg/enums/tasktype"
|
"github.com/krau/SaveAny-Bot/pkg/enums/tasktype"
|
||||||
"github.com/krau/SaveAny-Bot/pkg/tcbdata"
|
"github.com/krau/SaveAny-Bot/pkg/tcbdata"
|
||||||
@@ -20,16 +27,47 @@ import (
|
|||||||
func handleTextMessage(ctx *ext.Context, u *ext.Update) error {
|
func handleTextMessage(ctx *ext.Context, u *ext.Update) error {
|
||||||
logger := log.FromContext(ctx)
|
logger := log.FromContext(ctx)
|
||||||
text := u.EffectiveMessage.Text
|
text := u.EffectiveMessage.Text
|
||||||
item, err := parsers.ParseWithContext(ctx, text)
|
entityUrls := tgutil.ExtractMessageEntityUrls(u.EffectiveMessage.Message)
|
||||||
|
if len(entityUrls) > 0 {
|
||||||
|
text += "\n" + strings.Join(entityUrls, "\n")
|
||||||
|
}
|
||||||
|
// read lines and remove empty lines & duplicates
|
||||||
|
lines := strings.Split(text, "\n")
|
||||||
|
seen := make(map[string]struct{})
|
||||||
|
var processedLines []string
|
||||||
|
for _, line := range lines {
|
||||||
|
line = strings.TrimSpace(line)
|
||||||
|
if line == "" {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if _, ok := seen[line]; ok {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
seen[line] = struct{}{}
|
||||||
|
processedLines = append(processedLines, line)
|
||||||
|
}
|
||||||
|
source := strings.TrimSpace(strings.Join(processedLines, "\n"))
|
||||||
|
ok, pser := parsers.CanHandle(source)
|
||||||
|
if !ok {
|
||||||
|
return dispatcher.EndGroups
|
||||||
|
}
|
||||||
|
msg, err := ctx.Reply(u, ext.ReplyTextString(i18n.T(i18nk.BotMsgParseInfoParsing, nil)), nil)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
item, err := pser.Parse(ctx, source)
|
||||||
if errors.Is(err, parsers.ErrNoParserFound) {
|
if errors.Is(err, parsers.ErrNoParserFound) {
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Error("Failed to parse text", "error", err)
|
logger.Error("Failed to parse text", "error", err)
|
||||||
ctx.Reply(u, ext.ReplyTextString("Failed to parse text: "+err.Error()), nil)
|
ctx.Reply(u, ext.ReplyTextString(i18n.T(i18nk.BotMsgParseErrorParseTextFailed, map[string]any{
|
||||||
|
"Error": err.Error(),
|
||||||
|
})), nil)
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
logger.Debug("Parsed item from text message", "text", text, "item", item)
|
logger.Debug("Parsed item from text message", "title", item.Title, "url", item.URL)
|
||||||
userID := u.GetUserChat().GetID()
|
userID := u.GetUserChat().GetID()
|
||||||
markup, err := msgelem.BuildAddSelectStorageKeyboard(storage.GetUserStorages(ctx, userID), tcbdata.Add{
|
markup, err := msgelem.BuildAddSelectStorageKeyboard(storage.GetUserStorages(ctx, userID), tcbdata.Add{
|
||||||
TaskType: tasktype.TaskTypeParseditem,
|
TaskType: tasktype.TaskTypeParseditem,
|
||||||
@@ -37,23 +75,24 @@ func handleTextMessage(ctx *ext.Context, u *ext.Update) error {
|
|||||||
})
|
})
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Errorf("Failed to build storage selection keyboard: %s", err)
|
logger.Errorf("Failed to build storage selection keyboard: %s", err)
|
||||||
ctx.Reply(u, ext.ReplyTextString("Failed to build storage selection keyboard: "+err.Error()), nil)
|
ctx.Reply(u, ext.ReplyTextString(i18n.T(i18nk.BotMsgParseErrorBuildStorageSelectKeyboardFailed, map[string]any{
|
||||||
|
"Error": err.Error(),
|
||||||
|
})), nil)
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
text, entities, err := msgelem.BuildParsedTextEntity(*item)
|
text, entities, err := msgelem.BuildParsedTextEntity(*item)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Errorf("Failed to build parsed text entity: %s", err)
|
logger.Errorf("Failed to build parsed text entity: %s", err)
|
||||||
ctx.Reply(u, ext.ReplyTextString("Failed to build parsed text entity: "+err.Error()), nil)
|
ctx.Reply(u, ext.ReplyTextString(i18n.T(i18nk.BotMsgParseErrorBuildParsedTextEntityFailed, map[string]any{
|
||||||
|
"Error": err.Error(),
|
||||||
|
})), nil)
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
ctx.SendMessage(userID, &tg.MessagesSendMessageRequest{
|
ctx.EditMessage(userID, &tg.MessagesEditMessageRequest{
|
||||||
Message: text,
|
Message: text,
|
||||||
ReplyMarkup: markup,
|
ReplyMarkup: markup,
|
||||||
Entities: entities,
|
Entities: entities,
|
||||||
ReplyTo: &tg.InputReplyToMessage{
|
ID: msg.ID,
|
||||||
ReplyToMsgID: u.EffectiveMessage.ID,
|
|
||||||
ReplyToPeerID: u.GetUserChat().AsInputPeer(),
|
|
||||||
},
|
|
||||||
})
|
})
|
||||||
|
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
@@ -62,11 +101,6 @@ func handleTextMessage(ctx *ext.Context, u *ext.Update) error {
|
|||||||
func handleSilentSaveText(ctx *ext.Context, u *ext.Update) error {
|
func handleSilentSaveText(ctx *ext.Context, u *ext.Update) error {
|
||||||
logger := log.FromContext(ctx)
|
logger := log.FromContext(ctx)
|
||||||
stor := storage.FromContext(ctx)
|
stor := storage.FromContext(ctx)
|
||||||
if stor == nil {
|
|
||||||
logger.Warn("Context storage is nil")
|
|
||||||
ctx.Reply(u, ext.ReplyTextString("未找到存储"), nil)
|
|
||||||
return dispatcher.EndGroups
|
|
||||||
}
|
|
||||||
text := u.EffectiveMessage.Text
|
text := u.EffectiveMessage.Text
|
||||||
if text == "" {
|
if text == "" {
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
@@ -77,15 +111,19 @@ func handleSilentSaveText(ctx *ext.Context, u *ext.Update) error {
|
|||||||
}
|
}
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Error("Failed to parse text", "error", err)
|
logger.Error("Failed to parse text", "error", err)
|
||||||
ctx.Reply(u, ext.ReplyTextString("Failed to parse text: "+err.Error()), nil)
|
ctx.Reply(u, ext.ReplyTextString(i18n.T(i18nk.BotMsgParseErrorParseTextFailed, map[string]any{
|
||||||
|
"Error": err.Error(),
|
||||||
|
})), nil)
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
logger.Debug("Parsed item from text message", "text", text, "item", item)
|
logger.Debug("Parsed item from text message", "title", item.Title, "url", item.URL)
|
||||||
userID := u.GetUserChat().GetID()
|
userID := u.GetUserChat().GetID()
|
||||||
text, entities, err := msgelem.BuildParsedTextEntity(*item)
|
text, entities, err := msgelem.BuildParsedTextEntity(*item)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Errorf("Failed to build parsed text entity: %s", err)
|
logger.Errorf("Failed to build parsed text entity: %s", err)
|
||||||
ctx.Reply(u, ext.ReplyTextString("Failed to build parsed text entity: "+err.Error()), nil)
|
ctx.Reply(u, ext.ReplyTextString(i18n.T(i18nk.BotMsgParseErrorBuildParsedTextEntityFailed, map[string]any{
|
||||||
|
"Error": err.Error(),
|
||||||
|
})), nil)
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
msg, err := ctx.SendMessage(userID, &tg.MessagesSendMessageRequest{
|
msg, err := ctx.SendMessage(userID, &tg.MessagesSendMessageRequest{
|
||||||
@@ -100,5 +138,12 @@ func handleSilentSaveText(ctx *ext.Context, u *ext.Update) error {
|
|||||||
logger.Errorf("Failed to send message: %s", err)
|
logger.Errorf("Failed to send message: %s", err)
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
return shortcut.CreateAndAddParsedTaskWithEdit(ctx, stor, "", item, msg.ID, userID)
|
dirPath := ""
|
||||||
|
if len(item.Resources) > 1 {
|
||||||
|
dirPath = fsutil.NormalizePathname(item.Title)
|
||||||
|
}
|
||||||
|
if p := dirutil.PathFromContext(ctx); p != "" {
|
||||||
|
dirPath = path.Join(p, dirPath)
|
||||||
|
}
|
||||||
|
return shortcut.CreateAndAddParsedTaskWithEdit(ctx, stor, dirPath, item, msg.ID, userID)
|
||||||
}
|
}
|
||||||
|
|||||||
101
client/bot/handlers/parser.go
Normal file
101
client/bot/handlers/parser.go
Normal file
@@ -0,0 +1,101 @@
|
|||||||
|
package handlers
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
"github.com/celestix/gotgproto/dispatcher"
|
||||||
|
"github.com/celestix/gotgproto/ext"
|
||||||
|
"github.com/gotd/td/tg"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/i18n"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/i18n/i18nk"
|
||||||
|
"github.com/krau/SaveAny-Bot/config"
|
||||||
|
"github.com/krau/SaveAny-Bot/parsers"
|
||||||
|
)
|
||||||
|
|
||||||
|
func handleParserCmd(ctx *ext.Context, u *ext.Update) error {
|
||||||
|
args := strings.Split(u.EffectiveMessage.Text, " ")
|
||||||
|
help := i18n.T(i18nk.BotMsgParserHelpText, nil)
|
||||||
|
if len(args) < 2 {
|
||||||
|
ctx.Reply(u, ext.ReplyTextString(help), nil)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
switch args[1] {
|
||||||
|
// case "list":
|
||||||
|
// return handleParserListCmd(ctx, u)
|
||||||
|
case "install":
|
||||||
|
return handleParserInstallCmd(ctx, u)
|
||||||
|
// case "uninstall":
|
||||||
|
// return handleParserUninstallCmd(ctx, u)
|
||||||
|
default:
|
||||||
|
}
|
||||||
|
return dispatcher.EndGroups
|
||||||
|
}
|
||||||
|
|
||||||
|
func handleParserInstallCmd(ctx *ext.Context, u *ext.Update) error {
|
||||||
|
if !config.C().Parser.PluginEnable {
|
||||||
|
ctx.Reply(u, ext.ReplyTextString(i18n.T(i18nk.BotMsgParserPluginNotEnabled, nil)), nil)
|
||||||
|
return dispatcher.EndGroups
|
||||||
|
}
|
||||||
|
if u.EffectiveMessage.ReplyToMessage == nil || u.EffectiveMessage.ReplyToMessage.Media == nil {
|
||||||
|
ctx.Reply(u, ext.ReplyTextString(i18n.T(i18nk.BotMsgParserPromptReplyWithParserFile, nil)), nil)
|
||||||
|
return dispatcher.EndGroups
|
||||||
|
}
|
||||||
|
media := u.EffectiveMessage.ReplyToMessage.Media
|
||||||
|
document, ok := media.(*tg.MessageMediaDocument)
|
||||||
|
if !ok {
|
||||||
|
ctx.Reply(u, ext.ReplyTextString(i18n.T(i18nk.BotMsgParserErrorNoValidFileInReply, nil)), nil)
|
||||||
|
return dispatcher.EndGroups
|
||||||
|
}
|
||||||
|
value, ok := document.GetDocument()
|
||||||
|
if !ok {
|
||||||
|
ctx.Reply(u, ext.ReplyTextString(i18n.T(i18nk.BotMsgParserErrorNoValidFileInReply, nil)), nil)
|
||||||
|
return dispatcher.EndGroups
|
||||||
|
}
|
||||||
|
doc, ok := value.AsNotEmpty()
|
||||||
|
if !ok {
|
||||||
|
ctx.Reply(u, ext.ReplyTextString(i18n.T(i18nk.BotMsgParserErrorNoValidFileInReply, nil)), nil)
|
||||||
|
return dispatcher.EndGroups
|
||||||
|
}
|
||||||
|
if !strings.HasPrefix(doc.MimeType, "text/") {
|
||||||
|
ctx.Reply(u, ext.ReplyTextString(i18n.T(i18nk.BotMsgParserErrorWrongFileType, nil)), nil)
|
||||||
|
return dispatcher.EndGroups
|
||||||
|
}
|
||||||
|
if doc.Size > 1024*1024*10 {
|
||||||
|
ctx.Reply(u, ext.ReplyTextString(i18n.T(i18nk.BotMsgParserErrorFileTooLarge, nil)), nil)
|
||||||
|
return dispatcher.EndGroups
|
||||||
|
}
|
||||||
|
var fileName string
|
||||||
|
for _, attr := range doc.Attributes {
|
||||||
|
if fileNameAttr, ok := attr.(*tg.DocumentAttributeFilename); ok {
|
||||||
|
fileName = fileNameAttr.FileName
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if fileName == "" {
|
||||||
|
ctx.Reply(u, ext.ReplyTextString(i18n.T(i18nk.BotMsgParserErrorGetFilenameFailed, nil)), nil)
|
||||||
|
return dispatcher.EndGroups
|
||||||
|
}
|
||||||
|
if !strings.HasSuffix(fileName, ".js") {
|
||||||
|
ctx.Reply(u, ext.ReplyTextString(i18n.T(i18nk.BotMsgParserErrorOnlyJsSupported, nil)), nil)
|
||||||
|
return dispatcher.EndGroups
|
||||||
|
}
|
||||||
|
data := bytes.NewBuffer(nil)
|
||||||
|
_, err := ctx.DownloadMedia(media, ext.DownloadOutputStream{Writer: data}, nil)
|
||||||
|
if err != nil {
|
||||||
|
ctx.Reply(u, ext.ReplyTextString(i18n.T(i18nk.BotMsgParserErrorDownloadFileFailed, map[string]any{
|
||||||
|
"Error": err.Error(),
|
||||||
|
})), nil)
|
||||||
|
return dispatcher.EndGroups
|
||||||
|
}
|
||||||
|
if err := parsers.AddPlugin(ctx, data.String(), fileName); err != nil {
|
||||||
|
ctx.Reply(u, ext.ReplyTextString(i18n.T(i18nk.BotMsgParserErrorInstallPluginFailed, map[string]any{
|
||||||
|
"Error": err.Error(),
|
||||||
|
})), nil)
|
||||||
|
return dispatcher.EndGroups
|
||||||
|
}
|
||||||
|
ctx.Reply(u, ext.ReplyTextString(i18n.T(i18nk.BotMsgParserInfoInstallPluginSuccess, map[string]any{
|
||||||
|
"Name": fileName,
|
||||||
|
})), nil)
|
||||||
|
return dispatcher.EndGroups
|
||||||
|
}
|
||||||
@@ -1,28 +1,49 @@
|
|||||||
package handlers
|
package handlers
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"path"
|
|
||||||
"regexp"
|
"regexp"
|
||||||
"strings"
|
|
||||||
|
|
||||||
"github.com/celestix/gotgproto/dispatcher"
|
"github.com/celestix/gotgproto/dispatcher"
|
||||||
"github.com/celestix/gotgproto/dispatcher/handlers"
|
"github.com/celestix/gotgproto/dispatcher/handlers"
|
||||||
"github.com/celestix/gotgproto/dispatcher/handlers/filters"
|
"github.com/celestix/gotgproto/dispatcher/handlers/filters"
|
||||||
"github.com/celestix/gotgproto/ext"
|
"github.com/celestix/gotgproto/ext"
|
||||||
"github.com/charmbracelet/log"
|
sabotfilters "github.com/krau/SaveAny-Bot/client/bot/handlers/utils/filters"
|
||||||
"github.com/krau/SaveAny-Bot/client/bot/handlers/utils/re"
|
"github.com/krau/SaveAny-Bot/client/bot/handlers/utils/re"
|
||||||
"github.com/krau/SaveAny-Bot/client/bot/handlers/utils/ruleutil"
|
|
||||||
userclient "github.com/krau/SaveAny-Bot/client/user"
|
userclient "github.com/krau/SaveAny-Bot/client/user"
|
||||||
"github.com/krau/SaveAny-Bot/common/utils/tgutil"
|
"github.com/krau/SaveAny-Bot/common/i18n/i18nk"
|
||||||
"github.com/krau/SaveAny-Bot/config"
|
"github.com/krau/SaveAny-Bot/config"
|
||||||
"github.com/krau/SaveAny-Bot/core"
|
|
||||||
"github.com/krau/SaveAny-Bot/core/tasks/tfile"
|
|
||||||
"github.com/krau/SaveAny-Bot/database"
|
|
||||||
"github.com/krau/SaveAny-Bot/pkg/tcbdata"
|
"github.com/krau/SaveAny-Bot/pkg/tcbdata"
|
||||||
"github.com/krau/SaveAny-Bot/storage"
|
|
||||||
"github.com/rs/xid"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
|
type DescCommandHandler struct {
|
||||||
|
Cmd string
|
||||||
|
Desc i18nk.Key
|
||||||
|
handler func(ctx *ext.Context, u *ext.Update) error
|
||||||
|
}
|
||||||
|
|
||||||
|
var CommandHandlers = []DescCommandHandler{
|
||||||
|
{"start", i18nk.BotMsgCmdStart, handleHelpCmd},
|
||||||
|
{"silent", i18nk.BotMsgCmdSilent, handleSilentCmd},
|
||||||
|
{"storage", i18nk.BotMsgCmdStorage, handleStorageCmd},
|
||||||
|
{"dir", i18nk.BotMsgCmdDir, handleDirCmd},
|
||||||
|
{"rule", i18nk.BotMsgCmdRule, handleRuleCmd},
|
||||||
|
{"save", i18nk.BotMsgCmdSave, handleSilentMode(handleSaveCmd, handleSilentSaveReplied)},
|
||||||
|
{"dl", i18nk.BotMsgCmdDl, handleDlCmd},
|
||||||
|
{"aria2dl", i18nk.BotMsgCmdAria2dl, handleAria2DlCmd},
|
||||||
|
{"ytdlp", i18nk.BotMsgCmdYtdlp, handleYtdlpCmd},
|
||||||
|
{"task", i18nk.BotMsgCmdTask, handleTaskCmd},
|
||||||
|
{"cancel", i18nk.BotMsgCmdCancel, handleCancelCmd},
|
||||||
|
{"config", i18nk.BotMsgCmdConfig, handleConfigCmd},
|
||||||
|
{"fnametmpl", i18nk.BotMsgCmdFnametmpl, handleConfigFnameTmpl},
|
||||||
|
{"help", i18nk.BotMsgCmdHelp, handleHelpCmd},
|
||||||
|
{"parser", i18nk.BotMsgCmdParser, handleParserCmd},
|
||||||
|
{"watch", i18nk.BotMsgCmdWatch, handleWatchCmd},
|
||||||
|
{"unwatch", i18nk.BotMsgCmdUnwatch, handleUnwatchCmd},
|
||||||
|
{"lswatch", i18nk.BotMsgCmdLswatch, handleLswatchCmd},
|
||||||
|
{"syncpeers", i18nk.BotMsgCmdSyncpeers, handleSyncpeersCmd},
|
||||||
|
{"update", i18nk.BotMsgCmdUpdate, handleUpdateCmd},
|
||||||
|
}
|
||||||
|
|
||||||
func Register(disp dispatcher.Dispatcher) {
|
func Register(disp dispatcher.Dispatcher) {
|
||||||
disp.AddHandler(handlers.NewMessage(filters.Message.ChatType(filters.ChatTypeChannel), func(ctx *ext.Context, u *ext.Update) error {
|
disp.AddHandler(handlers.NewMessage(filters.Message.ChatType(filters.ChatTypeChannel), func(ctx *ext.Context, u *ext.Update) error {
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
@@ -31,108 +52,20 @@ func Register(disp dispatcher.Dispatcher) {
|
|||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
}))
|
}))
|
||||||
disp.AddHandler(handlers.NewMessage(filters.Message.All, checkPermission))
|
disp.AddHandler(handlers.NewMessage(filters.Message.All, checkPermission))
|
||||||
disp.AddHandler(handlers.NewCommand("start", handleHelpCmd))
|
for _, info := range CommandHandlers {
|
||||||
disp.AddHandler(handlers.NewCommand("help", handleHelpCmd))
|
disp.AddHandler(handlers.NewCommand(info.Cmd, info.handler))
|
||||||
disp.AddHandler(handlers.NewCommand("silent", handleSilentCmd))
|
}
|
||||||
disp.AddHandler(handlers.NewCommand("storage", handleStorageCmd))
|
disp.AddHandler(handlers.NewCallbackQuery(filters.CallbackQuery.Prefix("update"), handleUpdateCallback))
|
||||||
disp.AddHandler(handlers.NewCommand("dir", handleDirCmd))
|
|
||||||
disp.AddHandler(handlers.NewCommand("rule", handleRuleCmd))
|
|
||||||
disp.AddHandler(handlers.NewCommand("watch", handleWatchCmd))
|
|
||||||
disp.AddHandler(handlers.NewCommand("unwatch", handleUnwatchCmd))
|
|
||||||
disp.AddHandler(handlers.NewCommand("save", handleSilentMode(handleSaveCmd, handleSilentSaveReplied)))
|
|
||||||
disp.AddHandler(handlers.NewCallbackQuery(filters.CallbackQuery.Prefix(tcbdata.TypeAdd), handleAddCallback))
|
disp.AddHandler(handlers.NewCallbackQuery(filters.CallbackQuery.Prefix(tcbdata.TypeAdd), handleAddCallback))
|
||||||
disp.AddHandler(handlers.NewCallbackQuery(filters.CallbackQuery.Prefix(tcbdata.TypeSetDefault), handleSetDefaultCallback))
|
disp.AddHandler(handlers.NewCallbackQuery(filters.CallbackQuery.Prefix(tcbdata.TypeSetDefault), handleSetDefaultCallback))
|
||||||
disp.AddHandler(handlers.NewCallbackQuery(filters.CallbackQuery.Prefix("cancel"), handleCancelCallback))
|
disp.AddHandler(handlers.NewCallbackQuery(filters.CallbackQuery.Prefix(tcbdata.TypeCancel), handleCancelCallback))
|
||||||
linkRegexFilter, err := filters.Message.Regex(re.TgMessageLinkRegexString)
|
disp.AddHandler(handlers.NewCallbackQuery(filters.CallbackQuery.Prefix(tcbdata.TypeConfig), handleConfigCallback))
|
||||||
if err != nil {
|
disp.AddHandler(handlers.NewMessage(sabotfilters.RegexUrl(regexp.MustCompile(re.TgMessageLinkRegexString)), handleSilentMode(handleMessageLink, handleSilentSaveLink)))
|
||||||
panic("failed to create regex filter: " + err.Error())
|
disp.AddHandler(handlers.NewMessage(sabotfilters.RegexUrl(regexp.MustCompile(re.TelegraphUrlRegexString)), handleSilentMode(handleTelegraphUrlMessage, handleSilentSaveTelegraph)))
|
||||||
}
|
|
||||||
disp.AddHandler(handlers.NewMessage(linkRegexFilter, handleSilentMode(handleMessageLink, handleSilentSaveLink)))
|
|
||||||
telegraphUrlRegexFilter, err := filters.Message.Regex(re.TelegraphUrlRegexString)
|
|
||||||
if err != nil {
|
|
||||||
panic("failed to create Telegraph URL regex filter: " + err.Error())
|
|
||||||
}
|
|
||||||
disp.AddHandler(handlers.NewMessage(telegraphUrlRegexFilter, handleSilentMode(handleTelegraphUrlMessage, handleSilentSaveTelegraph)))
|
|
||||||
disp.AddHandler(handlers.NewMessage(filters.Message.Media, handleSilentMode(handleMediaMessage, handleSilentSaveMedia)))
|
disp.AddHandler(handlers.NewMessage(filters.Message.Media, handleSilentMode(handleMediaMessage, handleSilentSaveMedia)))
|
||||||
disp.AddHandler(handlers.NewMessage(filters.Message.Text, handleSilentMode(handleTextMessage, handleSilentSaveText)))
|
disp.AddHandler(handlers.NewMessage(filters.Message.Text, handleSilentMode(handleTextMessage, handleSilentSaveText)))
|
||||||
|
|
||||||
if config.Cfg.Telegram.Userbot.Enable {
|
if config.C().Telegram.Userbot.Enable {
|
||||||
go listenMediaMessageEvent(userclient.GetMediaMessageCh())
|
go listenMediaMessageEvent(userclient.GetMediaMessageCh())
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func listenMediaMessageEvent(ch chan userclient.MediaMessageEvent) {
|
|
||||||
logger := log.FromContext(userclient.GetCtx())
|
|
||||||
for event := range ch {
|
|
||||||
logger.Debug("Received media message event", "chat_id", event.ChatID, "file_name", event.File.Name())
|
|
||||||
ctx := event.Ctx
|
|
||||||
file := event.File
|
|
||||||
chats, err := database.GetWatchChatsByChatID(ctx, event.ChatID)
|
|
||||||
if err != nil {
|
|
||||||
logger.Errorf("Failed to get watch chats for chat ID %d: %v", event.ChatID, err)
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
msgText := event.File.Message().GetMessage()
|
|
||||||
for _, chat := range chats {
|
|
||||||
if chat.Filter != "" {
|
|
||||||
filter := strings.Split(chat.Filter, ":")
|
|
||||||
if len(filter) != 2 {
|
|
||||||
logger.Warnf("Invalid filter format in chat %d, skipping", chat.ChatID)
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
filterType := filter[0]
|
|
||||||
filterData := filter[1]
|
|
||||||
switch filterType {
|
|
||||||
case "msgre": // [TODO] enums for filter types
|
|
||||||
if ok, err := regexp.MatchString(filterData, msgText); err != nil {
|
|
||||||
continue
|
|
||||||
} else if !ok {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
default:
|
|
||||||
logger.Warnf("Unsupported filter type %s in chat %d, skipping", filterType, chat.ChatID)
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
}
|
|
||||||
user, err := database.GetUserByID(ctx, chat.UserID)
|
|
||||||
if err != nil {
|
|
||||||
logger.Errorf("Failed to get user by ID %d: %v", chat.UserID, err)
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
if user.DefaultStorage == "" {
|
|
||||||
logger.Warnf("User %d has no default storage set, skipping media message handling", chat.UserID)
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
stor, err := storage.GetStorageByUserIDAndName(ctx, user.ChatID, user.DefaultStorage)
|
|
||||||
if err != nil {
|
|
||||||
logger.Errorf("Failed to get storage by user ID %d and name %s: %v", user.ChatID, user.DefaultStorage, err)
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
var dirPath string
|
|
||||||
if user.ApplyRule && user.Rules != nil {
|
|
||||||
matchedStorageName, matchedDirPath := ruleutil.ApplyRule(ctx, user.Rules, ruleutil.NewInput(file))
|
|
||||||
dirPath = matchedDirPath.String()
|
|
||||||
if matchedStorageName.IsUsable() {
|
|
||||||
stor, err = storage.GetStorageByUserIDAndName(ctx, user.ChatID, matchedStorageName.String())
|
|
||||||
if err != nil {
|
|
||||||
logger.Errorf("Failed to get storage by user ID and name: %s", err)
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
storagePath := stor.JoinStoragePath(path.Join(dirPath, file.Name()))
|
|
||||||
injectCtx := tgutil.ExtWithContext(ctx.Context, ctx)
|
|
||||||
taskid := xid.New().String()
|
|
||||||
task, err := tfile.NewTGFileTask(taskid, injectCtx, file, stor, storagePath, nil)
|
|
||||||
if err != nil {
|
|
||||||
logger.Errorf("create task failed: %s", err)
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
if err := core.AddTask(injectCtx, task); err != nil {
|
|
||||||
logger.Errorf("add task failed: %s", err)
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
logger.Infof("Added media message task for user %d in chat %d: %s", chat.UserID, event.ChatID, file.Name())
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|||||||
@@ -10,18 +10,21 @@ import (
|
|||||||
"github.com/charmbracelet/log"
|
"github.com/charmbracelet/log"
|
||||||
"github.com/duke-git/lancet/v2/slice"
|
"github.com/duke-git/lancet/v2/slice"
|
||||||
"github.com/krau/SaveAny-Bot/client/bot/handlers/utils/msgelem"
|
"github.com/krau/SaveAny-Bot/client/bot/handlers/utils/msgelem"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/i18n"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/i18n/i18nk"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/utils/strutil"
|
||||||
"github.com/krau/SaveAny-Bot/database"
|
"github.com/krau/SaveAny-Bot/database"
|
||||||
"github.com/krau/SaveAny-Bot/pkg/enums/rule"
|
"github.com/krau/SaveAny-Bot/pkg/rule"
|
||||||
)
|
)
|
||||||
|
|
||||||
func handleRuleCmd(ctx *ext.Context, update *ext.Update) error {
|
func handleRuleCmd(ctx *ext.Context, update *ext.Update) error {
|
||||||
logger := log.FromContext(ctx)
|
logger := log.FromContext(ctx)
|
||||||
args := strings.Split(update.EffectiveMessage.Text, " ")
|
args := strutil.ParseArgsRespectQuotes(update.EffectiveMessage.Text)
|
||||||
userChatID := update.GetUserChat().GetID()
|
userChatID := update.GetUserChat().GetID()
|
||||||
user, err := database.GetUserByChatID(ctx, userChatID)
|
user, err := database.GetUserByChatID(ctx, userChatID)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Errorf("获取用户规则失败: %s", err)
|
logger.Errorf("Failed to get user rules: %s", err)
|
||||||
ctx.Reply(update, ext.ReplyTextString("获取用户规则失败"), nil)
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgRuleErrorGetUserRulesFailed, nil)), nil)
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
if len(args) < 2 {
|
if len(args) < 2 {
|
||||||
@@ -33,11 +36,14 @@ func handleRuleCmd(ctx *ext.Context, update *ext.Update) error {
|
|||||||
// /rule switch
|
// /rule switch
|
||||||
applyRule := !user.ApplyRule
|
applyRule := !user.ApplyRule
|
||||||
if err := database.UpdateUserApplyRule(ctx, user.ChatID, applyRule); err != nil {
|
if err := database.UpdateUserApplyRule(ctx, user.ChatID, applyRule); err != nil {
|
||||||
logger.Errorf("更新用户失败: %s", err)
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgRuleErrorUpdateUserFailed, nil)), nil)
|
||||||
ctx.Reply(update, ext.ReplyTextString("更新用户失败"), nil)
|
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
ctx.Reply(update, ext.ReplyTextString(fmt.Sprintf("已%s规则模式", map[bool]string{true: "启用", false: "禁用"}[applyRule])), nil)
|
if applyRule {
|
||||||
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgRuleInfoRuleModeEnabled, nil)), nil)
|
||||||
|
} else {
|
||||||
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgRuleInfoRuleModeDisabled, nil)), nil)
|
||||||
|
}
|
||||||
case "add":
|
case "add":
|
||||||
// /rule add <type> <data> <storage> <dirpath>
|
// /rule add <type> <data> <storage> <dirpath>
|
||||||
if len(args) < 6 {
|
if len(args) < 6 {
|
||||||
@@ -51,10 +57,13 @@ func handleRuleCmd(ctx *ext.Context, update *ext.Update) error {
|
|||||||
return t, nil
|
return t, nil
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
return rule.RuleType(""), fmt.Errorf("无效的规则类型: %s\n可用: %v", ruleTypeArg, slice.Join(rule.Values(), ", "))
|
return rule.RuleType(""), fmt.Errorf("invalid rule type: %s\navailable: %v", ruleTypeArg, slice.Join(rule.Values(), ", "))
|
||||||
}()
|
}()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
ctx.Reply(update, ext.ReplyTextString(err.Error()), nil)
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgRuleErrorInvalidRuleType, map[string]any{
|
||||||
|
"Type": ruleTypeArg,
|
||||||
|
"Available": slice.Join(rule.Values(), ", "),
|
||||||
|
})), nil)
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -70,29 +79,29 @@ func handleRuleCmd(ctx *ext.Context, update *ext.Update) error {
|
|||||||
UserID: user.ID,
|
UserID: user.ID,
|
||||||
}
|
}
|
||||||
if err := database.CreateRule(ctx, rd); err != nil {
|
if err := database.CreateRule(ctx, rd); err != nil {
|
||||||
logger.Errorf("创建规则失败: %s", err)
|
logger.Errorf("failed to create rule: %s", err)
|
||||||
ctx.Reply(update, ext.ReplyTextString("创建规则失败"), nil)
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgRuleErrorCreateRuleFailed, nil)), nil)
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
ctx.Reply(update, ext.ReplyTextString("创建规则成功"), nil)
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgRuleInfoCreateRuleSuccess, nil)), nil)
|
||||||
case "del":
|
case "del":
|
||||||
// /rule del <id>
|
// /rule del <id>
|
||||||
if len(args) < 3 {
|
if len(args) < 3 {
|
||||||
ctx.Reply(update, ext.ReplyTextString("请提供规则ID"), nil)
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgRulePromptProvideRuleId, nil)), nil)
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
ruleID := args[2]
|
ruleID := args[2]
|
||||||
id, err := strconv.Atoi(ruleID)
|
id, err := strconv.Atoi(ruleID)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
ctx.Reply(update, ext.ReplyTextString("无效的规则ID"), nil)
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgRuleErrorInvalidRuleId, nil)), nil)
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
if err := database.DeleteRule(ctx, uint(id)); err != nil {
|
if err := database.DeleteRule(ctx, uint(id)); err != nil {
|
||||||
logger.Errorf("删除规则失败: %s", err)
|
logger.Errorf("failed to delete rule %d: %s", id, err)
|
||||||
ctx.Reply(update, ext.ReplyTextString("删除规则失败"), nil)
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgRuleErrorDeleteRuleFailed, nil)), nil)
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
ctx.Reply(update, ext.ReplyTextString("删除规则成功"), nil)
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgRuleInfoDeleteRuleSuccess, nil)), nil)
|
||||||
default:
|
default:
|
||||||
ctx.Reply(update, ext.ReplyTextStyledTextArray(msgelem.BuildRuleHelpStyling(user.ApplyRule, user.Rules)), nil)
|
ctx.Reply(update, ext.ReplyTextStyledTextArray(msgelem.BuildRuleHelpStyling(user.ApplyRule, user.Rules)), nil)
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
|
|||||||
@@ -1,19 +1,24 @@
|
|||||||
package handlers
|
package handlers
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
|
||||||
"regexp"
|
"regexp"
|
||||||
"strings"
|
"strings"
|
||||||
|
|
||||||
"github.com/celestix/gotgproto/dispatcher"
|
"github.com/celestix/gotgproto/dispatcher"
|
||||||
"github.com/celestix/gotgproto/ext"
|
"github.com/celestix/gotgproto/ext"
|
||||||
"github.com/charmbracelet/log"
|
"github.com/charmbracelet/log"
|
||||||
|
"github.com/duke-git/lancet/v2/validator"
|
||||||
"github.com/gotd/td/tg"
|
"github.com/gotd/td/tg"
|
||||||
|
"github.com/krau/SaveAny-Bot/client/bot/handlers/utils/dirutil"
|
||||||
"github.com/krau/SaveAny-Bot/client/bot/handlers/utils/mediautil"
|
"github.com/krau/SaveAny-Bot/client/bot/handlers/utils/mediautil"
|
||||||
"github.com/krau/SaveAny-Bot/client/bot/handlers/utils/msgelem"
|
"github.com/krau/SaveAny-Bot/client/bot/handlers/utils/msgelem"
|
||||||
"github.com/krau/SaveAny-Bot/client/bot/handlers/utils/shortcut"
|
"github.com/krau/SaveAny-Bot/client/bot/handlers/utils/shortcut"
|
||||||
|
"github.com/krau/SaveAny-Bot/client/user"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/i18n"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/i18n/i18nk"
|
||||||
"github.com/krau/SaveAny-Bot/common/utils/strutil"
|
"github.com/krau/SaveAny-Bot/common/utils/strutil"
|
||||||
"github.com/krau/SaveAny-Bot/common/utils/tgutil"
|
"github.com/krau/SaveAny-Bot/common/utils/tgutil"
|
||||||
|
"github.com/krau/SaveAny-Bot/database"
|
||||||
"github.com/krau/SaveAny-Bot/pkg/tcbdata"
|
"github.com/krau/SaveAny-Bot/pkg/tcbdata"
|
||||||
"github.com/krau/SaveAny-Bot/pkg/tfile"
|
"github.com/krau/SaveAny-Bot/pkg/tfile"
|
||||||
|
|
||||||
@@ -22,27 +27,25 @@ import (
|
|||||||
|
|
||||||
func handleSaveCmd(ctx *ext.Context, update *ext.Update) error {
|
func handleSaveCmd(ctx *ext.Context, update *ext.Update) error {
|
||||||
logger := log.FromContext(ctx)
|
logger := log.FromContext(ctx)
|
||||||
args := strings.Split(string(update.EffectiveMessage.Text), " ")
|
args := strings.Split(update.EffectiveMessage.Text, " ")
|
||||||
if len(args) >= 3 {
|
if len(args) >= 3 {
|
||||||
return handleBatchSave(ctx, update, args[1:])
|
return handleBatchSave(ctx, update, args[1:])
|
||||||
}
|
}
|
||||||
replyTo := update.EffectiveMessage.ReplyToMessage
|
replyTo := update.EffectiveMessage.ReplyToMessage
|
||||||
if replyTo == nil || replyTo.Message == nil {
|
if replyTo == nil || replyTo.Message == nil {
|
||||||
ctx.Reply(update, ext.ReplyTextString(msgelem.SaveHelpText), nil)
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgSaveHelpText)), nil)
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
genFilename := func() string {
|
userDB, err := database.GetUserByChatID(ctx, update.GetUserChat().GetID())
|
||||||
if len(args) > 1 {
|
if err != nil {
|
||||||
return args[1]
|
return err
|
||||||
}
|
|
||||||
filename := tgutil.GenFileNameFromMessage(*replyTo.Message)
|
|
||||||
return filename
|
|
||||||
}()
|
|
||||||
option := tfile.WithNameIfEmpty(genFilename)
|
|
||||||
if len(args) > 1 {
|
|
||||||
option = tfile.WithName(genFilename)
|
|
||||||
}
|
}
|
||||||
msg, file, err := shortcut.GetFileFromMessageWithReply(ctx, update, replyTo.Message, option)
|
opts := mediautil.TfileOptions(ctx, userDB, replyTo.Message)
|
||||||
|
if len(args) > 1 {
|
||||||
|
// custom filename via command arg
|
||||||
|
opts = append(opts, tfile.WithName(strings.Join(args[1:], " ")))
|
||||||
|
}
|
||||||
|
msg, file, err := shortcut.GetFileFromMessageWithReply(ctx, update, replyTo.Message, opts...)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
@@ -50,8 +53,8 @@ func handleSaveCmd(ctx *ext.Context, update *ext.Update) error {
|
|||||||
stors := storage.GetUserStorages(ctx, userId)
|
stors := storage.GetUserStorages(ctx, userId)
|
||||||
req, err := msgelem.BuildAddOneSelectStorageMessage(ctx, stors, file, msg.ID)
|
req, err := msgelem.BuildAddOneSelectStorageMessage(ctx, stors, file, msg.ID)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Errorf("构建存储选择消息失败: %s", err)
|
logger.Errorf("Failed to build storage selection message: %s", err)
|
||||||
ctx.Reply(update, ext.ReplyTextString("构建存储选择消息失败: "+err.Error()), nil)
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgCommonErrorBuildStorageSelectMessageFailed, map[string]any{"Error": err.Error()})), nil)
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
ctx.EditMessage(update.EffectiveChat().GetID(), req)
|
ctx.EditMessage(update.EffectiveChat().GetID(), req)
|
||||||
@@ -63,34 +66,26 @@ func handleSilentSaveReplied(ctx *ext.Context, update *ext.Update) error {
|
|||||||
if len(args) >= 3 {
|
if len(args) >= 3 {
|
||||||
return handleBatchSave(ctx, update, args[1:])
|
return handleBatchSave(ctx, update, args[1:])
|
||||||
}
|
}
|
||||||
logger := log.FromContext(ctx)
|
|
||||||
stor := storage.FromContext(ctx)
|
stor := storage.FromContext(ctx)
|
||||||
if stor == nil {
|
|
||||||
logger.Warn("Context storage is nil")
|
|
||||||
ctx.Reply(update, ext.ReplyTextString("未找到存储"), nil)
|
|
||||||
return dispatcher.EndGroups
|
|
||||||
}
|
|
||||||
replyTo := update.EffectiveMessage.ReplyToMessage
|
replyTo := update.EffectiveMessage.ReplyToMessage
|
||||||
if replyTo == nil || replyTo.Message == nil {
|
if replyTo == nil || replyTo.Message == nil {
|
||||||
ctx.Reply(update, ext.ReplyTextString(msgelem.SaveHelpText), nil)
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgSaveHelpText)), nil)
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
genFilename := func() string {
|
userDB, err := database.GetUserByChatID(ctx, update.GetUserChat().GetID())
|
||||||
if len(args) > 1 {
|
|
||||||
return args[1]
|
|
||||||
}
|
|
||||||
filename := tgutil.GenFileNameFromMessage(*replyTo.Message)
|
|
||||||
return filename
|
|
||||||
}()
|
|
||||||
option := tfile.WithNameIfEmpty(genFilename)
|
|
||||||
if len(args) > 1 {
|
|
||||||
option = tfile.WithName(genFilename)
|
|
||||||
}
|
|
||||||
msg, file, err := shortcut.GetFileFromMessageWithReply(ctx, update, replyTo.Message, option)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
return shortcut.CreateAndAddTGFileTaskWithEdit(ctx, update.GetUserChat().GetID(), stor, "", file, msg.GetID())
|
opts := mediautil.TfileOptions(ctx, userDB, replyTo.Message)
|
||||||
|
if len(args) > 1 {
|
||||||
|
// custom filename via command arg
|
||||||
|
opts = append(opts, tfile.WithName(strings.Join(args[1:], " ")))
|
||||||
|
}
|
||||||
|
msg, file, err := shortcut.GetFileFromMessageWithReply(ctx, update, replyTo.Message, opts...)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
return shortcut.CreateAndAddTGFileTaskWithEdit(ctx, update.GetUserChat().GetID(), stor, dirutil.PathFromContext(ctx), file, msg.GetID())
|
||||||
}
|
}
|
||||||
|
|
||||||
func handleBatchSave(ctx *ext.Context, update *ext.Update, args []string) error {
|
func handleBatchSave(ctx *ext.Context, update *ext.Update, args []string) error {
|
||||||
@@ -103,35 +98,40 @@ func handleBatchSave(ctx *ext.Context, update *ext.Update, args []string) error
|
|||||||
var err error
|
var err error
|
||||||
filter, err = regexp.Compile(filterStr)
|
filter, err = regexp.Compile(filterStr)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
ctx.Reply(update, ext.ReplyTextString("无效的正则表达式: "+err.Error()), nil)
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgCommonErrorInvalidRegex, map[string]any{"Error": err.Error()})), nil)
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
startID, endID, err := strutil.ParseIntStrRange(msgIdRangeArg, "-")
|
startID, endID, err := strutil.ParseIntStrRange(msgIdRangeArg, "-")
|
||||||
if err != nil {
|
if err != nil {
|
||||||
ctx.Reply(update, ext.ReplyTextString("无效的消息ID范围: "+err.Error()), nil)
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgCommonErrorInvalidMsgIdRange, map[string]any{"Error": err.Error()})), nil)
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
chatID, err := tgutil.ParseChatID(ctx, chatArg)
|
tctx := ctx
|
||||||
|
uctx := user.GetCtx()
|
||||||
|
if uctx != nil && validator.IsIntStr(chatArg) {
|
||||||
|
tctx = uctx
|
||||||
|
}
|
||||||
|
chatID, err := tgutil.ParseChatID(tctx, chatArg)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
ctx.Reply(update, ext.ReplyTextString("无效的ID或用户名: "+err.Error()), nil)
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgCommonErrorInvalidIdOrUsername, map[string]any{"Error": err.Error()})), nil)
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
|
|
||||||
replied, err := ctx.Reply(update, ext.ReplyTextString("正在获取消息..."), nil)
|
replied, err := ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgCommonInfoFetchingMessages)), nil)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.FromContext(ctx).Errorf("回复失败: %s", err)
|
log.FromContext(ctx).Errorf("Failed to reply: %s", err)
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
|
|
||||||
// TODO: generator istead of get all messages
|
// [TODO]: generator istead of get all messages
|
||||||
msgs, err := tgutil.GetMessagesRange(ctx, chatID, int(startID), int(endID))
|
msgs, err := tgutil.GetMessagesRange(tctx, chatID, int(startID), int(endID))
|
||||||
if err != nil {
|
if err != nil {
|
||||||
ctx.Reply(update, ext.ReplyTextString("获取消息失败: "+err.Error()), nil)
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgCommonErrorGetMessagesFailed, map[string]any{"Error": err.Error()})), nil)
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
if len(msgs) == 0 {
|
if len(msgs) == 0 {
|
||||||
ctx.Reply(update, ext.ReplyTextString("没有找到指定范围内的消息"), nil)
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgCommonErrorNoMessagesInRange)), nil)
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
files := make([]tfile.TGFileMessage, 0, len(msgs))
|
files := make([]tfile.TGFileMessage, 0, len(msgs))
|
||||||
@@ -148,9 +148,9 @@ func handleBatchSave(ctx *ext.Context, update *ext.Update, args []string) error
|
|||||||
if !supported {
|
if !supported {
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
file, err := tfile.FromMediaMessage(media, ctx.Raw, msg, tfile.WithNameIfEmpty(tgutil.GenFileNameFromMessage(*msg)))
|
file, err := tfile.FromMediaMessage(media, tctx.Raw, msg, tfile.WithNameIfEmpty(tgutil.GenFileNameFromMessage(*msg)))
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.FromContext(ctx).Errorf("获取文件失败: %s", err)
|
log.FromContext(ctx).Errorf("Failed to get file from message: %s", err)
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
if filter != nil {
|
if filter != nil {
|
||||||
@@ -166,7 +166,7 @@ func handleBatchSave(ctx *ext.Context, update *ext.Update, args []string) error
|
|||||||
files = append(files, file)
|
files = append(files, file)
|
||||||
}
|
}
|
||||||
if len(files) == 0 {
|
if len(files) == 0 {
|
||||||
ctx.Reply(update, ext.ReplyTextString("没有找到指定范围内的可保存消息"), nil)
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgCommonErrorNoSavableMessagesInRange)), nil)
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
stor := storage.FromContext(ctx)
|
stor := storage.FromContext(ctx)
|
||||||
@@ -177,16 +177,16 @@ func handleBatchSave(ctx *ext.Context, update *ext.Update, args []string) error
|
|||||||
Files: files,
|
Files: files,
|
||||||
})
|
})
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.FromContext(ctx).Errorf("构建存储选择键盘失败: %s", err)
|
log.FromContext(ctx).Errorf("Failed to build storage selection keyboard: %s", err)
|
||||||
ctx.EditMessage(update.EffectiveChat().GetID(), &tg.MessagesEditMessageRequest{
|
ctx.EditMessage(update.EffectiveChat().GetID(), &tg.MessagesEditMessageRequest{
|
||||||
ID: replied.ID,
|
ID: replied.ID,
|
||||||
Message: "构建存储选择键盘失败: " + err.Error(),
|
Message: i18n.T(i18nk.BotMsgCommonErrorBuildStorageSelectKeyboardFailed, map[string]any{"Error": err.Error()}),
|
||||||
})
|
})
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
ctx.EditMessage(update.EffectiveChat().GetID(), &tg.MessagesEditMessageRequest{
|
ctx.EditMessage(update.EffectiveChat().GetID(), &tg.MessagesEditMessageRequest{
|
||||||
ID: replied.ID,
|
ID: replied.ID,
|
||||||
Message: fmt.Sprintf("找到 %d 个文件, 请选择存储位置", len(files)),
|
Message: i18n.T(i18nk.BotMsgCommonInfoFoundFilesSelectStorage, map[string]any{"Count": len(files)}),
|
||||||
ReplyMarkup: markup,
|
ReplyMarkup: markup,
|
||||||
})
|
})
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
|
|||||||
@@ -8,6 +8,8 @@ import (
|
|||||||
"github.com/gotd/td/tg"
|
"github.com/gotd/td/tg"
|
||||||
"github.com/krau/SaveAny-Bot/client/bot/handlers/utils/msgelem"
|
"github.com/krau/SaveAny-Bot/client/bot/handlers/utils/msgelem"
|
||||||
"github.com/krau/SaveAny-Bot/common/cache"
|
"github.com/krau/SaveAny-Bot/common/cache"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/i18n"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/i18n/i18nk"
|
||||||
"github.com/krau/SaveAny-Bot/database"
|
"github.com/krau/SaveAny-Bot/database"
|
||||||
"github.com/krau/SaveAny-Bot/pkg/tcbdata"
|
"github.com/krau/SaveAny-Bot/pkg/tcbdata"
|
||||||
"github.com/krau/SaveAny-Bot/storage"
|
"github.com/krau/SaveAny-Bot/storage"
|
||||||
@@ -16,71 +18,113 @@ import (
|
|||||||
func handleSilentCmd(ctx *ext.Context, update *ext.Update) error {
|
func handleSilentCmd(ctx *ext.Context, update *ext.Update) error {
|
||||||
user, err := database.GetUserByChatID(ctx, update.GetUserChat().GetID())
|
user, err := database.GetUserByChatID(ctx, update.GetUserChat().GetID())
|
||||||
if err != nil {
|
if err != nil {
|
||||||
ctx.Reply(update, ext.ReplyTextString("获取用户信息失败: "+err.Error()), nil)
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgCommonErrorGetUserInfoFailed, map[string]any{
|
||||||
|
"Error": err.Error(),
|
||||||
|
})), nil)
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
if !user.Silent && user.DefaultStorage == "" {
|
if !user.Silent && user.DefaultStorage == "" {
|
||||||
ctx.Reply(update, ext.ReplyTextString("请先使用 /storage 设置默认存储位置"), nil)
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgCommonErrorDefaultStorageNotSet, nil)), nil)
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
user.Silent = !user.Silent
|
user.Silent = !user.Silent
|
||||||
if err := database.UpdateUser(ctx, user); err != nil {
|
if err := database.UpdateUser(ctx, user); err != nil {
|
||||||
ctx.Reply(update, ext.ReplyTextString("更新用户信息失败: "+err.Error()), nil)
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgCommonErrorUpdateUserInfoFailed, map[string]any{
|
||||||
|
"Error": err.Error(),
|
||||||
|
})), nil)
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
responseText := "已" + map[bool]string{true: "开启", false: "关闭"}[user.Silent] + "静默模式"
|
if user.Silent {
|
||||||
ctx.Reply(update, ext.ReplyTextString(responseText), nil)
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgCommonInfoSilentModeOn, nil)), nil)
|
||||||
|
} else {
|
||||||
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgCommonInfoSilentModeOff, nil)), nil)
|
||||||
|
}
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
|
|
||||||
func handleSetDefaultCallback(ctx *ext.Context, update *ext.Update) error {
|
func handleSetDefaultCallback(ctx *ext.Context, update *ext.Update) error {
|
||||||
dataid := strings.Split(string(update.CallbackQuery.Data), " ")[1]
|
dataid := strings.Split(string(update.CallbackQuery.Data), " ")[1]
|
||||||
data, ok := cache.Get[tcbdata.SetDefaultStorage](dataid)
|
data, ok := cache.Get[tcbdata.SetDefaultStorage](dataid)
|
||||||
if !ok {
|
|
||||||
|
failedAnswer := func(message string) error {
|
||||||
ctx.AnswerCallback(&tg.MessagesSetBotCallbackAnswerRequest{
|
ctx.AnswerCallback(&tg.MessagesSetBotCallbackAnswerRequest{
|
||||||
QueryID: update.CallbackQuery.GetQueryID(),
|
QueryID: update.CallbackQuery.GetQueryID(),
|
||||||
Alert: true,
|
Alert: true,
|
||||||
Message: "数据已过期",
|
Message: message,
|
||||||
CacheTime: 5,
|
CacheTime: 5,
|
||||||
})
|
})
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if !ok {
|
||||||
|
return failedAnswer(i18n.T(i18nk.BotMsgCommonErrorDataExpired, nil))
|
||||||
|
}
|
||||||
userID := update.CallbackQuery.GetUserID()
|
userID := update.CallbackQuery.GetUserID()
|
||||||
|
|
||||||
storageName := data.StorageName
|
storageName := data.StorageName
|
||||||
selectedStorage, err := storage.GetStorageByUserIDAndName(ctx, userID, storageName)
|
selectedStorage, err := storage.GetStorageByUserIDAndName(ctx, userID, storageName)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
ctx.AnswerCallback(&tg.MessagesSetBotCallbackAnswerRequest{
|
return failedAnswer(i18n.T(i18nk.BotMsgCommonErrorGetStorageFailed, map[string]any{
|
||||||
QueryID: update.CallbackQuery.GetQueryID(),
|
"Error": err.Error(),
|
||||||
Alert: true,
|
}))
|
||||||
Message: "存储获取失败: " + err.Error(),
|
|
||||||
CacheTime: 5,
|
|
||||||
})
|
|
||||||
return dispatcher.EndGroups
|
|
||||||
}
|
}
|
||||||
user, err := database.GetUserByChatID(ctx, userID)
|
user, err := database.GetUserByChatID(ctx, userID)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
ctx.AnswerCallback(&tg.MessagesSetBotCallbackAnswerRequest{
|
return failedAnswer(i18n.T(i18nk.BotMsgCommonErrorGetUserInfoFailed, map[string]any{
|
||||||
QueryID: update.CallbackQuery.GetQueryID(),
|
"Error": err.Error(),
|
||||||
Alert: true,
|
}))
|
||||||
Message: "获取用户信息失败: " + err.Error(),
|
}
|
||||||
CacheTime: 5,
|
var dir *database.Dir
|
||||||
})
|
if data.DirID != 0 {
|
||||||
return dispatcher.EndGroups
|
// 已经选择了文件夹
|
||||||
|
var err error
|
||||||
|
dir, err = database.GetDirByID(ctx, data.DirID)
|
||||||
|
if err != nil {
|
||||||
|
return failedAnswer(i18n.T(i18nk.BotMsgDirErrorGetUserDirsFailed, nil))
|
||||||
|
}
|
||||||
|
user.DefaultDir = dir.ID
|
||||||
|
} else {
|
||||||
|
// 检查是否有可用的文件夹
|
||||||
|
dirs, err := database.GetDirsByUserIDAndStorageName(ctx, user.ID, storageName)
|
||||||
|
if err != nil {
|
||||||
|
return failedAnswer(i18n.T(i18nk.BotMsgCommonErrorGetDirFailed, map[string]any{
|
||||||
|
"Error": err.Error(),
|
||||||
|
}))
|
||||||
|
}
|
||||||
|
if len(dirs) > 0 {
|
||||||
|
// 要求选择文件夹
|
||||||
|
markup, err := msgelem.BuildSetDefaultDirMarkup(ctx, storageName, dirs)
|
||||||
|
if err != nil {
|
||||||
|
return failedAnswer(i18n.T(i18nk.BotMsgCommonErrorBuildDirSelectKeyboardFailed, map[string]any{
|
||||||
|
"Error": err.Error(),
|
||||||
|
}))
|
||||||
|
}
|
||||||
|
ctx.EditMessage(userID, &tg.MessagesEditMessageRequest{
|
||||||
|
ID: update.CallbackQuery.GetMsgID(),
|
||||||
|
Message: i18n.T(i18nk.BotMsgCommonPromptSelectDefaultDir, nil),
|
||||||
|
ReplyMarkup: markup,
|
||||||
|
})
|
||||||
|
return dispatcher.EndGroups
|
||||||
|
}
|
||||||
}
|
}
|
||||||
user.DefaultStorage = selectedStorage.Name()
|
user.DefaultStorage = selectedStorage.Name()
|
||||||
if err := database.UpdateUser(ctx, user); err != nil {
|
if err := database.UpdateUser(ctx, user); err != nil {
|
||||||
ctx.AnswerCallback(&tg.MessagesSetBotCallbackAnswerRequest{
|
return failedAnswer(i18n.T(i18nk.BotMsgCommonErrorUpdateUserInfoFailed, map[string]any{
|
||||||
QueryID: update.CallbackQuery.GetQueryID(),
|
"Error": err.Error(),
|
||||||
Alert: true,
|
}))
|
||||||
Message: "更新用户信息失败: " + err.Error(),
|
}
|
||||||
CacheTime: 5,
|
msg := i18n.T(i18nk.BotMsgCommonInfoDefaultStorageSet, map[string]any{
|
||||||
|
"Name": selectedStorage.Name(),
|
||||||
|
})
|
||||||
|
if dir != nil {
|
||||||
|
msg = i18n.T(i18nk.BotMsgCommonInfoDefaultStorageWithDirSet, map[string]any{
|
||||||
|
"Name": selectedStorage.Name(),
|
||||||
|
"Dir": strings.TrimPrefix(dir.Path, "/"),
|
||||||
})
|
})
|
||||||
return dispatcher.EndGroups
|
|
||||||
}
|
}
|
||||||
ctx.EditMessage(userID, &tg.MessagesEditMessageRequest{
|
ctx.EditMessage(userID, &tg.MessagesEditMessageRequest{
|
||||||
ID: update.CallbackQuery.GetMsgID(),
|
ID: update.CallbackQuery.GetMsgID(),
|
||||||
Message: "已将默认存储位置设置为: " + selectedStorage.Name(),
|
Message: msg,
|
||||||
})
|
})
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
@@ -89,15 +133,17 @@ func handleStorageCmd(ctx *ext.Context, update *ext.Update) error {
|
|||||||
userID := update.GetUserChat().GetID()
|
userID := update.GetUserChat().GetID()
|
||||||
storages := storage.GetUserStorages(ctx, userID)
|
storages := storage.GetUserStorages(ctx, userID)
|
||||||
if len(storages) == 0 {
|
if len(storages) == 0 {
|
||||||
ctx.Reply(update, ext.ReplyTextString("无可用的存储"), nil)
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgCommonErrorNoAvailableStorage, nil)), nil)
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
markup, err := msgelem.BuildSetDefaultStorageMarkup(ctx, userID, storages)
|
markup, err := msgelem.BuildSetDefaultStorageMarkup(ctx, storages)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
ctx.Reply(update, ext.ReplyTextString("获取存储失败: "+err.Error()), nil)
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgCommonErrorGetStorageFailed, map[string]any{
|
||||||
|
"Error": err.Error(),
|
||||||
|
})), nil)
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
ctx.Reply(update, ext.ReplyTextString("请选择要设为默认的存储位置"), &ext.ReplyOpts{
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgCommonPromptSelectDefaultStorage, nil)), &ext.ReplyOpts{
|
||||||
Markup: markup,
|
Markup: markup,
|
||||||
})
|
})
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
|
|||||||
62
client/bot/handlers/sync_peers.go
Normal file
62
client/bot/handlers/sync_peers.go
Normal file
@@ -0,0 +1,62 @@
|
|||||||
|
package handlers
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"sync"
|
||||||
|
|
||||||
|
"github.com/celestix/gotgproto/dispatcher"
|
||||||
|
"github.com/celestix/gotgproto/ext"
|
||||||
|
"github.com/celestix/gotgproto/storage"
|
||||||
|
"github.com/charmbracelet/log"
|
||||||
|
"github.com/gotd/td/telegram/query/dialogs"
|
||||||
|
"github.com/krau/SaveAny-Bot/client/user"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/i18n"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/i18n/i18nk"
|
||||||
|
"github.com/krau/SaveAny-Bot/config"
|
||||||
|
)
|
||||||
|
|
||||||
|
var syncpeerMu sync.Mutex
|
||||||
|
|
||||||
|
func handleSyncpeersCmd(ctx *ext.Context, u *ext.Update) error {
|
||||||
|
if !config.C().Telegram.Userbot.Enable {
|
||||||
|
return dispatcher.EndGroups
|
||||||
|
}
|
||||||
|
syncpeerMu.Lock()
|
||||||
|
defer syncpeerMu.Unlock()
|
||||||
|
uctx := user.GetCtx()
|
||||||
|
if uctx == nil {
|
||||||
|
return dispatcher.EndGroups
|
||||||
|
}
|
||||||
|
ctx.Reply(u, ext.ReplyTextString(i18n.T(i18nk.BotMsgSyncpeersStart)), nil)
|
||||||
|
tapi := uctx.Raw
|
||||||
|
peerStorage := uctx.PeerStorage
|
||||||
|
log.FromContext(ctx).Info("Starting to sync peers...")
|
||||||
|
count := 0
|
||||||
|
err := dialogs.NewQueryBuilder(tapi).GetDialogs().BatchSize(50).ForEach(ctx, func(ctx context.Context, e dialogs.Elem) error {
|
||||||
|
for cid, channel := range e.Entities.Channels() {
|
||||||
|
peerStorage.AddPeer(cid, channel.AccessHash, storage.TypeChannel, channel.Username)
|
||||||
|
count++
|
||||||
|
}
|
||||||
|
for uid, user := range e.Entities.Users() {
|
||||||
|
peerStorage.AddPeer(uid, user.AccessHash, storage.TypeUser, user.Username)
|
||||||
|
count++
|
||||||
|
}
|
||||||
|
for gid := range e.Entities.Chats() {
|
||||||
|
peerStorage.AddPeer(gid, storage.DefaultAccessHash, storage.TypeChat, storage.DefaultUsername)
|
||||||
|
count++
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
log.FromContext(ctx).Error("Failed to sync peers", "error", err)
|
||||||
|
ctx.Reply(u, ext.ReplyTextString(i18n.T(i18nk.BotMsgSyncpeersFailed, map[string]any{
|
||||||
|
"Error": err.Error(),
|
||||||
|
})), nil)
|
||||||
|
return dispatcher.EndGroups
|
||||||
|
}
|
||||||
|
log.FromContext(ctx).Info("Finished syncing peers")
|
||||||
|
ctx.Reply(u, ext.ReplyTextString(i18n.T(i18nk.BotMsgSyncpeersSuccess, map[string]any{
|
||||||
|
"Count": count,
|
||||||
|
})), nil)
|
||||||
|
return dispatcher.EndGroups
|
||||||
|
}
|
||||||
114
client/bot/handlers/tasks.go
Normal file
114
client/bot/handlers/tasks.go
Normal file
@@ -0,0 +1,114 @@
|
|||||||
|
package handlers
|
||||||
|
|
||||||
|
import (
|
||||||
|
"strings"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/celestix/gotgproto/dispatcher"
|
||||||
|
"github.com/celestix/gotgproto/ext"
|
||||||
|
"github.com/charmbracelet/log"
|
||||||
|
"github.com/gotd/td/telegram/message/styling"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/i18n"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/i18n/i18nk"
|
||||||
|
"github.com/krau/SaveAny-Bot/core"
|
||||||
|
)
|
||||||
|
|
||||||
|
func handleTaskCmd(ctx *ext.Context, update *ext.Update) error {
|
||||||
|
logger := log.FromContext(ctx)
|
||||||
|
args := strings.Fields(update.EffectiveMessage.Text)
|
||||||
|
if len(args) == 1 {
|
||||||
|
showRunningTasks(ctx, update)
|
||||||
|
return dispatcher.EndGroups
|
||||||
|
}
|
||||||
|
|
||||||
|
switch args[1] {
|
||||||
|
case "running", "run", "r":
|
||||||
|
showRunningTasks(ctx, update)
|
||||||
|
case "queued", "queue", "q", "waiting":
|
||||||
|
showQueuedTasks(ctx, update)
|
||||||
|
case "cancel", "c":
|
||||||
|
if len(args) < 3 {
|
||||||
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgTasksUsageCancel)), nil)
|
||||||
|
return dispatcher.EndGroups
|
||||||
|
}
|
||||||
|
taskID := args[2]
|
||||||
|
if err := core.CancelTask(ctx, taskID); err != nil {
|
||||||
|
logger.Errorf("Failed to cancel task %s: %v", taskID, err)
|
||||||
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgTasksCancelFailed, map[string]any{"Error": err.Error()})), nil)
|
||||||
|
return dispatcher.EndGroups
|
||||||
|
}
|
||||||
|
ctx.Reply(update, ext.ReplyTextStyledTextArray([]styling.StyledTextOption{
|
||||||
|
styling.Plain(i18n.T(i18nk.BotMsgTasksCancelRequestedPrefix)),
|
||||||
|
styling.Code(taskID),
|
||||||
|
}), nil)
|
||||||
|
default:
|
||||||
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgTasksUsage)), nil)
|
||||||
|
}
|
||||||
|
return dispatcher.EndGroups
|
||||||
|
}
|
||||||
|
|
||||||
|
func showRunningTasks(ctx *ext.Context, update *ext.Update) {
|
||||||
|
tasks := core.GetRunningTasks(ctx)
|
||||||
|
if len(tasks) == 0 {
|
||||||
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgTasksRunningEmpty)), nil)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
opts := make([]styling.StyledTextOption, 0, 2+len(tasks)*4)
|
||||||
|
opts = append(opts,
|
||||||
|
styling.Bold(i18n.T(i18nk.BotMsgTasksRunningTitle)),
|
||||||
|
styling.Plain(i18n.T(i18nk.BotMsgTasksTotalPrefix, map[string]any{"Count": len(tasks)})),
|
||||||
|
)
|
||||||
|
for _, t := range tasks {
|
||||||
|
created := t.Created.In(time.Local).Format("2006-01-02 15:04:05")
|
||||||
|
status := i18n.T(i18nk.BotMsgTasksStatusRunning)
|
||||||
|
if t.Cancelled {
|
||||||
|
status = i18n.T(i18nk.BotMsgTasksStatusCancelRequested)
|
||||||
|
}
|
||||||
|
opts = append(opts,
|
||||||
|
styling.Plain("\n"+i18n.T(i18nk.BotMsgTasksFieldId)),
|
||||||
|
styling.Code(t.ID),
|
||||||
|
styling.Plain("\n"+i18n.T(i18nk.BotMsgTasksFieldTitle)),
|
||||||
|
styling.Code(t.Title),
|
||||||
|
styling.Plain("\n"+i18n.T(i18nk.BotMsgTasksFieldCreated)),
|
||||||
|
styling.Code(created),
|
||||||
|
styling.Plain("\n"+i18n.T(i18nk.BotMsgTasksFieldStatus)),
|
||||||
|
styling.Code(status),
|
||||||
|
)
|
||||||
|
}
|
||||||
|
ctx.Reply(update, ext.ReplyTextStyledTextArray(opts), nil)
|
||||||
|
}
|
||||||
|
|
||||||
|
func showQueuedTasks(ctx *ext.Context, update *ext.Update) {
|
||||||
|
tasks := core.GetQueuedTasks(ctx)
|
||||||
|
if len(tasks) == 0 {
|
||||||
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgTasksQueuedEmpty)), nil)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
opts := make([]styling.StyledTextOption, 0, 2+len(tasks)*3)
|
||||||
|
opts = append(opts,
|
||||||
|
styling.Bold(i18n.T(i18nk.BotMsgTasksQueuedTitle)),
|
||||||
|
styling.Plain(i18n.T(i18nk.BotMsgTasksTotalPrefix, map[string]any{"Count": len(tasks)})),
|
||||||
|
)
|
||||||
|
for _, t := range tasks {
|
||||||
|
created := t.Created.In(time.Local).Format("2006-01-02 15:04:05")
|
||||||
|
status := i18n.T(i18nk.BotMsgTasksStatusQueued)
|
||||||
|
if t.Cancelled {
|
||||||
|
status = i18n.T(i18nk.BotMsgTasksStatusCancelRequested)
|
||||||
|
}
|
||||||
|
opts = append(opts,
|
||||||
|
styling.Plain("\n"+i18n.T(i18nk.BotMsgTasksFieldId)),
|
||||||
|
styling.Code(t.ID),
|
||||||
|
styling.Plain("\n"+i18n.T(i18nk.BotMsgTasksFieldTitle)),
|
||||||
|
styling.Code(t.Title),
|
||||||
|
styling.Plain("\n"+i18n.T(i18nk.BotMsgTasksFieldCreated)),
|
||||||
|
styling.Code(created),
|
||||||
|
styling.Plain("\n"+i18n.T(i18nk.BotMsgTasksFieldStatus)),
|
||||||
|
styling.Code(status),
|
||||||
|
)
|
||||||
|
if len(tasks) > 10 {
|
||||||
|
opts = append(opts, styling.Plain("\n"+i18n.T(i18nk.BotMsgTasksTruncatedNote, map[string]any{"Count": len(tasks)})))
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
ctx.Reply(update, ext.ReplyTextStyledTextArray(opts), nil)
|
||||||
|
}
|
||||||
@@ -2,6 +2,7 @@ package handlers
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
|
"path"
|
||||||
|
|
||||||
"github.com/celestix/gotgproto/dispatcher"
|
"github.com/celestix/gotgproto/dispatcher"
|
||||||
"github.com/celestix/gotgproto/ext"
|
"github.com/celestix/gotgproto/ext"
|
||||||
@@ -9,8 +10,11 @@ import (
|
|||||||
"github.com/gotd/td/telegram/message/entity"
|
"github.com/gotd/td/telegram/message/entity"
|
||||||
"github.com/gotd/td/telegram/message/styling"
|
"github.com/gotd/td/telegram/message/styling"
|
||||||
"github.com/gotd/td/tg"
|
"github.com/gotd/td/tg"
|
||||||
|
"github.com/krau/SaveAny-Bot/client/bot/handlers/utils/dirutil"
|
||||||
"github.com/krau/SaveAny-Bot/client/bot/handlers/utils/msgelem"
|
"github.com/krau/SaveAny-Bot/client/bot/handlers/utils/msgelem"
|
||||||
"github.com/krau/SaveAny-Bot/client/bot/handlers/utils/shortcut"
|
"github.com/krau/SaveAny-Bot/client/bot/handlers/utils/shortcut"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/i18n"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/i18n/i18nk"
|
||||||
"github.com/krau/SaveAny-Bot/pkg/enums/tasktype"
|
"github.com/krau/SaveAny-Bot/pkg/enums/tasktype"
|
||||||
"github.com/krau/SaveAny-Bot/pkg/tcbdata"
|
"github.com/krau/SaveAny-Bot/pkg/tcbdata"
|
||||||
"github.com/krau/SaveAny-Bot/storage"
|
"github.com/krau/SaveAny-Bot/storage"
|
||||||
@@ -32,18 +36,20 @@ func handleTelegraphUrlMessage(ctx *ext.Context, update *ext.Update) error {
|
|||||||
TphPics: result.Pics,
|
TphPics: result.Pics,
|
||||||
})
|
})
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Errorf("构建存储选择键盘失败: %s", err)
|
logger.Errorf("Failed to build storage selection keyboard: %s", err)
|
||||||
ctx.Reply(update, ext.ReplyTextString("构建存储选择键盘失败: "+err.Error()), nil)
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgTelegraphErrorBuildStorageSelectKeyboardFailed, map[string]any{
|
||||||
|
"Error": err.Error(),
|
||||||
|
})), nil)
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
|
|
||||||
eb := entity.Builder{}
|
eb := entity.Builder{}
|
||||||
if err := styling.Perform(&eb,
|
if err := styling.Perform(&eb,
|
||||||
styling.Plain("标题: "),
|
styling.Plain(i18n.T(i18nk.BotMsgTelegraphInfoTitlePrefix, nil)),
|
||||||
styling.Code(result.Page.Title),
|
styling.Code(result.Page.Title),
|
||||||
styling.Plain("\n图片数量: "),
|
styling.Plain(i18n.T(i18nk.BotMsgTelegraphInfoPicCountPrefix, nil)),
|
||||||
styling.Code(fmt.Sprintf("%d", len(result.Pics))),
|
styling.Code(fmt.Sprintf("%d", len(result.Pics))),
|
||||||
styling.Plain("\n请选择存储位置"),
|
styling.Plain(i18n.T(i18nk.BotMsgTelegraphInfoPromptSelectStorage, nil)),
|
||||||
); err != nil {
|
); err != nil {
|
||||||
log.FromContext(ctx).Errorf("Failed to build entity: %s", err)
|
log.FromContext(ctx).Errorf("Failed to build entity: %s", err)
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
@@ -59,18 +65,16 @@ func handleTelegraphUrlMessage(ctx *ext.Context, update *ext.Update) error {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func handleSilentSaveTelegraph(ctx *ext.Context, update *ext.Update) error {
|
func handleSilentSaveTelegraph(ctx *ext.Context, update *ext.Update) error {
|
||||||
logger := log.FromContext(ctx)
|
|
||||||
stor := storage.FromContext(ctx)
|
stor := storage.FromContext(ctx)
|
||||||
if stor == nil {
|
|
||||||
logger.Warn("Context storage is nil")
|
|
||||||
ctx.Reply(update, ext.ReplyTextString("未找到存储"), nil)
|
|
||||||
return dispatcher.EndGroups
|
|
||||||
}
|
|
||||||
msg, result, err := shortcut.GetTphPicsFromMessageWithReply(ctx, update)
|
msg, result, err := shortcut.GetTphPicsFromMessageWithReply(ctx, update)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
userID := update.GetUserChat().GetID()
|
userID := update.GetUserChat().GetID()
|
||||||
return shortcut.CreateAndAddtelegraphWithEdit(ctx, userID, result.Page, result.TphDir, result.Pics, stor, msg.ID)
|
dirpath := result.TphDir
|
||||||
|
if p := dirutil.PathFromContext(ctx); p != "" {
|
||||||
|
dirpath = path.Join(p, dirpath)
|
||||||
|
}
|
||||||
|
return shortcut.CreateAndAddtelegraphWithEdit(ctx, userID, result.Page, dirpath, result.Pics, stor, msg.ID)
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|||||||
128
client/bot/handlers/update.go
Normal file
128
client/bot/handlers/update.go
Normal file
@@ -0,0 +1,128 @@
|
|||||||
|
package handlers
|
||||||
|
|
||||||
|
import (
|
||||||
|
"errors"
|
||||||
|
"regexp"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
"github.com/blang/semver"
|
||||||
|
"github.com/celestix/gotgproto/dispatcher"
|
||||||
|
"github.com/celestix/gotgproto/ext"
|
||||||
|
"github.com/gotd/td/telegram/message/html"
|
||||||
|
"github.com/gotd/td/tg"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/i18n"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/i18n/i18nk"
|
||||||
|
"github.com/krau/SaveAny-Bot/config"
|
||||||
|
"github.com/unvgo/ghselfupdate"
|
||||||
|
)
|
||||||
|
|
||||||
|
func handleUpdateCmd(ctx *ext.Context, u *ext.Update) error {
|
||||||
|
currentV, err := semver.Parse(config.Version)
|
||||||
|
if err != nil {
|
||||||
|
ctx.Reply(u, ext.ReplyTextString(i18n.T(i18nk.BotMsgUpdateErrorVersionVarInvalid, map[string]any{
|
||||||
|
"Error": err.Error(),
|
||||||
|
})), nil)
|
||||||
|
return dispatcher.EndGroups
|
||||||
|
}
|
||||||
|
latest, ok, err := ghselfupdate.DetectLatest(config.GitRepo)
|
||||||
|
if err != nil {
|
||||||
|
ctx.Reply(u, ext.ReplyTextString(i18n.T(i18nk.BotMsgUpdateErrorCheckLatestFailed, map[string]any{
|
||||||
|
"Error": err.Error(),
|
||||||
|
})), nil)
|
||||||
|
return dispatcher.EndGroups
|
||||||
|
}
|
||||||
|
if !ok {
|
||||||
|
ctx.Reply(u, ext.ReplyTextString(i18n.T(i18nk.BotMsgUpdateErrorNoReleaseFound, nil)), nil)
|
||||||
|
return dispatcher.EndGroups
|
||||||
|
}
|
||||||
|
if latest.Version.Major != currentV.Major {
|
||||||
|
ctx.Reply(u, ext.ReplyTextString(i18n.T(i18nk.BotMsgUpdateInfoMajorUpgradeRequired, map[string]any{
|
||||||
|
"Current": currentV.String(),
|
||||||
|
"Latest": latest.Version.String(),
|
||||||
|
})), nil)
|
||||||
|
return dispatcher.EndGroups
|
||||||
|
}
|
||||||
|
if latest.Version.LT(currentV) || latest.Version.Equals(currentV) {
|
||||||
|
ctx.Reply(u, ext.ReplyTextString(i18n.T(i18nk.BotMsgUpdateInfoAlreadyLatest, map[string]any{
|
||||||
|
"Version": config.Version,
|
||||||
|
})), nil)
|
||||||
|
return dispatcher.EndGroups
|
||||||
|
}
|
||||||
|
indocker := config.Docker == "true"
|
||||||
|
ctx.Sender.To(u.GetUserChat().AsInputPeer()).StyledText(ctx, html.String(nil, func() string {
|
||||||
|
md := latest.ReleaseNotes
|
||||||
|
md = regexp.MustCompile(`(?m)^###\s+ (.+)$`).ReplaceAllString(md, "<b>$1</b>")
|
||||||
|
md = regexp.MustCompile(`(?m)^#####\s+ (.+)$`).ReplaceAllString(md, "<i>$1</i>")
|
||||||
|
|
||||||
|
md = regexp.MustCompile(`(?m)^- `).ReplaceAllString(md, "• ")
|
||||||
|
|
||||||
|
md = regexp.MustCompile(`\[\((\w{6,})\)\]\((https?://[^\s)]+)\)`).ReplaceAllString(md, `(<a href="$2">$1</a>)`)
|
||||||
|
|
||||||
|
md = regexp.MustCompile(`\[(.+?)\]\((https?://[^\s)]+)\)`).ReplaceAllString(md, `<a href="$2">$1</a>`)
|
||||||
|
|
||||||
|
md = strings.ReplaceAll(md, " ", " ")
|
||||||
|
|
||||||
|
return `<blockquote expandable>` + md + `</blockquote>`
|
||||||
|
}()))
|
||||||
|
if indocker {
|
||||||
|
text := i18n.T(i18nk.BotMsgUpdateInfoNewVersionInDocker, map[string]any{
|
||||||
|
"Latest": latest.Version.String(),
|
||||||
|
"Current": config.Version,
|
||||||
|
"PublishedAt": latest.PublishedAt.Format("2006-01-02 15:04:05"),
|
||||||
|
})
|
||||||
|
ctx.Reply(u, ext.ReplyTextString(text), nil)
|
||||||
|
return dispatcher.EndGroups
|
||||||
|
}
|
||||||
|
text := i18n.T(i18nk.BotMsgUpdateInfoNewVersionPromptUpgrade, map[string]any{
|
||||||
|
"Latest": latest.Version.String(),
|
||||||
|
"Current": config.Version,
|
||||||
|
"SizeMB": float64(latest.AssetByteSize) / (1024 * 1024),
|
||||||
|
"URL": latest.AssetURL,
|
||||||
|
"PublishedAt": latest.PublishedAt.Format("2006-01-02 15:04:05"),
|
||||||
|
})
|
||||||
|
ctx.Reply(u, ext.ReplyTextString(text), &ext.ReplyOpts{
|
||||||
|
Markup: &tg.ReplyInlineMarkup{
|
||||||
|
Rows: []tg.KeyboardButtonRow{
|
||||||
|
{
|
||||||
|
Buttons: []tg.KeyboardButtonClass{
|
||||||
|
&tg.KeyboardButtonCallback{
|
||||||
|
Text: i18n.T(i18nk.BotMsgUpdateButtonUpgrade, nil),
|
||||||
|
Data: []byte("update"),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
})
|
||||||
|
return dispatcher.EndGroups
|
||||||
|
}
|
||||||
|
|
||||||
|
func handleUpdateCallback(ctx *ext.Context, u *ext.Update) error {
|
||||||
|
currentV, err := semver.Parse(config.Version)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
ctx.EditMessage(u.GetUserChat().GetID(), &tg.MessagesEditMessageRequest{
|
||||||
|
ID: u.CallbackQuery.GetMsgID(),
|
||||||
|
Message: i18n.T(i18nk.BotMsgUpdateInfoUpgradingWithVersion, map[string]any{
|
||||||
|
"Current": config.Version,
|
||||||
|
}),
|
||||||
|
})
|
||||||
|
latest, err := ghselfupdate.UpdateSelf(currentV, config.GitRepo)
|
||||||
|
if err != nil {
|
||||||
|
ctx.EditMessage(u.GetUserChat().GetID(), &tg.MessagesEditMessageRequest{
|
||||||
|
ID: u.CallbackQuery.GetMsgID(),
|
||||||
|
Message: i18n.T(i18nk.BotMsgUpdateErrorUpgradeFailed, map[string]any{
|
||||||
|
"Error": err.Error(),
|
||||||
|
}),
|
||||||
|
})
|
||||||
|
return dispatcher.EndGroups
|
||||||
|
}
|
||||||
|
ctx.EditMessage(u.GetUserChat().GetID(), &tg.MessagesEditMessageRequest{
|
||||||
|
ID: u.CallbackQuery.GetMsgID(),
|
||||||
|
Message: i18n.T(i18nk.BotMsgUpdateInfoUpgradeSuccess, map[string]any{
|
||||||
|
"Version": latest.Version.String(),
|
||||||
|
}),
|
||||||
|
})
|
||||||
|
return errors.New("SAVEANTBOT-RESTART")
|
||||||
|
}
|
||||||
37
client/bot/handlers/utils/dirutil/context.go
Normal file
37
client/bot/handlers/utils/dirutil/context.go
Normal file
@@ -0,0 +1,37 @@
|
|||||||
|
package dirutil
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
|
||||||
|
"github.com/krau/SaveAny-Bot/database"
|
||||||
|
)
|
||||||
|
|
||||||
|
type contextKey struct{}
|
||||||
|
|
||||||
|
var dirContextKey = contextKey{}
|
||||||
|
|
||||||
|
func WithContext(ctx context.Context, dir *database.Dir) context.Context {
|
||||||
|
if dir == nil {
|
||||||
|
return ctx
|
||||||
|
}
|
||||||
|
return context.WithValue(ctx, dirContextKey, dir)
|
||||||
|
}
|
||||||
|
|
||||||
|
func FromContext(ctx context.Context) *database.Dir {
|
||||||
|
dir, ok := ctx.Value(dirContextKey).(*database.Dir)
|
||||||
|
if !ok {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
return dir
|
||||||
|
}
|
||||||
|
|
||||||
|
// PathFromContext returns the directory path stored in the context.
|
||||||
|
//
|
||||||
|
// If no directory is found, an empty string is returned.
|
||||||
|
func PathFromContext(ctx context.Context) string {
|
||||||
|
dir := FromContext(ctx)
|
||||||
|
if dir == nil {
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
return dir.Path
|
||||||
|
}
|
||||||
26
client/bot/handlers/utils/filters/url.go
Normal file
26
client/bot/handlers/utils/filters/url.go
Normal file
@@ -0,0 +1,26 @@
|
|||||||
|
package filters
|
||||||
|
|
||||||
|
import (
|
||||||
|
"regexp"
|
||||||
|
"slices"
|
||||||
|
|
||||||
|
"github.com/celestix/gotgproto/dispatcher/handlers/filters"
|
||||||
|
"github.com/celestix/gotgproto/types"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/utils/tgutil"
|
||||||
|
)
|
||||||
|
|
||||||
|
func RegexUrl(r *regexp.Regexp) filters.MessageFilter {
|
||||||
|
return func(m *types.Message) bool {
|
||||||
|
if m.Text == "" {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
if r.MatchString(m.Text) {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
urls := tgutil.ExtractMessageEntityUrls(m.Message)
|
||||||
|
if len(urls) == 0 {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
return slices.ContainsFunc(urls, r.MatchString)
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -1,6 +1,20 @@
|
|||||||
package mediautil
|
package mediautil
|
||||||
|
|
||||||
import "github.com/gotd/td/tg"
|
import (
|
||||||
|
"context"
|
||||||
|
"fmt"
|
||||||
|
"strings"
|
||||||
|
"text/template"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/charmbracelet/log"
|
||||||
|
"github.com/gotd/td/tg"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/utils/strutil"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/utils/tgutil"
|
||||||
|
"github.com/krau/SaveAny-Bot/database"
|
||||||
|
"github.com/krau/SaveAny-Bot/pkg/enums/fnamest"
|
||||||
|
"github.com/krau/SaveAny-Bot/pkg/tfile"
|
||||||
|
)
|
||||||
|
|
||||||
func IsSupported(media tg.MessageMediaClass) bool {
|
func IsSupported(media tg.MessageMediaClass) bool {
|
||||||
switch media.(type) {
|
switch media.(type) {
|
||||||
@@ -10,3 +24,122 @@ func IsSupported(media tg.MessageMediaClass) bool {
|
|||||||
return false
|
return false
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
type FilenameTemplateData struct {
|
||||||
|
MsgID string `json:"msgid,omitempty"`
|
||||||
|
MsgTags string `json:"msgtags,omitempty"`
|
||||||
|
MsgGen string `json:"msggen,omitempty"`
|
||||||
|
MsgDate string `json:"msgdate,omitempty"`
|
||||||
|
MsgRaw string `json:"msgraw,omitempty"`
|
||||||
|
OrigName string `json:"origname,omitempty"`
|
||||||
|
ChatID string `json:"chatid,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (f FilenameTemplateData) ToMap() map[string]string {
|
||||||
|
return map[string]string{
|
||||||
|
"msgid": f.MsgID,
|
||||||
|
"msgtags": f.MsgTags,
|
||||||
|
"msggen": f.MsgGen,
|
||||||
|
"msgraw": f.MsgRaw,
|
||||||
|
"msgdate": f.MsgDate,
|
||||||
|
"origname": f.OrigName,
|
||||||
|
"chatid": f.ChatID,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TfileOptions(ctx context.Context, user *database.User, message *tg.Message) []tfile.TGFileOption {
|
||||||
|
opts := make([]tfile.TGFileOption, 0)
|
||||||
|
var fnameOpt tfile.TGFileOption
|
||||||
|
switch user.FilenameStrategy {
|
||||||
|
case fnamest.Message.String():
|
||||||
|
fnameOpt = tfile.WithName(tgutil.GenFileNameFromMessage(*message))
|
||||||
|
case fnamest.Template.String():
|
||||||
|
if user.FilenameTemplate == "" {
|
||||||
|
log.FromContext(ctx).Warnf("empty filename template")
|
||||||
|
fnameOpt = tfile.WithNameIfEmpty(tgutil.GenFileNameFromMessage(*message))
|
||||||
|
break
|
||||||
|
}
|
||||||
|
tmpl, err := template.New("filename").Parse(user.FilenameTemplate)
|
||||||
|
if err != nil {
|
||||||
|
log.FromContext(ctx).Errorf("failed to parse filename template: %s", err)
|
||||||
|
fnameOpt = tfile.WithNameIfEmpty(tgutil.GenFileNameFromMessage(*message))
|
||||||
|
break
|
||||||
|
}
|
||||||
|
data := BuildFilenameTemplateData(message)
|
||||||
|
var sb strings.Builder
|
||||||
|
err = tmpl.Execute(&sb, data)
|
||||||
|
if err != nil {
|
||||||
|
log.FromContext(ctx).Errorf("failed to execute filename template: %s", err)
|
||||||
|
fnameOpt = tfile.WithNameIfEmpty(tgutil.GenFileNameFromMessage(*message))
|
||||||
|
break
|
||||||
|
}
|
||||||
|
fnameOpt = tfile.WithName(sb.String())
|
||||||
|
default:
|
||||||
|
fnameOpt = tfile.WithNameIfEmpty(tgutil.GenFileNameFromMessage(*message))
|
||||||
|
}
|
||||||
|
opts = append(opts, fnameOpt, tfile.WithMessage(message))
|
||||||
|
return opts
|
||||||
|
}
|
||||||
|
|
||||||
|
func BuildFilenameTemplateData(message *tg.Message) map[string]string {
|
||||||
|
data := FilenameTemplateData{
|
||||||
|
MsgID: func() string {
|
||||||
|
id := message.GetID()
|
||||||
|
if id == 0 {
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
return fmt.Sprintf("%d", id)
|
||||||
|
}(),
|
||||||
|
MsgTags: func() string {
|
||||||
|
tags := strutil.ExtractTagsFromText(message.GetMessage())
|
||||||
|
if len(tags) == 0 {
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
return strings.Join(tags, "_")
|
||||||
|
}(),
|
||||||
|
MsgGen: tgutil.GenFileNameFromMessage(*message),
|
||||||
|
OrigName: func() string {
|
||||||
|
f, _ := tgutil.GetMediaFileName(message.Media)
|
||||||
|
return f
|
||||||
|
}(),
|
||||||
|
MsgDate: func() string {
|
||||||
|
date := message.GetDate()
|
||||||
|
if date == 0 {
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
t := time.Unix(int64(date), 0)
|
||||||
|
return t.Format("2006-01-02_15-04-05")
|
||||||
|
}(),
|
||||||
|
MsgRaw: message.GetMessage(),
|
||||||
|
ChatID: func() string {
|
||||||
|
// 如果消息是频道的(从消息链接中fetch的) 直接使用其chat id,
|
||||||
|
// 无论它是否是从其他来源转发的
|
||||||
|
if message.GetPost() {
|
||||||
|
peer := message.GetPeerID()
|
||||||
|
switch p := peer.(type) {
|
||||||
|
case *tg.PeerChannel:
|
||||||
|
return intToStringOmitZero(p.ChannelID)
|
||||||
|
default: // impossible case
|
||||||
|
return intToStringOmitZero(tgutil.ChatIdFromPeer(peer))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
fwdHeader, ok := message.GetFwdFrom()
|
||||||
|
if !ok {
|
||||||
|
return intToStringOmitZero(tgutil.ChatIdFromPeer(message.GetPeerID()))
|
||||||
|
}
|
||||||
|
fwdFrom, ok := fwdHeader.GetFromID()
|
||||||
|
if !ok {
|
||||||
|
return intToStringOmitZero(tgutil.ChatIdFromPeer(message.GetPeerID()))
|
||||||
|
}
|
||||||
|
return intToStringOmitZero(tgutil.ChatIdFromPeer(fwdFrom))
|
||||||
|
}(),
|
||||||
|
}.ToMap()
|
||||||
|
return data
|
||||||
|
}
|
||||||
|
|
||||||
|
func intToStringOmitZero(i int64) string {
|
||||||
|
if i == 0 {
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
return fmt.Sprintf("%d", i)
|
||||||
|
}
|
||||||
|
|||||||
@@ -5,26 +5,28 @@ import (
|
|||||||
"strings"
|
"strings"
|
||||||
|
|
||||||
"github.com/gotd/td/telegram/message/styling"
|
"github.com/gotd/td/telegram/message/styling"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/i18n"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/i18n/i18nk"
|
||||||
"github.com/krau/SaveAny-Bot/database"
|
"github.com/krau/SaveAny-Bot/database"
|
||||||
)
|
)
|
||||||
|
|
||||||
func BuildDirHelpStyling(dirs []database.Dir) []styling.StyledTextOption {
|
func BuildDirHelpStyling(dirs []database.Dir) []styling.StyledTextOption {
|
||||||
return []styling.StyledTextOption{
|
return []styling.StyledTextOption{
|
||||||
styling.Bold("使用方法: /dir <操作> <参数...>"),
|
styling.Bold(i18n.T(i18nk.BotMsgDirHelpUsage, nil)),
|
||||||
styling.Plain("\n\n可用操作:\n"),
|
styling.Plain(i18n.T(i18nk.BotMsgDirHelpAvailableOps, nil)),
|
||||||
styling.Code("add"),
|
styling.Code("add"),
|
||||||
styling.Plain(" <存储名> <路径> - 添加路径\n"),
|
styling.Plain(i18n.T(i18nk.BotMsgDirHelpAddSuffix, nil)),
|
||||||
styling.Code("del"),
|
styling.Code("del"),
|
||||||
styling.Plain(" <路径ID> - 删除路径\n"),
|
styling.Plain(i18n.T(i18nk.BotMsgDirHelpDelSuffix, nil)),
|
||||||
styling.Plain("\n添加路径示例:\n"),
|
styling.Plain(i18n.T(i18nk.BotMsgDirHelpAddExamplePrefix, nil)),
|
||||||
styling.Code("/dir add local1 path/to/dir"),
|
styling.Code(i18n.T(i18nk.BotMsgDirHelpAddExampleCmd, nil)),
|
||||||
styling.Plain("\n\n删除路径示例:\n"),
|
styling.Plain(i18n.T(i18nk.BotMsgDirHelpDelExamplePrefix, nil)),
|
||||||
styling.Code("/dir del 3"),
|
styling.Code(i18n.T(i18nk.BotMsgDirHelpDelExampleCmd, nil)),
|
||||||
styling.Plain("\n\n当前已添加的路径:\n"),
|
styling.Plain(i18n.T(i18nk.BotMsgDirHelpExistingDirsPrefix, nil)),
|
||||||
styling.Blockquote(func() string {
|
styling.Blockquote(func() string {
|
||||||
var sb strings.Builder
|
var sb strings.Builder
|
||||||
for _, dir := range dirs {
|
for _, dir := range dirs {
|
||||||
sb.WriteString(fmt.Sprintf("%d: ", dir.ID))
|
fmt.Fprintf(&sb, "%d: ", dir.ID)
|
||||||
sb.WriteString(dir.StorageName)
|
sb.WriteString(dir.StorageName)
|
||||||
sb.WriteString(" - ")
|
sb.WriteString(" - ")
|
||||||
sb.WriteString(dir.Path)
|
sb.WriteString(dir.Path)
|
||||||
|
|||||||
@@ -7,6 +7,8 @@ import (
|
|||||||
"github.com/gotd/td/telegram/message/entity"
|
"github.com/gotd/td/telegram/message/entity"
|
||||||
"github.com/gotd/td/telegram/message/styling"
|
"github.com/gotd/td/telegram/message/styling"
|
||||||
"github.com/gotd/td/tg"
|
"github.com/gotd/td/tg"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/i18n"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/i18n/i18nk"
|
||||||
"github.com/krau/SaveAny-Bot/pkg/parser"
|
"github.com/krau/SaveAny-Bot/pkg/parser"
|
||||||
)
|
)
|
||||||
|
|
||||||
@@ -14,15 +16,15 @@ func BuildParsedTextEntity(item parser.Item) (string, []tg.MessageEntityClass, e
|
|||||||
eb := entity.Builder{}
|
eb := entity.Builder{}
|
||||||
if err := styling.Perform(&eb,
|
if err := styling.Perform(&eb,
|
||||||
styling.Bold(fmt.Sprintf("[%s]%s", item.Site, item.Title)),
|
styling.Bold(fmt.Sprintf("[%s]%s", item.Site, item.Title)),
|
||||||
styling.Plain("\n链接: "),
|
styling.Plain(i18n.T(i18nk.BotMsgParseInfoLinkPrefix, nil)),
|
||||||
styling.Code(item.URL),
|
styling.Code(item.URL),
|
||||||
styling.Plain("\n作者: "),
|
styling.Plain(i18n.T(i18nk.BotMsgParseInfoAuthorPrefix, nil)),
|
||||||
styling.Code(item.Author),
|
styling.Code(item.Author),
|
||||||
styling.Plain("\n描述: "),
|
styling.Plain(i18n.T(i18nk.BotMsgParseInfoDescriptionPrefix, nil)),
|
||||||
styling.Code(strutil.Ellipsis(item.Description, 233)),
|
styling.Blockquote(strutil.Ellipsis(item.Description, 233), true),
|
||||||
styling.Plain("\n文件数量: "),
|
styling.Plain(i18n.T(i18nk.BotMsgParseInfoFileCountPrefix, nil)),
|
||||||
styling.Code(fmt.Sprintf("%d", len(item.Resources))),
|
styling.Code(fmt.Sprintf("%d", len(item.Resources))),
|
||||||
styling.Plain("\n预计总大小: "),
|
styling.Plain(i18n.T(i18nk.BotMsgParseInfoTotalSizePrefix, nil)),
|
||||||
styling.Code(fmt.Sprintf("%.2f MB", func() float64 {
|
styling.Code(fmt.Sprintf("%.2f MB", func() float64 {
|
||||||
var totalSize int64
|
var totalSize int64
|
||||||
for _, res := range item.Resources {
|
for _, res := range item.Resources {
|
||||||
@@ -30,9 +32,9 @@ func BuildParsedTextEntity(item parser.Item) (string, []tg.MessageEntityClass, e
|
|||||||
}
|
}
|
||||||
return float64(totalSize) / 1024 / 1024
|
return float64(totalSize) / 1024 / 1024
|
||||||
}())),
|
}())),
|
||||||
styling.Plain("\n请选择存储位置"),
|
styling.Plain(i18n.T(i18nk.BotMsgParseInfoPromptSelectStorage, nil)),
|
||||||
); err != nil {
|
); err != nil {
|
||||||
return "", nil, fmt.Errorf("构建消息失败: %w", err)
|
return "", nil, fmt.Errorf("failed to build parsed text entity: %w", err)
|
||||||
}
|
}
|
||||||
text, entities := eb.Complete()
|
text, entities := eb.Complete()
|
||||||
return text, entities, nil
|
return text, entities, nil
|
||||||
|
|||||||
@@ -5,21 +5,28 @@ import (
|
|||||||
"strings"
|
"strings"
|
||||||
|
|
||||||
"github.com/gotd/td/telegram/message/styling"
|
"github.com/gotd/td/telegram/message/styling"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/i18n"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/i18n/i18nk"
|
||||||
"github.com/krau/SaveAny-Bot/database"
|
"github.com/krau/SaveAny-Bot/database"
|
||||||
)
|
)
|
||||||
|
|
||||||
func BuildRuleHelpStyling(enabled bool, rules []database.Rule) []styling.StyledTextOption {
|
func BuildRuleHelpStyling(enabled bool, rules []database.Rule) []styling.StyledTextOption {
|
||||||
return []styling.StyledTextOption{
|
return []styling.StyledTextOption{
|
||||||
styling.Bold("使用方法: /rule <操作> <参数...>"),
|
styling.Bold(i18n.T(i18nk.BotMsgRuleHelpUsage, nil)),
|
||||||
styling.Bold(fmt.Sprintf("\n当前已%s规则模式", map[bool]string{true: "启用", false: "禁用"}[enabled])),
|
styling.Bold(func() string {
|
||||||
styling.Plain("\n\n可用操作:\n"),
|
if enabled {
|
||||||
|
return i18n.T(i18nk.BotMsgRuleHelpCurrentModeEnabled, nil)
|
||||||
|
}
|
||||||
|
return i18n.T(i18nk.BotMsgRuleHelpCurrentModeDisabled, nil)
|
||||||
|
}()),
|
||||||
|
styling.Plain(i18n.T(i18nk.BotMsgRuleHelpAvailableOps, nil)),
|
||||||
styling.Code("switch"),
|
styling.Code("switch"),
|
||||||
styling.Plain(" - 开关规则模式\n"),
|
styling.Plain(i18n.T(i18nk.BotMsgRuleHelpSwitchSuffix, nil)),
|
||||||
styling.Code("add"),
|
styling.Code("add"),
|
||||||
styling.Plain(" <类型> <数据> <存储名> <路径> - 添加规则\n"),
|
styling.Plain(i18n.T(i18nk.BotMsgRuleHelpAddSuffix, nil)),
|
||||||
styling.Code("del"),
|
styling.Code("del"),
|
||||||
styling.Plain(" <规则ID> - 删除规则\n"),
|
styling.Plain(i18n.T(i18nk.BotMsgRuleHelpDelSuffix, nil)),
|
||||||
styling.Plain("\n当前已添加的规则:\n"),
|
styling.Plain(i18n.T(i18nk.BotMsgRuleHelpExistingRulesPrefix, nil)),
|
||||||
styling.Blockquote(func() string {
|
styling.Blockquote(func() string {
|
||||||
var sb strings.Builder
|
var sb strings.Builder
|
||||||
for _, rule := range rules {
|
for _, rule := range rules {
|
||||||
|
|||||||
@@ -1,15 +0,0 @@
|
|||||||
package msgelem
|
|
||||||
|
|
||||||
const (
|
|
||||||
SaveHelpText = `
|
|
||||||
使用方法:
|
|
||||||
|
|
||||||
1. 使用该命令回复要保存的文件, 可选文件名参数.
|
|
||||||
示例:
|
|
||||||
/save custom_file_name.mp4
|
|
||||||
|
|
||||||
2. 设置默认存储后, 发送 /save <频道ID/用户名> <消息ID范围> 来批量保存文件. 遵从存储规则, 若未匹配到任何规则则使用默认存储.
|
|
||||||
示例:
|
|
||||||
/save @acherkrau 114-514
|
|
||||||
`
|
|
||||||
)
|
|
||||||
@@ -9,6 +9,8 @@ import (
|
|||||||
"github.com/gotd/td/telegram/message/styling"
|
"github.com/gotd/td/telegram/message/styling"
|
||||||
"github.com/gotd/td/tg"
|
"github.com/gotd/td/tg"
|
||||||
"github.com/krau/SaveAny-Bot/common/cache"
|
"github.com/krau/SaveAny-Bot/common/cache"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/i18n"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/i18n/i18nk"
|
||||||
"github.com/krau/SaveAny-Bot/database"
|
"github.com/krau/SaveAny-Bot/database"
|
||||||
"github.com/krau/SaveAny-Bot/pkg/enums/tasktype"
|
"github.com/krau/SaveAny-Bot/pkg/enums/tasktype"
|
||||||
"github.com/krau/SaveAny-Bot/pkg/tcbdata"
|
"github.com/krau/SaveAny-Bot/pkg/tcbdata"
|
||||||
@@ -45,6 +47,12 @@ func BuildAddSelectStorageKeyboard(stors []storage.Storage, adddata tcbdata.Add)
|
|||||||
TphDirPath: adddata.TphDirPath,
|
TphDirPath: adddata.TphDirPath,
|
||||||
|
|
||||||
ParsedItem: adddata.ParsedItem,
|
ParsedItem: adddata.ParsedItem,
|
||||||
|
|
||||||
|
DirectLinks: adddata.DirectLinks,
|
||||||
|
|
||||||
|
Aria2URIs: adddata.Aria2URIs,
|
||||||
|
YtdlpURLs: adddata.YtdlpURLs,
|
||||||
|
YtdlpFlags: adddata.YtdlpFlags,
|
||||||
}
|
}
|
||||||
dataid := xid.New().String()
|
dataid := xid.New().String()
|
||||||
err := cache.Set(dataid, data)
|
err := cache.Set(dataid, data)
|
||||||
@@ -68,11 +76,14 @@ func BuildAddSelectStorageKeyboard(stors []storage.Storage, adddata tcbdata.Add)
|
|||||||
func BuildAddOneSelectStorageMessage(ctx context.Context, stors []storage.Storage, file tfile.TGFileMessage, msgId int) (*tg.MessagesEditMessageRequest, error) {
|
func BuildAddOneSelectStorageMessage(ctx context.Context, stors []storage.Storage, file tfile.TGFileMessage, msgId int) (*tg.MessagesEditMessageRequest, error) {
|
||||||
eb := entity.Builder{}
|
eb := entity.Builder{}
|
||||||
var entities []tg.MessageEntityClass
|
var entities []tg.MessageEntityClass
|
||||||
text := fmt.Sprintf("文件名: %s\n请选择存储位置", file.Name())
|
text := i18n.T(i18nk.BotMsgTasksInfoAddedToQueueFull, map[string]any{
|
||||||
|
"Filename": file.Name(),
|
||||||
|
"QueueLength": 0,
|
||||||
|
})
|
||||||
if err := styling.Perform(&eb,
|
if err := styling.Perform(&eb,
|
||||||
styling.Plain("文件名: "),
|
styling.Plain(i18n.T(i18nk.BotMsgStorageInfoFilenamePrefix, nil)),
|
||||||
styling.Code(file.Name()),
|
styling.Code(file.Name()),
|
||||||
styling.Plain("\n请选择存储位置"),
|
styling.Plain(i18n.T(i18nk.BotMsgStorageInfoPromptSelectStorage, nil)),
|
||||||
); err != nil {
|
); err != nil {
|
||||||
log.FromContext(ctx).Errorf("Failed to build entity: %s", err)
|
log.FromContext(ctx).Errorf("Failed to build entity: %s", err)
|
||||||
} else {
|
} else {
|
||||||
@@ -94,7 +105,10 @@ func BuildAddOneSelectStorageMessage(ctx context.Context, stors []storage.Storag
|
|||||||
}, nil
|
}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func BuildSetDefaultStorageMarkup(ctx context.Context, userID int64, stors []storage.Storage) (*tg.ReplyInlineMarkup, error) {
|
// Builds the inline keyboard for setting default storage
|
||||||
|
func BuildSetDefaultStorageMarkup(
|
||||||
|
ctx context.Context,
|
||||||
|
stors []storage.Storage) (*tg.ReplyInlineMarkup, error) {
|
||||||
buttons := make([]tg.KeyboardButtonClass, 0)
|
buttons := make([]tg.KeyboardButtonClass, 0)
|
||||||
for _, storage := range stors {
|
for _, storage := range stors {
|
||||||
data := tcbdata.SetDefaultStorage{
|
data := tcbdata.SetDefaultStorage{
|
||||||
@@ -119,7 +133,35 @@ func BuildSetDefaultStorageMarkup(ctx context.Context, userID int64, stors []sto
|
|||||||
return markup, nil
|
return markup, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func BuildSetDirKeyboard(dirs []database.Dir, dataid string) (*tg.ReplyInlineMarkup, error) {
|
func BuildSetDefaultDirMarkup(ctx context.Context,
|
||||||
|
seletedStorage string,
|
||||||
|
dirs []database.Dir) (*tg.ReplyInlineMarkup, error) {
|
||||||
|
buttons := make([]tg.KeyboardButtonClass, 0)
|
||||||
|
for _, dir := range dirs {
|
||||||
|
dataid := xid.New().String()
|
||||||
|
data := tcbdata.SetDefaultStorage{
|
||||||
|
StorageName: seletedStorage,
|
||||||
|
DirID: dir.ID,
|
||||||
|
}
|
||||||
|
err := cache.Set(dataid, data)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
buttons = append(buttons, &tg.KeyboardButtonCallback{
|
||||||
|
Text: dir.Path,
|
||||||
|
Data: fmt.Appendf(nil, "%s %s", tcbdata.TypeSetDefault, dataid),
|
||||||
|
})
|
||||||
|
}
|
||||||
|
markup := &tg.ReplyInlineMarkup{}
|
||||||
|
for i := 0; i < len(buttons); i += 3 {
|
||||||
|
row := tg.KeyboardButtonRow{}
|
||||||
|
row.Buttons = buttons[i:min(i+3, len(buttons))]
|
||||||
|
markup.Rows = append(markup.Rows, row)
|
||||||
|
}
|
||||||
|
return markup, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func BuildSetDirMarkupForAdd(dirs []database.Dir, dataid string) (*tg.ReplyInlineMarkup, error) {
|
||||||
data, ok := cache.Get[tcbdata.Add](dataid)
|
data, ok := cache.Get[tcbdata.Add](dataid)
|
||||||
if !ok {
|
if !ok {
|
||||||
return nil, fmt.Errorf("failed to get data from cache: %s", dataid)
|
return nil, fmt.Errorf("failed to get data from cache: %s", dataid)
|
||||||
@@ -152,7 +194,7 @@ func BuildSetDirKeyboard(dirs []database.Dir, dataid string) (*tg.ReplyInlineMar
|
|||||||
return nil, fmt.Errorf("failed to set default directory data in cache: %w", err)
|
return nil, fmt.Errorf("failed to set default directory data in cache: %w", err)
|
||||||
}
|
}
|
||||||
buttons = append(buttons, &tg.KeyboardButtonCallback{
|
buttons = append(buttons, &tg.KeyboardButtonCallback{
|
||||||
Text: "默认",
|
Text: i18n.T(i18nk.BotMsgDirButtonDefault, nil),
|
||||||
Data: fmt.Appendf(nil, "%s %s", tcbdata.TypeAdd, dirDefaultDataId),
|
Data: fmt.Appendf(nil, "%s %s", tcbdata.TypeAdd, dirDefaultDataId),
|
||||||
})
|
})
|
||||||
markup := &tg.ReplyInlineMarkup{}
|
markup := &tg.ReplyInlineMarkup{}
|
||||||
|
|||||||
@@ -2,13 +2,14 @@ package msgelem
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
"fmt"
|
|
||||||
"strconv"
|
"strconv"
|
||||||
|
|
||||||
"github.com/charmbracelet/log"
|
"github.com/charmbracelet/log"
|
||||||
"github.com/gotd/td/telegram/message/entity"
|
"github.com/gotd/td/telegram/message/entity"
|
||||||
"github.com/gotd/td/telegram/message/styling"
|
"github.com/gotd/td/telegram/message/styling"
|
||||||
"github.com/gotd/td/tg"
|
"github.com/gotd/td/tg"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/i18n"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/i18n/i18nk"
|
||||||
)
|
)
|
||||||
|
|
||||||
func BuildTaskAddedEntities(
|
func BuildTaskAddedEntities(
|
||||||
@@ -18,11 +19,15 @@ func BuildTaskAddedEntities(
|
|||||||
) (string, []tg.MessageEntityClass) {
|
) (string, []tg.MessageEntityClass) {
|
||||||
entityBuilder := entity.Builder{}
|
entityBuilder := entity.Builder{}
|
||||||
var entities []tg.MessageEntityClass
|
var entities []tg.MessageEntityClass
|
||||||
text := fmt.Sprintf("已添加到任务队列\n文件名: %s\n当前排队任务数: %d", filename, queueLength)
|
text := i18n.T(i18nk.BotMsgTasksInfoAddedToQueueFull, map[string]any{
|
||||||
|
"Filename": filename,
|
||||||
|
"QueueLength": queueLength,
|
||||||
|
})
|
||||||
if err := styling.Perform(&entityBuilder,
|
if err := styling.Perform(&entityBuilder,
|
||||||
styling.Plain("已添加到任务队列\n文件名: "),
|
styling.Plain(i18n.T(i18nk.BotMsgTasksInfoAddedToQueuePrefix, nil)),
|
||||||
|
styling.Plain(i18n.T(i18nk.BotMsgTasksInfoFilenamePrefix, nil)),
|
||||||
styling.Code(filename),
|
styling.Code(filename),
|
||||||
styling.Plain("\n当前排队任务数: "),
|
styling.Plain(i18n.T(i18nk.BotMsgTasksInfoQueueLengthPrefix, nil)),
|
||||||
styling.Bold(strconv.Itoa(queueLength)),
|
styling.Bold(strconv.Itoa(queueLength)),
|
||||||
); err != nil {
|
); err != nil {
|
||||||
log.FromContext(ctx).Errorf("Failed to build entity: %s", err)
|
log.FromContext(ctx).Errorf("Failed to build entity: %s", err)
|
||||||
|
|||||||
@@ -1,19 +0,0 @@
|
|||||||
package msgelem
|
|
||||||
|
|
||||||
const (
|
|
||||||
WatchHelpText = `
|
|
||||||
使用 /watch 命令监听一个聊天的消息, 并自动保存到默认存储中, 遵从存储规则.
|
|
||||||
|
|
||||||
命令语法:
|
|
||||||
/watch <chat_id> [filter]
|
|
||||||
|
|
||||||
参数:
|
|
||||||
- <chat_id>: 聊天的 ID 或用户名
|
|
||||||
- [filter]: 可选, 格式为 过滤器类型:表达式 , 所有支持类型的过滤器请查看文档
|
|
||||||
|
|
||||||
命令示例:
|
|
||||||
/watch 2229835658 msgre:.*plana.*
|
|
||||||
|
|
||||||
这将监听 ID 为 2229835658 的聊天, 并转存所有包含 "plana" 的媒体消息
|
|
||||||
`
|
|
||||||
)
|
|
||||||
@@ -7,8 +7,6 @@ import (
|
|||||||
|
|
||||||
"github.com/charmbracelet/log"
|
"github.com/charmbracelet/log"
|
||||||
"github.com/krau/SaveAny-Bot/database"
|
"github.com/krau/SaveAny-Bot/database"
|
||||||
"github.com/krau/SaveAny-Bot/pkg/consts"
|
|
||||||
ruleenum "github.com/krau/SaveAny-Bot/pkg/enums/rule"
|
|
||||||
"github.com/krau/SaveAny-Bot/pkg/rule"
|
"github.com/krau/SaveAny-Bot/pkg/rule"
|
||||||
"github.com/krau/SaveAny-Bot/pkg/tfile"
|
"github.com/krau/SaveAny-Bot/pkg/tfile"
|
||||||
)
|
)
|
||||||
@@ -36,8 +34,8 @@ func (m matchedStorName) String() string {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// can we use this storage name directly?
|
// can we use this storage name directly?
|
||||||
func (m matchedStorName) IsUsable() bool {
|
func (m matchedStorName) Usable() bool {
|
||||||
return m != "" && m != consts.RuleStorNameChosen
|
return m != "" && m != rule.RuleStorNameChosen
|
||||||
}
|
}
|
||||||
|
|
||||||
type MatchedDirPath string
|
type MatchedDirPath string
|
||||||
@@ -47,17 +45,17 @@ func (m MatchedDirPath) String() string {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (m MatchedDirPath) NeedNewForAlbum() bool {
|
func (m MatchedDirPath) NeedNewForAlbum() bool {
|
||||||
return m != "" && m == consts.RuleDirPathNewForAlbum
|
return m != "" && m == rule.RuleDirPathNewForAlbum
|
||||||
}
|
}
|
||||||
|
|
||||||
func ApplyRule(ctx context.Context, rules []database.Rule, inputs *ruleInput) (matchedStorageName matchedStorName, dirPath MatchedDirPath) {
|
func ApplyRule(ctx context.Context, rules []database.Rule, inputs *ruleInput) (matched bool, matchedStorageName matchedStorName, dirPath MatchedDirPath) {
|
||||||
if inputs == nil || len(rules) == 0 {
|
if inputs == nil || len(rules) == 0 {
|
||||||
return "", ""
|
return false, "", ""
|
||||||
}
|
}
|
||||||
logger := log.FromContext(ctx)
|
logger := log.FromContext(ctx)
|
||||||
for _, ur := range rules {
|
for _, ur := range rules {
|
||||||
switch ur.Type {
|
switch ur.Type {
|
||||||
case ruleenum.FileNameRegex.String():
|
case rule.FileNameRegex.String():
|
||||||
ru, err := rule.NewRuleFileNameRegex(ur.StorageName, ur.DirPath, ur.Data)
|
ru, err := rule.NewRuleFileNameRegex(ur.StorageName, ur.DirPath, ur.Data)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Errorf("Failed to create rule: %s", err)
|
logger.Errorf("Failed to create rule: %s", err)
|
||||||
@@ -72,7 +70,7 @@ func ApplyRule(ctx context.Context, rules []database.Rule, inputs *ruleInput) (m
|
|||||||
dirPath = MatchedDirPath(ru.StoragePath())
|
dirPath = MatchedDirPath(ru.StoragePath())
|
||||||
matchedStorageName = matchedStorName(ru.StorageName())
|
matchedStorageName = matchedStorName(ru.StorageName())
|
||||||
}
|
}
|
||||||
case ruleenum.MessageRegex.String():
|
case rule.MessageRegex.String():
|
||||||
ru, err := rule.NewRuleMessageRegex(ur.StorageName, ur.DirPath, ur.Data)
|
ru, err := rule.NewRuleMessageRegex(ur.StorageName, ur.DirPath, ur.Data)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Errorf("Failed to create rule: %s", err)
|
logger.Errorf("Failed to create rule: %s", err)
|
||||||
@@ -87,7 +85,7 @@ func ApplyRule(ctx context.Context, rules []database.Rule, inputs *ruleInput) (m
|
|||||||
dirPath = MatchedDirPath(ru.StoragePath())
|
dirPath = MatchedDirPath(ru.StoragePath())
|
||||||
matchedStorageName = matchedStorName(ru.StorageName())
|
matchedStorageName = matchedStorName(ru.StorageName())
|
||||||
}
|
}
|
||||||
case ruleenum.IsAlbum.String():
|
case rule.IsAlbum.String():
|
||||||
matchAlbum, err := convertor.ToBool(ur.Data)
|
matchAlbum, err := convertor.ToBool(ur.Data)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
matchAlbum = false
|
matchAlbum = false
|
||||||
@@ -108,5 +106,8 @@ func ApplyRule(ctx context.Context, rules []database.Rule, inputs *ruleInput) (m
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
return
|
if matchedStorageName != "" || dirPath != "" {
|
||||||
|
return true, matchedStorageName, dirPath
|
||||||
|
}
|
||||||
|
return false, "", ""
|
||||||
}
|
}
|
||||||
|
|||||||
65
client/bot/handlers/utils/shortcut/aria2.go
Normal file
65
client/bot/handlers/utils/shortcut/aria2.go
Normal file
@@ -0,0 +1,65 @@
|
|||||||
|
package shortcut
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/celestix/gotgproto/dispatcher"
|
||||||
|
"github.com/celestix/gotgproto/ext"
|
||||||
|
"github.com/charmbracelet/log"
|
||||||
|
"github.com/gotd/td/tg"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/i18n"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/i18n/i18nk"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/utils/tgutil"
|
||||||
|
"github.com/krau/SaveAny-Bot/core"
|
||||||
|
"github.com/krau/SaveAny-Bot/core/tasks/aria2dl"
|
||||||
|
"github.com/krau/SaveAny-Bot/pkg/aria2"
|
||||||
|
"github.com/krau/SaveAny-Bot/storage"
|
||||||
|
"github.com/rs/xid"
|
||||||
|
)
|
||||||
|
|
||||||
|
func CreateAndAddAria2TaskWithEdit(ctx *ext.Context, stor storage.Storage, dirPath string, uris []string, aria2Client *aria2.Client, msgID int, userID int64) error {
|
||||||
|
logger := log.FromContext(ctx)
|
||||||
|
injectCtx := tgutil.ExtWithContext(ctx.Context, ctx)
|
||||||
|
|
||||||
|
// Now add to aria2 after user selected storage
|
||||||
|
logger.Infof("Adding download to aria2, uris type: %T, value: %+v", uris, uris)
|
||||||
|
|
||||||
|
// Ensure uris is valid
|
||||||
|
if len(uris) == 0 {
|
||||||
|
logger.Error("URIs list is empty")
|
||||||
|
ctx.EditMessage(userID, &tg.MessagesEditMessageRequest{
|
||||||
|
ID: msgID,
|
||||||
|
Message: i18n.T(i18nk.BotMsgDlErrorNoValidLinks, nil),
|
||||||
|
})
|
||||||
|
return dispatcher.EndGroups
|
||||||
|
}
|
||||||
|
|
||||||
|
gid, err := aria2Client.AddURI(ctx, uris, nil)
|
||||||
|
if err != nil {
|
||||||
|
logger.Errorf("Failed to add aria2 download: %s", err)
|
||||||
|
ctx.EditMessage(userID, &tg.MessagesEditMessageRequest{
|
||||||
|
ID: msgID,
|
||||||
|
Message: i18n.T(i18nk.BotMsgAria2ErrorAddingAria2Download, map[string]any{
|
||||||
|
"Error": err.Error(),
|
||||||
|
}),
|
||||||
|
})
|
||||||
|
return dispatcher.EndGroups
|
||||||
|
}
|
||||||
|
logger.Infof("Aria2 download added with GID: %s", gid)
|
||||||
|
|
||||||
|
// Create task with the GID
|
||||||
|
task := aria2dl.NewTask(xid.New().String(), injectCtx, gid, uris, aria2Client, stor, stor.JoinStoragePath(dirPath), aria2dl.NewProgress(msgID, userID))
|
||||||
|
if err := core.AddTask(injectCtx, task); err != nil {
|
||||||
|
logger.Errorf("Failed to add task: %s", err)
|
||||||
|
ctx.EditMessage(userID, &tg.MessagesEditMessageRequest{
|
||||||
|
ID: msgID,
|
||||||
|
Message: i18n.T(i18nk.BotMsgCommonErrorTaskAddFailed, map[string]any{
|
||||||
|
"Error": err.Error(),
|
||||||
|
}),
|
||||||
|
})
|
||||||
|
return dispatcher.EndGroups
|
||||||
|
}
|
||||||
|
ctx.EditMessage(userID, &tg.MessagesEditMessageRequest{
|
||||||
|
ID: msgID,
|
||||||
|
Message: i18n.T(i18nk.BotMsgCommonInfoTaskAdded, nil),
|
||||||
|
})
|
||||||
|
return dispatcher.EndGroups
|
||||||
|
}
|
||||||
34
client/bot/handlers/utils/shortcut/directlinks.go
Normal file
34
client/bot/handlers/utils/shortcut/directlinks.go
Normal file
@@ -0,0 +1,34 @@
|
|||||||
|
package shortcut
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/celestix/gotgproto/dispatcher"
|
||||||
|
"github.com/celestix/gotgproto/ext"
|
||||||
|
"github.com/charmbracelet/log"
|
||||||
|
"github.com/gotd/td/tg"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/i18n"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/i18n/i18nk"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/utils/tgutil"
|
||||||
|
"github.com/krau/SaveAny-Bot/core"
|
||||||
|
"github.com/krau/SaveAny-Bot/core/tasks/directlinks"
|
||||||
|
"github.com/krau/SaveAny-Bot/storage"
|
||||||
|
"github.com/rs/xid"
|
||||||
|
)
|
||||||
|
|
||||||
|
func CreateAndAddDirectTaskWithEdit(ctx *ext.Context, stor storage.Storage, dirPath string, links []string, msgID int, userID int64) error {
|
||||||
|
injectCtx := tgutil.ExtWithContext(ctx.Context, ctx)
|
||||||
|
task := directlinks.NewTask(xid.New().String(), injectCtx, links, stor, stor.JoinStoragePath(dirPath), directlinks.NewProgress(msgID, userID))
|
||||||
|
if err := core.AddTask(injectCtx, task); err != nil {
|
||||||
|
log.FromContext(ctx).Errorf("Failed to add task: %s", err)
|
||||||
|
ctx.EditMessage(userID, &tg.MessagesEditMessageRequest{
|
||||||
|
ID: msgID,
|
||||||
|
Message: i18n.T(i18nk.BotMsgCommonErrorTaskAddFailed, map[string]any{
|
||||||
|
"Error": err.Error(),
|
||||||
|
}),
|
||||||
|
})
|
||||||
|
return dispatcher.EndGroups
|
||||||
|
}
|
||||||
|
ctx.EditMessage(userID, &tg.MessagesEditMessageRequest{
|
||||||
|
Message: i18n.T(i18nk.BotMsgCommonInfoTaskAdded, nil),
|
||||||
|
})
|
||||||
|
return dispatcher.EndGroups
|
||||||
|
}
|
||||||
@@ -17,15 +17,18 @@ import (
|
|||||||
"github.com/krau/SaveAny-Bot/client/bot/handlers/utils/re"
|
"github.com/krau/SaveAny-Bot/client/bot/handlers/utils/re"
|
||||||
uc "github.com/krau/SaveAny-Bot/client/user"
|
uc "github.com/krau/SaveAny-Bot/client/user"
|
||||||
"github.com/krau/SaveAny-Bot/common/cache"
|
"github.com/krau/SaveAny-Bot/common/cache"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/i18n"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/i18n/i18nk"
|
||||||
"github.com/krau/SaveAny-Bot/common/utils/tgutil"
|
"github.com/krau/SaveAny-Bot/common/utils/tgutil"
|
||||||
"github.com/krau/SaveAny-Bot/common/utils/tphutil"
|
"github.com/krau/SaveAny-Bot/common/utils/tphutil"
|
||||||
"github.com/krau/SaveAny-Bot/config"
|
"github.com/krau/SaveAny-Bot/config"
|
||||||
|
"github.com/krau/SaveAny-Bot/database"
|
||||||
"github.com/krau/SaveAny-Bot/pkg/telegraph"
|
"github.com/krau/SaveAny-Bot/pkg/telegraph"
|
||||||
"github.com/krau/SaveAny-Bot/pkg/tfile"
|
"github.com/krau/SaveAny-Bot/pkg/tfile"
|
||||||
)
|
)
|
||||||
|
|
||||||
// 获取消息中的文件并回复等待消息, 返回等待消息, 获取到的文件
|
// 获取消息中的文件并回复等待消息, 返回等待消息, 获取到的文件
|
||||||
func GetFileFromMessageWithReply(ctx *ext.Context, update *ext.Update, message *tg.Message, tfileopts ...tfile.TGFileOptions) (replied *types.Message,
|
func GetFileFromMessageWithReply(ctx *ext.Context, update *ext.Update, message *tg.Message, tfileopts ...tfile.TGFileOption) (replied *types.Message,
|
||||||
file tfile.TGFileMessage, err error,
|
file tfile.TGFileMessage, err error,
|
||||||
) {
|
) {
|
||||||
logger := log.FromContext(ctx)
|
logger := log.FromContext(ctx)
|
||||||
@@ -35,23 +38,25 @@ func GetFileFromMessageWithReply(ctx *ext.Context, update *ext.Update, message *
|
|||||||
return nil, nil, dispatcher.ContinueGroups
|
return nil, nil, dispatcher.ContinueGroups
|
||||||
}
|
}
|
||||||
|
|
||||||
replied, err = ctx.Reply(update, ext.ReplyTextString("正在获取文件信息..."), nil)
|
replied, err = ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgCommonInfoFetchingFileInfo, nil)), nil)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Errorf("Failed to reply: %s", err)
|
logger.Errorf("Failed to reply: %s", err)
|
||||||
return nil, nil, dispatcher.EndGroups
|
return nil, nil, dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
options := []tfile.TGFileOptions{
|
// options := []tfile.TGFileOption{
|
||||||
tfile.WithMessage(message),
|
// tfile.WithMessage(message),
|
||||||
}
|
// }
|
||||||
if len(tfileopts) > 0 {
|
// if len(tfileopts) > 0 {
|
||||||
options = append(options, tfileopts...)
|
// options = append(options, tfileopts...)
|
||||||
} else {
|
// } else {
|
||||||
options = append(options, tfile.WithNameIfEmpty(tgutil.GenFileNameFromMessage(*message)))
|
// options = append(options, tfile.WithNameIfEmpty(tgutil.GenFileNameFromMessage(*message)))
|
||||||
}
|
// }
|
||||||
file, err = tfile.FromMediaMessage(media, ctx.Raw, message, options...)
|
file, err = tfile.FromMediaMessage(media, ctx.Raw, message, tfileopts...)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Errorf("Failed to get file from media: %s", err)
|
logger.Errorf("Failed to get file from media: %s", err)
|
||||||
ctx.Reply(update, ext.ReplyTextString("获取文件失败: "+err.Error()), nil)
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgCommonErrorGetFileFailed, map[string]any{
|
||||||
|
"Error": err.Error(),
|
||||||
|
})), nil)
|
||||||
return nil, nil, dispatcher.EndGroups
|
return nil, nil, dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
return replied, file, nil
|
return replied, file, nil
|
||||||
@@ -62,12 +67,12 @@ type EditMessageFunc func(text string, markup tg.ReplyMarkupClass)
|
|||||||
// 获取链接中的文件并回复等待消息
|
// 获取链接中的文件并回复等待消息
|
||||||
func GetFilesFromUpdateLinkMessageWithReplyEdit(ctx *ext.Context, update *ext.Update) (replied *types.Message, files []tfile.TGFileMessage, editReplied EditMessageFunc, err error) {
|
func GetFilesFromUpdateLinkMessageWithReplyEdit(ctx *ext.Context, update *ext.Update) (replied *types.Message, files []tfile.TGFileMessage, editReplied EditMessageFunc, err error) {
|
||||||
logger := log.FromContext(ctx)
|
logger := log.FromContext(ctx)
|
||||||
msgLinks := re.TgMessageLinkRegexp.FindAllString(update.EffectiveMessage.GetMessage(), -1)
|
msgLinks := re.TgMessageLinkRegexp.FindAllString(tgutil.ExtractMessageEntityUrlsText(update.EffectiveMessage.Message), -1)
|
||||||
if len(msgLinks) == 0 {
|
if len(msgLinks) == 0 {
|
||||||
logger.Warn("no matched message links but called handleMessageLink")
|
logger.Warn("no matched message links but called handleMessageLink")
|
||||||
return nil, nil, nil, dispatcher.EndGroups
|
return nil, nil, nil, dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
replied, err = ctx.Reply(update, ext.ReplyTextString("正在获取消息..."), nil)
|
replied, err = ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgCommonInfoFetchingMessages, nil)), nil)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Errorf("failed to reply: %s", err)
|
logger.Errorf("failed to reply: %s", err)
|
||||||
return nil, nil, nil, dispatcher.EndGroups
|
return nil, nil, nil, dispatcher.EndGroups
|
||||||
@@ -81,7 +86,14 @@ func GetFilesFromUpdateLinkMessageWithReplyEdit(ctx *ext.Context, update *ext.Up
|
|||||||
logger.Errorf("failed to edit message: %s", err)
|
logger.Errorf("failed to edit message: %s", err)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
user, err := database.GetUserByChatID(ctx, update.GetUserChat().GetID())
|
||||||
|
if err != nil {
|
||||||
|
logger.Errorf("failed to get user from db: %s", err)
|
||||||
|
editReplied(i18n.T(i18nk.BotMsgCommonErrorGetUserInfoFailed, map[string]any{
|
||||||
|
"Error": err.Error(),
|
||||||
|
}), nil)
|
||||||
|
return nil, nil, nil, dispatcher.EndGroups
|
||||||
|
}
|
||||||
files = make([]tfile.TGFileMessage, 0, len(msgLinks))
|
files = make([]tfile.TGFileMessage, 0, len(msgLinks))
|
||||||
addFile := func(client downloader.Client, msg *tg.Message) {
|
addFile := func(client downloader.Client, msg *tg.Message) {
|
||||||
if msg == nil || msg.Media == nil {
|
if msg == nil || msg.Media == nil {
|
||||||
@@ -93,7 +105,8 @@ func GetFilesFromUpdateLinkMessageWithReplyEdit(ctx *ext.Context, update *ext.Up
|
|||||||
logger.Debugf("message %d has no media", msg.GetID())
|
logger.Debugf("message %d has no media", msg.GetID())
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
file, err := tfile.FromMediaMessage(media, client, msg, tfile.WithNameIfEmpty(tgutil.GenFileNameFromMessage(*msg)))
|
opts := mediautil.TfileOptions(ctx, user, msg)
|
||||||
|
file, err := tfile.FromMediaMessage(media, client, msg, opts...)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Errorf("failed to create file from media: %s", err)
|
logger.Errorf("failed to create file from media: %s", err)
|
||||||
return
|
return
|
||||||
@@ -102,8 +115,10 @@ func GetFilesFromUpdateLinkMessageWithReplyEdit(ctx *ext.Context, update *ext.Up
|
|||||||
}
|
}
|
||||||
|
|
||||||
tctx := ctx
|
tctx := ctx
|
||||||
if config.Cfg.Telegram.Userbot.Enable {
|
if config.C().Telegram.Userbot.Enable {
|
||||||
tctx = uc.GetCtx()
|
if uc.GetCtx() != nil {
|
||||||
|
tctx = uc.GetCtx()
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
for _, link := range msgLinks {
|
for _, link := range msgLinks {
|
||||||
@@ -119,12 +134,12 @@ func GetFilesFromUpdateLinkMessageWithReplyEdit(ctx *ext.Context, update *ext.Up
|
|||||||
}
|
}
|
||||||
msg, err := tgutil.GetMessageByID(tctx, chatId, msgId)
|
msg, err := tgutil.GetMessageByID(tctx, chatId, msgId)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Errorf("failed to get message by ID: %s", err)
|
logger.Error(err)
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
groupID, isGroup := msg.GetGroupedID()
|
groupID, isGroup := msg.GetGroupedID()
|
||||||
if isGroup && groupID != 0 && !linkUrl.Query().Has("single") {
|
if isGroup && groupID != 0 && !linkUrl.Query().Has("single") {
|
||||||
gmsgs, err := tgutil.GetGroupedMessages(ctx, chatId, msg)
|
gmsgs, err := tgutil.GetGroupedMessages(tctx, chatId, msg)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Errorf("failed to get grouped messages: %s", err)
|
logger.Errorf("failed to get grouped messages: %s", err)
|
||||||
} else {
|
} else {
|
||||||
@@ -137,7 +152,7 @@ func GetFilesFromUpdateLinkMessageWithReplyEdit(ctx *ext.Context, update *ext.Up
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
if len(files) == 0 {
|
if len(files) == 0 {
|
||||||
editReplied("没有找到可保存的文件", nil)
|
editReplied(i18n.T(i18nk.BotMsgCommonErrorNoSavableFilesFound, nil), nil)
|
||||||
return nil, nil, nil, dispatcher.EndGroups
|
return nil, nil, nil, dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
return replied, files, editReplied, nil
|
return replied, files, editReplied, nil
|
||||||
@@ -148,7 +163,7 @@ func GetCallbackDataWithAnswer[DataType any](ctx *ext.Context, update *ext.Updat
|
|||||||
if !ok {
|
if !ok {
|
||||||
log.FromContext(ctx).Warnf("Invalid data ID: %s", dataid)
|
log.FromContext(ctx).Warnf("Invalid data ID: %s", dataid)
|
||||||
queryID := update.CallbackQuery.GetQueryID()
|
queryID := update.CallbackQuery.GetQueryID()
|
||||||
ctx.AnswerCallback(msgelem.AlertCallbackAnswer(queryID, "数据已过期或无效"))
|
ctx.AnswerCallback(msgelem.AlertCallbackAnswer(queryID, i18n.T(i18nk.BotMsgCommonErrorDataExpired, nil)))
|
||||||
var zero DataType
|
var zero DataType
|
||||||
return zero, dispatcher.EndGroups
|
return zero, dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
@@ -164,7 +179,7 @@ type TelegraphResult struct {
|
|||||||
// return replied message, image urls, telegraph path(unescaped), error
|
// return replied message, image urls, telegraph path(unescaped), error
|
||||||
func GetTphPicsFromMessageWithReply(ctx *ext.Context, update *ext.Update) (*types.Message, *TelegraphResult, error) {
|
func GetTphPicsFromMessageWithReply(ctx *ext.Context, update *ext.Update) (*types.Message, *TelegraphResult, error) {
|
||||||
logger := log.FromContext(ctx)
|
logger := log.FromContext(ctx)
|
||||||
tphurl := re.TelegraphUrlRegexp.FindString(update.EffectiveMessage.GetMessage()) // TODO: batch urls
|
tphurl := re.TelegraphUrlRegexp.FindString(tgutil.ExtractMessageEntityUrlsText(update.EffectiveMessage.Message))
|
||||||
if tphurl == "" {
|
if tphurl == "" {
|
||||||
logger.Warnf("No telegraph url found but called handleTelegraph")
|
logger.Warnf("No telegraph url found but called handleTelegraph")
|
||||||
return nil, nil, dispatcher.ContinueGroups
|
return nil, nil, dispatcher.ContinueGroups
|
||||||
@@ -173,18 +188,24 @@ func GetTphPicsFromMessageWithReply(ctx *ext.Context, update *ext.Update) (*type
|
|||||||
tphdir, err := url.PathUnescape(pagepath)
|
tphdir, err := url.PathUnescape(pagepath)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Errorf("Failed to unescape telegraph path: %s", err)
|
logger.Errorf("Failed to unescape telegraph path: %s", err)
|
||||||
ctx.Reply(update, ext.ReplyTextString("解析 telegraph 路径失败: "+err.Error()), nil)
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgCommonErrorParseTelegraphPathFailed, map[string]any{
|
||||||
|
"Error": err.Error(),
|
||||||
|
})), nil)
|
||||||
return nil, nil, dispatcher.EndGroups
|
return nil, nil, dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
msg, err := ctx.Reply(update, ext.ReplyTextString("正在获取 telegraph 页面..."), nil)
|
tphdir = strings.TrimSpace(tphdir)
|
||||||
|
msg, err := ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgCommonInfoFetchingTelegraphPage, nil)), nil)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Errorf("Failed to reply to update: %s", err)
|
logger.Errorf("Failed to reply to update: %s", err)
|
||||||
return nil, nil, dispatcher.EndGroups
|
return nil, nil, dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
|
logger.Debugf("Fetching telegraph page: %s", pagepath)
|
||||||
page, err := tphutil.DefaultClient().GetPage(ctx, pagepath)
|
page, err := tphutil.DefaultClient().GetPage(ctx, pagepath)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Errorf("Failed to get telegraph page: %s", err)
|
logger.Errorf("Failed to get telegraph page: %s", err)
|
||||||
ctx.Reply(update, ext.ReplyTextString("获取 telegraph 页面失败: "+err.Error()), nil)
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgCommonErrorGetTelegraphPageFailed, map[string]any{
|
||||||
|
"Error": err.Error(),
|
||||||
|
})), nil)
|
||||||
return nil, nil, dispatcher.EndGroups
|
return nil, nil, dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
imgs := make([]string, 0)
|
imgs := make([]string, 0)
|
||||||
@@ -208,13 +229,17 @@ func GetTphPicsFromMessageWithReply(ctx *ext.Context, update *ext.Update) (*type
|
|||||||
}
|
}
|
||||||
if node.Tag == "img" {
|
if node.Tag == "img" {
|
||||||
if src, ok := node.Attrs["src"]; ok {
|
if src, ok := node.Attrs["src"]; ok {
|
||||||
|
if strings.HasPrefix(src, "/file/") {
|
||||||
|
// handle images on telegra.ph server
|
||||||
|
src = "https://telegra.ph" + src
|
||||||
|
}
|
||||||
imgs = append(imgs, src)
|
imgs = append(imgs, src)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
if len(imgs) == 0 {
|
if len(imgs) == 0 {
|
||||||
logger.Warn("No images found in telegraph page")
|
logger.Warn("No images found in telegraph page")
|
||||||
ctx.Reply(update, ext.ReplyTextString("在 telegraph 页面中未找到图片"), nil)
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgCommonErrorNoImagesInTelegraphPage, nil)), nil)
|
||||||
return nil, nil, dispatcher.EndGroups
|
return nil, nil, dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
return msg, &TelegraphResult{
|
return msg, &TelegraphResult{
|
||||||
|
|||||||
@@ -6,9 +6,11 @@ import (
|
|||||||
"github.com/charmbracelet/log"
|
"github.com/charmbracelet/log"
|
||||||
"github.com/gotd/td/tg"
|
"github.com/gotd/td/tg"
|
||||||
"github.com/krau/SaveAny-Bot/client/bot/handlers/utils/msgelem"
|
"github.com/krau/SaveAny-Bot/client/bot/handlers/utils/msgelem"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/i18n"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/i18n/i18nk"
|
||||||
"github.com/krau/SaveAny-Bot/common/utils/tgutil"
|
"github.com/krau/SaveAny-Bot/common/utils/tgutil"
|
||||||
"github.com/krau/SaveAny-Bot/core"
|
"github.com/krau/SaveAny-Bot/core"
|
||||||
"github.com/krau/SaveAny-Bot/core/tasks/parsed"
|
parsed "github.com/krau/SaveAny-Bot/core/tasks/parsed"
|
||||||
"github.com/krau/SaveAny-Bot/pkg/parser"
|
"github.com/krau/SaveAny-Bot/pkg/parser"
|
||||||
"github.com/krau/SaveAny-Bot/storage"
|
"github.com/krau/SaveAny-Bot/storage"
|
||||||
"github.com/rs/xid"
|
"github.com/rs/xid"
|
||||||
@@ -20,8 +22,10 @@ func CreateAndAddParsedTaskWithEdit(ctx *ext.Context, stor storage.Storage, dirP
|
|||||||
if err := core.AddTask(injectCtx, task); err != nil {
|
if err := core.AddTask(injectCtx, task); err != nil {
|
||||||
log.FromContext(ctx).Errorf("Failed to add task: %s", err)
|
log.FromContext(ctx).Errorf("Failed to add task: %s", err)
|
||||||
ctx.EditMessage(userID, &tg.MessagesEditMessageRequest{
|
ctx.EditMessage(userID, &tg.MessagesEditMessageRequest{
|
||||||
ID: msgID,
|
ID: msgID,
|
||||||
Message: "任务添加失败: " + err.Error(),
|
Message: i18n.T(i18nk.BotMsgCommonErrorTaskAddFailed, map[string]any{
|
||||||
|
"Error": err.Error(),
|
||||||
|
}),
|
||||||
})
|
})
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,7 +1,6 @@
|
|||||||
package shortcut
|
package shortcut
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
|
||||||
"path"
|
"path"
|
||||||
"strings"
|
"strings"
|
||||||
|
|
||||||
@@ -11,6 +10,8 @@ import (
|
|||||||
"github.com/gotd/td/tg"
|
"github.com/gotd/td/tg"
|
||||||
"github.com/krau/SaveAny-Bot/client/bot/handlers/utils/msgelem"
|
"github.com/krau/SaveAny-Bot/client/bot/handlers/utils/msgelem"
|
||||||
"github.com/krau/SaveAny-Bot/client/bot/handlers/utils/ruleutil"
|
"github.com/krau/SaveAny-Bot/client/bot/handlers/utils/ruleutil"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/i18n"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/i18n/i18nk"
|
||||||
"github.com/krau/SaveAny-Bot/common/utils/tgutil"
|
"github.com/krau/SaveAny-Bot/common/utils/tgutil"
|
||||||
"github.com/krau/SaveAny-Bot/core"
|
"github.com/krau/SaveAny-Bot/core"
|
||||||
"github.com/krau/SaveAny-Bot/core/tasks/batchtfile"
|
"github.com/krau/SaveAny-Bot/core/tasks/batchtfile"
|
||||||
@@ -28,27 +29,36 @@ func CreateAndAddTGFileTaskWithEdit(ctx *ext.Context, userID int64, stor storage
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Errorf("Failed to get user by chat ID: %s", err)
|
logger.Errorf("Failed to get user by chat ID: %s", err)
|
||||||
ctx.EditMessage(userID, &tg.MessagesEditMessageRequest{
|
ctx.EditMessage(userID, &tg.MessagesEditMessageRequest{
|
||||||
ID: trackMsgID,
|
ID: trackMsgID,
|
||||||
Message: "获取用户失败: " + err.Error(),
|
Message: i18n.T(i18nk.BotMsgCommonErrorGetUserWithErrFailed, map[string]any{
|
||||||
|
"Error": err.Error(),
|
||||||
|
}),
|
||||||
})
|
})
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
if user.ApplyRule && user.Rules != nil {
|
if user.ApplyRule && user.Rules != nil {
|
||||||
matchedStorageName, matchedDirPath := ruleutil.ApplyRule(ctx, user.Rules, ruleutil.NewInput(file))
|
matched, matchedStorageName, matchedDirPath := ruleutil.ApplyRule(ctx, user.Rules, ruleutil.NewInput(file))
|
||||||
dirPath = matchedDirPath.String()
|
if !matched {
|
||||||
if matchedStorageName.IsUsable() {
|
goto startCreateTask
|
||||||
|
}
|
||||||
|
if matchedDirPath != "" {
|
||||||
|
dirPath = matchedDirPath.String()
|
||||||
|
}
|
||||||
|
if matchedStorageName.Usable() {
|
||||||
stor, err = storage.GetStorageByUserIDAndName(ctx, user.ChatID, matchedStorageName.String())
|
stor, err = storage.GetStorageByUserIDAndName(ctx, user.ChatID, matchedStorageName.String())
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Errorf("Failed to get storage by user ID and name: %s", err)
|
logger.Errorf("Failed to get storage by user ID and name: %s", err)
|
||||||
ctx.EditMessage(userID, &tg.MessagesEditMessageRequest{
|
ctx.EditMessage(userID, &tg.MessagesEditMessageRequest{
|
||||||
ID: trackMsgID,
|
ID: trackMsgID,
|
||||||
Message: "获取存储失败: " + err.Error(),
|
Message: i18n.T(i18nk.BotMsgCommonErrorGetStorageFailed, map[string]any{
|
||||||
|
"Error": err.Error(),
|
||||||
|
}),
|
||||||
})
|
})
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
startCreateTask:
|
||||||
storagePath := stor.JoinStoragePath(path.Join(dirPath, file.Name()))
|
storagePath := stor.JoinStoragePath(path.Join(dirPath, file.Name()))
|
||||||
injectCtx := tgutil.ExtWithContext(ctx.Context, ctx)
|
injectCtx := tgutil.ExtWithContext(ctx.Context, ctx)
|
||||||
taskid := xid.New().String()
|
taskid := xid.New().String()
|
||||||
@@ -59,16 +69,20 @@ func CreateAndAddTGFileTaskWithEdit(ctx *ext.Context, userID int64, stor storage
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Errorf("create task failed: %s", err)
|
logger.Errorf("create task failed: %s", err)
|
||||||
ctx.EditMessage(userID, &tg.MessagesEditMessageRequest{
|
ctx.EditMessage(userID, &tg.MessagesEditMessageRequest{
|
||||||
ID: trackMsgID,
|
ID: trackMsgID,
|
||||||
Message: "创建任务失败: " + err.Error(),
|
Message: i18n.T(i18nk.BotMsgCommonErrorTaskCreateFailed, map[string]any{
|
||||||
|
"Error": err.Error(),
|
||||||
|
}),
|
||||||
})
|
})
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
if err := core.AddTask(injectCtx, task); err != nil {
|
if err := core.AddTask(injectCtx, task); err != nil {
|
||||||
logger.Errorf("add task failed: %s", err)
|
logger.Errorf("add task failed: %s", err)
|
||||||
ctx.EditMessage(userID, &tg.MessagesEditMessageRequest{
|
ctx.EditMessage(userID, &tg.MessagesEditMessageRequest{
|
||||||
ID: trackMsgID,
|
ID: trackMsgID,
|
||||||
Message: "添加任务失败: " + err.Error(),
|
Message: i18n.T(i18nk.BotMsgCommonErrorTaskAddFailed, map[string]any{
|
||||||
|
"Error": err.Error(),
|
||||||
|
}),
|
||||||
})
|
})
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
@@ -89,8 +103,10 @@ func CreateAndAddBatchTGFileTaskWithEdit(ctx *ext.Context, userID int64, stor st
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Errorf("Failed to get user by chat ID: %s", err)
|
logger.Errorf("Failed to get user by chat ID: %s", err)
|
||||||
ctx.EditMessage(userID, &tg.MessagesEditMessageRequest{
|
ctx.EditMessage(userID, &tg.MessagesEditMessageRequest{
|
||||||
ID: trackMsgID,
|
ID: trackMsgID,
|
||||||
Message: "获取用户失败: " + err.Error(),
|
Message: i18n.T(i18nk.BotMsgCommonErrorGetUserWithErrFailed, map[string]any{
|
||||||
|
"Error": err.Error(),
|
||||||
|
}),
|
||||||
})
|
})
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
@@ -101,10 +117,12 @@ func CreateAndAddBatchTGFileTaskWithEdit(ctx *ext.Context, userID int64, stor st
|
|||||||
if !useRule {
|
if !useRule {
|
||||||
return stor.Name(), ruleutil.MatchedDirPath(dirPath)
|
return stor.Name(), ruleutil.MatchedDirPath(dirPath)
|
||||||
}
|
}
|
||||||
storName, dirP := ruleutil.ApplyRule(ctx, user.Rules, ruleutil.NewInput(file))
|
matched, storName, dirP := ruleutil.ApplyRule(ctx, user.Rules, ruleutil.NewInput(file))
|
||||||
|
if !matched {
|
||||||
|
return stor.Name(), ruleutil.MatchedDirPath(dirPath)
|
||||||
|
}
|
||||||
storname := storName.String()
|
storname := storName.String()
|
||||||
if !storName.IsUsable() {
|
if !storName.Usable() {
|
||||||
storname = stor.Name()
|
storname = stor.Name()
|
||||||
}
|
}
|
||||||
return storname, dirP
|
return storname, dirP
|
||||||
@@ -124,8 +142,10 @@ func CreateAndAddBatchTGFileTaskWithEdit(ctx *ext.Context, userID int64, stor st
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Errorf("Failed to get storage by user ID and name: %s", err)
|
logger.Errorf("Failed to get storage by user ID and name: %s", err)
|
||||||
ctx.EditMessage(userID, &tg.MessagesEditMessageRequest{
|
ctx.EditMessage(userID, &tg.MessagesEditMessageRequest{
|
||||||
ID: trackMsgID,
|
ID: trackMsgID,
|
||||||
Message: "获取存储失败: " + err.Error(),
|
Message: i18n.T(i18nk.BotMsgCommonErrorGetStorageFailed, map[string]any{
|
||||||
|
"Error": err.Error(),
|
||||||
|
}),
|
||||||
})
|
})
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
@@ -136,8 +156,10 @@ func CreateAndAddBatchTGFileTaskWithEdit(ctx *ext.Context, userID int64, stor st
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Errorf("Failed to create task element: %s", err)
|
logger.Errorf("Failed to create task element: %s", err)
|
||||||
ctx.EditMessage(userID, &tg.MessagesEditMessageRequest{
|
ctx.EditMessage(userID, &tg.MessagesEditMessageRequest{
|
||||||
ID: trackMsgID,
|
ID: trackMsgID,
|
||||||
Message: "任务创建失败: " + err.Error(),
|
Message: i18n.T(i18nk.BotMsgCommonErrorTaskCreateFailed, map[string]any{
|
||||||
|
"Error": err.Error(),
|
||||||
|
}),
|
||||||
})
|
})
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
@@ -171,8 +193,10 @@ func CreateAndAddBatchTGFileTaskWithEdit(ctx *ext.Context, userID int64, stor st
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Errorf("Failed to create task element for album file: %s", err)
|
logger.Errorf("Failed to create task element for album file: %s", err)
|
||||||
ctx.EditMessage(userID, &tg.MessagesEditMessageRequest{
|
ctx.EditMessage(userID, &tg.MessagesEditMessageRequest{
|
||||||
ID: trackMsgID,
|
ID: trackMsgID,
|
||||||
Message: "任务创建失败: " + err.Error(),
|
Message: i18n.T(i18nk.BotMsgCommonErrorTaskCreateFailed, map[string]any{
|
||||||
|
"Error": err.Error(),
|
||||||
|
}),
|
||||||
})
|
})
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
@@ -186,14 +210,18 @@ func CreateAndAddBatchTGFileTaskWithEdit(ctx *ext.Context, userID int64, stor st
|
|||||||
if err := core.AddTask(injectCtx, task); err != nil {
|
if err := core.AddTask(injectCtx, task); err != nil {
|
||||||
logger.Errorf("Failed to add batch task: %s", err)
|
logger.Errorf("Failed to add batch task: %s", err)
|
||||||
ctx.EditMessage(userID, &tg.MessagesEditMessageRequest{
|
ctx.EditMessage(userID, &tg.MessagesEditMessageRequest{
|
||||||
ID: trackMsgID,
|
ID: trackMsgID,
|
||||||
Message: "批量任务添加失败: " + err.Error(),
|
Message: i18n.T(i18nk.BotMsgCommonErrorTaskAddFailed, map[string]any{
|
||||||
|
"Error": err.Error(),
|
||||||
|
}),
|
||||||
})
|
})
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
ctx.EditMessage(userID, &tg.MessagesEditMessageRequest{
|
ctx.EditMessage(userID, &tg.MessagesEditMessageRequest{
|
||||||
ID: trackMsgID,
|
ID: trackMsgID,
|
||||||
Message: fmt.Sprintf("已添加批量任务, 共 %d 个文件", len(files)),
|
Message: i18n.T(i18nk.BotMsgCommonInfoBatchTasksAdded, map[string]any{
|
||||||
|
"Count": len(files),
|
||||||
|
}),
|
||||||
ReplyMarkup: nil,
|
ReplyMarkup: nil,
|
||||||
})
|
})
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
|
|||||||
@@ -6,6 +6,8 @@ import (
|
|||||||
"github.com/charmbracelet/log"
|
"github.com/charmbracelet/log"
|
||||||
"github.com/gotd/td/tg"
|
"github.com/gotd/td/tg"
|
||||||
"github.com/krau/SaveAny-Bot/client/bot/handlers/utils/msgelem"
|
"github.com/krau/SaveAny-Bot/client/bot/handlers/utils/msgelem"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/i18n"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/i18n/i18nk"
|
||||||
"github.com/krau/SaveAny-Bot/common/utils/tgutil"
|
"github.com/krau/SaveAny-Bot/common/utils/tgutil"
|
||||||
"github.com/krau/SaveAny-Bot/common/utils/tphutil"
|
"github.com/krau/SaveAny-Bot/common/utils/tphutil"
|
||||||
"github.com/krau/SaveAny-Bot/core"
|
"github.com/krau/SaveAny-Bot/core"
|
||||||
@@ -23,7 +25,7 @@ func CreateAndAddtelegraphWithEdit(
|
|||||||
pics []string,
|
pics []string,
|
||||||
stor storage.Storage,
|
stor storage.Storage,
|
||||||
trackMsgID int) error {
|
trackMsgID int) error {
|
||||||
|
|
||||||
injectCtx := tgutil.ExtWithContext(ctx.Context, ctx)
|
injectCtx := tgutil.ExtWithContext(ctx.Context, ctx)
|
||||||
task := tphtask.NewTask(xid.New().String(),
|
task := tphtask.NewTask(xid.New().String(),
|
||||||
injectCtx,
|
injectCtx,
|
||||||
@@ -37,8 +39,10 @@ func CreateAndAddtelegraphWithEdit(
|
|||||||
if err := core.AddTask(injectCtx, task); err != nil {
|
if err := core.AddTask(injectCtx, task); err != nil {
|
||||||
log.FromContext(ctx).Errorf("Failed to add task: %s", err)
|
log.FromContext(ctx).Errorf("Failed to add task: %s", err)
|
||||||
ctx.EditMessage(userID, &tg.MessagesEditMessageRequest{
|
ctx.EditMessage(userID, &tg.MessagesEditMessageRequest{
|
||||||
ID: trackMsgID,
|
ID: trackMsgID,
|
||||||
Message: "任务添加失败: " + err.Error(),
|
Message: i18n.T(i18nk.BotMsgCommonErrorTaskAddFailed, map[string]any{
|
||||||
|
"Error": err.Error(),
|
||||||
|
}),
|
||||||
})
|
})
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
|
|||||||
63
client/bot/handlers/utils/shortcut/ytdlp.go
Normal file
63
client/bot/handlers/utils/shortcut/ytdlp.go
Normal file
@@ -0,0 +1,63 @@
|
|||||||
|
package shortcut
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/celestix/gotgproto/dispatcher"
|
||||||
|
"github.com/celestix/gotgproto/ext"
|
||||||
|
"github.com/charmbracelet/log"
|
||||||
|
"github.com/gotd/td/tg"
|
||||||
|
"github.com/rs/xid"
|
||||||
|
|
||||||
|
"github.com/krau/SaveAny-Bot/common/i18n"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/i18n/i18nk"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/utils/tgutil"
|
||||||
|
"github.com/krau/SaveAny-Bot/core"
|
||||||
|
"github.com/krau/SaveAny-Bot/core/tasks/ytdlp"
|
||||||
|
"github.com/krau/SaveAny-Bot/storage"
|
||||||
|
)
|
||||||
|
|
||||||
|
func CreateAndAddYtdlpTaskWithEdit(ctx *ext.Context, stor storage.Storage, dirPath string, urls []string, flags []string, msgID int, userID int64) error {
|
||||||
|
logger := log.FromContext(ctx)
|
||||||
|
injectCtx := tgutil.ExtWithContext(ctx.Context, ctx)
|
||||||
|
|
||||||
|
// Validate URLs
|
||||||
|
if len(urls) == 0 {
|
||||||
|
logger.Error("URLs list is empty")
|
||||||
|
ctx.EditMessage(userID, &tg.MessagesEditMessageRequest{
|
||||||
|
ID: msgID,
|
||||||
|
Message: i18n.T(i18nk.BotMsgYtdlpErrorNoValidUrls, nil),
|
||||||
|
})
|
||||||
|
return dispatcher.EndGroups
|
||||||
|
}
|
||||||
|
|
||||||
|
logger.Infof("Creating yt-dlp task for %d URL(s) with %d flag(s)", len(urls), len(flags))
|
||||||
|
|
||||||
|
// Create yt-dlp task
|
||||||
|
task := ytdlp.NewTask(
|
||||||
|
xid.New().String(),
|
||||||
|
injectCtx,
|
||||||
|
urls,
|
||||||
|
flags,
|
||||||
|
stor,
|
||||||
|
stor.JoinStoragePath(dirPath),
|
||||||
|
ytdlp.NewProgress(msgID, userID),
|
||||||
|
)
|
||||||
|
|
||||||
|
// Add task to queue
|
||||||
|
if err := core.AddTask(injectCtx, task); err != nil {
|
||||||
|
logger.Errorf("Failed to add yt-dlp task: %s", err)
|
||||||
|
ctx.EditMessage(userID, &tg.MessagesEditMessageRequest{
|
||||||
|
ID: msgID,
|
||||||
|
Message: i18n.T(i18nk.BotMsgCommonErrorTaskAddFailed, map[string]any{
|
||||||
|
"Error": err.Error(),
|
||||||
|
}),
|
||||||
|
})
|
||||||
|
return dispatcher.EndGroups
|
||||||
|
}
|
||||||
|
|
||||||
|
ctx.EditMessage(userID, &tg.MessagesEditMessageRequest{
|
||||||
|
ID: msgID,
|
||||||
|
Message: i18n.T(i18nk.BotMsgCommonInfoTaskAdded, nil),
|
||||||
|
})
|
||||||
|
|
||||||
|
return dispatcher.EndGroups
|
||||||
|
}
|
||||||
@@ -1,39 +1,55 @@
|
|||||||
package handlers
|
package handlers
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"fmt"
|
||||||
|
"path"
|
||||||
"regexp"
|
"regexp"
|
||||||
"strings"
|
"strings"
|
||||||
|
"sync"
|
||||||
|
"text/template"
|
||||||
|
"time"
|
||||||
|
|
||||||
"github.com/celestix/gotgproto/dispatcher"
|
"github.com/celestix/gotgproto/dispatcher"
|
||||||
"github.com/celestix/gotgproto/ext"
|
"github.com/celestix/gotgproto/ext"
|
||||||
"github.com/charmbracelet/log"
|
"github.com/charmbracelet/log"
|
||||||
"github.com/krau/SaveAny-Bot/client/bot/handlers/utils/msgelem"
|
"github.com/krau/SaveAny-Bot/client/bot/handlers/utils/mediautil"
|
||||||
|
"github.com/krau/SaveAny-Bot/client/bot/handlers/utils/ruleutil"
|
||||||
|
userclient "github.com/krau/SaveAny-Bot/client/user"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/i18n"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/i18n/i18nk"
|
||||||
"github.com/krau/SaveAny-Bot/common/utils/tgutil"
|
"github.com/krau/SaveAny-Bot/common/utils/tgutil"
|
||||||
|
"github.com/krau/SaveAny-Bot/config"
|
||||||
|
"github.com/krau/SaveAny-Bot/core"
|
||||||
|
coretfile "github.com/krau/SaveAny-Bot/core/tasks/tfile"
|
||||||
"github.com/krau/SaveAny-Bot/database"
|
"github.com/krau/SaveAny-Bot/database"
|
||||||
|
"github.com/krau/SaveAny-Bot/pkg/enums/fnamest"
|
||||||
|
"github.com/krau/SaveAny-Bot/pkg/tfile"
|
||||||
|
"github.com/krau/SaveAny-Bot/storage"
|
||||||
|
"github.com/rs/xid"
|
||||||
)
|
)
|
||||||
|
|
||||||
func handleWatchCmd(ctx *ext.Context, update *ext.Update) error {
|
func handleWatchCmd(ctx *ext.Context, update *ext.Update) error {
|
||||||
logger := log.FromContext(ctx)
|
logger := log.FromContext(ctx)
|
||||||
args := strings.Split(string(update.EffectiveMessage.Text), " ")
|
args := strings.Split(update.EffectiveMessage.Text, " ")
|
||||||
if len(args) < 2 {
|
if len(args) < 2 {
|
||||||
ctx.Reply(update, ext.ReplyTextString(msgelem.WatchHelpText), nil)
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgWatchHelpText)), nil)
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
userChatID := update.GetUserChat().GetID()
|
userChatID := update.GetUserChat().GetID()
|
||||||
user, err := database.GetUserByChatID(ctx, userChatID)
|
user, err := database.GetUserByChatID(ctx, userChatID)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Errorf("获取用户失败: %s", err)
|
logger.Errorf("Failed to get user: %s", err)
|
||||||
ctx.Reply(update, ext.ReplyTextString("获取用户失败"), nil)
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgCommonErrorGetUserFailed)), nil)
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
if user.DefaultStorage == "" {
|
if user.DefaultStorage == "" {
|
||||||
ctx.Reply(update, ext.ReplyTextString("请先设置默认存储, 使用 /storage 命令"), nil)
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgCommonErrorDefaultStorageNotSet)), nil)
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
chatArg := args[1]
|
chatArg := args[1]
|
||||||
chatID, err := tgutil.ParseChatID(ctx, chatArg)
|
chatID, err := tgutil.ParseChatID(ctx, chatArg)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
ctx.Reply(update, ext.ReplyTextString("无效的ID或用户名: "+err.Error()), nil)
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgCommonErrorInvalidIdOrUsername, map[string]any{"Error": err.Error()})), nil)
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
watching, err := user.WatchingChat(ctx, chatID)
|
watching, err := user.WatchingChat(ctx, chatID)
|
||||||
@@ -42,7 +58,7 @@ func handleWatchCmd(ctx *ext.Context, update *ext.Update) error {
|
|||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
if watching {
|
if watching {
|
||||||
ctx.Reply(update, ext.ReplyTextString("已经在监听此聊天"), nil)
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgWatchInfoAlreadyWatchingChat)), nil)
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
filter := ""
|
filter := ""
|
||||||
@@ -51,19 +67,19 @@ func handleWatchCmd(ctx *ext.Context, update *ext.Update) error {
|
|||||||
filterType := strings.Split(filterArg, ":")[0]
|
filterType := strings.Split(filterArg, ":")[0]
|
||||||
filterData := strings.Split(filterArg, ":")[1]
|
filterData := strings.Split(filterArg, ":")[1]
|
||||||
if filterType == "" || filterData == "" {
|
if filterType == "" || filterData == "" {
|
||||||
ctx.Reply(update, ext.ReplyTextString("过滤器格式错误, 请使用 <过滤器类型>:<表达式>"), nil)
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgWatchErrorFilterFormatInvalid)), nil)
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
switch filterType {
|
switch filterType {
|
||||||
case "msgre":
|
case "msgre":
|
||||||
_, err := regexp.Compile(filterData)
|
_, err := regexp.Compile(filterData)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
ctx.Reply(update, ext.ReplyTextString("正则表达式格式错误: "+err.Error()), nil)
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgCommonErrorInvalidRegex, map[string]any{"Error": err.Error()})), nil)
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
filter = filterType + ":" + filterData
|
filter = filterType + ":" + filterData
|
||||||
default:
|
default:
|
||||||
ctx.Reply(update, ext.ReplyTextString("不支持的过滤器类型, 请参阅文档"), nil)
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgWatchErrorFilterTypeUnsupported)), nil)
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -73,38 +89,333 @@ func handleWatchCmd(ctx *ext.Context, update *ext.Update) error {
|
|||||||
Filter: filter,
|
Filter: filter,
|
||||||
}); err != nil {
|
}); err != nil {
|
||||||
logger.Errorf("Failed to watch chat %d: %s", chatID, err)
|
logger.Errorf("Failed to watch chat %d: %s", chatID, err)
|
||||||
ctx.Reply(update, ext.ReplyTextString("监听聊天失败: "+err.Error()), nil)
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgWatchErrorWatchChatFailed, map[string]any{"Error": err.Error()})), nil)
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
ctx.Reply(update, ext.ReplyTextString("已开始监听聊天: "+chatArg), nil)
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgWatchInfoWatchChatStarted, map[string]any{"Chat": chatArg})), nil)
|
||||||
|
return dispatcher.EndGroups
|
||||||
|
}
|
||||||
|
|
||||||
|
func handleLswatchCmd(ctx *ext.Context, update *ext.Update) error {
|
||||||
|
logger := log.FromContext(ctx)
|
||||||
|
userChatID := update.GetUserChat().GetID()
|
||||||
|
user, err := database.GetUserByChatID(ctx, userChatID)
|
||||||
|
if err != nil {
|
||||||
|
logger.Errorf("Failed to get user: %s", err)
|
||||||
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgCommonErrorGetUserFailed)), nil)
|
||||||
|
return dispatcher.EndGroups
|
||||||
|
}
|
||||||
|
chats := user.WatchChats
|
||||||
|
if len(chats) == 0 {
|
||||||
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgWatchInfoWatchListEmpty)), nil)
|
||||||
|
return dispatcher.EndGroups
|
||||||
|
}
|
||||||
|
var sb strings.Builder
|
||||||
|
sb.WriteString(i18n.T(i18nk.BotMsgWatchInfoWatchListHeader))
|
||||||
|
for _, chat := range chats {
|
||||||
|
sb.WriteString("- ")
|
||||||
|
sb.WriteString(fmt.Sprintf("%d", chat.ChatID))
|
||||||
|
if chat.Filter != "" {
|
||||||
|
sb.WriteString(i18n.T(i18nk.BotMsgWatchInfoWatchListFilterPrefix))
|
||||||
|
sb.WriteString(chat.Filter)
|
||||||
|
sb.WriteString(")")
|
||||||
|
}
|
||||||
|
sb.WriteString("\n")
|
||||||
|
}
|
||||||
|
ctx.Reply(update, ext.ReplyTextString(sb.String()), nil)
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
|
|
||||||
func handleUnwatchCmd(ctx *ext.Context, update *ext.Update) error {
|
func handleUnwatchCmd(ctx *ext.Context, update *ext.Update) error {
|
||||||
logger := log.FromContext(ctx)
|
logger := log.FromContext(ctx)
|
||||||
args := strings.Split(string(update.EffectiveMessage.Text), " ")
|
args := strings.Split(update.EffectiveMessage.Text, " ")
|
||||||
if len(args) < 2 {
|
if len(args) < 2 {
|
||||||
ctx.Reply(update, ext.ReplyTextString("请提供要取消监听的聊天ID或用户名"), nil)
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgWatchErrorUnwatchNoChatProvided)), nil)
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
userChatID := update.GetUserChat().GetID()
|
userChatID := update.GetUserChat().GetID()
|
||||||
user, err := database.GetUserByChatID(ctx, userChatID)
|
user, err := database.GetUserByChatID(ctx, userChatID)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Errorf("获取用户失败: %s", err)
|
logger.Errorf("Failed to get user: %s", err)
|
||||||
ctx.Reply(update, ext.ReplyTextString("获取用户失败"), nil)
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgCommonErrorGetUserFailed)), nil)
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
chatArg := args[1]
|
chatArg := args[1]
|
||||||
chatID, err := tgutil.ParseChatID(ctx, chatArg)
|
chatID, err := tgutil.ParseChatID(ctx, chatArg)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
ctx.Reply(update, ext.ReplyTextString("无效的ID或用户名: "+err.Error()), nil)
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgCommonErrorInvalidIdOrUsername, map[string]any{"Error": err.Error()})), nil)
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
if err := user.UnwatchChat(ctx, chatID); err != nil {
|
if err := user.UnwatchChat(ctx, chatID); err != nil {
|
||||||
logger.Errorf("Failed to unwatch chat %d: %s", chatID, err)
|
logger.Errorf("Failed to unwatch chat %d: %s", chatID, err)
|
||||||
ctx.Reply(update, ext.ReplyTextString("取消监听聊天失败: "+err.Error()), nil)
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgWatchErrorUnwatchChatFailed, map[string]any{"Error": err.Error()})), nil)
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
ctx.Reply(update, ext.ReplyTextString("已取消监听聊天: "+chatArg), nil)
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgWatchInfoWatchChatStopped, map[string]any{"Chat": chatArg})), nil)
|
||||||
return dispatcher.EndGroups
|
return dispatcher.EndGroups
|
||||||
}
|
}
|
||||||
|
|
||||||
|
type watchMediaGroupHandler struct {
|
||||||
|
groups map[int64]map[uint][]tfile.TGFileMessage // chatID -> userID -> files
|
||||||
|
timers map[int64]map[uint]*time.Timer
|
||||||
|
mu sync.Mutex
|
||||||
|
}
|
||||||
|
|
||||||
|
var watchMediaGroupMgr = &watchMediaGroupHandler{
|
||||||
|
groups: make(map[int64]map[uint][]tfile.TGFileMessage),
|
||||||
|
timers: make(map[int64]map[uint]*time.Timer),
|
||||||
|
}
|
||||||
|
|
||||||
|
func (w *watchMediaGroupHandler) addFile(chatID int64, userID uint, file tfile.TGFileMessage, timeout time.Duration, callback func([]tfile.TGFileMessage)) {
|
||||||
|
w.mu.Lock()
|
||||||
|
defer w.mu.Unlock()
|
||||||
|
|
||||||
|
if w.groups[chatID] == nil {
|
||||||
|
w.groups[chatID] = make(map[uint][]tfile.TGFileMessage)
|
||||||
|
}
|
||||||
|
if w.timers[chatID] == nil {
|
||||||
|
w.timers[chatID] = make(map[uint]*time.Timer)
|
||||||
|
}
|
||||||
|
|
||||||
|
if timer, exists := w.timers[chatID][userID]; exists {
|
||||||
|
timer.Stop()
|
||||||
|
}
|
||||||
|
|
||||||
|
w.groups[chatID][userID] = append(w.groups[chatID][userID], file)
|
||||||
|
|
||||||
|
w.timers[chatID][userID] = time.AfterFunc(timeout, func() {
|
||||||
|
w.mu.Lock()
|
||||||
|
files := w.groups[chatID][userID]
|
||||||
|
delete(w.groups[chatID], userID)
|
||||||
|
delete(w.timers[chatID], userID)
|
||||||
|
if len(w.groups[chatID]) == 0 {
|
||||||
|
delete(w.groups, chatID)
|
||||||
|
}
|
||||||
|
if len(w.timers[chatID]) == 0 {
|
||||||
|
delete(w.timers, chatID)
|
||||||
|
}
|
||||||
|
w.mu.Unlock()
|
||||||
|
|
||||||
|
if len(files) > 0 {
|
||||||
|
callback(files)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
func listenMediaMessageEvent(ch chan userclient.MediaMessageEvent) {
|
||||||
|
if userclient.GetCtx() == nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
logger := log.FromContext(userclient.GetCtx())
|
||||||
|
for event := range ch {
|
||||||
|
logger.Debug("Received media message event", "chat_id", event.ChatID, "file_name", event.File.Name())
|
||||||
|
ctx := event.Ctx
|
||||||
|
file := event.File
|
||||||
|
chats, err := database.GetWatchChatsByChatID(ctx, event.ChatID)
|
||||||
|
if err != nil {
|
||||||
|
logger.Errorf("Failed to get watch chats for chat ID %d: %v", event.ChatID, err)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
msgText := event.File.Message().GetMessage()
|
||||||
|
for _, chat := range chats {
|
||||||
|
if chat.Filter != "" {
|
||||||
|
filter := strings.Split(chat.Filter, ":")
|
||||||
|
if len(filter) != 2 {
|
||||||
|
logger.Warnf("Invalid filter format in chat %d, skipping", chat.ChatID)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
filterType := filter[0]
|
||||||
|
filterData := filter[1]
|
||||||
|
switch filterType {
|
||||||
|
case "msgre": // [TODO] enums for filter types
|
||||||
|
if ok, err := regexp.MatchString(filterData, msgText); err != nil {
|
||||||
|
continue
|
||||||
|
} else if !ok {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
default:
|
||||||
|
logger.Warnf("Unsupported filter type %s in chat %d, skipping", filterType, chat.ChatID)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
}
|
||||||
|
user, err := database.GetUserByID(ctx, chat.UserID)
|
||||||
|
if err != nil {
|
||||||
|
logger.Errorf("Failed to get user by ID %d: %v", chat.UserID, err)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if user.DefaultStorage == "" {
|
||||||
|
logger.Warnf("User %d has no default storage set, skipping media message handling", chat.UserID)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
stor, err := storage.GetStorageByUserIDAndName(ctx, user.ChatID, user.DefaultStorage)
|
||||||
|
if err != nil {
|
||||||
|
logger.Errorf("Failed to get storage by user ID %d and name %s: %v", user.ChatID, user.DefaultStorage, err)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
switch user.FilenameStrategy {
|
||||||
|
case fnamest.Message.String():
|
||||||
|
file.SetName(tgutil.GenFileNameFromMessage(*file.Message()))
|
||||||
|
case fnamest.Template.String():
|
||||||
|
if user.FilenameTemplate == "" {
|
||||||
|
logger.Warnf("Empty filename template for user %d, using default filename", user.ChatID)
|
||||||
|
break
|
||||||
|
}
|
||||||
|
message := file.Message()
|
||||||
|
tmpl, err := template.New("filename").Parse(user.FilenameTemplate)
|
||||||
|
if err != nil {
|
||||||
|
logger.Errorf("Failed to parse filename template for user %d: %s", user.ChatID, err)
|
||||||
|
break
|
||||||
|
}
|
||||||
|
data := mediautil.BuildFilenameTemplateData(message)
|
||||||
|
var sb strings.Builder
|
||||||
|
err = tmpl.Execute(&sb, data)
|
||||||
|
if err != nil {
|
||||||
|
log.FromContext(ctx).Errorf("failed to execute filename template: %s", err)
|
||||||
|
break
|
||||||
|
}
|
||||||
|
file.SetName(sb.String())
|
||||||
|
}
|
||||||
|
|
||||||
|
// Check if this is a media group and if rules specify NEW-FOR-ALBUM
|
||||||
|
groupID, isGroup := file.Message().GetGroupedID()
|
||||||
|
needAlbumHandling := false
|
||||||
|
if isGroup && groupID != 0 && user.ApplyRule && user.Rules != nil {
|
||||||
|
_, _, matchedDirPath := ruleutil.ApplyRule(ctx, user.Rules, ruleutil.NewInput(file))
|
||||||
|
needAlbumHandling = matchedDirPath.NeedNewForAlbum()
|
||||||
|
}
|
||||||
|
|
||||||
|
if needAlbumHandling {
|
||||||
|
// For media groups with NEW-FOR-ALBUM rule, collect all files of the same group
|
||||||
|
watchMediaGroupMgr.addFile(event.ChatID, user.ID, file, time.Duration(config.C().Telegram.MediaGroupTimeout)*time.Second, func(files []tfile.TGFileMessage) {
|
||||||
|
processWatchMediaGroup(ctx, user, stor, "", files)
|
||||||
|
})
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
// Process single file or media group without album folder creation
|
||||||
|
var dirPath string
|
||||||
|
if user.ApplyRule && user.Rules != nil {
|
||||||
|
matched, matchedStorageName, matchedDirPath := ruleutil.ApplyRule(ctx, user.Rules, ruleutil.NewInput(file))
|
||||||
|
if !matched {
|
||||||
|
goto startCreateTask
|
||||||
|
}
|
||||||
|
dirPath = matchedDirPath.String()
|
||||||
|
if matchedStorageName.Usable() {
|
||||||
|
stor, err = storage.GetStorageByUserIDAndName(ctx, user.ChatID, matchedStorageName.String())
|
||||||
|
if err != nil {
|
||||||
|
logger.Errorf("Failed to get storage by user ID and name: %s", err)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
startCreateTask:
|
||||||
|
storagePath := stor.JoinStoragePath(path.Join(dirPath, file.Name()))
|
||||||
|
injectCtx := tgutil.ExtWithContext(ctx.Context, ctx)
|
||||||
|
taskid := xid.New().String()
|
||||||
|
task, err := coretfile.NewTGFileTask(taskid, injectCtx, file, stor, storagePath, nil)
|
||||||
|
if err != nil {
|
||||||
|
logger.Errorf("create task failed: %s", err)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if err := core.AddTask(injectCtx, task); err != nil {
|
||||||
|
logger.Errorf("add task failed: %s", err)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
logger.Infof("Added media message task for user %d in chat %d: %s", chat.UserID, event.ChatID, file.Name())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func processWatchMediaGroup(ctx *ext.Context, user *database.User, stor storage.Storage, dirPath string, files []tfile.TGFileMessage) {
|
||||||
|
logger := log.FromContext(ctx)
|
||||||
|
if len(files) == 0 {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
useRule := user.ApplyRule && user.Rules != nil
|
||||||
|
|
||||||
|
applyRule := func(file tfile.TGFileMessage) (string, ruleutil.MatchedDirPath) {
|
||||||
|
if !useRule {
|
||||||
|
return stor.Name(), ruleutil.MatchedDirPath(dirPath)
|
||||||
|
}
|
||||||
|
matched, storName, dirP := ruleutil.ApplyRule(ctx, user.Rules, ruleutil.NewInput(file))
|
||||||
|
if !matched {
|
||||||
|
return stor.Name(), ruleutil.MatchedDirPath(dirPath)
|
||||||
|
}
|
||||||
|
storname := storName.String()
|
||||||
|
if !storName.Usable() {
|
||||||
|
storname = stor.Name()
|
||||||
|
}
|
||||||
|
return storname, dirP
|
||||||
|
}
|
||||||
|
|
||||||
|
type albumFile struct {
|
||||||
|
file tfile.TGFileMessage
|
||||||
|
storage storage.Storage
|
||||||
|
}
|
||||||
|
albumFiles := make(map[int64][]albumFile)
|
||||||
|
|
||||||
|
// Collect files by group ID
|
||||||
|
for _, file := range files {
|
||||||
|
storName, ruleDirPath := applyRule(file)
|
||||||
|
fileStor := stor
|
||||||
|
if storName != stor.Name() && storName != "" {
|
||||||
|
var err error
|
||||||
|
fileStor, err = storage.GetStorageByUserIDAndName(ctx, user.ChatID, storName)
|
||||||
|
if err != nil {
|
||||||
|
logger.Errorf("Failed to get storage by user ID and name: %s", err)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
groupId, isGroup := file.Message().GetGroupedID()
|
||||||
|
if !isGroup || groupId == 0 {
|
||||||
|
logger.Warnf("File %s is not in a group, skipping", file.Name())
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
if !ruleDirPath.NeedNewForAlbum() {
|
||||||
|
logger.Warnf("File %s does not need album folder, skipping", file.Name())
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
if _, ok := albumFiles[groupId]; !ok {
|
||||||
|
albumFiles[groupId] = make([]albumFile, 0)
|
||||||
|
}
|
||||||
|
albumFiles[groupId] = append(albumFiles[groupId], albumFile{
|
||||||
|
file: file,
|
||||||
|
storage: fileStor,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
// Process album files with folder creation
|
||||||
|
injectCtx := tgutil.ExtWithContext(ctx.Context, ctx)
|
||||||
|
totalTasks := 0
|
||||||
|
for groupID, afiles := range albumFiles {
|
||||||
|
if len(afiles) <= 1 {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
// Use first file's name (without extension) as album folder name
|
||||||
|
albumDir := strings.TrimSuffix(path.Base(afiles[0].file.Name()), path.Ext(afiles[0].file.Name()))
|
||||||
|
albumStor := afiles[0].storage
|
||||||
|
|
||||||
|
logger.Infof("Creating album folder for group %d: %s with %d files", groupID, albumDir, len(afiles))
|
||||||
|
|
||||||
|
for _, af := range afiles {
|
||||||
|
afstorPath := af.storage.JoinStoragePath(path.Join(dirPath, albumDir, af.file.Name()))
|
||||||
|
taskid := xid.New().String()
|
||||||
|
task, err := coretfile.NewTGFileTask(taskid, injectCtx, af.file, albumStor, afstorPath, nil)
|
||||||
|
if err != nil {
|
||||||
|
logger.Errorf("create task failed for album file: %s", err)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if err := core.AddTask(injectCtx, task); err != nil {
|
||||||
|
logger.Errorf("add task failed: %s", err)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
totalTasks++
|
||||||
|
}
|
||||||
|
}
|
||||||
|
logger.Infof("Added %d watch media tasks for user %d", totalTasks, user.ChatID)
|
||||||
|
}
|
||||||
|
|||||||
92
client/bot/handlers/ytdlp.go
Normal file
92
client/bot/handlers/ytdlp.go
Normal file
@@ -0,0 +1,92 @@
|
|||||||
|
package handlers
|
||||||
|
|
||||||
|
import (
|
||||||
|
"net/url"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
"github.com/celestix/gotgproto/dispatcher"
|
||||||
|
"github.com/celestix/gotgproto/ext"
|
||||||
|
"github.com/charmbracelet/log"
|
||||||
|
|
||||||
|
"github.com/krau/SaveAny-Bot/client/bot/handlers/utils/msgelem"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/i18n"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/i18n/i18nk"
|
||||||
|
"github.com/krau/SaveAny-Bot/pkg/enums/tasktype"
|
||||||
|
"github.com/krau/SaveAny-Bot/pkg/tcbdata"
|
||||||
|
"github.com/krau/SaveAny-Bot/storage"
|
||||||
|
)
|
||||||
|
|
||||||
|
func handleYtdlpCmd(ctx *ext.Context, update *ext.Update) error {
|
||||||
|
logger := log.FromContext(ctx)
|
||||||
|
args := strings.Split(update.EffectiveMessage.Text, " ")
|
||||||
|
if len(args) < 2 {
|
||||||
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgYtdlpUsage)), nil)
|
||||||
|
return dispatcher.EndGroups
|
||||||
|
}
|
||||||
|
|
||||||
|
// Separate URLs and flags from arguments
|
||||||
|
var urls []string
|
||||||
|
var flags []string
|
||||||
|
|
||||||
|
for i := 1; i < len(args); i++ {
|
||||||
|
arg := strings.TrimSpace(args[i])
|
||||||
|
if arg == "" {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
// Check if it's a flag (starts with - or --)
|
||||||
|
if strings.HasPrefix(arg, "-") {
|
||||||
|
flags = append(flags, arg)
|
||||||
|
// Check if the next argument might be a value for this flag
|
||||||
|
// Don't consume it if it starts with - or looks like a URL with scheme
|
||||||
|
if i+1 < len(args) {
|
||||||
|
nextArg := strings.TrimSpace(args[i+1])
|
||||||
|
if nextArg != "" && !strings.HasPrefix(nextArg, "-") {
|
||||||
|
// Check if it's clearly a URL (has ://)
|
||||||
|
// This handles common video URLs (http://, https://)
|
||||||
|
// For other yt-dlp inputs, users should ensure proper formatting
|
||||||
|
if strings.Contains(nextArg, "://") {
|
||||||
|
// It's a URL, don't consume it as a flag value
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
// Otherwise, treat it as a flag value
|
||||||
|
flags = append(flags, nextArg)
|
||||||
|
i++ // Skip the next argument as it's been consumed
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
// Try to parse as URL
|
||||||
|
u, err := url.Parse(arg)
|
||||||
|
if err != nil || u.Scheme == "" || u.Host == "" {
|
||||||
|
logger.Warnf("Invalid URL: %s", arg)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
urls = append(urls, arg)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(urls) == 0 {
|
||||||
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgYtdlpErrorNoValidUrls)), nil)
|
||||||
|
return dispatcher.EndGroups
|
||||||
|
}
|
||||||
|
|
||||||
|
logger.Debugf("Preparing yt-dlp download for %d URL(s) with %d flag(s)", len(urls), len(flags))
|
||||||
|
|
||||||
|
// Build storage selection keyboard
|
||||||
|
markup, err := msgelem.BuildAddSelectStorageKeyboard(storage.GetUserStorages(ctx, update.GetUserChat().GetID()), tcbdata.Add{
|
||||||
|
TaskType: tasktype.TaskTypeYtdlp,
|
||||||
|
YtdlpURLs: urls,
|
||||||
|
YtdlpFlags: flags,
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgYtdlpInfoUrlsSelectStorage, map[string]any{
|
||||||
|
"Count": len(urls),
|
||||||
|
})), &ext.ReplyOpts{
|
||||||
|
Markup: markup,
|
||||||
|
})
|
||||||
|
|
||||||
|
return dispatcher.EndGroups
|
||||||
|
}
|
||||||
129
client/bot/handlers/ytdlp_test.go
Normal file
129
client/bot/handlers/ytdlp_test.go
Normal file
@@ -0,0 +1,129 @@
|
|||||||
|
package handlers
|
||||||
|
|
||||||
|
import (
|
||||||
|
"net/url"
|
||||||
|
"strings"
|
||||||
|
"testing"
|
||||||
|
)
|
||||||
|
|
||||||
|
// TestYtdlpArgumentParsing tests the URL and flag separation logic
|
||||||
|
func TestYtdlpArgumentParsing(t *testing.T) {
|
||||||
|
tests := []struct {
|
||||||
|
name string
|
||||||
|
input string
|
||||||
|
expectedURLs []string
|
||||||
|
expectedFlags []string
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
name: "Single URL without flags",
|
||||||
|
input: "/ytdlp https://example.com/video",
|
||||||
|
expectedURLs: []string{"https://example.com/video"},
|
||||||
|
expectedFlags: []string{},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "Multiple URLs without flags",
|
||||||
|
input: "/ytdlp https://example.com/v1 https://example.com/v2",
|
||||||
|
expectedURLs: []string{"https://example.com/v1", "https://example.com/v2"},
|
||||||
|
expectedFlags: []string{},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "URL with format flag",
|
||||||
|
input: "/ytdlp --format best https://example.com/video",
|
||||||
|
expectedURLs: []string{"https://example.com/video"},
|
||||||
|
expectedFlags: []string{"--format", "best"},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "URL with extract-audio flag",
|
||||||
|
input: "/ytdlp --extract-audio --audio-format mp3 https://example.com/video",
|
||||||
|
expectedURLs: []string{"https://example.com/video"},
|
||||||
|
expectedFlags: []string{"--extract-audio", "--audio-format", "mp3"},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "Multiple URLs with flags",
|
||||||
|
input: "/ytdlp --format best https://example.com/v1 https://example.com/v2",
|
||||||
|
expectedURLs: []string{"https://example.com/v1", "https://example.com/v2"},
|
||||||
|
expectedFlags: []string{"--format", "best"},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "Flags mixed with URLs",
|
||||||
|
input: "/ytdlp https://example.com/v1 --format best https://example.com/v2",
|
||||||
|
expectedURLs: []string{"https://example.com/v1", "https://example.com/v2"},
|
||||||
|
expectedFlags: []string{"--format", "best"},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "Short flag",
|
||||||
|
input: "/ytdlp -f best https://example.com/video",
|
||||||
|
expectedURLs: []string{"https://example.com/video"},
|
||||||
|
expectedFlags: []string{"-f", "best"},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "Boolean flag",
|
||||||
|
input: "/ytdlp --extract-audio https://example.com/video",
|
||||||
|
expectedURLs: []string{"https://example.com/video"},
|
||||||
|
expectedFlags: []string{"--extract-audio"},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, tt := range tests {
|
||||||
|
t.Run(tt.name, func(t *testing.T) {
|
||||||
|
args := strings.Split(tt.input, " ")
|
||||||
|
|
||||||
|
// Simulate the parsing logic from handleYtdlpCmd
|
||||||
|
var urls []string
|
||||||
|
var flags []string
|
||||||
|
|
||||||
|
for i := 1; i < len(args); i++ {
|
||||||
|
arg := strings.TrimSpace(args[i])
|
||||||
|
if arg == "" {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
// Check if it's a flag (starts with - or --)
|
||||||
|
if strings.HasPrefix(arg, "-") {
|
||||||
|
flags = append(flags, arg)
|
||||||
|
// Check if the next argument might be a value for this flag
|
||||||
|
if i+1 < len(args) {
|
||||||
|
nextArg := strings.TrimSpace(args[i+1])
|
||||||
|
if nextArg != "" && !strings.HasPrefix(nextArg, "-") {
|
||||||
|
// Check if it's clearly a URL (has ://)
|
||||||
|
if strings.Contains(nextArg, "://") {
|
||||||
|
// It's a URL, don't consume it as a flag value
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
// Otherwise, treat it as a flag value
|
||||||
|
flags = append(flags, nextArg)
|
||||||
|
i++ // Skip the next argument as it's been consumed
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
// Try to parse as URL
|
||||||
|
u, err := url.Parse(arg)
|
||||||
|
if err != nil || u.Scheme == "" || u.Host == "" {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
urls = append(urls, arg)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Verify URLs
|
||||||
|
if len(urls) != len(tt.expectedURLs) {
|
||||||
|
t.Errorf("Expected %d URLs, got %d", len(tt.expectedURLs), len(urls))
|
||||||
|
}
|
||||||
|
for i, expectedURL := range tt.expectedURLs {
|
||||||
|
if i >= len(urls) || urls[i] != expectedURL {
|
||||||
|
t.Errorf("Expected URL[%d] to be '%s', got '%s'", i, expectedURL, urls[i])
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Verify flags
|
||||||
|
if len(flags) != len(tt.expectedFlags) {
|
||||||
|
t.Errorf("Expected %d flags, got %d", len(tt.expectedFlags), len(flags))
|
||||||
|
}
|
||||||
|
for i, expectedFlag := range tt.expectedFlags {
|
||||||
|
if i >= len(flags) || flags[i] != expectedFlag {
|
||||||
|
t.Errorf("Expected flag[%d] to be '%s', got '%s'", i, expectedFlag, flags[i])
|
||||||
|
}
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -16,7 +16,7 @@ import (
|
|||||||
func NewDefaultMiddlewares(ctx context.Context, timeout time.Duration) []telegram.Middleware {
|
func NewDefaultMiddlewares(ctx context.Context, timeout time.Duration) []telegram.Middleware {
|
||||||
return []telegram.Middleware{
|
return []telegram.Middleware{
|
||||||
recovery.New(ctx, newBackoff(timeout)),
|
recovery.New(ctx, newBackoff(timeout)),
|
||||||
retry.New(config.Cfg.Telegram.RpcRetry),
|
retry.New(config.C().Telegram.RpcRetry),
|
||||||
floodwait.NewSimpleWaiter(),
|
floodwait.NewSimpleWaiter(),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,80 +1,57 @@
|
|||||||
package user
|
package user
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"bufio"
|
||||||
|
"fmt"
|
||||||
|
"os"
|
||||||
"strings"
|
"strings"
|
||||||
|
|
||||||
"github.com/celestix/gotgproto"
|
"github.com/celestix/gotgproto"
|
||||||
"github.com/charmbracelet/huh"
|
"golang.org/x/term"
|
||||||
"github.com/charmbracelet/log"
|
|
||||||
"github.com/fatih/color"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
type terminalAuthConversator struct{}
|
type terminalAuthConversator struct{}
|
||||||
|
|
||||||
func (t *terminalAuthConversator) AskPhoneNumber() (string, error) {
|
func readLine(prompt string) (string, error) {
|
||||||
phone := ""
|
fmt.Print(prompt)
|
||||||
err := huh.NewInput().Title("Your Phone Number").
|
reader := bufio.NewReader(os.Stdin)
|
||||||
Placeholder("+44 123456").
|
text, err := reader.ReadString('\n')
|
||||||
Prompt("> ").
|
|
||||||
Value(&phone).
|
|
||||||
WithTheme(huh.ThemeCatppuccin()).
|
|
||||||
Run()
|
|
||||||
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return "", err
|
return "", err
|
||||||
}
|
}
|
||||||
|
return strings.TrimSpace(text), nil
|
||||||
|
}
|
||||||
|
|
||||||
log.Info("Sending code to your phone number...")
|
func (t *terminalAuthConversator) AskPhoneNumber() (string, error) {
|
||||||
|
fmt.Println("Your Phone Number (e.g. +44 123456):")
|
||||||
return strings.TrimSpace(phone), nil
|
return readLine("> ")
|
||||||
}
|
}
|
||||||
|
|
||||||
func (t *terminalAuthConversator) AskCode() (string, error) {
|
func (t *terminalAuthConversator) AskCode() (string, error) {
|
||||||
code := ""
|
fmt.Println("Your Code (e.g. 123456):")
|
||||||
err := huh.NewInput().Title("Your Code").
|
return readLine("> ")
|
||||||
Placeholder("123456").
|
|
||||||
Value(&code).
|
|
||||||
Prompt("> ").
|
|
||||||
WithTheme(huh.ThemeCatppuccin()).
|
|
||||||
Run()
|
|
||||||
|
|
||||||
if err != nil {
|
|
||||||
return "", err
|
|
||||||
}
|
|
||||||
|
|
||||||
return strings.TrimSpace(code), nil
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (t *terminalAuthConversator) AskPassword() (string, error) {
|
func (t *terminalAuthConversator) AskPassword() (string, error) {
|
||||||
pwd := ""
|
fmt.Println("Your 2FA Password:")
|
||||||
|
fmt.Print("> ")
|
||||||
err := huh.NewInput().Title("Your 2FA Password").
|
bytePwd, err := term.ReadPassword(int(os.Stdin.Fd()))
|
||||||
EchoMode(huh.EchoModePassword).
|
fmt.Println()
|
||||||
Value(&pwd).
|
|
||||||
Prompt("> ").
|
|
||||||
WithTheme(huh.ThemeCatppuccin()).
|
|
||||||
Run()
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return "", err
|
return "", err
|
||||||
}
|
}
|
||||||
|
|
||||||
return strings.TrimSpace(pwd), nil
|
return strings.TrimSpace(string(bytePwd)), nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (t *terminalAuthConversator) AuthStatus(authStatus gotgproto.AuthStatus) {
|
func (t *terminalAuthConversator) AuthStatus(authStatus gotgproto.AuthStatus) {
|
||||||
switch authStatus.Event {
|
switch authStatus.Event {
|
||||||
case gotgproto.AuthStatusPhoneRetrial:
|
case gotgproto.AuthStatusPhoneRetrial:
|
||||||
color.Red("The phone number you just entered seems to be incorrect,")
|
fmt.Printf("The phone number is incorrect. Attempts left: %d\n", authStatus.AttemptsLeft)
|
||||||
color.Red("Attempts Left: %d", authStatus.AttemptsLeft)
|
|
||||||
color.Red("Please try again....")
|
|
||||||
case gotgproto.AuthStatusPasswordRetrial:
|
case gotgproto.AuthStatusPasswordRetrial:
|
||||||
color.Red("The 2FA password you just entered seems to be incorrect,")
|
fmt.Printf("The 2FA password is incorrect. Attempts left: %d\n", authStatus.AttemptsLeft)
|
||||||
color.Red("Attempts Left: %d", authStatus.AttemptsLeft)
|
|
||||||
color.Red("Please try again....")
|
|
||||||
case gotgproto.AuthStatusPhoneCodeRetrial:
|
case gotgproto.AuthStatusPhoneCodeRetrial:
|
||||||
color.Red("The OTP you just entered seems to be incorrect,")
|
fmt.Printf("The OTP code is incorrect. Attempts left: %d\n", authStatus.AttemptsLeft)
|
||||||
color.Red("Attempts Left: %d", authStatus.AttemptsLeft)
|
|
||||||
color.Red("Please try again....")
|
|
||||||
default:
|
default:
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -12,37 +12,27 @@ import (
|
|||||||
"github.com/celestix/gotgproto/sessionMaker"
|
"github.com/celestix/gotgproto/sessionMaker"
|
||||||
|
|
||||||
"github.com/charmbracelet/log"
|
"github.com/charmbracelet/log"
|
||||||
"github.com/gotd/td/telegram/dcs"
|
|
||||||
"github.com/gotd/td/tg"
|
"github.com/gotd/td/tg"
|
||||||
"github.com/krau/SaveAny-Bot/client/middleware"
|
"github.com/krau/SaveAny-Bot/client/middleware"
|
||||||
"github.com/krau/SaveAny-Bot/common/utils/netutil"
|
"github.com/krau/SaveAny-Bot/common/utils/tgutil"
|
||||||
"github.com/krau/SaveAny-Bot/config"
|
"github.com/krau/SaveAny-Bot/config"
|
||||||
"github.com/krau/SaveAny-Bot/database"
|
"github.com/krau/SaveAny-Bot/database"
|
||||||
"github.com/ncruces/go-sqlite3/gormlite"
|
|
||||||
"golang.org/x/net/proxy"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
var uc *gotgproto.Client
|
var uc *gotgproto.Client
|
||||||
var ectx *ext.Context
|
var ectx *ext.Context
|
||||||
|
|
||||||
func GetCtx() *ext.Context {
|
func GetCtx() *ext.Context {
|
||||||
if uc == nil {
|
|
||||||
panic("User client is not initialized, please call Login first")
|
|
||||||
}
|
|
||||||
if ectx != nil {
|
if ectx != nil {
|
||||||
return ectx
|
return ectx
|
||||||
}
|
}
|
||||||
|
if uc == nil {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
ectx = uc.CreateContext()
|
ectx = uc.CreateContext()
|
||||||
return ectx
|
return ectx
|
||||||
}
|
}
|
||||||
|
|
||||||
func GetClient() *gotgproto.Client {
|
|
||||||
if uc == nil {
|
|
||||||
panic("User client is not initialized, please call Login first")
|
|
||||||
}
|
|
||||||
return uc
|
|
||||||
}
|
|
||||||
|
|
||||||
func Login(ctx context.Context) (*gotgproto.Client, error) {
|
func Login(ctx context.Context) (*gotgproto.Client, error) {
|
||||||
log.FromContext(ctx).Debug("Logging in user client")
|
log.FromContext(ctx).Debug("Logging in user client")
|
||||||
if uc != nil {
|
if uc != nil {
|
||||||
@@ -53,33 +43,25 @@ func Login(ctx context.Context) (*gotgproto.Client, error) {
|
|||||||
err error
|
err error
|
||||||
})
|
})
|
||||||
go func() {
|
go func() {
|
||||||
var resolver dcs.Resolver
|
resolver, err := tgutil.NewConfigProxyResolver()
|
||||||
if config.Cfg.Telegram.Proxy.Enable && config.Cfg.Telegram.Proxy.URL != "" {
|
if err != nil {
|
||||||
dialer, err := netutil.NewProxyDialer(config.Cfg.Telegram.Proxy.URL)
|
res <- struct {
|
||||||
if err != nil {
|
client *gotgproto.Client
|
||||||
res <- struct {
|
err error
|
||||||
client *gotgproto.Client
|
}{nil, err}
|
||||||
err error
|
return
|
||||||
}{nil, err}
|
|
||||||
return
|
|
||||||
}
|
|
||||||
resolver = dcs.Plain(dcs.PlainOptions{
|
|
||||||
Dial: dialer.(proxy.ContextDialer).DialContext,
|
|
||||||
})
|
|
||||||
} else {
|
|
||||||
resolver = dcs.DefaultResolver()
|
|
||||||
}
|
}
|
||||||
tclient, err := gotgproto.NewClient(
|
tclient, err := gotgproto.NewClient(
|
||||||
config.Cfg.Telegram.AppID,
|
config.C().Telegram.AppID,
|
||||||
config.Cfg.Telegram.AppHash,
|
config.C().Telegram.AppHash,
|
||||||
gotgproto.ClientTypePhone(""),
|
gotgproto.ClientTypePhone(""),
|
||||||
&gotgproto.ClientOpts{
|
&gotgproto.ClientOpts{
|
||||||
Session: sessionMaker.SqlSession(gormlite.Open(config.Cfg.Telegram.Userbot.Session)),
|
Session: sessionMaker.SqlSession(database.GetDialect(config.C().Telegram.Userbot.Session)),
|
||||||
AuthConversator: &terminalAuthConversator{},
|
AuthConversator: &terminalAuthConversator{},
|
||||||
Context: ctx,
|
Context: ctx,
|
||||||
DisableCopyright: true,
|
DisableCopyright: true,
|
||||||
Resolver: resolver,
|
Resolver: resolver,
|
||||||
MaxRetries: config.Cfg.Telegram.RpcRetry,
|
MaxRetries: config.C().Telegram.RpcRetry,
|
||||||
AutoFetchReply: true,
|
AutoFetchReply: true,
|
||||||
Middlewares: middleware.NewDefaultMiddlewares(ctx, 5*time.Minute),
|
Middlewares: middleware.NewDefaultMiddlewares(ctx, 5*time.Minute),
|
||||||
ErrorHandler: func(ctx *ext.Context, u *ext.Update, s string) error {
|
ErrorHandler: func(ctx *ext.Context, u *ext.Update, s string) error {
|
||||||
|
|||||||
@@ -3,14 +3,17 @@ package main
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"bufio"
|
"bufio"
|
||||||
|
"bytes"
|
||||||
"flag"
|
"flag"
|
||||||
"fmt"
|
"fmt"
|
||||||
|
"go/format"
|
||||||
"io/fs"
|
"io/fs"
|
||||||
"os"
|
"os"
|
||||||
"path/filepath"
|
"path/filepath"
|
||||||
"regexp"
|
|
||||||
"sort"
|
"sort"
|
||||||
"strings"
|
"strings"
|
||||||
|
|
||||||
|
"github.com/goccy/go-yaml"
|
||||||
)
|
)
|
||||||
|
|
||||||
func main() {
|
func main() {
|
||||||
@@ -20,28 +23,27 @@ func main() {
|
|||||||
flag.Parse()
|
flag.Parse()
|
||||||
|
|
||||||
keys := make(map[string]struct{})
|
keys := make(map[string]struct{})
|
||||||
re := regexp.MustCompile(`^\s*\[+\s*([^\]\[]+)\s*\]+`)
|
|
||||||
|
|
||||||
err := filepath.WalkDir(*dir, func(path string, d fs.DirEntry, err error) error {
|
err := filepath.WalkDir(*dir, func(path string, d fs.DirEntry, err error) error {
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
if d.IsDir() || !strings.HasSuffix(d.Name(), ".toml") {
|
if d.IsDir() || !(strings.HasSuffix(d.Name(), ".yaml") || strings.HasSuffix(d.Name(), ".yml")) {
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
f, err := os.Open(path)
|
|
||||||
|
data, err := os.ReadFile(path)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
defer f.Close()
|
|
||||||
|
|
||||||
s := bufio.NewScanner(f)
|
var content map[string]any
|
||||||
for s.Scan() {
|
if err := yaml.Unmarshal(data, &content); err != nil {
|
||||||
if m := re.FindStringSubmatch(s.Text()); m != nil {
|
return fmt.Errorf("failed to parse yaml %s: %w", path, err)
|
||||||
keys[m[1]] = struct{}{}
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
return s.Err()
|
|
||||||
|
collectKeys(content, "", keys)
|
||||||
|
return nil
|
||||||
})
|
})
|
||||||
if err != nil {
|
if err != nil {
|
||||||
fmt.Fprintf(os.Stderr, "Error walking directory: %v\n", err)
|
fmt.Fprintf(os.Stderr, "Error walking directory: %v\n", err)
|
||||||
@@ -54,31 +56,60 @@ func main() {
|
|||||||
}
|
}
|
||||||
sort.Strings(list)
|
sort.Strings(list)
|
||||||
|
|
||||||
f, err := os.Create(*out)
|
// Generate code to buffer
|
||||||
if err != nil {
|
var buf bytes.Buffer
|
||||||
fmt.Fprintf(os.Stderr, "Error creating output file: %v\n", err)
|
w := bufio.NewWriter(&buf)
|
||||||
os.Exit(1)
|
fmt.Fprintf(w, "// Code generated by cmd/geni18n. DO NOT EDIT.\n")
|
||||||
}
|
|
||||||
defer f.Close()
|
|
||||||
|
|
||||||
w := bufio.NewWriter(f)
|
|
||||||
fmt.Fprintf(w, "// Code generated by cmd/gen_i18n. DO NOT EDIT.\n")
|
|
||||||
fmt.Fprintf(w, "package %s\n\n", *pkg)
|
fmt.Fprintf(w, "package %s\n\n", *pkg)
|
||||||
|
fmt.Fprintf(w, "type Key string\n\n")
|
||||||
fmt.Fprintf(w, "const (\n")
|
fmt.Fprintf(w, "const (\n")
|
||||||
for _, key := range list {
|
for _, key := range list {
|
||||||
name := toPascal(key)
|
name := toPascal(key)
|
||||||
fmt.Fprintf(w, "\t%s = %q\n", name, key)
|
fmt.Fprintf(w, "\t%s Key = %q\n", name, key)
|
||||||
}
|
}
|
||||||
fmt.Fprintf(w, ")\n")
|
fmt.Fprintf(w, ")\n")
|
||||||
w.Flush()
|
w.Flush()
|
||||||
|
|
||||||
|
// Format the generated code
|
||||||
|
formatted, err := format.Source(buf.Bytes())
|
||||||
|
if err != nil {
|
||||||
|
fmt.Fprintf(os.Stderr, "Warning: failed to format generated code: %v\n", err)
|
||||||
|
formatted = buf.Bytes()
|
||||||
|
}
|
||||||
|
|
||||||
|
// Write to output file
|
||||||
|
if err := os.WriteFile(*out, formatted, 0644); err != nil {
|
||||||
|
fmt.Fprintf(os.Stderr, "Error writing output file: %v\n", err)
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func collectKeys(node map[string]any, prefix string, keys map[string]struct{}) {
|
||||||
|
for k, v := range node {
|
||||||
|
fullKey := k
|
||||||
|
if prefix != "" {
|
||||||
|
fullKey = prefix + "." + k
|
||||||
|
}
|
||||||
|
switch val := v.(type) {
|
||||||
|
case map[string]any:
|
||||||
|
collectKeys(val, fullKey, keys)
|
||||||
|
default:
|
||||||
|
keys[fullKey] = struct{}{}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 转 PascalCase
|
||||||
func toPascal(key string) string {
|
func toPascal(key string) string {
|
||||||
parts := strings.Split(key, ".")
|
parts := strings.Split(key, ".")
|
||||||
for i, p := range parts {
|
for i, p := range parts {
|
||||||
if len(p) > 0 {
|
subs := strings.Split(p, "_")
|
||||||
parts[i] = strings.ToUpper(string(p[0])) + p[1:]
|
for j, s := range subs {
|
||||||
|
if len(s) > 0 {
|
||||||
|
subs[j] = strings.ToUpper(s[:1]) + s[1:]
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
parts[i] = strings.Join(subs, "")
|
||||||
}
|
}
|
||||||
return strings.Join(parts, "")
|
return strings.Join(parts, "")
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -4,6 +4,8 @@ import (
|
|||||||
"context"
|
"context"
|
||||||
"fmt"
|
"fmt"
|
||||||
|
|
||||||
|
"github.com/krau/SaveAny-Bot/cmd/upload"
|
||||||
|
"github.com/krau/SaveAny-Bot/config"
|
||||||
"github.com/spf13/cobra"
|
"github.com/spf13/cobra"
|
||||||
)
|
)
|
||||||
|
|
||||||
@@ -13,6 +15,11 @@ var rootCmd = &cobra.Command{
|
|||||||
Run: Run,
|
Run: Run,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
config.RegisterFlags(rootCmd)
|
||||||
|
upload.Register(rootCmd)
|
||||||
|
}
|
||||||
|
|
||||||
func Execute(ctx context.Context) {
|
func Execute(ctx context.Context) {
|
||||||
if err := rootCmd.ExecuteContext(ctx); err != nil {
|
if err := rootCmd.ExecuteContext(ctx); err != nil {
|
||||||
fmt.Println(err)
|
fmt.Println(err)
|
||||||
|
|||||||
69
cmd/run.go
69
cmd/run.go
@@ -14,7 +14,6 @@ import (
|
|||||||
userclient "github.com/krau/SaveAny-Bot/client/user"
|
userclient "github.com/krau/SaveAny-Bot/client/user"
|
||||||
"github.com/krau/SaveAny-Bot/common/cache"
|
"github.com/krau/SaveAny-Bot/common/cache"
|
||||||
"github.com/krau/SaveAny-Bot/common/i18n"
|
"github.com/krau/SaveAny-Bot/common/i18n"
|
||||||
"github.com/krau/SaveAny-Bot/common/i18n/i18nk"
|
|
||||||
"github.com/krau/SaveAny-Bot/common/utils/fsutil"
|
"github.com/krau/SaveAny-Bot/common/utils/fsutil"
|
||||||
"github.com/krau/SaveAny-Bot/config"
|
"github.com/krau/SaveAny-Bot/config"
|
||||||
"github.com/krau/SaveAny-Bot/core"
|
"github.com/krau/SaveAny-Bot/core"
|
||||||
@@ -25,7 +24,7 @@ import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
func Run(cmd *cobra.Command, _ []string) {
|
func Run(cmd *cobra.Command, _ []string) {
|
||||||
ctx := cmd.Context()
|
ctx, cancel := context.WithCancel(cmd.Context())
|
||||||
logger := log.NewWithOptions(os.Stdout, log.Options{
|
logger := log.NewWithOptions(os.Stdout, log.Options{
|
||||||
Level: log.DebugLevel,
|
Level: log.DebugLevel,
|
||||||
ReportTimestamp: true,
|
ReportTimestamp: true,
|
||||||
@@ -34,77 +33,75 @@ func Run(cmd *cobra.Command, _ []string) {
|
|||||||
})
|
})
|
||||||
ctx = log.WithContext(ctx, logger)
|
ctx = log.WithContext(ctx, logger)
|
||||||
|
|
||||||
initAll(ctx)
|
exitChan, err := initAll(ctx, cmd)
|
||||||
|
if err != nil {
|
||||||
|
logger.Fatal("Init failed", "error", err)
|
||||||
|
}
|
||||||
|
go func() {
|
||||||
|
<-exitChan
|
||||||
|
cancel()
|
||||||
|
}()
|
||||||
|
|
||||||
core.Run(ctx)
|
core.Run(ctx)
|
||||||
|
|
||||||
<-ctx.Done()
|
<-ctx.Done()
|
||||||
logger.Info(i18n.T(i18nk.Exiting))
|
logger.Info("Exiting...")
|
||||||
defer logger.Info(i18n.T(i18nk.Bye))
|
defer logger.Info("Exit complete")
|
||||||
cleanCache()
|
cleanCache()
|
||||||
}
|
}
|
||||||
|
|
||||||
func initAll(ctx context.Context) {
|
func initAll(ctx context.Context, cmd *cobra.Command) (<-chan struct{}, error) {
|
||||||
if err := config.Init(ctx); err != nil {
|
configFile := config.GetConfigFile(cmd)
|
||||||
fmt.Println("Failed to load config:", err)
|
if err := config.Init(ctx, configFile); err != nil {
|
||||||
os.Exit(1)
|
return nil, fmt.Errorf("failed to load config: %w", err)
|
||||||
}
|
}
|
||||||
cache.Init()
|
cache.Init()
|
||||||
logger := log.FromContext(ctx)
|
logger := log.FromContext(ctx)
|
||||||
i18n.Init(config.Cfg.Lang)
|
i18n.Init(config.C().Lang)
|
||||||
logger.Info(i18n.T(i18nk.Initing))
|
logger.Info("Initializing...")
|
||||||
database.Init(ctx)
|
database.Init(ctx)
|
||||||
storage.LoadStorages(ctx)
|
storage.LoadStorages(ctx)
|
||||||
if config.Cfg.Parser.PluginEnable {
|
if config.C().Parser.PluginEnable {
|
||||||
for _, dir := range config.Cfg.Parser.PluginDirs {
|
for _, dir := range config.C().Parser.PluginDirs {
|
||||||
if err := parsers.LoadPlugins(ctx, dir); err != nil {
|
if err := parsers.LoadPlugins(ctx, dir); err != nil {
|
||||||
logger.Error("Failed to load parser plugins", "dir", dir, "error", err)
|
logger.Error("Failed to load parser plugins", "dir", dir, "error", err)
|
||||||
} else {
|
} else {
|
||||||
logger.Debug("Loaded parser plugins", "dir", dir)
|
logger.Debug("Loaded parser plugins from directory", "dir", dir)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
if config.Cfg.Telegram.Userbot.Enable {
|
if config.C().Telegram.Userbot.Enable {
|
||||||
_, err := userclient.Login(ctx)
|
_, err := userclient.Login(ctx)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Fatalf("User client login failed: %s", err)
|
logger.Fatal("User login failed", "error", err)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
bot.Init(ctx)
|
return bot.Init(ctx), nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func cleanCache() {
|
func cleanCache() {
|
||||||
if config.Cfg.NoCleanCache {
|
if config.C().NoCleanCache {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
if config.Cfg.Temp.BasePath != "" && !config.Cfg.Stream {
|
if config.C().Temp.BasePath != "" && !config.C().Stream {
|
||||||
if slices.Contains([]string{"/", ".", "\\", ".."}, filepath.Clean(config.Cfg.Temp.BasePath)) {
|
if slices.Contains([]string{"/", ".", "\\", ".."}, filepath.Clean(config.C().Temp.BasePath)) {
|
||||||
log.Error(i18n.T(i18nk.InvalidCacheDir, map[string]any{
|
log.Error("Invalid cache directory", "path", config.C().Temp.BasePath)
|
||||||
"Path": config.Cfg.Temp.BasePath,
|
|
||||||
}))
|
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
currentDir, err := os.Getwd()
|
currentDir, err := os.Getwd()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Error(i18n.T(i18nk.GetWorkdirFailed, map[string]any{
|
log.Error("Failed to get working directory", "error", err)
|
||||||
"Error": err,
|
|
||||||
}))
|
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
cachePath := filepath.Join(currentDir, config.Cfg.Temp.BasePath)
|
cachePath := filepath.Join(currentDir, config.C().Temp.BasePath)
|
||||||
cachePath, err = filepath.Abs(cachePath)
|
cachePath, err = filepath.Abs(cachePath)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Error(i18n.T(i18nk.GetCacheAbsPathFailed, map[string]any{
|
log.Error("Failed to get absolute cache path", "error", err)
|
||||||
"Error": err,
|
|
||||||
}))
|
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
log.Info(i18n.T(i18nk.CleaningCache, map[string]any{
|
log.Info("Cleaning cache directory", "path", cachePath)
|
||||||
"Path": cachePath,
|
|
||||||
}))
|
|
||||||
if err := fsutil.RemoveAllInDir(cachePath); err != nil {
|
if err := fsutil.RemoveAllInDir(cachePath); err != nil {
|
||||||
log.Error(i18n.T(i18nk.CleanCacheFailed, map[string]any{
|
log.Error("Failed to clean cache directory", "error", err)
|
||||||
"Error": err,
|
|
||||||
}))
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
130
cmd/upload/cmd.go
Normal file
130
cmd/upload/cmd.go
Normal file
@@ -0,0 +1,130 @@
|
|||||||
|
package upload
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"fmt"
|
||||||
|
"io"
|
||||||
|
"os"
|
||||||
|
"path"
|
||||||
|
"path/filepath"
|
||||||
|
|
||||||
|
"github.com/charmbracelet/log"
|
||||||
|
"github.com/krau/SaveAny-Bot/client/bot"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/cache"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/i18n"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/utils/ioutil"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/utils/tgutil"
|
||||||
|
"github.com/krau/SaveAny-Bot/config"
|
||||||
|
"github.com/krau/SaveAny-Bot/database"
|
||||||
|
"github.com/krau/SaveAny-Bot/pkg/enums/ctxkey"
|
||||||
|
stortype "github.com/krau/SaveAny-Bot/pkg/enums/storage"
|
||||||
|
"github.com/krau/SaveAny-Bot/storage"
|
||||||
|
"github.com/spf13/cobra"
|
||||||
|
)
|
||||||
|
|
||||||
|
var uploadCmd = &cobra.Command{
|
||||||
|
Use: "upload",
|
||||||
|
Short: "upload local files to storage",
|
||||||
|
RunE: Upload,
|
||||||
|
}
|
||||||
|
|
||||||
|
func Register(root *cobra.Command) {
|
||||||
|
uploadCmd.Flags().StringP("file", "f", "", "file path to upload")
|
||||||
|
uploadCmd.MarkFlagRequired("file")
|
||||||
|
uploadCmd.Flags().StringP("storage", "s", "", "storage name to upload to")
|
||||||
|
uploadCmd.MarkFlagRequired("storage")
|
||||||
|
uploadCmd.Flags().StringP("dir", "d", "", "storage dir to upload to, default is the base_path of the storage")
|
||||||
|
uploadCmd.Flags().Bool("no-progress", false, "disable progress bar")
|
||||||
|
root.AddCommand(uploadCmd)
|
||||||
|
}
|
||||||
|
|
||||||
|
func Upload(cmd *cobra.Command, args []string) error {
|
||||||
|
storname, err := cmd.Flags().GetString("storage")
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
fp, err := cmd.Flags().GetString("file")
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
dirPath, err := cmd.Flags().GetString("dir")
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
noProgress, err := cmd.Flags().GetBool("no-progress")
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
ctx := cmd.Context()
|
||||||
|
log := log.FromContext(ctx)
|
||||||
|
configFile := config.GetConfigFile(cmd)
|
||||||
|
if err := config.Init(ctx, configFile); err != nil {
|
||||||
|
return fmt.Errorf("failed to load config: %w", err)
|
||||||
|
}
|
||||||
|
i18n.Init(config.C().Lang)
|
||||||
|
cache.Init()
|
||||||
|
database.Init(ctx)
|
||||||
|
|
||||||
|
stor, err := storage.GetStorageByName(ctx, storname)
|
||||||
|
if err != nil {
|
||||||
|
log.Fatal("Failed to get storage", "error", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
switch stor.Type() {
|
||||||
|
case stortype.Telegram:
|
||||||
|
bot.Init(ctx)
|
||||||
|
default:
|
||||||
|
// placeholder for other storage types that may need special initialization
|
||||||
|
}
|
||||||
|
|
||||||
|
file, err := os.Open(filepath.Clean(fp))
|
||||||
|
if err != nil {
|
||||||
|
log.Fatal("Failed to open file", "error", err)
|
||||||
|
}
|
||||||
|
defer file.Close()
|
||||||
|
fileInfo, err := file.Stat()
|
||||||
|
if err != nil {
|
||||||
|
log.Fatal("Failed to get file info", "error", err)
|
||||||
|
}
|
||||||
|
fileName := fileInfo.Name()
|
||||||
|
fileSize := fileInfo.Size()
|
||||||
|
|
||||||
|
uploadPath := stor.JoinStoragePath(path.Join(dirPath, fileName))
|
||||||
|
|
||||||
|
ctx = context.WithValue(ctx, ctxkey.ContentLength, fileSize)
|
||||||
|
ctx = tgutil.ExtWithContext(ctx, bot.ExtContext())
|
||||||
|
|
||||||
|
// Create progress reader and UI
|
||||||
|
var reader io.Reader
|
||||||
|
var progressUI *UploadProgress
|
||||||
|
log.Info("Uploading file...", "file", fp, "to", storname, "as", uploadPath)
|
||||||
|
|
||||||
|
if !noProgress && fileSize > 0 {
|
||||||
|
progressUI = NewUploadProgress(ctx, fileName, fileSize)
|
||||||
|
progressUI.Start()
|
||||||
|
|
||||||
|
reader = ioutil.NewProgressReader(file, fileSize, func(read int64, total int64) {
|
||||||
|
if total > 0 {
|
||||||
|
progressUI.UpdateProgress(float64(read) / float64(total))
|
||||||
|
}
|
||||||
|
})
|
||||||
|
} else {
|
||||||
|
reader = file
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := stor.Save(ctx, reader, uploadPath); err != nil {
|
||||||
|
if progressUI != nil {
|
||||||
|
progressUI.SetError(err)
|
||||||
|
progressUI.Wait()
|
||||||
|
}
|
||||||
|
log.Fatal("Failed to upload file", "error", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if progressUI != nil {
|
||||||
|
progressUI.Done()
|
||||||
|
progressUI.Wait()
|
||||||
|
}
|
||||||
|
log.Info("File uploaded successfully")
|
||||||
|
return nil
|
||||||
|
}
|
||||||
35
cmd/upload/progress_stub.go
Normal file
35
cmd/upload/progress_stub.go
Normal file
@@ -0,0 +1,35 @@
|
|||||||
|
//go:build no_bubbletea
|
||||||
|
|
||||||
|
package upload
|
||||||
|
|
||||||
|
import "context"
|
||||||
|
|
||||||
|
type uploadModel struct {
|
||||||
|
}
|
||||||
|
|
||||||
|
// UploadProgress manages the progress UI for uploads
|
||||||
|
type UploadProgress struct {
|
||||||
|
}
|
||||||
|
|
||||||
|
// NewUploadProgress creates a new upload progress tracker
|
||||||
|
func NewUploadProgress(ctx context.Context, fileName string, fileSize int64) *UploadProgress {
|
||||||
|
return &UploadProgress{}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Start starts the progress UI in a goroutine and returns immediately
|
||||||
|
func (up *UploadProgress) Start() {}
|
||||||
|
|
||||||
|
// UpdateProgress updates the progress bar with a new percentage (0.0 - 1.0)
|
||||||
|
func (up *UploadProgress) UpdateProgress(percent float64) {}
|
||||||
|
|
||||||
|
// SetError sets an error and quits the progress UI
|
||||||
|
func (up *UploadProgress) SetError(err error) {}
|
||||||
|
|
||||||
|
// Done signals that the upload is complete
|
||||||
|
func (up *UploadProgress) Done() {}
|
||||||
|
|
||||||
|
// Wait waits for the progress UI to finish
|
||||||
|
func (up *UploadProgress) Wait() {}
|
||||||
|
|
||||||
|
// Quit quits the progress UI
|
||||||
|
func (up *UploadProgress) Quit() {}
|
||||||
178
cmd/upload/progress_tea.go
Normal file
178
cmd/upload/progress_tea.go
Normal file
@@ -0,0 +1,178 @@
|
|||||||
|
//go:build !no_bubbletea
|
||||||
|
|
||||||
|
package upload
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"fmt"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
"github.com/charmbracelet/bubbles/progress"
|
||||||
|
tea "github.com/charmbracelet/bubbletea"
|
||||||
|
"github.com/charmbracelet/lipgloss"
|
||||||
|
"github.com/dustin/go-humanize"
|
||||||
|
)
|
||||||
|
|
||||||
|
var (
|
||||||
|
helpStyle = lipgloss.NewStyle().Foreground(lipgloss.Color("#626262"))
|
||||||
|
)
|
||||||
|
|
||||||
|
// progressMsg is sent to update the progress bar
|
||||||
|
type progressMsg float64
|
||||||
|
|
||||||
|
// progressErrMsg is sent when an error occurs
|
||||||
|
type progressErrMsg struct{ err error }
|
||||||
|
|
||||||
|
// progressDoneMsg is sent when the upload is complete
|
||||||
|
type progressDoneMsg struct{}
|
||||||
|
|
||||||
|
// uploadModel is the bubbletea model for the upload progress UI
|
||||||
|
type uploadModel struct {
|
||||||
|
progress progress.Model
|
||||||
|
fileName string
|
||||||
|
fileSize int64
|
||||||
|
bytesRead int64
|
||||||
|
err error
|
||||||
|
done bool
|
||||||
|
quitting bool
|
||||||
|
width int
|
||||||
|
}
|
||||||
|
|
||||||
|
func newUploadModel(fileName string, fileSize int64) uploadModel {
|
||||||
|
p := progress.New(
|
||||||
|
progress.WithDefaultGradient(),
|
||||||
|
progress.WithWidth(50),
|
||||||
|
)
|
||||||
|
return uploadModel{
|
||||||
|
progress: p,
|
||||||
|
fileName: fileName,
|
||||||
|
fileSize: fileSize,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m uploadModel) Init() tea.Cmd {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m uploadModel) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
|
||||||
|
switch msg := msg.(type) {
|
||||||
|
case tea.WindowSizeMsg:
|
||||||
|
m.width = msg.Width
|
||||||
|
m.progress.Width = min(msg.Width-10, 80)
|
||||||
|
return m, nil
|
||||||
|
|
||||||
|
case progressMsg:
|
||||||
|
var cmds []tea.Cmd
|
||||||
|
percent := float64(msg)
|
||||||
|
m.bytesRead = int64(percent * float64(m.fileSize))
|
||||||
|
|
||||||
|
cmds = append(cmds, m.progress.SetPercent(percent))
|
||||||
|
return m, tea.Batch(cmds...)
|
||||||
|
|
||||||
|
case progressErrMsg:
|
||||||
|
m.err = msg.err
|
||||||
|
return m, tea.Quit
|
||||||
|
|
||||||
|
case progressDoneMsg:
|
||||||
|
m.done = true
|
||||||
|
m.progress.SetPercent(1.0)
|
||||||
|
return m, tea.Quit
|
||||||
|
|
||||||
|
case progress.FrameMsg:
|
||||||
|
// Don't process frame messages if we're done or quitting
|
||||||
|
if m.done || m.quitting {
|
||||||
|
return m, nil
|
||||||
|
}
|
||||||
|
progressModel, cmd := m.progress.Update(msg)
|
||||||
|
m.progress = progressModel.(progress.Model)
|
||||||
|
return m, cmd
|
||||||
|
}
|
||||||
|
|
||||||
|
return m, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m uploadModel) View() string {
|
||||||
|
if m.err != nil {
|
||||||
|
return fmt.Sprintf("\n ❌ Error: %s\n\n", m.err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
var sb strings.Builder
|
||||||
|
sb.WriteString("\n")
|
||||||
|
|
||||||
|
// File info
|
||||||
|
sb.WriteString(fmt.Sprintf(" 📁 %s\n", m.fileName))
|
||||||
|
sb.WriteString(fmt.Sprintf(" 📊 %s / %s\n\n",
|
||||||
|
humanize.Bytes(uint64(m.bytesRead)),
|
||||||
|
humanize.Bytes(uint64(m.fileSize)),
|
||||||
|
))
|
||||||
|
|
||||||
|
// Progress bar
|
||||||
|
sb.WriteString(" ")
|
||||||
|
sb.WriteString(m.progress.View())
|
||||||
|
sb.WriteString("\n\n")
|
||||||
|
|
||||||
|
if m.done {
|
||||||
|
sb.WriteString(" √ Upload complete!\n\n")
|
||||||
|
} else {
|
||||||
|
sb.WriteString(helpStyle.Render(" Press Ctrl+C to cancel"))
|
||||||
|
sb.WriteString("\n\n")
|
||||||
|
}
|
||||||
|
|
||||||
|
return sb.String()
|
||||||
|
}
|
||||||
|
|
||||||
|
// UploadProgress manages the progress UI for uploads
|
||||||
|
type UploadProgress struct {
|
||||||
|
program *tea.Program
|
||||||
|
ctx context.Context
|
||||||
|
cancel context.CancelFunc
|
||||||
|
}
|
||||||
|
|
||||||
|
// NewUploadProgress creates a new upload progress tracker
|
||||||
|
func NewUploadProgress(ctx context.Context, fileName string, fileSize int64) *UploadProgress {
|
||||||
|
model := newUploadModel(fileName, fileSize)
|
||||||
|
ctx, cancel := context.WithCancel(ctx)
|
||||||
|
p := tea.NewProgram(
|
||||||
|
model,
|
||||||
|
tea.WithoutSignalHandler(),
|
||||||
|
tea.WithContext(ctx),
|
||||||
|
tea.WithInput(nil), // Disable keyboard input, rely on context cancellation
|
||||||
|
)
|
||||||
|
return &UploadProgress{
|
||||||
|
program: p,
|
||||||
|
ctx: ctx,
|
||||||
|
cancel: cancel,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Start starts the progress UI in a goroutine and returns immediately
|
||||||
|
func (up *UploadProgress) Start() {
|
||||||
|
go func() {
|
||||||
|
up.program.Run()
|
||||||
|
}()
|
||||||
|
}
|
||||||
|
|
||||||
|
// UpdateProgress updates the progress bar with a new percentage (0.0 - 1.0)
|
||||||
|
func (up *UploadProgress) UpdateProgress(percent float64) {
|
||||||
|
up.program.Send(progressMsg(percent))
|
||||||
|
}
|
||||||
|
|
||||||
|
// SetError sets an error and quits the progress UI
|
||||||
|
func (up *UploadProgress) SetError(err error) {
|
||||||
|
up.program.Send(progressErrMsg{err: err})
|
||||||
|
}
|
||||||
|
|
||||||
|
// Done signals that the upload is complete
|
||||||
|
func (up *UploadProgress) Done() {
|
||||||
|
up.program.Send(progressDoneMsg{})
|
||||||
|
}
|
||||||
|
|
||||||
|
// Wait waits for the progress UI to finish
|
||||||
|
func (up *UploadProgress) Wait() {
|
||||||
|
up.program.Wait()
|
||||||
|
}
|
||||||
|
|
||||||
|
// Quit quits the progress UI
|
||||||
|
func (up *UploadProgress) Quit() {
|
||||||
|
up.program.Quit()
|
||||||
|
}
|
||||||
@@ -4,8 +4,8 @@ import (
|
|||||||
"fmt"
|
"fmt"
|
||||||
"runtime"
|
"runtime"
|
||||||
|
|
||||||
"github.com/krau/SaveAny-Bot/pkg/consts"
|
"github.com/krau/SaveAny-Bot/config"
|
||||||
"github.com/rhysd/go-github-selfupdate/selfupdate"
|
"github.com/unvgo/ghselfupdate"
|
||||||
|
|
||||||
"github.com/blang/semver"
|
"github.com/blang/semver"
|
||||||
"github.com/spf13/cobra"
|
"github.com/spf13/cobra"
|
||||||
@@ -16,7 +16,7 @@ var VersionCmd = &cobra.Command{
|
|||||||
Aliases: []string{"v"},
|
Aliases: []string{"v"},
|
||||||
Short: "Print the version number of saveany-bot",
|
Short: "Print the version number of saveany-bot",
|
||||||
Run: func(cmd *cobra.Command, args []string) {
|
Run: func(cmd *cobra.Command, args []string) {
|
||||||
fmt.Printf("saveany-bot version: %s %s/%s\nBuildTime: %s, Commit: %s\n", consts.Version, runtime.GOOS, runtime.GOARCH, consts.BuildTime, consts.GitCommit)
|
fmt.Printf("saveany-bot version: %s %s/%s\nBuildTime: %s, Commit: %s\n", config.Version, runtime.GOOS, runtime.GOARCH, config.BuildTime, config.GitCommit)
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -25,18 +25,33 @@ var upgradeCmd = &cobra.Command{
|
|||||||
Aliases: []string{"up"},
|
Aliases: []string{"up"},
|
||||||
Short: "Upgrade saveany-bot to the latest version",
|
Short: "Upgrade saveany-bot to the latest version",
|
||||||
Run: func(cmd *cobra.Command, args []string) {
|
Run: func(cmd *cobra.Command, args []string) {
|
||||||
v := semver.MustParse(consts.Version)
|
v := semver.MustParse(config.Version)
|
||||||
latest, err := selfupdate.UpdateSelf(v, "krau/SaveAny-Bot")
|
latest, found, err := ghselfupdate.DetectLatest(config.GitRepo)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
fmt.Println("Binary update failed:", err)
|
fmt.Println("Error occurred while detecting latest version:", err)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
if latest.Version.Equals(v) {
|
if !found {
|
||||||
fmt.Println("Current binary is the latest version", consts.Version)
|
fmt.Println("No releases found")
|
||||||
} else {
|
return
|
||||||
fmt.Println("Successfully updated to version", latest.Version)
|
|
||||||
fmt.Println("Release note:\n", latest.ReleaseNotes)
|
|
||||||
}
|
}
|
||||||
|
if latest.Version.Major != v.Major {
|
||||||
|
fmt.Printf("Major version upgrade detected: %s -> %s. Please manually download the latest version and check the migration guide.\n", v, latest.Version)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if latest.Version.Equals(v) || latest.Version.LT(v) {
|
||||||
|
fmt.Println("Current binary is the latest version", config.Version)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
fmt.Printf("Updating to version %s...\n", latest.Version)
|
||||||
|
latest, err = ghselfupdate.UpdateSelf(v, config.GitRepo)
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println("Update failed:", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
fmt.Println("Successfully updated to version", latest.Version)
|
||||||
|
fmt.Println("Release note:\n", latest.ReleaseNotes)
|
||||||
|
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
6
common/cache/ristretto.go
vendored
6
common/cache/ristretto.go
vendored
@@ -16,8 +16,8 @@ func Init() {
|
|||||||
panic("cache already initialized")
|
panic("cache already initialized")
|
||||||
}
|
}
|
||||||
c, err := ristretto.NewCache(&ristretto.Config[string, any]{
|
c, err := ristretto.NewCache(&ristretto.Config[string, any]{
|
||||||
NumCounters: config.Cfg.Cache.NumCounters,
|
NumCounters: config.C().Cache.NumCounters,
|
||||||
MaxCost: config.Cfg.Cache.MaxCost,
|
MaxCost: config.C().Cache.MaxCost,
|
||||||
BufferItems: 64,
|
BufferItems: 64,
|
||||||
OnReject: func(item *ristretto.Item[any]) {
|
OnReject: func(item *ristretto.Item[any]) {
|
||||||
log.Warnf("Cache item rejected: key=%d, value=%v", item.Key, item.Value)
|
log.Warnf("Cache item rejected: key=%d, value=%v", item.Key, item.Value)
|
||||||
@@ -30,7 +30,7 @@ func Init() {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func Set(key string, value any) error {
|
func Set(key string, value any) error {
|
||||||
ok := cache.SetWithTTL(key, value, 0, time.Duration(config.Cfg.Cache.TTL)*time.Second)
|
ok := cache.SetWithTTL(key, value, 0, time.Duration(config.C().Cache.TTL)*time.Second)
|
||||||
if !ok {
|
if !ok {
|
||||||
return fmt.Errorf("failed to set value in cache")
|
return fmt.Errorf("failed to set value in cache")
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,3 +1,5 @@
|
|||||||
|
// [TODO] complete the i18n support
|
||||||
|
|
||||||
package i18n
|
package i18n
|
||||||
|
|
||||||
import (
|
import (
|
||||||
@@ -5,12 +7,13 @@ import (
|
|||||||
|
|
||||||
"maps"
|
"maps"
|
||||||
|
|
||||||
|
"github.com/goccy/go-yaml"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/i18n/i18nk"
|
||||||
"github.com/nicksnyder/go-i18n/v2/i18n"
|
"github.com/nicksnyder/go-i18n/v2/i18n"
|
||||||
"github.com/pelletier/go-toml/v2"
|
|
||||||
"golang.org/x/text/language"
|
"golang.org/x/text/language"
|
||||||
)
|
)
|
||||||
|
|
||||||
//go:embed locale/*.toml
|
//go:embed locale/*
|
||||||
var localesFS embed.FS
|
var localesFS embed.FS
|
||||||
|
|
||||||
var (
|
var (
|
||||||
@@ -20,7 +23,7 @@ var (
|
|||||||
|
|
||||||
func Init(lang string) {
|
func Init(lang string) {
|
||||||
bundle = i18n.NewBundle(language.SimplifiedChinese)
|
bundle = i18n.NewBundle(language.SimplifiedChinese)
|
||||||
bundle.RegisterUnmarshalFunc("toml", toml.Unmarshal)
|
bundle.RegisterUnmarshalFunc("yaml", yaml.Unmarshal)
|
||||||
files, err := localesFS.ReadDir("locale")
|
files, err := localesFS.ReadDir("locale")
|
||||||
if err != nil {
|
if err != nil {
|
||||||
panic("failed to read locale directory: " + err.Error())
|
panic("failed to read locale directory: " + err.Error())
|
||||||
@@ -39,70 +42,20 @@ func Init(lang string) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func T(key string, templateData ...map[string]any) string {
|
func T(key i18nk.Key, templateData ...map[string]any) string {
|
||||||
if localizer == nil || bundle == nil {
|
if localizer == nil || bundle == nil {
|
||||||
panic("localizer or bundle is not initialized, call Init() first")
|
Init("zh-Hans")
|
||||||
}
|
}
|
||||||
templateDataMap := make(map[string]any)
|
templateDataMap := make(map[string]any)
|
||||||
for _, data := range templateData {
|
for _, data := range templateData {
|
||||||
maps.Copy(templateDataMap, data)
|
maps.Copy(templateDataMap, data)
|
||||||
}
|
}
|
||||||
msg, err := localizer.Localize(&i18n.LocalizeConfig{
|
msg, err := localizer.Localize(&i18n.LocalizeConfig{
|
||||||
MessageID: key,
|
MessageID: string(key),
|
||||||
TemplateData: templateDataMap,
|
TemplateData: templateDataMap,
|
||||||
})
|
})
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return key
|
return string(key)
|
||||||
}
|
|
||||||
return msg
|
|
||||||
}
|
|
||||||
|
|
||||||
func TWithLang(lang, key string, templateData ...map[string]any) string {
|
|
||||||
if bundle == nil {
|
|
||||||
panic("bundle is not initialized, call Init() first")
|
|
||||||
}
|
|
||||||
templateDataMap := make(map[string]any)
|
|
||||||
for _, data := range templateData {
|
|
||||||
maps.Copy(templateDataMap, data)
|
|
||||||
}
|
|
||||||
localizerWithLang := i18n.NewLocalizer(bundle, lang)
|
|
||||||
msg, err := localizerWithLang.Localize(&i18n.LocalizeConfig{
|
|
||||||
MessageID: key,
|
|
||||||
TemplateData: templateDataMap,
|
|
||||||
})
|
|
||||||
if err != nil {
|
|
||||||
return key
|
|
||||||
}
|
|
||||||
return msg
|
|
||||||
}
|
|
||||||
|
|
||||||
// Only use in tests or packages that load before i18n
|
|
||||||
func TWithoutInit(lang, key string, templateData ...map[string]any) string {
|
|
||||||
bundle := i18n.NewBundle(language.SimplifiedChinese)
|
|
||||||
bundle.RegisterUnmarshalFunc("toml", toml.Unmarshal)
|
|
||||||
files, err := localesFS.ReadDir("locale")
|
|
||||||
if err != nil {
|
|
||||||
return key
|
|
||||||
}
|
|
||||||
for _, file := range files {
|
|
||||||
if _, err := bundle.LoadMessageFileFS(localesFS, "locale/"+file.Name()); err != nil {
|
|
||||||
return key
|
|
||||||
}
|
|
||||||
}
|
|
||||||
localizer := i18n.NewLocalizer(bundle, lang)
|
|
||||||
if localizer == nil {
|
|
||||||
return key
|
|
||||||
}
|
|
||||||
templateDataMap := make(map[string]any)
|
|
||||||
for _, data := range templateData {
|
|
||||||
maps.Copy(templateDataMap, data)
|
|
||||||
}
|
|
||||||
msg, err := localizer.Localize(&i18n.LocalizeConfig{
|
|
||||||
MessageID: key,
|
|
||||||
TemplateData: templateDataMap,
|
|
||||||
})
|
|
||||||
if err != nil {
|
|
||||||
return key
|
|
||||||
}
|
}
|
||||||
return msg
|
return msg
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,19 +1,260 @@
|
|||||||
// Code generated by cmd/gen_i18n. DO NOT EDIT.
|
// Code generated by cmd/geni18n. DO NOT EDIT.
|
||||||
package i18nk
|
package i18nk
|
||||||
|
|
||||||
|
type Key string
|
||||||
|
|
||||||
const (
|
const (
|
||||||
CleanCacheFailed = "CleanCacheFailed"
|
BotMsgAria2ErrorAddingAria2Download Key = "bot.msg.aria2.error_adding_aria2_download"
|
||||||
CleaningCache = "CleaningCache"
|
BotMsgAria2ErrorAria2ClientInitFailed Key = "bot.msg.aria2.error_aria2_client_init_failed"
|
||||||
ConfigInvalidDuplicateStorageName = "ConfigInvalid.DuplicateStorageName"
|
BotMsgAria2ErrorAria2NotEnabled Key = "bot.msg.aria2.error_aria2_not_enabled"
|
||||||
ConfigInvalidWorkersOrRetry = "ConfigInvalid.WorkersOrRetry"
|
BotMsgAria2InfoAddingAria2Download Key = "bot.msg.aria2.info_adding_aria2_download"
|
||||||
CreateRmTimerFailed = "CreateRmTimerFailed"
|
BotMsgAria2InfoAria2DownloadAdded Key = "bot.msg.aria2.info_aria2_download_added"
|
||||||
GetCacheAbsPathFailed = "GetCacheAbsPathFailed"
|
BotMsgAria2InfoSelectStorage Key = "bot.msg.aria2.info_select_storage"
|
||||||
GetWorkdirFailed = "GetWorkdirFailed"
|
BotMsgCancelErrorCancelFailed Key = "bot.msg.cancel.error_cancel_failed"
|
||||||
InvalidCacheDir = "InvalidCacheDir"
|
BotMsgCancelInfoCancelRequested Key = "bot.msg.cancel.info_cancel_requested"
|
||||||
LoadedStorages = "LoadedStorages"
|
BotMsgCancelInfoCancellingTask Key = "bot.msg.cancel.info_cancelling_task"
|
||||||
RemoveFileAfter = "RemoveFileAfter"
|
BotMsgCancelUsage Key = "bot.msg.cancel.usage"
|
||||||
RemoveFileFailed = "RemoveFileFailed"
|
BotMsgCmdAria2dl Key = "bot.msg.cmd.aria2dl"
|
||||||
Bye = "bye"
|
BotMsgCmdCancel Key = "bot.msg.cmd.cancel"
|
||||||
Exiting = "exiting"
|
BotMsgCmdConfig Key = "bot.msg.cmd.config"
|
||||||
Initing = "initing"
|
BotMsgCmdDir Key = "bot.msg.cmd.dir"
|
||||||
|
BotMsgCmdDl Key = "bot.msg.cmd.dl"
|
||||||
|
BotMsgCmdFnametmpl Key = "bot.msg.cmd.fnametmpl"
|
||||||
|
BotMsgCmdHelp Key = "bot.msg.cmd.help"
|
||||||
|
BotMsgCmdLswatch Key = "bot.msg.cmd.lswatch"
|
||||||
|
BotMsgCmdParser Key = "bot.msg.cmd.parser"
|
||||||
|
BotMsgCmdRule Key = "bot.msg.cmd.rule"
|
||||||
|
BotMsgCmdSave Key = "bot.msg.cmd.save"
|
||||||
|
BotMsgCmdSilent Key = "bot.msg.cmd.silent"
|
||||||
|
BotMsgCmdStart Key = "bot.msg.cmd.start"
|
||||||
|
BotMsgCmdStorage Key = "bot.msg.cmd.storage"
|
||||||
|
BotMsgCmdSyncpeers Key = "bot.msg.cmd.syncpeers"
|
||||||
|
BotMsgCmdTask Key = "bot.msg.cmd.task"
|
||||||
|
BotMsgCmdUnwatch Key = "bot.msg.cmd.unwatch"
|
||||||
|
BotMsgCmdUpdate Key = "bot.msg.cmd.update"
|
||||||
|
BotMsgCmdWatch Key = "bot.msg.cmd.watch"
|
||||||
|
BotMsgCmdYtdlp Key = "bot.msg.cmd.ytdlp"
|
||||||
|
BotMsgCommonCancelButtonText Key = "bot.msg.common.cancel_button_text"
|
||||||
|
BotMsgCommonErrorBuildDirSelectKeyboardFailed Key = "bot.msg.common.error_build_dir_select_keyboard_failed"
|
||||||
|
BotMsgCommonErrorBuildStorageSelectKeyboardFailed Key = "bot.msg.common.error_build_storage_select_keyboard_failed"
|
||||||
|
BotMsgCommonErrorBuildStorageSelectMessageFailed Key = "bot.msg.common.error_build_storage_select_message_failed"
|
||||||
|
BotMsgCommonErrorDataExpired Key = "bot.msg.common.error_data_expired"
|
||||||
|
BotMsgCommonErrorDefaultStorageNotSet Key = "bot.msg.common.error_default_storage_not_set"
|
||||||
|
BotMsgCommonErrorGetDirFailed Key = "bot.msg.common.error_get_dir_failed"
|
||||||
|
BotMsgCommonErrorGetFileFailed Key = "bot.msg.common.error_get_file_failed"
|
||||||
|
BotMsgCommonErrorGetMessagesFailed Key = "bot.msg.common.error_get_messages_failed"
|
||||||
|
BotMsgCommonErrorGetStorageFailed Key = "bot.msg.common.error_get_storage_failed"
|
||||||
|
BotMsgCommonErrorGetTelegraphPageFailed Key = "bot.msg.common.error_get_telegraph_page_failed"
|
||||||
|
BotMsgCommonErrorGetUserFailed Key = "bot.msg.common.error_get_user_failed"
|
||||||
|
BotMsgCommonErrorGetUserInfoFailed Key = "bot.msg.common.error_get_user_info_failed"
|
||||||
|
BotMsgCommonErrorGetUserWithErrFailed Key = "bot.msg.common.error_get_user_with_err_failed"
|
||||||
|
BotMsgCommonErrorInvalidIdOrUsername Key = "bot.msg.common.error_invalid_id_or_username"
|
||||||
|
BotMsgCommonErrorInvalidMsgIdRange Key = "bot.msg.common.error_invalid_msg_id_range"
|
||||||
|
BotMsgCommonErrorInvalidRegex Key = "bot.msg.common.error_invalid_regex"
|
||||||
|
BotMsgCommonErrorNoAvailableStorage Key = "bot.msg.common.error_no_available_storage"
|
||||||
|
BotMsgCommonErrorNoImagesInTelegraphPage Key = "bot.msg.common.error_no_images_in_telegraph_page"
|
||||||
|
BotMsgCommonErrorNoMessagesInRange Key = "bot.msg.common.error_no_messages_in_range"
|
||||||
|
BotMsgCommonErrorNoPermission Key = "bot.msg.common.error_no_permission"
|
||||||
|
BotMsgCommonErrorNoSavableFilesFound Key = "bot.msg.common.error_no_savable_files_found"
|
||||||
|
BotMsgCommonErrorNoSavableMessagesInRange Key = "bot.msg.common.error_no_savable_messages_in_range"
|
||||||
|
BotMsgCommonErrorParseTelegraphPathFailed Key = "bot.msg.common.error_parse_telegraph_path_failed"
|
||||||
|
BotMsgCommonErrorTaskAddFailed Key = "bot.msg.common.error_task_add_failed"
|
||||||
|
BotMsgCommonErrorTaskCreateFailed Key = "bot.msg.common.error_task_create_failed"
|
||||||
|
BotMsgCommonErrorUpdateUserInfoFailed Key = "bot.msg.common.error_update_user_info_failed"
|
||||||
|
BotMsgCommonInfoBatchTasksAdded Key = "bot.msg.common.info_batch_tasks_added"
|
||||||
|
BotMsgCommonInfoDefaultStorageSet Key = "bot.msg.common.info_default_storage_set"
|
||||||
|
BotMsgCommonInfoDefaultStorageWithDirSet Key = "bot.msg.common.info_default_storage_with_dir_set"
|
||||||
|
BotMsgCommonInfoFetchingFileInfo Key = "bot.msg.common.info_fetching_file_info"
|
||||||
|
BotMsgCommonInfoFetchingMessages Key = "bot.msg.common.info_fetching_messages"
|
||||||
|
BotMsgCommonInfoFetchingTelegraphPage Key = "bot.msg.common.info_fetching_telegraph_page"
|
||||||
|
BotMsgCommonInfoFoundFilesSelectStorage Key = "bot.msg.common.info_found_files_select_storage"
|
||||||
|
BotMsgCommonInfoSilentModeOff Key = "bot.msg.common.info_silent_mode_off"
|
||||||
|
BotMsgCommonInfoSilentModeOn Key = "bot.msg.common.info_silent_mode_on"
|
||||||
|
BotMsgCommonInfoTaskAdded Key = "bot.msg.common.info_task_added"
|
||||||
|
BotMsgCommonPromptSelectDefaultDir Key = "bot.msg.common.prompt_select_default_dir"
|
||||||
|
BotMsgCommonPromptSelectDefaultStorage Key = "bot.msg.common.prompt_select_default_storage"
|
||||||
|
BotMsgCommonPromptSelectDir Key = "bot.msg.common.prompt_select_dir"
|
||||||
|
BotMsgConfigButtonFilenameStrategy Key = "bot.msg.config.button_filename_strategy"
|
||||||
|
BotMsgConfigErrorInvalidCallbackData Key = "bot.msg.config.error_invalid_callback_data"
|
||||||
|
BotMsgConfigErrorInvalidTemplate Key = "bot.msg.config.error_invalid_template"
|
||||||
|
BotMsgConfigFnametmplHelp Key = "bot.msg.config.fnametmpl_help"
|
||||||
|
BotMsgConfigInfoCurrentTemplatePrefix Key = "bot.msg.config.info_current_template_prefix"
|
||||||
|
BotMsgConfigInfoFilenameStrategySet Key = "bot.msg.config.info_filename_strategy_set"
|
||||||
|
BotMsgConfigInfoTemplateUpdated Key = "bot.msg.config.info_template_updated"
|
||||||
|
BotMsgConfigPromptSelectFilenameStrategy Key = "bot.msg.config.prompt_select_filename_strategy"
|
||||||
|
BotMsgConfigPromptSelectOption Key = "bot.msg.config.prompt_select_option"
|
||||||
|
BotMsgDirButtonDefault Key = "bot.msg.dir.button_default"
|
||||||
|
BotMsgDirErrorCreateDirFailed Key = "bot.msg.dir.error_create_dir_failed"
|
||||||
|
BotMsgDirErrorDeleteDirFailed Key = "bot.msg.dir.error_delete_dir_failed"
|
||||||
|
BotMsgDirErrorGetUserDirsFailed Key = "bot.msg.dir.error_get_user_dirs_failed"
|
||||||
|
BotMsgDirErrorGetUserFailed Key = "bot.msg.dir.error_get_user_failed"
|
||||||
|
BotMsgDirErrorInvalidDirId Key = "bot.msg.dir.error_invalid_dir_id"
|
||||||
|
BotMsgDirErrorUnknownOperation Key = "bot.msg.dir.error_unknown_operation"
|
||||||
|
BotMsgDirHelpAddExampleCmd Key = "bot.msg.dir.help_add_example_cmd"
|
||||||
|
BotMsgDirHelpAddExamplePrefix Key = "bot.msg.dir.help_add_example_prefix"
|
||||||
|
BotMsgDirHelpAddSuffix Key = "bot.msg.dir.help_add_suffix"
|
||||||
|
BotMsgDirHelpAvailableOps Key = "bot.msg.dir.help_available_ops"
|
||||||
|
BotMsgDirHelpDelExampleCmd Key = "bot.msg.dir.help_del_example_cmd"
|
||||||
|
BotMsgDirHelpDelExamplePrefix Key = "bot.msg.dir.help_del_example_prefix"
|
||||||
|
BotMsgDirHelpDelSuffix Key = "bot.msg.dir.help_del_suffix"
|
||||||
|
BotMsgDirHelpExistingDirsPrefix Key = "bot.msg.dir.help_existing_dirs_prefix"
|
||||||
|
BotMsgDirHelpUsage Key = "bot.msg.dir.help_usage"
|
||||||
|
BotMsgDirInfoCreateDirSuccess Key = "bot.msg.dir.info_create_dir_success"
|
||||||
|
BotMsgDirInfoDeleteDirSuccess Key = "bot.msg.dir.info_delete_dir_success"
|
||||||
|
BotMsgDlErrorNoValidLinks Key = "bot.msg.dl.error_no_valid_links"
|
||||||
|
BotMsgDlInfoFilesSelectStorage Key = "bot.msg.dl.info_files_select_storage"
|
||||||
|
BotMsgDlUsage Key = "bot.msg.dl.usage"
|
||||||
|
BotMsgHelpTextFmt Key = "bot.msg.help_text_fmt"
|
||||||
|
BotMsgMediaGroupErrorBuildStorageSelectKeyboardFailed Key = "bot.msg.media_group.error_build_storage_select_keyboard_failed"
|
||||||
|
BotMsgMediaGroupInfoGroupFoundFilesSelectStorage Key = "bot.msg.media_group.info_group_found_files_select_storage"
|
||||||
|
BotMsgMediaGroupInfoSavingFiles Key = "bot.msg.media_group.info_saving_files"
|
||||||
|
BotMsgParseErrorBuildParsedTextEntityFailed Key = "bot.msg.parse.error_build_parsed_text_entity_failed"
|
||||||
|
BotMsgParseErrorBuildStorageSelectKeyboardFailed Key = "bot.msg.parse.error_build_storage_select_keyboard_failed"
|
||||||
|
BotMsgParseErrorParseTextFailed Key = "bot.msg.parse.error_parse_text_failed"
|
||||||
|
BotMsgParseInfoAuthorPrefix Key = "bot.msg.parse.info_author_prefix"
|
||||||
|
BotMsgParseInfoDescriptionPrefix Key = "bot.msg.parse.info_description_prefix"
|
||||||
|
BotMsgParseInfoFileCountPrefix Key = "bot.msg.parse.info_file_count_prefix"
|
||||||
|
BotMsgParseInfoLinkPrefix Key = "bot.msg.parse.info_link_prefix"
|
||||||
|
BotMsgParseInfoParsing Key = "bot.msg.parse.info_parsing"
|
||||||
|
BotMsgParseInfoPromptSelectStorage Key = "bot.msg.parse.info_prompt_select_storage"
|
||||||
|
BotMsgParseInfoTotalSizePrefix Key = "bot.msg.parse.info_total_size_prefix"
|
||||||
|
BotMsgParserErrorDownloadFileFailed Key = "bot.msg.parser.error_download_file_failed"
|
||||||
|
BotMsgParserErrorFileTooLarge Key = "bot.msg.parser.error_file_too_large"
|
||||||
|
BotMsgParserErrorGetFilenameFailed Key = "bot.msg.parser.error_get_filename_failed"
|
||||||
|
BotMsgParserErrorInstallPluginFailed Key = "bot.msg.parser.error_install_plugin_failed"
|
||||||
|
BotMsgParserErrorNoValidFileInReply Key = "bot.msg.parser.error_no_valid_file_in_reply"
|
||||||
|
BotMsgParserErrorOnlyJsSupported Key = "bot.msg.parser.error_only_js_supported"
|
||||||
|
BotMsgParserErrorWrongFileType Key = "bot.msg.parser.error_wrong_file_type"
|
||||||
|
BotMsgParserHelpText Key = "bot.msg.parser.help_text"
|
||||||
|
BotMsgParserInfoInstallPluginSuccess Key = "bot.msg.parser.info_install_plugin_success"
|
||||||
|
BotMsgParserPluginNotEnabled Key = "bot.msg.parser.plugin_not_enabled"
|
||||||
|
BotMsgParserPromptReplyWithParserFile Key = "bot.msg.parser.prompt_reply_with_parser_file"
|
||||||
|
BotMsgProgressAria2Done Key = "bot.msg.progress.aria2_done"
|
||||||
|
BotMsgProgressAria2Downloading Key = "bot.msg.progress.aria2_downloading"
|
||||||
|
BotMsgProgressAria2Start Key = "bot.msg.progress.aria2_start"
|
||||||
|
BotMsgProgressAvgSpeedPrefix Key = "bot.msg.progress.avg_speed_prefix"
|
||||||
|
BotMsgProgressBatchDonePrefix Key = "bot.msg.progress.batch_done_prefix"
|
||||||
|
BotMsgProgressBatchProcessingPrefix Key = "bot.msg.progress.batch_processing_prefix"
|
||||||
|
BotMsgProgressBatchStartPrefix Key = "bot.msg.progress.batch_start_prefix"
|
||||||
|
BotMsgProgressCurrentProgressPrefix Key = "bot.msg.progress.current_progress_prefix"
|
||||||
|
BotMsgProgressCurrentSpeedPrefix Key = "bot.msg.progress.current_speed_prefix"
|
||||||
|
BotMsgProgressDirectDonePrefix Key = "bot.msg.progress.direct_done_prefix"
|
||||||
|
BotMsgProgressDirectStart Key = "bot.msg.progress.direct_start"
|
||||||
|
BotMsgProgressDownloadDonePrefix Key = "bot.msg.progress.download_done_prefix"
|
||||||
|
BotMsgProgressDownloadFailedPrefix Key = "bot.msg.progress.download_failed_prefix"
|
||||||
|
BotMsgProgressDownloadedPrefix Key = "bot.msg.progress.downloaded_prefix"
|
||||||
|
BotMsgProgressDownloadingPrefix Key = "bot.msg.progress.downloading_prefix"
|
||||||
|
BotMsgProgressErrorPrefix Key = "bot.msg.progress.error_prefix"
|
||||||
|
BotMsgProgressFileNamePrefix Key = "bot.msg.progress.file_name_prefix"
|
||||||
|
BotMsgProgressFileProcessingPrefix Key = "bot.msg.progress.file_processing_prefix"
|
||||||
|
BotMsgProgressFileSizePrefix Key = "bot.msg.progress.file_size_prefix"
|
||||||
|
BotMsgProgressFileStartPrefix Key = "bot.msg.progress.file_start_prefix"
|
||||||
|
BotMsgProgressParsedDonePrefix Key = "bot.msg.progress.parsed_done_prefix"
|
||||||
|
BotMsgProgressParsedStartPrefix Key = "bot.msg.progress.parsed_start_prefix"
|
||||||
|
BotMsgProgressProcessingListPrefix Key = "bot.msg.progress.processing_list_prefix"
|
||||||
|
BotMsgProgressProcessingNone Key = "bot.msg.progress.processing_none"
|
||||||
|
BotMsgProgressSavePathPrefix Key = "bot.msg.progress.save_path_prefix"
|
||||||
|
BotMsgProgressTaskCanceled Key = "bot.msg.progress.task_canceled"
|
||||||
|
BotMsgProgressTaskCanceledWithId Key = "bot.msg.progress.task_canceled_with_id"
|
||||||
|
BotMsgProgressTaskFailedWithError Key = "bot.msg.progress.task_failed_with_error"
|
||||||
|
BotMsgProgressTelegraphDonePrefix Key = "bot.msg.progress.telegraph_done_prefix"
|
||||||
|
BotMsgProgressTelegraphProgressPrefix Key = "bot.msg.progress.telegraph_progress_prefix"
|
||||||
|
BotMsgProgressTelegraphStartPrefix Key = "bot.msg.progress.telegraph_start_prefix"
|
||||||
|
BotMsgProgressTotalSizePrefix Key = "bot.msg.progress.total_size_prefix"
|
||||||
|
BotMsgProgressYtdlpDone Key = "bot.msg.progress.ytdlp_done"
|
||||||
|
BotMsgProgressYtdlpDownloading Key = "bot.msg.progress.ytdlp_downloading"
|
||||||
|
BotMsgProgressYtdlpStart Key = "bot.msg.progress.ytdlp_start"
|
||||||
|
BotMsgRuleErrorCreateRuleFailed Key = "bot.msg.rule.error_create_rule_failed"
|
||||||
|
BotMsgRuleErrorDeleteRuleFailed Key = "bot.msg.rule.error_delete_rule_failed"
|
||||||
|
BotMsgRuleErrorGetUserRulesFailed Key = "bot.msg.rule.error_get_user_rules_failed"
|
||||||
|
BotMsgRuleErrorInvalidRuleId Key = "bot.msg.rule.error_invalid_rule_id"
|
||||||
|
BotMsgRuleErrorInvalidRuleType Key = "bot.msg.rule.error_invalid_rule_type"
|
||||||
|
BotMsgRuleErrorUpdateUserFailed Key = "bot.msg.rule.error_update_user_failed"
|
||||||
|
BotMsgRuleHelpAddSuffix Key = "bot.msg.rule.help_add_suffix"
|
||||||
|
BotMsgRuleHelpAvailableOps Key = "bot.msg.rule.help_available_ops"
|
||||||
|
BotMsgRuleHelpCurrentModeDisabled Key = "bot.msg.rule.help_current_mode_disabled"
|
||||||
|
BotMsgRuleHelpCurrentModeEnabled Key = "bot.msg.rule.help_current_mode_enabled"
|
||||||
|
BotMsgRuleHelpDelSuffix Key = "bot.msg.rule.help_del_suffix"
|
||||||
|
BotMsgRuleHelpExistingRulesPrefix Key = "bot.msg.rule.help_existing_rules_prefix"
|
||||||
|
BotMsgRuleHelpSwitchSuffix Key = "bot.msg.rule.help_switch_suffix"
|
||||||
|
BotMsgRuleHelpUsage Key = "bot.msg.rule.help_usage"
|
||||||
|
BotMsgRuleInfoCreateRuleSuccess Key = "bot.msg.rule.info_create_rule_success"
|
||||||
|
BotMsgRuleInfoDeleteRuleSuccess Key = "bot.msg.rule.info_delete_rule_success"
|
||||||
|
BotMsgRuleInfoRuleModeDisabled Key = "bot.msg.rule.info_rule_mode_disabled"
|
||||||
|
BotMsgRuleInfoRuleModeEnabled Key = "bot.msg.rule.info_rule_mode_enabled"
|
||||||
|
BotMsgRulePromptProvideRuleId Key = "bot.msg.rule.prompt_provide_rule_id"
|
||||||
|
BotMsgSaveErrorInvalidIdOrUsername Key = "bot.msg.save.error_invalid_id_or_username"
|
||||||
|
BotMsgSaveHelpText Key = "bot.msg.save_help_text"
|
||||||
|
BotMsgStorageInfoFilenamePrefix Key = "bot.msg.storage.info_filename_prefix"
|
||||||
|
BotMsgStorageInfoPromptSelectStorage Key = "bot.msg.storage.info_prompt_select_storage"
|
||||||
|
BotMsgSyncpeersDone Key = "bot.msg.syncpeers.done"
|
||||||
|
BotMsgSyncpeersFailed Key = "bot.msg.syncpeers.failed"
|
||||||
|
BotMsgSyncpeersStart Key = "bot.msg.syncpeers.start"
|
||||||
|
BotMsgSyncpeersSuccess Key = "bot.msg.syncpeers.success"
|
||||||
|
BotMsgTasksCancelFailed Key = "bot.msg.tasks.cancel_failed"
|
||||||
|
BotMsgTasksCancelRequestedPrefix Key = "bot.msg.tasks.cancel_requested_prefix"
|
||||||
|
BotMsgTasksFieldCreated Key = "bot.msg.tasks.field_created"
|
||||||
|
BotMsgTasksFieldId Key = "bot.msg.tasks.field_id"
|
||||||
|
BotMsgTasksFieldStatus Key = "bot.msg.tasks.field_status"
|
||||||
|
BotMsgTasksFieldTitle Key = "bot.msg.tasks.field_title"
|
||||||
|
BotMsgTasksInfoAddedToQueueFull Key = "bot.msg.tasks.info_added_to_queue_full"
|
||||||
|
BotMsgTasksInfoAddedToQueuePrefix Key = "bot.msg.tasks.info_added_to_queue_prefix"
|
||||||
|
BotMsgTasksInfoFilenamePrefix Key = "bot.msg.tasks.info_filename_prefix"
|
||||||
|
BotMsgTasksInfoQueueLengthPrefix Key = "bot.msg.tasks.info_queue_length_prefix"
|
||||||
|
BotMsgTasksQueuedEmpty Key = "bot.msg.tasks.queued_empty"
|
||||||
|
BotMsgTasksQueuedTitle Key = "bot.msg.tasks.queued_title"
|
||||||
|
BotMsgTasksRunningEmpty Key = "bot.msg.tasks.running_empty"
|
||||||
|
BotMsgTasksRunningTitle Key = "bot.msg.tasks.running_title"
|
||||||
|
BotMsgTasksStatusCancelRequested Key = "bot.msg.tasks.status_cancel_requested"
|
||||||
|
BotMsgTasksStatusQueued Key = "bot.msg.tasks.status_queued"
|
||||||
|
BotMsgTasksStatusRunning Key = "bot.msg.tasks.status_running"
|
||||||
|
BotMsgTasksTotalPrefix Key = "bot.msg.tasks.total_prefix"
|
||||||
|
BotMsgTasksTruncatedNote Key = "bot.msg.tasks.truncated_note"
|
||||||
|
BotMsgTasksUsage Key = "bot.msg.tasks.usage"
|
||||||
|
BotMsgTasksUsageCancel Key = "bot.msg.tasks.usage_cancel"
|
||||||
|
BotMsgTelegraphErrorBuildStorageSelectKeyboardFailed Key = "bot.msg.telegraph.error_build_storage_select_keyboard_failed"
|
||||||
|
BotMsgTelegraphInfoPicCountPrefix Key = "bot.msg.telegraph.info_pic_count_prefix"
|
||||||
|
BotMsgTelegraphInfoPromptSelectStorage Key = "bot.msg.telegraph.info_prompt_select_storage"
|
||||||
|
BotMsgTelegraphInfoTitlePrefix Key = "bot.msg.telegraph.info_title_prefix"
|
||||||
|
BotMsgUpdateButtonUpgrade Key = "bot.msg.update.button_upgrade"
|
||||||
|
BotMsgUpdateErrorCheckLatestFailed Key = "bot.msg.update.error_check_latest_failed"
|
||||||
|
BotMsgUpdateErrorNoReleaseFound Key = "bot.msg.update.error_no_release_found"
|
||||||
|
BotMsgUpdateErrorUpgradeFailed Key = "bot.msg.update.error_upgrade_failed"
|
||||||
|
BotMsgUpdateErrorVersionVarInvalid Key = "bot.msg.update.error_version_var_invalid"
|
||||||
|
BotMsgUpdateInfoAlreadyLatest Key = "bot.msg.update.info_already_latest"
|
||||||
|
BotMsgUpdateInfoMajorUpgradeRequired Key = "bot.msg.update.info_major_upgrade_required"
|
||||||
|
BotMsgUpdateInfoNewVersionInDocker Key = "bot.msg.update.info_new_version_in_docker"
|
||||||
|
BotMsgUpdateInfoNewVersionPromptUpgrade Key = "bot.msg.update.info_new_version_prompt_upgrade"
|
||||||
|
BotMsgUpdateInfoUpgradeSuccess Key = "bot.msg.update.info_upgrade_success"
|
||||||
|
BotMsgUpdateInfoUpgradingWithVersion Key = "bot.msg.update.info_upgrading_with_version"
|
||||||
|
BotMsgWatchErrorFilterFormatInvalid Key = "bot.msg.watch.error_filter_format_invalid"
|
||||||
|
BotMsgWatchErrorFilterTypeUnsupported Key = "bot.msg.watch.error_filter_type_unsupported"
|
||||||
|
BotMsgWatchErrorUnwatchChatFailed Key = "bot.msg.watch.error_unwatch_chat_failed"
|
||||||
|
BotMsgWatchErrorUnwatchNoChatProvided Key = "bot.msg.watch.error_unwatch_no_chat_provided"
|
||||||
|
BotMsgWatchErrorWatchChatFailed Key = "bot.msg.watch.error_watch_chat_failed"
|
||||||
|
BotMsgWatchInfoAlreadyWatchingChat Key = "bot.msg.watch.info_already_watching_chat"
|
||||||
|
BotMsgWatchInfoWatchChatStarted Key = "bot.msg.watch.info_watch_chat_started"
|
||||||
|
BotMsgWatchInfoWatchChatStopped Key = "bot.msg.watch.info_watch_chat_stopped"
|
||||||
|
BotMsgWatchInfoWatchListEmpty Key = "bot.msg.watch.info_watch_list_empty"
|
||||||
|
BotMsgWatchInfoWatchListFilterPrefix Key = "bot.msg.watch.info_watch_list_filter_prefix"
|
||||||
|
BotMsgWatchInfoWatchListHeader Key = "bot.msg.watch.info_watch_list_header"
|
||||||
|
BotMsgWatchHelpText Key = "bot.msg.watch_help_text"
|
||||||
|
BotMsgYtdlpErrorDownloadFailed Key = "bot.msg.ytdlp.error_download_failed"
|
||||||
|
BotMsgYtdlpErrorNoValidUrls Key = "bot.msg.ytdlp.error_no_valid_urls"
|
||||||
|
BotMsgYtdlpInfoDownloading Key = "bot.msg.ytdlp.info_downloading"
|
||||||
|
BotMsgYtdlpInfoUrlsSelectStorage Key = "bot.msg.ytdlp.info_urls_select_storage"
|
||||||
|
BotMsgYtdlpUsage Key = "bot.msg.ytdlp.usage"
|
||||||
|
ConfigErrDuplicateStorageName Key = "config.err.duplicate_storage_name"
|
||||||
|
ConfigErrInvalidCacheDir Key = "config.err.invalid_cache_dir"
|
||||||
|
ErrCleanCacheFailed Key = "err.clean_cache_failed"
|
||||||
|
ErrGetCacheAbsPathFailed Key = "err.get_cache_abs_path_failed"
|
||||||
|
ErrGetWorkdirFailed Key = "err.get_workdir_failed"
|
||||||
|
LifetimeBye Key = "lifetime.bye"
|
||||||
|
LifetimeCleaningCache Key = "lifetime.cleaning_cache"
|
||||||
|
LifetimeExiting Key = "lifetime.exiting"
|
||||||
|
LifetimeInitfailed Key = "lifetime.initfailed"
|
||||||
|
LifetimeIniting Key = "lifetime.initing"
|
||||||
|
LifetimeUserLoginFailed Key = "lifetime.user_login_failed"
|
||||||
|
ParserPluginLoadFailed Key = "parser.plugin.load_failed"
|
||||||
|
ParserPluginLoadedDir Key = "parser.plugin.loaded_dir"
|
||||||
)
|
)
|
||||||
|
|||||||
355
common/i18n/locale/en.yaml
Normal file
355
common/i18n/locale/en.yaml
Normal file
@@ -0,0 +1,355 @@
|
|||||||
|
lifetime:
|
||||||
|
initing: Starting up
|
||||||
|
initfailed: Initialization failed
|
||||||
|
exiting: Shutting down
|
||||||
|
user_login_failed: "User login failed: {{.Error}}"
|
||||||
|
cleaning_cache: "Cleaning cache {{.Path}}"
|
||||||
|
bye: Exited
|
||||||
|
config:
|
||||||
|
err:
|
||||||
|
invalid_cache_dir: "Invalid cache directory: {{.Path}}, please check the config file"
|
||||||
|
duplicate_storage_name: "Storage name '{{.Name}}' is duplicated, please check the config file"
|
||||||
|
err:
|
||||||
|
get_workdir_failed: "Failed to get working directory: {{.Error}}"
|
||||||
|
get_cache_abs_path_failed: "Failed to get absolute cache path: {{.Error}}"
|
||||||
|
clean_cache_failed: "Failed to clean cache: {{.Error}}"
|
||||||
|
parser:
|
||||||
|
plugin:
|
||||||
|
load_failed: Failed to load parser plugins
|
||||||
|
loaded_dir: Parser plugins loaded
|
||||||
|
bot:
|
||||||
|
msg:
|
||||||
|
help_text_fmt: |
|
||||||
|
Save Any Bot - Save your Telegram files
|
||||||
|
Version: %s , Commit: %s
|
||||||
|
|
||||||
|
Commands:
|
||||||
|
/start - Start using the bot
|
||||||
|
/help - Show help
|
||||||
|
/silent - Toggle silent mode
|
||||||
|
/storage - Set default storage
|
||||||
|
/save [custom filename] - Save file
|
||||||
|
/dir - Manage storage directories
|
||||||
|
/rule - Manage rules
|
||||||
|
/config - Modify configuration
|
||||||
|
/fnametmpl - Set custom filename template
|
||||||
|
/parser - Manage parser plugins
|
||||||
|
/task - Manage task queue
|
||||||
|
/watch - Watch chats and auto save (UserBot)
|
||||||
|
/unwatch - Stop watching chats (UserBot)
|
||||||
|
/lswatch - List watched chats (UserBot)
|
||||||
|
/syncpeers - Sync peer chats (UserBot)
|
||||||
|
/update - Check and upgrade to latest version
|
||||||
|
|
||||||
|
Usage guide: https://sabot.unv.app/usage
|
||||||
|
cmd:
|
||||||
|
start: "Start using"
|
||||||
|
silent: "Toggle silent mode"
|
||||||
|
storage: "Set default storage"
|
||||||
|
dir: "Manage storage directories"
|
||||||
|
rule: "Manage auto-save rules"
|
||||||
|
save: "Save files"
|
||||||
|
dl: "Download files from given links"
|
||||||
|
aria2dl: "Download files using Aria2"
|
||||||
|
ytdlp: "Download video/audio using yt-dlp"
|
||||||
|
task: "Manage task queue"
|
||||||
|
cancel: "Cancel task"
|
||||||
|
watch: "Watch chats (UserBot)"
|
||||||
|
unwatch: "Stop watching chats (UserBot)"
|
||||||
|
lswatch: "List watched chats (UserBot)"
|
||||||
|
config: "Modify configuration"
|
||||||
|
fnametmpl: "Set filename template"
|
||||||
|
help: "Show help"
|
||||||
|
parser: "Manage parsers"
|
||||||
|
update: "Check for updates"
|
||||||
|
syncpeers: "Sync peer chats (UserBot)"
|
||||||
|
save_help_text: |
|
||||||
|
Usage:
|
||||||
|
|
||||||
|
1. Reply to the file you want to save with this command, optional filename parameter.
|
||||||
|
Example:
|
||||||
|
/save custom_file_name.mp4
|
||||||
|
|
||||||
|
2. After setting default storage, send /save <channel_id/username> <message_id_range> to batch save files. Rules will be applied; if no rule matches, default storage will be used.
|
||||||
|
Example:
|
||||||
|
/save @acherkrau 114-514
|
||||||
|
watch_help_text: |
|
||||||
|
Use /watch to watch messages in a chat and automatically save them to the default storage, following storage rules.
|
||||||
|
|
||||||
|
Syntax:
|
||||||
|
/watch <chat_id> [filter]
|
||||||
|
|
||||||
|
Parameters:
|
||||||
|
- <chat_id>: Chat ID or username
|
||||||
|
- [filter]: Optional, format is filter_type:expression , see docs for all supported filters
|
||||||
|
|
||||||
|
Example:
|
||||||
|
/watch -1002229835658 msgre:.*plana.*
|
||||||
|
|
||||||
|
This will watch chat with ID -1002229835658 and save all media messages containing "plana".
|
||||||
|
common:
|
||||||
|
cancel_button_text: "Cancel"
|
||||||
|
error_invalid_regex: "Invalid regex: {{.Error}}"
|
||||||
|
error_invalid_msg_id_range: "Invalid message ID range: {{.Error}}"
|
||||||
|
error_invalid_id_or_username: "Invalid ID or username: {{.Error}}"
|
||||||
|
error_get_messages_failed: "Failed to get messages: {{.Error}}"
|
||||||
|
info_fetching_messages: "Fetching messages..."
|
||||||
|
error_no_messages_in_range: "No messages found in the specified range"
|
||||||
|
error_no_savable_messages_in_range: "No savable messages found in the specified range"
|
||||||
|
error_build_storage_select_message_failed: "Failed to build storage selection message: {{.Error}}"
|
||||||
|
error_build_storage_select_keyboard_failed: "Failed to build storage selection keyboard: {{.Error}}"
|
||||||
|
info_found_files_select_storage: "Found {{.Count}} files, please select storage"
|
||||||
|
error_get_user_failed: "Failed to get user"
|
||||||
|
error_get_user_with_err_failed: "Failed to get user: {{.Error}}"
|
||||||
|
error_default_storage_not_set: "Please set a default storage first with /storage"
|
||||||
|
error_no_available_storage: "No available storage"
|
||||||
|
error_get_storage_failed: "Failed to get storage: {{.Error}}"
|
||||||
|
prompt_select_default_storage: "Please select a storage to set as default"
|
||||||
|
error_data_expired: "Data has expired or is invalid"
|
||||||
|
error_task_add_failed: "Failed to add task: {{.Error}}"
|
||||||
|
info_task_added: "Task added"
|
||||||
|
info_batch_tasks_added: "Batch tasks added, total {{.Count}} files"
|
||||||
|
error_task_create_failed: "Failed to create task: {{.Error}}"
|
||||||
|
error_get_dir_failed: "Failed to get directory: {{.Error}}"
|
||||||
|
prompt_select_dir: "Please select a directory to store to"
|
||||||
|
prompt_select_default_dir: "Please select a default directory to save to"
|
||||||
|
info_default_storage_set: "Default storage set to: {{.Name}}"
|
||||||
|
info_default_storage_with_dir_set: "Default storage set to: {{.Name}}:/{{.Dir}}"
|
||||||
|
error_get_user_info_failed: "Failed to get user info: {{.Error}}"
|
||||||
|
error_update_user_info_failed: "Failed to update user info: {{.Error}}"
|
||||||
|
info_silent_mode_on: "Silent mode enabled"
|
||||||
|
info_silent_mode_off: "Silent mode disabled"
|
||||||
|
error_get_file_failed: "Failed to get file: {{.Error}}"
|
||||||
|
info_fetching_file_info: "Fetching file info..."
|
||||||
|
error_no_savable_files_found: "No savable files found"
|
||||||
|
error_parse_telegraph_path_failed: "Failed to parse telegraph path: {{.Error}}"
|
||||||
|
info_fetching_telegraph_page: "Fetching telegraph page..."
|
||||||
|
error_get_telegraph_page_failed: "Failed to get telegraph page: {{.Error}}"
|
||||||
|
error_no_images_in_telegraph_page: "No images found in telegraph page"
|
||||||
|
error_build_dir_select_keyboard_failed: "Failed to build directory selection keyboard: {{.Error}}"
|
||||||
|
error_no_permission: |
|
||||||
|
You are not in the whitelist and cannot use this bot.
|
||||||
|
You can deploy your own instance: https://github.com/krau/SaveAny-Bot
|
||||||
|
save:
|
||||||
|
error_invalid_id_or_username: "Invalid ID or username: {{.Error}}"
|
||||||
|
watch:
|
||||||
|
error_filter_format_invalid: "Invalid filter format, please use <type>:<expression>"
|
||||||
|
error_filter_type_unsupported: "Unsupported filter type, please see the docs"
|
||||||
|
error_watch_chat_failed: "Failed to watch chat: {{.Error}}"
|
||||||
|
info_watch_chat_started: "Started watching chat: {{.Chat}}"
|
||||||
|
info_already_watching_chat: "Already watching this chat"
|
||||||
|
info_watch_list_empty: "No chats are being watched currently"
|
||||||
|
info_watch_list_header: "Currently watched chats:\n"
|
||||||
|
info_watch_list_filter_prefix: " (filter: "
|
||||||
|
error_unwatch_no_chat_provided: "Please provide a chat ID or username to unwatch"
|
||||||
|
error_unwatch_chat_failed: "Failed to unwatch chat: {{.Error}}"
|
||||||
|
info_watch_chat_stopped: "Stopped watching chat: {{.Chat}}"
|
||||||
|
tasks:
|
||||||
|
usage_cancel: "Usage: /tasks cancel <task_id>"
|
||||||
|
usage: "Usage: /tasks [running|queued|cancel <task_id>]"
|
||||||
|
cancel_failed: "Failed to cancel task: {{.Error}}"
|
||||||
|
cancel_requested_prefix: "Cancel requested for task: "
|
||||||
|
running_empty: "No running tasks"
|
||||||
|
running_title: "Currently running tasks:"
|
||||||
|
total_prefix: "Total: {{.Count}}\n"
|
||||||
|
field_id: "ID: "
|
||||||
|
field_title: "Title: "
|
||||||
|
field_created: "Created at: "
|
||||||
|
field_status: "Status: "
|
||||||
|
status_running: "Running"
|
||||||
|
status_queued: "Queued"
|
||||||
|
status_cancel_requested: "Cancel requested"
|
||||||
|
queued_empty: "No queued tasks"
|
||||||
|
queued_title: "Currently queued tasks:"
|
||||||
|
truncated_note: "...\nShowing first 10 tasks, total {{.Count}} tasks"
|
||||||
|
info_added_to_queue_full: "Added to task queue\nFilename: {{.Filename}}\nCurrent queued tasks: {{.QueueLength}}"
|
||||||
|
info_added_to_queue_prefix: "Added to task queue\n"
|
||||||
|
info_filename_prefix: "Filename: "
|
||||||
|
info_queue_length_prefix: "\nCurrent queued tasks: "
|
||||||
|
rule:
|
||||||
|
error_get_user_rules_failed: "Failed to get user rules"
|
||||||
|
error_update_user_failed: "Failed to update user"
|
||||||
|
info_rule_mode_enabled: "Rule mode enabled"
|
||||||
|
info_rule_mode_disabled: "Rule mode disabled"
|
||||||
|
error_invalid_rule_type: "Invalid rule type: {{.Type}}\nAvailable: {{.Available}}"
|
||||||
|
error_create_rule_failed: "Failed to create rule"
|
||||||
|
info_create_rule_success: "Rule created successfully"
|
||||||
|
prompt_provide_rule_id: "Please provide rule ID"
|
||||||
|
error_invalid_rule_id: "Invalid rule ID"
|
||||||
|
error_delete_rule_failed: "Failed to delete rule"
|
||||||
|
info_delete_rule_success: "Rule deleted successfully"
|
||||||
|
help_usage: "Usage: /rule <op> <args...>"
|
||||||
|
help_current_mode_enabled: "\nRule mode is currently enabled"
|
||||||
|
help_current_mode_disabled: "\nRule mode is currently disabled"
|
||||||
|
help_available_ops: "\n\nAvailable operations:\n"
|
||||||
|
help_switch_suffix: " - Toggle rule mode\n"
|
||||||
|
help_add_suffix: " <type> <data> <storage_name> <path> - Add rule\n"
|
||||||
|
help_del_suffix: " <rule_id> - Delete rule\n"
|
||||||
|
help_existing_rules_prefix: "\nCurrent rules:\n"
|
||||||
|
dir:
|
||||||
|
error_get_user_dirs_failed: "Failed to get user directories"
|
||||||
|
error_get_user_failed: "Failed to get user"
|
||||||
|
error_create_dir_failed: "Failed to create directory"
|
||||||
|
info_create_dir_success: "Directory created successfully"
|
||||||
|
error_invalid_dir_id: "Invalid directory ID"
|
||||||
|
error_delete_dir_failed: "Failed to delete directory"
|
||||||
|
info_delete_dir_success: "Directory deleted successfully"
|
||||||
|
error_unknown_operation: "Unknown operation"
|
||||||
|
help_usage: "Usage: /dir <op> <args...>"
|
||||||
|
help_available_ops: "\n\nAvailable operations:\n"
|
||||||
|
help_add_suffix: " <storage_name> <path> - Add path\n"
|
||||||
|
help_del_suffix: " <path_id> - Delete path\n"
|
||||||
|
help_add_example_prefix: "\nAdd path example:\n"
|
||||||
|
help_add_example_cmd: "/dir add local1 path/to/dir"
|
||||||
|
help_del_example_prefix: "\n\nDelete path example:\n"
|
||||||
|
help_del_example_cmd: "/dir del 3"
|
||||||
|
help_existing_dirs_prefix: "\n\nCurrent paths:\n"
|
||||||
|
button_default: "Default"
|
||||||
|
parser:
|
||||||
|
help_text: |
|
||||||
|
Usage:
|
||||||
|
|
||||||
|
/parser install <reply a file> - Install parser
|
||||||
|
plugin_not_enabled: "Parser plugin feature is not enabled"
|
||||||
|
prompt_reply_with_parser_file: "Please reply with a message containing the parser file"
|
||||||
|
error_no_valid_file_in_reply: "The replied message does not contain a valid file"
|
||||||
|
error_wrong_file_type: "Wrong file type"
|
||||||
|
error_file_too_large: "File too large"
|
||||||
|
error_get_filename_failed: "Failed to get filename"
|
||||||
|
error_only_js_supported: "Only .js files are supported as parsers"
|
||||||
|
error_download_file_failed: "Failed to download file: {{.Error}}"
|
||||||
|
error_install_plugin_failed: "Failed to install plugin: {{.Error}}"
|
||||||
|
info_install_plugin_success: "Plugin installed: {{.Name}}"
|
||||||
|
parse:
|
||||||
|
info_parsing: "Parsing..."
|
||||||
|
error_parse_text_failed: "Failed to parse text: {{.Error}}"
|
||||||
|
error_build_storage_select_keyboard_failed: "Failed to build storage selection keyboard: {{.Error}}"
|
||||||
|
error_build_parsed_text_entity_failed: "Failed to build parsed text entity: {{.Error}}"
|
||||||
|
info_link_prefix: "\nLink: "
|
||||||
|
info_author_prefix: "\nAuthor: "
|
||||||
|
info_description_prefix: "\nDescription: "
|
||||||
|
info_file_count_prefix: "\nFile count: "
|
||||||
|
info_total_size_prefix: "\nEstimated total size: "
|
||||||
|
info_prompt_select_storage: "\nPlease select storage"
|
||||||
|
telegraph:
|
||||||
|
error_build_storage_select_keyboard_failed: "Failed to build storage selection keyboard: {{.Error}}"
|
||||||
|
info_title_prefix: "Title: "
|
||||||
|
info_pic_count_prefix: "\nImage count: "
|
||||||
|
info_prompt_select_storage: "\nPlease select storage"
|
||||||
|
update:
|
||||||
|
error_version_var_invalid: "Currently in development version or version info injection failed: {{.Error}}"
|
||||||
|
error_check_latest_failed: "Failed to check latest version: {{.Error}}"
|
||||||
|
error_no_release_found: "No release found"
|
||||||
|
info_major_upgrade_required: "Major upgrade detected: {{.Current}} -> {{.Latest}} , please download the latest version from GitHub and check the migration guide"
|
||||||
|
info_already_latest: "Already on latest version: {{.Version}}"
|
||||||
|
info_new_version_in_docker: |-
|
||||||
|
New version found: {{.Latest}}
|
||||||
|
Current version: {{.Current}}
|
||||||
|
Published at: {{.PublishedAt}}
|
||||||
|
Since you are using Docker, please update it on your deployment platform
|
||||||
|
info_new_version_prompt_upgrade: |-
|
||||||
|
New version found: {{.Latest}}
|
||||||
|
Current version: {{.Current}}
|
||||||
|
|
||||||
|
File size: {{.SizeMB}} MB
|
||||||
|
Download URL: {{.URL}}
|
||||||
|
Published at: {{.PublishedAt}}
|
||||||
|
|
||||||
|
Upgrading will restart the bot. Proceed?
|
||||||
|
info_upgrading_with_version: "Upgrading, current version: {{.Current}}"
|
||||||
|
error_upgrade_failed: "Upgrade failed: {{.Error}}"
|
||||||
|
info_upgrade_success: "Upgraded to version {{.Version}}\nIf the bot did not restart automatically, please start it manually"
|
||||||
|
button_upgrade: "Upgrade"
|
||||||
|
config:
|
||||||
|
prompt_select_option: "Please select an option to configure"
|
||||||
|
button_filename_strategy: "Filename strategy"
|
||||||
|
error_invalid_callback_data: "Invalid callback data"
|
||||||
|
error_invalid_template: "Invalid template, please check syntax\n{{.Error}}"
|
||||||
|
info_filename_strategy_set: "Filename strategy set to: {{.Strategy}}"
|
||||||
|
prompt_select_filename_strategy: "Please select filename strategy, current strategy: {{.Strategy}}"
|
||||||
|
fnametmpl_help: |-
|
||||||
|
Use this command to set filename template, for example:
|
||||||
|
/fnametmpl Image_{{"{{.msgid}}"}}_{{"{{.msgdate}}"}}.jpg
|
||||||
|
|
||||||
|
Available variables:
|
||||||
|
- {{"{{.msgid}}"}}: Message ID
|
||||||
|
- {{"{{.msgtags}}"}}: Tags in the message, joined with underscore
|
||||||
|
- {{"{{.msggen}}"}}: Generated filename from the message
|
||||||
|
- {{"{{.msgdate}}"}}: Message date, format YYYY-MM-DD_HH-MM-SS
|
||||||
|
- {{"{{.msgraw}}"}}: Raw message text (unprocessed)
|
||||||
|
- {{"{{.origname}}"}}: Original media filename (if any)
|
||||||
|
- {{"{{.chatid}}"}}: Chat ID of the message
|
||||||
|
|
||||||
|
Template only takes effect when filename strategy is set to 'Custom template'.
|
||||||
|
If template parsing fails, it will fall back to default filename.
|
||||||
|
info_template_updated: "Filename template updated"
|
||||||
|
info_current_template_prefix: "Current template: {{.Template}}"
|
||||||
|
dl:
|
||||||
|
usage: "Usage: /dl <url1> <url2> ..."
|
||||||
|
error_no_valid_links: "No valid links to download"
|
||||||
|
info_files_select_storage: "Total {{.Count}} files, please select storage"
|
||||||
|
ytdlp:
|
||||||
|
usage: "Usage: /ytdlp [OPTIONS] <URL1> [URL2] ...\nExamples:\n /ytdlp https://example.com/video\n /ytdlp --format best https://example.com/video\n /ytdlp --extract-audio --audio-format mp3 https://example.com/video"
|
||||||
|
error_no_valid_urls: "No valid URLs"
|
||||||
|
info_urls_select_storage: "Found {{.Count}} links, please select storage"
|
||||||
|
info_downloading: "Downloading via yt-dlp..."
|
||||||
|
error_download_failed: "yt-dlp download failed: {{.Error}}"
|
||||||
|
cancel:
|
||||||
|
usage: "Usage: /cancel <task_id>"
|
||||||
|
error_cancel_failed: "Failed to cancel task: {{.Error}}"
|
||||||
|
info_cancel_requested: "Cancel requested for task: {{.TaskID}}"
|
||||||
|
info_cancelling_task: "Cancelling task..."
|
||||||
|
media_group:
|
||||||
|
info_saving_files: "Saving files..."
|
||||||
|
error_build_storage_select_keyboard_failed: "Failed to build storage selection keyboard: {{.Error}}"
|
||||||
|
info_group_found_files_select_storage: "Total {{.Count}} files, please select storage"
|
||||||
|
storage:
|
||||||
|
info_filename_prefix: "Filename: "
|
||||||
|
info_prompt_select_storage: "\nPlease select storage"
|
||||||
|
progress:
|
||||||
|
batch_start_prefix: "Starting batch download task\nTotal size: "
|
||||||
|
batch_processing_prefix: "Processing batch download task\nTotal size: "
|
||||||
|
downloading_prefix: "Downloading\nTotal size: "
|
||||||
|
processing_list_prefix: "\nProcessing:\n"
|
||||||
|
processing_none: " - None"
|
||||||
|
avg_speed_prefix: "\nAverage speed: "
|
||||||
|
current_progress_prefix: "\nCurrent progress: "
|
||||||
|
task_canceled: "Task canceled"
|
||||||
|
task_canceled_with_id: "Processing canceled: {{.TaskID}}"
|
||||||
|
task_failed_with_error: "Processing failed: {{.Error}}"
|
||||||
|
batch_done_prefix: "Completed\nFile count: "
|
||||||
|
direct_done_prefix: "Completed, file count: "
|
||||||
|
parsed_start_prefix: "Starting download from {{.Site}}\nTotal size: "
|
||||||
|
parsed_done_prefix: "Completed, resource count: "
|
||||||
|
telegraph_start_prefix: "Starting Telegraph download\nImage count: "
|
||||||
|
telegraph_progress_prefix: "Downloading\nCurrent progress: "
|
||||||
|
telegraph_done_prefix: "Completed\nImage count: "
|
||||||
|
file_start_prefix: "Starting download\nFilename: "
|
||||||
|
file_processing_prefix: "Processing download task\nFilename: "
|
||||||
|
download_failed_prefix: "Download failed\nFilename: "
|
||||||
|
download_done_prefix: "Download completed\nFilename: "
|
||||||
|
file_size_prefix: "\nFile size: "
|
||||||
|
save_path_prefix: "\nSave path: "
|
||||||
|
total_size_prefix: "\nTotal size: "
|
||||||
|
direct_start: "Starting download, total size: {{.SizeMB}} MB ({{.Count}} files)"
|
||||||
|
file_name_prefix: "Filename: "
|
||||||
|
error_prefix: "\nError: "
|
||||||
|
aria2_start: "Waiting for Aria2 to complete download (GID: {{.GID}})..."
|
||||||
|
aria2_downloading: "Aria2 downloading (GID: {{.GID}})\n"
|
||||||
|
aria2_done: "Aria2 download completed and transferred (GID: {{.GID}})\n"
|
||||||
|
ytdlp_start: "Starting yt-dlp download ({{.Count}} links)..."
|
||||||
|
ytdlp_downloading: "yt-dlp downloading ({{.Count}} links)\n"
|
||||||
|
ytdlp_done: "yt-dlp download completed and transferred ({{.Count}} files)\n"
|
||||||
|
downloaded_prefix: "\nDownloaded: "
|
||||||
|
current_speed_prefix: "\nCurrent speed: "
|
||||||
|
syncpeers:
|
||||||
|
start: "Starting to sync peers..."
|
||||||
|
done: "Peer sync completed, total {{.Count}} chats synced"
|
||||||
|
failed: "Peer sync failed: {{.Error}}"
|
||||||
|
aria2:
|
||||||
|
error_aria2_not_enabled: "Aria2 feature is not enabled in the configuration"
|
||||||
|
error_aria2_client_init_failed: "Aria2 client initialization failed: {{.Error}}"
|
||||||
|
info_adding_aria2_download: "Adding Aria2 download task..."
|
||||||
|
error_adding_aria2_download: "Failed to add Aria2 download task: {{.Error}}"
|
||||||
|
info_aria2_download_added: "Aria2 download task added, GID: {{.GID}}"
|
||||||
|
info_select_storage: "Please select storage, the task will be added to Aria2 download queue after selection"
|
||||||
@@ -1,28 +0,0 @@
|
|||||||
[initing]
|
|
||||||
other = "正在启动..."
|
|
||||||
[exiting]
|
|
||||||
other = "正在退出..."
|
|
||||||
[bye]
|
|
||||||
other = "已退出"
|
|
||||||
[InvalidCacheDir]
|
|
||||||
other = "无效的缓存文件夹: {{.Path}}"
|
|
||||||
[GetWorkdirFailed]
|
|
||||||
other = "获取工作目录失败: {{.Error}}"
|
|
||||||
[GetCacheAbsPathFailed]
|
|
||||||
other = "获取缓存绝对路径失败: {{.Error}}"
|
|
||||||
[CleaningCache]
|
|
||||||
other = "正在清理缓存文件夹: {{.Path}}"
|
|
||||||
[CleanCacheFailed]
|
|
||||||
other = "清理缓存失败: {{.Error}}"
|
|
||||||
[CreateRmTimerFailed]
|
|
||||||
other = "创建清理定时器失败, 路径: {{.Path}}, 错误: {{.Error}}"
|
|
||||||
[RemoveFileAfter]
|
|
||||||
other = "将在 {{.Duration}} 后删除文件: {{.Path}}"
|
|
||||||
[RemoveFileFailed]
|
|
||||||
other = "删除文件失败: {{.Path}}, 错误: {{.Error}}"
|
|
||||||
[LoadedStorages]
|
|
||||||
other = "已加载 {{.Count}} 个存储"
|
|
||||||
[ConfigInvalid.WorkersOrRetry]
|
|
||||||
other = "配置无效: workers 或 retry 必须大于 0, 但当前值为: workers={{.Workers}}, retry={{.Retry}}"
|
|
||||||
[ConfigInvalid.DuplicateStorageName]
|
|
||||||
other = "存储名称重复: {{.Name}}"
|
|
||||||
356
common/i18n/locale/zh-Hans.yaml
Normal file
356
common/i18n/locale/zh-Hans.yaml
Normal file
@@ -0,0 +1,356 @@
|
|||||||
|
lifetime:
|
||||||
|
initing: 正在启动
|
||||||
|
initfailed: 初始化失败
|
||||||
|
exiting: 正在退出
|
||||||
|
user_login_failed: "用户登录失败: {{.Error}}"
|
||||||
|
cleaning_cache: "正在清理缓存 {{.Path}}"
|
||||||
|
bye: 已退出
|
||||||
|
config:
|
||||||
|
err:
|
||||||
|
invalid_cache_dir: "无效的缓存目录: {{.Path}},请检查配置文件"
|
||||||
|
duplicate_storage_name: "存储名称 '{{.Name}}' 重复,请检查配置文件"
|
||||||
|
err:
|
||||||
|
get_workdir_failed: "获取工作目录失败: {{.Error}}"
|
||||||
|
get_cache_abs_path_failed: "获取缓存绝对路径失败: {{.Error}}"
|
||||||
|
clean_cache_failed: "清理缓存失败: {{.Error}}"
|
||||||
|
parser:
|
||||||
|
plugin:
|
||||||
|
load_failed: 加载解析器插件失败
|
||||||
|
loaded_dir: 解析器插件已加载
|
||||||
|
bot:
|
||||||
|
msg:
|
||||||
|
help_text_fmt: |
|
||||||
|
Save Any Bot - 转存你的 Telegram 文件
|
||||||
|
版本: %s , 提交: %s
|
||||||
|
|
||||||
|
命令:
|
||||||
|
/start - 开始使用
|
||||||
|
/help - 显示帮助
|
||||||
|
/silent - 开关静默模式
|
||||||
|
/storage - 设置默认存储位置
|
||||||
|
/save [自定义文件名] - 保存文件
|
||||||
|
/dl <链接1> <链接2> ... - 下载给定链接的文件
|
||||||
|
/dir - 管理存储目录
|
||||||
|
/rule - 管理规则
|
||||||
|
/config - 修改配置
|
||||||
|
/fnametmpl - 设置文件自定义命名模板
|
||||||
|
/parser - 管理解析器插件
|
||||||
|
/task - 管理任务队列
|
||||||
|
/watch - 监听聊天并自动保存 (UserBot)
|
||||||
|
/unwatch - 取消监听聊天 (UserBot)
|
||||||
|
/lswatch - 列出正在监听的聊天 (UserBot)
|
||||||
|
/syncpeers - 同步对话列表 (UserBot)
|
||||||
|
/update - 检查更新并升级
|
||||||
|
|
||||||
|
使用帮助: https://sabot.unv.app/usage
|
||||||
|
cmd:
|
||||||
|
start: "开始使用"
|
||||||
|
silent: "切换静默模式"
|
||||||
|
storage: "设置默认存储端"
|
||||||
|
dir: "管理存储文件夹"
|
||||||
|
rule: "管理自动存储规则"
|
||||||
|
save: "保存文件"
|
||||||
|
dl: "下载给定链接的文件"
|
||||||
|
aria2dl: "使用 Aria2 下载给定链接的文件"
|
||||||
|
ytdlp: "使用 yt-dlp 下载视频/音频"
|
||||||
|
task: "管理任务队列"
|
||||||
|
cancel: "取消任务"
|
||||||
|
watch: "监听聊天(UserBot)"
|
||||||
|
unwatch: "取消监听聊天(UserBot)"
|
||||||
|
lswatch: "列出监听的聊天(UserBot)"
|
||||||
|
syncpeers: "同步对话列表(UserBot)"
|
||||||
|
config: "修改配置"
|
||||||
|
fnametmpl: "设置文件命名模板"
|
||||||
|
help: "显示帮助"
|
||||||
|
parser: "管理解析器"
|
||||||
|
update: "检查更新"
|
||||||
|
save_help_text: |
|
||||||
|
使用方法:
|
||||||
|
|
||||||
|
1. 使用该命令回复要保存的文件, 可选文件名参数.
|
||||||
|
示例:
|
||||||
|
/save custom_file_name.mp4
|
||||||
|
|
||||||
|
2. 设置默认存储后, 发送 /save <频道ID/用户名> <消息ID范围> 来批量保存文件. 遵从存储规则, 若未匹配到任何规则则使用默认存储.
|
||||||
|
示例:
|
||||||
|
/save @acherkrau 114-514
|
||||||
|
watch_help_text: |
|
||||||
|
使用 /watch 命令监听一个聊天的消息, 并自动保存到默认存储中, 遵从存储规则.
|
||||||
|
|
||||||
|
命令语法:
|
||||||
|
/watch <chat_id> [filter]
|
||||||
|
|
||||||
|
参数:
|
||||||
|
- <chat_id>: 聊天的 ID 或用户名
|
||||||
|
- [filter]: 可选, 格式为 过滤器类型:表达式 , 所有支持类型的过滤器请查看文档
|
||||||
|
|
||||||
|
命令示例:
|
||||||
|
/watch -1002229835658 msgre:.*plana.*
|
||||||
|
|
||||||
|
这将监听 ID 为 -1002229835658 的聊天, 并转存所有包含 "plana" 的媒体消息
|
||||||
|
common:
|
||||||
|
cancel_button_text: "取消任务"
|
||||||
|
error_invalid_regex: "无效的正则表达式: {{.Error}}"
|
||||||
|
error_invalid_msg_id_range: "无效的消息ID范围: {{.Error}}"
|
||||||
|
error_invalid_id_or_username: "无效的ID或用户名: {{.Error}}"
|
||||||
|
error_get_messages_failed: "获取消息失败: {{.Error}}"
|
||||||
|
info_fetching_messages: "正在获取消息..."
|
||||||
|
error_no_messages_in_range: "没有找到指定范围内的消息"
|
||||||
|
error_no_savable_messages_in_range: "没有找到指定范围内的可保存消息"
|
||||||
|
error_build_storage_select_message_failed: "构建存储选择消息失败: {{.Error}}"
|
||||||
|
error_build_storage_select_keyboard_failed: "构建存储选择键盘失败: {{.Error}}"
|
||||||
|
info_found_files_select_storage: "找到 {{.Count}} 个文件, 请选择存储位置"
|
||||||
|
error_get_user_failed: "获取用户失败"
|
||||||
|
error_get_user_with_err_failed: "获取用户失败: {{.Error}}"
|
||||||
|
error_default_storage_not_set: "请先设置默认存储, 使用 /storage 命令"
|
||||||
|
error_no_available_storage: "无可用的存储"
|
||||||
|
error_get_storage_failed: "获取存储失败: {{.Error}}"
|
||||||
|
prompt_select_default_storage: "请选择要设为默认的存储位置"
|
||||||
|
error_data_expired: "数据已过期或无效"
|
||||||
|
error_task_add_failed: "任务添加失败: {{.Error}}"
|
||||||
|
info_task_added: "任务已添加"
|
||||||
|
info_batch_tasks_added: "已添加批量任务, 共 {{.Count}} 个文件"
|
||||||
|
error_task_create_failed: "任务创建失败: {{.Error}}"
|
||||||
|
error_get_dir_failed: "获取目录失败: {{.Error}}"
|
||||||
|
prompt_select_dir: "请选择要存储到的目录"
|
||||||
|
prompt_select_default_dir: "请选择要保存到的默认文件夹"
|
||||||
|
info_default_storage_set: "已将默认存储位置设为: {{.Name}}"
|
||||||
|
info_default_storage_with_dir_set: "已将默认存储位置设为: {{.Name}}:/{{.Dir}}"
|
||||||
|
error_get_user_info_failed: "获取用户信息失败: {{.Error}}"
|
||||||
|
error_update_user_info_failed: "更新用户信息失败: {{.Error}}"
|
||||||
|
info_silent_mode_on: "已开启静默模式"
|
||||||
|
info_silent_mode_off: "已关闭静默模式"
|
||||||
|
error_get_file_failed: "获取文件失败: {{.Error}}"
|
||||||
|
info_fetching_file_info: "正在获取文件信息..."
|
||||||
|
error_no_savable_files_found: "没有找到可保存的文件"
|
||||||
|
error_parse_telegraph_path_failed: "解析 telegraph 路径失败: {{.Error}}"
|
||||||
|
info_fetching_telegraph_page: "正在获取 telegraph 页面..."
|
||||||
|
error_get_telegraph_page_failed: "获取 telegraph 页面失败: {{.Error}}"
|
||||||
|
error_no_images_in_telegraph_page: "在 telegraph 页面中未找到图片"
|
||||||
|
error_build_dir_select_keyboard_failed: "构建目录选择键盘失败: {{.Error}}"
|
||||||
|
error_no_permission: |
|
||||||
|
您不在白名单中, 无法使用此 Bot.
|
||||||
|
您可以部署自己的实例: https://github.com/krau/SaveAny-Bot
|
||||||
|
save:
|
||||||
|
error_invalid_id_or_username: "无效的ID或用户名: {{.Error}}"
|
||||||
|
watch:
|
||||||
|
error_filter_format_invalid: "过滤器格式错误, 请使用 <过滤器类型>:<表达式>"
|
||||||
|
error_filter_type_unsupported: "不支持的过滤器类型, 请参阅文档"
|
||||||
|
error_watch_chat_failed: "监听聊天失败: {{.Error}}"
|
||||||
|
info_watch_chat_started: "已开始监听聊天: {{.Chat}}"
|
||||||
|
info_already_watching_chat: "已经在监听此聊天"
|
||||||
|
info_watch_list_empty: "当前没有监听任何聊天"
|
||||||
|
info_watch_list_header: "当前监听的聊天:\n"
|
||||||
|
info_watch_list_filter_prefix: " (过滤器: "
|
||||||
|
error_unwatch_no_chat_provided: "请提供要取消监听的聊天ID或用户名"
|
||||||
|
error_unwatch_chat_failed: "取消监听聊天失败: {{.Error}}"
|
||||||
|
info_watch_chat_stopped: "已取消监听聊天: {{.Chat}}"
|
||||||
|
tasks:
|
||||||
|
usage_cancel: "用法: /tasks cancel <task_id>"
|
||||||
|
usage: "用法: /tasks [running|queued|cancel <task_id>]"
|
||||||
|
cancel_failed: "取消任务失败: {{.Error}}"
|
||||||
|
cancel_requested_prefix: "已请求取消任务: "
|
||||||
|
running_empty: "当前没有正在运行的任务"
|
||||||
|
running_title: "当前正在运行的任务:"
|
||||||
|
total_prefix: "总数: {{.Count}}\n"
|
||||||
|
field_id: "ID: "
|
||||||
|
field_title: "名称: "
|
||||||
|
field_created: "创建时间: "
|
||||||
|
field_status: "状态: "
|
||||||
|
status_running: "运行中"
|
||||||
|
status_queued: "排队中"
|
||||||
|
status_cancel_requested: "已请求取消"
|
||||||
|
queued_empty: "当前没有排队中的任务"
|
||||||
|
queued_title: "当前排队中的任务:"
|
||||||
|
truncated_note: "...\n只显示前 10 个任务, 共 {{.Count}} 个任务"
|
||||||
|
info_added_to_queue_full: "已添加到任务队列\n文件名: {{.Filename}}\n当前排队任务数: {{.QueueLength}}"
|
||||||
|
info_added_to_queue_prefix: "已添加到任务队列\n"
|
||||||
|
info_filename_prefix: "文件名: "
|
||||||
|
info_queue_length_prefix: "\n当前排队任务数: "
|
||||||
|
rule:
|
||||||
|
error_get_user_rules_failed: "获取用户规则失败"
|
||||||
|
error_update_user_failed: "更新用户失败"
|
||||||
|
info_rule_mode_enabled: "已启用规则模式"
|
||||||
|
info_rule_mode_disabled: "已禁用规则模式"
|
||||||
|
error_invalid_rule_type: "无效的规则类型: {{.Type}}\n可用: {{.Available}}"
|
||||||
|
error_create_rule_failed: "创建规则失败"
|
||||||
|
info_create_rule_success: "创建规则成功"
|
||||||
|
prompt_provide_rule_id: "请提供规则ID"
|
||||||
|
error_invalid_rule_id: "无效的规则ID"
|
||||||
|
error_delete_rule_failed: "删除规则失败"
|
||||||
|
info_delete_rule_success: "删除规则成功"
|
||||||
|
help_usage: "使用方法: /rule <操作> <参数...>"
|
||||||
|
help_current_mode_enabled: "\n当前已启用规则模式"
|
||||||
|
help_current_mode_disabled: "\n当前已禁用规则模式"
|
||||||
|
help_available_ops: "\n\n可用操作:\n"
|
||||||
|
help_switch_suffix: " - 开关规则模式\n"
|
||||||
|
help_add_suffix: " <类型> <数据> <存储名> <路径> - 添加规则\n"
|
||||||
|
help_del_suffix: " <规则ID> - 删除规则\n"
|
||||||
|
help_existing_rules_prefix: "\n当前已添加的规则:\n"
|
||||||
|
dir:
|
||||||
|
error_get_user_dirs_failed: "获取用户文件夹失败"
|
||||||
|
error_get_user_failed: "获取用户失败"
|
||||||
|
error_create_dir_failed: "创建文件夹失败"
|
||||||
|
info_create_dir_success: "文件夹添加成功"
|
||||||
|
error_invalid_dir_id: "文件夹ID无效"
|
||||||
|
error_delete_dir_failed: "删除文件夹失败"
|
||||||
|
info_delete_dir_success: "文件夹删除成功"
|
||||||
|
error_unknown_operation: "未知操作"
|
||||||
|
help_usage: "使用方法: /dir <操作> <参数...>"
|
||||||
|
help_available_ops: "\n\n可用操作:\n"
|
||||||
|
help_add_suffix: " <存储名> <路径> - 添加路径\n"
|
||||||
|
help_del_suffix: " <路径ID> - 删除路径\n"
|
||||||
|
help_add_example_prefix: "\n添加路径示例:\n"
|
||||||
|
help_add_example_cmd: "/dir add local1 path/to/dir"
|
||||||
|
help_del_example_prefix: "\n\n删除路径示例:\n"
|
||||||
|
help_del_example_cmd: "/dir del 3"
|
||||||
|
help_existing_dirs_prefix: "\n\n当前已添加的路径:\n"
|
||||||
|
button_default: "默认"
|
||||||
|
parser:
|
||||||
|
help_text: |
|
||||||
|
用法:
|
||||||
|
|
||||||
|
/parser install <回复一个文件> - 安装解析器
|
||||||
|
plugin_not_enabled: "解析器插件功能未启用"
|
||||||
|
prompt_reply_with_parser_file: "请回复一个包含解析器文件的消息"
|
||||||
|
error_no_valid_file_in_reply: "回复的消息不包含有效的文件"
|
||||||
|
error_wrong_file_type: "错误的文件类型"
|
||||||
|
error_file_too_large: "文件过大"
|
||||||
|
error_get_filename_failed: "无法获取文件名"
|
||||||
|
error_only_js_supported: "仅支持 .js 文件作为解析器"
|
||||||
|
error_download_file_failed: "文件下载失败: {{.Error}}"
|
||||||
|
error_install_plugin_failed: "插件安装失败: {{.Error}}"
|
||||||
|
info_install_plugin_success: "插件安装成功: {{.Name}}"
|
||||||
|
parse:
|
||||||
|
info_parsing: "正在解析..."
|
||||||
|
error_parse_text_failed: "Failed to parse text: {{.Error}}"
|
||||||
|
error_build_storage_select_keyboard_failed: "Failed to build storage selection keyboard: {{.Error}}"
|
||||||
|
error_build_parsed_text_entity_failed: "Failed to build parsed text entity: {{.Error}}"
|
||||||
|
info_link_prefix: "\n链接: "
|
||||||
|
info_author_prefix: "\n作者: "
|
||||||
|
info_description_prefix: "\n描述: "
|
||||||
|
info_file_count_prefix: "\n文件数量: "
|
||||||
|
info_total_size_prefix: "\n预计总大小: "
|
||||||
|
info_prompt_select_storage: "\n请选择存储位置"
|
||||||
|
telegraph:
|
||||||
|
error_build_storage_select_keyboard_failed: "构建存储选择键盘失败: {{.Error}}"
|
||||||
|
info_title_prefix: "标题: "
|
||||||
|
info_pic_count_prefix: "\n图片数量: "
|
||||||
|
info_prompt_select_storage: "\n请选择存储位置"
|
||||||
|
update:
|
||||||
|
error_version_var_invalid: "当前处于开发版本或版本信息注入失败: {{.Error}}"
|
||||||
|
error_check_latest_failed: "检测最新版本失败: {{.Error}}"
|
||||||
|
error_no_release_found: "没有找到版本信息"
|
||||||
|
info_major_upgrade_required: "检测到大版本更新: {{.Current}} -> {{.Latest}} , 请前往 GitHub 手动下载最新版本并查看迁移指南"
|
||||||
|
info_already_latest: "当前已经是最新版本: {{.Version}}"
|
||||||
|
info_new_version_in_docker: |-
|
||||||
|
发现新版本: {{.Latest}}
|
||||||
|
当前版本: {{.Current}}
|
||||||
|
发布时间: {{.PublishedAt}}
|
||||||
|
由于您正在使用 Docker 部署, 请自行在部署平台上执行更新命令
|
||||||
|
info_new_version_prompt_upgrade: |-
|
||||||
|
发现新版本: {{.Latest}}
|
||||||
|
当前版本: {{.Current}}
|
||||||
|
|
||||||
|
文件大小: {{.SizeMB}} MB
|
||||||
|
下载链接: {{.URL}}
|
||||||
|
发布时间: {{.PublishedAt}}
|
||||||
|
|
||||||
|
升级将重启 Bot , 是否升级?
|
||||||
|
info_upgrading_with_version: "正在升级中, 当前版本: {{.Current}}"
|
||||||
|
error_upgrade_failed: "升级失败: {{.Error}}"
|
||||||
|
info_upgrade_success: "已升级至版本 {{.Version}}\n若 Bot 未自动重启请手动启动"
|
||||||
|
button_upgrade: "升级"
|
||||||
|
config:
|
||||||
|
prompt_select_option: "请选择要配置的选项"
|
||||||
|
button_filename_strategy: "文件名策略"
|
||||||
|
error_invalid_callback_data: "无效的回调数据"
|
||||||
|
error_invalid_template: "无效的模板, 请检查语法\n{{.Error}}"
|
||||||
|
info_filename_strategy_set: "已将文件名策略设置为: {{.Strategy}}"
|
||||||
|
prompt_select_filename_strategy: "请选择文件名策略, 当前策略: {{.Strategy}}"
|
||||||
|
fnametmpl_help: |-
|
||||||
|
使用该命令设置文件名模板, 示例:
|
||||||
|
/fnametmpl 图片_{{"{{.msgid}}"}}_{{"{{.msgdate}}"}}.jpg
|
||||||
|
|
||||||
|
可用变量:
|
||||||
|
- {{"{{.msgid}}"}}: 消息ID
|
||||||
|
- {{"{{.msgtags}}"}}: 消息中的标签, 将以下划线分隔输出
|
||||||
|
- {{"{{.msggen}}"}}: 根据消息生成的文件名
|
||||||
|
- {{"{{.msgdate}}"}}: 消息日期, 格式 YYYY-MM-DD_HH-MM-SS
|
||||||
|
- {{"{{.msgraw}}"}}: 消息的原始文本内容 (不经任何处理)
|
||||||
|
- {{"{{.origname}}"}}: 媒体的原始文件名 (如果有)
|
||||||
|
- {{"{{.chatid}}"}}: 消息的聊天ID
|
||||||
|
|
||||||
|
模板仅在文件名策略设置为 '自定义模板' 时生效,
|
||||||
|
且模板解析错误时会回退到默认文件名
|
||||||
|
info_template_updated: "已更新文件名模板"
|
||||||
|
info_current_template_prefix: "当前模板: {{.Template}}"
|
||||||
|
dl:
|
||||||
|
usage: "用法: /dl <链接1> <链接2> ..."
|
||||||
|
error_no_valid_links: "没有有效的链接可供下载"
|
||||||
|
info_files_select_storage: "共 {{.Count}} 个文件, 请选择存储位置"
|
||||||
|
ytdlp:
|
||||||
|
usage: "用法: /ytdlp [选项] <URL1> [URL2] ...\n示例:\n /ytdlp https://example.com/video\n /ytdlp --format best https://example.com/video\n /ytdlp --extract-audio --audio-format mp3 https://example.com/video"
|
||||||
|
error_no_valid_urls: "没有有效的 URL"
|
||||||
|
info_urls_select_storage: "共 {{.Count}} 个链接, 请选择存储位置"
|
||||||
|
info_downloading: "正在通过 yt-dlp 下载..."
|
||||||
|
error_download_failed: "yt-dlp 下载失败: {{.Error}}"
|
||||||
|
cancel:
|
||||||
|
usage: "用法: /cancel <task_id>"
|
||||||
|
error_cancel_failed: "取消任务失败: {{.Error}}"
|
||||||
|
info_cancel_requested: "已请求取消任务: {{.TaskID}}"
|
||||||
|
info_cancelling_task: "正在取消任务..."
|
||||||
|
media_group:
|
||||||
|
info_saving_files: "正在保存文件..."
|
||||||
|
error_build_storage_select_keyboard_failed: "构建存储选择键盘失败: {{.Error}}"
|
||||||
|
info_group_found_files_select_storage: "共 {{.Count}} 个文件, 请选择存储位置"
|
||||||
|
storage:
|
||||||
|
info_filename_prefix: "文件名: "
|
||||||
|
info_prompt_select_storage: "\n请选择存储位置"
|
||||||
|
progress:
|
||||||
|
batch_start_prefix: "开始执行批量下载任务\n总大小: "
|
||||||
|
batch_processing_prefix: "正在处理批量下载任务\n总大小: "
|
||||||
|
downloading_prefix: "正在下载\n总大小: "
|
||||||
|
processing_list_prefix: "\n正在处理:\n"
|
||||||
|
processing_none: " - 无"
|
||||||
|
avg_speed_prefix: "\n平均速度: "
|
||||||
|
current_progress_prefix: "\n当前进度: "
|
||||||
|
task_canceled: "任务已取消"
|
||||||
|
task_canceled_with_id: "处理已取消: {{.TaskID}}"
|
||||||
|
task_failed_with_error: "处理失败: {{.Error}}"
|
||||||
|
batch_done_prefix: "处理完成\n文件数: "
|
||||||
|
direct_done_prefix: "处理完成, 文件数量: "
|
||||||
|
parsed_start_prefix: "开始下载 {{.Site}} 的资源\n总大小: "
|
||||||
|
parsed_done_prefix: "处理完成, 资源数量: "
|
||||||
|
telegraph_start_prefix: "开始下载Telegraph\n图片数量: "
|
||||||
|
telegraph_progress_prefix: "正在下载\n当前进度: "
|
||||||
|
telegraph_done_prefix: "处理完成\n图片数量: "
|
||||||
|
file_start_prefix: "开始下载\n文件名: "
|
||||||
|
file_processing_prefix: "正在处理下载任务\n文件名: "
|
||||||
|
download_failed_prefix: "下载失败\n文件名: "
|
||||||
|
download_done_prefix: "下载完成\n文件名: "
|
||||||
|
file_size_prefix: "\n文件大小: "
|
||||||
|
save_path_prefix: "\n保存路径: "
|
||||||
|
total_size_prefix: "\n总大小: "
|
||||||
|
direct_start: "开始下载, 总大小: {{.SizeMB}} MB ({{.Count}} 个文件)"
|
||||||
|
file_name_prefix: "文件名: "
|
||||||
|
error_prefix: "\n错误: "
|
||||||
|
aria2_start: "等待 Aria2 下载完成 (GID: {{.GID}})..."
|
||||||
|
aria2_downloading: "Aria2 正在下载 (GID: {{.GID}})\n"
|
||||||
|
aria2_done: "Aria2 下载完成并已转存 (GID: {{.GID}})\n"
|
||||||
|
ytdlp_start: "开始使用 yt-dlp 下载 ({{.Count}} 个链接)..."
|
||||||
|
ytdlp_downloading: "yt-dlp 正在下载 ({{.Count}} 个链接)\n"
|
||||||
|
ytdlp_done: "yt-dlp 下载完成并已转存 ({{.Count}} 个文件)\n"
|
||||||
|
downloaded_prefix: "\n已下载: "
|
||||||
|
current_speed_prefix: "\n当前速度: "
|
||||||
|
syncpeers:
|
||||||
|
start: "正在同步对话列表..."
|
||||||
|
success: "对话列表同步完成, 共同步 {{.Count}} 个对话"
|
||||||
|
failed: "对话列表同步失败: {{.Error}}"
|
||||||
|
aria2:
|
||||||
|
error_aria2_not_enabled: "Aria2 功能未启用, 请在配置文件中启用"
|
||||||
|
error_aria2_client_init_failed: "Aria2 客户端初始化失败: {{.Error}}"
|
||||||
|
info_adding_aria2_download: "正在添加 Aria2 下载任务..."
|
||||||
|
error_adding_aria2_download: "添加 Aria2 下载任务失败: {{.Error}}"
|
||||||
|
info_aria2_download_added: "Aria2 下载任务已添加, GID: {{.GID}}"
|
||||||
|
info_select_storage: "请选择存储位置, 选择后将添加到 Aria2 下载队列"
|
||||||
@@ -1,13 +1,14 @@
|
|||||||
package tfile
|
package tdler
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"github.com/gotd/td/telegram/downloader"
|
"github.com/gotd/td/telegram/downloader"
|
||||||
"github.com/krau/SaveAny-Bot/common/utils/dlutil"
|
"github.com/krau/SaveAny-Bot/common/utils/dlutil"
|
||||||
"github.com/krau/SaveAny-Bot/config"
|
"github.com/krau/SaveAny-Bot/config"
|
||||||
"github.com/krau/SaveAny-Bot/pkg/consts/tglimit"
|
"github.com/krau/SaveAny-Bot/pkg/consts/tglimit"
|
||||||
|
"github.com/krau/SaveAny-Bot/pkg/tfile"
|
||||||
)
|
)
|
||||||
|
|
||||||
func NewDownloader(file TGFile) *downloader.Builder {
|
func NewDownloader(file tfile.TGFile) *downloader.Builder {
|
||||||
return downloader.NewDownloader().WithPartSize(tglimit.MaxPartSize).
|
return downloader.NewDownloader().WithPartSize(tglimit.MaxPartSize).
|
||||||
Download(file.Dler(), file.Location()).WithThreads(dlutil.BestThreads(file.Size(), config.Cfg.Threads))
|
Download(file.Dler(), file.Location()).WithThreads(dlutil.BestThreads(file.Size(), config.C().Threads))
|
||||||
}
|
}
|
||||||
@@ -3,6 +3,8 @@ package fsutil
|
|||||||
import (
|
import (
|
||||||
"os"
|
"os"
|
||||||
"path/filepath"
|
"path/filepath"
|
||||||
|
"strings"
|
||||||
|
"unicode"
|
||||||
|
|
||||||
"github.com/gabriel-vasile/mimetype"
|
"github.com/gabriel-vasile/mimetype"
|
||||||
)
|
)
|
||||||
@@ -55,3 +57,21 @@ func CreateFile(fp string) (*File, error) {
|
|||||||
}
|
}
|
||||||
return &File{File: file}, nil
|
return &File{File: file}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func NormalizePathname(s string) string {
|
||||||
|
specials := `\/:*?"<>|` + "\n\r\t"
|
||||||
|
var builder strings.Builder
|
||||||
|
for _, ch := range s {
|
||||||
|
if strings.ContainsRune(specials, ch) || unicode.IsControl(ch) {
|
||||||
|
builder.WriteRune('_')
|
||||||
|
} else {
|
||||||
|
builder.WriteRune(ch)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
result := strings.TrimRightFunc(builder.String(), func(r rune) bool {
|
||||||
|
return r == '.' || r == '_' || unicode.IsSpace(r)
|
||||||
|
})
|
||||||
|
|
||||||
|
return result
|
||||||
|
}
|
||||||
|
|||||||
46
common/utils/fsutil/normalize_pathname_test.go
Normal file
46
common/utils/fsutil/normalize_pathname_test.go
Normal file
@@ -0,0 +1,46 @@
|
|||||||
|
package fsutil_test
|
||||||
|
|
||||||
|
import (
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"github.com/krau/SaveAny-Bot/common/utils/fsutil"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestNormalizePathname(t *testing.T) {
|
||||||
|
tests := []struct {
|
||||||
|
input string
|
||||||
|
expected string
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
input: "hello/world?.txt ",
|
||||||
|
expected: "hello_world_.txt",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
input: "bad|name:\nfile\r.",
|
||||||
|
expected: "bad_name__file",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
input: "normal.txt",
|
||||||
|
expected: "normal.txt",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
input: "test.... ",
|
||||||
|
expected: "test",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
input: "abc<>def",
|
||||||
|
expected: "abc__def",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
input: "with\tcontrol",
|
||||||
|
expected: "with_control",
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, tc := range tests {
|
||||||
|
got := fsutil.NormalizePathname(tc.input)
|
||||||
|
if got != tc.expected {
|
||||||
|
t.Errorf("NormalizePathname(%q) = %q; want %q", tc.input, got, tc.expected)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
65
common/utils/ioutil/progress_reader.go
Normal file
65
common/utils/ioutil/progress_reader.go
Normal file
@@ -0,0 +1,65 @@
|
|||||||
|
package ioutil
|
||||||
|
|
||||||
|
import (
|
||||||
|
"io"
|
||||||
|
"sync/atomic"
|
||||||
|
)
|
||||||
|
|
||||||
|
var _ io.ReadSeeker = (*ProgressReadSeeker)(nil)
|
||||||
|
|
||||||
|
// ProgressReadSeeker wraps an io.ReadSeeker and tracks read progress
|
||||||
|
type ProgressReadSeeker struct {
|
||||||
|
reader io.ReadSeeker
|
||||||
|
total atomic.Int64
|
||||||
|
read atomic.Int64
|
||||||
|
onProgress func(read int64, total int64)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Seek implements io.ReadSeeker.
|
||||||
|
func (pr *ProgressReadSeeker) Seek(offset int64, whence int) (int64, error) {
|
||||||
|
return pr.reader.Seek(offset, whence)
|
||||||
|
}
|
||||||
|
|
||||||
|
// NewProgressReader creates a new ProgressReader
|
||||||
|
func NewProgressReader(rs io.ReadSeeker, total int64, onProgress func(read int64, total int64)) *ProgressReadSeeker {
|
||||||
|
prs := &ProgressReadSeeker{
|
||||||
|
reader: rs,
|
||||||
|
total: atomic.Int64{},
|
||||||
|
read: atomic.Int64{},
|
||||||
|
onProgress: onProgress,
|
||||||
|
}
|
||||||
|
prs.total.Store(total)
|
||||||
|
return prs
|
||||||
|
}
|
||||||
|
|
||||||
|
// Read implements io.Reader
|
||||||
|
func (pr *ProgressReadSeeker) Read(p []byte) (int, error) {
|
||||||
|
n, err := pr.reader.Read(p)
|
||||||
|
if n > 0 {
|
||||||
|
pr.read.Add(int64(n))
|
||||||
|
read := pr.read.Load()
|
||||||
|
|
||||||
|
if pr.onProgress != nil {
|
||||||
|
pr.onProgress(read, pr.total.Load())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return n, err
|
||||||
|
}
|
||||||
|
|
||||||
|
// Progress returns the current progress as a float64 between 0 and 1
|
||||||
|
func (pr *ProgressReadSeeker) Progress() float64 {
|
||||||
|
if pr.total.Load() <= 0 {
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
return float64(pr.read.Load()) / float64(pr.total.Load())
|
||||||
|
}
|
||||||
|
|
||||||
|
// Read returns the number of bytes read so far
|
||||||
|
func (pr *ProgressReadSeeker) BytesRead() int64 {
|
||||||
|
return pr.read.Load()
|
||||||
|
}
|
||||||
|
|
||||||
|
// Total returns the total number of bytes
|
||||||
|
func (pr *ProgressReadSeeker) Total() int64 {
|
||||||
|
return pr.total.Load()
|
||||||
|
}
|
||||||
@@ -1,6 +1,8 @@
|
|||||||
package ioutil
|
package ioutil
|
||||||
|
|
||||||
import "io"
|
import (
|
||||||
|
"io"
|
||||||
|
)
|
||||||
|
|
||||||
type ProgressWriterAt struct {
|
type ProgressWriterAt struct {
|
||||||
wrAt io.WriterAt
|
wrAt io.WriterAt
|
||||||
|
|||||||
@@ -1,15 +1,78 @@
|
|||||||
package netutil
|
package netutil
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"context"
|
||||||
|
"fmt"
|
||||||
|
"net"
|
||||||
|
"net/http"
|
||||||
"net/url"
|
"net/url"
|
||||||
|
"sync"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/charmbracelet/log"
|
||||||
|
"github.com/krau/SaveAny-Bot/config"
|
||||||
"golang.org/x/net/proxy"
|
"golang.org/x/net/proxy"
|
||||||
)
|
)
|
||||||
|
|
||||||
func NewProxyDialer(proxyUrl string) (proxy.Dialer, error) {
|
func NewProxyHTTPClient(proxyUrl string) (*http.Client, error) {
|
||||||
url, err := url.Parse(proxyUrl)
|
if proxyUrl == "" {
|
||||||
|
return http.DefaultClient, nil
|
||||||
|
}
|
||||||
|
transport, err := NewProxyTransport(proxyUrl)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
return proxy.FromURL(url, proxy.Direct)
|
return &http.Client{
|
||||||
|
Transport: transport,
|
||||||
|
}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
var (
|
||||||
|
defaultProxyHttpClient *http.Client
|
||||||
|
onceLoadDefaultProxyHttpClient sync.Once
|
||||||
|
)
|
||||||
|
|
||||||
|
func DefaultParserHTTPClient() *http.Client {
|
||||||
|
onceLoadDefaultProxyHttpClient.Do(func() {
|
||||||
|
client, err := NewProxyHTTPClient(config.C().Parser.Proxy)
|
||||||
|
if err != nil {
|
||||||
|
log.Warn("Failed to create default proxy HTTP client, using http.DefaultClient", "error", err)
|
||||||
|
defaultProxyHttpClient = http.DefaultClient
|
||||||
|
} else {
|
||||||
|
defaultProxyHttpClient = client
|
||||||
|
}
|
||||||
|
})
|
||||||
|
return defaultProxyHttpClient
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewProxyTransport(proxyStr string) (*http.Transport, error) {
|
||||||
|
proxyURL, err := url.Parse(proxyStr)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
transport := &http.Transport{
|
||||||
|
ForceAttemptHTTP2: true,
|
||||||
|
MaxIdleConns: 100,
|
||||||
|
IdleConnTimeout: 90 * time.Second,
|
||||||
|
TLSHandshakeTimeout: 10 * time.Second,
|
||||||
|
ExpectContinueTimeout: 1 * time.Second,
|
||||||
|
}
|
||||||
|
switch proxyURL.Scheme {
|
||||||
|
case "http", "https":
|
||||||
|
transport.Proxy = http.ProxyURL(proxyURL)
|
||||||
|
|
||||||
|
case "socks5", "socks5h":
|
||||||
|
dialer, err := proxy.FromURL(proxyURL, proxy.Direct)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
transport.DialContext = func(ctx context.Context, network, addr string) (net.Conn, error) {
|
||||||
|
return dialer.(proxy.ContextDialer).DialContext(ctx, network, addr)
|
||||||
|
}
|
||||||
|
|
||||||
|
default:
|
||||||
|
return nil, fmt.Errorf("unsupported proxy type: %s", proxyURL.Scheme)
|
||||||
|
}
|
||||||
|
|
||||||
|
return transport, nil
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -48,3 +48,46 @@ func ParseIntStrRange(input string, sep string) (int64, int64, error) {
|
|||||||
}
|
}
|
||||||
return min, max, nil
|
return min, max, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func ParseArgsRespectQuotes(input string) []string {
|
||||||
|
var args []string
|
||||||
|
var current strings.Builder
|
||||||
|
inQuotes := false
|
||||||
|
escaped := false
|
||||||
|
|
||||||
|
for _, r := range input {
|
||||||
|
switch {
|
||||||
|
case escaped:
|
||||||
|
if r == '"' || r == '\\' {
|
||||||
|
current.WriteRune(r)
|
||||||
|
} else {
|
||||||
|
current.WriteRune('\\')
|
||||||
|
current.WriteRune(r)
|
||||||
|
}
|
||||||
|
escaped = false
|
||||||
|
|
||||||
|
case r == '\\':
|
||||||
|
escaped = true
|
||||||
|
|
||||||
|
case r == '"':
|
||||||
|
inQuotes = !inQuotes
|
||||||
|
|
||||||
|
case r == ' ' || r == '\t':
|
||||||
|
if inQuotes {
|
||||||
|
current.WriteRune(r)
|
||||||
|
} else if current.Len() > 0 {
|
||||||
|
args = append(args, current.String())
|
||||||
|
current.Reset()
|
||||||
|
}
|
||||||
|
|
||||||
|
default:
|
||||||
|
current.WriteRune(r)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if current.Len() > 0 {
|
||||||
|
args = append(args, current.String())
|
||||||
|
}
|
||||||
|
|
||||||
|
return args
|
||||||
|
}
|
||||||
|
|||||||
148
common/utils/strutil/string_test.go
Normal file
148
common/utils/strutil/string_test.go
Normal file
@@ -0,0 +1,148 @@
|
|||||||
|
package strutil_test
|
||||||
|
|
||||||
|
import (
|
||||||
|
"reflect"
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"github.com/krau/SaveAny-Bot/common/utils/strutil"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestExtractTagsFromText(t *testing.T) {
|
||||||
|
tests := []struct {
|
||||||
|
text string
|
||||||
|
expected []string
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
text: `初音ミクHappy 16th Birthday -Dear Creators-
|
||||||
|
✨エンドイラスト公開!✨
|
||||||
|
https://piapro.net/miku16thbd/
|
||||||
|
#初音ミク #miku16th`,
|
||||||
|
expected: []string{"初音ミク", "miku16th"},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
text: `ひっつきむし
|
||||||
|
#創作百合`,
|
||||||
|
expected: []string{"創作百合"},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
text: `#創作百合 #原创`,
|
||||||
|
expected: []string{"創作百合", "原创"},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
text: `プラニャ #ブルアカ`,
|
||||||
|
expected: []string{"ブルアカ"},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
text: `原神是一款#开放世界#冒险游戏,由中国著名游戏公司#miHoYo开发。`,
|
||||||
|
expected: []string{},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, test := range tests {
|
||||||
|
result := strutil.ExtractTagsFromText(test.text)
|
||||||
|
if !reflect.DeepEqual(result, test.expected) {
|
||||||
|
t.Fatalf("ExtractTagsFromText(%s) = %v, expected %v", test.text, result, test.expected)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestParseIntStrRange(t *testing.T) {
|
||||||
|
tests := []struct {
|
||||||
|
name string
|
||||||
|
input string
|
||||||
|
sep string
|
||||||
|
wantMin int64
|
||||||
|
wantMax int64
|
||||||
|
wantErr bool
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
name: "normal range",
|
||||||
|
input: "10-20",
|
||||||
|
sep: "-",
|
||||||
|
wantMin: 10,
|
||||||
|
wantMax: 20,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "reverse order",
|
||||||
|
input: "30 - 10",
|
||||||
|
sep: "-",
|
||||||
|
wantMin: 10,
|
||||||
|
wantMax: 30,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "invalid format",
|
||||||
|
input: "10",
|
||||||
|
sep: "-",
|
||||||
|
wantErr: true,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "invalid number",
|
||||||
|
input: "a-b",
|
||||||
|
sep: "-",
|
||||||
|
wantErr: true,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, tt := range tests {
|
||||||
|
t.Run(tt.name, func(t *testing.T) {
|
||||||
|
min, max, err := strutil.ParseIntStrRange(tt.input, tt.sep)
|
||||||
|
if (err != nil) != tt.wantErr {
|
||||||
|
t.Errorf("ParseIntStrRange(%q) error = %v, wantErr %v", tt.input, err, tt.wantErr)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if !tt.wantErr {
|
||||||
|
if min != tt.wantMin || max != tt.wantMax {
|
||||||
|
t.Errorf("ParseIntStrRange(%q) = (%d, %d), want (%d, %d)", tt.input, min, max, tt.wantMin, tt.wantMax)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestParseArgsRespectQuotes(t *testing.T) {
|
||||||
|
tests := []struct {
|
||||||
|
name string
|
||||||
|
input string
|
||||||
|
want []string
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
name: "simple split",
|
||||||
|
input: `/rule add FILENAME-REGEX (?i)\.(mp4|mkv)$ "我的 Alist" /视频`,
|
||||||
|
want: []string{"/rule", "add", "FILENAME-REGEX", "(?i)\\.(mp4|mkv)$", "我的 Alist", "/视频"},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "escaped quotes",
|
||||||
|
input: `/rule add "My \"Awesome\" Folder"`,
|
||||||
|
want: []string{"/rule", "add", `My "Awesome" Folder`},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "escaped backslash",
|
||||||
|
input: `/cmd "C:\\Users\\Admin" test`,
|
||||||
|
want: []string{"/cmd", `C:\Users\Admin`, "test"},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "multiple quoted parts",
|
||||||
|
input: `"Hello World" "你好 世界"`,
|
||||||
|
want: []string{"Hello World", "你好 世界"},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "unquoted words",
|
||||||
|
input: "a b c",
|
||||||
|
want: []string{"a", "b", "c"},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "mixed quotes and plain",
|
||||||
|
input: `cmd "quoted arg" plain`,
|
||||||
|
want: []string{"cmd", "quoted arg", "plain"},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, tt := range tests {
|
||||||
|
t.Run(tt.name, func(t *testing.T) {
|
||||||
|
got := strutil.ParseArgsRespectQuotes(tt.input)
|
||||||
|
if !reflect.DeepEqual(got, tt.want) {
|
||||||
|
t.Errorf("ParseArgsRespectQuotes(%q) = %#v, want %#v", tt.input, got, tt.want)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -4,17 +4,21 @@ import (
|
|||||||
"fmt"
|
"fmt"
|
||||||
"strconv"
|
"strconv"
|
||||||
"strings"
|
"strings"
|
||||||
|
"unicode"
|
||||||
|
"unicode/utf16"
|
||||||
|
|
||||||
"github.com/celestix/gotgproto/ext"
|
"github.com/celestix/gotgproto/ext"
|
||||||
"github.com/duke-git/lancet/v2/maputil"
|
"github.com/duke-git/lancet/v2/maputil"
|
||||||
|
|
||||||
"github.com/duke-git/lancet/v2/mathutil"
|
"github.com/duke-git/lancet/v2/mathutil"
|
||||||
"github.com/duke-git/lancet/v2/slice"
|
"github.com/duke-git/lancet/v2/slice"
|
||||||
lcstrutil "github.com/duke-git/lancet/v2/strutil"
|
lcstrutil "github.com/duke-git/lancet/v2/strutil"
|
||||||
"github.com/duke-git/lancet/v2/validator"
|
"github.com/duke-git/lancet/v2/validator"
|
||||||
"github.com/gabriel-vasile/mimetype"
|
"github.com/gabriel-vasile/mimetype"
|
||||||
|
"github.com/gotd/td/constant"
|
||||||
"github.com/gotd/td/tg"
|
"github.com/gotd/td/tg"
|
||||||
"github.com/krau/SaveAny-Bot/common/cache"
|
"github.com/krau/SaveAny-Bot/common/cache"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/i18n"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/i18n/i18nk"
|
||||||
"github.com/krau/SaveAny-Bot/common/utils/strutil"
|
"github.com/krau/SaveAny-Bot/common/utils/strutil"
|
||||||
"github.com/rs/xid"
|
"github.com/rs/xid"
|
||||||
)
|
)
|
||||||
@@ -61,16 +65,12 @@ func GenFileNameFromMessage(message tg.Message) string {
|
|||||||
return fmt.Sprintf("%s_%s", tagStr, strconv.Itoa(message.GetID()))
|
return fmt.Sprintf("%s_%s", tagStr, strconv.Itoa(message.GetID()))
|
||||||
}
|
}
|
||||||
text = lcstrutil.Substring(strings.Map(func(r rune) rune {
|
text = lcstrutil.Substring(strings.Map(func(r rune) rune {
|
||||||
if r < 0x20 || r == 0x7F {
|
|
||||||
return '_'
|
|
||||||
}
|
|
||||||
switch r {
|
switch r {
|
||||||
// invalid characters
|
|
||||||
case '/', '\\',
|
case '/', '\\',
|
||||||
':', '*', '?', '"', '<', '>', '|':
|
':', '*', '?', '"', '<', '>', '|':
|
||||||
return '_'
|
return '_'
|
||||||
// empty
|
}
|
||||||
case ' ', '\t', '\r', '\n':
|
if unicode.IsControl(r) || unicode.IsSpace(r) {
|
||||||
return '_'
|
return '_'
|
||||||
}
|
}
|
||||||
if validator.IsPrintable(string(r)) {
|
if validator.IsPrintable(string(r)) {
|
||||||
@@ -98,7 +98,7 @@ func GenFileNameFromMessage(message tg.Message) string {
|
|||||||
|
|
||||||
func BuildCancelButton(taskID string) tg.KeyboardButtonClass {
|
func BuildCancelButton(taskID string) tg.KeyboardButtonClass {
|
||||||
return &tg.KeyboardButtonCallback{
|
return &tg.KeyboardButtonCallback{
|
||||||
Text: "取消任务",
|
Text: i18n.T(i18nk.BotMsgCommonCancelButtonText, nil),
|
||||||
Data: fmt.Appendf(nil, "cancel %s", taskID),
|
Data: fmt.Appendf(nil, "cancel %s", taskID),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -113,7 +113,31 @@ func InputMessageClassSliceFromInt(ids []int) []tg.InputMessageClass {
|
|||||||
return result
|
return result
|
||||||
}
|
}
|
||||||
|
|
||||||
func GetMessagesRange(ctx *ext.Context, chatID int64, minId, maxId int) ([]*tg.Message, error) {
|
func GetMessagesRange(ctx *ext.Context, chatID int64, minId, maxId int) (msg []*tg.Message, err error) {
|
||||||
|
if msg, err = getMessagesRange(ctx, chatID, minId, maxId); err == nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
in := constant.TDLibPeerID(chatID)
|
||||||
|
plain := in.ToPlain()
|
||||||
|
var channel constant.TDLibPeerID
|
||||||
|
channel.Channel(plain)
|
||||||
|
if msg, err = getMessagesRange(ctx, int64(channel), minId, maxId); err == nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
var userID constant.TDLibPeerID
|
||||||
|
userID.User(plain)
|
||||||
|
if msg, err = getMessagesRange(ctx, int64(userID), minId, maxId); err == nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
var chat constant.TDLibPeerID
|
||||||
|
chat.Chat(plain)
|
||||||
|
if msg, err = getMessagesRange(ctx, int64(chat), minId, maxId); err == nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
return nil, fmt.Errorf("failed to get messages range for chat %d: %w", chatID, err)
|
||||||
|
}
|
||||||
|
|
||||||
|
func getMessagesRange(ctx *ext.Context, chatID int64, minId, maxId int) ([]*tg.Message, error) {
|
||||||
if minId > maxId {
|
if minId > maxId {
|
||||||
return nil, fmt.Errorf("minId (%d) cannot be greater than maxId (%d)", minId, maxId)
|
return nil, fmt.Errorf("minId (%d) cannot be greater than maxId (%d)", minId, maxId)
|
||||||
}
|
}
|
||||||
@@ -169,97 +193,98 @@ func GetMessagesRange(ctx *ext.Context, chatID int64, minId, maxId int) ([]*tg.M
|
|||||||
return result, nil
|
return result, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
type MessageItem struct {
|
// [TODO]
|
||||||
Message *tg.Message
|
// type MessageItem struct {
|
||||||
Error error
|
// Message *tg.Message
|
||||||
}
|
// Error error
|
||||||
|
// }
|
||||||
|
|
||||||
func IterMessages(ctx *ext.Context, chatID int64, minId, maxId int) (<-chan MessageItem, error) {
|
// func IterMessages(ctx *ext.Context, chatID int64, minId, maxId int) (<-chan MessageItem, error) {
|
||||||
total := maxId - minId + 1
|
// total := maxId - minId + 1
|
||||||
ch := make(chan MessageItem, 100)
|
// ch := make(chan MessageItem, 100)
|
||||||
|
|
||||||
go func() {
|
// go func() {
|
||||||
defer close(ch)
|
// defer close(ch)
|
||||||
if !ctx.Self.Bot {
|
// if !ctx.Self.Bot {
|
||||||
perr := ctx.PeerStorage.GetInputPeerById(chatID)
|
// perr := ctx.PeerStorage.GetInputPeerById(chatID)
|
||||||
if perr == nil || perr.(*tg.InputPeerEmpty) != nil {
|
// if perr == nil || perr.(*tg.InputPeerEmpty) != nil {
|
||||||
ch <- MessageItem{
|
// ch <- MessageItem{
|
||||||
Error: fmt.Errorf("peer not found: %d", chatID),
|
// Error: fmt.Errorf("peer not found: %d", chatID),
|
||||||
}
|
// }
|
||||||
return
|
// return
|
||||||
}
|
// }
|
||||||
|
|
||||||
for i := 0; i < total; i += 100 {
|
// for i := 0; i < total; i += 100 {
|
||||||
start := minId + i
|
// start := minId + i
|
||||||
end := min(start+100, maxId)
|
// end := min(start+100, maxId)
|
||||||
msgs, err := ctx.Raw.MessagesGetHistory(ctx, &tg.MessagesGetHistoryRequest{
|
// msgs, err := ctx.Raw.MessagesGetHistory(ctx, &tg.MessagesGetHistoryRequest{
|
||||||
Peer: perr,
|
// Peer: perr,
|
||||||
OffsetID: start,
|
// OffsetID: start,
|
||||||
AddOffset: start - end,
|
// AddOffset: start - end,
|
||||||
Limit: 100,
|
// Limit: 100,
|
||||||
})
|
// })
|
||||||
if err != nil {
|
// if err != nil {
|
||||||
ch <- MessageItem{
|
// ch <- MessageItem{
|
||||||
Error: fmt.Errorf("failed to get messages: %w", err),
|
// Error: fmt.Errorf("failed to get messages: %w", err),
|
||||||
}
|
// }
|
||||||
return
|
// return
|
||||||
}
|
// }
|
||||||
var msgClass []tg.MessageClass
|
// var msgClass []tg.MessageClass
|
||||||
switch msgsv := msgs.(type) {
|
// switch msgsv := msgs.(type) {
|
||||||
case *tg.MessagesMessages:
|
// case *tg.MessagesMessages:
|
||||||
msgClass = msgsv.GetMessages()
|
// msgClass = msgsv.GetMessages()
|
||||||
case *tg.MessagesMessagesSlice:
|
// case *tg.MessagesMessagesSlice:
|
||||||
msgClass = msgsv.GetMessages()
|
// msgClass = msgsv.GetMessages()
|
||||||
case *tg.MessagesChannelMessages:
|
// case *tg.MessagesChannelMessages:
|
||||||
msgClass = msgsv.GetMessages()
|
// msgClass = msgsv.GetMessages()
|
||||||
default:
|
// default:
|
||||||
ch <- MessageItem{
|
// ch <- MessageItem{
|
||||||
Error: fmt.Errorf("unsupported message type: %T", msgsv),
|
// Error: fmt.Errorf("unsupported message type: %T", msgsv),
|
||||||
}
|
// }
|
||||||
continue
|
// continue
|
||||||
}
|
// }
|
||||||
for _, msg := range msgClass {
|
// for _, msg := range msgClass {
|
||||||
msg, ok := msg.AsNotEmpty()
|
// msg, ok := msg.AsNotEmpty()
|
||||||
if !ok {
|
// if !ok {
|
||||||
continue
|
// continue
|
||||||
}
|
// }
|
||||||
switch msg := msg.(type) {
|
// switch msg := msg.(type) {
|
||||||
case *tg.Message:
|
// case *tg.Message:
|
||||||
key := fmt.Sprintf("tgmsg:%d:%d:%d", ctx.Self.ID, chatID, msg.GetID())
|
// key := fmt.Sprintf("tgmsg:%d:%d:%d", ctx.Self.ID, chatID, msg.GetID())
|
||||||
cache.Set(key, msg)
|
// cache.Set(key, msg)
|
||||||
ch <- MessageItem{
|
// ch <- MessageItem{
|
||||||
Message: msg,
|
// Message: msg,
|
||||||
}
|
// }
|
||||||
}
|
// }
|
||||||
}
|
// }
|
||||||
}
|
// }
|
||||||
} else {
|
// } else {
|
||||||
for i := 0; i < total; i += 100 {
|
// for i := 0; i < total; i += 100 {
|
||||||
start := minId + i
|
// start := minId + i
|
||||||
end := min(start+100, maxId)
|
// end := min(start+100, maxId)
|
||||||
msgs, err := GetMessagesRange(ctx, chatID, start, end)
|
// msgs, err := GetMessagesRange(ctx, chatID, start, end)
|
||||||
if err != nil {
|
// if err != nil {
|
||||||
ch <- MessageItem{
|
// ch <- MessageItem{
|
||||||
Error: fmt.Errorf("failed to get messages: %w", err),
|
// Error: fmt.Errorf("failed to get messages: %w", err),
|
||||||
}
|
// }
|
||||||
return
|
// return
|
||||||
}
|
// }
|
||||||
for _, msg := range msgs {
|
// for _, msg := range msgs {
|
||||||
if msg == nil {
|
// if msg == nil {
|
||||||
continue
|
// continue
|
||||||
}
|
// }
|
||||||
ch <- MessageItem{
|
// ch <- MessageItem{
|
||||||
Message: msg,
|
// Message: msg,
|
||||||
}
|
// }
|
||||||
}
|
// }
|
||||||
}
|
// }
|
||||||
}
|
// }
|
||||||
}()
|
// }()
|
||||||
|
|
||||||
return ch, nil
|
// return ch, nil
|
||||||
}
|
// }
|
||||||
|
|
||||||
func GetMessageByID(ctx *ext.Context, chatID int64, msgID int) (*tg.Message, error) {
|
func getMessageByID(ctx *ext.Context, chatID int64, msgID int) (*tg.Message, error) {
|
||||||
key := fmt.Sprintf("tgmsg:%d:%d:%d", ctx.Self.ID, chatID, msgID)
|
key := fmt.Sprintf("tgmsg:%d:%d:%d", ctx.Self.ID, chatID, msgID)
|
||||||
if msg, ok := cache.Get[*tg.Message](key); ok {
|
if msg, ok := cache.Get[*tg.Message](key); ok {
|
||||||
return msg, nil
|
return msg, nil
|
||||||
@@ -282,6 +307,33 @@ func GetMessageByID(ctx *ext.Context, chatID int64, msgID int) (*tg.Message, err
|
|||||||
return tgm, nil
|
return tgm, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// f**k gotgproto's breaking changes
|
||||||
|
func GetMessageByID(ctx *ext.Context, chatID int64, msgID int) (*tg.Message, error) {
|
||||||
|
// we don't know what the input chatID is bot api style(e.g. channel with -100 prefix) or plain tdlib style(no any prefix and every id is positive)
|
||||||
|
if msg, err := getMessageByID(ctx, chatID, msgID); err == nil {
|
||||||
|
return msg, nil
|
||||||
|
}
|
||||||
|
in := constant.TDLibPeerID(chatID)
|
||||||
|
plain := in.ToPlain()
|
||||||
|
var channel constant.TDLibPeerID
|
||||||
|
channel.Channel(plain)
|
||||||
|
if msg, err := getMessageByID(ctx, int64(channel), msgID); err == nil {
|
||||||
|
return msg, nil
|
||||||
|
}
|
||||||
|
var chat constant.TDLibPeerID
|
||||||
|
chat.Chat(plain)
|
||||||
|
if msg, err := getMessageByID(ctx, int64(chat), msgID); err == nil {
|
||||||
|
return msg, nil
|
||||||
|
}
|
||||||
|
var userID constant.TDLibPeerID
|
||||||
|
userID.User(plain)
|
||||||
|
if msg, err := getMessageByID(ctx, int64(userID), msgID); err == nil {
|
||||||
|
return msg, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil, fmt.Errorf("failed to get message by ID: chatID=%d, msgID=%d", chatID, msgID)
|
||||||
|
}
|
||||||
|
|
||||||
func GetGroupedMessages(ctx *ext.Context, chatID int64, msg *tg.Message) ([]*tg.Message, error) {
|
func GetGroupedMessages(ctx *ext.Context, chatID int64, msg *tg.Message) ([]*tg.Message, error) {
|
||||||
groupID, isGroup := msg.GetGroupedID()
|
groupID, isGroup := msg.GetGroupedID()
|
||||||
if !isGroup || groupID == 0 {
|
if !isGroup || groupID == 0 {
|
||||||
@@ -295,7 +347,7 @@ func GetGroupedMessages(ctx *ext.Context, chatID int64, msg *tg.Message) ([]*tg.
|
|||||||
}
|
}
|
||||||
msgs, err := GetMessagesRange(ctx, chatID, minID, maxID)
|
msgs, err := GetMessagesRange(ctx, chatID, minID, maxID)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, fmt.Errorf("failed to get grouped messages: %w", err)
|
return nil, err
|
||||||
}
|
}
|
||||||
groupedMessages := make([]*tg.Message, 0, len(msgs))
|
groupedMessages := make([]*tg.Message, 0, len(msgs))
|
||||||
for _, m := range msgs {
|
for _, m := range msgs {
|
||||||
@@ -309,3 +361,49 @@ func GetGroupedMessages(ctx *ext.Context, chatID int64, msg *tg.Message) ([]*tg.
|
|||||||
}
|
}
|
||||||
return groupedMessages, nil
|
return groupedMessages, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func ExtractMessageEntityUrls(msg *tg.Message) []string {
|
||||||
|
if len(msg.Entities) == 0 {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
msgText := msg.GetMessage()
|
||||||
|
if msgText == "" {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
runes := []rune(msgText)
|
||||||
|
utf16Codes := utf16.Encode(runes)
|
||||||
|
|
||||||
|
var urls []string
|
||||||
|
for _, entity := range msg.Entities {
|
||||||
|
switch ent := entity.(type) {
|
||||||
|
case *tg.MessageEntityTextURL:
|
||||||
|
urls = append(urls, ent.GetURL())
|
||||||
|
case *tg.MessageEntityURL:
|
||||||
|
start := ent.Offset
|
||||||
|
end := ent.Offset + ent.Length
|
||||||
|
if start < 0 || end > len(utf16Codes) {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
subRunes := utf16.Decode(utf16Codes[start:end])
|
||||||
|
urls = append(urls, string(subRunes))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return urls
|
||||||
|
}
|
||||||
|
|
||||||
|
func ExtractMessageEntityUrlsText(msg *tg.Message) string {
|
||||||
|
if msg == nil {
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
urls := ExtractMessageEntityUrls(msg)
|
||||||
|
if len(urls) == 0 {
|
||||||
|
return msg.GetMessage()
|
||||||
|
}
|
||||||
|
var sb strings.Builder
|
||||||
|
for _, url := range urls {
|
||||||
|
sb.WriteString(url)
|
||||||
|
sb.WriteString(" ")
|
||||||
|
}
|
||||||
|
return sb.String()
|
||||||
|
}
|
||||||
|
|||||||
131
common/utils/tgutil/net.go
Normal file
131
common/utils/tgutil/net.go
Normal file
@@ -0,0 +1,131 @@
|
|||||||
|
package tgutil
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bufio"
|
||||||
|
"context"
|
||||||
|
"encoding/base64"
|
||||||
|
"fmt"
|
||||||
|
"net"
|
||||||
|
"net/http"
|
||||||
|
"net/url"
|
||||||
|
|
||||||
|
"github.com/gotd/td/telegram/dcs"
|
||||||
|
"github.com/krau/SaveAny-Bot/config"
|
||||||
|
"golang.org/x/net/proxy"
|
||||||
|
)
|
||||||
|
|
||||||
|
// httpProxyDialer implements proxy.ContextDialer for HTTP CONNECT proxies
|
||||||
|
type httpProxyDialer struct {
|
||||||
|
proxyURL *url.URL
|
||||||
|
forward proxy.Dialer
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *httpProxyDialer) Dial(network, addr string) (net.Conn, error) {
|
||||||
|
return d.DialContext(context.Background(), network, addr)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *httpProxyDialer) DialContext(ctx context.Context, network, addr string) (net.Conn, error) {
|
||||||
|
proxyAddr := d.proxyURL.Host
|
||||||
|
if d.proxyURL.Port() == "" {
|
||||||
|
if d.proxyURL.Scheme == "https" {
|
||||||
|
proxyAddr = net.JoinHostPort(d.proxyURL.Hostname(), "443")
|
||||||
|
} else {
|
||||||
|
proxyAddr = net.JoinHostPort(d.proxyURL.Hostname(), "80")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
var conn net.Conn
|
||||||
|
var err error
|
||||||
|
if ctxDialer, ok := d.forward.(proxy.ContextDialer); ok {
|
||||||
|
conn, err = ctxDialer.DialContext(ctx, "tcp", proxyAddr)
|
||||||
|
} else {
|
||||||
|
conn, err = d.forward.Dial("tcp", proxyAddr)
|
||||||
|
}
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("failed to connect to proxy: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Send CONNECT request
|
||||||
|
connectReq := &http.Request{
|
||||||
|
Method: "CONNECT",
|
||||||
|
URL: &url.URL{Opaque: addr},
|
||||||
|
Host: addr,
|
||||||
|
Header: make(http.Header),
|
||||||
|
}
|
||||||
|
|
||||||
|
// Add proxy authentication if provided
|
||||||
|
if d.proxyURL.User != nil {
|
||||||
|
username := d.proxyURL.User.Username()
|
||||||
|
password, _ := d.proxyURL.User.Password()
|
||||||
|
auth := base64.StdEncoding.EncodeToString([]byte(username + ":" + password))
|
||||||
|
connectReq.Header.Set("Proxy-Authorization", "Basic "+auth)
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := connectReq.Write(conn); err != nil {
|
||||||
|
conn.Close()
|
||||||
|
return nil, fmt.Errorf("failed to write CONNECT request: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Read response
|
||||||
|
br := bufio.NewReader(conn)
|
||||||
|
resp, err := http.ReadResponse(br, connectReq)
|
||||||
|
if err != nil {
|
||||||
|
conn.Close()
|
||||||
|
return nil, fmt.Errorf("failed to read CONNECT response: %w", err)
|
||||||
|
}
|
||||||
|
defer resp.Body.Close()
|
||||||
|
|
||||||
|
if resp.StatusCode != http.StatusOK {
|
||||||
|
conn.Close()
|
||||||
|
return nil, fmt.Errorf("proxy CONNECT failed with status: %s", resp.Status)
|
||||||
|
}
|
||||||
|
|
||||||
|
return conn, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func newProxyDialer(proxyUrl string) (proxy.ContextDialer, error) {
|
||||||
|
parsedURL, err := url.Parse(proxyUrl)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
switch parsedURL.Scheme {
|
||||||
|
case "http", "https":
|
||||||
|
return &httpProxyDialer{
|
||||||
|
proxyURL: parsedURL,
|
||||||
|
forward: proxy.Direct,
|
||||||
|
}, nil
|
||||||
|
case "socks5", "socks5h":
|
||||||
|
dialer, err := proxy.FromURL(parsedURL, proxy.Direct)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return dialer.(proxy.ContextDialer), nil
|
||||||
|
default:
|
||||||
|
return nil, fmt.Errorf("unsupported proxy scheme: %s", parsedURL.Scheme)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewConfigProxyResolver() (dcs.Resolver, error) {
|
||||||
|
resolver := dcs.DefaultResolver()
|
||||||
|
if config.C().Proxy != "" {
|
||||||
|
// global proxy, which has lower priority
|
||||||
|
dialer, err := newProxyDialer(config.C().Proxy)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
resolver = dcs.Plain(dcs.PlainOptions{
|
||||||
|
Dial: dialer.DialContext,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
if config.C().Telegram.Proxy.Enable && config.C().Telegram.Proxy.URL != "" {
|
||||||
|
dialer, err := newProxyDialer(config.C().Telegram.Proxy.URL)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
resolver = dcs.Plain(dcs.PlainOptions{
|
||||||
|
Dial: dialer.DialContext,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
return resolver, nil
|
||||||
|
}
|
||||||
16
common/utils/tgutil/peer.go
Normal file
16
common/utils/tgutil/peer.go
Normal file
@@ -0,0 +1,16 @@
|
|||||||
|
package tgutil
|
||||||
|
|
||||||
|
import "github.com/gotd/td/tg"
|
||||||
|
|
||||||
|
func ChatIdFromPeer(peer tg.PeerClass) int64 {
|
||||||
|
switch peer := peer.(type) {
|
||||||
|
case *tg.PeerChannel:
|
||||||
|
return peer.ChannelID
|
||||||
|
case *tg.PeerUser:
|
||||||
|
return peer.UserID
|
||||||
|
case *tg.PeerChat:
|
||||||
|
return peer.ChatID
|
||||||
|
default:
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -86,7 +86,7 @@ func ParseMessageLink(ctx *ext.Context, link string) (int64, int, error) {
|
|||||||
return chatID, msgID, nil
|
return chatID, msgID, nil
|
||||||
case 3:
|
case 3:
|
||||||
// https://t.me/c/123456789/123
|
// https://t.me/c/123456789/123
|
||||||
// https://t.me/acherkrau/123/456 , 456: message thread ID
|
// https://t.me/acherkrau/123/456 , 123: topic id
|
||||||
chatPart, msgPart := paths[1], paths[2]
|
chatPart, msgPart := paths[1], paths[2]
|
||||||
if paths[0] != "c" {
|
if paths[0] != "c" {
|
||||||
chatPart = paths[0]
|
chatPart = paths[0]
|
||||||
|
|||||||
@@ -2,28 +2,38 @@ package tphutil
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"encoding/json"
|
"encoding/json"
|
||||||
|
"strings"
|
||||||
|
"sync"
|
||||||
|
|
||||||
"github.com/krau/SaveAny-Bot/config"
|
"github.com/krau/SaveAny-Bot/config"
|
||||||
"github.com/krau/SaveAny-Bot/pkg/telegraph"
|
"github.com/krau/SaveAny-Bot/pkg/telegraph"
|
||||||
)
|
)
|
||||||
|
|
||||||
var tphClient *telegraph.Client
|
var (
|
||||||
|
tphClient *telegraph.Client
|
||||||
|
once sync.Once
|
||||||
|
)
|
||||||
|
|
||||||
func DefaultClient() *telegraph.Client {
|
func DefaultClient() *telegraph.Client {
|
||||||
if tphClient != nil {
|
once.Do(func() {
|
||||||
return tphClient
|
tphClient = initDefault()
|
||||||
}
|
})
|
||||||
if config.Cfg.Telegram.Proxy.Enable && config.Cfg.Telegram.Proxy.URL != "" {
|
return tphClient
|
||||||
proxyUrl := config.Cfg.Telegram.Proxy.URL
|
}
|
||||||
|
|
||||||
|
func initDefault() *telegraph.Client {
|
||||||
|
var client *telegraph.Client
|
||||||
|
if config.C().Telegram.Proxy.Enable && config.C().Telegram.Proxy.URL != "" {
|
||||||
|
proxyUrl := config.C().Telegram.Proxy.URL
|
||||||
var err error
|
var err error
|
||||||
tphClient, err = telegraph.NewClientWithProxy(proxyUrl)
|
client, err = telegraph.NewClientWithProxy(proxyUrl)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
tphClient = telegraph.NewClient()
|
client = telegraph.NewClient()
|
||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
tphClient = telegraph.NewClient()
|
client = telegraph.NewClient()
|
||||||
}
|
}
|
||||||
return tphClient
|
return client
|
||||||
}
|
}
|
||||||
|
|
||||||
func GetNodeImages(node telegraph.Node) []string {
|
func GetNodeImages(node telegraph.Node) []string {
|
||||||
@@ -41,6 +51,10 @@ func GetNodeImages(node telegraph.Node) []string {
|
|||||||
|
|
||||||
if nodeElement.Tag == "img" {
|
if nodeElement.Tag == "img" {
|
||||||
if src, exists := nodeElement.Attrs["src"]; exists {
|
if src, exists := nodeElement.Attrs["src"]; exists {
|
||||||
|
if strings.HasPrefix(src, "/file/") {
|
||||||
|
// handle images on telegra.ph server
|
||||||
|
src = "https://telegra.ph" + src
|
||||||
|
}
|
||||||
srcs = append(srcs, src)
|
srcs = append(srcs, src)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -14,15 +14,26 @@ token = ""
|
|||||||
# app_id = 1025907
|
# app_id = 1025907
|
||||||
# app_hash = "452b0359b988148995f22ff0f4229750"
|
# app_hash = "452b0359b988148995f22ff0f4229750"
|
||||||
[telegram.proxy]
|
[telegram.proxy]
|
||||||
# 启用代理连接 telegram, 只支持 socks5
|
# 启用代理连接 telegram
|
||||||
enable = false
|
enable = false
|
||||||
url = "socks5://127.0.0.1:7890"
|
url = "socks5://127.0.0.1:7890"
|
||||||
|
|
||||||
|
# Aria2 配置
|
||||||
|
[aria2]
|
||||||
|
# 启用 Aria2 下载支持
|
||||||
|
enable = false
|
||||||
|
# Aria2 RPC URL
|
||||||
|
url = "http://localhost:6800/jsonrpc"
|
||||||
|
# Aria2 RPC Secret (如果配置了 rpc-secret)
|
||||||
|
secret = ""
|
||||||
|
# 转存完成后删除 Aria2 下载的本地文件
|
||||||
|
remove_after_transfer = true
|
||||||
|
|
||||||
# 存储列表
|
# 存储列表
|
||||||
[[storages]]
|
[[storages]]
|
||||||
# 标识名, 需要唯一
|
# 标识名, 需要唯一
|
||||||
name = "本机1"
|
name = "本机1"
|
||||||
# 存储类型, 目前可用: local, alist, webdav, minio, telegram
|
# 存储类型, 目前可用: local, alist, webdav, s3, telegram
|
||||||
type = "local"
|
type = "local"
|
||||||
# 启用存储
|
# 启用存储
|
||||||
enable = true
|
enable = true
|
||||||
|
|||||||
83
config/flags.go
Normal file
83
config/flags.go
Normal file
@@ -0,0 +1,83 @@
|
|||||||
|
package config
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/spf13/cobra"
|
||||||
|
"github.com/spf13/viper"
|
||||||
|
)
|
||||||
|
|
||||||
|
func RegisterFlags(cmd *cobra.Command) {
|
||||||
|
flags := cmd.Flags()
|
||||||
|
|
||||||
|
// 基础配置
|
||||||
|
flags.StringP("config", "c", "", "config file path")
|
||||||
|
flags.StringP("lang", "l", "", "language (e.g., zh-Hans, en)")
|
||||||
|
flags.IntP("workers", "w", 0, "number of workers")
|
||||||
|
flags.Int("retry", 0, "retry times")
|
||||||
|
flags.Int("threads", 0, "number of threads")
|
||||||
|
flags.Bool("stream", false, "enable stream mode")
|
||||||
|
flags.Bool("no-clean-cache", false, "do not clean cache on exit")
|
||||||
|
flags.String("proxy", "", "proxy URL (http, https, socks5, socks5h)")
|
||||||
|
|
||||||
|
// Telegram 配置
|
||||||
|
flags.String("telegram-token", "", "telegram bot token")
|
||||||
|
flags.Int("telegram-app-id", 0, "telegram app id")
|
||||||
|
flags.String("telegram-app-hash", "", "telegram app hash")
|
||||||
|
flags.Int("telegram-rpc-retry", 0, "telegram rpc retry times")
|
||||||
|
flags.Bool("telegram-userbot-enable", false, "enable userbot")
|
||||||
|
flags.String("telegram-userbot-session", "", "userbot session path")
|
||||||
|
flags.Bool("telegram-proxy-enable", false, "enable telegram proxy")
|
||||||
|
flags.String("telegram-proxy-url", "", "telegram proxy URL")
|
||||||
|
|
||||||
|
// 数据库配置
|
||||||
|
flags.String("db-path", "", "database path")
|
||||||
|
flags.String("db-session", "", "session database path")
|
||||||
|
|
||||||
|
// 临时目录配置
|
||||||
|
flags.String("temp-base-path", "", "temp directory base path")
|
||||||
|
|
||||||
|
// Parser 配置
|
||||||
|
flags.Bool("parser-plugin-enable", false, "enable parser plugins")
|
||||||
|
flags.StringSlice("parser-plugin-dirs", nil, "parser plugin directories")
|
||||||
|
flags.String("parser-proxy", "", "parser proxy URL")
|
||||||
|
|
||||||
|
// 绑定到 viper
|
||||||
|
bindFlags(cmd)
|
||||||
|
}
|
||||||
|
|
||||||
|
func bindFlags(cmd *cobra.Command) {
|
||||||
|
flags := cmd.Flags()
|
||||||
|
|
||||||
|
viper.BindPFlag("lang", flags.Lookup("lang"))
|
||||||
|
viper.BindPFlag("workers", flags.Lookup("workers"))
|
||||||
|
viper.BindPFlag("retry", flags.Lookup("retry"))
|
||||||
|
viper.BindPFlag("threads", flags.Lookup("threads"))
|
||||||
|
viper.BindPFlag("stream", flags.Lookup("stream"))
|
||||||
|
viper.BindPFlag("no_clean_cache", flags.Lookup("no-clean-cache"))
|
||||||
|
viper.BindPFlag("proxy", flags.Lookup("proxy"))
|
||||||
|
|
||||||
|
// Telegram
|
||||||
|
viper.BindPFlag("telegram.token", flags.Lookup("telegram-token"))
|
||||||
|
viper.BindPFlag("telegram.app_id", flags.Lookup("telegram-app-id"))
|
||||||
|
viper.BindPFlag("telegram.app_hash", flags.Lookup("telegram-app-hash"))
|
||||||
|
viper.BindPFlag("telegram.rpc_retry", flags.Lookup("telegram-rpc-retry"))
|
||||||
|
viper.BindPFlag("telegram.userbot.enable", flags.Lookup("telegram-userbot-enable"))
|
||||||
|
viper.BindPFlag("telegram.userbot.session", flags.Lookup("telegram-userbot-session"))
|
||||||
|
viper.BindPFlag("telegram.proxy.enable", flags.Lookup("telegram-proxy-enable"))
|
||||||
|
viper.BindPFlag("telegram.proxy.url", flags.Lookup("telegram-proxy-url"))
|
||||||
|
|
||||||
|
// database
|
||||||
|
viper.BindPFlag("db.path", flags.Lookup("db-path"))
|
||||||
|
viper.BindPFlag("db.session", flags.Lookup("db-session"))
|
||||||
|
// 临时目录
|
||||||
|
viper.BindPFlag("temp.base_path", flags.Lookup("temp-base-path"))
|
||||||
|
|
||||||
|
// Parser
|
||||||
|
viper.BindPFlag("parser.plugin_enable", flags.Lookup("parser-plugin-enable"))
|
||||||
|
viper.BindPFlag("parser.plugin_dirs", flags.Lookup("parser-plugin-dirs"))
|
||||||
|
viper.BindPFlag("parser.proxy", flags.Lookup("parser-proxy"))
|
||||||
|
}
|
||||||
|
|
||||||
|
func GetConfigFile(cmd *cobra.Command) string {
|
||||||
|
configFile, _ := cmd.Flags().GetString("config")
|
||||||
|
return configFile
|
||||||
|
}
|
||||||
@@ -1,6 +1,15 @@
|
|||||||
package config
|
package config
|
||||||
|
|
||||||
type parserConfig struct {
|
type parserConfig struct {
|
||||||
PluginEnable bool `toml:"plugin_enable" mapstructure:"plugin_enable" json:"plugin_enable"`
|
PluginEnable bool `toml:"plugin_enable" mapstructure:"plugin_enable" json:"plugin_enable"`
|
||||||
PluginDirs []string `toml:"plugin_dirs" mapstructure:"plugin_dirs" json:"plugin_dirs"`
|
PluginDirs []string `toml:"plugin_dirs" mapstructure:"plugin_dirs" json:"plugin_dirs"`
|
||||||
|
Proxy string `toml:"proxy" mapstructure:"proxy" json:"proxy"`
|
||||||
|
ParserCfgs map[string]map[string]any `mapstructure:",remain"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c Config) GetParserConfigByName(name string) map[string]any {
|
||||||
|
if c.Parser.ParserCfgs == nil {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
return c.Parser.ParserCfgs[name]
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -14,6 +14,7 @@ var storageFactories = map[storenum.StorageType]func(cfg *BaseConfig) (StorageCo
|
|||||||
storenum.Alist: createStorageConfig(&AlistStorageConfig{}),
|
storenum.Alist: createStorageConfig(&AlistStorageConfig{}),
|
||||||
storenum.Webdav: createStorageConfig(&WebdavStorageConfig{}),
|
storenum.Webdav: createStorageConfig(&WebdavStorageConfig{}),
|
||||||
storenum.Minio: createStorageConfig(&MinioStorageConfig{}),
|
storenum.Minio: createStorageConfig(&MinioStorageConfig{}),
|
||||||
|
storenum.S3: createStorageConfig(&S3StorageConfig{}),
|
||||||
storenum.Telegram: createStorageConfig(&TelegramStorageConfig{}),
|
storenum.Telegram: createStorageConfig(&TelegramStorageConfig{}),
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
43
config/storage/s3.go
Normal file
43
config/storage/s3.go
Normal file
@@ -0,0 +1,43 @@
|
|||||||
|
package storage
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
|
||||||
|
storenum "github.com/krau/SaveAny-Bot/pkg/enums/storage"
|
||||||
|
)
|
||||||
|
|
||||||
|
type S3StorageConfig struct {
|
||||||
|
BaseConfig
|
||||||
|
Endpoint string `toml:"endpoint" mapstructure:"endpoint" json:"endpoint"`
|
||||||
|
AccessKeyID string `toml:"access_key_id" mapstructure:"access_key_id" json:"access_key_id"`
|
||||||
|
SecretAccessKey string `toml:"secret_access_key" mapstructure:"secret_access_key" json:"secret_access_key"`
|
||||||
|
BucketName string `toml:"bucket_name" mapstructure:"bucket_name" json:"bucket_name"`
|
||||||
|
UseSSL bool `toml:"use_ssl" mapstructure:"use_ssl" json:"use_ssl"`
|
||||||
|
BasePath string `toml:"base_path" mapstructure:"base_path" json:"base_path"`
|
||||||
|
Region string `toml:"region" mapstructure:"region" json:"region"`
|
||||||
|
VirtualHost bool `toml:"virtual_host" mapstructure:"virtual_host" json:"virtual_host"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *S3StorageConfig) Validate() error {
|
||||||
|
if m.Endpoint == "" {
|
||||||
|
return fmt.Errorf("endpoint is required for s3 storage")
|
||||||
|
}
|
||||||
|
if m.AccessKeyID == "" || m.SecretAccessKey == "" {
|
||||||
|
return fmt.Errorf("access_key_id and secret_access_key are required for s3 storage")
|
||||||
|
}
|
||||||
|
if m.BucketName == "" {
|
||||||
|
return fmt.Errorf("bucket_name is required for s3 storage")
|
||||||
|
}
|
||||||
|
if m.BasePath == "" {
|
||||||
|
return fmt.Errorf("base_path is required for s3 storage")
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *S3StorageConfig) GetType() storenum.StorageType {
|
||||||
|
return storenum.S3
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *S3StorageConfig) GetName() string {
|
||||||
|
return m.Name
|
||||||
|
}
|
||||||
@@ -9,8 +9,14 @@ import (
|
|||||||
type TelegramStorageConfig struct {
|
type TelegramStorageConfig struct {
|
||||||
BaseConfig
|
BaseConfig
|
||||||
ChatID int64 `toml:"chat_id" mapstructure:"chat_id" json:"chat_id"`
|
ChatID int64 `toml:"chat_id" mapstructure:"chat_id" json:"chat_id"`
|
||||||
|
ForceFile bool `toml:"force_file" mapstructure:"force_file" json:"force_file"`
|
||||||
RateLimit int `toml:"rate_limit" mapstructure:"rate_limit" json:"rate_limit"`
|
RateLimit int `toml:"rate_limit" mapstructure:"rate_limit" json:"rate_limit"`
|
||||||
RateBurst int `toml:"rate_burst" mapstructure:"rate_burst" json:"rate_burst"`
|
RateBurst int `toml:"rate_burst" mapstructure:"rate_burst" json:"rate_burst"`
|
||||||
|
SkipLarge bool `toml:"skip_large" mapstructure:"skip_large" json:"skip_large"` // skip files larger than Telegram limit(2GB)
|
||||||
|
// split files larger than Telegram limit(2GB) into parts of specified size, in MB, leave 0 to set default(2000MB)
|
||||||
|
// only effective when SkipLarge is false
|
||||||
|
// use zip when splitting
|
||||||
|
SplitSizeMB int64 `toml:"split_size_mb" mapstructure:"split_size_mb" json:"split_size_mb"`
|
||||||
}
|
}
|
||||||
|
|
||||||
func (m *TelegramStorageConfig) Validate() error {
|
func (m *TelegramStorageConfig) Validate() error {
|
||||||
|
|||||||
13
config/tg.go
13
config/tg.go
@@ -1,12 +1,13 @@
|
|||||||
package config
|
package config
|
||||||
|
|
||||||
type telegramConfig struct {
|
type telegramConfig struct {
|
||||||
Token string `toml:"token" mapstructure:"token"`
|
Token string `toml:"token" mapstructure:"token"`
|
||||||
AppID int `toml:"app_id" mapstructure:"app_id" json:"app_id"`
|
AppID int `toml:"app_id" mapstructure:"app_id" json:"app_id"`
|
||||||
AppHash string `toml:"app_hash" mapstructure:"app_hash" json:"app_hash"`
|
AppHash string `toml:"app_hash" mapstructure:"app_hash" json:"app_hash"`
|
||||||
Proxy tgProxyConfig `toml:"proxy" mapstructure:"proxy"`
|
Proxy tgProxyConfig `toml:"proxy" mapstructure:"proxy"`
|
||||||
RpcRetry int `toml:"rpc_retry" mapstructure:"rpc_retry" json:"rpc_retry"`
|
RpcRetry int `toml:"rpc_retry" mapstructure:"rpc_retry" json:"rpc_retry"`
|
||||||
Userbot userbotConfig `toml:"userbot" mapstructure:"userbot" json:"userbot"`
|
Userbot userbotConfig `toml:"userbot" mapstructure:"userbot" json:"userbot"`
|
||||||
|
MediaGroupTimeout int `toml:"media_group_timeout" mapstructure:"media_group_timeout" json:"media_group_timeout"`
|
||||||
}
|
}
|
||||||
|
|
||||||
type userbotConfig struct {
|
type userbotConfig struct {
|
||||||
|
|||||||
@@ -14,7 +14,7 @@ var userIDs []int64
|
|||||||
var storages []string
|
var storages []string
|
||||||
var userStorages = make(map[int64][]string)
|
var userStorages = make(map[int64][]string)
|
||||||
|
|
||||||
func (c *Config) GetStorageNamesByUserID(userID int64) []string {
|
func (c Config) GetStorageNamesByUserID(userID int64) []string {
|
||||||
us, ok := userStorages[userID]
|
us, ok := userStorages[userID]
|
||||||
if ok {
|
if ok {
|
||||||
return us
|
return us
|
||||||
@@ -22,11 +22,11 @@ func (c *Config) GetStorageNamesByUserID(userID int64) []string {
|
|||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *Config) GetUsersID() []int64 {
|
func (c Config) GetUsersID() []int64 {
|
||||||
return userIDs
|
return userIDs
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *Config) HasStorage(userID int64, storageName string) bool {
|
func (c Config) HasStorage(userID int64, storageName string) bool {
|
||||||
us, ok := userStorages[userID]
|
us, ok := userStorages[userID]
|
||||||
if !ok {
|
if !ok {
|
||||||
return false
|
return false
|
||||||
|
|||||||
14
config/version.go
Normal file
14
config/version.go
Normal file
@@ -0,0 +1,14 @@
|
|||||||
|
package config
|
||||||
|
|
||||||
|
// inject version by '-X' flag
|
||||||
|
// go build -ldflags "-X github.com/krau/SaveAny-Bot/config.Version=${{ env.VERSION }}"
|
||||||
|
var (
|
||||||
|
Version string = "dev"
|
||||||
|
BuildTime string = "unknown"
|
||||||
|
GitCommit string = "unknown"
|
||||||
|
Docker string = "false" // whether built inside Docker
|
||||||
|
)
|
||||||
|
|
||||||
|
const (
|
||||||
|
GitRepo = "krau/SaveAny-Bot"
|
||||||
|
)
|
||||||
158
config/viper.go
158
config/viper.go
@@ -2,25 +2,28 @@ package config
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
"errors"
|
|
||||||
"fmt"
|
"fmt"
|
||||||
"os"
|
"net"
|
||||||
|
"net/http"
|
||||||
|
"net/url"
|
||||||
"strings"
|
"strings"
|
||||||
|
"time"
|
||||||
|
|
||||||
"github.com/duke-git/lancet/v2/slice"
|
"github.com/duke-git/lancet/v2/slice"
|
||||||
"github.com/krau/SaveAny-Bot/common/i18n"
|
|
||||||
"github.com/krau/SaveAny-Bot/common/i18n/i18nk"
|
|
||||||
"github.com/krau/SaveAny-Bot/config/storage"
|
"github.com/krau/SaveAny-Bot/config/storage"
|
||||||
"github.com/spf13/viper"
|
"github.com/spf13/viper"
|
||||||
|
"golang.org/x/net/proxy"
|
||||||
)
|
)
|
||||||
|
|
||||||
type Config struct {
|
type Config struct {
|
||||||
Lang string `toml:"lang" mapstructure:"lang" json:"lang"`
|
Lang string `toml:"lang" mapstructure:"lang" json:"lang"`
|
||||||
Workers int `toml:"workers" mapstructure:"workers"`
|
Workers int `toml:"workers" mapstructure:"workers"`
|
||||||
Retry int `toml:"retry" mapstructure:"retry"`
|
Retry int `toml:"retry" mapstructure:"retry"`
|
||||||
NoCleanCache bool `toml:"no_clean_cache" mapstructure:"no_clean_cache" json:"no_clean_cache"`
|
NoCleanCache bool `toml:"no_clean_cache" mapstructure:"no_clean_cache" json:"no_clean_cache"`
|
||||||
Threads int `toml:"threads" mapstructure:"threads" json:"threads"`
|
Threads int `toml:"threads" mapstructure:"threads" json:"threads"`
|
||||||
Stream bool `toml:"stream" mapstructure:"stream" json:"stream"`
|
Stream bool `toml:"stream" mapstructure:"stream" json:"stream"`
|
||||||
|
Proxy string `toml:"proxy" mapstructure:"proxy" json:"proxy"`
|
||||||
|
Aria2 aria2Config `toml:"aria2" mapstructure:"aria2" json:"aria2"`
|
||||||
|
|
||||||
Cache cacheConfig `toml:"cache" mapstructure:"cache" json:"cache"`
|
Cache cacheConfig `toml:"cache" mapstructure:"cache" json:"cache"`
|
||||||
Users []userConfig `toml:"users" mapstructure:"users" json:"users"`
|
Users []userConfig `toml:"users" mapstructure:"users" json:"users"`
|
||||||
@@ -32,7 +35,18 @@ type Config struct {
|
|||||||
Hook hookConfig `toml:"hook" mapstructure:"hook" json:"hook"`
|
Hook hookConfig `toml:"hook" mapstructure:"hook" json:"hook"`
|
||||||
}
|
}
|
||||||
|
|
||||||
var Cfg *Config = &Config{}
|
type aria2Config struct {
|
||||||
|
Enable bool `toml:"enable" mapstructure:"enable" json:"enable"`
|
||||||
|
Url string `toml:"url" mapstructure:"url" json:"url"`
|
||||||
|
Secret string `toml:"secret" mapstructure:"secret" json:"secret"`
|
||||||
|
KeepFile bool `toml:"keep_file" mapstructure:"keep_file" json:"keep_file"`
|
||||||
|
}
|
||||||
|
|
||||||
|
var cfg = &Config{}
|
||||||
|
|
||||||
|
func C() Config {
|
||||||
|
return *cfg
|
||||||
|
}
|
||||||
|
|
||||||
func (c Config) GetStorageByName(name string) storage.StorageConfig {
|
func (c Config) GetStorageByName(name string) storage.StorageConfig {
|
||||||
for _, storage := range c.Storages {
|
for _, storage := range c.Storages {
|
||||||
@@ -43,16 +57,39 @@ func (c Config) GetStorageByName(name string) storage.StorageConfig {
|
|||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func Init(ctx context.Context) error {
|
func Init(ctx context.Context, configFile ...string) error {
|
||||||
viper.SetConfigName("config")
|
|
||||||
viper.AddConfigPath(".")
|
|
||||||
viper.AddConfigPath("/etc/saveany/")
|
|
||||||
viper.SetConfigType("toml")
|
viper.SetConfigType("toml")
|
||||||
viper.SetEnvPrefix("SAVEANY")
|
viper.SetEnvPrefix("SAVEANY")
|
||||||
viper.AutomaticEnv()
|
viper.AutomaticEnv()
|
||||||
replacer := strings.NewReplacer(".", "_")
|
replacer := strings.NewReplacer(".", "_")
|
||||||
viper.SetEnvKeyReplacer(replacer)
|
viper.SetEnvKeyReplacer(replacer)
|
||||||
|
|
||||||
|
// 如果指定了配置文件路径,则使用指定的配置文件
|
||||||
|
// 配置文件支持传入一个 http(s) URL 地址
|
||||||
|
if len(configFile) > 0 && configFile[0] != "" {
|
||||||
|
cfg := configFile[0]
|
||||||
|
if strings.HasPrefix(cfg, "http://") || strings.HasPrefix(cfg, "https://") {
|
||||||
|
// 使用远程配置文件
|
||||||
|
resp, err := http.Get(cfg)
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("failed to fetch remote config file: %w", err)
|
||||||
|
}
|
||||||
|
defer resp.Body.Close()
|
||||||
|
if resp.StatusCode != http.StatusOK {
|
||||||
|
return fmt.Errorf("failed to fetch remote config file: status code %d", resp.StatusCode)
|
||||||
|
}
|
||||||
|
if err := viper.ReadConfig(resp.Body); err != nil {
|
||||||
|
return fmt.Errorf("failed to read remote config file: %w", err)
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
viper.SetConfigFile(cfg)
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
viper.SetConfigName("config")
|
||||||
|
viper.AddConfigPath(".")
|
||||||
|
viper.AddConfigPath("/etc/saveany/")
|
||||||
|
}
|
||||||
|
|
||||||
defaultConfigs := map[string]any{
|
defaultConfigs := map[string]any{
|
||||||
// 基础配置
|
// 基础配置
|
||||||
"lang": "zh-Hans",
|
"lang": "zh-Hans",
|
||||||
@@ -92,48 +129,42 @@ func Init(ctx context.Context) error {
|
|||||||
|
|
||||||
if err := viper.ReadInConfig(); err != nil {
|
if err := viper.ReadInConfig(); err != nil {
|
||||||
fmt.Println("Error reading config file, ", err)
|
fmt.Println("Error reading config file, ", err)
|
||||||
os.Exit(1)
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
if err := viper.Unmarshal(Cfg); err != nil {
|
if err := viper.Unmarshal(cfg); err != nil {
|
||||||
fmt.Println("Error unmarshalling config file, ", err)
|
fmt.Println("Error unmarshalling config file, ", err)
|
||||||
os.Exit(1)
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
storagesConfig, err := storage.LoadStorageConfigs(viper.GetViper())
|
storagesConfig, err := storage.LoadStorageConfigs(viper.GetViper())
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return fmt.Errorf("error loading storage configs: %w", err)
|
return fmt.Errorf("error loading storage configs: %w", err)
|
||||||
}
|
}
|
||||||
Cfg.Storages = storagesConfig
|
cfg.Storages = storagesConfig
|
||||||
|
|
||||||
storageNames := make(map[string]struct{})
|
storageNames := make(map[string]struct{})
|
||||||
for _, storage := range Cfg.Storages {
|
for _, storage := range cfg.Storages {
|
||||||
if _, ok := storageNames[storage.GetName()]; ok {
|
if _, ok := storageNames[storage.GetName()]; ok {
|
||||||
return errors.New(i18n.TWithoutInit(Cfg.Lang, i18nk.ConfigInvalidDuplicateStorageName, map[string]any{
|
return fmt.Errorf("duplicate storage name: %s", storage.GetName())
|
||||||
"Name": storage.GetName(),
|
|
||||||
}))
|
|
||||||
}
|
}
|
||||||
storageNames[storage.GetName()] = struct{}{}
|
storageNames[storage.GetName()] = struct{}{}
|
||||||
}
|
}
|
||||||
|
|
||||||
fmt.Println(i18n.TWithoutInit(Cfg.Lang, i18nk.LoadedStorages, map[string]any{
|
if cfg.Workers < 1 {
|
||||||
"Count": len(Cfg.Storages),
|
cfg.Workers = 1
|
||||||
}))
|
}
|
||||||
for _, storage := range Cfg.Storages {
|
if cfg.Threads < 1 {
|
||||||
fmt.Printf(" - %s (%s)\n", storage.GetName(), storage.GetType())
|
cfg.Threads = 1
|
||||||
|
}
|
||||||
|
if cfg.Retry < 1 {
|
||||||
|
cfg.Retry = 1
|
||||||
}
|
}
|
||||||
|
|
||||||
if Cfg.Workers < 1 || Cfg.Retry < 1 {
|
for _, storage := range cfg.Storages {
|
||||||
return errors.New(i18n.TWithoutInit(Cfg.Lang, i18nk.ConfigInvalidWorkersOrRetry, map[string]any{
|
|
||||||
"Workers": Cfg.Workers,
|
|
||||||
"Retry": Cfg.Retry,
|
|
||||||
}))
|
|
||||||
}
|
|
||||||
|
|
||||||
for _, storage := range Cfg.Storages {
|
|
||||||
storages = append(storages, storage.GetName())
|
storages = append(storages, storage.GetName())
|
||||||
}
|
}
|
||||||
for _, user := range Cfg.Users {
|
for _, user := range cfg.Users {
|
||||||
userIDs = append(userIDs, user.ID)
|
userIDs = append(userIDs, user.ID)
|
||||||
if user.Blacklist {
|
if user.Blacklist {
|
||||||
userStorages[user.ID] = slice.Compact(slice.Difference(storages, user.Storages))
|
userStorages[user.ID] = slice.Compact(slice.Difference(storages, user.Storages))
|
||||||
@@ -141,22 +172,43 @@ func Init(ctx context.Context) error {
|
|||||||
userStorages[user.ID] = user.Storages
|
userStorages[user.ID] = user.Storages
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
return nil
|
if cfg.Proxy != "" {
|
||||||
}
|
http.DefaultTransport, err = newProxyTransport(cfg.Proxy)
|
||||||
|
if err != nil {
|
||||||
func Set(key string, value any) {
|
return fmt.Errorf("failed to create proxy transport: %w", err)
|
||||||
viper.Set(key, value)
|
}
|
||||||
}
|
|
||||||
|
|
||||||
func ReloadConfig() error {
|
|
||||||
if err := viper.WriteConfig(); err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
if err := viper.ReadInConfig(); err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
if error := viper.Unmarshal(Cfg); error != nil {
|
|
||||||
return error
|
|
||||||
}
|
}
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func newProxyTransport(proxyStr string) (*http.Transport, error) {
|
||||||
|
proxyURL, err := url.Parse(proxyStr)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
transport := &http.Transport{
|
||||||
|
ForceAttemptHTTP2: true,
|
||||||
|
MaxIdleConns: 100,
|
||||||
|
IdleConnTimeout: 90 * time.Second,
|
||||||
|
TLSHandshakeTimeout: 10 * time.Second,
|
||||||
|
ExpectContinueTimeout: 1 * time.Second,
|
||||||
|
}
|
||||||
|
switch proxyURL.Scheme {
|
||||||
|
case "http", "https":
|
||||||
|
transport.Proxy = http.ProxyURL(proxyURL)
|
||||||
|
|
||||||
|
case "socks5", "socks5h":
|
||||||
|
dialer, err := proxy.FromURL(proxyURL, proxy.Direct)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
transport.DialContext = func(ctx context.Context, network, addr string) (net.Conn, error) {
|
||||||
|
return dialer.(proxy.ContextDialer).DialContext(ctx, network, addr)
|
||||||
|
}
|
||||||
|
|
||||||
|
default:
|
||||||
|
return nil, fmt.Errorf("unsupported proxy type: %s", proxyURL.Scheme)
|
||||||
|
}
|
||||||
|
|
||||||
|
return transport, nil
|
||||||
|
}
|
||||||
|
|||||||
50
core/core.go
50
core/core.go
@@ -10,17 +10,18 @@ import (
|
|||||||
"github.com/krau/SaveAny-Bot/pkg/queue"
|
"github.com/krau/SaveAny-Bot/pkg/queue"
|
||||||
)
|
)
|
||||||
|
|
||||||
var queueInstance *queue.TaskQueue[Exectable]
|
var queueInstance *queue.TaskQueue[Executable]
|
||||||
|
|
||||||
type Exectable interface {
|
type Executable interface {
|
||||||
Type() tasktype.TaskType
|
Type() tasktype.TaskType
|
||||||
|
Title() string
|
||||||
TaskID() string
|
TaskID() string
|
||||||
Execute(ctx context.Context) error
|
Execute(ctx context.Context) error
|
||||||
}
|
}
|
||||||
|
|
||||||
func worker(ctx context.Context, qe *queue.TaskQueue[Exectable], semaphore chan struct{}) {
|
func worker(ctx context.Context, qe *queue.TaskQueue[Executable], semaphore chan struct{}) {
|
||||||
logger := log.FromContext(ctx)
|
logger := log.FromContext(ctx)
|
||||||
execHooks := config.Cfg.Hook.Exec
|
execHooks := config.C().Hook.Exec
|
||||||
for {
|
for {
|
||||||
semaphore <- struct{}{}
|
semaphore <- struct{}{}
|
||||||
qtask, err := qe.Get()
|
qtask, err := qe.Get()
|
||||||
@@ -28,27 +29,27 @@ func worker(ctx context.Context, qe *queue.TaskQueue[Exectable], semaphore chan
|
|||||||
logger.Error("Failed to get task from queue:", err)
|
logger.Error("Failed to get task from queue:", err)
|
||||||
break // queue closed and empty
|
break // queue closed and empty
|
||||||
}
|
}
|
||||||
task := qtask.Data
|
exe := qtask.Data
|
||||||
logger.Infof("Processing task: %s", task.TaskID())
|
logger.Infof("Processing task: %s", exe.TaskID())
|
||||||
if err := ExecCommandString(qtask.Context(), execHooks.TaskBeforeStart); err != nil {
|
if err := ExecCommandString(qtask.Context(), execHooks.TaskBeforeStart); err != nil {
|
||||||
logger.Errorf("Failed to execute before start hook for task %s: %v", task.TaskID(), err)
|
logger.Errorf("Failed to execute before start hook for task %s: %v", exe.TaskID(), err)
|
||||||
}
|
}
|
||||||
if err := task.Execute(qtask.Context()); err != nil {
|
if err := exe.Execute(qtask.Context()); err != nil {
|
||||||
if errors.Is(err, context.Canceled) {
|
if errors.Is(err, context.Canceled) {
|
||||||
logger.Infof("Task %s was canceled", task.TaskID())
|
logger.Infof("Task %s was canceled", exe.TaskID())
|
||||||
if err := ExecCommandString(ctx, execHooks.TaskCancel); err != nil {
|
if err := ExecCommandString(ctx, execHooks.TaskCancel); err != nil {
|
||||||
logger.Errorf("Failed to execute cancel hook for task %s: %v", task.TaskID(), err)
|
logger.Errorf("Failed to execute cancel hook for task %s: %v", exe.TaskID(), err)
|
||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
logger.Errorf("Failed to execute task %s: %v", task.TaskID(), err)
|
logger.Errorf("Failed to execute task %s: %v", exe.TaskID(), err)
|
||||||
if err := ExecCommandString(ctx, execHooks.TaskFail); err != nil {
|
if err := ExecCommandString(ctx, execHooks.TaskFail); err != nil {
|
||||||
logger.Errorf("Failed to execute fail hook for task %s: %v", task.TaskID(), err)
|
logger.Errorf("Failed to execute fail hook for task %s: %v", exe.TaskID(), err)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
logger.Infof("Task %s completed successfully", task.TaskID())
|
logger.Infof("Task %s completed successfully", exe.TaskID())
|
||||||
if err := ExecCommandString(ctx, execHooks.TaskSuccess); err != nil {
|
if err := ExecCommandString(ctx, execHooks.TaskSuccess); err != nil {
|
||||||
logger.Errorf("Failed to execute success hook for task %s: %v", task.TaskID(), err)
|
logger.Errorf("Failed to execute success hook for task %s: %v", exe.TaskID(), err)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
qe.Done(qtask.ID)
|
qe.Done(qtask.ID)
|
||||||
@@ -58,18 +59,18 @@ func worker(ctx context.Context, qe *queue.TaskQueue[Exectable], semaphore chan
|
|||||||
|
|
||||||
func Run(ctx context.Context) {
|
func Run(ctx context.Context) {
|
||||||
log.FromContext(ctx).Info("Start processing tasks...")
|
log.FromContext(ctx).Info("Start processing tasks...")
|
||||||
semaphore := make(chan struct{}, config.Cfg.Workers)
|
semaphore := make(chan struct{}, config.C().Workers)
|
||||||
if queueInstance == nil {
|
if queueInstance == nil {
|
||||||
queueInstance = queue.NewTaskQueue[Exectable]()
|
queueInstance = queue.NewTaskQueue[Executable]()
|
||||||
}
|
}
|
||||||
for range config.Cfg.Workers {
|
for range config.C().Workers {
|
||||||
go worker(ctx, queueInstance, semaphore)
|
go worker(ctx, queueInstance, semaphore)
|
||||||
}
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func AddTask(ctx context.Context, task Exectable) error {
|
func AddTask(ctx context.Context, task Executable) error {
|
||||||
return queueInstance.Add(queue.NewTask(ctx, task.TaskID(), task))
|
return queueInstance.Add(queue.NewTask(ctx, task.TaskID(), task.Title(), task))
|
||||||
}
|
}
|
||||||
|
|
||||||
func CancelTask(ctx context.Context, id string) error {
|
func CancelTask(ctx context.Context, id string) error {
|
||||||
@@ -78,8 +79,13 @@ func CancelTask(ctx context.Context, id string) error {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func GetLength(ctx context.Context) int {
|
func GetLength(ctx context.Context) int {
|
||||||
if queueInstance == nil {
|
|
||||||
return 0
|
|
||||||
}
|
|
||||||
return queueInstance.ActiveLength()
|
return queueInstance.ActiveLength()
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func GetRunningTasks(ctx context.Context) []queue.TaskInfo {
|
||||||
|
return queueInstance.RunningTasks()
|
||||||
|
}
|
||||||
|
|
||||||
|
func GetQueuedTasks(ctx context.Context) []queue.TaskInfo {
|
||||||
|
return queueInstance.QueuedTasks()
|
||||||
|
}
|
||||||
|
|||||||
250
core/tasks/aria2dl/execute.go
Normal file
250
core/tasks/aria2dl/execute.go
Normal file
@@ -0,0 +1,250 @@
|
|||||||
|
package aria2dl
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
"os"
|
||||||
|
"path/filepath"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/charmbracelet/log"
|
||||||
|
"github.com/krau/SaveAny-Bot/config"
|
||||||
|
"github.com/krau/SaveAny-Bot/pkg/aria2"
|
||||||
|
"github.com/krau/SaveAny-Bot/pkg/enums/ctxkey"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Execute implements core.Executable.
|
||||||
|
func (t *Task) Execute(ctx context.Context) error {
|
||||||
|
logger := log.FromContext(ctx)
|
||||||
|
logger.Infof("Starting aria2 download task %s (GID: %s)", t.ID, t.GID)
|
||||||
|
|
||||||
|
if t.Progress != nil {
|
||||||
|
t.Progress.OnStart(ctx, t)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Wait for aria2 download to complete
|
||||||
|
if err := t.waitForDownload(ctx); err != nil {
|
||||||
|
// If context was canceled, also cancel the aria2 download
|
||||||
|
if errors.Is(err, context.Canceled) {
|
||||||
|
t.cancelAria2Download()
|
||||||
|
}
|
||||||
|
logger.Errorf("Aria2 download failed: %v", err)
|
||||||
|
if t.Progress != nil {
|
||||||
|
t.Progress.OnDone(ctx, t, err)
|
||||||
|
}
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
// Transfer downloaded files to storage
|
||||||
|
if err := t.transferFiles(ctx); err != nil {
|
||||||
|
logger.Errorf("File transfer failed: %v", err)
|
||||||
|
if t.Progress != nil {
|
||||||
|
t.Progress.OnDone(ctx, t, err)
|
||||||
|
}
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
logger.Infof("Aria2 task %s completed successfully", t.ID)
|
||||||
|
if t.Progress != nil {
|
||||||
|
t.Progress.OnDone(ctx, t, nil)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Clean up aria2 download result
|
||||||
|
if _, err := t.Aria2Client.RemoveDownloadResult(context.Background(), t.GID); err != nil {
|
||||||
|
logger.Warnf("Failed to remove aria2 download result: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// waitForDownload waits for aria2 to complete the download
|
||||||
|
func (t *Task) waitForDownload(ctx context.Context) error {
|
||||||
|
logger := log.FromContext(ctx)
|
||||||
|
ticker := time.NewTicker(2 * time.Second)
|
||||||
|
defer ticker.Stop()
|
||||||
|
|
||||||
|
for {
|
||||||
|
select {
|
||||||
|
case <-ctx.Done():
|
||||||
|
return ctx.Err()
|
||||||
|
case <-ticker.C:
|
||||||
|
status, err := t.getStatus(ctx)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
if t.Progress != nil {
|
||||||
|
t.Progress.OnProgress(ctx, t, status)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Check if download is complete
|
||||||
|
if status.IsDownloadComplete() {
|
||||||
|
// Handle metadata downloads (torrent/magnet) that spawn follow-up downloads
|
||||||
|
if len(status.FollowedBy) > 0 {
|
||||||
|
logger.Infof("Switching from metadata GID %s to actual download GID: %s", t.GID, status.FollowedBy[0])
|
||||||
|
t.GID = status.FollowedBy[0]
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
logger.Infof("Download completed for GID %s", t.GID)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// Check for errors
|
||||||
|
if status.IsDownloadError() {
|
||||||
|
return fmt.Errorf("aria2 download error: %s (code: %s)", status.ErrorMessage, status.ErrorCode)
|
||||||
|
}
|
||||||
|
|
||||||
|
if status.IsDownloadRemoved() {
|
||||||
|
return errors.New("aria2 download was removed")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// getStatus retrieves the current status of the download
|
||||||
|
func (t *Task) getStatus(ctx context.Context) (*aria2.Status, error) {
|
||||||
|
logger := log.FromContext(ctx)
|
||||||
|
|
||||||
|
// Try active/waiting queue first
|
||||||
|
status, err := t.Aria2Client.TellStatus(ctx, t.GID)
|
||||||
|
if err == nil {
|
||||||
|
return status, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// Check stopped queue
|
||||||
|
logger.Debugf("Task not in active queue, checking stopped queue")
|
||||||
|
stoppedTasks, stopErr := t.Aria2Client.TellStopped(ctx, -1, 100)
|
||||||
|
if stopErr != nil {
|
||||||
|
return nil, fmt.Errorf("failed to get aria2 status: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, task := range stoppedTasks {
|
||||||
|
if task.GID == t.GID {
|
||||||
|
logger.Debugf("Found task in stopped queue with status: %s", task.Status)
|
||||||
|
return &task, nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil, fmt.Errorf("task GID %s not found: %w", t.GID, err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// transferFiles transfers downloaded files from aria2 to storage
|
||||||
|
func (t *Task) transferFiles(ctx context.Context) error {
|
||||||
|
logger := log.FromContext(ctx)
|
||||||
|
|
||||||
|
status, err := t.Aria2Client.TellStatus(ctx, t.GID)
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("failed to get final status: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(status.Files) == 0 {
|
||||||
|
return errors.New("no files in aria2 download")
|
||||||
|
}
|
||||||
|
|
||||||
|
logger.Infof("Transferring %d file(s) to storage %s", len(status.Files), t.Storage.Name())
|
||||||
|
transferredCount := 0
|
||||||
|
|
||||||
|
for _, file := range status.Files {
|
||||||
|
if file.Selected != "true" {
|
||||||
|
logger.Debugf("Skipping unselected file: %s", file.Path)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
fileName := filepath.Base(file.Path)
|
||||||
|
|
||||||
|
// Skip torrent metadata files
|
||||||
|
if filepath.Ext(fileName) == ".torrent" {
|
||||||
|
logger.Debugf("Skipping torrent metadata file: %s", fileName)
|
||||||
|
t.removeFileIfNeeded(file.Path)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := t.transferFile(ctx, file.Path); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
transferredCount++
|
||||||
|
t.removeFileIfNeeded(file.Path)
|
||||||
|
}
|
||||||
|
|
||||||
|
if transferredCount == 0 {
|
||||||
|
return errors.New("no files were transferred")
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// transferFile transfers a single file to storage
|
||||||
|
func (t *Task) transferFile(ctx context.Context, filePath string) error {
|
||||||
|
logger := log.FromContext(ctx)
|
||||||
|
|
||||||
|
// Check if file exists
|
||||||
|
fileInfo, err := os.Stat(filePath)
|
||||||
|
if err != nil {
|
||||||
|
if os.IsNotExist(err) {
|
||||||
|
logger.Warnf("Downloaded file not found: %s", filePath)
|
||||||
|
return nil // Not a fatal error, continue with other files
|
||||||
|
}
|
||||||
|
return fmt.Errorf("failed to stat file %s: %w", filePath, err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Open file
|
||||||
|
f, err := os.Open(filePath)
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("failed to open file %s: %w", filePath, err)
|
||||||
|
}
|
||||||
|
defer f.Close()
|
||||||
|
|
||||||
|
// Set content length in context for storage
|
||||||
|
ctx = context.WithValue(ctx, ctxkey.ContentLength, fileInfo.Size())
|
||||||
|
|
||||||
|
// Save to storage
|
||||||
|
fileName := filepath.Base(filePath)
|
||||||
|
destPath := filepath.Join(t.StorPath, fileName)
|
||||||
|
|
||||||
|
logger.Infof("Transferring file %s to %s:%s", fileName, t.Storage.Name(), destPath)
|
||||||
|
|
||||||
|
if err := t.Storage.Save(ctx, f, destPath); err != nil {
|
||||||
|
return fmt.Errorf("failed to save file %s to storage: %w", fileName, err)
|
||||||
|
}
|
||||||
|
|
||||||
|
logger.Infof("Successfully transferred file %s", fileName)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// removeFileIfNeeded removes a file if RemoveAfterTransfer is enabled
|
||||||
|
func (t *Task) removeFileIfNeeded(filePath string) {
|
||||||
|
if config.C().Aria2.KeepFile {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
logger := log.FromContext(t.ctx)
|
||||||
|
if err := os.Remove(filePath); err != nil {
|
||||||
|
logger.Warnf("Failed to remove local file %s: %v", filePath, err)
|
||||||
|
} else {
|
||||||
|
logger.Debugf("Removed local file %s", filePath)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// cancelAria2Download cancels the aria2 download task
|
||||||
|
func (t *Task) cancelAria2Download() {
|
||||||
|
logger := log.FromContext(t.ctx)
|
||||||
|
logger.Infof("Canceling aria2 download GID: %s", t.GID)
|
||||||
|
|
||||||
|
// Use a background context with timeout for cleanup
|
||||||
|
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
|
||||||
|
defer cancel()
|
||||||
|
|
||||||
|
// Try to force remove the download
|
||||||
|
if _, err := t.Aria2Client.ForceRemove(ctx, t.GID); err != nil {
|
||||||
|
logger.Warnf("Failed to cancel aria2 download %s: %v", t.GID, err)
|
||||||
|
} else {
|
||||||
|
logger.Infof("Successfully canceled aria2 download %s", t.GID)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Also remove the download result to clean up
|
||||||
|
if _, err := t.Aria2Client.RemoveDownloadResult(ctx, t.GID); err != nil {
|
||||||
|
logger.Debugf("Failed to remove download result for %s: %v", t.GID, err)
|
||||||
|
}
|
||||||
|
}
|
||||||
189
core/tasks/aria2dl/progress.go
Normal file
189
core/tasks/aria2dl/progress.go
Normal file
@@ -0,0 +1,189 @@
|
|||||||
|
package aria2dl
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
"strconv"
|
||||||
|
"sync/atomic"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/charmbracelet/log"
|
||||||
|
"github.com/gotd/td/telegram/message/entity"
|
||||||
|
"github.com/gotd/td/telegram/message/styling"
|
||||||
|
"github.com/gotd/td/tg"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/i18n"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/i18n/i18nk"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/utils/dlutil"
|
||||||
|
"github.com/krau/SaveAny-Bot/common/utils/tgutil"
|
||||||
|
"github.com/krau/SaveAny-Bot/pkg/aria2"
|
||||||
|
)
|
||||||
|
|
||||||
|
type ProgressTracker interface {
|
||||||
|
OnStart(ctx context.Context, task *Task)
|
||||||
|
OnProgress(ctx context.Context, task *Task, status *aria2.Status)
|
||||||
|
OnDone(ctx context.Context, task *Task, err error)
|
||||||
|
}
|
||||||
|
|
||||||
|
type Progress struct {
|
||||||
|
msgID int
|
||||||
|
chatID int64
|
||||||
|
start time.Time
|
||||||
|
lastUpdatePercent atomic.Int32
|
||||||
|
}
|
||||||
|
|
||||||
|
// OnStart implements ProgressTracker.
|
||||||
|
func (p *Progress) OnStart(ctx context.Context, task *Task) {
|
||||||
|
logger := log.FromContext(ctx)
|
||||||
|
p.start = time.Now()
|
||||||
|
p.lastUpdatePercent.Store(0)
|
||||||
|
logger.Infof("Aria2 task started: message_id=%d, chat_id=%d, gid=%s", p.msgID, p.chatID, task.GID)
|
||||||
|
ext := tgutil.ExtFromContext(ctx)
|
||||||
|
if ext == nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
entityBuilder := entity.Builder{}
|
||||||
|
if err := styling.Perform(&entityBuilder,
|
||||||
|
styling.Plain(i18n.T(i18nk.BotMsgProgressAria2Start, map[string]any{
|
||||||
|
"GID": task.GID,
|
||||||
|
}))); err != nil {
|
||||||
|
log.FromContext(ctx).Errorf("Failed to build entities: %s", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
text, entities := entityBuilder.Complete()
|
||||||
|
req := &tg.MessagesEditMessageRequest{
|
||||||
|
ID: p.msgID,
|
||||||
|
}
|
||||||
|
req.SetMessage(text)
|
||||||
|
req.SetEntities(entities)
|
||||||
|
req.SetReplyMarkup(&tg.ReplyInlineMarkup{
|
||||||
|
Rows: []tg.KeyboardButtonRow{
|
||||||
|
{
|
||||||
|
Buttons: []tg.KeyboardButtonClass{
|
||||||
|
tgutil.BuildCancelButton(task.TaskID()),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
}},
|
||||||
|
)
|
||||||
|
ext.EditMessage(p.chatID, req)
|
||||||
|
}
|
||||||
|
|
||||||
|
// OnProgress implements ProgressTracker.
|
||||||
|
func (p *Progress) OnProgress(ctx context.Context, task *Task, status *aria2.Status) {
|
||||||
|
totalLength, _ := strconv.ParseInt(status.TotalLength, 10, 64)
|
||||||
|
completedLength, _ := strconv.ParseInt(status.CompletedLength, 10, 64)
|
||||||
|
downloadSpeed, _ := strconv.ParseInt(status.DownloadSpeed, 10, 64)
|
||||||
|
|
||||||
|
if totalLength == 0 {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
percent := int((completedLength * 100) / totalLength)
|
||||||
|
if p.lastUpdatePercent.Load() == int32(percent) {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
p.lastUpdatePercent.Store(int32(percent))
|
||||||
|
|
||||||
|
log.FromContext(ctx).Debugf("Aria2 progress update: %s, %d/%d", task.GID, completedLength, totalLength)
|
||||||
|
|
||||||
|
entityBuilder := entity.Builder{}
|
||||||
|
if err := styling.Perform(&entityBuilder,
|
||||||
|
styling.Plain(i18n.T(i18nk.BotMsgProgressAria2Downloading, map[string]any{
|
||||||
|
"GID": task.GID,
|
||||||
|
})),
|
||||||
|
styling.Plain(i18n.T(i18nk.BotMsgProgressDownloadedPrefix, nil)),
|
||||||
|
styling.Code(fmt.Sprintf("%.2f MB / %.2f MB", float64(completedLength)/(1024*1024), float64(totalLength)/(1024*1024))),
|
||||||
|
styling.Plain(i18n.T(i18nk.BotMsgProgressCurrentSpeedPrefix, nil)),
|
||||||
|
styling.Bold(fmt.Sprintf("%.2f MB/s", float64(downloadSpeed)/(1024*1024))),
|
||||||
|
styling.Plain(i18n.T(i18nk.BotMsgProgressAvgSpeedPrefix, nil)),
|
||||||
|
styling.Bold(fmt.Sprintf("%.2f MB/s", dlutil.GetSpeed(completedLength, p.start)/(1024*1024))),
|
||||||
|
styling.Plain(i18n.T(i18nk.BotMsgProgressCurrentProgressPrefix, nil)),
|
||||||
|
styling.Bold(fmt.Sprintf("%.2f%%", float64(percent))),
|
||||||
|
); err != nil {
|
||||||
|
log.FromContext(ctx).Errorf("Failed to build entities: %s", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
text, entities := entityBuilder.Complete()
|
||||||
|
req := &tg.MessagesEditMessageRequest{
|
||||||
|
ID: p.msgID,
|
||||||
|
}
|
||||||
|
req.SetMessage(text)
|
||||||
|
req.SetEntities(entities)
|
||||||
|
req.SetReplyMarkup(&tg.ReplyInlineMarkup{
|
||||||
|
Rows: []tg.KeyboardButtonRow{
|
||||||
|
{
|
||||||
|
Buttons: []tg.KeyboardButtonClass{
|
||||||
|
tgutil.BuildCancelButton(task.TaskID()),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
}},
|
||||||
|
)
|
||||||
|
ext := tgutil.ExtFromContext(ctx)
|
||||||
|
if ext != nil {
|
||||||
|
ext.EditMessage(p.chatID, req)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// OnDone implements ProgressTracker.
|
||||||
|
func (p *Progress) OnDone(ctx context.Context, task *Task, err error) {
|
||||||
|
logger := log.FromContext(ctx)
|
||||||
|
if err != nil {
|
||||||
|
if errors.Is(err, context.Canceled) {
|
||||||
|
logger.Infof("Aria2 task %s was canceled", task.TaskID())
|
||||||
|
ext := tgutil.ExtFromContext(ctx)
|
||||||
|
if ext != nil {
|
||||||
|
ext.EditMessage(p.chatID, &tg.MessagesEditMessageRequest{
|
||||||
|
ID: p.msgID,
|
||||||
|
Message: i18n.T(i18nk.BotMsgProgressTaskCanceledWithId, map[string]any{
|
||||||
|
"TaskID": task.TaskID(),
|
||||||
|
}),
|
||||||
|
})
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
logger.Errorf("Aria2 task %s failed: %s", task.TaskID(), err)
|
||||||
|
ext := tgutil.ExtFromContext(ctx)
|
||||||
|
if ext != nil {
|
||||||
|
ext.EditMessage(p.chatID, &tg.MessagesEditMessageRequest{
|
||||||
|
ID: p.msgID,
|
||||||
|
Message: i18n.T(i18nk.BotMsgProgressTaskFailedWithError, map[string]any{
|
||||||
|
"Error": err.Error(),
|
||||||
|
}),
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
logger.Infof("Aria2 task %s completed successfully", task.TaskID())
|
||||||
|
|
||||||
|
entityBuilder := entity.Builder{}
|
||||||
|
if err := styling.Perform(&entityBuilder,
|
||||||
|
styling.Plain(i18n.T(i18nk.BotMsgProgressAria2Done, map[string]any{
|
||||||
|
"GID": task.GID,
|
||||||
|
})),
|
||||||
|
styling.Plain(i18n.T(i18nk.BotMsgProgressSavePathPrefix, nil)),
|
||||||
|
styling.Code(fmt.Sprintf("[%s]:%s", task.Storage.Name(), task.StorPath)),
|
||||||
|
); err != nil {
|
||||||
|
logger.Errorf("Failed to build entities: %s", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
text, entities := entityBuilder.Complete()
|
||||||
|
req := &tg.MessagesEditMessageRequest{
|
||||||
|
ID: p.msgID,
|
||||||
|
}
|
||||||
|
req.SetMessage(text)
|
||||||
|
req.SetEntities(entities)
|
||||||
|
|
||||||
|
ext := tgutil.ExtFromContext(ctx)
|
||||||
|
if ext != nil {
|
||||||
|
ext.EditMessage(p.chatID, req)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
var _ ProgressTracker = (*Progress)(nil)
|
||||||
|
|
||||||
|
func NewProgress(msgID int, userID int64) ProgressTracker {
|
||||||
|
return &Progress{
|
||||||
|
msgID: msgID,
|
||||||
|
chatID: userID,
|
||||||
|
}
|
||||||
|
}
|
||||||
61
core/tasks/aria2dl/task.go
Normal file
61
core/tasks/aria2dl/task.go
Normal file
@@ -0,0 +1,61 @@
|
|||||||
|
package aria2dl
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"fmt"
|
||||||
|
|
||||||
|
"github.com/krau/SaveAny-Bot/core"
|
||||||
|
"github.com/krau/SaveAny-Bot/pkg/aria2"
|
||||||
|
"github.com/krau/SaveAny-Bot/pkg/enums/tasktype"
|
||||||
|
"github.com/krau/SaveAny-Bot/storage"
|
||||||
|
)
|
||||||
|
|
||||||
|
var _ core.Executable = (*Task)(nil)
|
||||||
|
|
||||||
|
type Task struct {
|
||||||
|
ID string
|
||||||
|
ctx context.Context
|
||||||
|
GID string
|
||||||
|
URIs []string
|
||||||
|
Aria2Client *aria2.Client
|
||||||
|
Storage storage.Storage
|
||||||
|
StorPath string
|
||||||
|
Progress ProgressTracker
|
||||||
|
}
|
||||||
|
|
||||||
|
// Title implements core.Executable.
|
||||||
|
func (t *Task) Title() string {
|
||||||
|
return fmt.Sprintf("[%s](Aria2 GID:%s->%s:%s)", t.Type(), t.GID, t.Storage.Name(), t.StorPath)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Type implements core.Executable.
|
||||||
|
func (t *Task) Type() tasktype.TaskType {
|
||||||
|
return tasktype.TaskTypeAria2
|
||||||
|
}
|
||||||
|
|
||||||
|
// TaskID implements core.Executable.
|
||||||
|
func (t *Task) TaskID() string {
|
||||||
|
return t.ID
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewTask(
|
||||||
|
id string,
|
||||||
|
ctx context.Context,
|
||||||
|
gid string,
|
||||||
|
uris []string,
|
||||||
|
aria2Client *aria2.Client,
|
||||||
|
stor storage.Storage,
|
||||||
|
storPath string,
|
||||||
|
progressTracker ProgressTracker,
|
||||||
|
) *Task {
|
||||||
|
return &Task{
|
||||||
|
ID: id,
|
||||||
|
ctx: ctx,
|
||||||
|
GID: gid,
|
||||||
|
URIs: uris,
|
||||||
|
Aria2Client: aria2Client,
|
||||||
|
Storage: stor,
|
||||||
|
StorPath: storPath,
|
||||||
|
Progress: progressTracker,
|
||||||
|
}
|
||||||
|
}
|
||||||
Some files were not shown because too many files have changed in this diff Show More
Reference in New Issue
Block a user