main.py

import subprocess
import json
from typing import Iterable, Optional, NamedTuple
from pathlib import Path

from ulauncher.api.client.Extension import Extension
from ulauncher.api.client.EventListener import EventListener
from ulauncher.api.shared.event import KeywordQueryEvent, ItemEnterEvent
from ulauncher.api.shared.item.ExtensionResultItem import ExtensionResultItem
from ulauncher.api.shared.action.BaseAction import BaseAction
from ulauncher.api.shared.action.RenderResultListAction import RenderResultListAction
from ulauncher.api.shared.action.ExtensionCustomAction import ExtensionCustomAction


class Profile(NamedTuple):
    dir: str
    name: str
    user_name: str


class VivaldiProfileScanner:
    config_dir: Path

    def __init__(self, config_dir: str) -> None:
        self.config_dir = Path(config_dir).expanduser()

    @property
    def __local_state(self):
        with open(self.config_dir / 'Local State', 'r') as f:
            return json.load(f)

    @property
    def _profile_info_cache(self) -> Iterable[Profile]:
        for dir_, profile_data in self.__local_state['profile']['info_cache'].items():
            yield Profile(
                dir=dir_,
                name=profile_data['name'],
                user_name= profile_data['user_name'],
            )


    def scan(self, query: Optional[str]) -> Iterable[Profile]:
        for profile in self._profile_info_cache:
            if (self.config_dir / profile.dir).exists():
                if query is None or query in profile.name:
                    yield profile


class KeywordQueryEventListener(EventListener):
    def on_event(self, event: KeywordQueryEvent, extension: Extension) -> BaseAction:
        vivaldi_dir = extension.preferences['vivaldi_dir']
        scanner = VivaldiProfileScanner(vivaldi_dir)
        query: Optional[str] = event.get_argument()

        # Create launcher entries
        return RenderResultListAction([
            ExtensionResultItem(
                icon='images/icon.png',
                name=profile.name,
                description=profile.user_name,
                on_enter=ExtensionCustomAction({
                    'profile': profile.dir,
                }, keep_app_open=True)
            )
            for profile in scanner.scan(query)
        ])


class ItemEnterEventListener(EventListener):
    def on_event(self, event: ItemEnterEvent, extension: Extension):
        # Open Chrome when user selects an entry
        data = event.get_data()
        subprocess.Popen([
            extension.preferences['vivaldi_path'],
            f'--profile-directory={data["profile"]}',
        ])


class VivaldiProfileExtension(Extension):
    def __init__(self):
        super(VivaldiProfileExtension, self).__init__()
        self.subscribe(KeywordQueryEvent, KeywordQueryEventListener())
        self.subscribe(ItemEnterEvent, ItemEnterEventListener())


if __name__ == '__main__':
    VivaldiProfileExtension().run()