1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71

use anyhow::{anyhow, Result};
use cid::Cid;
use libipld_cbor::DagCborCodec;
use noosphere_core::data::Header;
use noosphere_sphere::{HasMutableSphereContext, SphereContentWrite, SphereCursor};
use noosphere_storage::{BlockStore, MemoryStore};

use crate::native::workspace::{FileReference, Workspace};

/// TODO(#105): We may want to change this to take an optional list of paths to
/// consider, and allow the user to rely on their shell for glob filtering
pub async fn save(workspace: &Workspace) -> Result<()> {
    workspace.ensure_sphere_initialized()?;

    let mut memory_store = MemoryStore::default();
    let mut db = workspace.db().await?;

    let (content, content_changes) = match workspace
        .get_file_content_changes(&mut memory_store)
        .await?
    {
        Some((content, content_changes)) if !content_changes.is_empty() => {
            (content, content_changes)
        }
        _ => {
            return Err(anyhow!("No changes to save"));
        }
    };

    let content_entries = memory_store.entries.lock().await;

    for (cid_bytes, block) in content_entries.iter() {
        let cid = Cid::try_from(cid_bytes.as_slice())?;
        db.put_block(&cid, block).await?;
        db.put_links::<DagCborCodec>(&cid, block).await?;
    }

    let mut sphere_context = workspace.sphere_context().await?;

    for (slug, _) in content_changes
        .new
        .iter()
        .chain(content_changes.updated.iter())
    {
        if let Some(FileReference {
            cid,
            content_type,
            extension,
        }) = content.matched.get(slug)
        {
            info!("Saving {slug}...");
            let headers = extension
                .as_ref()
                .map(|extension| vec![(Header::FileExtension.to_string(), extension.clone())]);

            sphere_context
                .link(slug, &content_type.to_string(), cid, headers)
                .await?;
        }
    }

    for slug in content_changes.removed.keys() {
        info!("Removing {slug}...");
        sphere_context.remove(slug).await?;
    }

    let cid = SphereCursor::latest(sphere_context).save(None).await?;

    info!("Save complete!\nThe latest sphere revision is {cid}");
    Ok(())
}