quilt-rs 0.31.1

use std::path::PathBuf;

use tracing::debug;
use tracing::info;

use crate::Res;
use crate::flow;
use crate::flow::push::PushResult;
use crate::io::remote::HostConfig;
use crate::io::remote::Remote;
use crate::io::storage::Storage;
use crate::lineage::InstalledPackageStatus;
use crate::lineage::PackageLineage;
use crate::manifest::Manifest;
use crate::manifest::Workflow;
use crate::paths::DomainPaths;
use quilt_uri::Namespace;

/// Options passed to the commit half of [`publish_package`].
///
/// All fields are already resolved by the caller (template rendered,
/// metadata parsed, workflow looked up) — the library does not know
/// about templates or UI state.
pub struct CommitOptions {
    pub message: String,
    pub user_meta: Option<serde_json::Value>,
    pub workflow: Option<Workflow>,
}

/// Result of a successful publish — one variant per branch of the
/// two-state decision in [`publish_package`].
///
/// Generic over the push payload: the flow layer returns
/// `PublishOutcome<PushResult>`; the public API (`InstalledPackage::publish`)
/// maps it to `PublishOutcome<PushOutcome>` via the
/// `quilt::PublishOutcome` type alias.
#[derive(Debug)]
pub enum PublishOutcome<P> {
    /// Committed pending changes, then pushed the new revision.
    CommittedAndPushed(P),
    /// Pushed a previously-committed revision without a new commit
    /// (working directory had no changes).
    PushedOnly(P),
}

impl<P> PublishOutcome<P> {
    pub fn push(&self) -> &P {
        match self {
            Self::CommittedAndPushed(p) | Self::PushedOnly(p) => p,
        }
    }
}

