From f192cc469671e68569a72ca74a114c39756530b4 Mon Sep 17 00:00:00 2001
From: zanciks <zackarysemancik@gmail.com>
Date: Wed, 27 Nov 2024 16:12:08 -0700
Subject: [PATCH] detect if functions are public for fragment shaders

---
 crates/spirv-std/macros/src/lib.rs | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/crates/spirv-std/macros/src/lib.rs b/crates/spirv-std/macros/src/lib.rs
index 9ecc7aef8b..202454a85b 100644
--- a/crates/spirv-std/macros/src/lib.rs
+++ b/crates/spirv-std/macros/src/lib.rs
@@ -76,7 +76,7 @@ mod image;
 use proc_macro::TokenStream;
 use proc_macro2::{Delimiter, Group, Ident, Span, TokenTree};
 
-use syn::{punctuated::Punctuated, spanned::Spanned, visit_mut::VisitMut, ItemFn, Token};
+use syn::{punctuated::Punctuated, spanned::Spanned, visit_mut::VisitMut, ItemFn, Token, Visibility};
 
 use quote::{quote, ToTokens};
 use std::fmt::Write;
@@ -149,6 +149,14 @@ pub fn spirv(attr: TokenStream, item: TokenStream) -> TokenStream {
     let attr: proc_macro2::TokenStream = attr.into();
     tokens.extend(quote! { #[cfg_attr(target_arch="spirv", rust_gpu::spirv(#attr))] });
 
+    if attr.to_string().trim() == "fragment" {
+        let item_clone = item.clone();
+        let input = syn::parse_macro_input!(item_clone as ItemFn);
+        if !matches!(input.vis, Visibility::Public(_)) {
+            panic!("The `spirv` macro can only be applied to public functions.");
+        }
+    }
+
     let item: proc_macro2::TokenStream = item.into();
     for tt in item {
         match tt {