pytex/PyTeX/format/auto_format.py

from pathlib import Path
from typing import Optional, Dict, Union, Type

from .formatting_config import FormattingConfig

from .dict_formatter import DictFormatter
from .simple_tex_formatter import SimpleTeXFormatter
from .dtx_formatter import DTXFormatter
from .pytex_formatter import PyTeXFormatter
from .git_version_info import GitVersionInfo
from .docstrip_formatter import DocStripFormatter
from .nothing_formatter import NothingFormatter
from .copy_formatter import CopyFormatter
from .default_macros import get_default_macros
from .enums import TeXType, Target


def formatter_from_file_extension(
        input_file: Path,
        config: Optional[FormattingConfig] = None,
        git_version_info: Optional[GitVersionInfo] = None,
        locate_file_config: bool = True,
        allow_infile_config: bool = True,
        default_macros: bool = True,
        target: Optional[Target] = None,
) -> PyTeXFormatter:

    extension_switcher: Dict[str, TeXType] = {
        'dtx.pytex': TeXType.TeXDocstrip,
        'dtx': TeXType.TeXDocstrip,
        'sty.pytex': TeXType.TeXPackage,
        'sty': TeXType.TeXPackage,
        'cls.pytex': TeXType.TeXClass,
        'cls': TeXType.TeXClass,
        'dict.pytex': TeXType.TeXDictionary,
        'dict': TeXType.TeXDictionary,
        'tex.pytex': TeXType.TeXDocumentation,
        'tex': TeXType.TeXDocumentation,
    }
    source_formatter_switcher: Dict[str, Type[PyTeXFormatter]] = {
        'dtx.pytex': DTXFormatter,
        'sty.pytex': SimpleTeXFormatter,
        'cls.pytex': SimpleTeXFormatter,
        'dict.pytex': DictFormatter
    }
    tex_formatter_switcher = {
        'ins': NothingFormatter,
        'drv': NothingFormatter,
        'dtx': DocStripFormatter,
        'dict': CopyFormatter,
        'cls': CopyFormatter,
        'sty': CopyFormatter,
    }
    documentation_formatter_switcher = {

    }

    if target == Target.tex_source:
        switcher = source_formatter_switcher
    elif target == Target.tex:
        switcher = tex_formatter_switcher
    elif target == Target.documentation:
        switcher = documentation_formatter_switcher
    else:
        switcher = source_formatter_switcher | tex_formatter_switcher  # Default case
    try:
        [name, extension] = input_file.name.split('.', maxsplit=1)
    except ValueError:
        raise NotImplementedError

    config.tex_type = extension_switcher[extension]  # This sets the textype from file extension

    formatter = switcher[extension](
        input_file=input_file,
        config=config,
        git_version_info=git_version_info,
        locate_file_config=locate_file_config,
        allow_infile_config=allow_infile_config
    )
    if default_macros and target == Target.tex_source:
        formatter.macros = get_default_macros(formatter.config.tex_flavour, formatter.config.tex_type)
    return formatter