generator.rb

require 'yaml'
require 'set'

class Generator
  ABBREVIATIONS = {
    "Knight" => "KGT",
    "Monk" => "MNK",
    "Thief" => "THF",
    "Black-Mage" => "BLM",
    "White-Mage" => "WHM",
    "Blue-Mage" => "BLU",
    "Mystic-Knight" => "MYS",
    "Time-Mage" => "TIM",
    "Summoner" => "SUM",
    "Red-Mage" => "RDM",
    "Berserker" => "BER",
    "Beastmaster" => "BST",
    "Geomancer" => "GEO",
    "Ninja" => "NIN",
    "Bard" => "BRD",
    "Ranger" => "RAN",
    "Samurai" => "SAM",
    "Dragoon" => "DRG",
    "Dancer" => "DNC",
    "Chemist" => "CHM",
    "Freelancer" => "FRE",
    "Mime" => "MIM",
    "Cannoneer" => "CAN",
    "Gladiator" => "GLD",
    "Oracle" => "ORC"
  }

  def initialize(jobs)
    @output = ""
    @tags = Set.new()
    @job_data = YAML.load(File.read('data/jobs.yaml'))
    jobs.map{|j| j.gsub(" ", "-")}.each do |job|
      if job == "debug"
        add_all_tags
        @tags << "debug"
        break
      end
      @tags << job

      @job_data['Jobs'][job].each do |tag|
        @tags << tag
      end
    end
  end

  def add_all_tags
    @job_data['Jobs'].each do |job, subtags|
      @tags << job
      subtags.each do |tag|
        @tags << tag
      end
    end
  end

  def get_walkthrough(beginning_node='begin')
    generate!(beginning_node) if @output == ""
    @output
  end

  def generate!(beginning_node='begin')
    handle_node(beginning_node, nil)
  end

  def handle_node(nodename, previous_node)
    node = YAML.load(File.read("data/nodes/#{nodename}.yaml"))
    node_header = node.keys.first
    node_data = node[node_header]

    if previous_node && !node_data['Metadata']['previous-nodes'].include?(previous_node)
      print "WARNING: unexpected node transition #{previous_node} => #{nodename}\n"
    end

    emit_header(node_header)
    node_data.each do |condition, hints|
      condition = condition.gsub('`', '')
      next if condition == "Metadata"
      if condition == "Generic" || condition_true?(condition)
        hints.each do |hint|
          emit_list_element(hint, condition)
        end
      elsif @tags.include?("debug") && !condition_true?(condition)
        print "WARNING: condition unreachable: #{condition} in node #{nodename}\n" unless condition.include?("NOT")
      end
    end
    emit_paragraph_break

    next_node = node_data['Metadata']['next-node']
    handle_node(next_node, nodename) unless next_node == 'end'
  end

  # todo: support and, or with more than 2 elements
  def condition_true?(condition)
    token = condition.split.first
    if token == 'NOT'
      return !condition_true?(condition.split[1..-1].join(" "))
    elsif token == 'UNION'
      subconditions = condition.split[1..-1]
      return subconditions.all?{|c| condition_true?(c)}
    elsif token == 'INTERSECTION'
      subconditions = condition.split[1..-1]
      return subconditions.detect{|c| condition_true?(c)}
    else
      return @tags.include?(condition)
    end
  end

  def emit_header(contents)
    @output << "## #{contents}\n"
  end

  def emit_list_element(contents, condition)
    @output << "* #{format_condition(condition)}#{contents}\n"
  end

  def emit_paragraph_break()
    @output << "\n"
  end

  def format_condition(condition)
    tokens = condition.split(" ")
    abbreviations = tokens.map{|tok| ABBREVIATIONS[tok]}.compact
    if abbreviations.length == 0
      ""
    else
      if tokens.first == 'UNION'
        "[#{abbreviations.join('+')}] "
      elsif tokens.first == 'INTERSECTION'
        "[#{abbreviations.join('|')}] "
      elsif tokens.first == 'NOT'
        "[!#{abbreviations.join(',')}] "
      else
        "[#{abbreviations.join(' ')}] "
      end
    end
  end
end