language_modes.py

from talon import Context, Module, actions, app

from .code_languages import code_languages, code_special_file_map

mod = Module()
ctx = Context()

ctx_forced = Context()
ctx_forced.matches = r"""
tag: user.code_language_forced
"""


mod.tag("code_language_forced", "This tag is active when a language mode is forced")
mod.list("language_mode", desc="Name of a programming language mode.")

# Maps spoken forms to language ids
ctx.lists["user.language_mode"] = {
    spoken_form: language.id
    for language in code_languages
    for spoken_form in language.spoken_forms
}

# Maps extension to language ids
extension_lang_map = {
    f".{ext}": lang.id for lang in code_languages for ext in lang.extensions
}

language_ids = {lang.id for lang in code_languages}
forced_language = ""


@ctx.action_class("code")
class CodeActions:
    def language():
        file_name = actions.win.filename()
        if file_name in code_special_file_map:
            return code_special_file_map[file_name]

        file_extension = actions.win.file_ext()
        return extension_lang_map.get(file_extension, "")


@ctx_forced.action_class("code")
class ForcedCodeActions:
    def language():
        return forced_language


@mod.action_class
class Actions:
    def code_set_language_mode(language: str):
        """Sets the active language mode, and disables extension matching"""
        global forced_language
        assert language in language_ids
        forced_language = language
        # Update tags to force a context refresh. Otherwise `code.language` will not update.
        # Necessary to first set an empty list otherwise you can't move from one forced language to another.
        ctx.tags = []
        ctx.tags = ["user.code_language_forced"]

    def code_clear_language_mode():
        """Clears the active language mode, and re-enables code.language: extension matching"""
        global forced_language
        forced_language = ""
        ctx.tags = []

    def code_show_forced_language_mode():
        """Show the active language for this context"""
        if forced_language:
            app.notify(f"Forced language: {forced_language}")
        else:
            app.notify("No language forced")