#!/usr/bin/python3
#----------------------------------------------------------------------
#
# systemd service...
#
# see:
#	https://tecadmin.net/run-shell-script-as-systemd-service/
#	https://tecadmin.net/setup-autorun-python-script-using-systemd/
#
#
#----------------------------------------------------------------------

import telebot
import vk_api
import requests
import os
import io
import datetime
import re


#----------------------------------------------------------------------
# Helpers...

# XXX HACK -- mini/cute html2 -> markdown parser...
def html2md(text):
	text = re.sub(r'<b>(.*)</b>', r'*\1*', text, flags=re.I|re.M)
	text = re.sub(r'<i>(.*)</i>', r'_\1_', text, flags=re.I|re.M)
	text = re.sub(r'<a\s+[^>]*href="(.*)"[^>]*>(.*)</a>', r'[\2](\1)', text, flags=re.I|re.M)
	return text

def load(name):
	return open(os.path.join(BASE_PATH, name), 'r').read().strip()

def getDate():
	return datetime.datetime.now().strftime('%Y%m%d %H:%M:%S')


#----------------------------------------------------------------------

BASE_PATH = os.path.dirname(
	os.path.abspath(__file__))


telegram_token = load('telegram.token')
vk_token = load('vk.token')
vk_group_id = load('vk_group.id')


TEXT_SNIP = 50


# Content types supported: 
#	  "text", 
#	  "audio", 
#	  "document", 
#	  "photo", 
#	  "sticker", 
#	  "video", 
#	  "video_note", 
#	  "voice", 
#	  "location", 
#	  "contact",
#	  "new_chat_members",
#	  "left_chat_member",
#	  "new_chat_title",
#	  "new_chat_photo",
#	  "delete_chat_photo",
#	  "group_chat_created",
#	  "supergroup_chat_created",
#	  "channel_chat_created",
#	  "migrate_to_chat_id",
#	  "migrate_from_chat_id",
#	  "pinned_message",
TRACK_TYPES = [
	"text", 
	"photo", 
]


bot = telebot.TeleBot(telegram_token)
vk_session = vk_api.VkApi(token=vk_token)
vk = vk_session.get_api()




#----------------------------------------------------------------------
# Text post handler...
#
# XXX TODO:
#	- for some reason this does not see attached images at all...
#		 message.photo == None...
#	- get telegram formatting (markdown???)
#		(see notes for: repost_photo_to_vk(..) below)
@bot.channel_post_handler(content_types=['text'])
def repost_text_to_vk(message):
	print(f"{getDate()}: Received message: {(message.text or '')[:TEXT_SNIP]}")

	attachments = []

	# XXX for some reason we are not getting any photos here...
	print(f"  --- PHOTOS: {message.photo}")
	if message.photo:
		for photo in message.photo:
			# Скачиваем фото
			file_info = bot.get_file(photo.file_id)
			photo_url = "https://api.telegram.org/file/bot{}/{}".format(
				telegram_token, file_info.file_path)

			# Загружаем фото в ВКонтакте
			photo_id = upload_photo_to_vk(vk, photo_url)
			attachments.append(photo_id)

			print(f"  --- PHOTO {len(attachments)}: {attachments[-1]}")

	# Публикуем сообщение в группе ВКонтакте
	vk.wall.post(
		owner_id="-" + vk_group_id,
		message=message.text,
		attachments=",".join(attachments),
	)
	print(f"{getDate()}: Posted message: {(message.text or '')[:TEXT_SNIP]}")



#----------------------------------------------------------------------
# Image post handler...
#
# see:
#	https://github.com/python273/vk_api/blob/master/examples/upload_photo.py
#	https://vk-api.readthedocs.io/en/latest/upload.html (see: photo_wall(..))
#
# XXX TODO
#	- multiple images per post are treated	as separate posts for some reason -- need 
#		to find a way to group...
#	- get telegram formatting:
#		- links
#		- header
#		- ...
#	- Q: is the largest image alwways the last??
#		...check telegram API
@bot.channel_post_handler(content_types=['photo'])
def repost_photo_to_vk(message):
	print(f"{getDate()}: Received photo: {(message.caption or '')[:TEXT_SNIP]}")

	# download images (memory)...
	upload = vk_api.VkUpload(vk_session)
	ids = []
	photo = message.photo[-1]
	file_info = bot.get_file(photo.file_id)
	photo_url = "https://api.telegram.org/file/bot{}/{}".format(
		telegram_token, 
		file_info.file_path)
	response = requests.get(photo_url)

	# upload images...
	photo_info = upload.photo_wall(
		io.BytesIO(response.content),
		group_id=vk_group_id)[0]

	ids += [
		"photo{}_{}".format(
			photo_info["owner_id"], 
			photo_info["id"]) ]

	print(f"  --- PHOTO {len(ids)}: {ids[-1]}")

	vk.wall.post(
		owner_id="-" + vk_group_id,
		message=message.caption,
		#message=message.html_caption,
		attachments=",".join(ids))

	print(f"{getDate()}: Posted photo: {(message.caption or '')[:TEXT_SNIP]}")



#----------------------------------------------------------------------
# Start...

print(f'{getDate()}: Bot started...')

bot.polling(none_stop=False, interval=0, timeout=60)



#----------------------------------------------------------------------
# vim:set ts=4 sw=4 :