asciidoctor-diagram 2.0.2 → 2.1.2

data/lib/asciidoctor-diagram/tikz/converter.rb

@@ -13,6 +13,11 @@ module Asciidoctor
         [:pdf, :svg]
       end
 
+      def collect_options(source)
+        {
+            :preamble => source.attr('preamble') == 'true'
+        }
+      end
 
       def convert(source, format, options)
         latexpath = source.find_command('pdflatex')
@@ -23,14 +28,28 @@ module Asciidoctor
           svgpath = nil
         end
 
+        if options[:preamble]
+          preamble, body = source.to_s.split(/^~~~~$/, 2)
+          unless body
+            body = preamble
+            preamble = ''
+          end
+        else
+          preamble = ''
+          body = source.to_s
+        end
+
         latex = <<'END'
 \documentclass[border=2bp, tikz]{standalone}
 \usepackage{tikz}
+END
+        latex << preamble
+        latex << <<'END'
 \begin{document}
 \begingroup
 \tikzset{every picture/.style={scale=1}}
 END
-        latex << source.to_s
+        latex << body
         latex << <<'END'
 \endgroup
 \end{document}

data/lib/asciidoctor-diagram/umlet/converter.rb

@@ -1,6 +1,7 @@
 require_relative '../diagram_converter'
 require_relative '../util/cli_generator'
 require_relative '../util/platform'
+require_relative '../util/java'
 
 module Asciidoctor
   module Diagram
@@ -15,8 +16,16 @@ module Asciidoctor
       end
 
       def convert(source, format, options)
-        generate_file(source.find_command('umlet'), 'uxf', format.to_s, source.to_s) do |tool_path, input_path, output_path|
-          [tool_path, '-action=convert', "-format=#{format.to_s}", "-filename=#{Platform.native_path(input_path)}", "-output=#{Platform.native_path(output_path)}"]
+        java = ::Asciidoctor::Diagram::Java.java
+
+        umlet = source.find_command('umlet')
+        ext = File.extname(umlet)
+        if ext == '' || ext != '.jar'
+          umlet = File.expand_path(File.basename(umlet, '.*') + '.jar', File.dirname(umlet))
+        end
+
+        generate_file(java, 'uxf', format.to_s, source.to_s) do |tool_path, input_path, output_path|
+          [tool_path, '-jar', Platform.native_path(umlet), '-action=convert', "-format=#{format.to_s}", "-filename=#{Platform.native_path(input_path)}", "-output=#{Platform.native_path(output_path)}"]
         end
       end
     end

data/lib/asciidoctor-diagram/util/cli.rb

@@ -12,7 +12,16 @@ module Asciidoctor
           opts = args.pop.dup if args.last.is_a? Hash
           in_data = opts && opts[:stdin_data]
 
+          if Hash === args.first
+            env = args.shift.dup
+          else
+            env = {}
+          end
+
           pb = java.lang.ProcessBuilder.new(*args)
+          env.each_pair do |key, value|
+            pb.environment.put(key, value)
+          end
           p = pb.start
 
           stdout = ""
@@ -67,12 +76,18 @@ module Asciidoctor
             opts = {}
           end
 
+          if Hash === args.first
+            env = args.shift.dup
+          else
+            env = {}
+          end
+
           # When the first argument is an array, we force capture3 (or better the underlying Kernel#spawn)
           # to use a non-shell execution variant.
           cmd = File.basename(args[0])
           args[0] = [args[0], cmd]
 
-          stdout, stderr, status = Open3.capture3(*args, opts)
+          stdout, stderr, status = Open3.capture3(env, *args, opts)
 
           exit = status.exitstatus
 

data/lib/asciidoctor-diagram/util/cli_generator.rb

@@ -1,3 +1,4 @@
+require 'asciidoctor/logging'
 require 'tempfile'
 require_relative 'cli'
 
@@ -5,21 +6,25 @@ module Asciidoctor
   module Diagram
     # @private
     module CliGenerator
-      def generate_stdin(tool, format, code)
-        tool_name = File.basename(tool)
+      include Asciidoctor::Logging
 
-        target_file = Tempfile.new([tool_name, ".#{format}"])
+      def generate_stdin(tool, format, code)
+        target_file = Tempfile.new([File.basename(tool), ".#{format}"])
         begin
           target_file.close
-
-          opts = yield tool, target_file.path
-
-          generate(opts, target_file.path, :stdin_data => code)
+          generate_stdin_file(tool, code, target_file.path) do |t|
+            yield t, target_file.path
+          end
         ensure
           target_file.unlink
         end
       end
 
+      def generate_stdin_file(tool, code, target_file_path)
+        opts = yield tool
+        generate(opts, target_file_path, :stdin_data => code)
+      end
+
       def generate_stdin_stdout(tool, code)
         if block_given?
           opts = yield tool
@@ -51,20 +56,37 @@ module Asciidoctor
         end
       end
 
+      def generate_file_stdout(tool, input_ext, code)
+        tool_name = File.basename(tool)
+
+        source_file = Tempfile.new([tool_name, ".#{input_ext}"])
+        begin
+          File.write(source_file.path, code)
+
+          opts = yield tool, source_file.path
+          generate(opts, :stdout)
+        ensure
+          source_file.unlink
+        end
+      end
+
       private
       def generate(opts, target_file, open3_opts = {})
         case opts
           when Array
             args = opts
             out_file = nil
+            env = {}
           when Hash
             args = opts[:args]
             out_file = opts[:out_file]
+            env = opts[:env] || {}
           else
             raise "Block passed to generate_file should return an Array or a Hash"
         end
 
-        result = ::Asciidoctor::Diagram::Cli.run(*args, open3_opts)
+        logger.debug "Executing #{args} with options #{open3_opts} and environment #{env}"
+        result = ::Asciidoctor::Diagram::Cli.run(env, *args, open3_opts)
 
         data = target_file == :stdout ? result[:out] : read_result(target_file, out_file)
 

data/lib/asciidoctor-diagram/util/gif.rb

@@ -7,14 +7,14 @@ module Asciidoctor
       GIF87A_SIGNATURE = 'GIF87a'.force_encoding(Encoding::ASCII_8BIT)
       GIF89A_SIGNATURE = 'GIF89a'.force_encoding(Encoding::ASCII_8BIT)
 
-      def self.get_image_size(data)
+      def self.post_process_image(data, optimise)
         bio = BinaryIO.new(data)
         gif_signature = bio.read_string(6)
         raise "Invalid GIF signature" unless gif_signature == GIF87A_SIGNATURE || gif_signature == GIF89A_SIGNATURE
 
         width = bio.read_uint16_le
         height = bio.read_uint16_le
-        [width, height]
+        [data, width, height]
       end
     end
   end

data/lib/asciidoctor-diagram/util/java.rb

@@ -5,9 +5,7 @@ module Asciidoctor
     # @private
     module Java
       def self.classpath
-        @classpath ||= [
-            File.expand_path(File.join('../..', 'server-1.3.15.jar'), File.dirname(__FILE__))
-        ]
+        @classpath ||= Dir[File.join(File.dirname(__FILE__), '*.jar')]
       end
 
       CRLF = "\r\n".encode(Encoding::US_ASCII)
@@ -53,6 +51,10 @@ module Asciidoctor
         io.set_encoding Encoding::US_ASCII
         status_line = io.readline(CRLF)
         status_line_parts = STATUS_LINE.match status_line
+        unless status_line_parts
+          raise "Unexpected HTTP status line: #{status_line}"
+        end
+
         resp[:code] = status_line_parts[1].to_i
         resp[:reason] = status_line_parts[2]
 
@@ -88,6 +90,119 @@ module Asciidoctor
           raise "#{prefix_msg}: #{response[:reason]}"
         end
       end
