irpack 0.1.0 → 0.2.1

data/lib/irpack/missing.rb

@@ -0,0 +1,45 @@
+=begin
+Copyright (c) 2011 Ryuichi Sakamoto.
+
+This software is provided 'as-is', without any express or implied
+warranty. In no event will the authors be held liable for any damages
+arising from the use of this software.
+
+Permission is granted to anyone to use this software for any purpose,
+including commercial applications, and to alter it and redistribute it
+freely, subject to the following restrictions:
+
+    1. The origin of this software must not be misrepresented; you must not
+    claim that you wrote the original software. If you use this software
+    in a product, an acknowledgment in the product documentation would be
+    appreciated but is not required.
+
+    2. Altered source versions must be plainly marked as such, and must not be
+    misrepresented as being the original software.
+
+    3. This notice may not be removed or altered from any source
+    distribution.
+=end
+
+require 'fileutils'
+require 'tmpdir'
+
+if not Dir.respond_to?(:mktmpdir) or
+   not File::Stat.instance_methods.include?(:world_writable) then
+  def Dir.mktmpdir(prefix='d', tmpdir=Dir.tmpdir, &block)
+    n = 0
+    begin
+      path = File.join(tmpdir, "#{prefix}-#{Time.now.to_i}-#{$$}-#{rand(0x100000000).to_s(36)}-#{n}")
+      Dir.mkdir(path, 0700)
+    rescue Errno::EEXIST
+      n += 1
+      retry
+    end
+    begin
+      block.call(path)
+    ensure
+      FileUtils.remove_entry(path)
+    end
+  end
+end
+

data/lib/irpack/packager.rb

@@ -0,0 +1,51 @@
+=begin
+Copyright (c) 2011 Ryuichi Sakamoto.
+
+This software is provided 'as-is', without any express or implied
+warranty. In no event will the authors be held liable for any damages
+arising from the use of this software.
+
+Permission is granted to anyone to use this software for any purpose,
+including commercial applications, and to alter it and redistribute it
+freely, subject to the following restrictions:
+
+    1. The origin of this software must not be misrepresented; you must not
+    claim that you wrote the original software. If you use this software
+    in a product, an acknowledgment in the product documentation would be
+    appreciated but is not required.
+
+    2. Altered source versions must be plainly marked as such, and must not be
+    misrepresented as being the original software.
+
+    3. This notice may not be removed or altered from any source
+    distribution.
+=end
+
+require 'WindowsBase'
+
+module IRPack
+  module Packager
+    include System
+    include System::IO::Packaging
+
+    RelType = 'http://schemas.openxmlformats.org/package/2006/relationships/meta data/core-properties'
+    module_function
+    def pack(files, package_file, compress=false)
+      compress_option = compress ? CompressionOption.normal : CompressionOption.not_compressed
+      package = Package.open(package_file, System::IO::FileMode.create)
+      files.each do |src, dest|
+        uri = PackUriHelper.create_part_uri(Uri.new(dest, UriKind.relative))
+        part = package.create_part(uri, 'application/octet-stream', compress_option)
+        stream = part.get_stream
+        File.open(src, 'rb') do |f|
+          data = f.read
+          stream.write(data, 0, data.size)
+        end
+        stream.close
+        package.create_relationship(uri, TargetMode.internal, RelType)
+      end
+      package.close
+    end
+  end
+end
+

data/test/test_application.rb

