idzebra 0.1.0

data/bin/zebraidx_record

@@ -0,0 +1,55 @@
+#!/usr/bin/env ruby
+
+require 'idzebra'
+require 'trollop'
+
+SUB_COMMANDS = %w(add delete)
+global_opts = Trollop::options do
+  banner "zebra <options> (add|remove) [file]"
+  opt :config, "Zebra config file",
+    :type => :string, :short => '-c', :required => true
+  opt :verbose, "Verbosity level", :type => :string, :short => "-v"
+  opt :commit, "Commit on completion", :default => false, :short => "-C"
+  stop_on SUB_COMMANDS
+end
+
+Trollop::die :config, "must exist" unless File.exist?(global_opts[:config])
+
+if global_opts[:verbose]
+  level = global_opts[:verbose]
+  level = level =~ /^0x/ ? level.to_i : level.to_sym
+  IdZebra::log_level = level
+end
+
+
+cmd = ARGV.shift # get the subcommand
+cmd_opts = case cmd
+  when "add" # parse delete options
+    Trollop::options do
+      # No cmd specific options
+    end
+  when "remove"  # parse copy options
+    Trollop::options do
+      # No cmd specific options
+    end
+  else
+    Trollop::die "unknown subcommand #{cmd.inspect}"
+  end
+
+file = ARGV.shift
+data = case file
+  when nil, '-'
+    STDIN.read
+  else
+    File.open(file) { |f| f.read}
+  end
+
+IdZebra::API(global_opts[:config]) do |repo|
+  case cmd
+    when "add"
+      repo.add_record(data)
+    when "remove"
+      repo.delete_record(data)
+  end
+  repo.commit if global_opts[:commit]
+end

data/lib/idzebra.rb

@@ -0,0 +1,139 @@
+require 'ffi'
+
+module IdZebra
+  
+  module Native
+    extend FFI::Library
+    
+    UPDATE_ACTIONS = enum [
+      :action_insert, 1,
+      :action_replace,
+      :action_delete,
+      :action_update,
+      :action_a_delete ]
+    
+    ffi_lib_flags :now, :global
+    ffi_lib ['yaz', 'libyaz.so.4'],
+      ['yaz_server', 'libyaz_server.so.4'],
+      ['idzebra-2.0', 'libidzebra-2.0.so.0']
+      
+    typedef :pointer, :zebra_handle
+    typedef :pointer, :zebra_service
+    typedef :short,   :zebra_res
+    
+    # Yaz functions to set logging level
+    attach_function :yaz_log_init_level, [:int], :void
+    attach_function :yaz_log_mask_str, [:string], :int
+    
+    attach_function :zebra_start, [:string], :zebra_service
+    attach_function :zebra_stop, [:zebra_service], :zebra_res
+    attach_function :zebra_open, [:zebra_service, :pointer], :zebra_handle
+    attach_function :zebra_close, [:zebra_handle], :zebra_res
+    
+    attach_function :zebra_init,    [:zebra_handle], :zebra_res
+    attach_function :zebra_clean,   [:zebra_handle], :zebra_res
+    attach_function :zebra_commit,  [:zebra_handle], :zebra_res
+    attach_function :zebra_compact, [:zebra_handle], :zebra_res
+    
+    attach_function :zebra_add_record, [
+      :zebra_handle,
+      :string,        # buf
+      :int],          # buf_size
+      :zebra_res
+    attach_function :zebra_update_record, [
+      :zebra_handle,
+      UPDATE_ACTIONS, # action
+      :string,        # recordType
+      :long,          # sysno
+      :string,        # match
+      :string,        # fname
+      :string,        # buf
+      :int],          # buf_size
+      :zebra_res
+  end
+  
+  class << self
+    
+    def API(config_file, &block)
+      extend Native
+      log_level = :error
+      zs = zebra_start(config_file)
+      zh = zebra_open(zs, nil)
+      log_level = :default
+      yield Repository.new(zh)
+      log_level = :error
+      zebra_close zh
+      zebra_stop zs
+    end
+    
+    def log_level
+      extend Native
+      mask = yaz_log_mask_str('')
+      case mask
+      when yaz_log_mask_str('none,error')
+        :error
+      when yaz_log_mask_str('none,error,warn')
+        :warn
+      when yaz_log_mask_str('log')
+        :info
+      when yaz_log_mask_str('all')
+        :default
+      else
+        mask
+      end
+    end
+    
+    def log_level=(log_level)
+      extend Native
+      case log_level
+      when Numeric
+        yaz_log_init_level(log_level)
+      when :error
+        yaz_log_init_level(yaz_log_mask_str('none,error'))
+      when :warn
+        yaz_log_init_level(yaz_log_mask_str('none,error,warn'))
+      when :info, :default
+        yaz_log_init_level(yaz_log_mask_str('log'))
+      when :debug, :all
+        yaz_log_init_level(yaz_log_mask_str('all'))
+      end
+    end
+    
+  end
+  
+  class Repository < Struct.new(:zebra_handle)
+    include Native
+    
+    def init
+      zebra_init(zebra_handle)
+    end
+    
+    def compact
+      zebra_compact(zebra_handle)
+    end
+    
+    def clean
+      zebra_clean(zebra_handle)
+    end
+    
+    def commit
+      zebra_commit(zebra_handle)
+    end
+    
+    def add_record(record_str)
+      zebra_add_record(zebra_handle, record_str, 0)
+    end
+    
+    def update_record(record_str)
+      zebra_update_record(zebra_handle, 
+        :action_update, nil, 0, nil, nil, record_str, 0)
+    end
+    
+    def delete_record(record_str)
+      zebra_update_record(zebra_handle, 
+        :action_delete, nil, 0, nil, nil, record_str, 0)
+    end
+            
+  end
+  
+end

metadata

@@ -0,0 +1,116 @@
+--- !ruby/object:Gem::Specification
+name: idzebra
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+  prerelease: 
+platform: ruby
+authors:
+- Tim Dettrick
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-09-04 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: ffi
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: trollop
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+description: Ruby bindings for the Zebra open-source Z39.50/SRU server.
+email: tim@dettrick.net
+executables:
+- zebraidx_record
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/idzebra.rb
+- bin/zebraidx_record
+homepage: https://github.com/tjdett/idzebra
+licenses:
+- Simplified BSD
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: 1.9.1
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+      segments:
+      - 0
+      hash: 3872328755922669822
+requirements:
+- idzebra-2.0
+rubyforge_project: 
+rubygems_version: 1.8.24
+signing_key: 
+specification_version: 3
+summary: Ruby bindings for IdZebra
+test_files: []