neri 0.1.7 → 0.9.3

data/lib/neri/build.rb

@@ -1,37 +1,38 @@
 #!/usr/bin/env ruby
-# encoding: UTF-8
 
 require "neri"
 
 module Neri
   @data_files = []
-  @system_files = []
-  
+
   @options = {
     quiet:   false,
     verbose: false,
-    
+
+    external_encoding: nil,
+
     dlls:     [],
     libs:     [],
     gems:     [],
     encoding: "*",
-    
+
     enable_gems:         false,
     enable_did_you_mean: false,
     chdir_first:         false,
     pause_last:          nil,
     pause_text:          nil,
-    
-    output_dir: nil,
+
+    output_dir: "./",
     system_dir: "system",
-    
-    data_file:      nil,
+
+    datafile:       nil,
     encryption_key: nil,
-    
-    no_exe:    false,
-    b2ec_path: "Bat_To_Exe_Converter.exe",
+
+    no_exe: false,
+    use_b2ec: false,
+    b2ec_path: "Bat_To_Exe_Converter",
     b2ec: {
-      icon: "#{File.expand_path(File.dirname(__FILE__) + '/../../share/default.ico')}",
+      icon: File.expand_path("#{File.dirname(__FILE__)}/../../share/default.ico"),
       invisible:        nil,
       x64:              nil,
       uac_admin:        nil,
@@ -48,47 +49,49 @@ module Neri
       specialbuild:     nil,
       comments:         nil
     },
-    
+
     use_upx:     false,
-    upx_path:    "upx.exe",
+    upx_path:    "upx",
     upx_targets: ["bin/**/*.dll"],
     upx_options: "",
-    
+
     zipfile:       nil,
-    sevenzip_path: "7z.exe",
-    
+    sevenzip_path: "7z",
+
     inno_script: nil,
-    iscc_path:   "iscc",
+    iscc_path:   "iscc"
   }
   @rubyopt = ENV["RUBYOPT"].to_s
+  @args = ""
   @encryption_key = nil
-  
+
   @use_dxruby       = false
   @use_dxruby_tiled = false
   @use_ayame        = false
 
   class << self
-    
     attr_reader :options
-    
-    def relative_path(path, basedir=rubydir, prepath = "")
+
+    def relative_path(path, basedir = rubydir, prepath = "")
       basedir.concat(File::SEPARATOR) unless basedir.end_with?(File::SEPARATOR)
-      return path.start_with?(basedir) ? path.sub(basedir, prepath) : path
+      path.start_with?(basedir) ? path.sub(basedir, prepath) : path
     end
-    
+
     def to_winpath(path)
-      return File::ALT_SEPARATOR ? path.tr(File::SEPARATOR, File::ALT_SEPARATOR) : path
+      File::ALT_SEPARATOR ? path.tr(File::SEPARATOR, File::ALT_SEPARATOR) : path
     end
-    
-    def bindir(   ); RbConfig::CONFIG["bindir"] || File.join(rubydir, "bin"); end
-    def rubydir(  ); RbConfig::TOPDIR; end
-    def rubyexe(  ); RbConfig.ruby; end
-    def batchfile(); File.join(options[:output_dir], "#{File.basename(@data_files.first, ".*")}.bat"); end
-    def datafile( ); File.join(options[:output_dir], options[:system_dir], options[:data_file]); end
-    
+
+    def bindir    ; RbConfig::CONFIG["bindir"] || File.join(rubydir, "bin"); end
+    def rubydir   ; File.join(RbConfig::TOPDIR, ""); end
+    def rubyexe   ; RbConfig.ruby; end
+    def scriptfile; @data_files.first; end
+    def basename  ; File.basename(scriptfile, ".*"); end
+    def basepath  ; File.join(options[:output_dir], basename); end
+    def datafile  ; File.join(options[:output_dir], options[:system_dir], options[:datafile]); end
+
     # --help
     def output_help
-      puts <<-EOF
+      puts <<-HELP_MESSAGE
 usage: neri [options] script.rb (other_files...) -- script_arguments
 
 options:
@@ -96,27 +99,30 @@ options:
   --version or -v
   --quiet
   --verbose
-  
+
+  --external-encoding <encoding>
+
   --dll <dll1>,<dll2>,...
   --lib <lib1>,<lib2>,...
   --gem <gem1>,<gem2>,...
-  
+
   --no-enc
   --encoding <enc1>,<enc2>,...
-  
+
   --enable-gems
   --enable-did-you-mean
   --chdir-first
   --pause-last
   --no-pause-last
   --pause-text <text>
-  
+
   --output-dir <dirname>
   --system-dir <dirname>
-  --data-file <filename>
+  --datafile <filename>
   --encryption-key <key>
-  
+
   --no-exe
+  --use-b2ec
   --b2ec-path <bat_to_exe_converter_path>
   --icon <iconfile>
   --windows or --invisible
@@ -135,28 +141,28 @@ options:
   --privatebuild <string>
   --specialbuild <string>
   --comments <string>
-  
+
   --use-upx
   --upx-path <upx path>
   --upx_targets '<glob>'  # ex) 'bin/**/*.dll'
   --upx-options <options>
-  
+
   --zipfile <filename>
   --7zip-path <7-zip path>
-  
+
   --innosetup <inno_script>
   --iscc-path <iscc path>
-  
+
   --create-recipe <recipefile>
   --recipe <recipefile>
-      EOF
+      HELP_MESSAGE
     end
-    
+
     # --version
     def output_version
       puts "Neri #{Neri::VERSION}"
     end
-    
+
     # --create-recipe
     def create_recipe(file, hash = options, pre = "Neri.options")
       hash.each_pair do |key, value|
