"""

SaveUtils.py  -  core logic for the SaveUtils FreeCAD addon.

"""

import re

import os

import datetime

import FreeCAD

from PySide import QtCore

if FreeCAD.GuiUp:

    import FreeCADGui

    from PySide import QtGui

QT_TRANSLATE_NOOP = FreeCAD.Qt.QT_TRANSLATE_NOOP

FreeCAD.Console.PrintLog("=== SaveUtils: SaveUtils.py loading ===\n")

# ---------------------------------------------------------------------------

# Regex patterns

# ---------------------------------------------------------------------------

_TS_PATTERN = re.compile(r"-\d{8}-\d{6}\.FCStd$", re.IGNORECASE)

_INC_PATTERN = re.compile(r"-(\d{1,2})\.FCStd$", re.IGNORECASE)

_EXT_PATTERN = re.compile(r"\.FCStd$", re.IGNORECASE)

def _strip_fcstd(path):

    return _EXT_PATTERN.sub("", path)

def _get_base_path():

    doc = FreeCAD.ActiveDocument

    if doc is None:

        QtGui.QMessageBox.warning(

            FreeCADGui.getMainWindow(), "SaveUtils", "No active document."

        )

        return None

    return doc.FileName or None

def _ask_base_filename(title):

    path, _ = QtGui.QFileDialog.getSaveFileName(

        FreeCADGui.getMainWindow(),

        title,

        os.path.expanduser("~/untitled"),

        "FreeCAD files (*.FCStd)",

    )

    if not path:

        return None

    return _strip_fcstd(path)

def _save_as(new_path):

    doc = FreeCAD.ActiveDocument

    try:

        doc.saveAs(new_path)

        FreeCAD.Console.PrintMessage(f"SaveUtils: saved as '{new_path}'\n")

        return True

    except Exception as exc:

        QtGui.QMessageBox.critical(

            FreeCADGui.getMainWindow(),

            "SaveUtils error",

            f"Could not save file:\n{exc}",

        )

        return False

# ---------------------------------------------------------------------------

# Commands

# ---------------------------------------------------------------------------

class CmdSaveTimestamp:

    def GetResources(self):

        return {

            "MenuText": "Save As with Timestamp",

            "ToolTip": "Save a copy appending -YYYYMMDD-HHMMSS to the filename",

            "Pixmap": "",

        }

    def IsActive(self):

        return FreeCAD.ActiveDocument is not None

    def Activated(self):

        current = _get_base_path()

        if current:

            base = (

                _TS_PATTERN.sub("", current)

                if _TS_PATTERN.search(current)

                else _strip_fcstd(current)

            )

        else:

            base = _ask_base_filename("Save As with Timestamp – choose base filename")

            if base is None:

                return

        stamp = datetime.datetime.now().strftime("%Y%m%d-%H%M%S")

        _save_as(f"{base}-{stamp}.FCStd")

class CmdSaveIncrement:

    def GetResources(self):

        return {

            "MenuText": "Save As Increment",

            "ToolTip": "Save a copy incrementing the trailing -NN counter",

            "Pixmap": "",

        }

    def IsActive(self):

        return FreeCAD.ActiveDocument is not None

    def Activated(self):

        current = _get_base_path()

        if current:

            m = _INC_PATTERN.search(current)

            if m:

                new_num = int(m.group(1)) + 1

                base = current[: m.start()]

                _save_as(f"{base}-{new_num:02d}.FCStd")

            else:

                _save_as(f"{_strip_fcstd(current)}-01.FCStd")

        else:

            base = _ask_base_filename("Save As Increment – choose base filename")

            if base is None:

                return

            _save_as(f"{base}-01.FCStd")

# ---------------------------------------------------------------------------

# Menu injection

# ---------------------------------------------------------------------------

def install():

    FreeCAD.Console.PrintLog("=== SaveUtils: install() called ===\n")

    # Register commands

    FreeCADGui.addCommand("SaveUtils_Timestamp", CmdSaveTimestamp())

    FreeCADGui.addCommand("SaveUtils_Increment", CmdSaveIncrement())

    _inject_menu()

def _inject_menu():

    mw = FreeCADGui.getMainWindow()

    if mw is None:

        FreeCAD.Console.PrintError("=== SaveUtils: main window not found ===\n")

        return

    menubar = mw.menuBar()

    file_menu = None

    # Search by objectName first (most reliable), then fall back to display text

    for action in menubar.actions():

        menu = action.menu()

        if menu is None:

            continue

        obj_name = menu.objectName().replace("&", "").strip().lower()

        disp_name = action.text().replace("&", "").strip().lower()

        if obj_name in ("file", "&file") or disp_name == "file":

            file_menu = menu

            break

    if file_menu is None:

        FreeCAD.Console.PrintError("=== SaveUtils: File menu not found ===\n")

        FreeCAD.Console.PrintMessage(

            "=== SaveUtils: menus found: "

            + str(

                [

                    (a.text(), a.menu().objectName() if a.menu() else "")

                    for a in menubar.actions()

                ]

            )

            + " ===\n"

        )

        return

    # Avoid duplicates

    for act in file_menu.actions():

        if act.text() == "Save As with Timestamp":

            FreeCAD.Console.PrintMessage(

                "=== SaveUtils: already installed, skipping ===\n"

            )

            return

    # Build plain QActions

    ts_action = QtGui.QAction("Save As with Timestamp", file_menu)

    ts_action.setToolTip("Save a copy appending -YYYYMMDD-HHMMSS to the filename")

    ts_action.triggered.connect(lambda: FreeCADGui.runCommand("SaveUtils_Timestamp"))

    inc_action = QtGui.QAction("Save As Increment", file_menu)

    inc_action.setToolTip("Save a copy incrementing the trailing -NN counter")

    inc_action.triggered.connect(lambda: FreeCADGui.runCommand("SaveUtils_Increment"))

    # Insert before the first separator

    actions = file_menu.actions()

    insert_before = next((a for a in actions if a.isSeparator()), None)

    if insert_before:

        file_menu.insertAction(insert_before, ts_action)

        file_menu.insertAction(insert_before, inc_action)

        file_menu.insertSeparator(insert_before)

    else:

        file_menu.addSeparator()

        file_menu.addAction(ts_action)

        file_menu.addAction(inc_action)

    FreeCAD.Console.PrintLog("=== SaveUtils: menu items injected ===\n")