OBSPlugin/actions/ToggleStream/ToggleStream.py at 4dbc962f3a4358c655a6257176d45cd5ded5ff7f · StreamController/OBSPlugin · 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
# from ...OBSActionbase import OBSActionBase
import threading
from plugins.com_core447_OBSPlugin.OBSActionBase import OBSActionBase
from src.backend.DeckManagement.DeckController import DeckController
from src.backend.PageManagement.Page import Page
from src.backend.PluginManager.PluginBase import PluginBase

import os
class ToggleStream(OBSActionBase):
    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self.current_state = -1

    def on_ready(self):
        self.current_state = -1
        if self.plugin_base.backend is None:
            if not self.plugin_base.get_connected():
                self.reconnect_obs()

        threading.Thread(target=self.show_current_stream_status, daemon=True, name="show_current_stream_status").start()

    def show_current_stream_status(self, new_paused = False):
        if self.plugin_base.backend is None:
            self.current_state = -1
            self.show_error()
            return
        if not self.plugin_base.backend.get_connected():
            self.current_state = -1
            self.show_error()
            return
        status = self.plugin_base.backend.get_stream_status()
        if status is None:
            self.current_state = -1
            self.show_error()
            return
        self.hide_error()
        if status["active"]:
            if status["reconnecting"]:
                self.show_for_state(2)
            else:
                self.show_for_state(1)
        else:
            self.show_for_state(0)

    def show_for_state(self, state: int):
        """
        0: Not Streaming
        1: Streaming Connected
        2: Streaming Reconnecting
        """
        self.hide_error()
        if state in [1, 2]:
            self.show_stream_time()

        if state == self.current_state:
            return

        self.current_state = state
        image = "stream_inactive.png"
        if state == 0:
            self.set_bottom_label(None)
        elif state == 1:
            image = "stream_active.png"
        elif state == 2:
            image = "stream_reconnecting.png"

        self.set_media(media_path=os.path.join(self.plugin_base.PATH, "assets", image))

    def on_key_down(self):
        if self.plugin_base.backend is None:
            self.current_state = -1
            self.show_error()
            return
        if not self.plugin_base.backend.get_connected():
            self.current_state = -1
            self.show_error()
            return
        self.plugin_base.backend.toggle_stream()
        self.on_tick()

    def on_tick(self):
        self.show_current_stream_status()

    def show_stream_time(self):
        if not self.plugin_base.backend.get_connected():
            self.set_media(media_path=os.path.join(self.plugin_base.PATH, "assets", "error.png"))
            return
        status = self.plugin_base.backend.get_stream_status()
        if status is None:
            self.set_media(media_path=os.path.join(self.plugin_base.PATH, "assets", "error.png"))
            return

        if not status["active"]:
            self.set_bottom_label(None)
            return
        self.set_bottom_label(status["timecode"][:-4])