multi-user/ui.py

import bpy
from . import operators
from .libs.replication.replication.constants import FETCHED, ERROR
from .bl_types.bl_user import BlUser


PROP_STATES = ['KEYTYPE_EXTREME_VEC',  # ADDED
               'KEYTYPE_BREAKDOWN_VEC',  # COMMITED
               'KEYTYPE_KEYFRAME_VEC',  # PUSHED
               'KEYTYPE_KEYFRAME_VEC',  # FETCHED
               'KEYTYPE_JITTER_VEC',   # UP
               'KEYTYPE_KEYFRAME_VEC']  # CHANGED


class SESSION_PT_settings(bpy.types.Panel):
    """Settings panel"""
    bl_idname = "MULTIUSER_SETTINGS_PT_panel"
    bl_label = "Settings"
    bl_space_type = 'VIEW_3D'
    bl_region_type = 'UI'
    bl_category = "Multiuser"

    def draw_header(self, context):
        self.layout.label(text="", icon='TOOL_SETTINGS')

    def draw(self, context):
        layout = self.layout
        layout.use_property_split = True
        row = layout.row()

        if hasattr(context.window_manager, 'session'):
            # STATE INITIAL
            if not operators.client \
               or (operators.client and operators.client.state == 0):
                pass
                # REPLICATION SETTINGS
                # row = layout.row()
                # box = row.box()
                # row = box.row()
                # row.label(text="REPLICATION", icon='TRIA_RIGHT')
                # row = box.row()

                # for item in window_manager.session.supported_datablock:
                #     row.label(text=item.type_name,icon=ICONS[item.type_name])
                #     row.prop(item, "is_replicated", text="")
                #     row = box.row()
            else:
                # STATE ACTIVE
                if operators.client.state == 2:

                    row = layout.row()
                    row.operator("session.stop", icon='QUIT', text="Exit")
                    row = layout.row()

                # STATE SYNCING
                else:
                    status = "connecting..."
                    row.label(text=status)
                    row = layout.row()
                    row.operator("session.stop", icon='QUIT', text="CANCEL")


class SESSION_PT_settings_network(bpy.types.Panel):
    bl_idname = "MULTIUSER_SETTINGS_NETWORK_PT_panel"
    bl_label = "Network"
    bl_space_type = 'VIEW_3D'
    bl_region_type = 'UI'
    bl_category = "Multiuser"
    bl_parent_id = 'MULTIUSER_SETTINGS_PT_panel'

    @classmethod
    def poll(cls, context):
        return not operators.client \
            or (operators.client and operators.client.state == 0)

    def draw(self, context):
        layout = self.layout

        settings = context.window_manager.session

        row = layout.row()
        # USER SETTINGS
        row.label(text="draw overlay:")
        row.prop(settings, "enable_presence", text="")
        row = layout.row()

        row = layout.row()
        row.prop(settings, "session_mode", expand=True)
        row = layout.row()

        if settings.session_mode == 'HOST':
            box = row.box()
            row = box.row()
            row.label(text="clear blend:")
            row.prop(settings, "start_empty", text="")
            row = box.row()
            row.label(text="init scene:")
            row.prop(settings, "init_scene", text="")
            row = box.row()
            row.operator("session.start", text="HOST").host = True
        else:
            box = row.box()
            row = box.row()
            row.prop(settings, "ip", text="ip")
            row = box.row()
            row.label(text="port:")
            row.prop(settings, "port", text="")
            row = box.row()

            row = box.row()
            row.operator("session.start", text="CONNECT").host = False


class SESSION_PT_settings_user(bpy.types.Panel):
    bl_idname = "MULTIUSER_SETTINGS_USER_PT_panel"
    bl_label = "User"
    bl_space_type = 'VIEW_3D'
    bl_region_type = 'UI'
    bl_category = "Multiuser"
    bl_parent_id = 'MULTIUSER_SETTINGS_PT_panel'

    @classmethod
    def poll(cls, context):
        return not operators.client \
            or (operators.client and operators.client.state == 0)

    def draw(self, context):
        layout = self.layout

        settings = context.window_manager.session

        row = layout.row()
        # USER SETTINGS
        row.prop(settings, "username", text="id")

        row = layout.row()
        row.prop(settings, "client_color", text="color")
        row = layout.row()


