__init__.py

# -*- coding: utf-8 -*-
# Copyright (c) 2026 Ruben Lagerwerf

import subprocess
from pathlib import Path
from albert import *

md_iid = '4.0'
md_version = '1.0.0'
md_name = 'FreeRDP'
md_description = 'Launch FreeRDP sessions'
md_license = 'MIT'
md_url = 'https://github.com/RubenLWF/albert-freerdp'
md_authors = ['@RubenLWF']
md_maintainers = ['@RubenLWF']
md_bin_dependencies = ["xfreerdp3"]


class Plugin(PluginInstance, TriggerQueryHandler):
    
    hostname: str = ''
    username: str = ''
    password: str = ''
    domain: str = ''
    floatbar_enabled: bool = False
    dynamic_resolution_enabled: bool = False
    clipboard_sharing_enabled: bool = False
    multimon_enabled: bool = False

    logo_path: Path
    
    def __init__(self):
        PluginInstance.__init__(self)
        TriggerQueryHandler.__init__(self)

        # Load settings from config
        self.hostname = self.readConfig('hostname', str) or ''
        self.username = self.readConfig('username', str) or ''
        self.password = self.readConfig('password', str) or ''
        self.domain = self.readConfig('domain', str) or ''
        self.floatbar_enabled = self.readConfig('floatbar_enabled', bool) if self.readConfig('floatbar_enabled', bool) is not None else False
        self.dynamic_resolution_enabled = self.readConfig('dynamic_resolution_enabled', bool) if self.readConfig('dynamic_resolution_enabled', bool) is not None else False
        self.clipboard_sharing_enabled = self.readConfig('clipboard_sharing_enabled', bool) if self.readConfig('clipboard_sharing_enabled', bool) is not None else False
        self.multimon_enabled = self.readConfig('multimon_enabled', bool) if self.readConfig('multimon_enabled', bool) is not None else False

        # Get the icon path
        plugin_dir = Path(__file__).parent
        self.logo_path = plugin_dir / 'logo.png'

    def defaultTrigger(self):
        return 'rdp '
    
    def configWidget(self):
        return [
            {
                'type': 'lineedit',
                'label': 'Server hostname',
                'property': 'hostname',
                'widget_properties': {
                    'placeholderText': 'Enter your server hostname'
                }
            },
            {
                'type': 'lineedit',
                'label': 'Domain',
                'property': 'domain',
                'widget_properties': {
                    'placeholderText': 'Enter your domain'
                }
            },
            {
                'type': 'lineedit',
                'label': 'Username',
                'property': 'username',
                'widget_properties': {
                    'placeholderText': 'Enter your username'
                }
            },
            {
                'type': 'lineedit',
                'label': 'Password',
                'property': 'password',
                'widget_properties': {
                    'placeholderText': 'Enter your password',
                    'echoMode': 2  # Password mode
                }
            },
            {
                'type': 'checkbox',
                'label': 'Enable floatbar',
                'property': 'floatbar_enabled'
            },
            {
                'type': 'checkbox',
                'label': 'Enable dynamic resolution',
                'property': 'dynamic_resolution_enabled'
            },
            {
                'type': 'checkbox',
                'label': 'Enable clipboard sharing',
                'property': 'clipboard_sharing_enabled'
            },
            {
                'type': 'checkbox',
                'label': 'Enable multi-monitor support',
                'property': 'multimon_enabled'
            }
        ]
    
    def __setattr__(self, name, value):
        # Save config when settings are updated
        if name == 'hostname':
            super().__setattr__(name, value)
            self.writeConfig('hostname', value)
        elif name == 'username':
            super().__setattr__(name, value)
            self.writeConfig('username', value)
        elif name == 'password':
            super().__setattr__(name, value)
            self.writeConfig('password', value)
        elif name == 'domain':
            super().__setattr__(name, value)
            self.writeConfig('domain', value)
        elif name == 'floatbar_enabled':
            super().__setattr__(name, value)
            self.writeConfig('floatbar_enabled', value)
        elif name == 'dynamic_resolution_enabled':
            super().__setattr__(name, value)
            self.writeConfig('dynamic_resolution_enabled', value)
        elif name == 'clipboard_sharing_enabled':
            super().__setattr__(name, value)
            self.writeConfig('clipboard_sharing_enabled', value)
        elif name == 'multimon_enabled':
            super().__setattr__(name, value)
            self.writeConfig('multimon_enabled', value)
        else:
            super().__setattr__(name, value)

    def handleTriggerQuery(self, query):
        user_input = query.string.strip().lower()
        
        items = []

        def settings_valid():
            return all([
                self.hostname,
                self.username,
                self.password,
                self.domain
            ])

        if settings_valid():
            if 'connect'.startswith(user_input) or 'up'.startswith(user_input) or not user_input:
                def connect_rdp():
                    cmd = [
                        'xfreerdp3',
                        f'/v:{self.hostname}',
                        f'/u:{self.username}',
                        f'/p:{self.password}',
                        f'/d:{self.domain}'
                    ]
                    
                    if self.floatbar_enabled:
                        cmd.append('/floatbar:show:always')
                    if self.dynamic_resolution_enabled:
                        cmd.append('/dynamic-resolution')
                    if self.clipboard_sharing_enabled:
                        cmd.append('/clipboard:direction-to:all')
                    if self.multimon_enabled:
                        cmd.append('/multimon')
                    
                    subprocess.Popen(
                        cmd,
                        start_new_session=True
                    )
                
                items.append(
                    StandardItem(
                        id='freerdp-connect',
                        text='Connect via FreeRDP',
                        subtext=f'Launch a FreeRDP session to {self.hostname}',
                        icon_factory=lambda: makeImageIcon(str(self.logo_path)),
                        actions=[
                            Action( 
                                'connect',
                                'Connect',
                                connect_rdp
                            )
                        ]
                    )
                )
        else:
            items.append(
                StandardItem(
                    id='freerdp-invalid-settings',
                    text='Configure FreeRDP',
                    subtext='Configure all required settings in the plugin settings',
                    icon_factory=lambda: makeImageIcon(str(self.logo_path))
                )
            )

        
        query.add(items)