From 309c4111efbdf8bcb8a33d9e8eefbe424fd75284 Mon Sep 17 00:00:00 2001
From: Zergling_man <zerglingman@fedora.email>
Date: Mon, 2 Jan 2023 21:34:54 +1100
Subject: [PATCH] init commit

---
 birds.py        | 49 +++++++++++++++++++++++++++++++++++++++++++++++++
 example_conf.py |  5 +++++
 gelb.py         | 25 +++++++++++++++++++++++++
 3 files changed, 79 insertions(+)
 create mode 100644 birds.py
 create mode 100644 example_conf.py
 create mode 100644 gelb.py

diff --git a/birds.py b/birds.py
new file mode 100644
index 0000000..cae5d5a
--- /dev/null
+++ b/birds.py
@@ -0,0 +1,49 @@
+#!/bin/python
+
+from fedibot import main, api
+import gelb
+import conf
+import random
+api.instance=conf.instance
+api.token=conf.token
+gelb.conf=conf.gelb
+
+tags=conf.tags
+pointer=int(random.random()*len(tags))
+
+@main.loop(20*60)
+async def shitpost():
+	global pointer
+	tag=tags[pointer]
+	pointer+=1; pointer%=len(tags)
+	img,name,src,ecchi=await gelb.getgelbimg(tag)
+	await api.post(f'#{tag} {src}',[(name,img)],{'sensitive':ecchi})
+
+def unescape(thing):
+	out=[]
+	m=len(thing)
+	i=0
+	while i<m:
+		if thing[i]=='_':
+			try:
+				out.append(chr(int(thing[i+1:i+3],16)))
+				i+=3
+				continue
+			except ValueError: pass
+		out.append(thing[i])
+		i+=1
+	return ''.join(out)
+
+@main.loop(30)
+async def reeepost():
+	async for notif in api.stream('GET','notifications',{'included_types[]':['mention'],'types[]':['mention']},lambda x:{'max_id':x[-1]['id']}):
+		s=notif['status']
+		tag=s['tags']
+		if not tag: continue
+		tag=unescape(tag[0]['name'])
+		img,name,src,ecchi=await gelb.getgelbimg(tag)
+		if not img: await api.post(f'no images in {tag}, maybe you got the name backwards',json={'in_reply_to_id':s['id']}); continue
+		await api.post(f'#{tag} {src}',[(name,img)],{'sensitive':ecchi,'in_reply_to_id':s['id']})
+	await api.call('POST','notifications/clear')
+
+main.run()
\ No newline at end of file
diff --git a/example_conf.py b/example_conf.py
new file mode 100644
index 0000000..e926313
--- /dev/null
+++ b/example_conf.py
@@ -0,0 +1,5 @@
+instance='nigge.rs'
+token='XxXXxXxxxxxXXXXxxxxXXXxxxXXXXxxXXXxxXxX'
+gelb={"api_key": "fagfagfagfagfagfag", "user_id":9001}
+
+tags=['your_mum','lwmmg_(girls_frontline)']
\ No newline at end of file
diff --git a/gelb.py b/gelb.py
new file mode 100644
index 0000000..deee351
--- /dev/null
+++ b/gelb.py
@@ -0,0 +1,25 @@
+import aiohttp as ah
+import json as j
+from os import path
+
+#conf=j.load(open(path.join(path.dirname(__file__),'config.json')))[path.splitext(path.basename(__file__))[0]]
+
+baseurl='https://gelbooru.com/index.php'
+
+async def getgelbimg(tags):
+	params={'page':'dapi','s':'post','q':'index','limit':1,'json':1}
+	params|=conf
+	if isinstance(tags,str): params['tags']=f'{tags} sort:random'
+	if isinstance(tags,list): params['tags']=' '.join(tags+['sort:random'])
+	async with ah.ClientSession() as sess:
+		async with sess.get(baseurl,params=params) as resp:
+			try: data=(await resp.json())['post'][0]
+			except KeyError: print('no result'); return None,None,None,None
+		async with sess.get(data['file_url']) as resp:
+			img=await resp.read()
+	src=f'https://gelbooru.com/index.php?page=post&s=view&id={data["id"]}'
+	ecchi=(data['rating']!="general") #lmao they changed it, need to fix this now
+	name=data['image']
+	return img,name,src,ecchi
+
+#Gelb needs to put more API endpoints. Like edits.
\ No newline at end of file