#!/usr/bin/env python3

# This Source Code Form is subject to the terms of the
# Mozilla Public License, v. 2.0.
# If a copy of the MPL was not distributed with this file,
# You can obtain one at http://mozilla.org/MPL/2.0/.

import requests
import mastodon
from bs4 import BeautifulSoup
import json, re

cfg = json.load(open('config.json', 'r'))

client = mastodon.Mastodon(
	client_id=cfg['client']['id'],
	client_secret = cfg['client']['secret'], 
	access_token=cfg['secret'], 
	api_base_url=cfg['site'])

def extract_toot(toot):
	#copied from main.py, see there for comments
	soup = BeautifulSoup(toot, "html.parser")
	for lb in soup.select("br"):
		lb.insert_after("\n")
		lb.decompose()
	for p in soup.select("p"):
		p.insert_after("\n")
		p.unwrap()
	for ht in soup.select("a.hashtag"):
		ht.unwrap()
	for link in soup.select("a"):
		link.insert_after(link["href"])
		link.decompose()
	text = map(lambda a: a.strip(), soup.get_text().strip().split("\n"))
	text = "\n".join(list(text))
	text = re.sub("https?://([^/]+)/(@[^ ]+)", r"\2@\1", text) #put mentions back in
	text = re.sub("^@[^@]+@[^ ]+ *", r"", text) #...but remove the initial one
	text = text.lower() #for easier matching
	return text

class ReplyListener(mastodon.StreamListener):
	def on_notification(self, notification):
		if notification['type'] == 'mention':
			acct = "@" + notification['account']['acct']
			post_id = notification['status']['id']
			mention = extract_toot(notification['status']['content'])
			q = mention
			print(acct + " says " + mention)

			kind = 'general' #todo: support for images
			if kind == 'images':
				r = requests.get("https://searx.lynnesbian.space/?category_images=1&q={}&format=json".format(q), timeout=10)
			else:
				r = requests.get("https://searx.lynnesbian.space/?q={}&format=json".format(q), timeout=10)
			j = r.json()

			if kind == 'images':
				toot = "Here's what I found by searching for images with the query \"{}\".".format(q)
			else:
				result = j['results'][0]
				toot = result['title'] + "\n" + result['url'] + "\n" + result['content'] + "\n" + "(Score: {})\nMore results: {}{}".format(result['score'], "https://searx.lynnesbian.space/?q=", q.replace(" ", "%20"))

			toot = toot.replace("@", "@\u200B") # don't @ people

			toot = acct + " " + toot
			visibility = notification['status']['visibility']
			if visibility == "public":
				visibility = "unlisted"
			client.status_post(toot, post_id, visibility=visibility, spoiler_text = "Search result")
			print("replied with " + toot)

rl = ReplyListener()
client.stream_user(rl)