Add auto-completion for team URLs

Author: Kobzol

crates/generate_blog/src/main.rs

@@ -3,7 +3,9 @@ use std::{error::Error, fmt::Display, fs, path::PathBuf, process::Command};
 use front_matter::FrontMatter;
 use inquire::{Confirm, Select, Text, validator::Validation, Autocomplete, CustomUserError};
 use inquire::autocompletion::Replacement;
-use rust_team_data::v1::Teams;
+use rust_team_data::v1::{Team, Teams};
+
+const BASE_TEAM_WEBSITE_URL: &str = "https://www.rust-lang.org/governance/teams/";
 
 fn main() -> Result<(), Box<dyn Error>> {
     println!("\nHi, thanks for writing a post for the Rust blog!\n");
@@ -91,8 +93,12 @@ fn main() -> Result<(), Box<dyn Error>> {
         }
 
         let team = team_prompt.prompt()?;
+
+        let prefilled_url = team_data.as_ref().and_then(|teams| find_team_url(teams, &team))
+            .unwrap_or_else(|| BASE_TEAM_WEBSITE_URL.to_string());
+
         let url = Text::new("At what URL can people find the team?")
-            .with_initial_value("https://www.rust-lang.org/governance/teams/")
+            .with_initial_value(&prefilled_url)
             .prompt()?;
         (Some(team), Some(url))
     };
@@ -179,6 +185,24 @@ fn load_teams() -> Result<Teams, String> {
     Ok(teams)
 }
 
+fn find_team_url(teams: &Teams, team_name: &str) -> Option<String> {
+    let Some(team) = teams.teams.get(team_name)?;
+    let top_level_team = find_top_level_team(teams, team);
+
+    // E.g. <BASE>compiler#team-miri
+    Some(format!("{}{}#team-{team_name}", BASE_TEAM_WEBSITE_URL, top_level_team.name))
+}
+
+fn find_top_level_team<'a>(teams: &'a Teams, team: &'a Team) -> &'a Team {
+    if team.top_level.unwrap_or(false) {
+        return team;
+    }
+    if let Some(parent) = team.subteam_of.as_ref().and_then(|t| teams.teams.get(t)) {
+        return find_top_level_team(teams, parent);
+    }
+    team
+}
+
 #[derive(Clone)]
 struct TeamNames(Vec<String>);