Objective-C interop + a few improvements

1 files changed, 47 insertions, 0 deletions

diff --git a/src/codegen/compat.h b/src/codegen/compat.h
index 26b0df5..63a5af5 100644
--- a/src/codegen/compat.h
+++ b/src/codegen/compat.h
@@ -53,6 +53,27 @@
     "#endif\n"                                                                                     \
     "#endif\n"
 
+/* Generic selection string for C mode */
+#define ZC_C_GENERIC_STR                                                                           \
+    "#ifdef __OBJC__\n"                                                                            \
+    "#define _z_objc_map ,id: \"%s\", Class: \"%s\", SEL: \"%s\"\n"                                \
+    "#define _z_objc_arg_map(x) ,id: [(id)(x) description].UTF8String, Class: "                    \
+    "class_getName((Class)(x)), SEL: sel_getName((SEL)(x))\n"                                      \
+    "#else\n"                                                                                      \
+    "#define _z_objc_map\n"                                                                        \
+    "#define _z_objc_arg_map(x)\n"                                                                 \
+    "#endif\n"                                                                                     \
+    "\n"                                                                                           \
+    "#define _z_str(x) _Generic((x), _Bool: \"%s\", char: \"%c\", "                                \
+    "signed char: \"%c\", unsigned char: \"%u\", short: \"%d\", "                                  \
+    "unsigned short: \"%u\", int: \"%d\", unsigned int: \"%u\", "                                  \
+    "long: \"%ld\", unsigned long: \"%lu\", long long: \"%lld\", "                                 \
+    "unsigned long long: \"%llu\", float: \"%f\", double: \"%f\", "                                \
+    "char*: \"%s\", void*: \"%p\" _z_objc_map)\n"
+
+#define ZC_C_ARG_GENERIC_STR                                                                       \
+    "#define _z_arg(x) _Generic((x), _Bool: _z_bool_str(x) _z_objc_arg_map(x), default: (x))\n"
+
 #ifdef __cplusplus
 #include <type_traits>
 
@@ -126,6 +147,32 @@ inline const char *_zc_fmt(void *)
 }
 
 #define _z_str(x) _zc_fmt(x)
+
+#ifdef __OBJC__
+#include <objc/objc.h>
+#include <objc/runtime.h>
+#include <objc/message.h> // for direct calls if needed, but [x description] is fine
+
+inline const char *_zc_fmt(id x)
+{
+    return [[x description] UTF8String];
+}
+inline const char *_zc_fmt(Class x)
+{
+    return class_getName(x);
+}
+inline const char *_zc_fmt(SEL x)
+{
+    return sel_getName(x);
+}
+// BOOL is signed char usually, already handled?
+// "typedef signed char BOOL;" on standard apple headers.
+// If it maps to signed char, `_zc_fmt(signed char)` handles it ("%c").
+// We might want "YES"/"NO" for BOOL.
+// But we can't distinguish typedefs in C++ function overloads easily if underlying type is same.
+// We'll leave BOOL as %c or %d for now to avoid ambiguity errors.
+#endif
+
 #endif
 
 #endif