@@ -167,15 +173,12 @@ options:
           file.puts "#{pre}[:#{key}] = #{value}"
         when NilClass
           file.puts "#{pre}[:#{key}] = nil"
-        when String
-          file.puts "#{pre}[:#{key}] = '#{value.gsub("\\", "\\\\").gsub("'", "\\'")}'"
-        when Array
+        when String, Array
           file.puts "#{pre}[:#{key}] = " + JSON.generate(value)
         end
       end
     end
-    
-    
+
     def check_options
       nputs_v "Checking Neri options."
       while arg = ARGV.shift
@@ -188,14 +191,16 @@ options:
           exit
         when "--quiet", "-q"
           options[:quiet] = true
-        when "--verbose", "-v"
+        when "--verbose"
           options[:verbose] = true
+        when "--external_encoding"
+          options[:external_encoding] = ARGV.shift
         when "--dll"
-          options[:dlls] += ARGV.shift.split(/\s*,\s*/)
+          options[:dlls] += ARGV.shift.split(",").map(&:strip)
         when "--lib"
-          options[:libs] += ARGV.shift.split(/\s*,\s*/)
+          options[:libs] += ARGV.shift.split(",").map(&:strip)
         when "--gem"
-          options[:gems] += ARGV.shift.split(/\s*,\s*/)
+          options[:gems] += ARGV.shift.split(",").map(&:strip)
         when "--no-enc"
           options[:encoding] = nil
         when "--encoding"
@@ -211,22 +216,24 @@ options:
         when "--no-pause-last"
           options[:pause_last] = false
         when "--pause-text"
-          options[:pause_text] = ARGV.shift
+          options[:pause_text] = ARGV.shift.encode("utf-8")
           options[:pause_last] = true
         when "--output-dir"
-          options[:output_dir] = ARGV.shift
+          options[:output_dir] = ARGV.shift.encode("utf-8")
         when "--system-dir"
-          options[:system_dir] = ARGV.shift
-        when "--data-file"
-          options[:data_file] = ARGV.shift
+          options[:system_dir] = ARGV.shift.encode("utf-8")
+        when "--datafile"
+          options[:datafile] = ARGV.shift.encode("utf-8")
         when "--encryption-key"
-          options[:encryption_key] = ARGV.shift
+          options[:encryption_key] = ARGV.shift.encode("utf-8")
         when "--no-exe"
           options[:no_exe] = true
+        when "--use-b2ec"
+          options[:use_b2ec] = true
         when "--b2ec-path"
-          options[:b2ec_path] = ARGV.shift
+          options[:b2ec_path] = ARGV.shift.encode("utf-8")
         when "--icon"
-          options[:b2ec][:icon] = ARGV.shift
+          options[:b2ec][:icon] = ARGV.shift.encode("utf-8")
         when "--windows", "--invisible"
           options[:b2ec][:invisible] = true
         when "--console", "--visible"
@@ -240,164 +247,171 @@ options:
         when "--productversion"
           options[:b2ec][:productversion] = ARGV.shift
         when "--productname"
-          options[:b2ec][:productname] = ARGV.shift
+          options[:b2ec][:productname] = ARGV.shift.encode("utf-8")
         when "--originalfilename"
-          options[:b2ec][:originalfilename] = ARGV.shift
+          options[:b2ec][:originalfilename] = ARGV.shift.encode("utf-8")
         when "--internalname"
-          options[:b2ec][:internalname] = ARGV.shift
+          options[:b2ec][:internalname] = ARGV.shift.encode("utf-8")
         when "--description"
-          options[:b2ec][:description] = ARGV.shift
+          options[:b2ec][:description] = ARGV.shift.encode("utf-8")
         when "--company"
-          options[:b2ec][:company] = ARGV.shift
+          options[:b2ec][:company] = ARGV.shift.encode("utf-8")
         when "--trademarks"
-          options[:b2ec][:trademarks] = ARGV.shift
+          options[:b2ec][:trademarks] = ARGV.shift.encode("utf-8")
         when "--copyright"
-          options[:b2ec][:copyright] = ARGV.shift
+          options[:b2ec][:copyright] = ARGV.shift.encode("utf-8")
         when "--privatebuild"
-          options[:b2ec][:privatebuild] = ARGV.shift
+          options[:b2ec][:privatebuild] = ARGV.shift.encode("utf-8")
         when "--specialbuild"
-          options[:b2ec][:specialbuild] = ARGV.shift
+          options[:b2ec][:specialbuild] = ARGV.shift.encode("utf-8")
         when "--comments"
-          options[:b2ec][:comments] = ARGV.shift
+          options[:b2ec][:comments] = ARGV.shift.encode("utf-8")
         when "--use-upx"
           options[:use_upx] = true
         when "--upx-path"
-          options[:upx_path] = ARGV.shift
+          options[:upx_path] = ARGV.shift.encode("utf-8")
         when "--upx-targets"
-          options[:upx_targets] += ARGV.shift.split(/\s*,\s*/)
+          options[:upx_targets] += ARGV.shift.split(",").map(&:strip)
         when "--upx-options"
           options[:upx_options] = ARGV.shift
         when "--zipfile"
-          options[:zipfile] = ARGV.shift
+          options[:zipfile] = "#{ARGV.shift.encode('utf-8').sub(/\.zip$/, '')}.zip"
         when "--7zip-path"
-          options[:sevenzip_path] = ARGV.shift
+          options[:sevenzip_path] = ARGV.shift.encode("utf-8")
         when "--innosetup"
-          options[:inno_script] = ARGV.shift
+          options[:inno_script] = ARGV.shift.encode("utf-8")
         when "--iscc-path"
-          options[:iscc_path] = ARGV.shift
+          options[:iscc_path] = ARGV.shift.encode("utf-8")
         when "--create-recipe"
           require "json"
