Add initial methods for storing and retrieving templates

Author: tpoliaw

.sqlx/query-74ddd4d18b2b3529839b609905af032ed741abbd5f26fb9dd0c3079a313ef958.json

@@ -0,0 +1 @@
+DROP TABLE templates;

.sqlx/query-f3b05af5405d08d1577c12fbc483d19c75504c917b65e59ae7f396508251079b.json

@@ -0,0 +1,7 @@
+-- Add new table for additional templates
+CREATE TABLE templates (
+    id INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT,
+    name TEXT NOT NULL CHECK (length(name) > 0),
+    template TEXT NOT NULL CHECK (length(name) > 0),
+    beamline INTEGER NOT NULL REFERENCES beamline(id) ON DELETE CASCADE ON UPDATE CASCADE
+)

migrations/0003_template_table.down.sql

@@ -19,7 +19,7 @@ use std::path::Path;
 pub use error::ConfigurationError;
 use error::NewConfigurationError;
 use sqlx::sqlite::{SqliteConnectOptions, SqliteRow};
-use sqlx::{query_as, FromRow, QueryBuilder, Row, Sqlite, SqlitePool};
+use sqlx::{query, query_as, FromRow, QueryBuilder, Row, Sqlite, SqlitePool};
 use tracing::{info, instrument, trace};
 
 use crate::paths::{
@@ -35,6 +35,12 @@ pub struct SqliteScanPathService {
     pool: SqlitePool,
 }
 
+#[derive(Debug)]
+pub struct NamedTemplate {
+    pub name: String,
+    pub template: String,
+}
+
 #[derive(Debug, PartialEq, Eq)]
 struct RawPathTemplate<F>(String, PhantomData<F>);
 
@@ -341,6 +347,39 @@ impl SqliteScanPathService {
         .ok_or(ConfigurationError::MissingInstrument(instrument.into()))
     }
 
+    pub async fn additional_templates(
+        &self,
+        beamline: &str,
+    ) -> Result<Vec<NamedTemplate>, ConfigurationError> {
+        Ok(query_as!(
+            NamedTemplate,
+            "SELECT templates.name, templates.template
+                FROM beamline JOIN templates ON beamline.id = templates.beamline
+                WHERE beamline.name = ?",
+            beamline
+        )
+        .fetch_all(&self.pool)
+        .await?)
+    }
+
+    pub async fn register_template(
+        &self,
+        beamline: &str,
+        name: String,
+        template: String,
+    ) -> Result<(), ConfigurationError> {
+        query!(
+            "INSERT INTO templates (name, template, beamline)
+                VALUES (?, ?, (SELECT id FROM beamline WHERE name = ?));",
+            name,
+            template,
+            beamline
+        )
+        .execute(&self.pool)
+        .await?;
+        Ok(())
+    }
+
     /// Create a db service from a new empty/schema-less DB
     #[cfg(test)]
     pub(crate) async fn uninitialised() -> Self {