require 'rails/generators'
require 'rails/generators/migration'

module SystemMetrics
  module Generators
    class MigrationGenerator < Rails::Generators::Base
      include Rails::Generators::Migration

      desc "Create migration for System Metrics metrics table"

      source_root File.expand_path("../templates", __FILE__)

      def self.next_migration_number(dirname)
        if ActiveRecord::Base.timestamped_migrations
          Time.now.utc.strftime("%Y%m%d%H%M%S")
        else
          "%.3d" % (current_migration_number(dirname) + 1)
        end
      end

      def create_migration_file
        migration_template 'migration.rb', 'db/migrate/create_metrics_table.rb'
      end
    end
  end
end