# Copyright 2018 Amazon.com, Inc. or its affiliates. All Rights Reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License"). You may not
# use this file except in compliance with the License. A copy of the License is
# located at
#
#     http://aws.amazon.com/apache2.0/
#
# or in the "license" file accompanying this file. This file is distributed on
# an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
# or implied. See the License for the specific language governing permissions
# and limitations under the License.

require "rails/generators/rails/app/app_generator"
require "rails/generators/testing/behaviour"
require "rails/generators/testing/assertions"
require "fileutils"
require "minitest/spec"

module AwsRecord
  module Generators
    class GeneratorTestHelper
      include Minitest::Assertions
      attr_accessor :assertions

      include Rails::Generators::Testing::Behaviour
      include Rails::Generators::Testing::Assertions
      include FileUtils

      def initialize(klass, dest)
        @temp_root = File.expand_path(dest)

        GeneratorTestHelper.tests klass
        temp_app_dest = File.join(File.expand_path(@temp_root, __dir__), "test_app")
        GeneratorTestHelper.destination temp_app_dest
        
        destination_root_is_set?
        prepare_destination
        setup_test_app
        ensure_current_path

        self.assertions = 0
      end

      def run_in_test_app(cmd)
        cd destination_root
        a = `rails #{cmd}`

        ensure_current_path
      end

      def assert_file_fixture(generated_file, actual_file)
        assert File.exist?(generated_file), "Expected file #{generated_file.inspect} to exist, but does not"
        assert File.exist?(actual_file), "Expected file #{actual_file.inspect} to exist, but does not"
        assert identical? generated_file, actual_file
      end

      def assert_model_rand_table_name(generated_file, actual_file, table_name)
        assert File.exist?(generated_file), "Expected file #{generated_file.inspect} to exist, but does not"
        assert File.exist?(actual_file), "Expected file #{actual_file.inspect} to exist, but does not"

        fixture = File.read(actual_file)
        generated = File.read(generated_file)
        fixture = fixture.gsub(/#table_name#/, table_name)

        assert fixture == generated
      end

      def cleanup
        rm_rf @temp_root
      end

      def run_generator(args = default_arguments, config = {})
        result = nil
        capture(:stderr) do
          result = super
        end
        result
      end

      private

      def setup_test_app
        Rails::Generators::AppGenerator.start [destination_root, '--skip-bundle', '--skip-git', '--skip-spring', '--skip-test', '--force', '--quiet']
        `echo 'gem "aws-record-generator", :path => "../../"' >> "#{destination_root}/Gemfile"`
        `bundle install --gemfile "#{destination_root}/Gemfile"`
      end
    end
  end
end