gd-utils/src/tg.js

351 lines
14 KiB
JavaScript
Raw Normal View History

2020-06-27 14:19:43 +08:00
const Table = require('cli-table3')
const dayjs = require('dayjs')
const axios = require('@viegg/axios')
const HttpsProxyAgent = require('https-proxy-agent')
const { db } = require('../db')
2020-07-04 01:12:20 +08:00
const { gen_count_body, validate_fid, real_copy, get_name_by_id } = require('./gd')
2020-07-06 18:48:19 +08:00
const { AUTH, DEFAULT_TARGET, USE_PERSONAL_AUTH } = require('../config')
2020-06-27 14:19:43 +08:00
const { tg_token } = AUTH
2020-07-04 01:12:20 +08:00
const gen_link = (fid, text) => `<a href="https://drive.google.com/drive/folders/${fid}">${text || fid}</a>`
2020-06-27 14:19:43 +08:00
2020-07-06 10:27:46 +08:00
if (!tg_token) throw new Error('請先在config.js中設定tg_token')
2020-06-27 14:19:43 +08:00
const { https_proxy } = process.env
const axins = axios.create(https_proxy ? { httpsAgent: new HttpsProxyAgent(https_proxy) } : {})
2020-07-04 01:12:20 +08:00
const FID_TO_NAME = {}
async function get_folder_name (fid) {
let name = FID_TO_NAME[fid]
if (name) return name
name = await get_name_by_id(fid)
return FID_TO_NAME[fid] = name
}
2020-06-27 14:19:43 +08:00
function send_help (chat_id) {
const text = `<pre>[使用說明]
***不支持單檔分享***
命令 說明
2020-07-06 18:48:19 +08:00
=====================
/help | 使
2020-07-06 18:48:19 +08:00
=====================
/count sourceID [-u] | sourceID
sourceID可以是共享網址本身也可以是共享ID如果命令最后加上 -u則無視快取記錄強制從線上獲取適合一段時候後才更新完畢的分享連結
=====================
/copy sourceID targetID() [-u] | sourceIDtargetID
若無targetID則會複製到預設位置config.js中的DEFAULT_TARGET
如果設定了bookmark那麼targetID也可以是bookmark的標籤名
如果命令最後加上 -u則無視快取記錄強制從線上獲取源資料夾資訊返回拷貝任務的taskID
=====================
/task taskID() | taskID
若填 all 則返回所有任務列表(歷史紀錄)
=====================
/bm [action] [alias] [target] | bookmarkID
會在輸入共享連結後返回的文件統計開始複製這兩個按鈕的下方出現方便複製到常用位置
範例
/bm |
/bm set movie folder-id | folder-idmovie
/bm unset movie |
</pre>`
return sm({ chat_id, text, parse_mode: 'HTML' })
}
2020-06-27 14:19:43 +08:00
2020-07-06 18:48:19 +08:00
function send_bm_help (chat_id) {
const text = `<pre>/bm [action] [alias] [target] | bookmark添加常用目的資料夾ID
會在輸入共享連結後返回的文件統計開始複製這兩個按鈕的下方出現方便複製到常用位置
範例
/bm |
/bm set movie folder-id | folder-idmovie
/bm unset movie |
2020-06-27 14:19:43 +08:00
</pre>`
return sm({ chat_id, text, parse_mode: 'HTML' })
}
2020-07-06 18:48:19 +08:00
function send_all_bookmarks (chat_id) {
let records = db.prepare('select alias, target from bookmark').all()
if (!records.length) return sm({ chat_id, text: '資料庫中沒有收藏紀錄' })
const tb = new Table({ style: { head: [], border: [] } })
const headers = ['標籤名', 'dstID']
records = records.map(v => [v.alias, v.target])
tb.push(headers, ...records)
const text = tb.toString().replace(/─/g, '—')
return sm({ chat_id, text: `<pre>${text}</pre>`, parse_mode: 'HTML' })
}
function set_bookmark ({ chat_id, alias, target }) {
const record = db.prepare('select alias from bookmark where alias=?').get(alias)
if (record) return sm({ chat_id, text: '資料庫中已有同名的收藏' })
db.prepare('INSERT INTO bookmark (alias, target) VALUES (?, ?)').run(alias, target)
return sm({ chat_id, text: `成功設定收藏${alias} | ${target}` })
}
function unset_bookmark ({ chat_id, alias }) {
const record = db.prepare('select alias from bookmark where alias=?').get(alias)
if (!record) return sm({ chat_id, text: '未找到此標籤名的收藏' })
db.prepare('delete from bookmark where alias=?').run(alias)
return sm({ chat_id, text: '成功刪除收藏 ' + alias })
}
function get_target_by_alias (alias) {
const record = db.prepare('select target from bookmark where alias=?').get(alias)
return record && record.target
}
2020-06-27 14:19:43 +08:00
function send_choice ({ fid, chat_id }) {
if(BUTTON_LEVEL == 1){
return sm({
2020-06-27 14:19:43 +08:00
chat_id,
text: `辨識到分享ID ${fid},請選擇動作`,
2020-06-27 14:19:43 +08:00
reply_markup: {
inline_keyboard: [
[
2020-07-06 21:18:00 +08:00
{ text: '文件統計', callback_data: `count ${fid}` }
],
[
2020-07-06 18:48:19 +08:00
{ text: '開始複製', callback_data: `copy ${fid}` }
2020-06-27 14:19:43 +08:00
]
2020-07-06 18:48:19 +08:00
].concat(gen_bookmark_choices(fid))
2020-06-27 14:19:43 +08:00
}
})
}else{
return sm({
chat_id,
text: `辨識到分享ID ${fid},請選擇動作`,
reply_markup: {
inline_keyboard: [
[
{ text: '文件統計', callback_data: `count ${fid}` },
{ text: '開始複製', callback_data: `copy ${fid}` }
]
].concat(gen_bookmark_choices(fid))
}
})
}
2020-06-27 14:19:43 +08:00
}
2020-07-06 18:48:19 +08:00
// console.log(gen_bookmark_choices())
function gen_bookmark_choices (fid) {
let level = 1
if (BUTTON_LEVEL > 2){
level = 2
}else{
level = BUTTON_LEVEL
}
2020-07-06 18:48:19 +08:00
const gen_choice = v => ({text: `複製到 ${v.alias}`, callback_data: `copy ${fid} ${v.alias}`})
const records = db.prepare('select * from bookmark').all()
const result = []
2020-07-06 21:18:00 +08:00
for (let i = 0; i < records.length; i++) {
2020-07-06 18:48:19 +08:00
const line = [gen_choice(records[i])]
for(let j = 0; j < level-1; j ++){
if (records[i+1]) line.push(gen_choice(records[i+1]))
i++
}
2020-07-06 18:48:19 +08:00
result.push(line)
}
return result
}
2020-06-27 14:19:43 +08:00
async function send_all_tasks (chat_id) {
let records = db.prepare('select id, status, ctime from task').all()
if (!records.length) return sm({ chat_id, text: '資料庫中沒有任務記錄' })
2020-06-27 14:19:43 +08:00
const tb = new Table({ style: { head: [], border: [] } })
const headers = ['ID', 'status', 'ctime']
records = records.map(v => {
const { id, status, ctime } = v
2020-06-27 14:25:50 +08:00
return [id, status, dayjs(ctime).format('YYYY-MM-DD HH:mm:ss')]
2020-06-27 14:19:43 +08:00
})
tb.push(headers, ...records)
const text = tb.toString().replace(/─/g, '—')
const url = `https://api.telegram.org/bot${tg_token}/sendMessage`
return axins.post(url, {
chat_id,
parse_mode: 'HTML',
text: `所有拷貝任務:\n<pre>${text}</pre>`
2020-07-04 01:12:20 +08:00
}).catch(err => {
// const description = err.response && err.response.data && err.response.data.description
// if (description && description.includes('message is too long')) {
if (true) {
2020-06-27 14:19:43 +08:00
const text = [headers].concat(records).map(v => v.join('\t')).join('\n')
return sm({ chat_id, parse_mode: 'HTML', text: `所有拷貝任務:\n<pre>${text}</pre>` })
2020-06-27 14:19:43 +08:00
}
console.error(err)
})
}
2020-07-04 01:12:20 +08:00
async function get_task_info (task_id) {
2020-06-27 14:19:43 +08:00
const record = db.prepare('select * from task where id=?').get(task_id)
2020-07-04 01:12:20 +08:00
if (!record) return {}
2020-06-27 14:19:43 +08:00
const { source, target, status, copied, mapping, ctime, ftime } = record
2020-07-04 01:12:20 +08:00
const folder_mapping = mapping && mapping.trim().split('\n')
const new_folder = folder_mapping && folder_mapping[0].split(' ')[1]
2020-06-27 14:19:43 +08:00
const { summary } = db.prepare('select summary from gd where fid=?').get(source) || {}
const { file_count, folder_count, total_size } = summary ? JSON.parse(summary) : {}
const copied_files = copied ? copied.trim().split('\n').length : 0
2020-07-04 01:12:20 +08:00
const copied_folders = folder_mapping ? (folder_mapping.length - 1) : 0
let text = '任務ID' + task_id + '\n'
2020-07-04 01:12:20 +08:00
const folder_name = await get_folder_name(source)
text += '源資料夾:' + gen_link(source, folder_name) + '\n'
2020-07-04 01:12:20 +08:00
text += '目的位置:' + gen_link(target) + '\n'
text += '新資料夾:' + (new_folder ? gen_link(new_folder) : '尚未創建') + '\n'
text += '任務狀態:' + status + '\n'
text += '創建時間:' + dayjs(ctime).format('YYYY-MM-DD HH:mm:ss') + '\n'
text += '完成時間:' + (ftime ? dayjs(ftime).format('YYYY-MM-DD HH:mm:ss') : '未完成') + '\n'
var pct = copied_folders/(folder_count === undefined ? '未知數量' : folder_count)*100
pct = pct.toFixed(2);
text += '目錄進度:' + copied_folders + '/' + (folder_count === undefined ? '未知數量' : folder_count) + ' - ' + pct + '%\n'
pct = copied_files/(file_count === undefined ? '未知數量' : file_count)*100
pct = pct.toFixed(2);
text += '文件進度:' + copied_files + '/' + (file_count === undefined ? '未知數量' : file_count) + ' - ' + pct + '%\n'
text += '合計大小:' + (total_size || '未知大小')
2020-07-04 01:12:20 +08:00
const total_count = (folder_count || 0) + (file_count || 0)
return { text, status, total_count }
2020-06-27 14:19:43 +08:00
}
2020-07-04 01:12:20 +08:00
async function send_task_info ({ task_id, chat_id }) {
2020-07-06 18:48:19 +08:00
const { text, status, folder_count } = await get_task_info(task_id)
if (!text) return sm({ chat_id, text: '資料庫查無此任務ID' + task_id })
2020-07-04 01:12:20 +08:00
const url = `https://api.telegram.org/bot${tg_token}/sendMessage`
let message_id
try {
const { data } = await axins.post(url, { chat_id, text, parse_mode: 'HTML' })
message_id = data && data.result && data.result.message_id
} catch (e) {
console.log('fail to send message to tg', e.message)
}
2020-07-06 18:48:19 +08:00
// get_task_info 在task目录数超大时比较吃cpu如果超1万就不每10秒更新了以后如果把mapping 也另存一张表可以取消此限制
if (!message_id || status !== 'copying' || folder_count > 10000) return
2020-07-04 01:12:20 +08:00
const loop = setInterval(async () => {
const url = `https://api.telegram.org/bot${tg_token}/editMessageText`
const { text, status } = await get_task_info(task_id)
if (status !== 'copying') clearInterval(loop)
axins.post(url, { chat_id, message_id, text, parse_mode: 'HTML' }).catch(e => console.error(e.message))
}, 10 * 1000)
}
async function tg_copy ({ fid, target, chat_id, update }) { // return task_id
2020-06-27 14:19:43 +08:00
target = target || DEFAULT_TARGET
if (!target) {
sm({ chat_id, text: '請輸入目的地ID或先在config.js中設定預設複製的目的地ID(DEFAULT_TARGET)' })
2020-06-27 14:19:43 +08:00
return
}
let record = db.prepare('select id, status from task where source=? and target=?').get(fid, target)
if (record) {
if (record.status === 'copying') {
2020-07-06 18:48:19 +08:00
sm({ chat_id, text: '已有相同來源ID和目的ID的任務正在進行查詢進度可輸入 /task ' + record.id })
2020-06-27 14:19:43 +08:00
return
} else if (record.status === 'finished') {
sm({ chat_id, text: `檢測到已存在的任務 ${record.id},開始繼續拷貝` })
2020-06-27 14:19:43 +08:00
}
}
2020-07-06 18:48:19 +08:00
real_copy({ source: fid, update, target, service_account: !USE_PERSONAL_AUTH, is_server: true })
2020-07-04 01:12:20 +08:00
.then(async info => {
2020-06-27 14:19:43 +08:00
if (!record) record = {} // 防止无限循环
2020-07-04 01:12:20 +08:00
if (!info) return
const { task_id } = info
2020-07-06 18:48:19 +08:00
const { text } = await get_task_info(task_id)
2020-07-04 01:12:20 +08:00
sm({ chat_id, text, parse_mode: 'HTML' })
2020-06-27 14:19:43 +08:00
})
.catch(err => {
const task_id = record && record.id
if (task_id){
db.prepare('update task set status=? where id=?').run('error', task_id)
}
2020-06-27 14:19:43 +08:00
if (!record) record = {}
console.error('複製失敗', fid, '-->', target)
2020-06-27 14:19:43 +08:00
console.error(err)
sm({ chat_id, text: '複製失敗,失敗訊息:' + err.message })
2020-06-27 14:19:43 +08:00
})
while (!record) {
record = db.prepare('select id from task where source=? and target=?').get(fid, target)
await sleep(1000)
}
return record.id
}
function sleep (ms) {
return new Promise((resolve, reject) => {
setTimeout(resolve, ms)
})
}
function reply_cb_query ({ id, data }) {
const url = `https://api.telegram.org/bot${tg_token}/answerCallbackQuery`
return axins.post(url, {
callback_query_id: id,
text: '開始執行 ' + data
2020-06-27 14:19:43 +08:00
})
}
2020-07-04 01:12:20 +08:00
async function send_count ({ fid, chat_id, update }) {
2020-07-06 18:48:19 +08:00
sm({ chat_id, text: `開始獲取 ${fid} 所有檔案資訊,請稍後,建議統計完成前先不要開始複製,因为複製也需要先獲取來源資料夾資訊` })
const table = await gen_count_body({ fid, update, type: 'tg', service_account: !USE_PERSONAL_AUTH })
if (!table) return sm({ chat_id, parse_mode: 'HTML', text: gen_link(fid) + ' 資訊獲取失敗' })
2020-06-27 14:19:43 +08:00
const url = `https://api.telegram.org/bot${tg_token}/sendMessage`
const gd_link = `https://drive.google.com/drive/folders/${fid}`
2020-07-04 01:12:20 +08:00
const name = await get_folder_name(fid)
2020-06-27 14:19:43 +08:00
return axins.post(url, {
chat_id,
parse_mode: 'HTML',
text: `<pre>源資料夾名稱:${name}
源連結${gd_link}
2020-06-27 14:19:43 +08:00
${table}</pre>`
}).catch(async err => {
2020-07-04 01:12:20 +08:00
// const description = err.response && err.response.data && err.response.data.description
// const too_long_msgs = ['request entity too large', 'message is too long']
// if (description && too_long_msgs.some(v => description.toLowerCase().includes(v))) {
if (true) {
2020-07-06 18:48:19 +08:00
const smy = await gen_count_body({ fid, type: 'json', service_account: !USE_PERSONAL_AUTH })
2020-06-27 14:19:43 +08:00
const { file_count, folder_count, total_size } = JSON.parse(smy)
return sm({
chat_id,
parse_mode: 'HTML',
text: `連結:<a href="https://drive.google.com/drive/folders/${fid}">${fid}</a>\n<pre>
表格太長超出telegram訊息限制僅顯示概要
目錄名稱${name}
文件總數${file_count}
目錄總數${folder_count}
合計大小${total_size}
2020-06-27 14:19:43 +08:00
</pre>`
})
}
throw err
})
}
function sm (data) {
const url = `https://api.telegram.org/bot${tg_token}/sendMessage`
return axins.post(url, data).catch(err => {
2020-07-04 01:12:20 +08:00
// console.error('fail to post', url, data)
console.error('fail to send message to tg:', err.message)
2020-06-27 14:19:43 +08:00
})
}
function extract_fid (text) {
2020-07-04 01:12:20 +08:00
text = text.replace(/^\/count/, '').replace(/^\/copy/, '').replace(/\\/g, '').trim()
2020-06-27 14:19:43 +08:00
const [source, target] = text.split(' ').map(v => v.trim())
if (validate_fid(source)) return source
try {
if (!text.startsWith('http')) text = 'https://' + text
const u = new URL(text)
if (u.pathname.includes('/folders/')) {
2020-07-06 18:48:19 +08:00
const reg = /[^/?]+$/
2020-06-27 14:19:43 +08:00
const match = u.pathname.match(reg)
2020-07-04 01:12:20 +08:00
return match && match[0]
2020-06-27 14:19:43 +08:00
}
return u.searchParams.get('id')
} catch (e) {
return ''
}
}
2020-07-04 01:12:20 +08:00
function extract_from_text (text) {
const reg = /https?:\/\/drive.google.com\/[^\s]+/g
const m = text.match(reg)
return m && extract_fid(m[0])
}
2020-07-06 18:48:19 +08:00
module.exports = { send_count, send_help, sm, extract_fid, reply_cb_query, send_choice, send_task_info, send_all_tasks, tg_copy, extract_from_text, get_target_by_alias, send_bm_help, send_all_bookmarks, set_bookmark, unset_bookmark }