lol
This commit is contained in:
parent
e3d3311ddb
commit
80856ef17d
58
chatlib.py
Normal file
58
chatlib.py
Normal file
|
@ -0,0 +1,58 @@
|
||||||
|
import aiohttp as ah
|
||||||
|
import time
|
||||||
|
import asyncio
|
||||||
|
import datetime
|
||||||
|
import traceback
|
||||||
|
from .utils import Event,Listener
|
||||||
|
|
||||||
|
func=type(lambda:1) # Gross, but I can't actually find it.
|
||||||
|
# And yes, lamdas are <class 'function'> too.
|
||||||
|
|
||||||
|
tracing=False
|
||||||
|
def trace(*msg):
|
||||||
|
if tracing: print(datetime.datetime.now(),*msg)
|
||||||
|
|
||||||
|
class MatrixClient():
|
||||||
|
def __init__(self,homeserver,token):
|
||||||
|
self.event_queue=asyncio.Queue() #Contains Events. Everything in here must be an Event.
|
||||||
|
self.listeners=[]
|
||||||
|
self.timers=[]
|
||||||
|
|
||||||
|
async def request(self,endpoint='sync',method='GET', ver=0,headers={} *args,**kwargs):
|
||||||
|
async with self.session.request(method, f'{self.baseurl}/r{ver}/{endpoint}', headers=headers|{'Authorization':f'Bearer {self.token}'}, *args,**kwargs) as fetched:
|
||||||
|
if fetched.status_code!=200: raise Exception('fix ur shit')
|
||||||
|
try: return await fetched.json()
|
||||||
|
except JSONDecodeError: pass # TODO: Figure out what this is called
|
||||||
|
|
||||||
|
# This function just dumps /sync blobs in the event queue as a raw event.
|
||||||
|
# ALL handling is deferred to handlers.
|
||||||
|
# ... Except updating the since token. That's important to do here to guarantee that it never calls with the same token twice.
|
||||||
|
async def sync(self):
|
||||||
|
blob=await self.request(params={'timeout':30000,'since':self.since})
|
||||||
|
self.since=blob['next_batch']
|
||||||
|
self.event_queue.put(Event('m.sync',None,blob))
|
||||||
|
|
||||||
|
def addlistener(self,name=None,match=None):
|
||||||
|
if isinstance(name,func) and match==None: match=name; name=None
|
||||||
|
def __wrap__(funky):
|
||||||
|
self.listeners.append(Listener(name,match,funky))
|
||||||
|
return funky
|
||||||
|
|
||||||
|
def loop(timer=0):
|
||||||
|
def _loop(funky):
|
||||||
|
@functools.wraps(funky)
|
||||||
|
async def _wrapper(*args,**kwargs):
|
||||||
|
while True:
|
||||||
|
try: await funky(*args,**kwargs)
|
||||||
|
except Exception: open('error '+str(datetime.datetime.now()),'w').write(traceback.format_exc())
|
||||||
|
await asyncio.sleep(timer)
|
||||||
|
self.timers.append(_wrapper)
|
||||||
|
return _wrapper
|
||||||
|
return _loop
|
||||||
|
|
||||||
|
async def process_queue(self):
|
||||||
|
item=await self.event_queue.get():
|
||||||
|
for listener in self.listeners:
|
||||||
|
if listener==item:
|
||||||
|
await listener(item)
|
||||||
|
|
13
matrixbot.py
Normal file
13
matrixbot.py
Normal file
|
@ -0,0 +1,13 @@
|
||||||
|
from .matrix import *
|
||||||
|
import .listeners
|
||||||
|
real_listeners=dict(filter(lambda x:isinstance(x,Listener),listeners.__dict__.items()))
|
||||||
|
|
||||||
|
class MatrixClient(MatrixClient):
|
||||||
|
def __init__(self,*args,**kwargs):
|
||||||
|
super().__init__(*args,**kwargs)
|
||||||
|
self.listeners=real_listeners
|
||||||
|
self.homeserver=homeserver
|
||||||
|
self.token=token # Deal with login stuff later
|
||||||
|
self.baseurl=f'https://{self.homeserver}/_matrix/client'
|
||||||
|
self.session=ah.ClientSession()
|
||||||
|
self.since=None
|
Loading…
Reference in New Issue
Block a user