-          filename = ARGV.shift
+          filename = ARGV.shift.encode("utf-8")
           nputs "Creating recipe_file '#{filename}'."
-          open(filename, "w:utf-8"){|file| create_recipe(file)}
+          File.open(filename, "w:utf-8") { |file| create_recipe(file) }
           exit
         when "--recipe"
-          filename = ARGV.shift
+          filename = ARGV.shift.encode("utf-8")
           nputs_v "Loading recipe_file '#{filename}'."
-          load filename
+          load File.expand_path(filename)
         when "--"
           break
         when /^(--.+)/
-          puts "** Invalid Option '#{arg}'! **"
+          error "Invalid Option '#{arg}'!"
           output_help
           exit
         else
-          @data_files.push(arg)
+          @data_files.push(arg.encode("utf-8"))
         end
       end
-      
+
       if @data_files.empty?
-        puts "** No Script File! **"
+        error "No Script File!"
         output_help
         exit
       end
-      
+
+      @args = ARGV.map { |a| %( "#{a}") }.join("")
+      @options[:external_encoding] ||= Encoding.default_external.name
+      unless options[:enable_gems] || @rubyopt.index("--disable-gems")
+        @rubyopt += " --disable-gems"
+      end
+      unless options[:enable_did_you_mean] || @rubyopt.index("--disable-did_you_mean")
+        @rubyopt += " --disable-did_you_mean"
+      end
       if @data_files.size > 1 || options[:encryption_key]
-        options[:data_file] ||= File.basename(@data_files.first, ".*") + ".dat"
+        options[:datafile] ||= "#{basename}.dat"
       end
     end
-    
-    
+
     # check dependencies
-    def rb_dependencies()
-      return $LOADED_FEATURES.uniq
+    def rb_dependencies
+      $LOADED_FEATURES.uniq
     end
-    
-    def dll_dependencies()
+
+    def dll_dependencies
       require "Win32API"
-      
+
       enumprocessmodules = Win32API.new("psapi"   , "EnumProcessModules", ["L","P","L","P"], "L")
       getmodulefilename  = Win32API.new("kernel32", "GetModuleFileNameW", ["L","P","L"], "L")
       getcurrentprocess  = Win32API.new("kernel32", "GetCurrentProcess" , [], "L")
-      
+
       bytes_needed = 4 * 32
       module_handle_buffer = nil
-      process_handle = getcurrentprocess.call()
+      process_handle = getcurrentprocess.call
       loop do
         module_handle_buffer = "\x00" * bytes_needed
         bytes_needed_buffer = [0].pack("I")
-        r = enumprocessmodules.call(process_handle, module_handle_buffer, module_handle_buffer.size, bytes_needed_buffer)
-        bytes_needed = bytes_needed_buffer.unpack("I")[0]
+        enumprocessmodules.call(process_handle, module_handle_buffer, module_handle_buffer.size, bytes_needed_buffer)
+        bytes_needed = bytes_needed_buffer.unpack1("I")
         break if bytes_needed <= module_handle_buffer.size
       end
-      
+
       handles = module_handle_buffer.unpack("I*")
       dependencies = handles.select { |handle| handle > 0 }.map do |handle|
         str = "\x00\x00" * 256
         modulefilename_length = getmodulefilename.call(handle, str, str.size)
-        modulefilename = str[0, modulefilename_length * 2].force_encoding("UTF-16LE").encode("UTF-8")
-      end
-      
-      dependencies.map!{|dep| dep.sub(/^\\\\\?\\/, "")}
-      if File::ALT_SEPARATOR
-        dependencies.map!{|dep| dep.tr(File::ALT_SEPARATOR, File::SEPARATOR)}
+        str[0, modulefilename_length * 2].force_encoding("UTF-16LE").encode("UTF-8")
       end
+
+      dependencies.map! { |dep| dep.sub(/^\\\\\?\\/, "") }
+      dependencies.map! { |dep| dep.tr(File::ALT_SEPARATOR, File::SEPARATOR) } if File::ALT_SEPARATOR
       dependencies.delete(rubyexe)
-      
-      return dependencies.uniq
+
+      dependencies.uniq
     end
-    
-    def ruby_dependencies()
+
+    def ruby_dependencies
       dependencies = Dir.glob(File.join(bindir, "**", "*.manifest"))
       dependencies.push(rubyexe)
-      return dependencies.uniq
+      dependencies.uniq
     end
-    
-    def additional_dlls_dependencies()
+
+    def additional_dlls_dependencies
       dependencies = []
       options[:dlls].each do |dll|
         dependencies += Dir.glob(File.join(bindir, "**", dll))
-        dependencies += Dir.glob(File.join(bindir, "**", dll + ".*"))
+        dependencies += Dir.glob(File.join(bindir, "**", "#{dll}.*"))
       end
-      return dependencies.uniq
+      dependencies.uniq
     end
-    
-    def additional_libs_dependencies()
+
+    def additional_libs_dependencies
       dependencies = []
       options[:libs].each do |lib|
         $LOAD_PATH.each do |path|
           dependencies += Dir.glob(File.join(path, lib))
-          dependencies += Dir.glob(File.join(path, lib + ".*"))
+          dependencies += Dir.glob(File.join(path, "#{lib}.*"))
           dependencies += Dir.glob(File.join(path, lib, "**", "*"))
         end
       end
-      return dependencies.uniq
+      dependencies.uniq
     end
-    
-    def additional_gems_dependencies()
+
+    def additional_gems_dependencies
       require "rubygems"
       dependencies = []
       rubygems_dir = File.join(Gem.dir, "gems")
       options[:gems].each do |gem|
-        gem.sub!(/\:(.+)/, "")
-        targets = $1.to_s.split("|")
+        gem.sub!(/:(.+)/, "")
+        targets = Regexp.last_match(1).to_s.split("|")
         targets.push("lib/**/*")
         gem += "-*" unless gem.match("-")
