setup load path for running scripts against local repo

fix for movitto#3
alsemyonov · Apr 10, 2011 · 1d097de · 1d097de
1 parent cd6bfcc
commit 1d097de
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 2 deletions.
diff --git a/bin/rxsd_test b/bin/rxsd_test
@@ -6,7 +6,12 @@
 # Copyright (C) 2010 Mohammed Morsi <movitto@yahoo.com>
 # See COPYING for the License of this software
 
-require 'rxsd'
+begin
+  require 'rxsd'
+rescue LoadError
+  $: << File.join(File.dirname(File.expand_path(__FILE__)), '..', 'lib')
+  require 'rxsd'
+end
 
 if ARGV.size < 2
   puts "missing required arguments"

diff --git a/bin/xsd_to_ruby b/bin/xsd_to_ruby
@@ -7,7 +7,13 @@
 # See COPYING for the License of this software
 
 require 'fileutils'
-require 'rxsd'
+
+begin
+  require 'rxsd'
+rescue LoadError
+  $: << File.join(File.dirname(File.expand_path(__FILE__)), '..', 'lib')
+  require 'rxsd'
+end
 
 if ARGV.size < 2
   puts "missing required arguments"