use crate::database::edit::DbEdit; use crate::database::version::EditVersion; use crate::error::MyResult; use anyhow::anyhow; use diffy::{apply, Patch}; use rand::{distributions::Alphanumeric, thread_rng, Rng}; use url::{ParseError, Url}; pub fn generate_activity_id(domain: &Url) -> Result { let port = domain.port().unwrap(); let domain = domain.domain().unwrap(); let id: String = thread_rng() .sample_iter(&Alphanumeric) .take(7) .map(char::from) .collect(); Url::parse(&format!("http://{}:{}/objects/{}", domain, port, id)) } /// Starting from empty string, apply edits until the specified version is reached. If no version is /// given, apply all edits up to latest version. /// /// TODO: testing /// TODO: should cache all these generated versions pub fn generate_article_version(edits: &Vec, version: &EditVersion) -> MyResult { let mut generated = String::new(); if version == &EditVersion::default() { return Ok(generated); } for e in edits { let patch = Patch::from_str(&e.diff)?; generated = apply(&generated, &patch)?; if &e.hash == version { return Ok(generated); } } Err(anyhow!("failed to generate article version").into()) } #[cfg(test)] mod test { use super::*; use activitypub_federation::fetch::object_id::ObjectId; use diffy::create_patch; fn create_edits() -> MyResult> { let generate_edit = |a, b| -> MyResult { let diff = create_patch(a, b).to_string(); Ok(DbEdit { id: 0, creator_id: 0, hash: EditVersion::new(&diff)?, ap_id: ObjectId::parse("http://example.com")?, diff, article_id: 0, previous_version_id: Default::default(), }) }; Ok([ generate_edit("", "test\n")?, generate_edit("test\n", "sda\n")?, generate_edit("sda\n", "123\n")?, ] .to_vec()) } #[test] fn test_generate_article_version() -> MyResult<()> { let edits = create_edits()?; let generated = generate_article_version(&edits, &edits[1].hash)?; assert_eq!("sda\n", generated); Ok(()) } #[test] fn test_generate_invalid_version() -> MyResult<()> { let edits = create_edits()?; let generated = generate_article_version(&edits, &EditVersion::new("invalid")?); assert!(generated.is_err()); Ok(()) } #[test] fn test_generate_first_version() -> MyResult<()> { let edits = create_edits()?; let generated = generate_article_version(&edits, &EditVersion::default())?; assert_eq!("", generated); Ok(()) } }