"""
Generator PSD - Podziekowanie dla gosci weselnych (buteleczka, Wzor 5).

Otwiera szablon PSD, podmienia teksty w Smart Object "Warstwa 2"
z zachowaniem pozycji warstw, zapisuje jako nowy PSD w folderze _gotowe.

Szablon ma 8 instancji tego samego Smart Objectu — edycja jednego SO
aktualizuje wszystkie 8 buteleczek na arkuszu automatycznie.

Wymaga: uruchomiony Adobe Photoshop, photoshop-python-api.

Uzycie:
    python buteleczki_slub_wzor5.py --imie_zenskie "Milena" --imie_meskie "Grzegorz" \\
        --data "28.03.2026" --klient "Milena Kowalska"
"""

import argparse
import os
import photoshop.api as ps


PROJEKT_DIR = os.path.join(
    r"d:\pomysloweprezenty.pl\projekty\ślub - buteleczki",
    "Podziękowanie dla gości weselnych buteleczka z nadrukiem UV - Wzór 5",
)
SZABLON_PATH = os.path.join(PROJEKT_DIR, "szablon 370x300.psd")
GOTOWE_DIR = os.path.join(PROJEKT_DIR, "_gotowe")

SMART_OBJECT_LAYER = "Warstwa 2"


def open_smart_object(app):
    """Otwiera zawartosc aktywnej warstwy Smart Object do edycji."""
    desc = ps.ActionDescriptor()
    ref = ps.ActionReference()
    ref.putEnumerated(
        app.stringIDToTypeID("layer"),
        app.stringIDToTypeID("ordinal"),
        app.stringIDToTypeID("targetEnum"),
    )
    desc.putReference(app.stringIDToTypeID("null"), ref)
    app.executeAction(app.stringIDToTypeID("placedLayerEditContents"), desc)


def set_text(layer, new_text):
    """Zmienia tekst warstwy zachowujac srodek bounding boxa (centrowane warstwy)."""
    b = [float(x) for x in layer.bounds]
    cx, cy = (b[0] + b[2]) / 2, (b[1] + b[3]) / 2
    layer.textItem.contents = new_text
    b2 = [float(x) for x in layer.bounds]
    cx2, cy2 = (b2[0] + b2[2]) / 2, (b2[1] + b2[3]) / 2
    dx, dy = cx - cx2, cy - cy2
    if dx or dy:
        layer.translate(dx, dy)


def set_layer_text(container, layer_name, new_text):
    if new_text is None:
        return
    layer = container.artLayers[layer_name]
    old = layer.textItem.contents
    set_text(layer, new_text)
    print(f'    {layer_name}: "{old}" -> "{new_text}"')


def generate(imie_zenskie, imie_meskie, data, klient, naglowek=None):
    os.makedirs(GOTOWE_DIR, exist_ok=True)
    output_path = os.path.join(GOTOWE_DIR, f"{klient}.psd")

    app = ps.Application()
    doc = app.open(SZABLON_PATH)
    print(f"Otwarto szablon: {doc.name}")

    try:
        # Aktywuj Smart Object i wejdz do edycji
        so_layer = doc.artLayers[SMART_OBJECT_LAYER]
        doc.activeLayer = so_layer
        open_smart_object(app)

        so_doc = app.activeDocument
        tekst = so_doc.layerSets["Tekst"]

        print("  [SO/Tekst]:")
        set_layer_text(tekst, "imie_meskie", imie_meskie)
        set_layer_text(tekst, "imie_zenskie", imie_zenskie)
        set_layer_text(tekst, "data", data)
        set_layer_text(tekst, "naglowek", naglowek)

        so_doc.save()
        so_doc.close()
        print("Smart Object zapisany")

        psd_opts = ps.PhotoshopSaveOptions()
        app.activeDocument.saveAs(output_path, psd_opts, True)
        print(f"Zapisano: {output_path}")
    finally:
        app.activeDocument.close(ps.SaveOptions.DoNotSaveChanges)
        print("Gotowe!")

    return output_path


def main():
    parser = argparse.ArgumentParser(
        description="Generator PSD - buteleczki weselne Wzor 5"
    )
    parser.add_argument("--imie_zenskie", required=True, help="Imie panny mlodej")
    parser.add_argument("--imie_meskie", required=True, help="Imie pana mlodego")
    parser.add_argument("--data", required=True, help="Data uroczystosci (np. 28.03.2026)")
    parser.add_argument("--klient", required=True, help="Nazwa pliku wyjsciowego")
    parser.add_argument("--naglowek", default=None,
                        help='Naglowek (opcjonalnie, domyslnie "Dziękujemy, że jesteście!")')

    args = parser.parse_args()
    generate(
        imie_zenskie=args.imie_zenskie,
        imie_meskie=args.imie_meskie,
        data=args.data,
        klient=args.klient,
        naglowek=args.naglowek,
    )


if __name__ == "__main__":
    main()