Initial outline of backup/restore

src/db_service.rs

@@ -323,6 +323,42 @@ impl SqliteScanPathService {
             .collect())
     }
 
+    pub async fn insert_configurations(
+        &self,
+        configs: &[InstrumentConfiguration],
+    ) -> Result<(), sqlx::Error> {
+        let mut tx = self.pool.begin().await?;
+
+        sqlx::query!("DelETE FROM instrument").execute(&mut tx).await?;
+
+        for config in configs {
+            sqlx::query!(
+                "INSERT INTO instrument (name, scan_number, directory, scan, detector, tracker_file_extension) VALUES (?, ?, ?, ?, ?, ?)",
+                config.name,
+                config.scan_number,
+                config.directory,
+                config.scan,
+                config.detector,
+                config.tracker_file_extension
+            )
+            .execute(&mut tx)
+            .await?;
+        }
+
+        tx.commit().await?;
+        Ok(())
+    }
+}
+    pub async fn next_scan_configuration(
+        &self,
+        instrument: &str,
+        current_high: Option<u32>,
+    ) -> Result<InstrumentConfiguration, ConfigurationError> {
+        let exp = current_high.unwrap_or(0);
+        query_as!(
+            DbInstrumentConfig,
+    }
+
     pub async fn next_scan_configuration(
         &self,
         instrument: &str,

src/graphql/mod.rs

@@ -26,6 +26,7 @@ use async_graphql::{
 };
 use async_graphql_axum::{GraphQLRequest, GraphQLResponse};
 use auth::{AuthError, PolicyCheck};
+use axum::extract::State;
 use axum::http::StatusCode;
 use axum::response::{Html, IntoResponse};
 use axum::routing::{get, post};
@@ -67,14 +68,16 @@ pub async fn serve_graphql(opts: ServeOptions) {
     let schema = Schema::build(Query, Mutation, EmptySubscription)
         .extension(Tracing)
         .limit_directives(32)
-        .data(db)
+        .data(db.clone())
         .data(directory_numtracker)
         .data(opts.policy.map(PolicyCheck::new))
         .finish();
     let app = Router::new()
         // status check endpoint allows external processes to monitor status of server without
         // making graphql queries
         .route("/status", get(server_status))
+        .route("/admin/export", get(export_handler))
+        // .route("/admin/restore", post(restore_handler))
         .route("/graphql", post(graphql_handler))
         // make it obvious that /graphql isn't expected to work when visiting from a browser
         .route(
@@ -88,6 +91,7 @@ pub async fn serve_graphql(opts: ServeOptions) {
         // Interactive graphiql playground
         .route("/graphiql", get(graphiql))
         // Make it look less like something is broken when going to any other page
+        .with_state(db)
         .fallback((
             StatusCode::NOT_FOUND,
             Html(include_str!("../../static/404.html")),
@@ -102,6 +106,11 @@ pub async fn serve_graphql(opts: ServeOptions) {
         .expect("Can't serve graphql endpoint");
 }
 
+async fn export_handler(State(db): State<SqliteScanPathService>) -> String {
+    let configs = db.all_configurations().await;
+    return format!("{configs:?}");
+}
+
 async fn create_signal_handler() {
     let mut term = signal(SignalKind::terminate()).expect("Failed to create SIGTERM listener");
     let mut int = signal(SignalKind::interrupt()).expect("Failed to create SIGINT listener");