snappy-ruby 1.0.0 → 1.0.1

data/ext/snappy/snappy.c

@@ -1,282 +0,0 @@
-#include <ruby.h>
-#include <ruby/encoding.h>
-#include <snappy-c.h>
-#include <errno.h>
-
-static VALUE rb_mSnappy;
-static VALUE rb_eSnappyError;
-
-// Compression level validation
-#define MIN_COMPRESSION_LEVEL 1
-#define MAX_COMPRESSION_LEVEL 9
-#define DEFAULT_COMPRESSION_LEVEL 6
-
-/*
- * Compress a string using Snappy compression
- *
- * @param data [String] the data to compress
- * @param options [Hash] options hash
- * @option options [Integer] :level compression level (1-9, default: 6)
- * @return [String] compressed data
- */
-static VALUE rb_snappy_compress(int argc, VALUE *argv, VALUE self) {
-    VALUE data, options;
-    int level = DEFAULT_COMPRESSION_LEVEL;
-
-    rb_scan_args(argc, argv, "11", &data, &options);
-
-    // Validate input
-    Check_Type(data, T_STRING);
-
-    // Parse options
-    if (!NIL_P(options)) {
-        Check_Type(options, T_HASH);
-        VALUE level_val = rb_hash_aref(options, ID2SYM(rb_intern("level")));
-        if (!NIL_P(level_val)) {
-            level = NUM2INT(level_val);
-            if (level < MIN_COMPRESSION_LEVEL || level > MAX_COMPRESSION_LEVEL) {
-                rb_raise(rb_eArgError, "compression level must be between %d and %d",
-                        MIN_COMPRESSION_LEVEL, MAX_COMPRESSION_LEVEL);
-            }
-        }
-    }
-
-    const char *input = RSTRING_PTR(data);
-    size_t input_length = RSTRING_LEN(data);
-
-    // Get maximum compressed length
-    size_t max_compressed_length = snappy_max_compressed_length(input_length);
-
-    // Allocate buffer for compressed data
-    char *compressed = ALLOC_N(char, max_compressed_length);
-    size_t compressed_length = max_compressed_length;
-
-    // Perform compression
-    snappy_status status = snappy_compress(input, input_length, compressed, &compressed_length);
-
-    if (status != SNAPPY_OK) {
-        xfree(compressed);
-        rb_raise(rb_eSnappyError, "compression failed with status: %d", status);
-    }
-
-    // Create Ruby string with compressed data
-    VALUE result = rb_str_new(compressed, compressed_length);
-    xfree(compressed);
-
-    return result;
-}
-
-/*
- * Decompress Snappy compressed data
- *
- * @param data [String] the compressed data
- * @return [String] decompressed data
- */
-static VALUE rb_snappy_decompress(VALUE self, VALUE data) {
-    Check_Type(data, T_STRING);
-
-    const char *compressed = RSTRING_PTR(data);
-    size_t compressed_length = RSTRING_LEN(data);
-
-    // Get uncompressed length
-    size_t uncompressed_length;
-    snappy_status status = snappy_uncompressed_length(compressed, compressed_length, &uncompressed_length);
-
-    if (status != SNAPPY_OK) {
-        rb_raise(rb_eSnappyError, "failed to get uncompressed length, status: %d", status);
-    }
-
-    // Allocate buffer for decompressed data
-    char *uncompressed = ALLOC_N(char, uncompressed_length);
-
-    // Perform decompression
-    status = snappy_uncompress(compressed, compressed_length, uncompressed, &uncompressed_length);
-
-    if (status != SNAPPY_OK) {
-        xfree(uncompressed);
-        rb_raise(rb_eSnappyError, "decompression failed with status: %d", status);
-    }
-
-    // Create Ruby string with decompressed data
-    VALUE result = rb_str_new(uncompressed, uncompressed_length);
-    xfree(uncompressed);
-
-    // Try to set UTF-8 encoding if the data is valid UTF-8
-    // Otherwise, keep it as binary (ASCII-8BIT)
-    int utf8_index = rb_utf8_encindex();
-    rb_enc_associate_index(result, utf8_index);
-
-    // Check if the string is valid in the UTF-8 encoding
-    // If not, fall back to binary encoding
-    if (rb_enc_str_coderange(result) == ENC_CODERANGE_BROKEN) {
-        rb_enc_associate_index(result, rb_ascii8bit_encindex());
-    }
-
-    return result;
-}
-
-/*
- * Compress a file using Snappy compression
- *
- * @param input_path [String] path to input file
- * @param output_path [String] path to output file
- * @param options [Hash] options hash
- * @option options [Integer] :level compression level (1-9, default: 6)
- */
-static VALUE rb_snappy_compress_file(int argc, VALUE *argv, VALUE self) {
-    VALUE input_path, output_path, options;
-    int level = DEFAULT_COMPRESSION_LEVEL;
-
-    rb_scan_args(argc, argv, "21", &input_path, &output_path, &options);
-
-    Check_Type(input_path, T_STRING);
-    Check_Type(output_path, T_STRING);
-
-    // Parse options
-    if (!NIL_P(options)) {
-        Check_Type(options, T_HASH);
-        VALUE level_val = rb_hash_aref(options, ID2SYM(rb_intern("level")));
-        if (!NIL_P(level_val)) {
-            level = NUM2INT(level_val);
-            if (level < MIN_COMPRESSION_LEVEL || level > MAX_COMPRESSION_LEVEL) {
-                rb_raise(rb_eArgError, "compression level must be between %d and %d",
-                        MIN_COMPRESSION_LEVEL, MAX_COMPRESSION_LEVEL);
-            }
-        }
-    }
-
-    const char *input_file = StringValueCStr(input_path);
-    const char *output_file = StringValueCStr(output_path);
-
-    // Open input file
-    FILE *in = fopen(input_file, "rb");
-    if (!in) {
-        rb_sys_fail(input_file);
-    }
-
-    // Get file size
-    fseek(in, 0, SEEK_END);
-    long file_size = ftell(in);
-    fseek(in, 0, SEEK_SET);
-
-    // Read file contents
-    char *input_data = ALLOC_N(char, file_size);
-    size_t bytes_read = fread(input_data, 1, file_size, in);
-    fclose(in);
-
-    if (bytes_read != file_size) {
-        xfree(input_data);
-        rb_raise(rb_eIOError, "failed to read complete file");
-    }
-
-    // Compress data
-    size_t max_compressed_length = snappy_max_compressed_length(file_size);
-    char *compressed = ALLOC_N(char, max_compressed_length);
-    size_t compressed_length = max_compressed_length;
-
-    snappy_status status = snappy_compress(input_data, file_size, compressed, &compressed_length);
-    xfree(input_data);
-
-    if (status != SNAPPY_OK) {
-        xfree(compressed);
-        rb_raise(rb_eSnappyError, "compression failed with status: %d", status);
-    }
-
-    // Write compressed data to output file
-    FILE *out = fopen(output_file, "wb");
-    if (!out) {
-        xfree(compressed);
-        rb_sys_fail(output_file);
-    }
-
-    size_t bytes_written = fwrite(compressed, 1, compressed_length, out);
-    fclose(out);
-    xfree(compressed);
-
-    if (bytes_written != compressed_length) {
-        rb_raise(rb_eIOError, "failed to write complete compressed data");
-    }
-
-    return Qnil;
-}
-
-/*
- * Decompress a Snappy compressed file
- *
- * @param input_path [String] path to compressed input file
- * @param output_path [String] path to decompressed output file
- */
-static VALUE rb_snappy_decompress_file(VALUE self, VALUE input_path, VALUE output_path) {
-    Check_Type(input_path, T_STRING);
-    Check_Type(output_path, T_STRING);
-
-    const char *input_file = StringValueCStr(input_path);
-    const char *output_file = StringValueCStr(output_path);
-
-    // Open input file
-    FILE *in = fopen(input_file, "rb");
-    if (!in) {
-        rb_sys_fail(input_file);
-    }
-
-    // Get file size
-    fseek(in, 0, SEEK_END);
-    long file_size = ftell(in);
-    fseek(in, 0, SEEK_SET);
-
-    // Read compressed file contents
-    char *compressed_data = ALLOC_N(char, file_size);
-    size_t bytes_read = fread(compressed_data, 1, file_size, in);
-    fclose(in);
-
-    if (bytes_read != file_size) {
-        xfree(compressed_data);
-        rb_raise(rb_eIOError, "failed to read complete file");
-    }
-
-    // Get uncompressed length
-    size_t uncompressed_length;
-    snappy_status status = snappy_uncompressed_length(compressed_data, file_size, &uncompressed_length);
-
-    if (status != SNAPPY_OK) {
-        xfree(compressed_data);
-        rb_raise(rb_eSnappyError, "failed to get uncompressed length, status: %d", status);
-    }
-
-    // Decompress data
-    char *uncompressed = ALLOC_N(char, uncompressed_length);
-    status = snappy_uncompress(compressed_data, file_size, uncompressed, &uncompressed_length);
-    xfree(compressed_data);
-
-    if (status != SNAPPY_OK) {
-        xfree(uncompressed);
-        rb_raise(rb_eSnappyError, "decompression failed with status: %d", status);
-    }
-
-    // Write decompressed data to output file
-    FILE *out = fopen(output_file, "wb");
-    if (!out) {
-        xfree(uncompressed);
-        rb_sys_fail(output_file);
-    }
-
-    size_t bytes_written = fwrite(uncompressed, 1, uncompressed_length, out);
-    fclose(out);
-    xfree(uncompressed);
-
-    if (bytes_written != uncompressed_length) {
-        rb_raise(rb_eIOError, "failed to write complete decompressed data");
-    }
-
-    return Qnil;
-}
-
-void Init_snappy(void) {
-    rb_mSnappy = rb_define_module("Snappy");
-    rb_eSnappyError = rb_define_class_under(rb_mSnappy, "Error", rb_eStandardError);
-
-    rb_define_singleton_method(rb_mSnappy, "compress", rb_snappy_compress, -1);
-    rb_define_singleton_method(rb_mSnappy, "decompress", rb_snappy_decompress, 1);
-    rb_define_singleton_method(rb_mSnappy, "compress_file", rb_snappy_compress_file, -1);
-    rb_define_singleton_method(rb_mSnappy, "decompress_file", rb_snappy_decompress_file, 2);
-}