1use std::io::Write;
7
8use clap::{ArgAction, CommandFactory, Parser, Subcommand};
9use clap_complete::Shell;
10
11use crate::commands;
12use crate::context::{Ctx, GlobalArgs};
13use crate::errors::CliError;
14use crate::output::Format;
15
16#[derive(Debug, Parser)]
18#[command(
19 name = "qn",
20 version,
21 about = "Command-line interface for the Quicknode API.",
22 long_about = "qn lets you manage Quicknode endpoints, streams, webhooks, and the KV store from the terminal.\n\n\
23 Use `qn <noun> --help` (e.g. `qn endpoint --help`) for command details.\n\n\
24 Authentication is resolved in this order: --api-key flag, then the config file\n\
25 (--config-file path if given, else ~/.config/qn/config.toml). Run `qn auth login`\n\
26 to save a key the first time.",
27 propagate_version = true,
28 disable_help_subcommand = true,
29 disable_help_flag = true,
32 disable_version_flag = true,
33 after_help = "Examples:\n \
34 qn auth login\n \
35 qn endpoint create --chain ethereum --network mainnet\n \
36 qn endpoint list -o json\n \
37 qn endpoint logs ep-1234 --from 1h\n \
38 qn chain list\n\n\
39 AI agents: run 'qn agent context' for a machine-readable usage guide.",
40 next_help_heading = "Global options"
43)]
44pub struct Cli {
45 #[arg(long, global = true)]
47 pub api_key: Option<String>,
48
49 #[arg(long, global = true, value_name = "PATH")]
51 pub config_file: Option<std::path::PathBuf>,
52
53 #[arg(short = 'o', long = "format", global = true, value_enum)]
58 pub format: Option<Format>,
59
60 #[arg(long, global = true)]
62 pub no_color: bool,
63
64 #[arg(short, long, global = true)]
66 pub quiet: bool,
67
68 #[arg(short = 'w', long = "wide", global = true)]
72 pub wide: bool,
73
74 #[arg(short, long, global = true)]
76 pub verbose: bool,
77
78 #[arg(long, global = true)]
80 pub no_input: bool,
81
82 #[arg(long, global = true, default_value_t = 3, value_name = "N")]
86 pub retries: u32,
87
88 #[arg(short = 'y', long = "yes", global = true, action = ArgAction::Count)]
90 pub yes: u8,
91
92 #[arg(long, global = true, hide = true)]
95 pub base_url: Option<String>,
96
97 #[arg(long, global = true, hide = true)]
101 pub base_prefix: Option<String>,
102
103 #[arg(short = 'h', long, global = true, action = ArgAction::Help)]
105 pub help: Option<bool>,
106
107 #[arg(short = 'V', long, global = true, action = ArgAction::Version)]
109 pub version: Option<bool>,
110
111 #[command(subcommand)]
112 pub command: Command,
113}
114
115#[derive(Debug, Subcommand)]
116pub enum Command {
117 Auth(commands::auth::Args),
119
120 Agent(commands::agent::Args),
122
123 #[command(visible_alias = "endpoints")]
125 Endpoint(commands::endpoint::Args),
126
127 #[command(visible_alias = "teams")]
129 Team(commands::team::Args),
130
131 Usage(commands::usage::Args),
133
134 Metrics(commands::metrics::Args),
136
137 #[command(visible_alias = "chains")]
139 Chain(commands::chain::Args),
140
141 Billing(commands::billing::Args),
143
144 #[command(visible_alias = "streams")]
146 Stream(commands::stream::Args),
147
148 #[command(visible_alias = "webhooks")]
150 Webhook(commands::webhook::Args),
151
152 Kv(commands::kv::Args),
154
155 Sql(commands::sql::Args),
157
158 Rpc(commands::rpc::Args),
160
161 #[command(name = "tooling-access")]
163 ToolingAccess(commands::tooling_access::Args),
164
165 #[command(after_long_help = "### bash\n\n \
176 First, ensure that you install `bash-completion` using your package manager.\n\n \
177 After, add this to your `~/.bashrc`:\n\n \
178 eval \"$(qn completions bash)\"\n\n\
179 ### zsh\n\n \
180 Homebrew already creates this `_qn` file for you on `brew install`. To\n \
181 set it up manually, generate the script into a directory on your\n \
182 `$fpath` (Apple Silicon shown; Intel brew uses\n \
183 `/usr/local/share/zsh/site-functions`):\n\n \
184 qn completions zsh > /opt/homebrew/share/zsh/site-functions/_qn\n\n \
185 Ensure that the following is present in your `~/.zshrc`:\n\n \
186 autoload -U compinit\n \
187 compinit\n\n \
188 See the zsh manual for details:\n \
189 https://zsh.sourceforge.io/Doc/Release/Completion-System.html\n\n\
190 ### fish\n\n \
191 Generate a `qn.fish` completion script:\n\n \
192 qn completions fish > ~/.config/fish/completions/qn.fish\n\n\
193 ### PowerShell\n\n \
194 Add the following line to your profile script (`$PROFILE`):\n\n \
195 qn completions powershell | Out-String | Invoke-Expression\n\n \
196 Or append the generated script so it loads each session:\n\n \
197 qn completions powershell >> $PROFILE")]
198 Completions {
199 #[arg(value_enum)]
201 shell: Shell,
202 },
203}
204
205impl Cli {
206 pub fn global_args(&self) -> GlobalArgs {
208 GlobalArgs {
209 api_key: self.api_key.clone(),
210 config_file: self.config_file.clone(),
211 format: self.format,
212 wide: self.wide,
213 no_color: self.no_color,
216 quiet: self.quiet,
217 verbose: self.verbose,
218 no_input: self.no_input,
219 yes_count: self.yes,
220 retries: self.retries,
221 base_url: self.base_url.clone(),
222 base_prefix: self.base_prefix.clone(),
223 }
224 }
225
226 pub async fn run(self) -> Result<(), CliError> {
232 let global = self.global_args();
233 match self.command {
234 Command::Completions { shell } => {
235 let mut cmd = <Self as CommandFactory>::command();
236 let bin_name = cmd.get_name().to_string();
237 let mut out = std::io::stdout().lock();
238 clap_complete::generate(shell, &mut cmd, bin_name, &mut out);
239 out.flush()?;
240 Ok(())
241 }
242 Command::Auth(args) => commands::auth::run(args, global).await,
243 Command::Agent(args) => commands::agent::run(args, global).await,
244 Command::Endpoint(args) => {
245 commands::endpoint::run(args, Ctx::from_global(global)?).await
246 }
247 Command::Team(args) => commands::team::run(args, Ctx::from_global(global)?).await,
248 Command::Usage(args) => commands::usage::run(args, Ctx::from_global(global)?).await,
249 Command::Metrics(args) => commands::metrics::run(args, Ctx::from_global(global)?).await,
250 Command::Chain(args) => commands::chain::run(args, Ctx::from_global(global)?).await,
251 Command::Billing(args) => commands::billing::run(args, Ctx::from_global(global)?).await,
252 Command::Stream(args) => commands::stream::run(args, Ctx::from_global(global)?).await,
253 Command::Webhook(args) => commands::webhook::run(args, Ctx::from_global(global)?).await,
254 Command::Kv(args) => commands::kv::run(args, Ctx::from_global(global)?).await,
255 Command::Sql(args) => commands::sql::run(args, Ctx::from_global(global)?).await,
256 Command::Rpc(args) => commands::rpc::run(args, global).await,
259 Command::ToolingAccess(args) => {
260 commands::tooling_access::run(args, Ctx::from_global(global)?).await
261 }
262 }
263 }
264}