80 lines
2 KiB
Rust
80 lines
2 KiB
Rust
mod ides;
|
|
mod logging;
|
|
mod plugins;
|
|
|
|
use clap::{Parser, Subcommand};
|
|
use log::info;
|
|
use std::path::PathBuf;
|
|
use tokio::try_join;
|
|
|
|
#[derive(Parser)]
|
|
struct Cli {
|
|
#[arg(short, long)]
|
|
output_path: PathBuf,
|
|
#[clap(subcommand)]
|
|
command: Command,
|
|
}
|
|
|
|
#[derive(Subcommand)]
|
|
enum Command {
|
|
/// Generate the IDE JSON files and create/update all_plugins.json
|
|
Generate,
|
|
/// Remove all plugins from all_plugins.json that are no longer used in any IDE json file.
|
|
Cleanup,
|
|
}
|
|
|
|
const PLUGIN_INDICES: &[&str] = &[
|
|
"https://downloads.marketplace.jetbrains.com/files/pluginsXMLIds.json",
|
|
"https://downloads.marketplace.jetbrains.com/files/jbPluginsXMLIds.json",
|
|
];
|
|
|
|
#[tokio::main]
|
|
async fn main() -> anyhow::Result<()> {
|
|
let cli = Cli::parse();
|
|
_ = logging::setup_logging();
|
|
info!("Starting...");
|
|
|
|
match cli.command {
|
|
Command::Generate => generate(cli).await,
|
|
Command::Cleanup => cleanup(cli).await,
|
|
}
|
|
}
|
|
|
|
async fn generate(cli: Cli) -> anyhow::Result<()> {
|
|
info!("running generate.");
|
|
let (ides, mut plugins, jb_plugins) = try_join!(
|
|
ides::collect_ids(),
|
|
plugins::index(PLUGIN_INDICES[0]),
|
|
plugins::index(PLUGIN_INDICES[1])
|
|
)?;
|
|
|
|
info!(
|
|
"Indexing {} IDE versions, {} plugins and {} Jetbrains plugins.",
|
|
ides.len(),
|
|
plugins.len(),
|
|
jb_plugins.len()
|
|
);
|
|
plugins.extend_from_slice(&jb_plugins);
|
|
|
|
info!("Loading old database.");
|
|
let mut db = plugins::db_load(&cli.output_path).await?;
|
|
info!("Beginning plugin download...");
|
|
plugins::db_update(&mut db, &ides, &plugins).await?;
|
|
info!("Saving DB...");
|
|
plugins::db_save(&cli.output_path, db).await?;
|
|
|
|
Ok(())
|
|
}
|
|
|
|
async fn cleanup(cli: Cli) -> anyhow::Result<()> {
|
|
info!("Loading database and IDE mappings.");
|
|
let mut db = plugins::db_load_full(&cli.output_path).await?;
|
|
|
|
info!("Running cleanup...");
|
|
plugins::db_cleanup(&mut db).await?;
|
|
|
|
info!("Saving DB...");
|
|
plugins::db_save(&cli.output_path, db).await?;
|
|
|
|
Ok(())
|
|
}
|