preferences.py

"""Addon preferences and keymap management for YABQOLA."""

from __future__ import annotations

import bpy
from bpy.props import BoolProperty
from bpy.types import AddonPreferences, Context, KeyMap, KeyMapItem

__all__ = [
    "AnimationQOLPreferences",
    "get_addon_preferences",
    "refresh_keymaps",
    "clear_keymaps",
    "register",
    "unregister",
]

_addon_keymaps: list[tuple[KeyMap, KeyMapItem]] = []


def get_addon_preferences(context: Context | None = None) -> "AnimationQOLPreferences | None":
    context = context or bpy.context
    if context is None:
        return None

    preferences = context.preferences
    addon = preferences.addons.get(__package__)
    if addon is None:
        return None
    return addon.preferences


def _remove_existing_keymaps() -> None:
    while _addon_keymaps:
        km, kmi = _addon_keymaps.pop()
        km.keymap_items.remove(kmi)


def clear_keymaps() -> None:
    _remove_existing_keymaps()


def _install_keymaps(context: Context, prefs: "AnimationQOLPreferences") -> None:
    window_manager = context.window_manager
    if window_manager is None:
        return

    keyconfigs = window_manager.keyconfigs.addon
    if keyconfigs is None:
        return

    # No optional keymaps currently installed.


def refresh_keymaps(context: Context | None = None) -> None:
    context = context or bpy.context
    prefs = get_addon_preferences(context)
    if prefs is None:
        return

    _remove_existing_keymaps()
    _install_keymaps(context, prefs)


def _update_keymaps(self, _context: Context) -> None:
    refresh_keymaps()


