cut with column names

ccut

#!/usr/bin/env ruby
# cat data.tsv | ccut -d delim -c name,name,name...

require 'trollop'
opts = Trollop::options do
  opt :delim, 'delimitor', default: "\t"
  opt :column, 'column name(s)', type: String
end

delim = opts[:delim]
column_names = opts[:column].split(',')

positions = [ ]
$<.each_with_index do |l, i|
  ary = l.chomp.split(delim)
  if i.zero?
    header = [ ]
    column_names.each do |c|
      pos = ary.find_index(c)
      unless pos.nil?
        positions.push pos
        header.push c
      end
    end
    puts header.join(delim)
  else
    puts positions.map{|pos| ary[pos]}.join(delim)
  end
end