arc 0.0.1

data/.gitignore

@@ -0,0 +1,9 @@
+*.gem
+.bundle
+Gemfile.lock
+pkg/*
+.DS_Store
+.rvmrc
+tmp/
+/coverage
+doc/spec.html

data/.rspec

@@ -0,0 +1 @@
+-d

data/Gemfile

@@ -0,0 +1,21 @@
+source "http://rubygems.org"
+#this needs to get added to gemspec eventually
+gem 'q', :path => '~/Projects/jacobsimeon/q'
+
+group :test do
+  gem 'arel', :path => "../arel"
+  gem 'rspec'
+  gem 'simplecov', '>= 0.4.0', :require => false
+  gem 'ruby-debug19', :require => 'ruby-debug'
+  group :sqlite do
+    gem 'sqlite3'
+  end
+  group :postgres do
+    gem 'pg'
+  end
+  group :mysql do
+    gem 'mysql2'
+  end
+end
+
+gemspec

data/Rakefile

@@ -0,0 +1,35 @@
+require 'bundler/gem_tasks'
+task :test do
+  ['sqlite', 'postgres', 'mysql'].each do |environment|
+    puts "Running tests for environment #{environment}"
+    ENV['ARC_ENV'] = environment
+    system 'rspec spec'
+  end
+end
+
+task :setup do
+  #setup postgres
+  `createdb arc_development -O jacob`
+  puts "Setup complete"
+end
+
+namespace :test do
+  task :adapter, :env do |task, args|
+    ENV['ARC_ENV'] = args.env
+    system 'rspec spec/data_stores/integration/crud_spec.rb spec/data_stores/integration/schema_spec.rb'
+  end
+  task :docs do
+    `rspec spec -f h > doc/spec.html`
+  end
+end
+
+#example use of passing arguments to rake
+# task :say, :word, :person do |task, args|
+#   puts task
+#   puts '"args.word"'
+#   puts "\t- #{args.person}"
+# end
+# $ rake say[hello,"Jacob Morris"]
+# >>  say
+# >>  "hello"
+# >>    - Jacob Morris

data/arc.gemspec

@@ -0,0 +1,20 @@
+# -*- encoding: utf-8 -*-
+$:.push File.expand_path("../lib", __FILE__)
+require "arc/version"
+
+Gem::Specification.new do |s|
+  s.name        = "arc"
+  s.version     = Arc::VERSION
+  s.authors     = ["Jacob Morris"]
+  s.email       = ["jacob.s.morris@gmail.com"]
+  s.homepage    = ""
+  s.summary     = %q{A lightweight driver for multiple sql backends}
+  s.description = %q{Compatible with mysql, sqlite, and postgres}
+
+  s.rubyforge_project = "arc"
+  
+  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"]
+end

data/lib/arc.rb

@@ -0,0 +1,13 @@
+require 'bundler/setup'
+require 'time'
+
+require 'q/resource_pool'
+require 'q/hash'
+require 'q/array'
+require 'q/string'
+require 'q/symbol'
+require 'q/dispatcher'
+
+require "arc/version"
+require 'arc/data_stores'
+

data/lib/arc/casting.rb

@@ -0,0 +1,66 @@
+module Arc
+  module Casting
+    class CannotCastValueError < StandardError; end
+        
+    TYPES = { 
+      integer:  'integer',
+      int:      'integer',
+      char:     'string' ,
+      varchar:  'string' ,
+      binary:   'binary' ,
+      date:     'date'   ,
+      datetime: 'time'   ,
+      time:     'time'   ,
+      bool:     'boolean',
+      boolean:  'boolean', 
+      bit:      'boolean',
+      float:    'float'  ,
+    }
+    
+    CAST_METHODS = Hash.new do |cast_methods, key|
+      method = "cast_#{TYPES[key.to_sym]}"
+      cast_methods[key] = method 
+    end
+      
+    def cast value, type
+      method = CAST_METHODS[type]
+      return send method, value if respond_to? method, :include_private
+      return cast_string value
+    end
+  
+    private
+    def cast_string value
+      value.to_s
+    end
+    def cast_integer value
+      value.to_i
+    end
+    def cast_float value
+      value.to_f
+    end
+    def cast_time value
+      Time.parse value
+    end
+    def cast_date value
+      Date.parse value
+    end
+    def cast_binary value
+      value.to_s.force_encoding "BINARY"
+      value.gsub!(/%00|%25|\\\\/n) do |byte|
+        case byte
+          when "%00"  then "\0"
+          when "%25"  then "%"
+          when "\\\\" then "\\"
+        end
+      end
+    end
+    def cast_boolean value
+      case value
+      when true, 1, '1', 'true', 't'
+        true
+      when false, 0, '0', 'false', 'f'
+        false
+      end
+    end
+  end
+end

data/lib/arc/data_stores.rb

@@ -0,0 +1,9 @@
+require 'arc/data_stores/abstract/store'
+
+module Arc
+  module DataStores
+    extend Q::Dispatcher
+    require_pattern "arc/data_stores/%s/store.rb"
+    constant_suffix "DataStore"
+  end
+end

data/lib/arc/data_stores/abstract/arel_compatibility.rb

@@ -0,0 +1,50 @@
+require 'yaml'
+require 'arel'
+
+module Arel
+  module Visitors
+    VISITORS['postgres'] = VISITORS['postgresql']
+  end
+end
+
+module Arc
+  module DataStores
+    module ArelCompatibility
+      # tell arel how we'll be using it today
+      def visitor
+        Arel::Visitors.for self
+      end
+      def table_exists? name
+        !self[name.to_sym].nil?
+      end
+      # hashes, imo, have no business in a sql database
+      # add ability to quote a hash here to keep arel tests passing
+      def quote_hash h
+        quote_string h.to_yaml
+      end
+      # provide a method for getting at the primary key of a given table
+      def primary_key table
+        return nil unless table
+        table = self[table]
+        pk = table.column_names.find{ |c| table[c].primary_key? }
+      end
+      # the interface for each of these methods is already implemented by the existing class
+      # just return self
+      def columns_hash
+        self
+      end
+      def connection
+        self
+      end
+      def connection_pool
+        self
+      end
+      def spec
+        self
+      end
+      def with_connection
+        yield self
+      end
+    end
+  end
+end

data/lib/arc/data_stores/abstract/object_definitions.rb

@@ -0,0 +1,55 @@
+require 'q/collector'
+module Arc
+  module DataStores
+    module ObjectDefinitions
+
+      class Schema
+        include Collector
+        alias :table_names :keys
+        def initialize store
+          @data_store = store
+        end
+        def execute_ddl ddl
+          case ddl
+          when Array
+            ddl.each { |s| @data_store.execute s }            
+          when String
+            execute_ddl ddl.split(';')
+          end        
+        end
+      end
+    
+      class Table
+        include Collector
+        alias :column_names :keys
+        attr_reader :name
+      
+        def initialize name, store=nil
+          @name, @data_store = name.to_s, store
+        end
+      end
+    
+      class Column
+        attr_reader :name
+        attr_reader :default
+        def initialize store, properties={}
+          @data_store = store
+          @name = properties.delete :name
+          @allows_null = properties.delete :allows_null
+          @default = properties.delete :default
+          @is_pk = properties.delete(:pk) || properties.delete(:primary_key)
+          @stype = properties.delete(:type)
+        end
+        def pk?; @is_pk; end
+        alias :primary_key? :pk?
+        def type
+          raise NotImplementedError
+        end
+        def allows_null?
+          @allows_null
+        end
+      end
+    
+    end
+  end
+end

data/lib/arc/data_stores/abstract/store.rb

@@ -0,0 +1,55 @@
+require 'arc/data_stores/abstract/arel_compatibility'
+require 'arc/data_stores/abstract/object_definitions'
+require 'arc/quoting'
+
+module Arc
+  module DataStores
+    class AbstractDataStore < ResourcePool
+      include ArelCompatibility
+      include Arc::Quoting
+            
+      def [] table
+        schema[table]
+      end
+
+      def schema
+        raise NotImplementedError
+      end
+      
+      def create query
+        #add new data
+        raise NotImplementedError
+      end
+
+      def read query
+        #read existing data
+        raise NotImplementedError
+      end
+
+      def update query
+        #update existing data
+        raise NotImplementedError
+      end
+
+      def destroy query
+        #destroy existing data
+        raise NotImplementedError
+      end
+          
+      def execute query
+        #adapters should override this method to execute a query against the database
+        raise NotImplementedError
+      end
+      alias :with_store :with_resource
+      alias :columns :[]
+      private
+      #better semantics for a class that deals with 'connections' instead of 'resources'
+      def create_connection
+        raise NotImplementedError
+      end
+      def create_resource
+        create_connection
+      end
+    end
+  end
+end

data/lib/arc/data_stores/mysql/object_definitions.rb

@@ -0,0 +1,54 @@
+require 'arc/data_stores/abstract/object_definitions'
+
+module Arc
+  module DataStores
+    module ObjectDefinitions
+      class MysqlSchema < Schema
+        def fetch_keys
+          @data_store.read("SHOW TABLES").map do |r|
+            r[r.keys.first].to_sym
+          end
+        end
+      
+        def fetch_item name
+          MysqlTable.new name, @data_store
+        end
+      end
+    
+      class MysqlTable < Table
+        def raw_column_data
+          @data_store.read("SHOW FULL FIELDS FROM superheros").each do |r|
+            r
+          end
+        end
+        def fetch_keys
+          raw_column_data.map {|c| c[:Field].to_sym }
+        end
+        def fetch_item name
+          return {} unless keys.include? name.to_sym
+          c = raw_column_data.find{|c| c[:Field] == name.to_s}
+          MysqlColumn.new @data_store,{
+            name: c[:Field],
+            allows_null: c[:Null] == "YES",
+            default: c[:Default],
+            primary_key: c[:Key] == "PRI",
+            type: c[:Type]
+          }
+        end
+      end
+    
+      class MysqlColumn < Column
+        TYPES = {
+          :int => :integer,
+        }
+        def type
+          @type ||= begin
+            type_key = @stype.downcase.gsub(/\([\d,]*\)/, '').to_sym
+            TYPES[type_key] || type_key
+          end
+        end
+      end
+
+    end
+  end
+end

data/lib/arc/data_stores/mysql/store.rb

@@ -0,0 +1,51 @@
+require 'mysql2'
+require 'arc/data_stores/mysql/object_definitions'
+
+Mysql2::Client.default_query_options.merge!({
+  :symbolize_keys => true
+})
+
+module Arc::DataStores
+  class MysqlDataStore < AbstractDataStore
+    
+    def read query
+      execute(query)
+    end
+    
+    def create sql
+      table = sql.match(/\AINSERT into ([^ (]*)/i)[1]
+      execute sql
+      read("select * from #{table} where id = " + last_row_id.to_s).first
+    end
+    
+    def update sql
+      execute sql
+    end
+    
+    def destroy sql
+      execute sql
+    end
+
+    def execute query
+      with_store do |store|
+        store.query query
+      end
+    end
+    
+    def schema
+      @schema ||= ObjectDefinitions::MysqlSchema.new self
+    end
+    
+    private
+    def create_connection
+      Mysql2::Client.new @config
+    end
+    
+    def last_row_id
+      with_store do |store|
+        store.last_id
+      end
+    end
+          
+  end
+end