|
| 1 | +use std::path::PathBuf; |
| 2 | + |
| 3 | +use aqora_template::DatasetMarimoTemplate; |
| 4 | +use clap::Args; |
| 5 | +use graphql_client::GraphQLQuery; |
| 6 | +use serde::Serialize; |
| 7 | + |
| 8 | +use crate::commands::GlobalArgs; |
| 9 | +use crate::error::{self, format_permission_error, Result}; |
| 10 | +use crate::graphql_client::custom_scalars::*; |
| 11 | + |
| 12 | +#[derive(Args, Debug, Serialize)] |
| 13 | +pub struct DatasetMarimo { |
| 14 | + #[arg(short, long)] |
| 15 | + version: Option<String>, |
| 16 | + slug: String, |
| 17 | + dest: Option<PathBuf>, |
| 18 | +} |
| 19 | + |
| 20 | +#[derive(GraphQLQuery)] |
| 21 | +#[graphql( |
| 22 | + query_path = "src/graphql/get_latest_dataset_version.graphql", |
| 23 | + schema_path = "schema.graphql", |
| 24 | + response_derives = "Debug" |
| 25 | +)] |
| 26 | +pub struct GetLatestDatasetVersion; |
| 27 | + |
| 28 | +pub async fn dataset_marimo(args: DatasetMarimo, global: GlobalArgs) -> Result<()> { |
| 29 | + let pb = global.spinner().with_message(format!( |
| 30 | + "Creating dataset marimo notebook for '{}'", |
| 31 | + args.slug |
| 32 | + )); |
| 33 | + |
| 34 | + let (owner, local_slug) = args |
| 35 | + .slug |
| 36 | + .split_once('/') |
| 37 | + .ok_or_else(|| error::user("Malformed slug", "Expected a slug like: {owner}/{dataset}"))?; |
| 38 | + |
| 39 | + let version = if let Some(version) = args.version { |
| 40 | + version |
| 41 | + } else { |
| 42 | + global |
| 43 | + .graphql_client() |
| 44 | + .await? |
| 45 | + .send::<GetLatestDatasetVersion>(get_latest_dataset_version::Variables { |
| 46 | + owner: owner.to_string(), |
| 47 | + local_slug: local_slug.to_string(), |
| 48 | + }) |
| 49 | + .await? |
| 50 | + .dataset_by_slug |
| 51 | + .and_then(|d| d.latest_version) |
| 52 | + .map(|v| v.version) |
| 53 | + .ok_or_else(|| { |
| 54 | + error::user( |
| 55 | + "No version found", |
| 56 | + "Please publish a version or specify a draft version", |
| 57 | + ) |
| 58 | + })? |
| 59 | + }; |
| 60 | + |
| 61 | + let dest = args.dest.unwrap_or_else(|| PathBuf::from(local_slug)); |
| 62 | + DatasetMarimoTemplate::builder() |
| 63 | + .owner(owner) |
| 64 | + .local_slug(local_slug) |
| 65 | + .version(version) |
| 66 | + .render(&dest) |
| 67 | + .map_err(|e| format_permission_error("create use case", &dest, &e))?; |
| 68 | + |
| 69 | + pb.finish_with_message(format!( |
| 70 | + "Created dataset marimo noteboook in '{}'", |
| 71 | + dest.display() |
| 72 | + )); |
| 73 | + Ok(()) |
| 74 | +} |
0 commit comments