fossil 0.2.3 → 0.2.4

data/lib/fos_schema/fos_schema_snapshot.rb

@@ -0,0 +1,47 @@
+require 'rubygems'
+require 'sequel'
+require 'logger'
+require 'fastercsv'
+require File.dirname(__FILE__) +'/../sequel/pervasive'
+
+Sequel::Model.plugin(:schema)
+DB = Sequel.connect("odbc:/#{ARGV[0]}")#, :loggers => [ Logger.new( $stdout ) ] )
+DB.extend(Sequel::Pervasive::DatabaseMethods)
+
+class XFILE < Sequel::Model( :'x$file' )
+  #~ xf$name, xf$loc, xf$flags, xf$reserved
+  set_primary_key :'xf$id'
+  one_to_many :fields, :class => :XFIELD, :key => :'xe$file'
+  one_to_many :indices, :class => :XINDEX, :key => :'xi$file'
+end
+
+class XFIELD < Sequel::Model( :'x$field' )
+  #~ xe$file xe$name xe$datatype xe$offset xe$size xe$dec xe$flags
+  set_primary_key :'xe$id'
+  many_to_one :file, :class => :XFILE, :key => :'xe$file', :primary_key => :'xf$id'
+  def <=>( other )
+    self[:'xe$name'] <=> other[:'xe$name']
+  end
+end
+
+class XINDEX < Sequel::Model( :'x$index' )
+  #~ xi$file xi$field xi$number xi$part xi$flags
+  set_primary_key [:'xi$file', :'xi$field', :'xi$part']
+  many_to_one :field, :class => :XFIELD, :key => :'xi$field', :primary_key => :'xe$id'
+  def <=>( other )
+    val = self[:'xi$number'] <=> other[:'xi$number']
+    val = self[:'xi$part'] <=> other[:'xi$part'] if val == 0
+    val
+  end
+end
+
+FasterCSV.open("FOS_SCHEMA_#{ARGV[1]}.csv", 'w') do |csv|
+  csv << 'table_name,table_flags,column_name,column_datatype,column_size,column_dec,column_flags'.split(',')
+  tables = XFILE.filter(~:'xf$flags' => 16).and(:'xf$loc'.like('%.btr')).or(:'xf$loc'.like('%.BTR')).order(:'xf$name').all
+  tables.each do |table|
+    fields = table.fields.sort
+    fields.each do |field|
+      csv << [ table[:'xf$name'],table[:'xf$flags'],field[:'xe$name'],field[:'xe$datatype'],field[:'xe$size'],field[:'xe$dec'],field[:'xe$flags'] ]
+    end
+  end
+end

data/lib/fos_schema/fos_table_gen.rb

