import json
from dusk.event import Event
from PyQt5.QtWidgets import QFileDialog, QMessageBox
from PyQt5.QtCore import QTimer
import os

MAP_DEFAULT_PATH = os.path.join(os.path.dirname(__file__), '../../../assets/map/')
EDITOR_CONFIG_PATH = os.path.join(os.path.dirname(__file__), '.editor')

class MapData:
  def __init__(self):
    self.data = {}
    self.dataOriginal = {}
    self.onMapData = Event()
    self.mapFileName = None
    
    QTimer.singleShot(16, self.loadLastFile)

  def loadLastFile(self):
    if os.path.exists(EDITOR_CONFIG_PATH):
      try:
        with open(EDITOR_CONFIG_PATH, 'r') as f:
          config = json.load(f)
        lastFile = config.get('lastFile')
        if lastFile and os.path.exists(lastFile):
          self.load(lastFile)
      except Exception:
        pass

  def updateEditorConfig(self):
    try:
      config = {'lastFile': self.mapFileName if self.mapFileName else ""}
      with open(EDITOR_CONFIG_PATH, 'w') as f:
        json.dump(config, f, indent=2)
    except Exception:
      pass

  def newFile(self):
    self.data = {}
    self.dataOriginal = {}
    self.mapFileName = None
    self.onMapData.invoke(self.data)
    self.updateEditorConfig()

  def save(self, fname=None):
    if not self.mapFileName and fname is None:
      filePath, _ = QFileDialog.getSaveFileName(None, "Save Map File", MAP_DEFAULT_PATH, "Map Files (*.json)")
      if not filePath:
        return
      self.mapFileName = filePath
    if fname:
      self.mapFileName = fname
    if not self.isMapFileDirty():
      return
    try:
      with open(self.mapFileName, 'w') as f:
        json.dump(self.data, f, indent=2)
      self.dataOriginal = json.loads(json.dumps(self.data))  # Deep copy
      self.updateEditorConfig()
    except Exception as e:
      QMessageBox.critical(None, "Save Error", f"Failed to save map file:\n{e}")

  def load(self, fileName):
    try:
      with open(fileName, 'r') as f:
        self.data = json.load(f)
      self.mapFileName = fileName
      self.dataOriginal = json.loads(json.dumps(self.data))  # Deep copy
      self.onMapData.invoke(self.data)
      self.updateEditorConfig()
    except Exception as e:
      QMessageBox.critical(None, "Load Error", f"Failed to load map file:\n{e}")

  def isMapFileDirty(self):
    return json.dumps(self.data, sort_keys=True) != json.dumps(self.dataOriginal, sort_keys=True)

  def isDirty(self):
    return self.isMapFileDirty()