import os
import shutil
import tempfile

from unittest import TestCase, mock

from parameterized import parameterized_class

from aws_lambda_builders.builder import LambdaBuilder
from aws_lambda_builders.exceptions import WorkflowFailedError

import logging

logger = logging.getLogger("aws_lambda_builders.workflows.ruby_bundler.bundler")
workflow_logger = logging.getLogger("aws_lambda_builders.workflows.ruby_bundler.workflow")


@parameterized_class(
    ("runtime",),
    [
        ("ruby2.7",),
        ("ruby3.2",),
    ],
)
class TestRubyWorkflow(TestCase):
    """
    Verifies that `ruby` workflow works by building a Lambda using Bundler
    """

    runtime = ""

    TEST_DATA_FOLDER = os.path.join(os.path.dirname(__file__), "testdata")

    def setUp(self):
        self.artifacts_dir = tempfile.mkdtemp()
        self.scratch_dir = tempfile.mkdtemp()
        self.dependencies_dir = tempfile.mkdtemp()
        self.no_deps = os.path.join(self.TEST_DATA_FOLDER, "no-deps")
        self.builder = LambdaBuilder(language="ruby", dependency_manager="bundler", application_framework=None)

    def tearDown(self):
        shutil.rmtree(self.artifacts_dir)
        shutil.rmtree(self.scratch_dir)
        shutil.rmtree(self.dependencies_dir)

    def test_builds_project_without_dependencies(self):
        source_dir = os.path.join(self.TEST_DATA_FOLDER, "no-deps")
        self.builder.build(
            source_dir, self.artifacts_dir, self.scratch_dir, os.path.join(source_dir, "Gemfile"), runtime=self.runtime
        )
        expected_files = {"handler.rb", "Gemfile", "Gemfile.lock", ".bundle", "vendor"}
        output_files = set(os.listdir(self.artifacts_dir))
        self.assertEqual(expected_files, output_files)

    def test_builds_project_with_dependencies(self):
        source_dir = os.path.join(self.TEST_DATA_FOLDER, "with-deps")
        self.builder.build(
            source_dir, self.artifacts_dir, self.scratch_dir, os.path.join(source_dir, "Gemfile"), runtime=self.runtime
        )
        expected_files = {"handler.rb", "Gemfile", "Gemfile.lock", ".bundle", "vendor"}
        output_files = set(os.listdir(self.artifacts_dir))
        self.assertEqual(expected_files, output_files)

    def test_builds_project_and_ignores_excluded_files(self):
        source_dir = os.path.join(self.TEST_DATA_FOLDER, "excluded-files")
        self.builder.build(
            source_dir, self.artifacts_dir, self.scratch_dir, os.path.join(source_dir, "Gemfile"), runtime=self.runtime
        )
        expected_files = {"handler.rb", "Gemfile", "Gemfile.lock", ".bundle", "vendor"}
        output_files = set(os.listdir(self.artifacts_dir))
        self.assertEqual(expected_files, output_files)

    def test_fails_if_bundler_cannot_resolve_dependencies(self):
        source_dir = os.path.join(self.TEST_DATA_FOLDER, "broken-deps")
        with self.assertRaises(WorkflowFailedError) as ctx:
            self.builder.build(
                source_dir,
                self.artifacts_dir,
                self.scratch_dir,
                os.path.join(source_dir, "Gemfile"),
                runtime=self.runtime,
            )
        self.assertIn("RubyBundlerBuilder:RubyBundle - Bundler Failed: ", str(ctx.exception))

    def test_must_log_warning_if_gemfile_not_found(self):
        source_dir = os.path.join(self.TEST_DATA_FOLDER, "excludes-gemfile")
        with mock.patch.object(logger, "warning") as mock_warning:
            self.builder.build(
                source_dir,
                self.artifacts_dir,
                self.scratch_dir,
                os.path.join("non", "existent", "manifest"),
                runtime=self.runtime,
            )
        expected_files = {"handler.rb"}
        output_files = set(os.listdir(self.artifacts_dir))
        self.assertEqual(expected_files, output_files)
        mock_warning.assert_called_with("Gemfile not found. Continuing the build without dependencies.")

    def test_builds_project_without_downloaded_dependencies_with_dependencies_dir(self):
        source_dir = os.path.join(self.TEST_DATA_FOLDER, "with-deps")
        self.builder.build(
            source_dir,
            self.artifacts_dir,
            self.scratch_dir,
            os.path.join(source_dir, "Gemfile"),
            runtime=self.runtime,
            dependencies_dir=self.dependencies_dir,
            download_dependencies=False,
        )
        expected_files = {"handler.rb", "Gemfile"}
        output_files = set(os.listdir(self.artifacts_dir))
        self.assertEqual(expected_files, output_files)

    def test_builds_project_with_downloaded_dependencies_and_dependencies_dir(self):
        source_dir = os.path.join(self.TEST_DATA_FOLDER, "with-deps")
        self.builder.build(
            source_dir,
            self.artifacts_dir,
            self.scratch_dir,
            os.path.join(source_dir, "Gemfile"),
            runtime=self.runtime,
            dependencies_dir=self.dependencies_dir,
            download_dependencies=True,
        )
        expected_files = {"handler.rb", "Gemfile", "Gemfile.lock", ".bundle", "vendor"}
        output_files = set(os.listdir(self.artifacts_dir))
        self.assertEqual(expected_files, output_files)

        expected_dependencies_files = {"Gemfile.lock", ".bundle", "vendor"}
        output_dependencies_files = set(os.listdir(self.dependencies_dir))
        self.assertEqual(output_dependencies_files, expected_dependencies_files)

    def test_builds_project_without_downloaded_dependencies_without_dependencies_dir(self):
        source_dir = os.path.join(self.TEST_DATA_FOLDER, "with-deps")
        with mock.patch.object(workflow_logger, "info") as mock_info:
            self.builder.build(
                source_dir,
                self.artifacts_dir,
                self.scratch_dir,
                os.path.join(source_dir, "Gemfile"),
                runtime=self.runtime,
                dependencies_dir=None,
                download_dependencies=False,
            )
        expected_files = {"handler.rb", "Gemfile"}
        output_files = set(os.listdir(self.artifacts_dir))
        self.assertEqual(expected_files, output_files)

        mock_info.assert_called_with(
            "download_dependencies is False and dependencies_dir is None. Copying the source files into the "
            "artifacts directory. "
        )