Hive Color

# -*- coding: utf-8 -*-

class Hiveql

  def initialize(hiveql)
    @hiveql = hiveql
  end

  attr_reader :hiveql

  # 文字列からインスタンスを作る
  def self.load(str)
    return Hiveql.new(str)
  end

  # ファイルの絶対パスからインスタンスを作る
  def self.load_file(file)
    return load(`cat #{file}`)
  end

  # hiveql内のfrom句をパースし、db.tableをユニークな配列にして返す
  def source_tables
    tables = Array.new
    split(hiveql).each{|hiveql|
      from_expressions = search_from_expressions(hiveql)
      next if from_expressions.empty?

      from_expressions.each{|table|
        tables.concat(table)
      }
    }

    tables.uniq
  end

  # クエリをセミコロンで分割する
  def split(hiveql)
    hiveql.split(';')
  end

  # from句を見つける
  # db.table を正規表現で見つける。db名の省略にも対応している
  def search_from_expressions(hiveql)
    return [] if (hiveql =~ /use\\s+\\w+\\s*;/mi) # use db; を見つける正規表現

    # from db_name.table_name を見つける正規表現
    regexp = /from\\s+((?:\\w+\\.)?\\w+)\\s+/mi

    if (hiveql =~ regexp)
      return hiveql.scan(regexp).uniq
    end

    return []
  end

end

# ここから実行開始
hiveql = Hiveql.load_file(ARGV[0])
puts hiveql.source_tables