use crate::materialize::build_package_archive_bytes;
use crate::publish_version::published_registry_package_version;
use crate::transport_metadata::transport_package_metadata;
use crate::{registry_artifact_path, torrent_manifest_bytes, torrent_manifest_path};
use base64::{engine::general_purpose::STANDARD, Engine as _};
use pray_core::distribution::{parse_registry_distribution_settings, RegistryDistributionSettings};
use pray_core::registry::RegistryPackageMetadata;
use pray_core::resolve::ResolvedPackage;
use pray_core::ssh_client::{with_pray_ssh_session, SshRpcSession};
use pray_core::{PrayError, PrayResult};
use serde_json::json;
pub(crate) fn publish_to_ssh_server(
packages: &[&ResolvedPackage],
signer: &str,
signer_fingerprint: Option<&str>,
published_at: u64,
signing_key: Option<&ed25519_dalek::SigningKey>,
server_url: &str,
) -> PrayResult<()> {
with_pray_ssh_session(server_url, |session| {
let distribution = ssh_distribution_settings(session)?;
for package in packages {
let archive_bytes = build_package_archive_bytes(package)?;
let artifact_path =
registry_artifact_path(&package.declaration.name, &package.spec.version);
session.call_json(
"artifact.put",
json!({
"path": artifact_path,
"body": STANDARD.encode(&archive_bytes),
}),
)?;
if distribution.allows_torrent() {
let torrent_path = torrent_manifest_path(&artifact_path);
session.call_json(
"artifact.put",
json!({
"path": torrent_path,
"body": STANDARD.encode(&torrent_manifest_bytes(
package,
&artifact_path,
&archive_bytes,
distribution.bootstrap_trackers.clone(),
)?),
}),
)?;
}
let metadata = RegistryPackageMetadata {
name: package.declaration.name.clone(),
versions: vec![published_registry_package_version(
package,
signer,
signer_fingerprint,
published_at,
signing_key,
&artifact_path,
&archive_bytes,
)?],
};
session.call_json(
"sync.push",
json!({
"metadata": transport_package_metadata(&metadata),
}),
)?;
}
Ok(())
})
}
fn ssh_distribution_settings(
session: &mut SshRpcSession,
) -> PrayResult<RegistryDistributionSettings> {
match session.call_bytes("artifact.get", json!({ "path": "v1/distribution.json" })) {
Ok(bytes) => {
let text = String::from_utf8(bytes).map_err(|error| PrayError::Parse {
kind: "distribution config",
message: error.to_string(),
})?;
parse_registry_distribution_settings(&text)
}
Err(error)
if error.to_string().contains("not found")
|| error.to_string().contains("status 404") =>
{
Ok(RegistryDistributionSettings::default())
}
Err(error) => Err(error),
}
}