-        gemdir = Dir.glob(File.join(rubygems_dir, gem)).sort.last
+        gemdir = Dir.glob(File.join(rubygems_dir, gem)).max
         next unless gemdir
+
         targets.each do |target|
           dependencies += Dir.glob(File.join(gemdir, target))
         end
       end
-      return dependencies.uniq
+      dependencies.uniq
     end
-    
-    def encoding_dependencies()
+
+    def encoding_dependencies
       return [] unless options[:encoding]
+
       dependencies = []
       enc_dir = Dir.glob(File.join(RbConfig::CONFIG["archdir"] || RbConfig::TOPDIR, "**", "enc")).first
-      
-      options[:encoding].split(/\s*,\s*/).each do |enc|
+
+      options[:encoding].split(",").map(&:strip).each do |enc|
         case enc
         when "ja"
           %w[windows_31j.so japanese_sjis.so encdb.so].each do |enc_name|
@@ -405,25 +419,24 @@ options:
           end
         else
           dependencies += Dir.glob(File.join(enc_dir, "**", enc))
-          dependencies += Dir.glob(File.join(enc_dir, "**", enc + ".*"))
+          dependencies += Dir.glob(File.join(enc_dir, "**", "#{enc}.*"))
         end
       end
-      
-      return dependencies.uniq
+
+      dependencies.uniq
     end
-    
-    def check_dependencies()
-      nputs "Running script '#{@data_files.first}' to check dependencies."
+
+    def check_dependencies
+      nputs "Running script '#{scriptfile}' to check dependencies."
       begin
-        load @data_files.first
-      rescue SystemExit
+        load File.expand_path(scriptfile)
+      rescue SystemExit, Interrupt
       end
-      nputs "Script '#{@data_files.first}' end."
+      nputs "Script '#{scriptfile}' end."
 
       if defined? DXRuby
         require "neri/dxruby"
         @use_dxruby = true
-        options[:b2ec][:invisible] = true if options[:b2ec][:invisible] == nil
       end
       if defined? DXRuby::Tiled
         require "neri/dxruby_tiled"
@@ -434,13 +447,14 @@ options:
         @use_ayame = true
       end
 
-      if options[:b2ec][:invisible] == nil
-        options[:b2ec][:invisible] = true if File.extname(@data_files.first) == ".rbw"
+      if options[:b2ec][:invisible].nil? &&
+         (File.extname(scriptfile) == ".rbw" || @use_dxruby)
+        options[:b2ec][:invisible] = true
       end
-      if options[:pause_last] == nil
-        options[:pause_last] = true unless options[:b2ec][:invisible]
+      if options[:pause_last].nil? && !options[:b2ec][:invisible]
+        options[:pause_last] = true
       end
-      
+
       require "rbconfig"
       dependencies = []
       dependencies += rb_dependencies
@@ -452,7 +466,7 @@ options:
       dependencies += encoding_dependencies
       dependencies = select_dependencies(dependencies)
 
-      size = dependencies.map{|d| File.size(d)}.inject(&:+)
+      size = dependencies.map { |d| File.size(d) }.inject(&:+)
       nputs "#{dependencies.size} files, #{size} bytes dependencies."
       if options[:verbose]
         dependencies.each do |dependency|
@@ -460,259 +474,415 @@ options:
         end
       end
 
-      return dependencies
+      dependencies
     end
-    
+
     def select_dependencies(dependencies)
       dependencies.select! do |dependency|
-        dependency.start_with?(rubydir + File::SEPARATOR)
+        dependency.start_with?(rubydir)
       end
-      
+
       @data_files.each do |file|
         dependencies.delete(File.expand_path(file))
       end
-      
+
       unless options[:enable_gems]
         dependencies.delete_if do |dependency|
           File.basename(dependency) == "rubygems.rb" ||
-          dependency.split(File::SEPARATOR).index("rubygems")
+            dependency.split(File::SEPARATOR).index("rubygems")
         end
       end
       unless options[:enable_did_you_mean]
         dependencies.delete_if do |dependency|
           File.basename(dependency) == "did_you_mean.rb" ||
-          dependency.split(File::SEPARATOR).index("did_you_mean")
+            dependency.split(File::SEPARATOR).index("did_you_mean")
         end
       end
-      
-      return dependencies.uniq
+
+      dependencies.uniq
     end
-    
-    
+
     def copy_files(dependencies)
       nputs "Copying dependencies."
       require "fileutils"
-      options[:output_dir] ||= File.basename(@data_files.first, ".*")
-      src_dir  = File.join(rubydir, "")
+      src_dir  = rubydir
       desc_dir = File.join(options[:output_dir], options[:system_dir], "")
-      
-      @system_files = dependencies.map do |file|
+
+      system_files = dependencies.map do |file|
         [file, file.sub(src_dir, desc_dir)]
       end
       unless options[:enable_gems]
