Sha256: dc1181dff7733455f387a20e5dae7e51bcf509a0027c6d79dde940ef429d46ed

Contents?: true

Size: 1.74 KB

Versions: 9

Compression:

Stored size: 1.74 KB

Contents

begin
  require "posix-spawn"
rescue LoadError
  require "open3"
end

require "github/markup/implementation"


module GitHub
  module Markup
    class CommandError < RuntimeError
    end

    class CommandImplementation < Implementation
      attr_reader :command, :block, :name

      def initialize(regexp, languages, command, name, &block)
        super(regexp, languages)
        @command = command.to_s
        @block = block
        @name = name
      end

      def render(filename, content, options: {})
        rendered = execute(command, content)
        rendered = rendered.to_s.empty? ? content : rendered
        call_block(rendered, content)
      end

    private
      def call_block(rendered, content)
        if block && block.arity == 2
          block.call(rendered, content)
        elsif block
          block.call(rendered)
        else
          rendered
        end
      end

      if defined?(POSIX::Spawn)
        def execute(command, target)
          spawn = POSIX::Spawn::Child.new(*command, :input => target)
          if spawn.status.success?
            sanitize(spawn.out, target.encoding)
          else
            raise CommandError.new(spawn.err.strip)
          end
        end
      else
        def execute(command, target)
          output = Open3.popen3(*command) do |stdin, stdout, stderr, wait_thr|
            stdin.puts target
            stdin.close
            if wait_thr.value.success?
              stdout.readlines
            else
              raise CommandError.new(stderr.readlines.join('').strip)
            end
          end
          sanitize(output.join(''), target.encoding)
        end
      end

      def sanitize(input, encoding)
        input.gsub("\r", '').force_encoding(encoding)
      end

    end
  end
end

Version data entries

9 entries across 9 versions & 1 rubygems

Version Path
github-markup-4.0.2 lib/github/markup/command_implementation.rb
github-markup-4.0.1 lib/github/markup/command_implementation.rb
github-markup-4.0.0 lib/github/markup/command_implementation.rb
github-markup-3.0.5 lib/github/markup/command_implementation.rb
github-markup-3.0.4 lib/github/markup/command_implementation.rb
github-markup-3.0.3 lib/github/markup/command_implementation.rb
github-markup-3.0.2 lib/github/markup/command_implementation.rb
github-markup-3.0.1 lib/github/markup/command_implementation.rb
github-markup-3.0.0 lib/github/markup/command_implementation.rb