+
+      def self.java
+        @java_exe ||= find_java
+        raise "Could not find Java executable" unless @java_exe
+        @java_exe
+      end
+
+      private
+      def self.find_java
+        case ::Asciidoctor::Diagram::Platform.os
+          when :windows
+            path_to(ENV['JAVA_HOME'], 'bin/java.exe') || registry_lookup || ::Asciidoctor::Diagram::Which.which('java')
+          when :macosx
+            path_to(ENV['JAVA_HOME'], 'bin/java') || path_to(::Asciidoctor::Diagram::Cli.run('/usr/libexec/java_home')[:out].strip, 'bin/java') || ::Asciidoctor::Diagram::Which.which('java')
+          else
+            path_to(ENV['JAVA_HOME'], 'bin/java') || ::Asciidoctor::Diagram::Which.which('java')
+        end
+      end
+
+      def self.path_to(java_home, java_binary)
+        exe_path = File.expand_path(java_binary, java_home)
+        if File.executable?(exe_path)
+          exe_path
+        else
+          nil
+        end
+      end
+
+      JDK_KEY = 'HKEY_LOCAL_MACHINE\SOFTWARE\JavaSoft\Java Development Kit'
+      JRE_KEY = 'HKEY_LOCAL_MACHINE\SOFTWARE\JavaSoft\Java Runtime Environment'
+
+      def self.registry_lookup
+        registry_current(JRE_KEY) || registry_current(JDK_KEY) || registry_any()
+      end
+
+      def self.registry_current(key)
+        current_version = registry_query(key, 'CurrentVersion')
+        if current_version
+          java_home = registry_query("#{key}\\#{current_version}", 'JavaHome')
+          java_exe(java_home)
+        else
+          nil
+        end
+      end
+
+      def self.registry_any()
+        java_homes = registry_query('HKEY_LOCAL_MACHINE\SOFTWARE\JavaSoft', 'JavaHome', :recursive => true).values
+        java_homes.map { |path| java_exe(path) }.find { |exe| !exe.nil? }
+      end
+
+      def self.java_exe(java_home)
+        java = File.expand_path('bin/java.exe', java_home)
+
+        if File.executable?(java)
+          java
+        else
+          nil
+        end
+      end
+
+      def self.registry_query(key, value = nil, opts = {})
+        args = ['reg', 'query']
+        args << key
+        args << '/v' << value unless value.nil?
+        args << '/s' if opts[:recursive]
+
+        begin
+          lines = ::Asciidoctor::Diagram::Cli.run(*args)[:out].lines.reject { |l| l.strip.empty? }.each
+        rescue
+          lines = [].each
+        end
+
+        result = {}
+
+        while true
+          begin
+            begin
+              k = lines.next
+            rescue StopIteration
+              break
+            end
+
+            unless k.start_with? key
+              next
+            end
+
+            v = nil
+            begin
+              v = lines.next.strip if lines.peek.start_with?(' ')
+            rescue StopIteration
+              break
+            end
+
+            if !k.valid_encoding? || (v && !v.valid_encoding?)
+              next
+            end
+
+            if v && (md = /([^\s]+)\s+(REG_[^\s]+)\s+(.+)/.match(v))
+              v_name = md[1]
+              v_value = md[3]
+              result["#{k}\\#{v_name}"] = v_value
+            else
+              result[k] = v
+            end
+          end
+        end
+
+        if value && !opts[:recursive]
+          result.values[0]
+        else
+          result
+        end
+      end
     end
   end
 end

data/lib/asciidoctor-diagram/util/java_jruby.rb

@@ -10,7 +10,10 @@ module Asciidoctor
           return
         end
 
-        classpath.flatten.each { |j| require j }
+        classpath.flatten.each do |j|
+          raise "Classpath item #{j} does not exist" unless File.exist?(j)
+          require j
+        end
         @loaded = true
       end
 

data/lib/asciidoctor-diagram/util/java_socket.rb

@@ -10,9 +10,14 @@ module Asciidoctor
     module Java
       class CommandServer
         def initialize(java, classpath)
+          classpath.each do |file|
+            raise "Classpath item #{file} does not exist" unless File.exist?(file)
+          end
+
           args = []
           args << '-Djava.awt.headless=true'
           args << '-Djava.net.useSystemProxies=true'
+          args << '-Dfile.encoding=UTF-8'
           args << '-cp'
           args << classpath.flatten.map { |jar| ::Asciidoctor::Diagram::Platform.host_os_path(jar).strip }.join(::Asciidoctor::Diagram::Platform.host_os_path_separator)
           args << 'org.asciidoctor.diagram.StdInOutCommandServer'
