zhaoxi
6932294ddd
撤销独立仓库，直接纳入 KiloStar monorepo 的 subprojects/ 目录下。 Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>
85 lines
2.8 KiB
Rust
85 lines
2.8 KiB
Rust
/*
|
|
* // Copyright 2026 zhaoxi826
|
|
* //
|
|
* // Licensed under the Apache License, Version 2.0 (the "License");
|
|
* // you may not use this file except in compliance with the License.
|
|
* // You may obtain a copy of the License at
|
|
* //
|
|
* // http://www.apache.org/licenses/LICENSE-2.0
|
|
* //
|
|
* // Unless required by applicable law or agreed to in writing, software
|
|
* // distributed under the License is distributed on an "AS IS" BASIS,
|
|
* // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* // See the License for the specific language governing permissions and
|
|
* // limitations under the License.
|
|
*/
|
|
|
|
use clap::{Parser, Subcommand};
|
|
use std::path::PathBuf;
|
|
use viceroy::manifest::skill::analysis::process_and_save_skill;
|
|
use viceroy::manifest::skill::model::SkillModel;
|
|
use anyhow::Result;
|
|
|
|
#[derive(Parser)]
|
|
#[command(name = "viceroy")]
|
|
#[command(about = "Pretor's plugin management tool", long_about = None)]
|
|
struct Cli {
|
|
#[command(subcommand)]
|
|
command: Commands,
|
|
}
|
|
|
|
#[derive(Subcommand)]
|
|
enum Commands {
|
|
/// Parse a skill directory, extracting SKILL.md and analyzing python files.
|
|
Parse {
|
|
/// The path to the skill directory
|
|
#[arg(short, long, value_name = "DIR")]
|
|
path: PathBuf,
|
|
},
|
|
/// Install a skill from a Git repository and parse it
|
|
Install {
|
|
/// The Git repository URL
|
|
url: String,
|
|
|
|
/// Subdirectory path inside the repo (default is root)
|
|
#[arg(short = 'p', long, default_value = "")]
|
|
path: String,
|
|
|
|
/// Root cache directory to clone into
|
|
#[arg(short = 'c', long, default_value = ".cache")]
|
|
cache_dir: String,
|
|
|
|
/// Output directory to move the final skill into
|
|
#[arg(short = 'o', long)]
|
|
output: Option<String>,
|
|
},
|
|
}
|
|
|
|
fn main() -> Result<()> {
|
|
let cli = Cli::parse();
|
|
|
|
match &cli.command {
|
|
Commands::Parse { path } => {
|
|
if !path.exists() || !path.is_dir() {
|
|
anyhow::bail!("Error: path {:?} does not exist or is not a directory", path);
|
|
}
|
|
println!("Parsing skill directory: {:?}", path);
|
|
process_and_save_skill(path)?;
|
|
println!("Done.");
|
|
}
|
|
Commands::Install { url, path, cache_dir, output } => {
|
|
let mut final_url = url.clone();
|
|
if !final_url.starts_with("http://") && !final_url.starts_with("https://") && !final_url.starts_with("git@") {
|
|
final_url = format!("https://github.com/{}", final_url);
|
|
}
|
|
println!("Installing skill from {} into {}", final_url, cache_dir);
|
|
let skill = SkillModel::install(final_url, cache_dir.clone(), path.clone(), output.clone());
|
|
println!("Analyzing installed skill at {}", skill.skill_path);
|
|
skill.analysis()?;
|
|
println!("Done.");
|
|
}
|
|
}
|
|
|
|
Ok(())
|
|
}
|