Image Hosting
8 |Upload your images and get shareable links.
9 | 10 |
11 |
14 |
18 |
19 |
20 |
21 |
22 | Drag & Drop your images here or
15 | 16 | 17 |
23 |
85 | 24 | My Gallery 25 |
26 | 27 | {% if images %} 28 |
29 |
44 |
45 |
30 |
31 |
32 |
33 |
34 |
35 | Link Format:
36 |
42 |
43 |
46 |
47 |
52 | File
48 | Size
49 | Upload Date
50 | Actions
51 |
53 | {% for image in images %}
54 |
78 | {% else %}
79 |
55 |
76 | {% endfor %}
77 |
56 |
57 |
58 |
59 | 
60 | {{ image.filename }}
61 |
62 | {{ "%.2f KB"|format(image.size / 1024) }}
63 | {{ image.upload_date }}
64 |
65 |
68 |
69 |
70 |
71 |
74 |
75 |
80 |
81 |
83 | {% endif %}
84 | Your gallery is empty. Upload some images to get started!
82 |
24 | 
25 |
26 |
27 |
28 |
29 |
30 |
31 |
32 | ## 快速开始
33 |
34 | 推荐使用 Docker 来部署 `tgState`，这是最简单快捷的方式。
35 |
36 | ### 使用 Docker 部署
37 |
38 | ```bash
39 | docker run -d \
40 | --name tgstate \
41 | -p 8000:8000 \
42 | -e BOT_TOKEN="123456:ABC-DEF1234ghIkl-zyx57W2v1u123ew11" \
43 | -e CHANNEL_NAME="@my_test_channel" \
44 | -e PASS_WORD="supersecret" \
45 | -e BASE_URL="https://my-service.com" \
46 | -e PICGO_API_KEY="supersecret(可选不需要就删除这行)" \
47 | mitu233/python-tgstate:latest
48 | ```
49 |
50 |
51 | ### 本地开发与运行
52 |
53 | 如果您希望在本地环境直接运行 `tgState`：
54 |
55 | 1. **克隆项目并进入目录**:
56 |
57 | ```bash
58 | git clone https://github.com/your-repo/python-tgstate.git
59 | cd python-tgstate
60 | ```
61 |
62 | 2. **创建并激活虚拟环境**:
63 |
64 | ```bash
65 | # 创建虚拟环境
66 | python -m venv venv
67 | # 激活虚拟环境 (Windows)
68 | venv\Scripts\activate
69 | # 激活虚拟环境 (Linux/macOS)
70 | # source venv/bin/activate
71 | ```
72 |
73 | 3. **安装依赖**:
74 |
75 | ```bash
76 | pip install -r requirements.txt
77 | ```
78 |
79 | 4. **创建 `.env` 配置文件**: 参照 Docker 部分的说明，在项目根目录创建并配置 `.env` 文件。
80 |
81 | 5. **启动应用**:
82 |
83 | ```bash
84 | uvicorn app.main:app --reload
85 | ```
86 |
87 | 如果没有配置域名应用将在 `http://127.0.0.1:8000` 上运行。
88 |
89 | ## 配置 (环境变量)
90 |
91 | `tgState` 通过环境变量进行配置，应用会从根目录的 `.env` 文件中自动加载这些变量。
92 |
93 | | 变量 | 描述 | 是否必须 | 默认值 |
94 | | --------------- | ------------------------------------------------------------ | -------- | ----------------------- |
95 | | `BOT_TOKEN` | 您的 Telegram Bot API 令牌。可以从 [@BotFather](https://t.me/BotFather) 获取。 | **是** | `None` |
96 | | `CHANNEL_NAME` | 用于文件存储的目标聊天/频道。可以是公共频道的 `@username` 。 | **是** | `None` |
97 | | `PASS_WORD` | 用于保护 Web 界面访问的密码。如果留空，则应用将公开访问，无需密码。 | 否 | `None` |
98 | | `BASE_URL` | 您的服务的公共 URL。用于生成完整的下载链接。 | 否 | `http://127.0.0.1:8000` |
99 | | `PICGO_API_KEY` | 用于 PicGo 上传接口的 API 密钥。 | 否 | `None` |
100 |
101 | ## 注意密码相关
102 |
103 | 1. **两个密码都【没有】设置**：
104 | - **处理方式**：完全开放。
105 | - **结果**：无论是通过 Picogo/API 还是网页，**均可直接上传**，无需任何凭证。
106 | 2. **只设置了【Picogo密码】**：
107 | - **结果**：来自网页的上传请求会无条件允许。来自 Picogo/API 的请求会验证后允许。
108 | 3. **只设置了【登录密码】**：
109 | - **结果**：来自网页的上传只有**已登录**的网页用户能上传。来自 Picogo/API 的请求会无条件允许。
110 |
111 | ## 使用方法
112 |
113 | 1. **访问 Web 界面**: 启动应用后，在浏览器中访问 `http://127.0.0.1:8000` (或您配置的 `BASE_URL`)。
114 |
115 | 2. **密码验证**: 如果您在 `.env` 文件中设置了 `PASS_WORD`，应用会首先跳转到密码输入页面。输入正确密码后即可访问主界面。
116 |
117 | 3. **上传文件**: 在主界面，点击上传区域选择文件，或直接将文件拖拽到上传框中。上传成功后，文件会出现在下方的文件列表中。
118 |
119 | 4. **群组上传**支持转发到群组上传，支持小于20m的文件和照片（tg官方的限制）。
120 |
121 | 5. **获取链接**: 文件列表中的每个文件都会显示其文件名、大小和上传日期。点击文件名即可复制该文件的直接下载链接。
122 |
123 | 6. **群组获取链接: **群组中回复文件get获取下载链接
124 |
125 | 
126 |
127 | ## PicGo 配置指南
128 |
129 | ### 前置要求
130 |
131 | 1. 安装插件web-uploader
132 |
133 | ### 核心配置
134 |
135 | 1. **图床配置名**: 随便
136 |
137 | 2. **api地址**: 填写你的服务提供的上传地址。本地是 `http://<你的服务器IP>:8000/api/upload`。
138 |
139 | 3. **POST参数名**: `file`。
140 |
141 | 4. **JSON 参数名**: `url`。
142 |
143 | 5. **自定义请求头**：{"x-api-key": "PICGO_API_KEY"}（可选推荐）(验证方式二选一)
144 |
145 | 6. **自定义body**：{"key":"PICGO_API_KEY"}（可选推荐）(验证方式二选一)
146 |
147 | 
148 |
149 |
150 |
151 | ### 目前缺点
152 |
153 | 1. 删除文件慢，群组内删除前端不会更新需要手动删除。
154 |
155 | 用 roocode 和 白嫖的心 制作。****
--------------------------------------------------------------------------------
/app/bot_handler.py:
--------------------------------------------------------------------------------
1 | import asyncio
2 | import json
3 | from telegram import Update
4 | from telegram.ext import Application, MessageHandler, filters, ContextTypes
5 | from .core.config import get_settings
6 | from .services.telegram_service import get_telegram_service
7 | from . import database
8 | from .events import file_update_queue
9 |
10 | async def handle_new_file(update: Update, context: ContextTypes.DEFAULT_TYPE):
11 | """
12 | 处理新增的文件或照片，将其元数据存入数据库，并通过队列发送通知。
13 | 在函数内部检查消息来源是否为授权的聊天（私聊、群组或频道）。
14 | """
15 | settings = get_settings()
16 | message = update.message or update.channel_post
17 |
18 | # 1. 确保有消息
19 | if not message:
20 | return
21 |
22 | # 2. 检查消息来源是否为指定的频道/群组
23 | channel_identifier = settings.CHANNEL_NAME
24 | if not channel_identifier:
25 | print("错误: CHANNEL_NAME 未在 .env 中设置，无法处理文件。")
26 | return
27 |
28 | chat = message.chat
29 | is_allowed = False
30 | # 检查是公开频道 (e.g., "@username") 还是私密频道 (e.g., "-100123456789")
31 | if channel_identifier.startswith('@'):
32 | if chat.username and chat.username == channel_identifier.lstrip('@'):
33 | is_allowed = True
34 | else:
35 | if str(chat.id) == channel_identifier:
36 | is_allowed = True
37 |
38 | if not is_allowed:
39 | return
40 |
41 | # 3. 确定文件/照片信息
42 | file_obj = None
43 | file_name = None
44 |
45 | if message.document:
46 | file_obj = message.document
47 | file_name = file_obj.file_name
48 | elif message.photo:
49 | # 选择分辨率最高的照片
50 | file_obj = message.photo[-1]
51 | # 为照片创建一个默认文件名
52 | file_name = f"photo_{message.message_id}.jpg"
53 |
54 | # 4. 如果成功获取到文件或照片对象，则处理它
55 | if file_obj and file_name:
56 | # 我们只关心小于20MB的、非清单文件
57 | if file_obj.file_size < (20 * 1024 * 1024) and not file_name.endswith('.manifest'):
58 | # 使用复合ID "message_id:file_id"
59 | composite_id = f"{message.message_id}:{file_obj.file_id}"
60 |
61 | database.add_file_metadata(
62 | filename=file_name,
63 | file_id=composite_id,
64 | filesize=file_obj.file_size
65 | )
66 |
67 | # 将新文件信息放入队列，以便 SSE 端点可以推送给前端
68 | file_info = {
69 | "action": "add",
70 | "filename": file_name,
71 | "file_id": composite_id,
72 | "filesize": file_obj.file_size,
73 | "upload_date": message.date.isoformat()
74 | }
75 | await file_update_queue.put(json.dumps(file_info))
76 |
77 | async def handle_get_reply(update: Update, context: ContextTypes.DEFAULT_TYPE):
78 | """
79 | 处理对文件消息回复 "get" 的情况。
80 | """
81 | if not (update.message and update.message.reply_to_message and (update.message.reply_to_message.document or update.message.reply_to_message.photo)):
82 | return
83 |
84 | # 检查回复的文本是否完全是 "get"
85 | if update.message.text.lower().strip() != 'get':
86 | return
87 |
88 | document = update.message.reply_to_message.document or update.message.reply_to_message.photo[-1]
89 | file_id = document.file_id
90 | file_name = getattr(document, 'file_name', f"photo_{update.message.reply_to_message.message_id}.jpg")
91 | settings = get_settings()
92 |
93 | final_file_id = file_id
94 | final_file_name = file_name
95 |
96 | # 如果是清单文件，我们需要解析它以获取原始文件名
97 | if file_name.endswith('.manifest'):
98 | telegram_service = get_telegram_service()
99 | download_url = await telegram_service.get_download_url(file_id)
100 | if download_url:
101 | # 在实际应用中，我们应该流式处理而不是一次性下载
102 | # 但为了简化，我们先下载清单内容
103 | import httpx
104 | async with httpx.AsyncClient() as client:
105 | try:
106 | resp = await client.get(download_url)
107 | resp.raise_for_status()
108 | content = resp.content
109 | if content.startswith(b'tgstate-blob\n'):
110 | lines = content.decode('utf-8').strip().split('\n')
111 | final_file_name = lines[1] # 获取原始文件名
112 | # file_id 保持为清单文件的 ID，下载路由会处理它
113 | except httpx.RequestError as e:
114 | print(f"下载清单文件时出错: {e}")
115 | await update.message.reply_text("错误：无法获取清单文件内容。")
116 | return
117 |
118 | file_path = f"/d/{final_file_id}"
119 |
120 | if settings.BASE_URL:
121 | download_link = f"{settings.BASE_URL.strip('/')}{file_path}"
122 | reply_text = f"这是 '{final_file_name}' 的下载链接:\n{download_link}"
123 | else:
124 | reply_text = f"这是 '{final_file_name}' 的下载路径 (请自行拼接域名):\n`{file_path}`"
125 |
126 | await update.message.reply_text(reply_text)
127 |
128 | async def handle_deleted_message(update: Update, context: ContextTypes.DEFAULT_TYPE):
129 | """
130 | 处理消息删除事件，同步删除数据库中的文件记录。
131 | """
132 | # 在 `python-telegram-bot` v20+ 中，删除事件是通过 `update.edited_message` 捕获的，
133 | # 当消息被删除时，它会变成一个内容为空的 `edited_message`。
134 | # 我们通过检查 `update.edited_message` 是否存在来判断消息是否被删除。
135 | if update.edited_message and not update.edited_message.text:
136 | message_id = update.edited_message.message_id
137 | deleted_file_id = database.delete_file_by_message_id(message_id)
138 | if deleted_file_id:
139 | # 如果成功删除了数据库记录，就通知前端
140 | delete_info = {
141 | "action": "delete",
142 | "file_id": deleted_file_id
143 | }
144 | await file_update_queue.put(json.dumps(delete_info))
145 |
146 | def create_bot_app() -> Application:
147 | """
148 | 创建并配置 Telegram Bot 应用实例。
149 | """
150 | settings = get_settings()
151 | if not settings.BOT_TOKEN:
152 | print("错误: .env 文件中未设置 BOT_TOKEN。机器人无法创建。")
153 | raise ValueError("BOT_TOKEN not configured.")
154 |
155 | application = Application.builder().token(settings.BOT_TOKEN).build()
156 |
157 | # --- 添加处理器 ---
158 |
159 | # 1. 处理对文件消息回复 "get" 的情况 (在任何地方)
160 | get_handler = MessageHandler(
161 | filters.TEXT & (~filters.COMMAND) & filters.REPLY,
162 | handle_get_reply
163 | )
164 | application.add_handler(get_handler)
165 |
166 | # 2. 处理所有文件和照片消息
167 | new_file_handler = MessageHandler(filters.ALL, handle_new_file)
168 | application.add_handler(new_file_handler, group=0)
169 |
170 | # 3. 处理消息删除事件
171 | # 注意：机器人需要有管理员权限才能接收到此事件
172 | delete_handler = MessageHandler(filters.UpdateType.EDITED_MESSAGE, handle_deleted_message)
173 | application.add_handler(delete_handler, group=1)
174 |
175 | return application
176 |
--------------------------------------------------------------------------------
/app/templates/download.html:
--------------------------------------------------------------------------------
1 | {% extends "base.html" %}
2 |
3 | {% block title %}Download {{ file.filename }}{% endblock %}
4 |
5 | {% block content %}
6 |
133 |
134 |