@@ -0,0 +1,124 @@
+require 'rubygems'
+require 'sequel'
+require 'logger'
+require File.dirname(__FILE__) +'/../sequel/pervasive'
+
+Sequel::Model.plugin(:schema)
+$db = Sequel.connect('odbc:/FOS')#, :loggers => [ Logger.new( $stdout ) ] )
+$db.extend(Sequel::Pervasive::DatabaseMethods)
+
+class XFILE < Sequel::Model( :'x$file' )
+  #~ xf$name, xf$loc, xf$flags, xf$reserved
+  set_primary_key :'xf$id'
+  one_to_many :fields, :class => :XFIELD, :key => :'xe$file'
+  one_to_many :indices, :class => :XINDEX, :key => :'xi$file'
+end
+
+class XFIELD < Sequel::Model( :'x$field' )
+  #~ xe$file xe$name xe$datatype xe$offset xe$size xe$dec xe$flags
+  set_primary_key :'xe$id'
+  many_to_one :file, :class => :XFILE, :key => :'xe$file', :primary_key => :'xf$id'
+  def <=>( other )
+    self[:'xe$name'] <=> other[:'xe$name']
+  end
+end
+
+class XINDEX < Sequel::Model( :'x$index' )
+  #~ xi$file xi$field xi$number xi$part xi$flags
+  set_primary_key [:'xi$file', :'xi$field', :'xi$part']
+  many_to_one :field, :class => :XFIELD, :key => :'xi$field', :primary_key => :'xe$id'
+  def <=>( other )
+    val = self[:'xi$number'] <=> other[:'xi$number']
+    val = self[:'xi$part'] <=> other[:'xi$part'] if val == 0
+    val
+  end
+end
+
+class FOSClassGenerator
+  CLASS_TEXT = <<-STR
+class CLASS_NAME < Sequel::Model(:'TABLE_NAME')\n
+SEQUEL_INFO
+end
+  STR
+
+  SEQUEL_TEXT = <<-STR
+#### BEGIN GENERATED SECTION ####
+\tset_primary_key [:'kid - user', :'kid - mult', :'kid - comm', :'kid - date', :'kid - time']
+COLUMN_DEFINITIONS
+#### END GENERATED SECTION ####
+  STR
+
+  def initialize( file )
+    @file = file
+    @table_name = file[:'xf$name'].downcase
+    @escaped_table_name = @table_name.gsub('\'', "\\\\\\\\'")
+    @class_name = make_pretty_table_name(@table_name)
+    @file_name = make_pretty_file_name(@table_name)
+  end
+
+  def write_class_file
+    file_name = File.join("..","..","models","#{@file_name}.rb")
+    puts "filename=#{file_name}"
+    file_str = File.exists?(file_name) ? File.read(file_name) : new_class_text
+    sequel_part = build_sequel_section
+    #~ puts sequel_part if @file_name == "aircraft"
+    if( !file_str.sub!(/#### BEGIN GENERATED SECTION ####[^.|.]*#### END GENERATED SECTION ####/, sequel_part.chop) )
+      puts "CANNOT SUBSTITUTE NEW TEXT INTO OLD FILE for #{@file_name}"
+      return
+    end
+    File.open(file_name,"w"){ |file| file.write(file_str) }
+  end
+
+  private
+
+  def build_sequel_section
+    text = SEQUEL_TEXT.dup
+    column_info = @file.fields.sort.inject("") do |str, field|
+      field_name = field[:'xe$name'].downcase
+      str << "\n\tcolumn_alias :#{make_pretty_column_name(field_name)}, :'#{field_name.gsub("'", "\\\\\\\\\\\\\\\\'")}'"
+    end
+    text.sub!(/COLUMN_DEFINITIONS/,column_info)
+    text
+  end
+
+  def new_class_text
+    class_text = CLASS_TEXT.dup
+    class_text.sub!(/CLASS_NAME/,@class_name)
+    class_text.sub!(/TABLE_NAME/,@escaped_table_name)
+    class_text.sub!(/SEQUEL_INFO/,SEQUEL_TEXT.dup)
+    class_text
+  end
+
+  def make_pretty_table_name( ugly_name )
+    new_name = ugly_name.downcase.split( /[\s\/\'\-\&]/ ).collect { |v| v.capitalize }.join
+    new_name.chop! if new_name.split( '' ).last.downcase == "s"
+    new_name
+  end
+
+  def make_pretty_column_name( ugly_name )
+    better_name = ugly_name.gsub(' - ', ' ')
+    slightly_better_name = better_name.gsub('.', '')
+    even_better_name = slightly_better_name.gsub( /[\s\s]/, ' ' )
+    more_better_name = even_better_name.gsub('+', '_and_').gsub('<=', '_less_than_or_equal_to_').gsub('<', '_less_than_').gsub('>', '_more_than_').gsub('(text)', 'text').gsub('a/p', 'ap').gsub('a/c', 'ac')
+    super_good_name = more_better_name.downcase.gsub( /[\s\/\'\-\&]/, "_" )
+    excellent_name = super_good_name.gsub(/_+/,"_")
+    excellent_name
+  end
+  
+  def make_pretty_file_name( ugly_name ) 
+    perfect_name = make_pretty_column_name(ugly_name)
+    perfect_name.chop! if perfect_name[-1,1].downcase == "s"
+    perfect_name.chop! if perfect_name[-1,1].downcase == "_"
+    perfect_name = 'quote' if perfect_name == 'quotes_original'
+    perfect_name = 'quote_leg' if perfect_name == 'quote_legs_original'
+    perfect_name
+  end
+  
+end
+
+files = XFILE.filter(~:'xf$flags' => 16).and(:'xf$loc'.like('%.btr')).or(:'xf$loc'.like('%.BTR'))
+files.each do |file| 
+  p file[:'xf$name']
+  fos_class_gen = FOSClassGenerator.new(file) 
+  fos_class_gen.write_class_file
+end