# frozen_string_literal: true

require "bundler/gem_tasks"
require "rake/testtask"
require "yaml"
require_relative 'lib/rsyntaxtree'
require_relative 'lib/rsyntaxtree/utils'

# task default: "test"

Rake::TestTask.new do |task|
  task.pattern = "test/*_test.rb"
  task.warning = false
end

desc "Generate SVG and PNG example images"
task :generate do
  examples_dir = File.expand_path(File.join(__dir__, "docs", "_examples"))
  svg_dir = File.expand_path(File.join(__dir__, "docs", "assets", "svg"))
  png_dir = File.expand_path(File.join(__dir__, "docs", "assets", "img"))
  Dir.glob("*.md", base: examples_dir).map do |md|
    md = File.join(examples_dir, md)
    config = YAML.load_file(md)
    rst = File.read(md).scan(/```([^`]+)```/m).last.first

    opts = {
      format: "png",
      leafstyle: "auto",
      fontstyle: "sans",
      fontsize: 16,
      margin: 1,
      vheight: 2.0,
      color: "on",
      symmetrize: "on",
      transparent: "off",
      polyline: "off"
    }
    name = nil
    config.each do |key, value|
      next if value.to_s == ""

      case key
      when "name"
        name = value
        opts[:name] = name
      when "colorization"
        opts[:color] = value
      when "polyline"
        opts[:polyline] = value
      when "symmetrization"
        opts[:symmetrize] = value
      when "connector"
        opts[:leafstyle] = value
      when "font"
        opts[:fontstyle] = case value
                           when /mono/i
                             "mono"
                           when /sans/i
                             "sans"
                           when /serif/i
                             "serif"
                           when /wqy/i
                             "cjk"
                           else
                             "sans"
                           end
      end
    end
    opts[:data] = rst
    rsg = RSyntaxTree::RSGenerator.new(opts)

    File.open(File.join(svg_dir, "#{name}.svg"), "w") do |f|
      puts "Creating svg file: #{name}.svg"
      svg = rsg.draw_svg
      f.write(svg)
    end

    File.open(File.join(png_dir, "#{name}.png"), "w") do |f|
      puts "Creating png file: #{name}.png"
      png = rsg.draw_png
      f.write(png)
    end
  end
end