mirror of
https://github.com/myned/watcher.git
synced 2024-11-01 12:22:38 +00:00
98 lines
3 KiB
Python
98 lines
3 KiB
Python
import datetime as dt
|
|
import hikari
|
|
import lightbulb
|
|
from lightbulb.ext import tasks
|
|
|
|
import config as c
|
|
|
|
|
|
plugin = lightbulb.Plugin("activity")
|
|
|
|
|
|
# Check every minute if inactive (or config duration if under 60 secs)
|
|
@tasks.task(s=60 if c.config["duration"] >= 60 else c.config["duration"])
|
|
async def check_activity():
|
|
for author_id, timestamp in c.db.items():
|
|
# If time between now and timestamp >= duration
|
|
if dt.datetime.now(dt.timezone.utc) - timestamp >= dt.timedelta(seconds=c.config["duration"]):
|
|
try:
|
|
# Acquire member object
|
|
member = plugin.bot.cache.get_member(
|
|
c.config["guild"], author_id
|
|
) or await plugin.bot.rest.fetch_member(c.config["guild"], author_id)
|
|
|
|
# Delete member from db if it has excluded role
|
|
if c.config["exclude"] in member.role_ids:
|
|
del c.db[author_id]
|
|
continue
|
|
|
|
# Enforce activity roles
|
|
if c.config["active"] in member.role_ids:
|
|
await member.remove_role(c.config["active"])
|
|
if c.config["inactive"] not in member.role_ids:
|
|
await member.add_role(c.config["inactive"])
|
|
# Delete member from db if not found
|
|
except hikari.NotFoundError:
|
|
del c.db[author_id]
|
|
|
|
|
|
# Listener for bot ready
|
|
@plugin.listener(hikari.StartedEvent)
|
|
async def on_ready(event):
|
|
check_activity.start()
|
|
|
|
|
|
# Check activity and update timestamp
|
|
async def update_activity(event, member):
|
|
# Exclude other guilds
|
|
if event.guild_id != c.config["guild"]:
|
|
return
|
|
|
|
# Exclude and remove activity roles from excluded role
|
|
if c.config["exclude"] in member.role_ids:
|
|
if c.config["active"] in member.role_ids:
|
|
await member.remove_role(c.config["active"])
|
|
if c.config["inactive"] in member.role_ids:
|
|
await member.remove_role(c.config["inactive"])
|
|
return
|
|
|
|
# Insert current timestamp into db
|
|
c.db[member.id] = dt.datetime.now(dt.timezone.utc)
|
|
|
|
# Toggle activity roles
|
|
if c.config["active"] not in member.role_ids:
|
|
await member.add_role(c.config["active"])
|
|
if c.config["inactive"] in member.role_ids:
|
|
await member.remove_role(c.config["inactive"])
|
|
|
|
|
|
# Listener for guild messages
|
|
@plugin.listener(hikari.GuildMessageCreateEvent)
|
|
async def on_message(event):
|
|
# Exclude bots and webhooks
|
|
if event.is_human:
|
|
await update_activity(event, event.member)
|
|
|
|
|
|
# Listener for guild typing
|
|
@plugin.listener(hikari.GuildTypingEvent)
|
|
async def on_typing(event):
|
|
# Exclude bots
|
|
if not event.member.is_bot:
|
|
await update_activity(event, event.member)
|
|
|
|
|
|
# Listener for voice state
|
|
@plugin.listener(hikari.VoiceStateUpdateEvent)
|
|
async def on_voice(event):
|
|
# Exclude bots
|
|
if not event.state.member.is_bot:
|
|
await update_activity(event, event.state.member)
|
|
|
|
|
|
def load(bot):
|
|
bot.add_plugin(plugin)
|
|
|
|
|
|
def unload(bot):
|
|
bot.remove_plugin(plugin)
|