From aff11938989901327fdf0198d08f97b7105e7b3b Mon Sep 17 00:00:00 2001
From: Bryan Newbold <bnewbold@robocracy.org>
Date: Wed, 23 May 2018 20:19:41 -0700
Subject: get_or_create_editgroup() helper for POSTs

---
 rust/src/api_helpers.rs | 28 ++++++++++++++++++++++++++++
 1 file changed, 28 insertions(+)
 create mode 100644 rust/src/api_helpers.rs

(limited to 'rust/src/api_helpers.rs')

diff --git a/rust/src/api_helpers.rs b/rust/src/api_helpers.rs
new file mode 100644
index 00000000..200b70a8
--- /dev/null
+++ b/rust/src/api_helpers.rs
@@ -0,0 +1,28 @@
+
+use errors::*;
+use diesel;
+use diesel::prelude::*;
+use database_models::*;
+use database_schema::{editgroup, editor};
+
+pub fn get_or_create_editgroup(editor_id: i64, conn: &PgConnection) -> Result<i64> {
+    // check for current active
+    let ed_row: EditorRow = editor::table.find(editor_id).first(conn)?;
+    if let Some(current) = ed_row.active_editgroup_id {
+        return Ok(current);
+    }
+
+    // need to insert and update
+    conn.build_transaction().run(|| {
+
+        let eg_row: EditgroupRow = diesel::insert_into(editgroup::table)
+            .values((
+                editgroup::editor_id.eq(ed_row.id),
+            ))
+            .get_result(conn)?;
+        diesel::update(editor::table.find(ed_row.id))
+            .set(editor::active_editgroup_id.eq(eg_row.id))
+            .execute(conn)?;
+        Ok(eg_row.id)
+    })
+}
-- 
cgit v1.2.3