Add zstd_version

Author: SpringMT

README.md

@@ -6,8 +6,6 @@ Ruby binding for zstd(Zstandard - Fast real-time compression algorithm)
 
 See https://github.com/facebook/zstd
 
-Welcome to your new gem! In this directory, you'll find the files you need to be able to package up your Ruby library into a gem. Put your Ruby code in the file `lib/zstd-ruby`. To experiment with that code, run `bin/console` for an interactive prompt.
-
 ## Zstd version
 v1.1.2 (https://github.com/facebook/zstd/releases/tag/v1.1.2)
 
@@ -29,6 +27,22 @@ Or install it yourself as:
 
 ## Usage
 
+```
+require 'zstd-ruby'
+```
+
+### compression
+
+```
+Zstd.compress(data)
+```
+
+
+### decompression
+
+```
+Zstd.decompress(compressed_data)
+```
 
 ## Development
 

ext/zstdruby/extconf.rb

@@ -1,6 +1,6 @@
 require "mkmf"
 
-$CFLAGS = '-I. -O3 -std=c99 -pedantic'
+$CFLAGS = '-I. -O3 -std=c99'
 
 Dir.chdir File.expand_path('..', __FILE__) do
     $srcs = Dir['**/*.c']

ext/zstdruby/zstdruby.c

@@ -1,6 +1,11 @@
 #include "zstdruby.h"
 #include "./libzstd/zstd.h"
 
+static VALUE zstdVersion(VALUE self)
+{
+  unsigned version = ZSTD_versionNumber();
+  return INT2NUM(version);
+}
 
 static VALUE compress(int argc, VALUE *argv, VALUE self)
 {
@@ -27,7 +32,7 @@ static VALUE compress(int argc, VALUE *argv, VALUE self)
   char* output_data = RSTRING_PTR(output);
 
   size_t compressed_size = ZSTD_compress((void*)output_data, max_compressed_size,
-                                         (const void*)input_data, input_size, 1);
+                                         (const void*)input_data, input_size, compression_level_value);
 
   if (ZSTD_isError(compressed_size)) {
     rb_raise(rb_eRuntimeError, "%s: %s", "compress failed", ZSTD_getErrorName(compressed_size));
@@ -106,6 +111,7 @@ void
 Init_zstdruby(void)
 {
   rb_mZstd = rb_define_module("Zstd");
+  rb_define_module_function(rb_mZstd, "zstd_version", zstdVersion, 0);
   rb_define_module_function(rb_mZstd, "compress", compress, -1);
   rb_define_module_function(rb_mZstd, "decompress", decompress, 1);
 }

spec/zstd-ruby_spec.rb

@@ -6,6 +6,12 @@
     expect(Zstd::VERSION).not_to be nil
   end
 
+  describe 'zstd_version' do
+    it 'should work' do
+      expect(Zstd.zstd_version).to eq(10102)
+    end
+  end
+
   describe 'compress' do
     it 'should work' do
       compressed = Zstd.compress('abc')

zstd-ruby.gemspec

@@ -26,11 +26,9 @@ Gem::Specification.new do |spec|
   spec.files         = `git ls-files -z`.split("\x0").reject do |f|
     f.match(%r{^(test|spec|features)/})
   end
-  spec.files        += ["zstd/lib/zstd.h"]
   spec.bindir        = "exe"
   spec.executables   = spec.files.grep(%r{^exe/}) { |f| File.basename(f) }
   spec.require_paths = ["lib"]
-  #spec.extensions    = ["ext/zstdruby/extconf.rb"]
 
   spec.add_development_dependency "bundler", "~> 1.14"
   spec.add_development_dependency "rake", "~> 10.0"