@@ -0,0 +1,98 @@
+=begin
+Copyright (c) 2011 Ryuichi Sakamoto.
+
+This software is provided 'as-is', without any express or implied
+warranty. In no event will the authors be held liable for any damages
+arising from the use of this software.
+
+Permission is granted to anyone to use this software for any purpose,
+including commercial applications, and to alter it and redistribute it
+freely, subject to the following restrictions:
+
+    1. The origin of this software must not be misrepresented; you must not
+    claim that you wrote the original software. If you use this software
+    in a product, an acknowledgment in the product documentation would be
+    appreciated but is not required.
+
+    2. Altered source versions must be plainly marked as such, and must not be
+    misrepresented as being the original software.
+
+    3. This notice may not be removed or altered from any source
+    distribution.
+=end
+
+require 'test/unit'
+require 'irpack/application'
+require 'utils'
+require 'stringio'
+
+class TC_ApplicationArguments < Test::Unit::TestCase
+  include Utils
+  def test_parse_without_args
+    argv = []
+    $stderr = StringIO.new
+    assert_nil(IRPack::Application::Arguments.parse!(argv))
+    assert_not_equal('', $stderr.to_s)
+  end
+
+  def test_parse
+    argv = ['entry.rb']
+    args = IRPack::Application::Arguments.parse!(argv)
+    assert(!args.compress)
+    assert(args.embed_references)
+    assert_equal(:exe, args.target)
+    assert_equal(File.expand_path('entry.exe'), File.expand_path(args.output_file))
+    assert_equal('entry.rb', args.entry_file)
+    assert_equal(1, args.files.size)
+    assert_equal('entry.rb', args.files[File.expand_path('entry.rb')])
+  end
+
+  def test_parse_output_file
+    argv = ['-o', 'foo.exe', 'entry.rb']
+    args = IRPack::Application::Arguments.parse!(argv)
+    assert_equal(File.expand_path('foo.exe'), File.expand_path(args.output_file))
+  end
+
+  def test_parse_window_app
+    argv = ['--window', 'entry.rb']
+    args = IRPack::Application::Arguments.parse!(argv)
+    assert_equal(:winexe, args.target)
+  end
+
+  def test_parse_console_app
+    argv = ['--console', 'entry.rb']
+    args = IRPack::Application::Arguments.parse!(argv)
+    assert_equal(:exe, args.target)
+  end
+
+  def test_parse_no_embed_references
+    argv = ['--no-embed', 'entry.rb']
+    args = IRPack::Application::Arguments.parse!(argv)
+    assert(!args.embed_references)
+  end
+
+  def test_parse_compress
+    argv = ['--compress', 'entry.rb']
+    args = IRPack::Application::Arguments.parse!(argv)
+    assert(args.compress)
+  end
+
+  def test_parse_basedir
+    argv = [
+      '-b', 'foo/bar',
+      'entry.rb',
+      'foo/bar/baz.rb',
+      File.expand_path('foo/bar/hoge.rb'),
+      'foo/bar/hoge/fuga.rb',
+    ]
+    args = IRPack::Application::Arguments.parse!(argv)
+    assert_equal(File.expand_path('entry.exe'), File.expand_path(args.output_file))
+    assert_equal('../../entry.rb', args.entry_file)
+    assert_equal(4, args.files.size)
+    assert_equal('../../entry.rb', args.files[File.expand_path('entry.rb')])
+    assert_equal('baz.rb', args.files[File.expand_path('foo/bar/baz.rb')])
+    assert_equal('hoge.rb', args.files[File.expand_path('foo/bar/hoge.rb')])
+    assert_equal('hoge/fuga.rb', args.files[File.expand_path('foo/bar/hoge/fuga.rb')])
+  end
+end
+

data/test/test_bootloader.rb

@@ -0,0 +1,72 @@
+=begin
+Copyright (c) 2011 Ryuichi Sakamoto.
+
+This software is provided 'as-is', without any express or implied
+warranty. In no event will the authors be held liable for any damages
+arising from the use of this software.
+
+Permission is granted to anyone to use this software for any purpose,
+including commercial applications, and to alter it and redistribute it
+freely, subject to the following restrictions:
+
+    1. The origin of this software must not be misrepresented; you must not
+    claim that you wrote the original software. If you use this software
+    in a product, an acknowledgment in the product documentation would be
+    appreciated but is not required.
+
+    2. Altered source versions must be plainly marked as such, and must not be
+    misrepresented as being the original software.
+
+    3. This notice may not be removed or altered from any source
+    distribution.
+=end
+
+require 'test/unit'
+require 'irpack/bootloader'
+require 'utils'
+require 'erb'
+
+class TC_IRPack_BootLoader < Test::Unit::TestCase
+  include Utils
+  ENTRYPOINT_SRC = <<-CS
+    using System;
+    using System.IO;
+    using System.IO.Packaging;
+
+    namespace <%= module_name %> {
+      public class EntryPoint
+      {
+        public static int Main(Package package, string[] args)
+        {
+          Console.WriteLine("Hello World!");
+          return 0;
+        }
+      }
+    }
+  CS
+
+  def compile_entrypoint(output_file, module_name)
+    sysasm = IRPack::CSCompiler.system_assemblies.collect {|asm|
+      IRPack::CSCompiler.assembly_location(asm)
+    }
+    IRPack::CSCompiler.compile(:dll, output_file, ERB.new(ENTRYPOINT_SRC).result(binding), sysasm, [])
+    output_file
+  end
+
+  def test_compile
+    module_name = 'TestModule'
+    entrypoint = tempfilename
+    references = ironruby_assemblies
+    package_file = tempfilename
+
+    compile_entrypoint(entrypoint, module_name)
+    package = create_package(package_file, 'TestModule.EntryPoint.dll' => File.open(entrypoint, 'rb') {|f| f.read })
+
+    exe = tempfilename('.exe')
+    assert_equal(exe, IRPack::BootLoader.compile(:exe, exe, module_name, references, package_file))
+    res = `#{exe}`.chomp
+    assert_equal('Hello World!', res)
+    package.close
+  end
+end
+

