surgicalstrike 0.0.2 → 0.0.3

data/README.md

@@ -0,0 +1,29 @@
+# SurgicalStrike
+
+TODO: Write a gem description
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+    gem 'SurgicalStrike'
+
+And then execute:
+
+    $ bundle
+
+Or install it yourself as:
+
+    $ gem install SurgicalStrike
+
+## Usage
+
+TODO: Write usage instructions here
+
+## Contributing
+
+1. Fork it
+2. Create your feature branch (`git checkout -b my-new-feature`)
+3. Commit your changes (`git commit -am 'Added some feature'`)
+4. Push to the branch (`git push origin my-new-feature`)
+5. Create new Pull Request

data/lib/SurgicalStrike.rb

@@ -0,0 +1,4 @@
+require "SurgicalStrike/version"
+require "SurgicalStrike/generic/base"
+require "SurgicalStrike/generic/surgicalstrike"
+require "SurgicalStrike/active_record/surgicalstrike"

data/lib/SurgicalStrike/active_record/surgicalstrike.rb

@@ -0,0 +1,64 @@
+require 'active_record/base'
+require 'active_record/connection_adapters/abstract_adapter'
+require 'SurgicalStrike/active_record/base'
+require 'SurgicalStrike/generic/SurgicalStrike'
+
+module ActiveRecord
+  class Base
+    include ::DatabaseCleaner::Generic::Dataholder
+    extend ActiveRecord::Callbacks
+
+    after_create :add_vip
+
+    private
+    def add_vip
+      checktarget(self)
+    end
+  end
+
+  module ConnectionAdapters
+
+    class AbstractAdapter
+      def views
+        @views ||= select_values("select table_name from information_schema.views where table_schema = '#{current_database}'") rescue []
+      end
+    end
+  end
+end
+
+module DatabaseCleaner::ActiveRecord
+  class Surgicalstrike
+    include ::DatabaseCleaner::ActiveRecord::Base
+    include ::DatabaseCleaner::Generic::Dataholder
+    include ::DatabaseCleaner::Generic::Surgicalstrike
+
+    def start
+      connection = connection_klass.connection
+
+      $strike_targets = tables_to_strike(connection)
+      $vips = Array.new
+    end
+
+    def clean
+      strike_tables
+    end
+
+
+    private
+    def strike_tables
+      $vips = $vips.reverse
+      $vips.each do |record|
+          record.delete
+      end
+    end
+
+    def tables_to_strike(connection)
+      (@only || connection.tables) - @tables_to_exclude - connection.views
+    end
+
+    # overwritten
+    def migration_storage_name
+      'schema_migrations'
+    end
+  end
+end

data/lib/SurgicalStrike/generic/base.rb

@@ -0,0 +1,11 @@
+require 'active_record'
+require 'erb'
+
+module DatabaseCleaner
+  module ActiveRecord
+
+    def self.available_strategies
+      %w[truncation transaction deletion SurgicalStrike]
+    end
+  end
+end

data/lib/SurgicalStrike/generic/surgicalstrike.rb

@@ -0,0 +1,45 @@
+module DatabaseCleaner
+  module Generic
+    module Dataholder
+      def checktarget(target)
+        if ($strike_targets.include?(target.class.table_name.upcase))
+          $vips.push(target)
+        end
+      end
+    end
+    module Surgicalstrike
+      def initialize(opts={})
+        if !opts.empty? && !(opts.keys - [:only, :except]).empty?
+          raise ArgumentError, "The only valid options are :only and :except. You specified #{opts.keys.join(',')}."
+        end
+        if opts.has_key?(:only) && opts.has_key?(:except)
+          raise ArgumentError, "You may only specify either :only or :except.  Doing both doesn't really make sense does it?"
+        end
+
+        @only = opts[:only]
+        @tables_to_exclude = (opts[:except] || []).dup
+        @tables_to_exclude << migration_storage_name if migration_storage_name
+      end
+
+      def start
+        raise NotImplementedError
+      end
+
+      def clean
+        raise NotImplementedError
+      end
+
+      private
+
+      def tables_to_strike(connection)
+        raise NotImplementedError
+      end
+
+      # overwrite in subclasses
+      # default implementation given because migration storage need not be present
+      def migration_storage_name
+        nil
+      end
+    end
+  end
+end

data/lib/SurgicalStrike/version.rb

@@ -0,0 +1,3 @@
+module SurgicalStrike
+  VERSION = "0.0.3"
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: surgicalstrike
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
   prerelease: 
 platform: ruby
 authors:
@@ -17,7 +17,13 @@ email:
 executables: []
 extensions: []
 extra_rdoc_files: []
-files: []
+files:
+- README.md
+- lib/SurgicalStrike.rb
+- lib/SurgicalStrike/active_record/surgicalstrike.rb
+- lib/SurgicalStrike/generic/base.rb
+- lib/SurgicalStrike/generic/surgicalstrike.rb
+- lib/SurgicalStrike/version.rb
 homepage: ''
 licenses: []
 post_install_message: