start-os/backend/src/version/mod.rs

use std::cmp::Ordering;

use async_trait::async_trait;
use color_eyre::eyre::eyre;
use patch_db::DbHandle;
use rpc_toolkit::command;
use sqlx::PgPool;

use crate::init::InitReceipts;
use crate::Error;

mod v0_3_0;
mod v0_3_0_1;
mod v0_3_0_2;
mod v0_3_0_3;
mod v0_3_1;
mod v0_3_1_1;
mod v0_3_1_2;
mod v0_3_2;
mod v0_3_2_1;
mod v0_3_3;
mod v0_3_4;
mod v0_3_4_1;
mod v0_3_4_2;
mod v0_3_4_3;

pub type Current = v0_3_4_3::Version;

#[derive(serde::Serialize, serde::Deserialize, Debug, Clone)]
#[serde(untagged)]
enum Version {
    V0_3_0(Wrapper<v0_3_0::Version>),
    V0_3_0_1(Wrapper<v0_3_0_1::Version>),
    V0_3_0_2(Wrapper<v0_3_0_2::Version>),
    V0_3_0_3(Wrapper<v0_3_0_3::Version>),
    V0_3_1(Wrapper<v0_3_1::Version>),
    V0_3_1_1(Wrapper<v0_3_1_1::Version>),
    V0_3_1_2(Wrapper<v0_3_1_2::Version>),
    V0_3_2(Wrapper<v0_3_2::Version>),
    V0_3_2_1(Wrapper<v0_3_2_1::Version>),
    V0_3_3(Wrapper<v0_3_3::Version>),
    V0_3_4(Wrapper<v0_3_4::Version>),
    V0_3_4_1(Wrapper<v0_3_4_1::Version>),
    V0_3_4_2(Wrapper<v0_3_4_2::Version>),
    V0_3_4_3(Wrapper<v0_3_4_3::Version>),
    Other(emver::Version),
}

impl Version {
    fn from_util_version(version: crate::util::Version) -> Self {
        serde_json::to_value(version.clone())
            .and_then(serde_json::from_value)
            .unwrap_or_else(|_e| {
                tracing::warn!("Can't deserialize: {:?} and falling back to other", version);
                Version::Other(version.into_version())
            })
    }
    #[cfg(test)]
    fn as_sem_ver(&self) -> emver::Version {
        match self {
            Version::V0_3_0(Wrapper(x)) => x.semver(),
            Version::V0_3_0_1(Wrapper(x)) => x.semver(),
            Version::V0_3_0_2(Wrapper(x)) => x.semver(),
            Version::V0_3_0_3(Wrapper(x)) => x.semver(),
            Version::V0_3_1(Wrapper(x)) => x.semver(),
            Version::V0_3_1_1(Wrapper(x)) => x.semver(),
            Version::V0_3_1_2(Wrapper(x)) => x.semver(),
            Version::V0_3_2(Wrapper(x)) => x.semver(),
            Version::V0_3_2_1(Wrapper(x)) => x.semver(),
            Version::V0_3_3(Wrapper(x)) => x.semver(),
            Version::V0_3_4(Wrapper(x)) => x.semver(),
            Version::V0_3_4_1(Wrapper(x)) => x.semver(),
            Version::V0_3_4_2(Wrapper(x)) => x.semver(),
            Version::V0_3_4_3(Wrapper(x)) => x.semver(),
            Version::Other(x) => x.clone(),
        }
    }
}

