import os
import re
import json

from .utils import set_dict_value

_THIS_DIR = os.path.dirname(os.path.abspath(__file__))
_FILE_PY_PROJECT = os.path.join(_THIS_DIR, '..', 'pyproject.toml')

def read_pyproject():
  """Reads the pyproject.toml file"""
  data = {}
  last_key = ''
  lines = []
  # I'd like to use tomllib/tomli, but I'd much rather not introduce dependencies since I've yet to
  # need to and not everyone may have 3.11. We've got a controlled config file anyway.
  with open(_FILE_PY_PROJECT, "r", encoding='utf-8') as f:
    lines = f.readlines()
  for line in lines:
    line = line.strip()
    if re.match(r'\[([^\]]+)\]$', line):
      last_key = line[1:-1]
      set_dict_value(data, last_key, data[last_key] if last_key in data else {})
      continue
    value_matches = re.match(r'^([^\s\=]+)\s*=\s*(.*)$', line)
    if value_matches:
      try:
        set_dict_value(data, f'{last_key}.{value_matches[1]}', json.loads(value_matches[2]))
      except json.decoder.JSONDecodeError:
        # We don't handle multiline arrays or curly brackets; that's ok, we know the file.
        pass

  return data

_DATA = read_pyproject()

VERSION: str = _DATA['project']['version']
NAME: str = _DATA['project']['name']
DISPAY_NAME: str = _DATA['tool']['comfy']['DisplayName']
parts = DISPAY_NAME.split("_")
CATEGORY_NAME = f"{parts[0]}"