diff --git a/crates/zeta/src/zeta.rs b/crates/zeta/src/zeta.rs index 0be88b3c6a..6e68a957c9 100644 --- a/crates/zeta/src/zeta.rs +++ b/crates/zeta/src/zeta.rs @@ -953,20 +953,33 @@ impl LicenseDetectionWatcher { pub fn new(worktree: &Worktree, cx: &mut Context) -> Self { let (mut is_open_source_tx, is_open_source_rx) = watch::channel_with::(false); - let loaded_file_fut = worktree.load_file(Path::new("LICENSE"), cx); + const LICENSE_FILES_TO_CHECK: [&'static str; 2] = ["LICENSE", "LICENCE"]; // US and UK English spelling + + // Check if worktree is a single file, if so we do not need to check for a LICENSE file + let task = if worktree.abs_path().is_file() { + Task::ready(()) + } else { + let loaded_files_task = futures::future::join_all( + LICENSE_FILES_TO_CHECK + .iter() + .map(|file| worktree.load_file(Path::new(file), cx)), + ); + + cx.background_executor().spawn(async move { + for loaded_file in loaded_files_task.await { + if let Some(content) = loaded_file.log_err() { + if is_license_eligible_for_data_collection(&content.text) { + *is_open_source_tx.borrow_mut() = true; + break; + } + } + } + }) + }; Self { is_open_source_rx, - _is_open_source_task: cx.spawn(|_, _| async move { - let Ok(loaded_file) = loaded_file_fut.await else { - return; - }; - - let is_loaded_file_open_source_thing: bool = - is_license_eligible_for_data_collection(&loaded_file.text); - - *is_open_source_tx.borrow_mut() = is_loaded_file_open_source_thing; - }), + _is_open_source_task: task, } }