from __future__ import annotations

from typing import TYPE_CHECKING

import yaml

from commitizen.exceptions import InvalidConfigurationError

from commitizen.git import smart_open

from .base_config import BaseConfig

if TYPE_CHECKING:

import sys

from pathlib import Path

# Self is Python 3.11+ but backported in typing-extensions

if sys.version_info < (3, 11):

from typing_extensions import Self

else:

from typing import Self

class YAMLConfig(BaseConfig):

def __init__(self, *, data: bytes | str, path: Path) -> None:

super().__init__()

self.path = path

self._parse_setting(data)

def init_empty_config_content(self) -> None:

with smart_open(

self.path, "a", encoding=self._settings["encoding"]

) as json_file:

yaml.dump({"commitizen": {}}, json_file, explicit_start=True)

def _parse_setting(self, data: bytes | str) -> None:

"""We expect to have a section in cz.yaml looking like

```

commitizen:

name: cz_conventional_commits

```

"""

import yaml.scanner

try:

doc = yaml.safe_load(data)

except yaml.YAMLError as e:

raise InvalidConfigurationError(f"Failed to parse {self.path}: {e}")

try:

self.settings.update(doc["commitizen"])

except (KeyError, TypeError):

self.is_empty_config = True

def set_key(self, key: str, value: object) -> Self:

with open(self.path, "rb") as yaml_file:

config_doc = yaml.load(yaml_file, Loader=yaml.FullLoader)

config_doc["commitizen"][key] = value

with smart_open(

self.path, "w", encoding=self._settings["encoding"]

) as yaml_file:

yaml.dump(config_doc, yaml_file, explicit_start=True)

return self