From 478d02895473f09852eadacab5447d9d08526bd7 Mon Sep 17 00:00:00 2001
From: IG <aloneguid@outlook.com>
Date: Mon, 30 Sep 2024 12:15:58 +0100
Subject: [PATCH] switch to enums

---
 native/api.cpp | 15 +++++++++------
 native/api.h   |  2 +-
 2 files changed, 10 insertions(+), 7 deletions(-)

diff --git a/native/api.cpp b/native/api.cpp
index 22c8d10..ebfce08 100644
--- a/native/api.cpp
+++ b/native/api.cpp
@@ -304,24 +304,27 @@ bool compress_lz4(
     return false;
 }
 
-bool iron_compress(bool compress, int32_t codec, char* input_buffer, int32_t input_buffer_size,
+bool iron_compress(bool compress, compression_codec codec, char* input_buffer, int32_t input_buffer_size,
     char* output_buffer, int32_t* output_buffer_size, compression_level compression_level) {
+
+    if(!iron_is_supported(codec)) return false;
+
     switch(codec) {
 #ifndef NO_NATIVE_SNAPPY
-        case 1:
+        case compression_codec::snappy:
             return compress_snappy(compress, input_buffer, input_buffer_size, output_buffer, output_buffer_size);
 #endif
 #ifndef NO_NATIVE_ZSTD
-        case 2:
+        case compression_codec::zstd:
             return compress_zstd(compress, input_buffer, input_buffer_size, output_buffer, output_buffer_size, compression_level);
 #endif
         //case 3:
         //    return compress_gzip(compress, input_buffer, input_buffer_size, output_buffer, output_buffer_size, compression_level);
-        case 4:
+        case compression_codec::brotli:
             return compress_brotli(compress, input_buffer, input_buffer_size, output_buffer, output_buffer_size, compression_level);
-        case 5:
+        case compression_codec::lzo:
             return compress_lzo(compress, input_buffer, input_buffer_size, output_buffer, output_buffer_size);
-        case 6:
+        case compression_codec::lz4:
             return compress_lz4(compress, input_buffer, input_buffer_size, output_buffer, output_buffer_size, compression_level);
         default:
             return false;
diff --git a/native/api.h b/native/api.h
index ae3aed0..f766da0 100644
--- a/native/api.h
+++ b/native/api.h
@@ -42,7 +42,7 @@ extern "C"
    */
    EXPORTED bool iron_compress(
       bool compress,
-      int32_t codec,
+      compression_codec codec,
       char* input_buffer,
       int32_t input_buffer_size,
       char* output_buffer,