codetether-agent 4.7.0-a-002.4

//! Post-prompt worktree cleanup and result handling.
//!
//! Runs the prompt against the provider, then delegates
//! merge/PR logic to [`super::merge`] and cleans up.

use std::path::PathBuf;
use std::sync::Arc;

use tokio::sync::mpsc;

use crate::provider::ProviderRegistry;
use crate::session::{ImageAttachment, Session, SessionEvent};

use super::merge::{merge_locally, push_or_merge};
use super::pr_request::wants_pr;
use super::worktree::WorktreeState;

/// Run the prompt against the provider and restore the
/// original directory on completion.
pub(super) async fn run_prompt(
    session: &mut Session,
    prompt: &str,
    images: Vec<ImageAttachment>,
    event_tx: mpsc::Sender<SessionEvent>,
    registry: Arc<ProviderRegistry>,
    original_dir: Option<PathBuf>,
) -> anyhow::Result<()> {
    session
        .prompt_with_events_and_images(prompt, images, event_tx, registry)
        .await?;
    session.metadata.directory = original_dir;
    Ok(())
}

/// Handle worktree publishing or local merge after a prompt.
///
/// Creates a PR only when the prompt explicitly asks for one;
/// otherwise merges the successful worktree locally and cleans up.
pub(super) async fn handle_worktree_result(
    result: &anyhow::Result<Session>,
    worktree: Option<WorktreeState>,
    prompt: Option<&str>,
) {
    let Some((mgr, wt, base_branch)) = worktree else {
        return;
    };
    if result.is_ok() {
        match prompt.filter(|p| wants_pr(p)) {
            Some(p) => push_or_merge(&mgr, &wt, base_branch.as_deref(), Some(p)).await,
            None => merge_locally(&mgr, &wt).await,
        }
    }
    if let Err(e) = mgr.cleanup(&wt.name).await {
        tracing::warn!(error = %e, "Failed to cleanup TUI worktree");
    }
}