#[async_trait]
pub trait VersionT
where
    Self: Sized + Send + Sync,
{
    type Previous: VersionT;
    fn new() -> Self;
    fn semver(&self) -> emver::Version;
    fn compat(&self) -> &'static emver::VersionRange;
    async fn up<Db: DbHandle>(&self, db: &mut Db, secrets: &PgPool) -> Result<(), Error>;
    async fn down<Db: DbHandle>(&self, db: &mut Db, secrets: &PgPool) -> Result<(), Error>;
    async fn commit<Db: DbHandle>(
        &self,
        db: &mut Db,
        receipts: &InitReceipts,
    ) -> Result<(), Error> {
        receipts
            .version_range
            .set(db, self.compat().clone())
            .await?;
        receipts
            .server_version
            .set(db, self.semver().into())
            .await?;

        Ok(())
    }
    async fn migrate_to<V: VersionT, Db: DbHandle>(
        &self,
        version: &V,
        db: &mut Db,
        secrets: &PgPool,
        receipts: &InitReceipts,
    ) -> Result<(), Error> {
        match self.semver().cmp(&version.semver()) {
            Ordering::Greater => {
                self.rollback_to_unchecked(version, db, secrets, receipts)
                    .await
            }
            Ordering::Less => {
                version
                    .migrate_from_unchecked(self, db, secrets, receipts)
                    .await
            }
            Ordering::Equal => Ok(()),
        }
    }
    async fn migrate_from_unchecked<V: VersionT, Db: DbHandle>(
        &self,
        version: &V,
        db: &mut Db,
        secrets: &PgPool,
        receipts: &InitReceipts,
    ) -> Result<(), Error> {
        let previous = Self::Previous::new();
        if version.semver() < previous.semver() {
            previous
                .migrate_from_unchecked(version, db, secrets, receipts)
                .await?;
        } else if version.semver() > previous.semver() {
            return Err(Error::new(
                eyre!(
                    "NO PATH FROM {}, THIS IS LIKELY A MISTAKE IN THE VERSION DEFINITION",
                    version.semver()
                ),
                crate::ErrorKind::MigrationFailed,
            ));
        }
        tracing::info!("{} -> {}", previous.semver(), self.semver(),);
        self.up(db, secrets).await?;
        self.commit(db, receipts).await?;
        Ok(())
    }
    async fn rollback_to_unchecked<V: VersionT, Db: DbHandle>(
        &self,
        version: &V,
        db: &mut Db,
        secrets: &PgPool,
        receipts: &InitReceipts,
    ) -> Result<(), Error> {
        let previous = Self::Previous::new();
        tracing::info!("{} -> {}", self.semver(), previous.semver(),);
        self.down(db, secrets).await?;
        previous.commit(db, receipts).await?;
        if version.semver() < previous.semver() {
            previous
                .rollback_to_unchecked(version, db, secrets, receipts)
                .await?;
        } else if version.semver() > previous.semver() {
            return Err(Error::new(
                eyre!(
                    "NO PATH TO {}, THIS IS LIKELY A MISTAKE IN THE VERSION DEFINITION",
                    version.semver()
                ),
                crate::ErrorKind::MigrationFailed,
            ));
        }
        Ok(())
    }
}
#[derive(Debug, Clone)]
struct Wrapper<T>(T);
impl<T> serde::Serialize for Wrapper<T>
where
    T: VersionT,
{
    fn serialize<S: serde::Serializer>(&self, serializer: S) -> Result<S::Ok, S::Error> {
        self.0.semver().serialize(serializer)
    }
}
impl<'de, T> serde::Deserialize<'de> for Wrapper<T>
where
    T: VersionT,
{
    fn deserialize<D: serde::Deserializer<'de>>(deserializer: D) -> Result<Self, D::Error> {
        let v = crate::util::Version::deserialize(deserializer)?;
        let version = T::new();
        if *v == version.semver() {
            Ok(Wrapper(version))
        } else {
            Err(serde::de::Error::custom("Mismatched Version"))
        }
    }
}

