kjuulh/dagger-rs
1
0
Fork 0
mirror of https://github.com/kjuulh/dagger-rs.git synced 2024-12-23 02:43:31 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat(sdk): with return result instead of unwrap

Browse Source
This commit is contained in:
Kasper Juul Hermansen 2023-02-19 15:18:25 +01:00 committed by Kasper Juul Hermansen
parent 5d66736990
commit de063eae85
10 changed files with 116 additions and 98 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
crates/dagger-codegen/src/rust/functions.rs
View File

@ -1,5 +1,5 @@
use convert_case::{Case, Casing};
use dagger_core::introspection::FullTypeFields;
use dagger_core::introspection::{FullTypeFields, TypeRef};
use genco::prelude::rust;
use genco::quote;
use genco::tokens::quoted;
@ -37,7 +37,7 @@ pub fn format_function(funcs: &CommonFunctions, field: &FullTypeFields) -> Optio
let output_type = field
.type_
.pipe(|t| &t.type_ref)
.pipe(|t| funcs.format_output_type(t));
.pipe(|t| render_output_type(funcs, t));
Some(quote! {
$(signature)(
@ -122,6 +122,20 @@ fn render_optional_args(_funcs: &CommonFunctions, field: &FullTypeFields) -> Opt
}
}
fn render_output_type(funcs: &CommonFunctions, type_ref: &TypeRef) -> rust::Tokens {
let output_type = funcs.format_output_type(type_ref);
if type_ref_is_object(type_ref) || type_ref_is_list_of_objects(type_ref) {
return quote! {
$(output_type)
};
}
quote! {
eyre::Result<$output_type>
}
}
fn render_execution(funcs: &CommonFunctions, field: &FullTypeFields) -> rust::Tokens {
if let Some(true) = field.type_.pipe(|t| type_ref_is_object(&t.type_ref)) {
let output_type = funcs.format_output_type(&field.type_.as_ref().unwrap().type_ref);
@ -163,7 +177,7 @@ fn render_execution(funcs: &CommonFunctions, field: &FullTypeFields) -> rust::To
let graphql_client = rust::import("crate::client", "graphql_client");
quote! {
query.execute(&$graphql_client(&self.conn)).unwrap()
query.execute(&$graphql_client(&self.conn))
}
}

2
crates/dagger-sdk/examples/build-the-application/main.rs
View File

@ -14,7 +14,7 @@ fn main() -> eyre::Result<()> {
let source = client
.container(None)
.from("node:16".into())
.with_mounted_directory("/src".into(), host_source_dir.id());
.with_mounted_directory("/src".into(), host_source_dir.id()?);
let runner = source
.with_workdir("/src".into())

8
crates/dagger-sdk/examples/caching/main.rs
View File

@ -12,12 +12,12 @@ fn main() -> eyre::Result<()> {
}),
);
let node_cache = client.cache_volume("node".into()).id();
let node_cache = client.cache_volume("node".into()).id()?;
let source = client
.container(None)
.from("node:16".into())
.with_mounted_directory("/src".into(), host_source_dir.id())
.with_mounted_directory("/src".into(), host_source_dir.id()?)
.with_mounted_cache("/src/node_modules".into(), node_cache, None);
let runner = source
@ -43,11 +43,11 @@ fn main() -> eyre::Result<()> {
let ref_ = client
.container(None)
.from("nginx".into())
.with_directory("/usr/share/nginx/html".into(), build_dir.id(), None)
.with_directory("/usr/share/nginx/html".into(), build_dir.id()?, None)
.publish(
format!("ttl.sh/hello-dagger-rs-{}:1h", rng.gen::<u64>()),
None,
);
)?;
println!("published image to: {}", ref_);

5
crates/dagger-sdk/examples/existing-dockerfile/main.rs
View File

@ -1,4 +1,3 @@
use rand::Rng;
fn main() -> eyre::Result<()> {
@ -12,11 +11,11 @@ fn main() -> eyre::Result<()> {
let ref_ = client
.container(None)
.build(context_dir.id(), None)
.build(context_dir.id()?, None)
.publish(
format!("ttl.sh/hello-dagger-rs-{}:1h", rng.gen::<u64>()),
None,
);
)?;
println!("published image to: {}", ref_);

2
crates/dagger-sdk/examples/first-pipeline/main.rs
View File

@ -5,7 +5,7 @@ fn main() -> eyre::Result<()> {
.container(None)
.from("golang:1.19".into())
.with_exec(vec!["go".into(), "version".into()], None)
.stdout();
.stdout()?;
println!("Hello from Dagger and {}", version.trim());

6
crates/dagger-sdk/examples/multi-stage-build/main.rs
View File

@ -15,7 +15,7 @@ fn main() -> eyre::Result<()> {
let source = client
.container(None)
.from("node:16".into())
.with_mounted_directory("/src".into(), host_source_dir.id());
.with_mounted_directory("/src".into(), host_source_dir.id()?);
let runner = source
.with_workdir("/src".into())
@ -40,11 +40,11 @@ fn main() -> eyre::Result<()> {
let ref_ = client
.container(None)
.from("nginx".into())
.with_directory("/usr/share/nginx/html".into(), build_dir.id(), None)
.with_directory("/usr/share/nginx/html".into(), build_dir.id()?, None)
.publish(
format!("ttl.sh/hello-dagger-rs-{}:1h", rng.gen::<u64>()),
None,
);
)?;
println!("published image to: {}", ref_);

6
crates/dagger-sdk/examples/publish-the-application/main.rs
View File

@ -16,7 +16,7 @@ fn main() -> eyre::Result<()> {
let source = client
.container(None)
.from("node:16".into())
.with_mounted_directory("/src".into(), host_source_dir.id());
.with_mounted_directory("/src".into(), host_source_dir.id()?);
let runner = source
.with_workdir("/src".into())
@ -44,13 +44,13 @@ fn main() -> eyre::Result<()> {
.from("nginx".into())
.with_directory(
"/usr/share/nginx/html".into(),
client.host().directory(output.into(), None).id(),
client.host().directory(output.into(), None).id()?,
None,
)
.publish(
format!("ttl.sh/hello-dagger-rs-{}:1h", rng.gen::<u64>()),
None,
);
)?;
println!("published image to: {}", ref_);

4
crates/dagger-sdk/examples/test-the-application/main.rs
View File

@ -14,7 +14,7 @@ fn main() -> eyre::Result<()> {
let source = client
.container(None)
.from("node:16".into())
.with_mounted_directory("/src".into(), host_source_dir.id());
.with_mounted_directory("/src".into(), host_source_dir.id()?);
let runner = source
.with_workdir("/src".into())
@ -30,7 +30,7 @@ fn main() -> eyre::Result<()> {
],
None,
)
.stderr();
.stderr()?;
println!("{}", out);

158
crates/dagger-sdk/src/gen.rs
View File

@ -21,8 +21,8 @@ pub struct SecretId(String);
pub struct SocketId(String);
#[derive(Serialize, Deserialize)]
pub struct BuildArg {
pub value: String,
pub name: String,
pub value: String,
}
pub struct CacheVolume {
pub proc: Arc<Child>,
@ -31,10 +31,10 @@ pub struct CacheVolume {
}
impl CacheVolume {
pub fn id(&self) -> CacheId {
pub fn id(&self) -> eyre::Result<CacheId> {
let mut query = self.selection.select("id");
query.execute(&graphql_client(&self.conn)).unwrap()
query.execute(&graphql_client(&self.conn))
}
}
pub struct Container {
@ -112,10 +112,10 @@ impl Container {
conn: self.conn.clone(),
};
}
pub fn default_args(&self) -> Vec<String> {
pub fn default_args(&self) -> eyre::Result<Vec<String>> {
let mut query = self.selection.select("defaultArgs");
query.execute(&graphql_client(&self.conn)).unwrap()
query.execute(&graphql_client(&self.conn))
}
pub fn directory(&self, path: String) -> Directory {
let mut query = self.selection.select("directory");
@ -128,17 +128,17 @@ impl Container {
conn: self.conn.clone(),
};
}
pub fn entrypoint(&self) -> Vec<String> {
pub fn entrypoint(&self) -> eyre::Result<Vec<String>> {
let mut query = self.selection.select("entrypoint");
query.execute(&graphql_client(&self.conn)).unwrap()
query.execute(&graphql_client(&self.conn))
}
pub fn env_variable(&self, name: String) -> String {
pub fn env_variable(&self, name: String) -> eyre::Result<String> {
let mut query = self.selection.select("envVariable");
query = query.arg("name", name);
query.execute(&graphql_client(&self.conn)).unwrap()
query.execute(&graphql_client(&self.conn))
}
pub fn env_variables(&self) -> Vec<EnvVariable> {
let mut query = self.selection.select("envVariables");
@ -179,12 +179,12 @@ impl Container {
conn: self.conn.clone(),
};
}
pub fn exit_code(&self) -> isize {
pub fn exit_code(&self) -> eyre::Result<isize> {
let mut query = self.selection.select("exitCode");
query.execute(&graphql_client(&self.conn)).unwrap()
query.execute(&graphql_client(&self.conn))
}
pub fn export(&self, path: String, opts: Option<ContainerExportOpts>) -> bool {
pub fn export(&self, path: String, opts: Option<ContainerExportOpts>) -> eyre::Result<bool> {
let mut query = self.selection.select("export");
query = query.arg("path", path);
@ -194,7 +194,7 @@ impl Container {
}
}
query.execute(&graphql_client(&self.conn)).unwrap()
query.execute(&graphql_client(&self.conn))
}
pub fn file(&self, path: String) -> File {
let mut query = self.selection.select("file");
@ -227,17 +227,17 @@ impl Container {
conn: self.conn.clone(),
};
}
pub fn id(&self) -> ContainerId {
pub fn id(&self) -> eyre::Result<ContainerId> {
let mut query = self.selection.select("id");
query.execute(&graphql_client(&self.conn)).unwrap()
query.execute(&graphql_client(&self.conn))
}
pub fn label(&self, name: String) -> String {
pub fn label(&self, name: String) -> eyre::Result<String> {
let mut query = self.selection.select("label");
query = query.arg("name", name);
query.execute(&graphql_client(&self.conn)).unwrap()
query.execute(&graphql_client(&self.conn))
}
pub fn labels(&self) -> Vec<Label> {
let mut query = self.selection.select("labels");
@ -248,10 +248,10 @@ impl Container {
conn: self.conn.clone(),
}];
}
pub fn mounts(&self) -> Vec<String> {
pub fn mounts(&self) -> eyre::Result<Vec<String>> {
let mut query = self.selection.select("mounts");
query.execute(&graphql_client(&self.conn)).unwrap()
query.execute(&graphql_client(&self.conn))
}
pub fn pipeline(&self, name: String, opts: Option<ContainerPipelineOpts>) -> Container {
let mut query = self.selection.select("pipeline");
@ -269,12 +269,16 @@ impl Container {
conn: self.conn.clone(),
};
}
pub fn platform(&self) -> Platform {
pub fn platform(&self) -> eyre::Result<Platform> {
let mut query = self.selection.select("platform");
query.execute(&graphql_client(&self.conn)).unwrap()
query.execute(&graphql_client(&self.conn))
}
pub fn publish(&self, address: String, opts: Option<ContainerPublishOpts>) -> String {
pub fn publish(
&self,
address: String,
opts: Option<ContainerPublishOpts>,
) -> eyre::Result<String> {
let mut query = self.selection.select("publish");
query = query.arg("address", address);
@ -284,7 +288,7 @@ impl Container {
}
}
query.execute(&graphql_client(&self.conn)).unwrap()
query.execute(&graphql_client(&self.conn))
}
pub fn rootfs(&self) -> Directory {
let mut query = self.selection.select("rootfs");
@ -295,20 +299,20 @@ impl Container {
conn: self.conn.clone(),
};
}
pub fn stderr(&self) -> String {
pub fn stderr(&self) -> eyre::Result<String> {
let mut query = self.selection.select("stderr");
query.execute(&graphql_client(&self.conn)).unwrap()
query.execute(&graphql_client(&self.conn))
}
pub fn stdout(&self) -> String {
pub fn stdout(&self) -> eyre::Result<String> {
let mut query = self.selection.select("stdout");
query.execute(&graphql_client(&self.conn)).unwrap()
query.execute(&graphql_client(&self.conn))
}
pub fn user(&self) -> String {
pub fn user(&self) -> eyre::Result<String> {
let mut query = self.selection.select("user");
query.execute(&graphql_client(&self.conn)).unwrap()
query.execute(&graphql_client(&self.conn))
}
pub fn with_default_args(&self, opts: Option<ContainerWithDefaultArgsOpts>) -> Container {
let mut query = self.selection.select("withDefaultArgs");
@ -667,10 +671,10 @@ impl Container {
conn: self.conn.clone(),
};
}
pub fn workdir(&self) -> String {
pub fn workdir(&self) -> eyre::Result<String> {
let mut query = self.selection.select("workdir");
query.execute(&graphql_client(&self.conn)).unwrap()
query.execute(&graphql_client(&self.conn))
}
}
pub struct Directory {
@ -752,7 +756,7 @@ impl Directory {
conn: self.conn.clone(),
};
}
pub fn entries(&self, opts: Option<DirectoryEntriesOpts>) -> Vec<String> {
pub fn entries(&self, opts: Option<DirectoryEntriesOpts>) -> eyre::Result<Vec<String>> {
let mut query = self.selection.select("entries");
if let Some(opts) = opts {
@ -761,14 +765,14 @@ impl Directory {
}
}
query.execute(&graphql_client(&self.conn)).unwrap()
query.execute(&graphql_client(&self.conn))
}
pub fn export(&self, path: String) -> bool {
pub fn export(&self, path: String) -> eyre::Result<bool> {
let mut query = self.selection.select("export");
query = query.arg("path", path);
query.execute(&graphql_client(&self.conn)).unwrap()
query.execute(&graphql_client(&self.conn))
}
pub fn file(&self, path: String) -> File {
let mut query = self.selection.select("file");
@ -781,10 +785,10 @@ impl Directory {
conn: self.conn.clone(),
};
}
pub fn id(&self) -> DirectoryId {
pub fn id(&self) -> eyre::Result<DirectoryId> {
let mut query = self.selection.select("id");
query.execute(&graphql_client(&self.conn)).unwrap()
query.execute(&graphql_client(&self.conn))
}
pub fn load_project(&self, config_path: String) -> Project {
let mut query = self.selection.select("loadProject");
@ -943,15 +947,15 @@ pub struct EnvVariable {
}
impl EnvVariable {
pub fn name(&self) -> String {
pub fn name(&self) -> eyre::Result<String> {
let mut query = self.selection.select("name");
query.execute(&graphql_client(&self.conn)).unwrap()
query.execute(&graphql_client(&self.conn))
}
pub fn value(&self) -> String {
pub fn value(&self) -> eyre::Result<String> {
let mut query = self.selection.select("value");
query.execute(&graphql_client(&self.conn)).unwrap()
query.execute(&graphql_client(&self.conn))
}
}
pub struct File {
@ -961,22 +965,22 @@ pub struct File {
}
impl File {
pub fn contents(&self) -> String {
pub fn contents(&self) -> eyre::Result<String> {
let mut query = self.selection.select("contents");
query.execute(&graphql_client(&self.conn)).unwrap()
query.execute(&graphql_client(&self.conn))
}
pub fn export(&self, path: String) -> bool {
pub fn export(&self, path: String) -> eyre::Result<bool> {
let mut query = self.selection.select("export");
query = query.arg("path", path);
query.execute(&graphql_client(&self.conn)).unwrap()
query.execute(&graphql_client(&self.conn))
}
pub fn id(&self) -> FileId {
pub fn id(&self) -> eyre::Result<FileId> {
let mut query = self.selection.select("id");
query.execute(&graphql_client(&self.conn)).unwrap()
query.execute(&graphql_client(&self.conn))
}
pub fn secret(&self) -> Secret {
let mut query = self.selection.select("secret");
@ -987,10 +991,10 @@ impl File {
conn: self.conn.clone(),
};
}
pub fn size(&self) -> isize {
pub fn size(&self) -> eyre::Result<isize> {
let mut query = self.selection.select("size");
query.execute(&graphql_client(&self.conn)).unwrap()
query.execute(&graphql_client(&self.conn))
}
pub fn with_timestamps(&self, timestamp: isize) -> File {
let mut query = self.selection.select("withTimestamps");
@ -1016,10 +1020,10 @@ pub struct GitRefTreeOpts {
}
impl GitRef {
pub fn digest(&self) -> String {
pub fn digest(&self) -> eyre::Result<String> {
let mut query = self.selection.select("digest");
query.execute(&graphql_client(&self.conn)).unwrap()
query.execute(&graphql_client(&self.conn))
}
pub fn tree(&self, opts: Option<GitRefTreeOpts>) -> Directory {
let mut query = self.selection.select("tree");
@ -1058,10 +1062,10 @@ impl GitRepository {
conn: self.conn.clone(),
};
}
pub fn branches(&self) -> Vec<String> {
pub fn branches(&self) -> eyre::Result<Vec<String>> {
let mut query = self.selection.select("branches");
query.execute(&graphql_client(&self.conn)).unwrap()
query.execute(&graphql_client(&self.conn))
}
pub fn commit(&self, id: String) -> GitRef {
let mut query = self.selection.select("commit");
@ -1085,10 +1089,10 @@ impl GitRepository {
conn: self.conn.clone(),
};
}
pub fn tags(&self) -> Vec<String> {
pub fn tags(&self) -> eyre::Result<Vec<String>> {
let mut query = self.selection.select("tags");
query.execute(&graphql_client(&self.conn)).unwrap()
query.execute(&graphql_client(&self.conn))
}
}
pub struct Host {
@ -1183,10 +1187,10 @@ impl HostVariable {
conn: self.conn.clone(),
};
}
pub fn value(&self) -> String {
pub fn value(&self) -> eyre::Result<String> {
let mut query = self.selection.select("value");
query.execute(&graphql_client(&self.conn)).unwrap()
query.execute(&graphql_client(&self.conn))
}
}
pub struct Label {
@ -1196,15 +1200,15 @@ pub struct Label {
}
impl Label {
pub fn name(&self) -> String {
pub fn name(&self) -> eyre::Result<String> {
let mut query = self.selection.select("name");
query.execute(&graphql_client(&self.conn)).unwrap()
query.execute(&graphql_client(&self.conn))
}
pub fn value(&self) -> String {
pub fn value(&self) -> eyre::Result<String> {
let mut query = self.selection.select("value");
query.execute(&graphql_client(&self.conn)).unwrap()
query.execute(&graphql_client(&self.conn))
}
}
pub struct Project {
@ -1232,25 +1236,25 @@ impl Project {
conn: self.conn.clone(),
};
}
pub fn install(&self) -> bool {
pub fn install(&self) -> eyre::Result<bool> {
let mut query = self.selection.select("install");
query.execute(&graphql_client(&self.conn)).unwrap()
query.execute(&graphql_client(&self.conn))
}
pub fn name(&self) -> String {
pub fn name(&self) -> eyre::Result<String> {
let mut query = self.selection.select("name");
query.execute(&graphql_client(&self.conn)).unwrap()
query.execute(&graphql_client(&self.conn))
}
pub fn schema(&self) -> String {
pub fn schema(&self) -> eyre::Result<String> {
let mut query = self.selection.select("schema");
query.execute(&graphql_client(&self.conn)).unwrap()
query.execute(&graphql_client(&self.conn))
}
pub fn sdk(&self) -> String {
pub fn sdk(&self) -> eyre::Result<String> {
let mut query = self.selection.select("sdk");
query.execute(&graphql_client(&self.conn)).unwrap()
query.execute(&graphql_client(&self.conn))
}
}
pub struct Query {
@ -1306,10 +1310,10 @@ impl Query {
conn: self.conn.clone(),
};
}
pub fn default_platform(&self) -> Platform {
pub fn default_platform(&self) -> eyre::Result<Platform> {
let mut query = self.selection.select("defaultPlatform");
query.execute(&graphql_client(&self.conn)).unwrap()
query.execute(&graphql_client(&self.conn))
}
pub fn directory(&self, opts: Option<QueryDirectoryOpts>) -> Directory {
let mut query = self.selection.select("directory");
@ -1434,15 +1438,15 @@ pub struct Secret {
}
impl Secret {
pub fn id(&self) -> SecretId {
pub fn id(&self) -> eyre::Result<SecretId> {
let mut query = self.selection.select("id");
query.execute(&graphql_client(&self.conn)).unwrap()
query.execute(&graphql_client(&self.conn))
}
pub fn plaintext(&self) -> String {
pub fn plaintext(&self) -> eyre::Result<String> {
let mut query = self.selection.select("plaintext");
query.execute(&graphql_client(&self.conn)).unwrap()
query.execute(&graphql_client(&self.conn))
}
}
pub struct Socket {
@ -1452,10 +1456,10 @@ pub struct Socket {
}
impl Socket {
pub fn id(&self) -> SocketId {
pub fn id(&self) -> eyre::Result<SocketId> {
let mut query = self.selection.select("id");
query.execute(&graphql_client(&self.conn)).unwrap()
query.execute(&graphql_client(&self.conn))
}
}
#[derive(Serialize)]

3
crates/dagger-sdk/tests/mod.rs
View File

@ -15,7 +15,8 @@ fn test_example_container() {
redirect_stderr: None,
experimental_privileged_nesting: None,
}))
.stdout();
.stdout()
.unwrap();
assert_eq!(out, "3.16.2\n".to_string())
}