-        @system_files.each do |src, desc|
-          desc.sub!(/\/gems(\/\d+\.\d+\.\d+\/)gems\/(.+?)\-[^\/]+\/lib\//, "/vendor_ruby\\1")
+        system_files.each do |_src, desc|
+          desc.sub!(%r{/gems(/\d+\.\d+\.\d+/)gems/(.+?)-[^/]+/lib/}, "/vendor_ruby\\1")
         end
       end
-      
-      @system_files.each do |src, desc|
+
+      system_files.each do |src, desc|
         FileUtils.makedirs(File.dirname(desc))
         if File.file?(src)
           FileUtils.copy(src, desc)
           nputs_v "  #{src}\n  -> #{desc}"
         end
       end
-      FileUtils.copy(@data_files.first, desc_dir) unless options[:data_file]
+      FileUtils.copy(scriptfile, desc_dir) unless options[:datafile]
     end
-    
-    
-    def create_batch()
-      nputs "Creating batch_file '#{batchfile}'."
-      
-      enc = system(%(ruby --disable-gems -e "'#{@data_files.first}'" >NUL 2>&1)) ?
-        '' : ' -e "# coding: utf-8"'
-      
-        unless options[:enable_gems]
-        @rubyopt += " --disable-gems" unless @rubyopt.match("--disable-gems")
-      end
-      
-      ruby_code = ""
+
+    def create_datafile
+      return unless options[:datafile]
+
+      nputs "Creating datafile '#{datafile}'."
+      data_files = @data_files.select { |file| File.file? file }
+      @data_files.select { |file| File.directory? file }.each do |dir|
+        data_files += Dir.glob("#{dir}/**/*").select { |file| File.file? file }
+      end
       if options[:encryption_key]
         require "digest/sha2"
         @encryption_key = Digest::SHA2.hexdigest(options[:encryption_key])
-        ruby_code = "Neri.key='#{@encryption_key}';"
       end
-      if options[:data_file]
-        data_file = "%~dp0#{options[:system_dir]}#{File::ALT_SEPARATOR || File::SEPARATOR}#{options[:data_file]}"
-        ruby_code += "Neri.datafile='#{data_file}';"
-        ruby_code += "load '#{File.basename(@data_files.first)}'"
-      else
-        ruby_code += "load '%~dp0#{options[:system_dir]}#{File::ALT_SEPARATOR}#{File.basename(@data_files.first)}'"
+      Neri.key = @encryption_key || "0" * 64
+      File.open(datafile, "wb") do |f|
+        pos = 0
+        files_str = data_files.map { |file|
+          fullpath = File.expand_path(file)
+          filename = if fullpath.start_with?(rubydir)
+                       relative_path(fullpath, rubydir, "*neri*#{File::SEPARATOR}")
+                     else
+                       file
+                     end
+          filedata = [filename, File.size(file), pos]
+          pos += File.size(file)
+          pos += BLOCK_LENGTH - pos % BLOCK_LENGTH unless pos % BLOCK_LENGTH == 0
+          nputs_v "  - #{filename}:#{File.size(file)} bytes"
+          filedata.join("\t")
+        }.join("\n").encode(Encoding::UTF_8)
+
+        f.write(format("%#{BLOCK_LENGTH}d", files_str.bytesize))
+        f.write(xor(files_str))
+        data_files.each do |file|
+          f.write(xor(File.binread(file)))
+        end
       end
-      
-      pause_code = ""
+    end
+
+    def create_batfile
+      nputs "Creating batch_file '#{basepath}.bat'."
+
+      pause_command = ""
       if options[:pause_last]
+        pause_command += "echo.\n"
         if options[:pause_text]
-          pause_code = "echo.\necho #{options[:pause_text]}\npause > nul"
+          pause_command += "echo #{options[:pause_text]}\n" +
+                           "pause > nul"
         else
-          pause_code = "echo.\npause"
+          pause_command += "pause"
         end
       end
-      
-      r  = " -rneri"
-      r += " -rneri/dxruby"       if @use_dxruby
-      r += " -rneri/dxruby_tiled" if @use_dxruby_tiled
-      r += " -rneri/ayame"        if @use_ayame
-      
-      open(batchfile, "w:#{Encoding.default_external.name}") do |f|
-        f.puts <<-EOF
+      chdir = options[:chdir_first] ? 'cd /d "%~dp0"' : ""
+
+      File.open("#{basepath}.bat", "w:#{options[:external_encoding]}") do |f|
+        f.puts <<-BATCH
 @echo off
 setlocal
 set PATH=%~dp0#{options[:system_dir]}\\#{relative_path(bindir)};%PATH%
-#{options[:chdir_first] ? 'cd /d "%~dp0"' : ''}
+set NERI_EXECUTABLE=%~0
+#{chdir}
 if %~x0 == .exe ( shift )
-#{relative_path(rubyexe, bindir)}#{r} #{@rubyopt}#{enc} -e "#{ruby_code}" %1 %2 %3 %4 %5 %6 %7 %8 %9
-#{pause_code}
+#{ruby_command(options[:chdir_first] ? '' : '%~dp0')} %1 %2 %3 %4 %5 %6 %7 %8 %9
+#{pause_command}
 endlocal
-        EOF
+        BATCH
       end
     end
-    
-    
-    def create_datafile()
-      nputs "Creating data_file '#{datafile}'."
-      data_files = @data_files.select { |file| File.file? file }
-      @data_files.select { |file| File.directory? file }.each do |dir|
-        data_files += Dir.glob(dir + "/**/*").select { |file| File.file? file }
+
+    def create_exefile
+      unless system("gcc --version >nul 2>&1 && windres --version >nul 2>&1")
+        error "gcc or windres not found !"
+        create_batfile
+        return
       end
-      Neri.key = @encryption_key || "0" * 64
-      open(datafile, "wb") do |f|
-        pos = 0
-        files_str = data_files.map{|file|
-          filename = File.expand_path(file)
-          filename = relative_path(filename, rubydir, "*neri*" + File::SEPARATOR)
-          filename = relative_path(filename, Dir.pwd)
-          filedata = [filename, File.size(file), pos]
-          pos += File.size(file)
-          pos += BLOCK_LENGTH - pos % BLOCK_LENGTH unless pos % BLOCK_LENGTH == 0
-          nputs_v "  - #{filename}:#{File.size(file)} bytes"
-          filedata.join("\t")
-        }.join("\n").encode(Encoding::UTF_8)
-        
-        f.write(sprintf("%#{BLOCK_LENGTH}d", files_str.bytesize))
-        f.write(xor(files_str))
-        data_files.each do |file|
-          f.write(xor(File.binread(file)))
+
+      exe_file = to_winpath("#{basepath}.exe"   )
+      c_file   = to_winpath("#{basepath}_tmp.c" )
+      o_file   = to_winpath("#{basepath}_tmp.o" )
+      rc_file  = to_winpath("#{basepath}_tmp.rc")
+      system_dir = escape_cstr(to_winpath(File.join(options[:system_dir], "")))
+      nputs "Creating exe_file '#{exe_file}'."
+      File.open(c_file, "w:#{options[:external_encoding]}") do |f|
+        f.puts <<-CFILE
+#include <stdio.h>
+#include <stdlib.h>
+#include <windows.h>
+#include <unistd.h>
+
+int main(int argc, char *argv[])
+{
+    char exepath[_MAX_PATH  *  2 + 1],
+         drive  [_MAX_DRIVE      + 1],
+         dir    [_MAX_DIR   *  2 + 1],
+         fname  [_MAX_FNAME *  2 + 1],
+         ext    [_MAX_EXT   *  2 + 1],
+         paths  [_MAX_PATH  * 32 + 1],
+         runruby[_MAX_PATH  * 32 + 1];
+    PROCESS_INFORMATION pi;
+    STARTUPINFO si;
+    ZeroMemory(&si, sizeof(STARTUPINFO));
+
+    if(GetModuleFileName(NULL, exepath, MAX_PATH * 2) != 0){
+        _splitpath_s(exepath, drive, _MAX_DRIVE, dir, _MAX_DIR * 2, fname, _MAX_FNAME * 2, ext, _MAX_EXT * 2);
+    } else {
+        exit(EXIT_FAILURE);
+    }
+    snprintf(paths, sizeof(paths), "NERI_EXECUTABLE=%s", exepath);
+    putenv(paths);
+    snprintf(paths, sizeof(paths), "PATH=%s%s#{system_dir}bin;%s", drive, dir, getenv("PATH"));
+    putenv(paths);
+    #{options[:chdir_first] ? 'snprintf(paths, sizeof(paths), "%s%s", drive, dir);chdir(paths);' : ''}
+    snprintf(runruby, sizeof(runruby), "#{escape_cstr(ruby_command(options[:chdir_first] ? '' : '%s%s'))} %s %s %s %s %s %s %s %s %s",
+        #{options[:chdir_first] ? '' : 'drive, dir,'}
+        argc > 1 ? argv[1] : "",
+        argc > 2 ? argv[2] : "",
+        argc > 3 ? argv[3] : "",
+        argc > 4 ? argv[4] : "",
+        argc > 5 ? argv[5] : "",
+        argc > 6 ? argv[6] : "",
+        argc > 7 ? argv[7] : "",
+        argc > 8 ? argv[8] : "",
+        argc > 9 ? argv[9] : ""
+        );
+        CFILE
+        if options[:b2ec][:invisible]
+          f.puts %[    CreateProcess(NULL, runruby, NULL, NULL, FALSE, NORMAL_PRIORITY_CLASS | CREATE_NO_WINDOW, NULL, NULL, &si, &pi);]
+        else
+          f.puts %[    system(runruby);]
+        end
+        if options[:pause_last]
+          f.puts %[    system("echo.");]
+          if options[:pause_text]
+            f.puts %[    system("echo #{escape_cstr(options[:pause_text])}");]
+            f.puts %[    system("pause >nul");]
+          else
+            f.puts %[    system("pause");]
+          end
         end
+        f.puts "    return 0;\n}"
+      end
+
+      File.open(rc_file, "w:#{options[:external_encoding]}") do |f|
+        f.puts <<-RCFILE
+#include <winver.h>
+
+1 VERSIONINFO
+#{options[:b2ec][:fileversion   ] ? "FILEVERSION     #{escape_cstr(options[:b2ec][:fileversion   ])}" : ""}
+#{options[:b2ec][:productversion] ? "PRODUCTVERSION  #{escape_cstr(options[:b2ec][:productversion])}" : ""}
+FILETYPE        VFT_APP
+BEGIN
+    BLOCK "StringFileInfo"
+    BEGIN
+        BLOCK "000004b0"
+        BEGIN
+            #{options[:b2ec][:fileversion     ] ? 'VALUE "FileVersion",      "' + escape_cstr(options[:b2ec][:fileversion     ]) + '\0"' : ''}
+            #{options[:b2ec][:productversion  ] ? 'VALUE "ProductVersion",   "' + escape_cstr(options[:b2ec][:productversion  ]) + '\0"' : ''}
+            #{options[:b2ec][:productname     ] ? 'VALUE "ProductName",      "' + escape_cstr(options[:b2ec][:productname     ]) + '\0"' : ''}
+            #{options[:b2ec][:originalfilename] ? 'VALUE "OriginalFileName", "' + escape_cstr(options[:b2ec][:originalfilename]) + '\0"' : ''}
+            #{options[:b2ec][:internalname    ] ? 'VALUE "InternalName",     "' + escape_cstr(options[:b2ec][:internalname    ]) + '\0"' : ''}
+            #{options[:b2ec][:description     ] ? 'VALUE "FileDescription",  "' + escape_cstr(options[:b2ec][:description     ]) + '\0"' : ''}
+            #{options[:b2ec][:company         ] ? 'VALUE "CompanyName",      "' + escape_cstr(options[:b2ec][:company         ]) + '\0"' : ''}
+            #{options[:b2ec][:trademarks      ] ? 'VALUE "LegalTrademarks",  "' + escape_cstr(options[:b2ec][:trademarks      ]) + '\0"' : ''}
+            #{options[:b2ec][:copyright       ] ? 'VALUE "LegalCopyright",   "' + escape_cstr(options[:b2ec][:copyright       ]) + '\0"' : ''}
+            #{options[:b2ec][:privatebuild    ] ? 'VALUE "PrivateBuild",     "' + escape_cstr(options[:b2ec][:privatebuild    ]) + '\0"' : ''}
+            #{options[:b2ec][:specialbuild    ] ? 'VALUE "SpecialBuild",     "' + escape_cstr(options[:b2ec][:specialbuild    ]) + '\0"' : ''}
+            #{options[:b2ec][:comments        ] ? 'VALUE "Comments",         "' + escape_cstr(options[:b2ec][:comments        ]) + '\0"' : ''}
+        END
+    END
+
+    BLOCK "VarFileInfo"
+    BEGIN
+        VALUE "Translation", 0x0, 0x4b0
+    END
+END
+
+2 ICON "#{escape_cstr(options[:b2ec][:icon])}"
+        RCFILE
+      end
+      nsystem(%(windres -o "#{o_file}" "#{rc_file}"))
+      nsystem(%(gcc#{options[:b2ec][:invisible] ? ' -mwindows' : ''} -o "#{exe_file}" "#{c_file}" "#{o_file}"))
+      nsystem(%(strip "#{exe_file}"))
+      File.delete(c_file, rc_file, o_file)
+    end
+
+    def ruby_command(path)
+      system_dir = "#{path}#{File.join(options[:system_dir], '')}"
+      ruby_code = ""
+      ruby_code = "Neri.key='#{@encryption_key}';" if @encryption_key
+      if options[:datafile]
+        ruby_code += "Neri.datafile='#{system_dir}' + #{unpack_filename(options[:datafile])};"
+        ruby_code += "load #{unpack_filename(File.basename(scriptfile))}"
+      else
+        ruby_code += "load File.expand_path('#{system_dir}' + #{unpack_filename(scriptfile)})"
       end
+
+      r  = " -rneri"
+      r += " -rneri/dxruby"       if @use_dxruby
+      r += " -rneri/dxruby_tiled" if @use_dxruby_tiled
+      r += " -rneri/ayame"        if @use_ayame
+
+      ruby = to_winpath(relative_path(rubyexe, bindir))
+      %(#{ruby}#{r} #{@rubyopt} -e "# coding:utf-8" -e "#{ruby_code}" #{@args})
     end
-    
-    
-    def bat_to_exe_converter()
-      exefile = batchfile.sub(/\.bat$/, ".exe")
-      nputs "Creating exe_file '#{exefile}'."
-      File.delete(exefile) if File.exist?(exefile)
-      args = %[ /bat "#{batchfile}" /exe "#{exefile}"]
-      if options[:b2ec][:x64] == nil
-        options[:b2ec][:x64] = true if RbConfig::CONFIG["target"].to_s.index("64")
-      end
-      
-      args += options[:b2ec].map{|key, value|
+
+    def bat_to_exe_converter
+      create_batfile
+      begin
+        `#{options[:b2ec_path]} /help`
+      rescue
+        error "Bat To Exe Converter not found !"
+        return
+      end
+
+      batch_file = "#{basepath}.bat"
+      exe_file   = "#{basepath}.exe"
+      nputs "Creating exe_file '#{exe_file}' with Bat To Exe Converter."
+      File.delete(exe_file) if File.exist?(exe_file)
+      if options[:b2ec][:x64].nil? && RbConfig::CONFIG["target"].to_s.index("64")
+        options[:b2ec][:x64] = true
+      end
+      args = %( /bat "#{batch_file}" /exe "#{exe_file}")
+      args += options[:b2ec].map { |key, value|
         case value
-        when String; %[ /#{key.to_s.tr('_', '-')} "#{value}"]
-        when true;   %[ /#{key.to_s.tr('_', '-')}]
-        else;        %[]
+        when String then %( /#{key.to_s.tr('_', '-')} "#{value}")
+        when true   then %( /#{key.to_s.tr('_', '-')})
+        else;            %()
         end
       }.join("")
-      begin
-        exec = %(#{options[:b2ec_path]}#{args})
-        nputs_v exec
-        options[:quiet] ? `#{exec}` : system(exec)
-      rescue SystemCallError
-      end
-      if File.exist?(exefile)
-        File.delete(batchfile)
-      else
-        nputs "Failed to create exe_file."
-      end
+
+      error "Failed to create exe_file !" unless nsystem "#{options[:b2ec_path]}#{args}"
     end
-    
-    
-    def upx()
+
+    def upx
+      unless system("#{options[:upx_path]} --version >nul 2>&1")
+        error "UPX not found !"
+        return
+      end
+
       nputs "Compressing with UPX."
       options[:upx_targets].each do |target|
         Dir.glob(File.join(options[:output_dir], options[:system_dir], target)).each do |target_path|
-          exec = %(#{options[:upx_path]} #{options[:upx_options]} "#{target_path}")
-          nputs_v exec
-          options[:quiet] ? `#{exec}` : system(exec)
+          command = %("#{options[:upx_path]}" #{options[:upx_options]} "#{target_path}")
+          nsystem command
         end
       end
     end
-    
-    
-    def create_zipfile()
+
+    def create_zipfile
+      unless system("#{options[:sevenzip_path]} >nul 2>&1")
+        error "7-Zip not found !"
+        return
+      end
+
       nputs "Creating zip_file '#{options[:zipfile]}'."
       File.delete(options[:zipfile]) if File.exist?(options[:zipfile])
-      exec = %(#{options[:sevenzip_path]} a #{options[:zipfile]} "#{options[:output_dir]}")
-      nputs_v exec
-      options[:quiet] ? `#{exec}` : system(exec)
+      files = []
+      if options[:output_dir] == "./"
+        files.push(options[:system_dir])
+        files.push(File.exist?("#{basepath}.exe") ? "#{basepath}.exe" : "#{basepath}.bat")
+      else
+        files.push(options[:output_dir])
+      end
+      command = %("#{options[:sevenzip_path]}" a "#{options[:zipfile]}" "#{files.join('" "')}")
+      nsystem command
     end
-    
-    
-    def inno_setup()
+
+    def inno_setup
+      unless system("#{options[:iscc_path]} /? >nul 2>&1")
+        error("Inno Setup not found !")
+        return
+      end
+
       filename = options[:inno_script]
+      nputs "Creating Installer '#{filename}'."
       script = "[Setup]\n"
       if File.exist?(filename)
         script = File.read(filename, encoding: Encoding::UTF_8)
-        filename = File.basename(filename, ".*") + "_tmp" + File.extname(filename)
+        filename = "#{File.basename(filename, '.*')}_tmp#{File.extname(filename)}"
       end
-      
+
       version = options[:b2ec][:productversion] || options[:b2ec][:fileversion]
       if !script.match(/^AppName=/) && options[:b2ec][:productname]
-        script.sub!(/^(\[Setup\])(\s+)/i){ "#{$1}\nAppName=#{options[:b2ec][:productname]}#{$2}" }
+        script.sub!(/^(\[Setup\])(\s+)/i) { "#{$1}\nAppName=#{options[:b2ec][:productname]}#{$2}" }
       end
       if !script.match(/^AppVersion=/) && version
-        script.sub!(/^(\[Setup\])(\s+)/i){ "#{$1}\nAppVersion=#{version}#{$2}" }
+        script.sub!(/^(\[Setup\])(\s+)/i) { "#{$1}\nAppVersion=#{version}#{$2}" }
       end
       if !script.match(/^AppVerName=/) && options[:b2ec][:productname] && version
-        script.sub!(/^(\[Setup\])(\s+)/i){ "#{$1}\nAppVerName=#{options[:b2ec][:productname]} #{version}#{$2}" }
+        script.sub!(/^(\[Setup\])(\s+)/i) { "#{$1}\nAppVerName=#{options[:b2ec][:productname]} #{version}#{$2}" }
       end
       if !script.match(/^AppPublisher=/) && options[:b2ec][:company]
-        script.sub!(/^(\[Setup\])(\s+)/i){ "#{$1}\nAppPublisher=#{options[:b2ec][:company]}#{$2}" }
+        script.sub!(/^(\[Setup\])(\s+)/i) { "#{$1}\nAppPublisher=#{options[:b2ec][:company]}#{$2}" }
       end
       if !script.match(/^AppCopyright=/) && options[:b2ec][:copyright]
-        script.sub!(/^(\[Setup\])(\s+)/i){ "#{$1}\nAppCopyright=#{options[:b2ec][:copyright]}#{$2}" }
+        script.sub!(/^(\[Setup\])(\s+)/i) { "#{$1}\nAppCopyright=#{options[:b2ec][:copyright]}#{$2}" }
       end
-      
+
       script += "\n[Files]\n" unless script.match(/^\[Files\]/)
       dir = File.expand_path(options[:output_dir])
       files_str = ""
       Dir.glob(File.join(dir, "**", "*")).each do |file|
         next unless File.file? file
+
         dist_dir = to_winpath(File::SEPARATOR + File.dirname(relative_path(file, dir)))
         dist_dir = "" if dist_dir == "\\."
-        files_str += "\nSource: \"#{to_winpath(file)}\"; DistDir: \"{app}#{dist_dir}\""
+        files_str += "\nSource: \"#{to_winpath(file)}\"; DistDir: \"{app}#{dist_dir}"
         files_str += "; Flags: isreadme" if File.basename(file).match(/^readme/i)
       end
-      script.sub!(/^(\[Files\])(\s*)/i){ "#{$1}#{files_str}#{$2}" }
-      
+      script.sub!(/^(\[Files\])(\s*)/i) { "#{$1}#{files_str}#{$2}" }
+
       File.write(filename, script)
-      exec = %(#{options[:iscc_path]} "#{filename}")
-      nputs_v exec
-      options[:quiet] ? `#{exec}` : system(exec)
+      command = %(#{options[:iscc_path]} "#{filename}")
+      nsystem command
     end
-    
-    def run()
+
+    def run
       check_options
       dependencies = check_dependencies
       copy_files(dependencies)
-      create_batch
-      create_datafile          if options[:data_file]
-      bat_to_exe_converter unless options[:no_exe]
-      upx                      if options[:use_upx]
-      create_zipfile           if options[:zipfile]
-      inno_setup               if options[:inno_script]
+      create_datafile
+      if options[:no_exe]
+        create_batfile
+      else
+        options[:use_b2ec] ? bat_to_exe_converter : create_exefile
+      end
+      upx             if options[:use_upx]
+      create_zipfile  if options[:zipfile]
+      inno_setup      if options[:inno_script]
       nputs "Neri Finished."
     end
-    
+
     private
-    
+
     def nputs(str)
       puts "=== #{str}" unless options[:quiet]
     end
-    
+
     def nputs_v(str)
       puts str if options[:verbose]
     end
+
+    def error(str)
+      puts "\e[31m#{str}\e[0m"
+    end
+
+    def unpack_filename(filename)
+      "[#{filename.unpack('U*').map(&:to_s).join(',')}].pack('U*')"
+    end
+
+    def escape_cstr(str)
+      str.gsub("\\") { "\\\\" }.gsub('"') { '\\"' }.gsub("'") { "\\'" }
+    end
+
+    def nsystem(str)
+      nputs_v(str)
+      command = str.encode(options[:external_encoding])
+      system(command + (options[:quiet] ? " >nul 2>&1" : ""))
+    end
   end
 end