Compare commits
2 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| 0d4f9b9a9a | |||
| 733f3e3ae6 |
+43
-27
@@ -1,4 +1,14 @@
|
||||
from typing import Any, Awaitable, Callable, Concatenate, ParamSpec, TypeVar, Union, TYPE_CHECKING
|
||||
from typing import (
|
||||
Any,
|
||||
Generic,
|
||||
Awaitable,
|
||||
Callable,
|
||||
Concatenate,
|
||||
ParamSpec,
|
||||
TypeVar,
|
||||
Union,
|
||||
TYPE_CHECKING,
|
||||
)
|
||||
from functools import wraps, reduce
|
||||
import operator
|
||||
|
||||
@@ -22,9 +32,9 @@ __all__ = (
|
||||
"ternary_term",
|
||||
"run_function",
|
||||
"cascade_term",
|
||||
'ExecutorTerm',
|
||||
'LateTerm',
|
||||
'lazy_term',
|
||||
"ExecutorTerm",
|
||||
"LateTerm",
|
||||
"lazy_term",
|
||||
)
|
||||
|
||||
P = ParamSpec("P")
|
||||
@@ -37,7 +47,7 @@ T2 = TypeVar("T2")
|
||||
TermResult = TypeVar("TermResult")
|
||||
|
||||
|
||||
class ExecutorTerm[TermResult]:
|
||||
class ExecutorTerm(Generic[TermResult]):
|
||||
# May add more here for initial parsing context
|
||||
|
||||
def __init__(self, *args, node_text: str | None = None, **kwargs):
|
||||
@@ -145,7 +155,7 @@ async def cmdarg_term(ctx, argn: int, greedy: bool) -> str:
|
||||
if greedy:
|
||||
return " ".join(args[argn:])
|
||||
elif argn >= len(args):
|
||||
return ''
|
||||
return ""
|
||||
else:
|
||||
return args[argn]
|
||||
|
||||
@@ -158,33 +168,36 @@ async def cascade_term(ctx, *terms: ExecutorTerm):
|
||||
return eval
|
||||
return ""
|
||||
|
||||
|
||||
@lazy_term
|
||||
async def command_shell(ctx, component_term: ExecutorTerm):
|
||||
if isinstance(component_term, ExecutorTerm):
|
||||
result = await component_term(ctx)
|
||||
else:
|
||||
logger.warning(
|
||||
f"command_shell LazyTerm stringifying non-term {component_term}"
|
||||
)
|
||||
logger.warning(f"command_shell LazyTerm stringifying non-term {component_term}")
|
||||
result = component_term
|
||||
return str(result)
|
||||
|
||||
|
||||
@lazy_term
|
||||
async def define_equal(ctx, var: str, value: ExecutorTerm):
|
||||
ctx.this_scope[var] = await value.execute(ctx)
|
||||
return None
|
||||
|
||||
|
||||
@lazy_term
|
||||
async def define_walrus(ctx, var: str, value: ExecutorTerm):
|
||||
result = await value.execute(ctx)
|
||||
ctx.this_scope[var] = result
|
||||
return result
|
||||
|
||||
|
||||
@lazy_term
|
||||
async def define_term(ctx, var: str, value: ExecutorTerm):
|
||||
ctx.this_scope[var] = value
|
||||
return None
|
||||
|
||||
|
||||
@lazy_term
|
||||
async def run_function(ctx, var: ExecutorTerm, *args: ExecutorTerm):
|
||||
varlue = await var.execute(ctx)
|
||||
@@ -196,13 +209,14 @@ async def run_function(ctx, var: ExecutorTerm, *args: ExecutorTerm):
|
||||
else:
|
||||
return await varlue(ctx, *argues)
|
||||
|
||||
|
||||
@lazy_term
|
||||
async def get_var(ctx, root: str | ExecutorTerm, *attrs: str):
|
||||
if isinstance(root, ExecutorTerm):
|
||||
value = await root.execute(ctx)
|
||||
name = [root.text or '']
|
||||
name = [root.text or ""]
|
||||
elif root not in ctx.scope:
|
||||
return ' '.join((root, *attrs))
|
||||
return " ".join((root, *attrs))
|
||||
else:
|
||||
value = ctx.scope[root]
|
||||
name = [root]
|
||||
@@ -213,39 +227,41 @@ async def get_var(ctx, root: str | ExecutorTerm, *attrs: str):
|
||||
|
||||
for attr in attrs:
|
||||
try:
|
||||
value = getattr(value, '_exec_attr_'+attr)
|
||||
value = getattr(value, "_exec_attr_" + attr)
|
||||
name.append(attr)
|
||||
except AttributeError:
|
||||
raise CommandValueError(f"{'.'.join(name)} has no attribute {attr}")
|
||||
return value
|
||||
|
||||
|
||||
@lazy_term
|
||||
async def compare(ctx, term1: ExecutorTerm, op: str, term2: ExecutorTerm):
|
||||
opmap = {
|
||||
'==': operator.eq,
|
||||
'!=': operator.ne,
|
||||
'>': operator.gt,
|
||||
'>=': operator.ge,
|
||||
'<': operator.lt,
|
||||
'<=': operator.le,
|
||||
"==": operator.eq,
|
||||
"!=": operator.ne,
|
||||
">": operator.gt,
|
||||
">=": operator.ge,
|
||||
"<": operator.lt,
|
||||
"<=": operator.le,
|
||||
}
|
||||
value1 = await term1.execute(ctx)
|
||||
value2 = await term2.execute(ctx)
|
||||
oppy = opmap[op]
|
||||
return oppy(value1, value2)
|
||||
|
||||
|
||||
@lazy_term
|
||||
async def binop(ctx, op: str, *terms):
|
||||
opmap = {
|
||||
'||': operator.or_,
|
||||
'&&': operator.and_,
|
||||
'+': operator.add,
|
||||
'-': operator.sub,
|
||||
'*': operator.mul,
|
||||
'**': operator.pow,
|
||||
'/': operator.truediv,
|
||||
'//': operator.floordiv,
|
||||
'%': operator.mod,
|
||||
"||": operator.or_,
|
||||
"&&": operator.and_,
|
||||
"+": operator.add,
|
||||
"-": operator.sub,
|
||||
"*": operator.mul,
|
||||
"**": operator.pow,
|
||||
"/": operator.truediv,
|
||||
"//": operator.floordiv,
|
||||
"%": operator.mod,
|
||||
}
|
||||
# Left associative
|
||||
oppy = opmap[op]
|
||||
|
||||
+11
-12
@@ -1,11 +1,13 @@
|
||||
import twitchio
|
||||
|
||||
from customcmd.client import ExecClient
|
||||
from customcmd.data import CustomCmd, CustomCmdData
|
||||
from meta import Bot
|
||||
|
||||
from ..customcmd.client import ExecClient
|
||||
from ..customcmd.data import CustomCmd, CustomCmdData
|
||||
|
||||
from .context import CustomContext, CustomCmdUser
|
||||
|
||||
|
||||
class TwitchCmdClient(ExecClient):
|
||||
def __init__(self, bot: Bot, data: CustomCmdData):
|
||||
super().__init__()
|
||||
@@ -22,12 +24,12 @@ class TwitchCmdClient(ExecClient):
|
||||
if message.reply:
|
||||
# Strip the hidden mention prefix from the start of the message
|
||||
prefix = message.reply.parent_user.mention
|
||||
content = content[len(prefix) + 1:]
|
||||
content = content[len(prefix) + 1 :]
|
||||
|
||||
# If we start with a valid prefix, parse out a command name
|
||||
if prefix := next((p for p in prefixes if content.startswith(p)), None):
|
||||
# Strip the valid prefix
|
||||
content = content[len(prefix):].strip()
|
||||
content = content[len(prefix) :].strip()
|
||||
if not content:
|
||||
# Exit early, no command
|
||||
return
|
||||
@@ -53,20 +55,17 @@ class TwitchCmdClient(ExecClient):
|
||||
used_prefix=prefix,
|
||||
message=message,
|
||||
initial_globals={
|
||||
'author': CustomCmdUser(message.chatter),
|
||||
'channel': CustomCmdUser(message.broadcaster),
|
||||
}
|
||||
"author": CustomCmdUser(message.chatter),
|
||||
"channel": CustomCmdUser(message.broadcaster),
|
||||
},
|
||||
)
|
||||
result = await self.run_command(ctx, cmd.response)
|
||||
if result:
|
||||
await message.broadcaster.send_message(
|
||||
result,
|
||||
sender=self.bot.user
|
||||
)
|
||||
await message.broadcaster.send_message(result, sender=self.bot.user)
|
||||
|
||||
async def get_prefixes(self, message: twitchio.ChatMessage):
|
||||
# TODO:
|
||||
return ('!',)
|
||||
return ("!",)
|
||||
|
||||
async def get_commands_in(self, communityid: int) -> dict[str, CustomCmd]:
|
||||
"""
|
||||
|
||||
+7
-7
@@ -4,7 +4,6 @@ import twitchio
|
||||
from twitchio import PartialUser
|
||||
from twitchio.ext import commands as cmds
|
||||
|
||||
from customcmd.parser.context import ExecContext
|
||||
from meta import Bot
|
||||
from twitch.client import TwitchCmdClient
|
||||
from utils.lib import utc_now
|
||||
@@ -12,13 +11,12 @@ from utils.lib import utc_now
|
||||
from . import logger
|
||||
from ..customcmd.data import CustomCmd, CustomCmdData
|
||||
from ..customcmd.client import ExecClient
|
||||
from ..customcmd.parser import ExecContext
|
||||
|
||||
# TODO: Make an alphacroc or croccyalpha client for testing
|
||||
# TODO: Fix the tracker not actually guarding the commands properly
|
||||
|
||||
|
||||
|
||||
|
||||
class CustomCmdComponent(cmds.Component):
|
||||
def __init__(self, bot: Bot):
|
||||
self.bot = bot
|
||||
@@ -46,7 +44,7 @@ class CustomCmdComponent(cmds.Component):
|
||||
# TODO: Probably attempt to list commands or something here.
|
||||
pass
|
||||
|
||||
@cmd_grp.command(name="add", aliases=('create', 'new'))
|
||||
@cmd_grp.command(name="add", aliases=("create", "new"))
|
||||
@cmds.is_broadcaster() # TODO: Better command creation perm checks
|
||||
async def cmd_add(self, ctx: cmds.Context, name: str, *, response: str):
|
||||
comm = await self.bot.profiles.fetch_community(ctx.broadcaster)
|
||||
@@ -60,11 +58,13 @@ class CustomCmdComponent(cmds.Component):
|
||||
|
||||
# Make sure the command parses (can't guarantee it runs, but at least it can parse)
|
||||
try:
|
||||
await self.cmdclient.run_command(ExecContext(args=[], content=''), response)
|
||||
await self.cmdclient.run_command(ExecContext(args=[], content=""), response)
|
||||
except Exception:
|
||||
# TODO: Nicer errors for pinpointing problem
|
||||
await ctx.reply(f"Sorry, could not parse your command!")
|
||||
logger.info(f"Failed to parse attempted customcmd '{response}'", exc_info=True)
|
||||
logger.info(
|
||||
f"Failed to parse attempted customcmd '{response}'", exc_info=True
|
||||
)
|
||||
|
||||
# If everything is good, add it to the table
|
||||
# TODO: Again, permission resolution
|
||||
@@ -79,7 +79,7 @@ class CustomCmdComponent(cmds.Component):
|
||||
|
||||
await ctx.reply(f"Created a new command !{name}")
|
||||
|
||||
@cmd_grp.command(name="del", aliases=('delete', 'remove', 'rm'))
|
||||
@cmd_grp.command(name="del", aliases=("delete", "remove", "rm"))
|
||||
@cmds.is_broadcaster()
|
||||
async def cmd_del(self, ctx: cmds.Context, name: str):
|
||||
# Lookup the name to make sure a command does exist
|
||||
|
||||
+9
-8
@@ -1,11 +1,11 @@
|
||||
import twitchio
|
||||
|
||||
from customcmd.client import ExecClient
|
||||
from customcmd.data import CustomCmd, CustomCmdData
|
||||
from meta import Bot
|
||||
from utils.lib import utc_now, strfdur
|
||||
|
||||
from ..customcmd.parser import ExecContext
|
||||
from ..customcmd.client import ExecClient
|
||||
from ..customcmd.data import CustomCmd, CustomCmdData
|
||||
from .ctxvars import stuff
|
||||
|
||||
|
||||
@@ -25,10 +25,7 @@ class CustomCmdUser:
|
||||
return self.user.display_name or self.user.name or self.user.id
|
||||
|
||||
async def _exec_attr_whisper(self, ctx, message: str):
|
||||
await self.user.send_whisper(
|
||||
to_user=self.user,
|
||||
message=message
|
||||
)
|
||||
await self.user.send_whisper(to_user=self.user, message=message)
|
||||
|
||||
async def _exec_attr_followage(self, ctx, channel: twitchio.PartialUser):
|
||||
followed_at = None
|
||||
@@ -38,7 +35,9 @@ class CustomCmdUser:
|
||||
|
||||
if followed_at:
|
||||
durstr = strfdur(utc_now() - followed_at)
|
||||
resp = f"{self.user.mention} has been following {channel.mention} for {durstr}"
|
||||
resp = (
|
||||
f"{self.user.mention} has been following {channel.mention} for {durstr}"
|
||||
)
|
||||
else:
|
||||
return f"{self.user.mention} is not following {channel.mention}"
|
||||
|
||||
@@ -51,13 +50,15 @@ class CustomContext(ExecContext):
|
||||
"""
|
||||
CustomCommand ExecContext instantiated from a Twitch message
|
||||
"""
|
||||
|
||||
def __init__(
|
||||
self,
|
||||
bot: Bot,
|
||||
message: twitchio.ChatMessage,
|
||||
alias: str,
|
||||
used_prefix: str,
|
||||
**kwargs):
|
||||
**kwargs,
|
||||
):
|
||||
super().__init__(**kwargs)
|
||||
self.bot = bot
|
||||
self.message = message
|
||||
|
||||
Reference in New Issue
Block a user