100 lines
3.5 KiB
Lua
100 lines
3.5 KiB
Lua
local gdrive = {}
|
|
|
|
local utilities = require('otouto.utilities')
|
|
local https = require('ssl.https')
|
|
local ltn12 = require('ltn12')
|
|
local json = require('dkjson')
|
|
local bindings = require('otouto.bindings')
|
|
|
|
function gdrive:init(config)
|
|
if not cred_data.google_apikey then
|
|
print('Missing config value: google_apikey.')
|
|
print('gdrive.lua will not be enabled.')
|
|
return
|
|
end
|
|
|
|
gdrive.triggers = {
|
|
"docs.google.com/(.*)/d/([A-Za-z0-9-_-]+)",
|
|
"drive.google.com/(.*)/d/([A-Za-z0-9-_-]+)",
|
|
"drive.google.com/(open)%?id=([A-Za-z0-9-_-]+)"
|
|
}
|
|
end
|
|
|
|
local apikey = cred_data.google_apikey
|
|
|
|
local BASE_URL = 'https://www.googleapis.com/drive/v2'
|
|
|
|
function gdrive:get_drive_document_data (docid)
|
|
local apikey = cred_data.google_apikey
|
|
local url = BASE_URL..'/files/'..docid..'?key='..apikey..'&fields=id,title,mimeType,ownerNames,exportLinks,fileExtension'
|
|
local res,code = https.request(url)
|
|
local res = string.gsub(res, 'image/', '')
|
|
local res = string.gsub(res, 'application/', '')
|
|
if code ~= 200 then return nil end
|
|
local data = json.decode(res)
|
|
return data
|
|
end
|
|
|
|
function gdrive:send_drive_document_data(data, self, msg)
|
|
local title = data.title
|
|
local mimetype = data.mimeType
|
|
local id = data.id
|
|
local owner = data.ownerNames[1]
|
|
local text = '"'..title..'", freigegeben von '..owner
|
|
if data.exportLinks then
|
|
if data.exportLinks.png then
|
|
local image_url = data.exportLinks.png
|
|
utilities.send_typing(self, msg.chat.id, 'upload_photo')
|
|
local file = download_to_file(image_url, 'photo.png')
|
|
utilities.send_photo(self, msg.chat.id, file, text, msg.message_id)
|
|
return
|
|
else
|
|
local pdf_url = data.exportLinks.pdf
|
|
utilities.send_typing(self, msg.chat.id, 'upload_document')
|
|
local file = download_to_file(pdf_url, 'document.pdf')
|
|
utilities.send_document(self, msg.chat.id, file, text, msg.message_id)
|
|
return
|
|
end
|
|
else
|
|
local get_file_url = 'https://drive.google.com/uc?id='..id
|
|
local keyboard = '{"inline_keyboard":[[{"text":"Direktlink","url":"'..get_file_url..'"}]]}'
|
|
local ext = data.fileExtension
|
|
if mimetype == "png" or mimetype == "jpg" or mimetype == "jpeg" or mimetype == "gif" or mimetype == "webp" then
|
|
local respbody = {}
|
|
local options = {
|
|
url = get_file_url,
|
|
sink = ltn12.sink.table(respbody),
|
|
redirect = false
|
|
}
|
|
local response = {https.request(options)} -- luasec doesn't support 302 redirects, so we must contact gdrive again
|
|
local code = response[2]
|
|
local headers = response[3]
|
|
local file_url = headers.location
|
|
if ext == "jpg" or ext == "jpeg" or ext == "png" then
|
|
utilities.send_typing(self, msg.chat.id, 'upload_photo')
|
|
local file = download_to_file(file_url, 'photo.'..ext)
|
|
utilities.send_photo(self, msg.chat.id, file, text, msg.message_id, keyboard)
|
|
return
|
|
else
|
|
utilities.send_typing(self, msg.chat.id, 'upload_document')
|
|
local file = download_to_file(file_url, 'document.'..ext)
|
|
utilities.send_document(self, msg.chat.id, file, text, msg.message_id, keyboard)
|
|
return
|
|
end
|
|
else
|
|
local text = '*'..title..'*, freigegeben von _'..owner..'_'
|
|
utilities.send_reply(self, msg, text, true, keyboard)
|
|
return
|
|
end
|
|
end
|
|
end
|
|
|
|
function gdrive:action(msg, config, matches)
|
|
local docid = matches[2]
|
|
local data = gdrive:get_drive_document_data(docid)
|
|
if not data then utilities.send_reply(self, msg, config.errors.connection) return end
|
|
gdrive:send_drive_document_data(data, self, msg)
|
|
return
|
|
end
|
|
|
|
return gdrive |