data/test/test_cscompiler.rb

@@ -1,17 +1,56 @@
+=begin
+Copyright (c) 2011 Ryuichi Sakamoto.
 
-$:.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
-require 'test/unit'
-require 'tempfile'
-require 'tmpdir'
-require 'fileutils'
-require 'irpack'
+This software is provided 'as-is', without any express or implied
+warranty. In no event will the authors be held liable for any damages
+arising from the use of this software.
+
+Permission is granted to anyone to use this software for any purpose,
+including commercial applications, and to alter it and redistribute it
+freely, subject to the following restrictions:
+
+    1. The origin of this software must not be misrepresented; you must not
+    claim that you wrote the original software. If you use this software
+    in a product, an acknowledgment in the product documentation would be
+    appreciated but is not required.
 
-require 'WindowsBase'
-include System
-include System::IO
-include System::IO::Packaging
+    2. Altered source versions must be plainly marked as such, and must not be
+    misrepresented as being the original software.
+
+    3. This notice may not be removed or altered from any source
+    distribution.
+=end
+
+require 'test/unit'
+require 'irpack/cscompiler'
+require 'utils'
 
 class TC_CSCompiler < Test::Unit::TestCase
+  include Utils
+  def test_system_assemblies
+    assert_equal(IRPack::CSCompiler.system_assemblies(4), [
+      'System, Version=4.0.0.0, Culture=neutral, PublicKeyToken=b77a5c561934e089',
+      'WindowsBase, Version=4.0.0.0, Culture=neutral, PublicKeyToken=31bf3856ad364e35',
+      'System.Core, Version=4.0.0.0, Culture=neutral, PublicKeyToken=b77a5c561934e089',
+    ])
+    [2,3.0,3.5].each do |v|
+      assert_equal(IRPack::CSCompiler.system_assemblies(v), [
+        'System, Version=2.0.0.0, Culture=neutral, PublicKeyToken=b77a5c561934e089',
+        'WindowsBase, Version=3.0.0.0, Culture=neutral, PublicKeyToken=31bf3856ad364e35',
+        'System.Core, Version=3.5.0.0, Culture=neutral, PublicKeyToken=b77a5c561934e089',
+      ])
+      assert_raise(ArgumentError) do
+        IRPack::CSCompiler.system_assemblies(5)
+      end
+    end
+  end
+
+  def test_assembly_location
+    path = IRPack::CSCompiler.assembly_location(
+        'System, Version=2.0.0.0, Culture=neutral, PublicKeyToken=b77a5c561934e089')
+    assert_equal(File.basename(path), 'System.dll')
+  end
+
   def test_compile_dll
     src = <<-CS
     using System;
@@ -26,19 +65,17 @@ class TC_CSCompiler < Test::Unit::TestCase
     CS
     references = []
     resources = []
-    output_file = Tempfile.new(File.basename(__FILE__))
-    output_file.close
+    output_file = tempfilename('.dll')
     result = IRPack::CSCompiler.compile(
       :dll,
+      output_file,
       src,
       references,
-      resources,
-      output_file.path
-    )
-    assert_equal(output_file.path, result)
+      resources)
+    assert_equal(output_file, result)
     asm = nil
     assert_nothing_raised do
