multi-user/ui.py

import bpy
from . import operators


ICONS = {'Image': 'IMAGE_DATA', 'Curve':'CURVE_DATA', 'Client':'SOLO_ON','Collection': 'FILE_FOLDER', 'Mesh': 'MESH_DATA', 'Object': 'OBJECT_DATA', 'Material': 'MATERIAL_DATA',
                  'Texture': 'TEXTURE_DATA', 'Scene': 'SCENE_DATA','AreaLight':'LIGHT_DATA', 'Light': 'LIGHT_DATA', 'SpotLight': 'LIGHT_DATA', 'SunLight': 'LIGHT_DATA', 'PointLight': 'LIGHT_DATA', 'Camera': 'CAMERA_DATA', 'Action': 'ACTION', 'Armature': 'ARMATURE_DATA', 'GreasePencil': 'GREASEPENCIL'}

PROP_STATES = [ 'ADDED',
                'COMMITED',
                'PUSHED',
                'FETCHED',
                'UP']
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

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

            # 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")


            row = layout.row()

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
        scene = context.window_manager
        row = layout.row()
        # USER SETTINGS
        row.label(text="draw overlay:")
        row.prop(settings, "enable_presence", text="")
        row = layout.row()
        row.label(text="clear blend:")
        row.prop(settings, "start_empty", 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="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
        scene = context.window_manager
        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
        scene = context.window_manager
        # Create a simple row.
        row = layout.row()
        client_keys = operators.client.list()
        if client_keys and len(client_keys) > 0:
            for key in client_keys:
                pass
                # if 'Client' in key[0]:
                #     info = ""
                #     item_box = row.box()
                #     detail_item_box = item_box.row()

                #     username = key[0].split('/')[1]
                #     if username == settings.username:
                #         info = "(self)"
                #     # detail_item_box = item_box.row()
                #     detail_item_box.label(
                #         text="{} - {}".format(username, info))

                #     if settings.username not in key[0]:
                #         detail_item_box.operator(
                #             "session.snapview", text="", icon='VIEW_CAMERA').target_client = username
                #     row = layout.row()
        else:
            row.label(text="Empty")

        row = layout.row()


def get_client_key(item):
    return item.owner

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
            scene = context.window_manager

            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 item in sorted(client_keys, key=get_client_key):
                    area_msg = col.row(align = True)
                    item_box = area_msg.box()
                    name = "None"
                    #TODO: refactor that...
                    if hasattr(item.pointer,'name'):
                        name = item.pointer.name
                    else:
                        name = item.buffer['name']

                    detail_item_box = item_box.row()
                    detail_item_box.label(text="",icon=item.icon)
                    detail_item_box.label(text="{} ".format(name))
                    detail_item_box.label(text="{} ".format(item.owner))
                    detail_item_box.label(text="{} ".format(PROP_STATES[item.state]))
                        
                    
                    # right_icon = "DECORATE_UNLOCKED"
                    # if owner == settings.username:
                    #     right_icon="DECORATE_UNLOCKED"
                    # else:
                        
                    #     right_icon="DECORATE_LOCKED"
                    
                    # ro = detail_item_box.operator("session.right", text="",emboss=settings.is_admin, icon=right_icon)
                    # ro.key = item[0]
                    # detail_item_box.operator(
                    #     "session.remove_prop", text="", icon="X").property_path = key
            else:
                area_msg.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()