1 files changed, 126 insertions, 1 deletions
diff --git a/src/hydrilla/proxy/state_impl/items.py b/src/hydrilla/proxy/state_impl/items.py
index ddfef7c..919223b 100644
--- a/src/hydrilla/proxy/state_impl/items.py
+++ b/src/hydrilla/proxy/state_impl/items.py
@@ -48,7 +48,7 @@ def _set_installed_status(cursor: sqlite3.Cursor, id: str, new_status: str) \
         (new_status, id)
     )
 
-def _get_statuses(cursor: sqlite3.Cursor, id: str) -> t.Tuple[str, str]:
+def _get_statuses(cursor: sqlite3.Cursor, id: str) -> tuple[str, str]:
     cursor.execute(
         '''
         SELECT
@@ -114,6 +114,105 @@ def _uninstall_version(ref: VersionRefVar) -> t.Optional[VersionRefVar]:
 class ConcreteMappingRef(st.MappingRef):
     state: base.HaketiloStateWithFields = dc.field(hash=False, compare=False)
 
+    def _get_status_data(self, cursor: sqlite3.Cursor) \
+        -> tuple[str, str, int]:
+        cursor.execute(
+            '''
+            SELECT
+                    ms.enabled, ms.frozen, ms.active_version_id
+            FROM
+                    mapping_statuses
+            WHERE
+                    item_id = ?;
+            ''',
+            (self.id,)
+        )
+
+        rows = cursor.fetchall()
+
+        if rows == []:
+            raise st.MissingItemError()
+
+        (enabled_status, frozen_status, active_version_id), = rows
+
+        return (enabled_status, frozen_status, active_version_id)
+
+
+    def update_status(
+            self,
+            enabled:                st.EnabledStatus,
+            frozen:                 t.Optional[st.FrozenStatus] = None,
+            version_id_to_activate: t.Optional[str]             = None
+    ) -> None:
+        assert frozen is None or enabled == st.EnabledStatus.ENABLED
+        assert version_id_to_activate is None or \
+            frozen != st.FrozenStatus.NOT_FROZEN
+
+        with self.state.cursor(transaction=True) as cursor:
+            cursor.execute(
+                '''
+                SELECT
+                        enabled, frozen, active_version_id
+                FROM
+                        mapping_statuses
+                WHERE
+                        item_id = ?;
+                ''',
+                (self.id,)
+            )
+
+            rows = cursor.fetchall()
+
+            if rows == []:
+                raise st.MissingItemError()
+
+            (old_enabled_status, old_frozen_status,
+             old_active_version_id), = rows
+
+            if enabled.value == old_enabled_status and frozen is None:
+                return
+
+            new_enabled_status = enabled.value
+
+            new_frozen_status = None if frozen is None else frozen.value
+
+            if version_id_to_activate is not None:
+                new_active_version_id = version_id_to_activate
+            elif enabled == st.EnabledStatus.ENABLED:
+                new_active_version_id = str(old_active_version_id)
+            else:
+                new_active_version_id = None
+
+            cursor.execute(
+                '''
+                UPDATE
+                        mapping_statuses
+                SET
+                        enabled           = ?,
+                        frozen            = ?,
+                        active_version_id = ?
+                WHERE
+                        item_id = ?;
+                ''', (
+                    new_enabled_status,
+                    new_frozen_status,
+                    new_active_version_id,
+                    self.id
+                ))
+
+            if enabled == st.EnabledStatus.ENABLED:
+                if old_enabled_status == 'E'                           and \
+                   new_active_version_id == str(old_active_version_id) and \
+                   (new_frozen_status == 'E' or
+                    old_frozen_status == 'N' or
+                    new_frozen_status == old_frozen_status):
+                    return
+            else:
+                if old_active_version_id is None:
+                    return
+
+            self.state.recompute_dependencies([int(self.id)])
+
     def get_version_display_infos(self) \
         -> t.Sequence[st.MappingVersionDisplayInfo]:
         with self.state.cursor() as cursor:
@@ -267,6 +366,32 @@ class ConcreteMappingVersionRef(st.MappingVersionRef):
     def uninstall(self) -> t.Optional['ConcreteMappingVersionRef']:
         return _uninstall_version(self)
 
+    def update_mapping_status(
+            self,
+            enabled: st.EnabledStatus,
+            frozen:  t.Optional[st.FrozenStatus] = None
+    ) -> None:
+        with self.state.cursor(transaction=True) as cursor:
+            cursor.execute(
+                'SELECT item_id FROM item_versions WHERE item_version_id = ?;',
+                (self.id,)
+            )
+
+            rows = cursor.fetchall()
+
+            if rows == []:
+                raise st.MissingItemError()
+
+            (mapping_id,), = rows
+
+            mapping_ref = ConcreteMappingRef(str(mapping_id), self.state)
+
+            id_to_pass: t.Optional[str] = self.id
+            if enabled.value != 'E' or frozen is None or frozen.value == 'N':
+                id_to_pass = None
+
+            mapping_ref.update_status(enabled, frozen, id_to_pass)
+
     def get_all_version_display_infos(self) \
         -> t.Sequence[st.MappingVersionDisplayInfo]:
         with self.state.cursor() as cursor: