From cae1126e692914bb95dc3446342e984a1d5f9f68 Mon Sep 17 00:00:00 2001
From: amateurforger <amateurforger@kl.netlib.re>
Date: Sat, 21 Feb 2026 20:23:05 +0100
Subject: [PATCH] feat: Add helpers to switch between tables

---
 src/database/category.rs       | 8 ++++++++
 src/database/content_folder.rs | 8 ++++++++
 2 files changed, 16 insertions(+)

diff --git a/src/database/category.rs b/src/database/category.rs
index b8b73fb..083618f 100644
--- a/src/database/category.rs
+++ b/src/database/category.rs
@@ -4,6 +4,7 @@ use sea_orm::entity::prelude::*;
 use sea_orm::*;
 use snafu::prelude::*;
 
+use crate::database::operator::DatabaseOperator;
 use crate::database::{content_folder, operation::*};
 use crate::extractors::normalized_path::*;
 use crate::extractors::user::User;
@@ -64,6 +65,13 @@ impl CategoryOperator {
         Self { state, user }
     }
 
+    pub fn db(&self) -> DatabaseOperator {
+        DatabaseOperator {
+            state: self.state.clone(),
+            user: self.user.clone(),
+        }
+    }
+
     /// List categories
     ///
     /// Should not fail, unless SQLite was corrupted for some reason.
diff --git a/src/database/content_folder.rs b/src/database/content_folder.rs
index 2f37891..4d6ecaa 100644
--- a/src/database/content_folder.rs
+++ b/src/database/content_folder.rs
@@ -6,6 +6,7 @@ use snafu::prelude::*;
 
 use crate::database::category::{self, CategoryError, CategoryOperator};
 use crate::database::operation::{Operation, OperationId, OperationLog, OperationType, Table};
+use crate::database::operator::DatabaseOperator;
 use crate::extractors::user::User;
 use crate::routes::content_folder::ContentFolderForm;
 use crate::state::AppState;
@@ -64,6 +65,13 @@ impl ContentFolderOperator {
         Self { state, user }
     }
 
+    pub fn db(&self) -> DatabaseOperator {
+        DatabaseOperator {
+            state: self.state.clone(),
+            user: self.user.clone(),
+        }
+    }
+
     /// list All child folders for 1 folder
     ///
     /// Should not fail, unless SQLite was corrupted for some reason.