pub async fn init<Db: DbHandle>(
    db: &mut Db,
    secrets: &PgPool,
    receipts: &crate::init::InitReceipts,
) -> Result<(), Error> {
    let version = Version::from_util_version(receipts.server_version.get(db).await?);
    match version {
        Version::V0_3_0(v) => {
            v.0.migrate_to(&Current::new(), db, secrets, receipts)
                .await?
        }
        Version::V0_3_0_1(v) => {
            v.0.migrate_to(&Current::new(), db, secrets, receipts)
                .await?
        }
        Version::V0_3_0_2(v) => {
            v.0.migrate_to(&Current::new(), db, secrets, receipts)
                .await?
        }
        Version::V0_3_0_3(v) => {
            v.0.migrate_to(&Current::new(), db, secrets, receipts)
                .await?
        }
        Version::V0_3_1(v) => {
            v.0.migrate_to(&Current::new(), db, secrets, receipts)
                .await?
        }
        Version::V0_3_1_1(v) => {
            v.0.migrate_to(&Current::new(), db, secrets, receipts)
                .await?
        }
        Version::V0_3_1_2(v) => {
            v.0.migrate_to(&Current::new(), db, secrets, receipts)
                .await?
        }
        Version::V0_3_2(v) => {
            v.0.migrate_to(&Current::new(), db, secrets, receipts)
                .await?
        }
        Version::V0_3_2_1(v) => {
            v.0.migrate_to(&Current::new(), db, secrets, receipts)
                .await?
        }
        Version::V0_3_3(v) => {
            v.0.migrate_to(&Current::new(), db, secrets, receipts)
                .await?
        }
        Version::V0_3_4(v) => {
            v.0.migrate_to(&Current::new(), db, secrets, receipts)
                .await?
        }
        Version::V0_3_4_1(v) => {
            v.0.migrate_to(&Current::new(), db, secrets, receipts)
                .await?
        }
        Version::V0_3_4_2(v) => {
            v.0.migrate_to(&Current::new(), db, secrets, receipts)
                .await?
        }
        Version::V0_3_4_3(v) => {
            v.0.migrate_to(&Current::new(), db, secrets, receipts)
                .await?
        }
        Version::Other(_) => {
            return Err(Error::new(
                eyre!("Cannot downgrade"),
                crate::ErrorKind::InvalidRequest,
            ))
        }
    }
    Ok(())
}

pub const COMMIT_HASH: &str =
    git_version::git_version!(args = ["--always", "--abbrev=40", "--dirty=-modified"]);

#[command(rename = "git-info", local, metadata(authenticated = false))]
pub fn git_info() -> Result<&'static str, Error> {
    Ok(COMMIT_HASH)
}

#[cfg(test)]
mod tests {
    use proptest::prelude::*;

    use super::*;

    fn em_version() -> impl Strategy<Value = emver::Version> {
        any::<(usize, usize, usize, usize)>().prop_map(|(major, minor, patch, super_minor)| {
            emver::Version::new(major, minor, patch, super_minor)
        })
    }

    fn versions() -> impl Strategy<Value = Version> {
        prop_oneof![
            Just(Version::V0_3_0(Wrapper(v0_3_0::Version::new()))),
            Just(Version::V0_3_0_1(Wrapper(v0_3_0_1::Version::new()))),
            Just(Version::V0_3_0_2(Wrapper(v0_3_0_2::Version::new()))),
            Just(Version::V0_3_0_3(Wrapper(v0_3_0_3::Version::new()))),
            Just(Version::V0_3_1(Wrapper(v0_3_1::Version::new()))),
            Just(Version::V0_3_1_1(Wrapper(v0_3_1_1::Version::new()))),
            Just(Version::V0_3_1_2(Wrapper(v0_3_1_2::Version::new()))),
            Just(Version::V0_3_2(Wrapper(v0_3_2::Version::new()))),
            Just(Version::V0_3_2_1(Wrapper(v0_3_2_1::Version::new()))),
            Just(Version::V0_3_3(Wrapper(v0_3_3::Version::new()))),
            Just(Version::V0_3_4(Wrapper(v0_3_4::Version::new()))),
            Just(Version::V0_3_4_1(Wrapper(v0_3_4_1::Version::new()))),
            Just(Version::V0_3_4_2(Wrapper(v0_3_4_2::Version::new()))),
            Just(Version::V0_3_4_3(Wrapper(v0_3_4_3::Version::new()))),
            em_version().prop_map(Version::Other),
        ]
    }

    proptest! {
        #[test]
        fn emversion_isomorphic_version(original in em_version()) {
            let version = Version::from_util_version(original.clone().into());
            let back = version.as_sem_ver();
            prop_assert_eq!(original, back, "All versions should round trip");
        }
        #[test]
        fn version_isomorphic_em_version(version in versions()) {
            let sem_ver = version.as_sem_ver();
            let back = Version::from_util_version(sem_ver.into());
            prop_assert_eq!(format!("{:?}",version), format!("{:?}", back), "All versions should round trip");
        }
    }
}