strata 0.0.1

data/Gemfile

@@ -0,0 +1,6 @@
+source "http://rubygems.org"
+
+# Specify your gem's dependencies in strata.gemspec
+gemspec
+
+gem 'rspec'

data/Gemfile.lock

@@ -0,0 +1,31 @@
+PATH
+  remote: .
+  specs:
+    strata (0.0.1)
+      activesupport
+      i18n
+
+GEM
+  remote: http://rubygems.org/
+  specs:
+    activesupport (3.2.2)
+      i18n (~> 0.6)
+      multi_json (~> 1.0)
+    diff-lcs (1.1.3)
+    i18n (0.6.0)
+    multi_json (1.1.0)
+    rspec (2.8.0)
+      rspec-core (~> 2.8.0)
+      rspec-expectations (~> 2.8.0)
+      rspec-mocks (~> 2.8.0)
+    rspec-core (2.8.0)
+    rspec-expectations (2.8.0)
+      diff-lcs (~> 1.1.2)
+    rspec-mocks (2.8.0)
+
+PLATFORMS
+  ruby
+
+DEPENDENCIES
+  rspec
+  strata!

data/README

@@ -0,0 +1 @@
+A gem for defining string-based records.

data/Rakefile

@@ -0,0 +1,7 @@
+require "bundler/gem_tasks"
+require 'rspec/core/rake_task'
+
+RSpec::Core::RakeTask.new('spec')
+
+# If you want to make this the default task
+task :default => :spec

data/lib/strata.rb

@@ -0,0 +1,9 @@
+require "strata/version"
+require "active_support/core_ext/string"
+require "yaml"
+
+module Strata
+  CONFIG_DIR = File.expand_path(File.dirname(__FILE__)) + "/config"
+end
+
+require 'strata/record_writer'

data/lib/strata/record_writer.rb

