use api_helpers::uuid2fcid; use chrono; use database_schema::*; use errors::*; use fatcat_api::models::{ChangelogEntry, Editgroup, EntityEdit}; use serde_json; use uuid::Uuid; // Ugh. I thought the whole point was to *not* do this, but: // https://github.com/diesel-rs/diesel/issues/1589 pub enum EntityState { WorkInProgress, Active(i64), Redirect(Uuid, i64), Deleted, } impl EntityState { pub fn shortname(&self) -> String { match self { EntityState::WorkInProgress => "wip", EntityState::Active(_) => "active", EntityState::Redirect(_, _) => "redirect", EntityState::Deleted => "deleted", }.to_string() } } pub trait EntityIdentRow { fn state(&self) -> Result; } pub trait EntityEditRow { fn to_model(self) -> Result; } // Helper for constructing tables macro_rules! entity_structs { ($edit_table:expr, $edit_struct:ident, $ident_table:expr, $ident_struct:ident) => { #[derive(Debug, Queryable, Identifiable, Associations, AsChangeset, QueryableByName)] #[table_name = $edit_table] pub struct $edit_struct { pub id: i64, pub editgroup_id: i64, pub ident_id: Uuid, pub rev_id: Option, pub redirect_id: Option, pub extra_json: Option, } impl EntityEditRow for $edit_struct { /// Go from a row (SQL model) to an API model fn to_model(self) -> Result { Ok(EntityEdit { editgroup_id: self.editgroup_id, revision: self.rev_id, redirect_ident: self.redirect_id.map(|v| uuid2fcid(&v)), ident: uuid2fcid(&self.ident_id), edit_id: self.id, extra: self.extra_json, }) } } #[derive(Debug, Queryable, Identifiable, Associations, AsChangeset)] #[table_name = $ident_table] pub struct $ident_struct { pub id: Uuid, pub is_live: bool, pub rev_id: Option, pub redirect_id: Option, } impl EntityIdentRow for $ident_struct { fn state(&self) -> Result { if !self.is_live { return Ok(EntityState::WorkInProgress); } match (self.redirect_id, self.rev_id) { (None, None) => Ok(EntityState::Deleted), (Some(redir), Some(rev)) => Ok(EntityState::Redirect(redir, rev)), (None, Some(rev)) => Ok(EntityState::Active(rev)), _ => bail!("Invalid EntityIdentRow state"), } } } }; } #[derive(Debug, Queryable, Identifiable, Associations, AsChangeset)] #[table_name = "container_rev"] pub struct ContainerRevRow { pub id: i64, pub extra_json: Option, pub name: String, pub publisher: Option, pub issnl: Option, pub abbrev: Option, pub coden: Option, } entity_structs!( "container_edit", ContainerEditRow, "container_ident", ContainerIdentRow ); #[derive(Debug, Queryable, Identifiable, Associations, AsChangeset)] #[table_name = "creator_rev"] pub struct CreatorRevRow { pub id: i64, pub extra_json: Option, pub display_name: String, pub given_name: Option, pub surname: Option, pub orcid: Option, } entity_structs!( "creator_edit", CreatorEditRow, "creator_ident", CreatorIdentRow ); #[derive(Debug, Queryable, Identifiable, Associations, AsChangeset)] #[table_name = "file_rev"] pub struct FileRevRow { pub id: i64, pub extra_json: Option, pub size: Option, pub sha1: Option, pub sha256: Option, pub md5: Option, pub url: Option, pub mimetype: Option, } entity_structs!("file_edit", FileEditRow, "file_ident", FileIdentRow); #[derive(Debug, Queryable, Identifiable, Associations, AsChangeset)] #[table_name = "release_rev"] pub struct ReleaseRevRow { pub id: i64, pub extra_json: Option, pub work_ident_id: Uuid, pub container_ident_id: Option, pub title: String, pub release_type: Option, pub release_status: Option, pub release_date: Option, pub doi: Option, pub isbn13: Option, pub volume: Option, pub issue: Option, pub pages: Option, pub publisher: Option, pub language: Option, } entity_structs!( "release_edit", ReleaseEditRow, "release_ident", ReleaseIdentRow ); #[derive(Debug, Queryable, Identifiable, Associations, AsChangeset)] #[table_name = "work_rev"] pub struct WorkRevRow { pub id: i64, pub extra_json: Option, pub work_type: Option, pub primary_release_id: Option, } entity_structs!("work_edit", WorkEditRow, "work_ident", WorkIdentRow); #[derive(Debug, Queryable, Identifiable, Associations, AsChangeset)] #[table_name = "release_contrib"] pub struct ReleaseContribRow { pub id: i64, pub release_rev: i64, pub creator_ident_id: Option, pub role: Option, pub index: Option, pub raw: Option, } #[derive(Debug, Insertable)] #[table_name = "release_contrib"] pub struct ReleaseContribNewRow { pub release_rev: i64, pub creator_ident_id: Option, pub role: Option, pub index: Option, pub raw: Option, } #[derive(Debug, Queryable, Identifiable, Associations, AsChangeset)] #[table_name = "release_ref"] pub struct ReleaseRefRow { pub id: i64, pub release_rev: i64, pub target_release_ident_id: Option, pub index: Option, pub key: Option, pub raw: Option, pub container_title: Option, pub year: Option, pub title: Option, pub locator: Option, } #[derive(Debug, Insertable, AsChangeset)] #[table_name = "release_ref"] pub struct ReleaseRefNewRow { pub release_rev: i64, pub target_release_ident_id: Option, pub index: Option, pub key: Option, pub raw: Option, pub container_title: Option, pub year: Option, pub title: Option, pub locator: Option, } #[derive(Debug, Queryable, Insertable, Associations, AsChangeset)] #[table_name = "file_release"] pub struct FileReleaseRow { pub file_rev: i64, pub target_release_ident_id: Uuid, } #[derive(Debug, Queryable, Identifiable, Associations, AsChangeset)] #[table_name = "editgroup"] pub struct EditgroupRow { pub id: i64, pub extra_json: Option, pub editor_id: i64, pub description: Option, } impl EditgroupRow { /// Returns an Edigroup API model *without* the entity edits actually populated. Useful for, /// eg, entity history queries (where we already have the entity edit we want) pub fn to_model_partial(self) -> Editgroup { Editgroup { id: Some(self.id), editor_id: self.editor_id, description: self.description, extra: self.extra_json, edits: None, } } } #[derive(Debug, Queryable, Identifiable, Associations, AsChangeset)] #[table_name = "editor"] pub struct EditorRow { pub id: i64, pub username: String, pub is_admin: bool, pub active_editgroup_id: Option, } #[derive(Debug, Queryable, Identifiable, Associations, AsChangeset)] #[table_name = "changelog"] pub struct ChangelogRow { pub id: i64, pub editgroup_id: i64, pub timestamp: chrono::NaiveDateTime, } impl ChangelogRow { pub fn to_model(self) -> ChangelogEntry { ChangelogEntry { index: self.id, editgroup_id: self.editgroup_id, editgroup: None, timestamp: chrono::DateTime::from_utc(self.timestamp, chrono::Utc), } } }