Remote review (#1521)
This commit is contained in:
Louis Knight-Webb
committed by
GitHub
GitHub
parent
5710cc3371
commit
fd9e5e5d79
255
crates/review/src/main.rs
Normal file
255
crates/review/src/main.rs
Normal file
@@ -0,0 +1,255 @@
|
||||
mod api;
|
||||
mod archive;
|
||||
mod claude_session;
|
||||
mod config;
|
||||
mod error;
|
||||
mod github;
|
||||
mod session_selector;
|
||||
|
||||
use std::time::Duration;
|
||||
|
||||
use anyhow::Result;
|
||||
use api::{ReviewApiClient, ReviewStatus, StartRequest};
|
||||
use clap::Parser;
|
||||
use error::ReviewError;
|
||||
use github::{checkout_commit, clone_repo, get_pr_info, parse_pr_url};
|
||||
use indicatif::{ProgressBar, ProgressStyle};
|
||||
use tempfile::TempDir;
|
||||
use tracing::debug;
|
||||
use tracing_subscriber::EnvFilter;
|
||||
|
||||
const DEFAULT_API_URL: &str = "https://api.dev.vibekanban.com";
|
||||
const POLL_INTERVAL: Duration = Duration::from_secs(10);
|
||||
const TIMEOUT: Duration = Duration::from_secs(600); // 10 minutes
|
||||
|
||||
const BANNER: &str = r#"
|
||||
██████╗ ███████╗██╗ ██╗██╗███████╗██╗ ██╗ ███████╗ █████╗ ███████╗████████╗
|
||||
██╔══██╗██╔════╝██║ ██║██║██╔════╝██║ ██║ ██╔════╝██╔══██╗██╔════╝╚══██╔══╝
|
||||
██████╔╝█████╗ ██║ ██║██║█████╗ ██║ █╗ ██║ █████╗ ███████║███████╗ ██║
|
||||
██╔══██╗██╔══╝ ╚██╗ ██╔╝██║██╔══╝ ██║███╗██║ ██╔══╝ ██╔══██║╚════██║ ██║
|
||||
██║ ██║███████╗ ╚████╔╝ ██║███████╗╚███╔███╔╝██╗██║ ██║ ██║███████║ ██║
|
||||
╚═╝ ╚═╝╚══════╝ ╚═══╝ ╚═╝╚══════╝ ╚══╝╚══╝ ╚═╝╚═╝ ╚═╝ ╚═╝╚══════╝ ╚═╝
|
||||
|
||||
"#;
|
||||
|
||||
#[derive(Parser, Debug)]
|
||||
#[command(name = "review")]
|
||||
#[command(
|
||||
about = "Vibe-Kanban Review helps you review GitHub pull requests by turning them into a clear, story-driven summary instead of a wall of diffs. You provide a pull request URL, optionally link a Claude Code project for additional context, and it builds a narrative that highlights key events and important decisions, helping you prioritise what actually needs attention. It's particularly useful when reviewing large amounts of AI-generated code. Note that code is uploaded to and processed on Vibe-Kanban servers using AI."
|
||||
)]
|
||||
#[command(version)]
|
||||
struct Args {
|
||||
/// GitHub PR URL (e.g., https://github.com/owner/repo/pull/123)
|
||||
pr_url: String,
|
||||
|
||||
/// Enable verbose output
|
||||
#[arg(short, long, default_value_t = false)]
|
||||
verbose: bool,
|
||||
|
||||
/// API base URL
|
||||
#[arg(long, env = "REVIEW_API_URL", default_value = DEFAULT_API_URL)]
|
||||
api_url: String,
|
||||
}
|
||||
|
||||
fn show_disclaimer() {
|
||||
println!();
|
||||
println!(
|
||||
"DISCLAIMER: Your code will be processed on our secure remote servers, all artefacts (code, AI logs, etc...) will be deleted after 14 days."
|
||||
);
|
||||
println!();
|
||||
println!("Full terms and conditions and privacy policy: https://review.fast/terms");
|
||||
println!();
|
||||
println!("Press Enter to accept and continue...");
|
||||
|
||||
let mut input = String::new();
|
||||
std::io::stdin().read_line(&mut input).ok();
|
||||
}
|
||||
|
||||
fn prompt_email(config: &mut config::Config) -> String {
|
||||
use dialoguer::Input;
|
||||
|
||||
let mut input: Input<String> =
|
||||
Input::new().with_prompt("Email address (we'll send a link to the review here, no spam)");
|
||||
|
||||
if let Some(ref saved_email) = config.email {
|
||||
input = input.default(saved_email.clone());
|
||||
}
|
||||
|
||||
let email: String = input.interact_text().expect("Failed to read email");
|
||||
|
||||
// Save email for next time
|
||||
config.email = Some(email.clone());
|
||||
if let Err(e) = config.save() {
|
||||
debug!("Failed to save config: {}", e);
|
||||
}
|
||||
|
||||
email
|
||||
}
|
||||
|
||||
fn create_spinner(message: &str) -> ProgressBar {
|
||||
let spinner = ProgressBar::new_spinner();
|
||||
spinner.set_style(
|
||||
ProgressStyle::default_spinner()
|
||||
.template("{spinner:.green} {msg}")
|
||||
.expect("Invalid spinner template"),
|
||||
);
|
||||
spinner.set_message(message.to_string());
|
||||
spinner.enable_steady_tick(Duration::from_millis(100));
|
||||
spinner
|
||||
}
|
||||
|
||||
#[tokio::main]
|
||||
async fn main() -> Result<()> {
|
||||
let args = Args::parse();
|
||||
|
||||
// Initialize tracing
|
||||
let filter = if args.verbose {
|
||||
EnvFilter::new("debug")
|
||||
} else {
|
||||
EnvFilter::new("warn")
|
||||
};
|
||||
tracing_subscriber::fmt().with_env_filter(filter).init();
|
||||
|
||||
println!("{}", BANNER);
|
||||
|
||||
show_disclaimer();
|
||||
|
||||
debug!("Args: {:?}", args);
|
||||
|
||||
// Run the main flow and handle errors
|
||||
if let Err(e) = run(args).await {
|
||||
eprintln!("Error: {e}");
|
||||
std::process::exit(1);
|
||||
}
|
||||
|
||||
Ok(())
|
||||
}
|
||||
|
||||
async fn run(args: Args) -> Result<(), ReviewError> {
|
||||
// 1. Load config and prompt for email
|
||||
let mut config = config::Config::load();
|
||||
let email = prompt_email(&mut config);
|
||||
|
||||
// 2. Parse PR URL
|
||||
let spinner = create_spinner("Parsing PR URL...");
|
||||
let (owner, repo, pr_number) = parse_pr_url(&args.pr_url)?;
|
||||
spinner.finish_with_message(format!("PR: {owner}/{repo}#{pr_number}"));
|
||||
|
||||
// 3. Get PR info
|
||||
let spinner = create_spinner("Fetching PR information...");
|
||||
let pr_info = get_pr_info(&owner, &repo, pr_number)?;
|
||||
spinner.finish_with_message(format!("PR: {}", pr_info.title));
|
||||
|
||||
// 4. Select Claude Code session (optional)
|
||||
let session_files = match session_selector::select_session(&pr_info.head_ref_name) {
|
||||
Ok(session_selector::SessionSelection::Selected(files)) => {
|
||||
println!(" Selected {} session file(s)", files.len());
|
||||
Some(files)
|
||||
}
|
||||
Ok(session_selector::SessionSelection::Skipped) => {
|
||||
println!(" Skipping project attachment");
|
||||
None
|
||||
}
|
||||
Err(e) => {
|
||||
debug!("Session selection error: {}", e);
|
||||
println!(" No sessions found");
|
||||
None
|
||||
}
|
||||
};
|
||||
|
||||
// 5. Clone repository to temp directory
|
||||
let temp_dir = TempDir::new().map_err(|e| ReviewError::CloneFailed(e.to_string()))?;
|
||||
let repo_dir = temp_dir.path().join(&repo);
|
||||
|
||||
let spinner = create_spinner("Cloning repository...");
|
||||
clone_repo(&owner, &repo, &repo_dir)?;
|
||||
spinner.finish_with_message("Repository cloned");
|
||||
|
||||
// 6. Checkout PR head commit
|
||||
let spinner = create_spinner("Checking out PR...");
|
||||
checkout_commit(&pr_info.head_commit, &repo_dir)?;
|
||||
spinner.finish_with_message("PR checked out");
|
||||
|
||||
// 7. Create tarball (with optional session data)
|
||||
let spinner = create_spinner("Creating archive...");
|
||||
|
||||
// If sessions were selected, write .agent-messages.json to repo root
|
||||
if let Some(ref files) = session_files {
|
||||
let json_content = claude_session::concatenate_sessions_to_json(files)?;
|
||||
let agent_messages_path = repo_dir.join(".agent-messages.json");
|
||||
std::fs::write(&agent_messages_path, json_content)
|
||||
.map_err(|e| ReviewError::ArchiveFailed(e.to_string()))?;
|
||||
}
|
||||
|
||||
let payload = archive::create_tarball(&repo_dir)?;
|
||||
let size_mb = payload.len() as f64 / 1_048_576.0;
|
||||
spinner.finish_with_message(format!("Archive created ({size_mb:.2} MB)"));
|
||||
|
||||
// 8. Initialize review
|
||||
let client = ReviewApiClient::new(args.api_url.clone());
|
||||
let spinner = create_spinner("Initializing review...");
|
||||
let init_response = client.init(&args.pr_url, &email, &pr_info.title).await?;
|
||||
spinner.finish_with_message(format!("Review ID: {}", init_response.review_id));
|
||||
|
||||
// 9. Upload archive
|
||||
let spinner = create_spinner("Uploading archive...");
|
||||
client.upload(&init_response.upload_url, payload).await?;
|
||||
spinner.finish_with_message("Upload complete");
|
||||
|
||||
// 10. Start review
|
||||
let spinner = create_spinner("Starting review...");
|
||||
let codebase_url = format!("r2://{}", init_response.object_key);
|
||||
client
|
||||
.start(StartRequest {
|
||||
id: init_response.review_id.to_string(),
|
||||
title: pr_info.title,
|
||||
description: pr_info.description,
|
||||
org: pr_info.owner,
|
||||
repo: pr_info.repo,
|
||||
codebase_url,
|
||||
base_commit: pr_info.base_commit,
|
||||
})
|
||||
.await?;
|
||||
spinner.finish_with_message(format!("Review started, we'll send you an email at {} when the review is ready. This can take a few minutes, you may now close the terminal", email));
|
||||
|
||||
// 11. Poll for completion
|
||||
let spinner = create_spinner("Review in progress...");
|
||||
let start_time = std::time::Instant::now();
|
||||
|
||||
loop {
|
||||
tokio::time::sleep(POLL_INTERVAL).await;
|
||||
|
||||
// Check for timeout
|
||||
if start_time.elapsed() > TIMEOUT {
|
||||
spinner.finish_with_message("Timed out");
|
||||
return Err(ReviewError::Timeout);
|
||||
}
|
||||
|
||||
let status = client
|
||||
.poll_status(&init_response.review_id.to_string())
|
||||
.await?;
|
||||
|
||||
match status.status {
|
||||
ReviewStatus::Completed => {
|
||||
spinner.finish_with_message("Review completed!");
|
||||
break;
|
||||
}
|
||||
ReviewStatus::Failed => {
|
||||
spinner.finish_with_message("Review failed");
|
||||
let error_msg = status.error.unwrap_or_else(|| "Unknown error".to_string());
|
||||
return Err(ReviewError::ReviewFailed(error_msg));
|
||||
}
|
||||
_ => {
|
||||
let progress = status.progress.unwrap_or_else(|| status.status.to_string());
|
||||
spinner.set_message(format!("Review in progress: {progress}"));
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// 12. Print result URL
|
||||
let review_url = client.review_url(&init_response.review_id.to_string());
|
||||
println!("\nReview available at:");
|
||||
println!(" {review_url}");
|
||||
|
||||
Ok(())
|
||||
}
|
||||
Reference in New Issue
Block a user