@@ -41,11 +46,8 @@ module Asciidoctor
       end
 
       def self.instance
-        @java_exe ||= find_java
-        raise "Could not find Java executable" unless @java_exe
-
         unless defined?(@command_server) && @command_server
-          server = CommandServer.new(@java_exe, classpath)
+          server = CommandServer.new(java, classpath)
           @command_server = server
           at_exit do
             server.shutdown
@@ -61,113 +63,10 @@ module Asciidoctor
         # headers = req[:headers] ||= {}
         # headers['Host'] = "localhost:#{svr.port}"
         format_request(req, svr.io)
-        parse_response(svr.io)
-      end
-
-      private
-      def self.find_java
-        case ::Asciidoctor::Diagram::Platform.os
-          when :windows
-            path_to(ENV['JAVA_HOME'], 'bin/java.exe') || registry_lookup || ::Asciidoctor::Diagram::Which.which('java')
-          when :macosx
-            path_to(ENV['JAVA_HOME'], 'bin/java') || path_to(::Asciidoctor::Diagram::Cli.run('/usr/libexec/java_home')[:out].strip, 'bin/java') || ::Asciidoctor::Diagram::Which.which('java')
-          else
-            path_to(ENV['JAVA_HOME'], 'bin/java') || ::Asciidoctor::Diagram::Which.which('java')
-        end
-      end
-
-      def self.path_to(java_home, java_binary)
-        exe_path = File.expand_path(java_binary, java_home)
-        if File.executable?(exe_path)
-          exe_path
-        else
-          nil
-        end
-      end
-
-      JDK_KEY = 'HKEY_LOCAL_MACHINE\SOFTWARE\JavaSoft\Java Development Kit'
-      JRE_KEY = 'HKEY_LOCAL_MACHINE\SOFTWARE\JavaSoft\Java Runtime Environment'
-
-      def self.registry_lookup
-        registry_current(JRE_KEY) || registry_current(JDK_KEY) || registry_any()
-      end
-
-      def self.registry_current(key)
-        current_version = registry_query(key, 'CurrentVersion')
-        if current_version
-          java_home = registry_query("#{key}\\#{current_version}", 'JavaHome')
-          java_exe(java_home)
-        else
-          nil
-        end
-      end
-
-      def self.registry_any()
-        java_homes = registry_query('HKEY_LOCAL_MACHINE\SOFTWARE\JavaSoft', 'JavaHome', :recursive => true).values
-        java_homes.map { |path| java_exe(path) }.find { |exe| !exe.nil? }
-      end
-
-      def self.java_exe(java_home)
-        java = File.expand_path('bin/java.exe', java_home)
-
-        if File.executable?(java)
-          java
-        else
-          nil
-        end
-      end
-
-      def self.registry_query(key, value = nil, opts = {})
-        args = ['reg', 'query']
-        args << key
-        args << '/v' << value unless value.nil?
-        args << '/s' if opts[:recursive]
-
         begin
-          lines = ::Asciidoctor::Diagram::Cli.run(*args)[:out].lines.reject { |l| l.strip.empty? }.each
+          parse_response(svr.io)
         rescue
-          lines = [].each
-        end
-
-        result = {}
-
-        while true
-          begin
-            begin
-              k = lines.next
-            rescue StopIteration
-              break
-            end
-
-            unless k.start_with? key
-              next
-            end
-
-            v = nil
-            begin
-              v = lines.next.strip if lines.peek.start_with?(' ')
-            rescue StopIteration
-              break
-            end
-
-            if !k.valid_encoding? || (v && !v.valid_encoding?)
-              next
-            end
-
-            if v && (md = /([^\s]+)\s+(REG_[^\s]+)\s+(.+)/.match(v))
-              v_name = md[1]
-              v_value = md[3]
-              result["#{k}\\#{v_name}"] = v_value
-            else
-              result[k] = v
-            end
-          end
-        end
-
-        if value && !opts[:recursive]
-          result.values[0]
-        else
-          result
+          raise "Error processing request #{req}\nEncoding of input is #{req[:body].encoding}"
         end
       end
     end