LogalDeveloper
4fa7050f0a
CI / Formatting (push) Successful in 5s
CI / Linting (push) Successful in 5s
CI / Tests (Python 3.12) (push) Successful in 13s
CI / Tests (Python 3.13) (push) Successful in 13s
CI / Tests (Python 3.14) (push) Successful in 10s
CI / Type Checking (push) Successful in 10s
CI / Spelling (push) Successful in 5s
96 lines
2.9 KiB
Python
96 lines
2.9 KiB
Python
# Copyright 2026 Logan Fick
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
|
|
"""Event handlers for the timers module."""
|
|
|
|
from owlbot.api import (
|
|
ChatEvent,
|
|
EventContext,
|
|
EventType,
|
|
Priority,
|
|
StreamStartedEvent,
|
|
StreamStoppedEvent,
|
|
VisibilityUpdateEvent,
|
|
on_event,
|
|
)
|
|
|
|
from .manager import get_manager
|
|
|
|
|
|
@on_event(EventType.STREAM_STARTED)
|
|
async def handle_stream_started(ctx: EventContext[StreamStartedEvent]) -> None:
|
|
"""Start timers when the stream goes live.
|
|
|
|
Cancels any pending offline stop (stream returned within grace period)
|
|
and starts all enabled timers.
|
|
|
|
:param ctx: The event context.
|
|
"""
|
|
manager = get_manager(ctx.module)
|
|
manager.cancel_offline_stop()
|
|
await manager.start_all()
|
|
ctx.logger.info("Stream started. Timers activated.")
|
|
|
|
|
|
@on_event(EventType.STREAM_STOPPED)
|
|
async def handle_stream_stopped(ctx: EventContext[StreamStoppedEvent]) -> None:
|
|
"""Schedule timers to stop after a grace period.
|
|
|
|
If the stream returns within the grace period, the stop is cancelled
|
|
by handle_stream_started.
|
|
|
|
:param ctx: The event context.
|
|
"""
|
|
manager = get_manager(ctx.module)
|
|
manager.schedule_offline_stop()
|
|
ctx.logger.info(
|
|
"Stream stopped. Timers will stop in %.0f seconds.",
|
|
manager.OFFLINE_STOP_DELAY,
|
|
)
|
|
|
|
|
|
@on_event(EventType.CHAT, priority=Priority.LOWEST)
|
|
async def count_chat_message(ctx: EventContext[ChatEvent]) -> None:
|
|
"""Count a chat message for all active timers.
|
|
|
|
Runs at lowest priority so all other CHAT handlers (moderation, etc.)
|
|
execute first. Skips bot messages and hidden messages.
|
|
|
|
:param ctx: The event context with the chat event.
|
|
"""
|
|
event = ctx.event
|
|
|
|
if event.user.is_bot or not event.is_visible:
|
|
return
|
|
|
|
get_manager(ctx.module).count_message(event.message_id)
|
|
|
|
|
|
@on_event(EventType.VISIBILITY_UPDATE, priority=Priority.LOWEST)
|
|
async def handle_visibility_update(ctx: EventContext[VisibilityUpdateEvent]) -> None:
|
|
"""Remove hidden messages from chat counts.
|
|
|
|
Only handles the hide case. Un-hiding does not re-add messages because
|
|
we cannot distinguish previously counted user messages from bot messages
|
|
that were never counted.
|
|
|
|
:param ctx: The event context with the visibility update event.
|
|
"""
|
|
event = ctx.event
|
|
|
|
if event.is_visible:
|
|
return
|
|
|
|
get_manager(ctx.module).remove_messages(set(event.message_ids))
|