class AnimationQOLPreferences(AddonPreferences):
    """Addon preferences for shortcut management."""

    bl_idname = __package__

    enable_noise_randomizer_tools: BoolProperty(
        name="Noise Randomizer",
        description="Show the Noise Randomizer tools",
        default=True,
    )
    enable_keyframe_offset_tools: BoolProperty(
        name="Keyframe Offset",
        description="Show the Keyframe Offset tools",
        default=True,
    )
    enable_stagger_timing_tools: BoolProperty(
        name="Stagger Timing",
        description="Show the Stagger Timing tools",
        default=True,
    )
    enable_ease_presets_tools: BoolProperty(
        name="Ease Presets",
        description="Show the Ease Presets tools",
        default=True,
    )
    enable_motion_hold_tools: BoolProperty(
        name="Motion Hold",
        description="Show the Motion Hold tools",
        default=True,
    )
    enable_quick_flip_tools: BoolProperty(
        name="Quick Flip",
        description="Show the Quick Flip tools",
        default=True,
    )
    enable_scene_cleanup_tools: BoolProperty(
        name="Scene Cleanup",
        description="Show the Scene Cleanup tools",
        default=True,
    )
    enable_render_presets_tools: BoolProperty(
        name="Render Presets",
        description="Show the Render Presets tools",
        default=True,
    )
    enable_select_by_type_tools: BoolProperty(
        name="Select by Keyframe Type",
        description="Show the Select by Keyframe Type tools",
        default=True,
    )
    enable_scale_values_tools: BoolProperty(
        name="Scale Keyframe Values",
        description="Show the Keyframe Value Scaling tools",
        default=True,
    )
    enable_snap_keyframes_tools: BoolProperty(
        name="Snap Keyframes",
        description="Show the Snap Keyframes tools",
        default=True,
    )
    enable_loop_animation_tools: BoolProperty(
        name="Loop Animation",
        description="Show the Animation Loop Helper tools",
        default=True,
    )
    enable_overshoot_settle_tools: BoolProperty(
        name="Overshoot & Settle",
        description="Show the Overshoot & Settle tools",
        default=True,
    )
    enable_cycle_setup_tools: BoolProperty(
        name="Cycle Setup",
        description="Show the Cycle/Repeat Setup tools",
        default=True,
    )
    enable_time_stretch_tools: BoolProperty(
        name="Time Stretch",
        description="Show the Time Stretch tools",
        default=True,
    )
    enable_batch_render_tools: BoolProperty(
        name="Multi-Angle Batch Render",
        description="Show the Multi-Angle Batch Render tools",
        default=True,
    )
    enable_compositor_setup_tools: BoolProperty(
        name="Compositor Quick Setup",
        description="Show the Compositor Quick Setup tools",
        default=True,
    )
    enable_auto_framing_tools: BoolProperty(
        name="Auto Framing",
        description="Show the Auto Framing tools",
        default=True,
    )
    enable_silhouette_check_tools: BoolProperty(
        name="Silhouette Check",
        description="Show the Silhouette/Readability Check tools",
        default=True,
    )
    enable_pose_snapshots_tools: BoolProperty(
        name="Pose Snapshots",
        description="Show the Pose Snapshot & Compare tools",
        default=True,
    )
    enable_camera_snapshots_tools: BoolProperty(
        name="Camera Snapshots",
        description="Show the Camera Snapshots tools",
        default=True,
    )
    enable_lighting_presets_tools: BoolProperty(
        name="Lighting Presets",
        description="Show the Studio Lighting Preset tools",
        default=True,
    )
    enable_focus_helper_tools: BoolProperty(
        name="Focus Helper",
        description="Show the Quick DOF Setup tools",
        default=True,
    )
    enable_reference_camera_tools: BoolProperty(
        name="Reference Camera Matcher",
        description="Show the Reference Camera Matcher tools",
        default=True,
    )
    enable_output_presets_tools: BoolProperty(
        name="Output Size Presets",
        description="Show the Output Size Preset tools",
        default=True,
    )
    enable_background_swap_tools: BoolProperty(
        name="Background Swap",
        description="Show the Background Swap tools",
        default=True,
    )
    enable_viewport_hdri_tools: BoolProperty(
        name="Viewport HDRI Rotation",
        description="Show the Viewport HDRI Rotation tools",
        default=True,
    )
    enable_color_temperature_tools: BoolProperty(
        name="Color Temperature",
        description="Show the Color Temperature Shift tools",
        default=True,
    )
    enable_material_override_tools: BoolProperty(
        name="Material Override",
        description="Show the Material Override tools",
        default=True,
    )
    enable_quick_backdrop_tools: BoolProperty(
        name="Quick Backdrop",
        description="Show the Quick Backdrop tools",
        default=True,
    )
    enable_turntable_tools: BoolProperty(
        name="Turntable Generator",
        description="Show the Turntable Generator tools",
        default=True,
    )
    enable_expression_mixer_tools: BoolProperty(
        name="Expression Mixer",
        description="Show the Expression Mixer tools",
        default=True,
    )
    enable_physics_snapshot_tools: BoolProperty(
        name="Physics Snapshot",
        description="Show the Physics Snapshot tools",
        default=True,
    )
    enable_watermark_tools: BoolProperty(
        name="Watermark Overlay",
        description="Show the Watermark Overlay tools",
        default=True,
    )

    def draw(self, _context: Context):
        layout = self.layout
        layout.use_property_split = False
        layout.use_property_decorate = False

        feature_box = layout.box()
        feature_box.label(text="Feature Visibility", icon="PREFERENCES")
        column = feature_box.column(align=True)
        column.prop(self, "enable_noise_randomizer_tools")
        column.prop(self, "enable_keyframe_offset_tools")
        column.prop(self, "enable_stagger_timing_tools")
        column.prop(self, "enable_ease_presets_tools")
        column.prop(self, "enable_motion_hold_tools")
        column.prop(self, "enable_quick_flip_tools")
        column.prop(self, "enable_scene_cleanup_tools")
        column.prop(self, "enable_render_presets_tools")
        column.prop(self, "enable_select_by_type_tools")
        column.prop(self, "enable_scale_values_tools")
        column.prop(self, "enable_snap_keyframes_tools")
        column.prop(self, "enable_loop_animation_tools")
        column.prop(self, "enable_overshoot_settle_tools")
        column.prop(self, "enable_cycle_setup_tools")
        column.prop(self, "enable_time_stretch_tools")
        column.prop(self, "enable_batch_render_tools")
        column.prop(self, "enable_compositor_setup_tools")
        column.prop(self, "enable_auto_framing_tools")
        column.prop(self, "enable_silhouette_check_tools")
        column.prop(self, "enable_pose_snapshots_tools")
        column.prop(self, "enable_camera_snapshots_tools")
        column.prop(self, "enable_lighting_presets_tools")
        column.prop(self, "enable_focus_helper_tools")
        column.prop(self, "enable_reference_camera_tools")
        column.prop(self, "enable_output_presets_tools")
        column.prop(self, "enable_background_swap_tools")
        column.prop(self, "enable_viewport_hdri_tools")
        column.prop(self, "enable_color_temperature_tools")
        column.prop(self, "enable_material_override_tools")
        column.prop(self, "enable_quick_backdrop_tools")
        column.prop(self, "enable_turntable_tools")
        column.prop(self, "enable_expression_mixer_tools")
        column.prop(self, "enable_physics_snapshot_tools")
        column.prop(self, "enable_watermark_tools")


def register() -> None:
    bpy.utils.register_class(AnimationQOLPreferences)
    refresh_keymaps()


def unregister() -> None:
    clear_keymaps()
    bpy.utils.unregister_class(AnimationQOLPreferences)