Add routes to lookup resources by id

nexus/src/app/disk.rs

@@ -214,6 +214,18 @@ impl super::Nexus {
         Ok(db_disk)
     }
 
+    pub async fn disk_fetch_by_id(
+        &self,
+        opctx: &OpContext,
+        disk_id: Uuid,
+    ) -> LookupResult<db::model::Disk> {
+        let (.., db_disk) = LookupPath::new(opctx, &self.db_datastore)
+            .disk_id(disk_id)
+            .fetch()
+            .await?;
+        Ok(db_disk)
+    }
+
     /// Modifies the runtime state of the Disk as requested.  This generally
     /// means attaching or detaching the disk.
     // TODO(https://github.com/oxidecomputer/omicron/issues/811):
@@ -386,6 +398,13 @@ impl super::Nexus {
         Err(self.unimplemented_todo(opctx, unimp).await)
     }
 
+    pub async fn snapshot_fetch_by_id(&self, opctx: &OpContext, id: Uuid) -> LookupResult<db::model::Snapshot> {
+        let (.., db_snapshot) = LookupPath::new(opctx, &self.db_datastore)
+            .snapshot_id(id)
+            .fetch()
+            .await?;
+    }
+
     pub async fn project_delete_snapshot(
         self: &Arc<Self>,
         opctx: &OpContext,

nexus/src/app/image.rs

@@ -71,6 +71,14 @@ impl super::Nexus {
         Err(self.unimplemented_todo(opctx, unimp).await)
     }
 
+    pub async fn project_image_fetch_by_id(&self, opctx: &OpContext, image_id: Uuid) -> LookupResult<db::model::Image> {
+        let (.., db_image) = LookupPath::new(opctx, &self.db_datastore)
+            .image_id(image_id)
+            .fetch()
+            .await?;
+            Ok(db_image)
+    }
+
     pub async fn project_delete_image(
         self: &Arc<Self>,
         opctx: &OpContext,
@@ -296,6 +304,14 @@ impl super::Nexus {
         Ok(db_disk)
     }
 
+    pub async fn global_image_fetch_by_id(&self, opctx: &OpContext, global_image_id: Uuid) -> LookupResult<db::model::GlobalImage> {
+        let (.., db_global_image) = LookupPath::new(opctx, &self.db_datastore)
+            .global_image_id(global_image_id)
+            .fetch()
+            .await?;
+        Ok(db_global_image)
+    }
+
     pub async fn global_image_delete(
         self: &Arc<Self>,
         opctx: &OpContext,

nexus/src/app/instance.rs

@@ -178,6 +178,14 @@ impl super::Nexus {
         Ok(db_instance)
     }
 
+    pub async fn instance_fetch_by_id(&self, opctx: &OpContext, instance_id: Uuid) -> LookupResult<db::model::Instance> {
+        let (.., db_instance) = LookupPath::new(opctx, &self.db_datastore)
+            .instance_id(instance_id)
+            .fetch()
+            .await?;
+        Ok(db_instance)
+    }
+
     // This operation may only occur on stopped instances, which implies that
     // the attached disks do not have any running "upstairs" process running
     // within the sled.
@@ -760,6 +768,13 @@ impl super::Nexus {
         Ok(db_interface)
     }
 
+    pub async fn network_interface_fetch_by_id(&self, opctx: &OpContext, interface_id: Uuid) -> LookupResult<db::model::NetworkInterface> {
+        let (.., db_interface) = LookupPath::new(opctx, &self.db_datastore)
+            .network_interface_id(interface_id)
+            .await?;
+        Ok(db_interface)
+    }
+
     /// Update a network interface for the given instance.
     pub async fn network_interface_update(
         &self,

nexus/src/app/organization.rs

@@ -42,6 +42,18 @@ impl super::Nexus {
         Ok(db_organization)
     }
 
+    pub async fn organization_fetch_by_id(
+        &self,
+        opctx: &OpContext,
+        organization_id: Uuid,
+    ) -> LookupResult<db::model::Organization> {
+        let (.., db_organization) = LookupPath::new(opctx, &self.db_datastore)
+            .organization_id(organization_id)
+            .fetch()
+            .await?;
+        Ok(db_organization)
+    }
+
     pub async fn organizations_list_by_name(
         &self,
         opctx: &OpContext,

nexus/src/app/project.rs

@@ -88,6 +88,18 @@ impl super::Nexus {
         Ok(db_project)
     }
 
+    pub async fn project_fetch_by_id(
+        &self,
+        opctx: &OpContext,
+        project_id: Uuid,
+    ) -> LookupResult<db::model::Project> {
+        let (.., db_project) = LookupPath::new(opctx, &self.db_datastore)
+            .project_id(project_id)
+            .fetch()
+            .await?;
+        Ok(db_project)
+    }
+
     pub async fn projects_list_by_name(
         &self,
         opctx: &OpContext,

nexus/src/app/vpc.rs

@@ -209,6 +209,13 @@ impl super::Nexus {
         Ok(db_vpc)
     }
 
+    pub async fn vpc_fetch_by_id(&self, opctx: &OpContext, vpc_id: Uuid) -> LookupResult<db::model::Vpc> {
+        let (.., db_vpc) = LookupPath::new(opctx, &self.db_datastore)
+            .vpc_id(vpc_id)
+            .await?;
+        Ok(db_vpc)
+    }
+
     pub async fn project_update_vpc(
         &self,
         opctx: &OpContext,

nexus/src/app/vpc_router.rs

@@ -96,6 +96,14 @@ impl super::Nexus {
         Ok(db_router)
     }
 
+    pub async fn vpc_router_fetch_by_id(&self, opctx: &OpContext, vpc_router_id: Uuid) -> LookupResult<db::model::VpcRouter> {
+        let (.., db_router) = LookupPath::new(opctx, &self.db_datastore)
+            .vpc_router_id(vpc_router_id)
+            .fetch()
+            .await?;
+        Ok(db_router)
+    }
+
     pub async fn vpc_update_router(
         &self,
         opctx: &OpContext,
@@ -223,6 +231,14 @@ impl super::Nexus {
         Ok(db_route)
     }
 
+    pub async fn route_fetch_by_id(&self, opctx: &OpContext, route_id: Uuid) -> LookupResult<db::model::RouterRoute> {
+        let (.., db_route) = LookupPath::new(opctx, &self.db_datastore)
+            .router_route_id(route_id)
+            .fetch()
+            .await?;
+        Ok(db_route)
+    }
+
     #[allow(clippy::too_many_arguments)]
     pub async fn router_update_route(
         &self,

nexus/src/app/vpc_subnet.rs

@@ -218,6 +218,14 @@ impl super::Nexus {
         Ok(db_vpc)
     }
 
+    pub async fn vpc_subnet_fetch_by_id(&self, opctx: &OpContext, vpc_subnet_id: Uuid) -> LookupResult<db::model::VpcSubnet> {
+        let (.., db_vpc) = LookupPath::new(opctx, &self.db_datastore)
+            .vpc_subnet_id(vpc_subnet_id)
+            .fetch()
+            .await?;
+        Ok(db_vpc)
+    }
+
     pub async fn vpc_update_subnet(
         &self,
         opctx: &OpContext,

nexus/src/authz/api_resources.rs

@@ -482,6 +482,22 @@ authz_resource! {
     polar_snippet = InProject,
 }
 
+authz_resource! {
+    name = "Image",
+    parent = "Project",
+    primary_key = Uuid,
+    roles_allowed = false,
+    polar_snippet = InProject,
+}
+
+authz_resource! {
+    name = "Snapshot",
+    parent = "Project",
+    primary_key = Uuid,
+    roles_allowed = false,
+    polar_snippet = InProject,
+}
+
 authz_resource! {
     name = "Instance",
     parent = "Project",

nexus/src/db/lookup.rs

@@ -231,6 +231,23 @@ impl<'a> LookupPath<'a> {
         Disk { key: DiskKey::PrimaryKey(Root { lookup_root: self }, id) }
     }
 
+    /// Select a resource of type Image, identified by its id
+    pub fn image_id(self, id: Uuid) -> Image<'a> {
+        Image { key: ImageKey::PrimaryKey(Root { lookup_root: self }, id) }
+    }
+
+    /// Select a resource of type Snapshot, identified by its id
+    pub fn snapshot_id(self, id: Uuid) -> Snapshot<'a> {
+        Snapshot { key: SnapshotKey::PrimaryKey(Root { lookup_root: self }, id) }
+    }
+
+    /// Select a resource of type NetworkInterface, identified by its id
+    pub fn network_interface_id(self, id: Uuid) -> NetworkInterface<'a> {
+        NetworkInterface {
+            key: NetworkInterfaceKey::PrimaryKey(Root { lookup_root: self }, id),
+        }
+    }
+
     /// Select a resource of type Vpc, identified by its id
     pub fn vpc_id(self, id: Uuid) -> Vpc<'a> {
         Vpc { key: VpcKey::PrimaryKey(Root { lookup_root: self }, id) }
@@ -476,6 +493,24 @@ lookup_resource! {
     primary_key_columns = [ { column_name = "id", rust_type = Uuid } ]
 }
 
+lookup_resource! {
+    name = "Image",
+    ancestors = [ "Silo", "Organization", "Project" ],
+    children = [],
+    lookup_by_name = true,
+    soft_deletes = true,
+    primary_key_columns = [ { column_name = "id", rust_type = Uuid } ]
+}
+
+lookup_resource! {
+    name = "Snapshot",
+    ancestors = [ "Silo", "Organization", "Project" ],
+    children = [],
+    lookup_by_name = true,
+    soft_deletes = true,
+    primary_key_columns = [ { column_name = "id", rust_type = Uuid } ]
+}
+
 lookup_resource! {
     name = "Instance",
     ancestors = [ "Silo", "Organization", "Project" ],