class SESSION_PT_user(bpy.types.Panel):
    bl_idname = "MULTIUSER_USER_PT_panel"
    bl_label = "Users"
    bl_space_type = 'VIEW_3D'
    bl_region_type = 'UI'
    bl_category = "Multiuser"
    bl_parent_id = 'MULTIUSER_SETTINGS_PT_panel'

    @classmethod
    def poll(cls, context):
        return operators.client and operators.client.state == 2

    def draw(self, context):
        layout = self.layout

        settings = context.window_manager.session

        # Create a simple row.
        col = layout.column(align=True)

        client_keys = operators.client.list(filter=BlUser)
        if client_keys and len(client_keys) > 0:
            for key in client_keys:
                area_msg = col.row(align=True)
                item_box = area_msg.box()
                client = operators.client.get(uuid=key).buffer

                info = ""

                detail_item_row = item_box.row(align=True)

                username = client['name']

                is_local_user = username == settings.username

                if is_local_user:
                    info = "(self)"

                detail_item_row.label(
                    text="{} {}".format(username, info))

                if not is_local_user:
                    detail_item_row.operator(
                        "session.snapview",
                        text="",
                        icon='VIEW_CAMERA').target_client = key
                row = layout.row()
        else:
            row.label(text="Empty")

        row = layout.row()


def draw_property(context, parent, property_uuid, level=0):
    settings = context.window_manager.session
    item = operators.client.get(uuid=property_uuid)

    if item.str_type == 'BlUser' or item.state == ERROR:
        return

    area_msg = parent.row(align=True)
    if level > 0:
        for i in range(level):
            area_msg.label(text="")
    line = area_msg.box()

    name = item.buffer['name']

    detail_item_box = line.row(align=True)

    if item.state == FETCHED:
        detail_item_box.operator(
            "session.apply",
            text="",
            icon=PROP_STATES[item.state]).target = item.uuid
    else:
        detail_item_box.label(text="", icon=PROP_STATES[item.state])

    detail_item_box.label(text="", icon=item.icon)
    detail_item_box.label(text="{} ".format(name))

    right_icon = "DECORATE_UNLOCKED"
    if item.owner != settings.username:
        right_icon = "DECORATE_LOCKED"

    ro = detail_item_box.operator(
        "session.right", text="", icon=right_icon, emboss=settings.is_admin)
    ro.key = property_uuid

    detail_item_box.operator(
        "session.remove_prop", text="", icon="X").property_path = property_uuid


class SESSION_PT_outliner(bpy.types.Panel):
    bl_idname = "MULTIUSER_PROPERTIES_PT_panel"
    bl_label = "Properties"
    bl_space_type = 'VIEW_3D'
    bl_region_type = 'UI'
    bl_category = "Multiuser"

    @classmethod
    def poll(cls, context):
        return operators.client and operators.client.state == 2

    def draw_header(self, context):
        self.layout.label(text="", icon='OUTLINER_OB_GROUP_INSTANCE')

    def draw(self, context):
        layout = self.layout

        if hasattr(context.window_manager, 'session'):
            settings = context.window_manager.session

            row = layout.row()
            row.prop(settings, 'outliner_filter', text="")

            row = layout.row(align=True)
            # Property area
            # area_msg = row.box()
            client_keys = operators.client.list()

            if client_keys and len(client_keys) > 0:
                col = layout.column(align=True)
                for key in client_keys:
                    draw_property(context, col, key)

            else:
                col.label(text="Empty")


classes = (
    SESSION_PT_settings,
    SESSION_PT_settings_user,
    SESSION_PT_settings_network,
    SESSION_PT_user,
    SESSION_PT_outliner,

)


register, unregister = bpy.utils.register_classes_factory(classes)

if __name__ == "__main__":
    register()