add generic variants of CompileResult and ModuleResult

Author: Firestar99

crates/rustc_codegen_spirv-types/src/compile_result.rs

@@ -1,42 +1,46 @@
 use serde::{Deserialize, Serialize};
 use std::collections::BTreeMap;
 use std::fmt::Write;
-use std::path::{Path, PathBuf};
+use std::path::PathBuf;
+
+pub type ModuleResult = GenericModuleResult<PathBuf>;
 
 #[derive(Debug, Serialize, Deserialize)]
 #[serde(untagged)]
-pub enum ModuleResult {
-    SingleModule(PathBuf),
-    MultiModule(BTreeMap<String, PathBuf>),
+pub enum GenericModuleResult<T> {
+    SingleModule(T),
+    MultiModule(BTreeMap<String, T>),
 }
 
-impl ModuleResult {
-    pub fn unwrap_single(&self) -> &Path {
+impl<T> GenericModuleResult<T> {
+    pub fn unwrap_single(&self) -> &T {
         match self {
-            ModuleResult::SingleModule(result) => result,
-            ModuleResult::MultiModule(_) => {
+            GenericModuleResult::SingleModule(result) => result,
+            GenericModuleResult::MultiModule(_) => {
                 panic!("called `ModuleResult::unwrap_single()` on a `MultiModule` result")
             }
         }
     }
 
-    pub fn unwrap_multi(&self) -> &BTreeMap<String, PathBuf> {
+    pub fn unwrap_multi(&self) -> &BTreeMap<String, T> {
         match self {
-            ModuleResult::MultiModule(result) => result,
-            ModuleResult::SingleModule(_) => {
+            GenericModuleResult::MultiModule(result) => result,
+            GenericModuleResult::SingleModule(_) => {
                 panic!("called `ModuleResult::unwrap_multi()` on a `SingleModule` result")
             }
         }
     }
 }
 
+pub type CompileResult = GenericCompileResult<PathBuf>;
+
 #[derive(Debug, Serialize, Deserialize)]
-pub struct CompileResult {
+pub struct GenericCompileResult<T> {
     pub entry_points: Vec<String>,
-    pub module: ModuleResult,
+    pub module: GenericModuleResult<T>,
 }
 
-impl CompileResult {
+impl<T> GenericCompileResult<T> {
     pub fn codegen_entry_point_strings(&self) -> String {
         let trie = Trie::create_from(self.entry_points.iter().map(|x| x as &str));
         let mut builder = String::new();
@@ -110,9 +114,6 @@ impl<'a> Trie<'a> {
     }
 }
 
-#[allow(non_upper_case_globals)]
-pub const a: &str = "x::a";
-
 #[cfg(test)]
 mod test {
     use super::*;

crates/spirv-builder/src/lib.rs

@@ -88,8 +88,7 @@ use std::path::{Path, PathBuf};
 use std::process::{Command, Stdio};
 use thiserror::Error;
 
-pub use rustc_codegen_spirv_types::Capability;
-pub use rustc_codegen_spirv_types::{CompileResult, ModuleResult};
+pub use rustc_codegen_spirv_types::*;
 
 #[cfg(feature = "include-target-specs")]
 pub use rustc_codegen_spirv_target_specs::TARGET_SPEC_DIR_PATH;