switches 0.1.3 → 0.1.4

data/README.markdown

@@ -98,29 +98,29 @@ The switches will get applied to any role that matches <tt>/app/</tt> (so :app_m
 
 ## Throwing switches before you db:migrate ##
 
-I like to use Switches to turn off <tt>%w{memoization caching facebook campaign\_monitor delayed\_job}</tt> before running rake db:migrate, so I put this in <tt>lib/tasks/databases.rake</tt>:
+I like to use Switches to turn off <tt>%w{memoization caching facebook campaign\_monitor delayed\_job}</tt> before running rake db:migrate, so I put this in <tt>lib/tasks/zzz\_rake_switches.rake</tt>:
 
-    namespace :db do
-      # what to do before db:migrate
-      task :before_load_config do
+    namespace :rake_switches do
+      task :turn_stuff_off do
         Rake::Task['s:backup'].execute
         %w{memoization caching facebook campaign_monitor delayed_job}.each do |switch|
           Rake::Task['s:off'].execute(Rake::TaskArguments.new([:name], [switch]))
         end
       end
-      # what to do after db:migrate
-      task :after_load_config do
+      task :turn_stuff_back_on do
         Rake::Task['s:restore'].execute
         Rake::Task['cache:clear'].execute
       end
     end
 
-    # make it happen
-    Rake::Task['db:migrate'].enhance(['db:before_load_config']) do
-      Rake::Task['db:after_load_config'].invoke
+    # modify what happens on db:migrate, etc.
+    [ 'db:migrate', 'your:task:if:it:needs:wrapping' ].each do |task_to_wrap|
+      Rake::Task[task_to_wrap].enhance(['rake_switches:turn_stuff_off']) do
+        Rake::Task['rake_switches:turn_stuff_back_on'].invoke
+      end
     end
 
-Now my question is: how can I wrap that in a begin/ensure block so the switches are always restored?
+Note that 's:backup' and 's:restore' are not thread safe or really GFS safe, either.
 
 ## Usage ##
 

data/VERSION

@@ -1 +1 @@
-0.1.3
+0.1.4

data/lib/switches.rb

@@ -1,6 +1,6 @@
 require 'yaml'
-require 'rubygems'
-require 'activesupport'
+require 'fileutils'
+require 'active_support'
 
 # TODO not agnostic, expects RAILS_ROOT
 
@@ -13,6 +13,7 @@ module Switches
   CURRENT_PATH = File.join CONFIG_DIR, 'current.yml'
   DEFAULT_PATH = File.join CONFIG_DIR, 'default.yml'
   BACKUP_PATH  = File.join CONFIG_DIR, 'backup.yml'
+  TRANSACTION_PID_PATH = File.join CONFIG_DIR, 'transaction.pid'
   
   class << self
     def say(str)
@@ -20,8 +21,6 @@ module Switches
     end
     
     def setup
-      require 'fileutils'
-      
       say "Making #{CONFIG_DIR}."
       FileUtils.mkdir_p CONFIG_DIR
       
@@ -68,8 +67,12 @@ module Switches
       suffix = method_name.to_s[-1,1]
       key = method_name.to_s[0..-2]
       
-      if suffix == "?" and current.has_key?(key)
-        current[key]
+      if suffix == "?"
+        if current.has_key?(key)
+          current[key]      # set, so could be true or false
+        else
+          false             # unset, so always false
+        end
       elsif suffix == "="
         current[key] = args.first
         # TEMPORARY since we're not doing a write_current here
@@ -80,8 +83,9 @@ module Switches
     
     def default
       return @_default unless @_default.nil?
-      # say "file system read #{DEFAULT_PATH}"
-      @_default = YAML.load(File.read(DEFAULT_PATH))
+      # say "file system activity #{DEFAULT_PATH}"
+      resolve_transaction!
+      @_default = YAML.load(IO.read(DEFAULT_PATH))
       @_default.stringify_keys!
     rescue Errno::ENOENT
       say "Couldn't read defaults from #{DEFAULT_PATH}."
@@ -91,9 +95,10 @@ module Switches
     
     def current
       return @_current unless @_current.nil?
+      resolve_transaction!
       if File.exist?(CURRENT_PATH)
-        # say "file system read #{CURRENT_PATH}"
-        @_current = YAML.load(File.read(CURRENT_PATH))
+        # say "file system activity #{CURRENT_PATH}"
+        @_current = YAML.load(IO.read(CURRENT_PATH))
         @_current.stringify_keys!
       else
         @_current = default.dup
@@ -143,6 +148,8 @@ module Switches
     
     def backup
       write_current
+      start_transaction!
+      # say "file system activity #{BACKUP_PATH}"
       FileUtils.cp CURRENT_PATH, BACKUP_PATH
     end
     
@@ -152,6 +159,7 @@ module Switches
       else
         raise ArgumentError, "#{BACKUP_PATH} doesn't exist."
       end
+      end_transaction!
       @_current = nil
     end
     
@@ -159,5 +167,28 @@ module Switches
       current # load it first!
       File.open(CURRENT_PATH, 'w') { |f| f.write current.stringify_keys.to_yaml }
     end
+    
+    def transaction_pid
+      # say "file system activity #{TRANSACTION_PID_PATH}"
+      IO.readlines(TRANSACTION_PID_PATH).first.chomp.to_i if File.exists?(TRANSACTION_PID_PATH)
+    end
+    
+    def resolve_transaction!
+      if transaction_pid.present? and transaction_pid != Process.pid
+        say "Resolving... calling restore"
+        restore
+      end
+    end
+    
+    def start_transaction!
+      resolve_transaction!
+      say "Starting transaction"
+      File.open(TRANSACTION_PID_PATH, 'w') { |f| f.write Process.pid }
+    end
+    
+    def end_transaction!
+      say "Finishing transaction"
+      FileUtils.rm_f TRANSACTION_PID_PATH
+    end
   end
 end

data/switches.gemspec

@@ -5,11 +5,11 @@
 
 Gem::Specification.new do |s|
   s.name = %q{switches}
-  s.version = "0.1.3"
+  s.version = "0.1.4"
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
   s.authors = ["Seamus Abshere"]
-  s.date = %q{2009-11-05}
+  s.date = %q{2009-11-17}
   s.description = %q{
 Switches lets you turn on and off parts of your code from the commandline. There's a defaults.yml and a current.yml in the background.
 

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: switches
 version: !ruby/object:Gem::Version 
-  version: 0.1.3
+  version: 0.1.4
 platform: ruby
 authors: 
 - Seamus Abshere
@@ -9,7 +9,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2009-11-05 00:00:00 -05:00
+date: 2009-11-17 00:00:00 -05:00
 default_executable: 
 dependencies: 
 - !ruby/object:Gem::Dependency