MediaPlugin/MediaAction.py at main · StreamController/MediaPlugin · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
from src.backend.PluginManager.ActionBase import ActionBase
from src.backend.PluginManager.PluginBase import PluginBase
from src.backend.PageManagement.Page import Page

# Import gtk modules
import gi
gi.require_version("Gtk", "4.0")
gi.require_version("Adw", "1")
from gi.repository import Gtk, Adw

from PIL import Image, ImageEnhance
import os
import math

class MediaAction(ActionBase):
    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)

        self.has_configuration = True

        self.current_status = None

    def on_key_down(self):
        pass

    def on_key_up(self):
        pass

    def on_tick(self):
        pass

    def get_config_rows(self) -> "list[Adw.PreferencesRow]":
        # Init ui elements
        self.player_model = Gtk.StringList()
        self.player_selector = Adw.ComboRow(model=self.player_model, title=self.plugin_base.lm.get("actions.media-action.bind-to-player.label"), subtitle=self.plugin_base.lm.get("actions.media-action.bind-to-player.subtitle"))
        self.player_selector.set_enable_search(True) #TODO: Implement


        self.label_toggle = Adw.SwitchRow(title=self.plugin_base.lm.get("actions.media-action.show-name-switch.label"), subtitle=self.plugin_base.lm.get("actions.media-action.show-name-switch.subtitle"))
        self.thumbnail_toggle = Adw.SwitchRow(title=self.plugin_base.lm.get("actions.media-action.show-thumbnail-switch.label"), subtitle=self.plugin_base.lm.get("actions.media-action.show-thumbnail-switch.subtitle"))

        self.load_config_defaults()

        self.player_selector.connect("notify::selected-item", self.on_change_player)
        self.label_toggle.connect("notify::active", self.on_toggle_label)
        self.thumbnail_toggle.connect("notify::active", self.on_toggle_thumbnail)

        return [self.player_selector, self.label_toggle, self.thumbnail_toggle]

    ## Custom methods
    def load_config_defaults(self):
        settings = self.get_settings()
        if settings == None:
            return

        show_label = settings.setdefault("show_label", True)
        show_thumbnail = settings.setdefault("show_thumbnail", True)

        # Update ui
        self.label_toggle.set_active(show_label)
        self.thumbnail_toggle.set_active(show_thumbnail)
        self.update_player_selector()

    def update_player_selector(self):
        # Clear the model
        for i in range(self.player_model.get_n_items()):
            self.player_model.remove(0)

        players = self.plugin_base.mc.get_player_names(remove_duplicates=True)

        self.player_model.append(self.plugin_base.lm.get("actions.media-action-bind-to-player.all-players"))

        for player in players:
            self.player_model.append(player)

        # Add from settings if not already in the model
        if self.get_player_name() is not None:
            if self.get_player_name() not in players:
                self.player_model.append(self.get_player_name())

        # Select from settings
        if self.get_player_name() is not None:
            position = 0
            for i in range(self.player_model.get_n_items()):
                item = self.player_model.get_item(i).get_string()
                n = self.get_player_name()
                if self.player_model.get_item(i).get_string() == self.get_player_name():
                    position = i
                    break
            self.player_selector.set_selected(position)

    def on_change_player(self, combo, *args):
        settings = self.get_settings()
        if combo.get_selected_item().get_string() == self.plugin_base.lm.get("actions.media-action-bind-to-player.all-players"):
            del settings["player_name"]
        else:
            settings["player_name"] = combo.get_selected_item().get_string()
        self.set_settings(settings)

    def get_player_name(self):
        settings = self.get_settings()
        if settings is not None:
            return settings.get("player_name")

    def show_title(self, reload_key = True) -> bool:
        if self.get_settings() == None:
            return False
        title = self.plugin_base.mc.title(self.get_player_name())
        if self.get_settings().setdefault("show_label", True) and title is not None:
            label = None
            if isinstance(title, list):
                if len(title) > 0:
                    label = title[0]
                    # margins = [5, 0, 5, 10]
            if isinstance(title, str):
                if len(title) > 0:
                    label = title
                    # margins = [5, 0, 5, 10]
            self.set_bottom_label(str(self.shorten_label(label, 10)), font_size=12, update=reload_key)
            return True
        else:
            self.set_bottom_label(None, update=reload_key)
            return False

    def shorten_label(self, label, length):
        if label is None:
            return
        if len(label) > length:
            return label[:length-2] + ".."
        return label

    def on_toggle_label(self, switch, *args):
        settings = self.get_settings()
        settings["show_label"] = switch.get_active()
        self.set_settings(settings)
        # Update image
        self.on_tick()

    def on_toggle_thumbnail(self, switch, *args):
        settings = self.get_settings()
        settings["show_thumbnail"] = switch.get_active()
        self.set_settings(settings)
        # Update image
        self.on_tick()

    def generate_image(self, icon:Image.Image = None, background:Image.Image=None, valign: float = 0, halign: float = 0, size: float = 1):
        if background is None:
            background = Image.new("RGBA", (self.deck_controller.deck.key_image_format()["size"]), (0, 0, 0, 0))
        else:
            background = background.resize(self.deck_controller.deck.key_image_format()["size"])

        if icon is not None:
            # Resize
            lenght = int(self.deck_controller.deck.key_image_format()["size"][0] * size)
            icon = icon.resize((lenght, lenght))

        left_margin = int((background.width - icon.width) * (halign + 1) / 2)
        top_margin = int((background.height - icon.height) * (valign + 1) / 2)

        background.paste(icon, (left_margin, top_margin), icon)

        return background