bin/cfn-decompiler in aws-cfn-decompiler-0.2.5 vs bin/cfn-decompiler in aws-cfn-decompiler-0.5.0

@@ -1,28 +1,31 @@
 #!/usr/bin/env ruby
 
 require 'rubygems'
 
-# require 'safe_yaml'
-# SafeYAML::OPTIONS[:default_mode] = :safe
-
+path = File.dirname(__FILE__)
+if File.exists?("#{path}/../.gitignore")
+  %w(aws-cfn-dsl aws-cfn-compiler cloudformation-ruby-dsl).each do |mod|
+    add_path = File.expand_path(File.join(path, "../../#{mod}", "lib"))
+    $:.unshift(add_path)
+  end
+else
 # Borrowing from "whiches" gem ...
-cmd  = File.basename(__FILE__, '.rb')
-exes = []
-exts = ENV['PATHEXT'] ? ENV['PATHEXT'].split(';') : ['']
-ENV['PATH'].split(File::PATH_SEPARATOR).each do |path|
-  exts.each { |ext|
-    exe = File.join(path, "#{cmd}#{ext}")
-    exes << exe if File.executable? exe
-  }
-end
-path = if exes.size > 0
-         File.dirname(exes[0])
-       else
-         File.dirname(__FILE__)
-       end
+  cmd  = File.basename(__FILE__, '.rb')
+  exes = []
+  exts = ENV['PATHEXT'] ? ENV['PATHEXT'].split(';') : ['']
+  ENV['PATH'].split(File::PATH_SEPARATOR).each do |path|
+    exts.each { |ext|
+      exe = File.join(path, "#{cmd}#{ext}")
+      exes << exe if File.executable? exe
+    }
+  end
+  if exes.size > 0
+    path = File.dirname(exes[0])
+  end
 
+end
 add_path = File.expand_path(File.join(path, "..", "lib"))
 $:.unshift(add_path)
 
 require 'aws/cfn/decompiler'
-Aws::Cfn::DeCompiler::Main.new().run
+Aws::Cfn::DeCompiler::Main.new.run
\ No newline at end of file