/// Commit any pending working-directory changes and then push the resulting
/// revision to the remote in one step.
///
/// Branches on the pre-publish state:
///
/// - `status.changes` empty and `lineage.commit` is `Some` → push only
///   (reuse the unpushed prior commit; by invariant, a set `commit` is
///   always unpushed — [`flow::push`] clears it on success)
/// - otherwise → commit (with the caller's message/metadata/workflow),
///   then push. `status.changes` may be empty: a message- or
///   metadata-only revision is a legitimate user intent, and if the
///   resulting manifest header and rows happen to match the last-pushed
///   state the content-addressed top hash will match and push is a
///   no-op — so repeat clicks with identical input don't produce
///   divergent commits on the remote.
#[allow(clippy::too_many_arguments)]
pub async fn publish_package(
    lineage: PackageLineage,
    manifest: &mut Manifest,
    paths: &DomainPaths,
    storage: &(impl Storage + Sync),
    remote: &impl Remote,
    working_dir: PathBuf,
    status: InstalledPackageStatus,
    namespace: Namespace,
    host_config: HostConfig,
    commit_opts: CommitOptions,
) -> Res<PublishOutcome<PushResult>> {
    let has_changes = !status.changes.is_empty();
    let has_pending_commit = lineage.commit.is_some();

    let (lineage, push_manifest, committed) = if has_pending_commit && !has_changes {
        debug!("✔️ Publish: reusing pending commit, skipping commit");
        (lineage, manifest.clone(), false)
    } else {
        debug!("⏳ Publish: committing local changes");
        let (lineage, new_commit) = flow::commit(
            lineage,
            manifest,
            paths,
            storage,
            working_dir,
            status,
            namespace.clone(),
            commit_opts.message,
            commit_opts.user_meta,
            commit_opts.workflow,
        )
        .await?;
        // commit wrote a new manifest to disk; reload it so push uploads
        // the new rows, not the pre-commit manifest we were handed.
        let committed_path = paths.installed_manifest(&namespace, &new_commit.hash);
        let committed_manifest = Manifest::from_path(storage, &committed_path).await?;
        debug!("✔️ Publish: commit done");
        (lineage, committed_manifest, true)
    };

    info!("⏳ Publish: pushing revision");
    let push = flow::push(
        lineage,
        push_manifest,
        paths,
        storage,
        remote,
        Some(namespace),
        host_config,
    )
    .await?;
    info!("✔️ Publish: push done");

    Ok(if committed {
        PublishOutcome::CommittedAndPushed(push)
    } else {
        PublishOutcome::PushedOnly(push)
    })
}

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

    use std::collections::BTreeMap;

    use aws_sdk_s3::primitives::ByteStream;
    use test_log::test;

    use crate::fixtures;
    use crate::io::remote::mocks::MockRemote;
    use crate::io::storage::mocks::MockStorage;
    use crate::lineage::Change;
    use crate::lineage::CommitState;
    use crate::lineage::PathState;
    use crate::manifest::ManifestRow;
    use quilt_uri::ManifestUri;
    use quilt_uri::S3Uri;

    fn manifest_uri(hash: &str) -> ManifestUri {
        ManifestUri {
            bucket: "b".to_string(),
            namespace: ("foo", "bar").into(),
            hash: hash.to_string(),
            origin: None,
        }
    }

    fn first_push_uri() -> ManifestUri {
        // Empty hash triggers the "first push" branch in push_package, so
        // no remote manifest fetch is required to run the round trip.
        manifest_uri("")
    }

    async fn seed_remote_latest(remote: &MockRemote, latest_hash: &str) -> Res {
        remote
            .put_object(
                &None,
                &S3Uri::try_from("s3://b/.quilt/named_packages/foo/bar/latest")?,
                latest_hash.as_bytes().to_vec(),
            )
            .await?;
        Ok(())
    }

    #[test(tokio::test)]
    async fn test_publish_commits_message_only_and_pushes() -> Res {
        // No working-dir changes and no pending commit, but the caller
        // supplied a message. Publish still commits (recording a
        // message-only revision) and pushes it.
        let storage = MockStorage::default();
        let remote = MockRemote::default();

        let lineage = PackageLineage {
            remote_uri: Some(first_push_uri()),
            ..PackageLineage::default()
        };
        let mut manifest = Manifest::default();

        let outcome = publish_package(
            lineage,
            &mut manifest,
            &DomainPaths::default(),
            &storage,
            &remote,
            PathBuf::default(),
            InstalledPackageStatus::default(),
            ("foo", "bar").into(),
            HostConfig::default(),
            CommitOptions {
                message: "Custom message".to_string(),
                user_meta: None,
                workflow: None,
            },
        )
        .await?;

        let push = match &outcome {
            PublishOutcome::CommittedAndPushed(p) => p,
            PublishOutcome::PushedOnly(_) => {
                panic!("expected CommittedAndPushed even without file changes");
            }
        };
        assert!(push.certified_latest);
        assert!(push.lineage.commit.is_none());
        Ok(())
    }

    #[test(tokio::test)]
    async fn test_publish_skips_commit_when_no_changes() -> Res {
        // Package with a pending local commit, first push. No new
        // working-dir changes — commit should be skipped.
        let hash = fixtures::top_hash::EMPTY_NULL_TOP_HASH.to_string();
        let lineage = PackageLineage {
            commit: Some(CommitState {
                timestamp: chrono::Utc::now(),
                hash: hash.clone(),
                prev_hashes: Vec::new(),
            }),
            remote_uri: Some(first_push_uri()),
            ..PackageLineage::default()
        };

        let storage = MockStorage::default();
        storage
            .write_byte_stream(
                PathBuf::from(format!(".quilt/packages/b/{hash}")),
                ByteStream::from_static(b"foo"),
            )
            .await?;

        let remote = MockRemote::default();
        seed_remote_latest(&remote, &hash).await?;

        let mut manifest = Manifest::default();
        manifest.header.user_meta = Some(serde_json::Value::Null);

        let outcome = publish_package(
            lineage,
            &mut manifest,
            &DomainPaths::default(),
            &storage,
            &remote,
            PathBuf::default(),
            InstalledPackageStatus::default(),
            ("foo", "bar").into(),
            HostConfig::default(),
            CommitOptions {
                message: String::new(),
                user_meta: None,
                workflow: None,
            },
        )
        .await?;

        let push = match &outcome {
            PublishOutcome::PushedOnly(p) => p,
            PublishOutcome::CommittedAndPushed(_) => {
                panic!("should skip commit when no changes");
            }
        };
        assert!(push.certified_latest);
        assert_eq!(push.lineage.remote()?.hash, hash);
        Ok(())
    }

    #[test(tokio::test)]
    async fn test_publish_push_fails_after_successful_commit() -> Res {
        // Commit succeeds, but lineage has no remote — push bails out.
        let manifest_src = fixtures::manifest_with_objects_all_sizes::manifest().await?;
        let base_record = manifest_src.get_record(&PathBuf::from("0mb.bin")).unwrap();
        let added = ManifestRow {
            logical_key: PathBuf::from("foo"),
            hash: base_record.hash.clone(),
            size: base_record.size,
            physical_key: base_record.physical_key.clone(),
            ..ManifestRow::default()
        };

        let storage = MockStorage::default();
        storage
            .write_byte_stream(PathBuf::from("/working-dir/foo"), ByteStream::default())
            .await?;

        let status = InstalledPackageStatus {
            changes: BTreeMap::from([(PathBuf::from("foo"), Change::Added(added))]),
            ..InstalledPackageStatus::default()
        };

        let lineage = PackageLineage {
            paths: BTreeMap::from([(PathBuf::from("foo"), PathState::default())]),
            ..PackageLineage::default()
        };

        let remote = MockRemote::default();

        let mut manifest = Manifest::default();

        let err = publish_package(
            lineage,
            &mut manifest,
            &DomainPaths::new(PathBuf::from("/")),
            &storage,
            &remote,
            PathBuf::from("/working-dir"),
            status,
            ("foo", "bar").into(),
            HostConfig::default(),
            CommitOptions {
                message: "published".to_string(),
                user_meta: None,
                workflow: None,
            },
        )
        .await
        .unwrap_err();
        assert!(
            err.to_string().contains("remote"),
            "expected remote-missing error, got: {err}"
        );
        Ok(())
    }

    /// Shared setup for the commit-then-push publish tests.
    ///
    /// Seeds working-dir and remote object storage with an empty file at
    /// `{hash_hex}`, and returns a `(storage, remote)` pair ready for use
    /// by `publish_package` with a first-push lineage.
    async fn setup_storages_for_commit_and_push(hash_hex: &str) -> Res<(MockStorage, MockRemote)> {
        let storage = MockStorage::default();
        storage
            .write_byte_stream(PathBuf::from("/working-dir/foo"), ByteStream::default())
            .await?;

        let remote = MockRemote::default();
        // Commit rewrites the row's physical_key to file:///.quilt/objects/{hash}.
        // Push reads that path through MockRemote's own storage
        // (see MockRemote::upload_file), so seed the same empty file there too.
        let object_path = PathBuf::from(format!("/.quilt/objects/{hash_hex}"));
        remote
            .storage
            .write_byte_stream(object_path, ByteStream::default())
            .await?;
        Ok((storage, remote))
    }

    fn first_push_lineage_with_foo() -> PackageLineage {
        PackageLineage {
            paths: BTreeMap::from([(PathBuf::from("foo"), PathState::default())]),
            remote_uri: Some(first_push_uri()),
            ..PackageLineage::default()
        }
    }

    fn row_from_fixture(fixture: &Manifest, source_key: &str) -> ManifestRow {
        let base_record = fixture.get_record(&PathBuf::from(source_key)).unwrap();
        ManifestRow {
            logical_key: PathBuf::from("foo"),
            hash: base_record.hash.clone(),
            size: base_record.size,
            physical_key: base_record.physical_key.clone(),
            ..ManifestRow::default()
        }
    }

    /// Invariants a successful first-push revision of `foo/bar` must satisfy.
    ///
    /// Factored out so the four commit-and-push tests all enforce the same
    /// post-publish state without re-listing it each time: the new manifest
    /// hash is non-empty, push cleared the pending commit, and first-push
    /// certification pinned both `base_hash` and `latest_hash` to the
    /// revision we just uploaded.
    fn assert_first_push_of_foo_bar(push: &PushResult) -> Res {
        assert!(push.certified_latest);
        let pushed = push.lineage.remote()?;
        assert!(
            !pushed.hash.is_empty(),
            "pushed manifest should have a hash"
        );
        assert_eq!(pushed.namespace, ("foo", "bar").into());
        assert!(
            push.lineage.commit.is_none(),
            "publish should clear the pending commit after a successful push"
        );
        assert_eq!(
            push.lineage.base_hash, pushed.hash,
            "first push should pin base_hash to the uploaded revision"
        );
        assert_eq!(
            push.lineage.latest_hash, pushed.hash,
            "first push should pin latest_hash to the uploaded revision"
        );
        Ok(())
    }

    #[test(tokio::test)]
    async fn test_publish_commits_and_pushes_happy_path() -> Res {
        // Full happy path: working-dir changes → commit succeeds → push succeeds.
        // Mirrors the setup of `test_publish_push_fails_after_successful_commit`,
        // but gives the lineage a first-push `remote_uri` and seeds the remote
        // so `upload_row` / `tag_latest` can complete.
        let manifest_src = fixtures::manifest_with_objects_all_sizes::manifest().await?;
        let added = row_from_fixture(&manifest_src, "0mb.bin");

        let (storage, remote) =
            setup_storages_for_commit_and_push(fixtures::objects::ZERO_HASH_HEX).await?;

        let status = InstalledPackageStatus {
            changes: BTreeMap::from([(PathBuf::from("foo"), Change::Added(added))]),
            ..InstalledPackageStatus::default()
        };

        let mut manifest = Manifest::default();

        let outcome = publish_package(
            first_push_lineage_with_foo(),
            &mut manifest,
            &DomainPaths::new(PathBuf::from("/")),
            &storage,
            &remote,
            PathBuf::from("/working-dir"),
            status,
            ("foo", "bar").into(),
            HostConfig::default(),
            CommitOptions {
                message: "published".to_string(),
                user_meta: None,
                workflow: None,
            },
        )
        .await?;

        let push = match &outcome {
            PublishOutcome::CommittedAndPushed(p) => p,
            PublishOutcome::PushedOnly(_) => {
                panic!("expected CommittedAndPushed, got PushedOnly");
            }
        };
        assert_first_push_of_foo_bar(push)
    }

    #[test(tokio::test)]
    async fn test_publish_modifies_file_and_pushes() -> Res {
        // Mirrors `flow::commit::test_modifying_and_commit` on the commit
        // side: the initial manifest has a row at "foo", and `Change::Modified`
        // swaps its content to a new hash. The resulting revision is then
        // pushed end-to-end.
        let manifest_src = fixtures::manifest_with_objects_all_sizes::manifest().await?;
        let modified = row_from_fixture(&manifest_src, "less-then-8mb.txt");

        // Seed working-dir and remote objects with the *real* less-than-8mb
        // bytes so the declared row hash matches what MockRemote computes on
        // upload. If we seeded zero bytes here, push would compute the hash
        // of zero bytes and the top-hash check ("local == pushed") would fail.
        let storage = MockStorage::default();
        storage
            .write_byte_stream(
                PathBuf::from("/working-dir/foo"),
                ByteStream::from_static(fixtures::objects::less_than_8mb()),
            )
            .await?;
        let remote = MockRemote::default();
        let object_path = PathBuf::from(format!(
            "/.quilt/objects/{}",
            fixtures::objects::LESS_THAN_8MB_HASH_HEX
        ));
        remote
            .storage
            .write_byte_stream(
                object_path,
                ByteStream::from_static(fixtures::objects::less_than_8mb()),
            )
            .await?;

        let status = InstalledPackageStatus {
            changes: BTreeMap::from([(PathBuf::from("foo"), Change::Modified(modified))]),
            ..InstalledPackageStatus::default()
        };

        // Initial manifest has "foo" pointing at zero-byte content — this
        // is the row `Change::Modified` replaces.
        let mut manifest = Manifest::default();
        manifest
            .insert_record(row_from_fixture(&manifest_src, "0mb.bin"))
            .await?;

        let outcome = publish_package(
            first_push_lineage_with_foo(),
            &mut manifest,
            &DomainPaths::new(PathBuf::from("/")),
            &storage,
            &remote,
            PathBuf::from("/working-dir"),
            status,
            ("foo", "bar").into(),
            HostConfig::default(),
            CommitOptions {
                message: "modified".to_string(),
                user_meta: None,
                workflow: None,
            },
        )
        .await?;

        let push = match &outcome {
            PublishOutcome::CommittedAndPushed(p) => p,
            PublishOutcome::PushedOnly(_) => {
                panic!("expected CommittedAndPushed, got PushedOnly");
            }
        };
        assert_first_push_of_foo_bar(push)
    }

    #[test(tokio::test)]
    async fn test_publish_removes_file_and_pushes() -> Res {
        // Mirrors `flow::commit::test_removing_and_commit`: initial manifest
        // has "foo", `Change::Removed` drops it, and publish pushes the
        // resulting empty manifest. Push uploads zero rows but still writes
        // the manifest file and certifies it as latest.
        let manifest_src = fixtures::manifest_with_objects_all_sizes::manifest().await?;
        let existing = row_from_fixture(&manifest_src, "0mb.bin");

        // Remove has nothing to copy into the object store, but setup still
        // seeds /working-dir/foo so the helper stays uniform across tests.
        let (storage, remote) =
            setup_storages_for_commit_and_push(fixtures::objects::ZERO_HASH_HEX).await?;

        let status = InstalledPackageStatus {
            changes: BTreeMap::from([(PathBuf::from("foo"), Change::Removed(existing.clone()))]),
            ..InstalledPackageStatus::default()
        };

        let mut manifest = Manifest::default();
        manifest.insert_record(existing).await?;

        let outcome = publish_package(
            first_push_lineage_with_foo(),
            &mut manifest,
            &DomainPaths::new(PathBuf::from("/")),
            &storage,
            &remote,
            PathBuf::from("/working-dir"),
            status,
            ("foo", "bar").into(),
            HostConfig::default(),
            CommitOptions {
                message: "removed".to_string(),
                user_meta: None,
                workflow: None,
            },
        )
        .await?;

        let push = match &outcome {
            PublishOutcome::CommittedAndPushed(p) => p,
            PublishOutcome::PushedOnly(_) => {
                panic!("expected CommittedAndPushed, got PushedOnly");
            }
        };
        assert_first_push_of_foo_bar(push)?;
        assert!(
            !push.lineage.paths.contains_key(&PathBuf::from("foo")),
            "lineage.paths should no longer track the removed file"
        );
        Ok(())
    }

    #[test(tokio::test)]
    async fn test_publish_with_meta_and_pushes() -> Res {
        // Mirrors `flow::commit::test_commit_meta` in the publish flow:
        // a non-empty `user_meta` and commit message flow through
        // `CommitOptions` into `flow::commit`, then push succeeds.
        let manifest_src = fixtures::manifest_with_objects_all_sizes::manifest().await?;
        let added = row_from_fixture(&manifest_src, "0mb.bin");

        let (storage, remote) =
            setup_storages_for_commit_and_push(fixtures::objects::ZERO_HASH_HEX).await?;

        let status = InstalledPackageStatus {
            changes: BTreeMap::from([(PathBuf::from("foo"), Change::Added(added))]),
            ..InstalledPackageStatus::default()
        };

        let mut manifest = Manifest::default();

        let outcome = publish_package(
            first_push_lineage_with_foo(),
            &mut manifest,
            &DomainPaths::new(PathBuf::from("/")),
            &storage,
            &remote,
            PathBuf::from("/working-dir"),
            status,
            ("foo", "bar").into(),
            HostConfig::default(),
            CommitOptions {
                message: "Lorem ipsum".to_string(),
                user_meta: Some(serde_json::json!({"lorem": "ipsum"})),
                workflow: None,
            },
        )
        .await?;

        let push = match &outcome {
            PublishOutcome::CommittedAndPushed(p) => p,
            PublishOutcome::PushedOnly(_) => {
                panic!("expected CommittedAndPushed, got PushedOnly");
            }
        };
        assert_first_push_of_foo_bar(push)
    }
}