start-os/core/src/s9pk/mod.rs

pub mod git_hash;
pub mod merkle_archive;
pub mod rpc;
pub mod v1;
pub mod v2;

use std::sync::Arc;

use tokio::io::{AsyncReadExt, AsyncSeek};
pub use v2::{S9pk, manifest};

use crate::prelude::*;
use crate::progress::FullProgressTracker;
use crate::s9pk::merkle_archive::source::{ArchiveSource, DynFileSource};
use crate::s9pk::v1::reader::S9pkReader;
use crate::s9pk::v2::compat::MAGIC_AND_VERSION;
use crate::util::io::TmpDir;

pub async fn load<S, K>(
    source: S,
    key: K,
    progress: Option<&FullProgressTracker>,
) -> Result<S9pk<DynFileSource>, Error>
where
    S: ArchiveSource,
    S::FetchAllReader: AsyncSeek + Sync,
    K: FnOnce() -> Result<ed25519_dalek::SigningKey, Error>,
{
    // TODO: return s9pk
    const MAGIC_LEN: usize = MAGIC_AND_VERSION.len();
    let mut magic = [0_u8; MAGIC_LEN];
    source.fetch(0, 3).await?.read_exact(&mut magic).await?;
    if magic == v2::compat::MAGIC_AND_VERSION {
        let phase = if let Some(progress) = progress {
            let mut phase = progress.add_phase(
                "Converting Package to V2".into(),
                Some(source.size().await.unwrap_or(60)),
            );
            phase.start();
            Some(phase)
        } else {
            None
        };
        tracing::info!("Converting package to v2 s9pk");
        let tmp_dir = TmpDir::new().await?;
        let s9pk = S9pk::from_v1(
            S9pkReader::from_reader(source.fetch_all().await?, true).await?,
            Arc::new(tmp_dir),
            key()?,
        )
        .await?;
        tracing::info!("Converted s9pk successfully");
        if let Some(mut phase) = phase {
            phase.complete();
        }
        Ok(s9pk.into_dyn())
    } else {
        Ok(S9pk::deserialize(&Arc::new(source), None).await?.into_dyn())
    }
}