Class: Ned::Main

Inherits:

Object

• Object
• Ned::Main

Defined in:

lib/ned.rb

Instance Method Summary

• #create_command(command_args, inputs, output, error) ⇒ Object
• #help ⇒ Object
• #initialize ⇒ Main constructor

  A new instance of Main.

• #run(args, input, output, error) ⇒ Object

Constructor Details

#initialize ⇒ Main

Returns a new instance of Main.

# File 'lib/ned.rb', line 42

def initialize
  @options = { }

  @option_parser = OptionParser.new do |opts|
    opts.banner = "      Ned, a stream editor.\n\n      Usage: ned [[ned_arg ...] --] [[command] [: command ...]]\n\n      OPTIONS\n    EOF\n\n    opts.on('-i', '--inplace', 'Edit each file in place by processing each file independently.') do |inplace|\n      raise OptionParser::ParseError.new('duplicate flag') if @options[:inplace]\n\n      inplace\n    end\n\n    opts.on('-s', '--separate', 'Process input files one at a time rather than as one stream.') do |separate|\n      raise OptionParser::ParseError.new('duplicate flag') if @options[:separate]\n\n      separate\n    end\n\n    opts.on('-h', '--help [COMMAND]', 'Print this help.') do |help|\n      help || true\n    end\n\n    opts.on('-v', '--version', 'Print Ned version.') do |version|\n      version\n    end\n  end\nend\n"

Instance Method Details

#create_command(command_args, inputs, output, error) ⇒ Object

# File 'lib/ned.rb', line 249

def create_command(command_args, inputs, output, error)
  previous = Ned::Read.new(inputs)
  command_args.each do |args|
    command = Ned::CommandRegistry.find(args.shift).new(previous)
    begin
      command.parse(args)
      raise OptionParser::InvalidArgument.new(args[0]) unless args.empty?
    rescue OptionParser::ParseError  => e
      error.puts "fatal: error while parsing options for #{command.class.long_name}: #{e.message}"
      error.puts
      error.puts command.class.help
      return 1
    end

    if command.options[:help]
      output.puts command.class.help
      return 0
    end

    if command.options[:version]
      output.puts "Ned version #{Ned::VERSION}"
      return 0
    end

    previous = command
  end

  [nil, previous]
end

#help ⇒ Object

# File 'lib/ned.rb', line 76

def help
  help = @option_parser.help

  help << "\nCOMMANDS\n"

  Ned::CommandRegistry.all.each do |command|
    help << "\n#{command.long_name} (#{command.short_name})\n\n"
    command.help.split("\n").each do |line|
      help << "  #{line}\n"
    end
  end
  help
end

#run(args, input, output, error) ⇒ Object

# File 'lib/ned.rb', line 90

def run(args, input, output, error)
  args = args.map do |arg|
    arg = arg.dup
    Ned::Strings.unescape!(arg)
    arg
  end

  ned_args = []
  command_args = []
  has_stdin = !input.tty?

  if index = args.index('--')
    ned_args.concat(args.shift(index))
    args.shift
  elsif args[0]&.start_with?('-')
    ned_args = args
    args = []
  end

  command_args.concat(args.shift(args.size))

  begin
    @option_parser.parse!(ned_args, into: @options)
  rescue OptionParser::ParseError  => e
    error.puts "fatal: error while parsing options: #{e.message}"
    return 1
  end

  if help_option = @options[:help]
    if help_option.is_a? TrueClass
      output.puts help
      return 0
    elsif help_option.is_a? String
      command = Ned::CommandRegistry.find(help_option)
      unless command
        error.puts "fatal: \"#{help_option}\" is not a valid command"
        return 1
      end
      output.puts command.help
      return 0
    end
  end

  if @options[:version]
    output.puts "Ned version #{Ned::VERSION}"
    return 0
  end

  inputs = ned_args.map do |current|
    if current == '-'
      input
    else
      unless File.exist? current
        error.puts "fatal: file not found: \"#{current}\""
        return 1
      end
      File.new(current)
    end
  end

  inputs << input if inputs.empty?
  inputs.unshift input if !inputs.index(input) && has_stdin

  command_args = command_args.inject([[]]) do |array, arg|
    if arg == ':'
      array << []
    else
      array[-1] << arg
    end
    array
  end

  if command_args.size == 1 && command_args[0] == []
    command_args.clear
  end

  if command_args.index([])
    error.puts 'fatal: empty command found'
    return 1
  end

  command_args.each do |array|
    array.each_with_index do |arg, i|
      array[i] = arg[1..-1] if arg == '\:'
    end
  end

  command_args.each do |args|
    unless Ned::CommandRegistry.find(args[0])
      error.puts "fatal: \"#{args[0]}\" is not a valid command"
      return 1
    end
  end

  if inputs.count(input) > 1
    error.puts 'fatal: can\'t read from stdin more than once'
    return 1
  end

  if @options[:inplace] && inputs.index(input)
    error.puts "fatal: option -i specified with stdin"
    return 1
  end

  if @options[:inplace]
    inputs.each do |i|
      dup = command_args.map { |a| a.dup }
      result, command = create_command(dup, [i], output, error)

      return result if result

      if inputs.index(input) && !has_stdin
        error.puts 'fatal: can\'t read from stdin'
        return 1
      end

      temp = Tempfile.new(File.expand_path(i))
      print = Ned::Print.new(command, temp)

      while line = print.execute; end

      temp.rewind
      IO.copy_stream(temp, File.expand_path(i))
      temp.unlink
    end
  elsif @options[:separate]
    inputs.each do |i|
      dup = command_args.map { |a| a.dup }
      result, command = create_command(dup, [i], output, error)

      return result if result

      if inputs.index(input) && !has_stdin
        error.puts 'fatal: can\'t read from stdin'
        return 1
      end

      print = Ned::Print.new(command, output)

      while line = print.execute; end
    end
  else
    result, command = create_command(command_args, inputs, output, error)

    return result if result

    if inputs.index(input) && !has_stdin
      error.puts 'fatal: can\'t read from stdin'
      return 1
    end

    print = Ned::Print.new(command, output)

    while line = print.execute; end
  end

  0
end