-      asm = System::Reflection::Assembly.load_from(output_file.path)
+      asm = System::Reflection::Assembly.load_from(output_file)
     end
     assert(asm.get_type('hoge.Hoge'))
     assert_nil(asm.entry_point)

data/test/test_entrypoint.rb

@@ -0,0 +1,99 @@
+=begin
+Copyright (c) 2011 Ryuichi Sakamoto.
+
+This software is provided 'as-is', without any express or implied
+warranty. In no event will the authors be held liable for any damages
+arising from the use of this software.
+
+Permission is granted to anyone to use this software for any purpose,
+including commercial applications, and to alter it and redistribute it
+freely, subject to the following restrictions:
+
+    1. The origin of this software must not be misrepresented; you must not
+    claim that you wrote the original software. If you use this software
+    in a product, an acknowledgment in the product documentation would be
+    appreciated but is not required.
+
+    2. Altered source versions must be plainly marked as such, and must not be
+    misrepresented as being the original software.
+
+    3. This notice may not be removed or altered from any source
+    distribution.
+=end
+
+require 'test/unit'
+require 'irpack/entrypoint'
+require 'utils'
+
+class TC_IRPack_EntryPoint < Test::Unit::TestCase
+  include Utils
+
+  def test_compile
+    output_file = tempfilename('.dll')
+    module_name = 'TestModule'
+    entry_file = 'foo.rb'
+    references = ironruby_assemblies
+
+    assert_equal(output_file, IRPack::EntryPoint.compile(output_file, module_name, entry_file, references))
+    assert(File.exist?(output_file))
+    asm = nil
+    assert_nothing_raised do
+      asm = System::Reflection::Assembly.load_from(output_file)
+    end
+    entrypoint = asm.get_type("#{module_name}.EntryPoint")
+    assert_not_nil(entrypoint)
+    main = entrypoint.get_method('Main')
+    assert_not_nil(main)
+    create_package('foo.rb' => 'exit ARGV.size') do |package|
+      res = main.invoke(nil, System::Array[System::Object].new([package, System::Array[System::String].new(['hoge', 'fuga', 'piyo'])]))
+      assert_equal(3, res)
+    end
+  end
+
+  def test_run
+    output_file = tempfilename('.dll')
+    module_name = 'TestModule'
+    entry_file = 'main.rb'
+    references = ironruby_assemblies
+
+    IRPack::EntryPoint.compile(output_file, module_name, entry_file, references)
+    asm = System::Reflection::Assembly.load_from(output_file)
+    main = asm.get_type("#{module_name}.EntryPoint").get_method('Main')
+    assert_not_nil(main)
+    main_rb = <<-RB
+    puts 'Hello World!'
+    RB
+    create_package('main.rb' => main_rb) do |package|
+      res = main.invoke(nil, System::Array[System::Object].new([package, System::Array[System::String].new(0)]))
+      assert_equal(0, res)
+    end
+  end
+
+  def test_run_raised
+    output_file = tempfilename('.dll')
+    module_name = 'TestModule'
+    entry_file = 'main.rb'
+    references = ironruby_assemblies
+
+    IRPack::EntryPoint.compile(output_file, module_name, entry_file, references)
+    asm = System::Reflection::Assembly.load_from(output_file)
+    main = asm.get_type("#{module_name}.EntryPoint").get_method('Main')
+    assert_not_nil(main)
+    main_rb = <<-RB
+    raise RuntimeError, 'Hello Exception!'
+    RB
+    create_package('main.rb' => main_rb) do |package|
+      res = nil
+      err = System::IO::StringWriter.new
+      assert_nothing_raised do
+        prev_err = System::Console.error
+        System::Console.set_error(err)
+        res = main.invoke(nil, System::Array[System::Object].new([package, System::Array[System::String].new(0)]))
+        System::Console.set_error(prev_err)
+      end
+      assert_match(/Hello Exception!/, err.get_string_builder.to_string)
+      assert_equal(-1, res)
+    end
+  end
+end
+