From caebb18e56a9220721079577f33eb9e1b80923c4 Mon Sep 17 00:00:00 2001
From: Michael Polidori <contact@michaelpolidori.com>
Date: Wed, 24 Jul 2024 13:14:11 -0400
Subject: [PATCH] Fix page order on update/delete

---
 ckanext/pages/actions.py | 102 +++++++++++++++++++++++++--------------
 1 file changed, 67 insertions(+), 35 deletions(-)

diff --git a/ckanext/pages/actions.py b/ckanext/pages/actions.py
index b8facc2..51636ec 100644
--- a/ckanext/pages/actions.py
+++ b/ckanext/pages/actions.py
@@ -16,6 +16,10 @@
 
 from ckanext.pages import db
 
+import logging
+
+log = logging.getLogger(__name__)
+
 
 class HTMLFirstImage(HTMLParser):
     def __init__(self):
@@ -90,16 +94,22 @@ def _pages_list(context, data_dict):
 def _pages_delete(context, data_dict):
     org_id = data_dict.get('org_id')
     page = data_dict.get('page')
+
     out = db.Page.get(group_id=org_id, name=page)
+
     if out:
         session = context['session']
+
+        _reorder_pages(session, org_id, updated_page=out, remove=True)
+
         session.delete(out)
         session.commit()
 
 
 def _pages_update(context, data_dict):
     org_id = data_dict.get('org_id')
-    page = data_dict.get('page')
+    page = data_dict.get('page') or data_dict.get('name')
+
     # we need the page in the context for name validation
     context['page'] = page
     context['group_id'] = org_id
@@ -111,6 +121,7 @@ def _pages_update(context, data_dict):
         raise p.toolkit.ValidationError(errors)
 
     out = db.Page.get(group_id=org_id, name=page)
+
     if not out:
         out = db.Page()
         out.group_id = org_id
@@ -122,39 +133,6 @@ def _pages_update(context, data_dict):
     for item in items:
         setattr(out, item, data.get(item, 'page' if item == 'page_type' else None))
 
-    new_order = data.get("order")
-
-    if new_order is not None:
-        new_order = int(new_order)
-        session = context["session"]
-        existing_pages = (
-            session.query(db.Page).order_by(sa.cast(db.Page.order, sa.Integer)).all()
-        )
-
-        current_order = int(out.order) if out.order and out.order.isdigit() else None
-
-        for p in existing_pages:
-            p_order = int(p.order)
-
-            if p.name == page:
-                continue
-
-            if current_order is None or current_order == 0:
-                if p_order >= new_order:
-                    p.order = str(p_order + 1)
-            elif new_order < current_order:
-                if new_order <= p_order < current_order:
-                    p.order = str(p_order + 1)
-            elif new_order > current_order:
-                if current_order < p_order <= new_order:
-                    p.order = str(p_order - 1)
-
-            session.add(p)
-
-        out.order = str(new_order)
-        session.add(out)
-        session.commit()
-
     extras = {}
 
     extra_keys = set(schema.keys()) - set(items + ['id', 'created'])
@@ -166,12 +144,66 @@ def _pages_update(context, data_dict):
     out.modified = datetime.datetime.utcnow()
     user = model.User.get(context['user'])
     out.user_id = user.id
-    out.save()
+
     session = context['session']
+
+    _reorder_pages(session, org_id, data.get('order'), out)
+
+    out.save()
     session.add(out)
     session.commit()
 
 
+def _sanitize_order(order):
+    if order in (None, "", 'None'):
+        return None
+    try:
+        value = int(order)
+        return value if value > 0 else None
+    except (ValueError, TypeError):
+        return None
+
+
+def _reorder_pages(session, org_id, new_order=None, updated_page=None, remove=False):
+    new_order = _sanitize_order(new_order)
+
+    existing_pages = (
+        session.query(db.Page)
+        .filter(db.Page.group_id == org_id)
+        .filter(db.Page.order.isnot(None))
+        .filter(db.Page.order != "")
+        .order_by(sa.cast(db.Page.order, sa.Integer))
+        .all()
+    )
+
+    hidden_pages = (
+        session.query(db.Page)
+        .filter(db.Page.group_id == org_id)
+        .filter((db.Page.order.is_(None)) | (db.Page.order == ""))
+        .all()
+    )
+
+    if updated_page:
+        existing_pages = [p for p in existing_pages if p.name != updated_page.name]
+        hidden_pages = [p for p in hidden_pages if p.name != updated_page.name]
+
+    if updated_page and not remove:
+        if new_order is None:
+            updated_page.order = None
+            hidden_pages.append(updated_page)
+        else:
+            updated_page.order = new_order
+            existing_pages.insert(min(new_order - 1, len(existing_pages)), updated_page)
+
+    for i, p in enumerate(existing_pages, start=1):
+        p.order = i
+        session.add(p)
+
+    for p in hidden_pages:
+        p.order = None
+        session.add(p)
+
+
 def pages_upload(context, data_dict):
     """ Upload a file to the CKAN server.