@@ -0,0 +1,199 @@
+module Strata
+  module RecordWriter
+  
+    def layout_rules
+      @layout_rules ||= self.class.class_layout_rules  
+    end
+  
+    def exposed_rules
+      layout_rules.select {|key, rule| !(rule["expose"] == false && rule.has_key?("expose")) }
+    end
+  
+    def filler_rules
+      @filler_rules ||= self.class.filler_layout_rules  
+    end
+  
+    def set_layout_variables(options = {})
+      self.class.define_attribute_accessors
+    
+      options.each do |k,v|
+        raise "#{k}: Argument is not a string" unless v.is_a? String
+        self.class.send :attr_accessor, k
+        self.send "#{k}=", v.upcase
+      end
+    
+      layout_rules.each do |k,v|
+        self.class.send(:attr_accessor, k) unless (v["expose"] && v["expose"] == false)
+        # self.send "#{k}=", v['fixed_val'] if v.has_key? ""
+      end
+    end
+  
+    def set_filler(string)
+      filler_rules.each do |key, value|
+        string[(value["offset"] - 1), value["length"]] = value["fixed_val"]
+      end
+    
+      return string
+    end
+  
+    def record_length
+      self.class.class_variable_get(:@@record_length) || 0
+    end
+  
+    def record_delimiter
+      self.class.class_variable_get(:@@record_delimiter) || ""
+    end
+    
+    def allowed_characters
+      if self.class.class_variable_defined?(:@@record_allowed_characters)
+        self.class.class_variable_get(:@@record_allowed_characters)
+      end
+    end
+          
+    def to_s
+      @string = " " * record_length + record_delimiter
+      @string = set_filler(@string)
+    
+      @string = "#{@string}"
+    
+      self.exposed_rules.each do |field_name,rule|
+        value = self.send(field_name) || ""
+
+        value = value.rjust(rule['length'], "0") if rule['data_type'] == 'N'
+        value = value.ljust(rule['length'], " ") if rule['data_type'] == 'A'
+      
+        offset = rule['offset'] - 1
+        length = rule['length']
+      
+        @string[(offset), length] = value
+      end 
+    
+      @string
+    end
+    
+    def valid_characters(string)
+      return true unless allowed_characters
+      string.each_char.all? {|c| allowed_characters.include?(c)}
+    end
+    
+    def validate_field(field_name, field_value)
+      if field_name != :user_ref
+        raise "#{field_name}: Invalid character used in #{field_value}" unless valid_characters(field_value)
+      end
+      
+      raise "#{field_name}: Argument is not a string" unless field_value.is_a? String
+    end
+  
+    def validate!(options)
+      options.each do |k,v|
+        rule = layout_rules[k.to_s]
+        
+        validate_field(k, v)
+        raise "#{k}: Input too long" if v.length > rule['length']
+        raise "#{k}: Invalid data" if rule['regex'] && ((v =~ /#{rule['regex']}/) != 0)
+        raise "#{k}: Invalid data - expected #{rule['fixed_val']}, got #{v}" if rule['fixed_val'] && (v != rule['fixed_val'])
+        raise "#{k}: Numeric value required" if (rule['data_type'] == 'N') && !(Float(v) rescue false)
+      end
+    end
+  
+    module ClassMethods
+    
+      def set_record_length(length)
+        class_variable_set(:@@record_length, length)
+      end
+    
+      def set_delimiter(delimiter)
+        class_variable_set(:@@record_delimiter, delimiter)
+      end
+      
+      def set_allowed_characters(chars)
+        class_variable_set(:@@record_allowed_characters, chars)
+      end
+      
+      # TODO: remove this method or require config path to be specified OR.. make it redundant by refactoring behind an add_rule method
+      def class_layout_rules
+        file_name = "#{Absa::H2h::CONFIG_DIR}/#{self.name.split("::")[-2].underscore}.yml"
+        record_type = self.name.split("::")[-1].underscore
+      
+        YAML.load(File.open(file_name))[record_type]
+      end
+    
+      def exposed_class_layout_rules
+        self.class_layout_rules.select {|key, rule| !(rule["expose"] == false && rule.has_key?("expose")) }
+      end
+    
+      def filler_layout_rules
+        class_layout_rules.select {|key, rule| rule.has_key?("expose") && rule["expose"] == false && rule.has_key?("fixed_val")}
+      end
+      
+      def define_attribute_accessors
+        self.class_layout_rules.each do |k,v|
+          (class << self; self; end).send :attr_accessor, k
+          self.send :attr_accessor, k
+        end
+      end
+      
+      def template_options
+        hash = {}
+
+        self.exposed_class_layout_rules.each do |field, rule|
+          value = rule.has_key?('fixed_val') ? rule['fixed_val'] : nil
+
+          if value
+            value = value.rjust(rule['length'], "0") if rule['data_type'] == 'N'
+            value = value.ljust(rule['length'], " ") if rule['data_type'] == 'A'
+          end
+
+          hash[field.to_sym] = value
+        end
+
+        hash
+      end
+      
+      def retrieve_field_value(string, field, rule)
+        offset = rule['offset'] - 1
+        length = rule['length']
+        field_type = rule['data_type']
+
+        value = string[offset, length]
+
+        unless rule['fixed_val'] || rule['no_strip']
+          value = value.rstrip if field_type == 'A'
+          value = value.to_i.to_s if field_type == 'N'
+        end
+
+        value
+      end
+      
+      def matches_definition?(string)
+        self.class_layout_rules.each do |field, rule|
+          regex = rule['regex']
+          fixed_val = rule['fixed_val']
+          value = self.retrieve_field_value(string, field, rule)
+
+          return false if fixed_val and value != fixed_val
+          return false if regex and not value =~ /#{regex}/
+        end
+
+        true
+      end
+
+      def string_to_hash(string)
+        hash = {}
+
+        self.exposed_class_layout_rules.each do |field, rule|
+          hash[field.to_sym] = self.retrieve_field_value(string, field, rule)
+        end
+
+        hash
+      end
+
+      def from_s(string)
+        options = self.string_to_hash(string)
+        record = self.new(options)
+      end
+    
+    end
+  
+  end
+end

data/lib/strata/version.rb

@@ -0,0 +1,3 @@
+module Strata
+  VERSION = "0.0.1"
+end

data/spec/lib/record_writer_spec.rb

@@ -0,0 +1,44 @@
+require 'spec_helper'
+
+describe Strata::RecordWriter do
+  
+  it "should allow the record length to be specified" do
+    
+    class String
+      include Strata::RecordWriter
+      extend Strata::RecordWriter::ClassMethods
+      
+      set_record_length 200
+    end
+    
+    string = "test1234"
+    string.record_length.should == 200
+  end
+  
+  it "should not validate characters if no character set was specified" do
+    class String
+      include Strata::RecordWriter
+      extend Strata::RecordWriter::ClassMethods
+      
+      set_record_length 200
+      
+      string = "test1234"
+      string.valid_characters("ADAWVDAWVGr#&^ @").should == true
+    end
+  end
+  
+  it "should validate that a record contains only valid characters" do
+    class String
+      include Strata::RecordWriter
+      extend Strata::RecordWriter::ClassMethods
+      
+      set_record_length 200
+      set_allowed_characters ('A'..'Z').to_a
+      
+      string = "test1234"
+      string.valid_characters("ADAWVDAWVGR").should == true
+      string.valid_characters("ADAWVDAWVGr").should == false
+    end
+  end
+  
+end

data/spec/spec_helper.rb

@@ -0,0 +1,8 @@
+require 'rubygems'
+require 'bundler/setup'
+
+require 'strata'
+
+RSpec.configure do |config|
+  # some (optional) config here
+end

data/strata.gemspec

@@ -0,0 +1,23 @@
+# -*- encoding: utf-8 -*-
+$:.push File.expand_path("../lib", __FILE__)
+require "strata/version"
+
+Gem::Specification.new do |s|
+  s.name        = "strata"
+  s.version     = Strata::VERSION
+  s.authors     = ["Douglas Anderson", "Jeffrey van Aswegen"]
+  s.email       = ["i.am.douglas.anderson@gmail.com, jeffmess@gmail.com"]
+  s.homepage    = ""
+  s.summary     = %q{A gem for manipulating string-based records.}
+  s.description = %q{A gem for manipulating string-based records.}
+
+  s.rubyforge_project = "strata"
+
+  s.files         = `git ls-files`.split("\n")
+  s.test_files    = `git ls-files -- {test,spec,features}/*`.split("\n")
+  s.executables   = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
+  s.require_paths = ["lib"]
+  
+  s.add_dependency "activesupport"
+  s.add_dependency "i18n"
+end

metadata

@@ -0,0 +1,80 @@
+--- !ruby/object:Gem::Specification
+name: strata
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- Douglas Anderson
+- Jeffrey van Aswegen
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-04-04 00:00:00.000000000Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: activesupport
+  requirement: &70149324763260 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *70149324763260
+- !ruby/object:Gem::Dependency
+  name: i18n
+  requirement: &70149324762840 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *70149324762840
+description: A gem for manipulating string-based records.
+email:
+- i.am.douglas.anderson@gmail.com, jeffmess@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- Gemfile
+- Gemfile.lock
+- README
+- Rakefile
+- lib/strata.rb
+- lib/strata/record_writer.rb
+- lib/strata/version.rb
+- spec/lib/record_writer_spec.rb
+- spec/spec_helper.rb
+- strata.gemspec
+homepage: ''
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: strata
+rubygems_version: 1.8.6
+signing_key: 
+specification_version: 3
+summary: A gem for manipulating string-based records.
+test_files:
+- spec/lib/record_writer_spec.rb
+- spec/spec_helper.rb