use anyhow::Result;
use azure_devops_rust_api::search;
use azure_devops_rust_api::search::models::{
CodeSearchRequest, EntitySearchRequest, EntitySearchRequestBase,
};
use serde_json::json;
use std::env;
mod utils;
#[tokio::main]
async fn main() -> Result<()> {
let credential = utils::get_credential()?;
let organization = env::var("ADO_ORGANIZATION").expect("Must define ADO_ORGANIZATION");
let project = env::var("ADO_PROJECT").expect("Must define ADO_PROJECT");
let repository_name = env::args()
.nth(1)
.expect("Usage: search_code <repository-name> <branch> <search-text>");
let branch_name = env::args()
.nth(2)
.expect("Usage: search_code <repository-name> <branch> <search-text>");
let search_text = env::args()
.nth(3)
.expect("Usage: search_code <repository-name> <branch> <search-text>");
println!("Create search client");
let search_client = search::ClientBuilder::new(credential).build();
let entity_search_request_base = EntitySearchRequestBase {
filters: Some(json!({
"Project": [
project
],
"Repository": [
repository_name
],
"Branch": [
branch_name
]
})),
search_text: Some(search_text),
};
let entity_search_request = EntitySearchRequest {
entity_search_request_base,
top: Some(10), ..Default::default()
};
let code_search_request = CodeSearchRequest {
entity_search_request,
include_snippet: Some(true),
};
println!("Search...");
let search_results = search_client
.code_search_results_client()
.fetch_code_search_results(organization, code_search_request, project)
.await?
.results;
println!("Found {} results", search_results.len());
if let Some(result) = search_results.first() {
println!("Example search result:\n{result:#?}");
}
Ok(())
}