import sys
import io
from typing import Optional

from pygments.formatters import HtmlFormatter


def configure_io():
    sys.stdin = io.TextIOWrapper(sys.stdin.buffer, encoding='utf-8', errors='replace')
    sys.stdout = io.TextIOWrapper(sys.stdout.buffer, encoding='utf-8', errors='replace')


def write_style(formatter: HtmlFormatter, opt_style: Optional[str] = None):
    sys.stdout.write('<style>')

    if opt_style:
        sys.stdout.write(opt_style)

    sys.stdout.write(formatter.get_style_defs('.highlight'))
    sys.stdout.write('</style>')