rubyistokei.rb

require 'bundler'
Bundler.require

require 'yaml'
require 'json'

class Database
  def initialize(path)
    @data = Dir[File.join(path, '*.yaml')].sort.map do |yaml_path|
      hash = YAML.load_file(yaml_path)
      id = File.basename(yaml_path, '.yaml')
      hash.merge(id: id)
    end
  end

  attr_reader :data
end

module Rubyistokei
  class Application < Sinatra::Application
    configure do
      set :protection, :except => :frame_options

      data_path = File.join(__dir__, 'data')
      database = Database.new(data_path)
      DATA_JSON = JSON.dump(database.data)
    end

    get '/' do
      haml :index
    end

    get '/css/screen.css' do
      scss :screen
    end

    get '/data.json' do
      content_type :json
      DATA_JSON
    end
  end
end