import os
import shutil
import sys
from tempfile import mkdtemp, mkstemp
from unittest import TestCase, mock

from fs_uae_wrapper import base, utils


class TestBase(TestCase):

    def setUp(self):
        fd, self.fname = mkstemp()
        self.dirname = mkdtemp()
        self.confdir = mkdtemp()
        os.close(fd)
        self._argv = sys.argv[:]
        sys.argv = ['fs-uae-wrapper']
        self.curdir = os.path.abspath(os.curdir)

    def tearDown(self):
        os.chdir(self.curdir)
        try:
            shutil.rmtree(self.dirname)
        except OSError:
            pass
        try:
            shutil.rmtree(self.confdir)
        except OSError:
            pass
        os.unlink(self.fname)
        sys.argv = self._argv[:]

    def test_clean(self):

        bobj = base.Base('Config.fs-uae', utils.CmdOption(), {})
        bobj.clean()
        self.assertTrue(os.path.exists(self.dirname))

        bobj.dir = self.dirname
        bobj.clean()
        self.assertFalse(os.path.exists(self.dirname))

    @mock.patch('os.path.exists')
    @mock.patch('fs_uae_wrapper.utils.get_config')
    def test_normalize_options(self, get_config, os_exists):

        os_exists.return_value = True
        bobj = base.Base('Config.fs-uae', utils.CmdOption(), {})

        get_config.return_value = {'kickstarts_dir': '/some/path'}
        bobj._normalize_options()
        self.assertDictEqual(bobj.fsuae_options, {})

        os.chdir(self.dirname)
        get_config.return_value = {'fmv_rom': 'bar'}
        bobj._normalize_options()
        self.assertDictEqual(bobj.fsuae_options,
                             {'fmv_rom': os.path.join(self.dirname, 'bar')})

        get_config.return_value = {'floppies_dir': '../some/path'}
        bobj.fsuae_options = utils.CmdOption()
        result = os.path.abspath(os.path.join(self.dirname, '../some/path'))
        bobj._normalize_options()
        self.assertDictEqual(bobj.fsuae_options, {'floppies_dir': result})

        bobj.conf_file = os.path.join(self.dirname, 'Config.fs-uae')
        get_config.return_value = {'cdroms_dir': '$CONFIG/../path'}
        bobj.fsuae_options = utils.CmdOption()
        result = os.path.abspath(os.path.join(self.dirname, '../path'))
        bobj._normalize_options()
        self.assertDictEqual(bobj.fsuae_options, {'cdroms_dir': result})

        get_config.return_value = {'cdroms_dir': '$HOME/path'}
        bobj.fsuae_options = utils.CmdOption()
        bobj._normalize_options()
        self.assertDictEqual(bobj.fsuae_options, {})

        get_config.return_value = {'cdroms_dir': '$WRAPPER/path'}
        bobj.fsuae_options = utils.CmdOption()
        bobj.dir = self.dirname
        bobj._normalize_options()
        self.assertDictEqual(bobj.fsuae_options,
                             {'cdroms_dir': os.path.join(bobj.dir, 'path')})

        get_config.return_value = {'cdroms_dir': '~/path'}
        bobj.fsuae_options = utils.CmdOption()
        bobj._normalize_options()
        self.assertDictEqual(bobj.fsuae_options, {})

        get_config.return_value = {'random_item': 10}
        bobj.fsuae_options = utils.CmdOption()
        bobj._normalize_options()
        self.assertDictEqual(bobj.fsuae_options, {})

    @mock.patch('os.path.exists')
    @mock.patch('fs_uae_wrapper.utils.get_config')
    def test_normalize_options_path_not_exists(self, get_config, os_exists):

        os_exists.return_value = False
        bobj = base.Base('Config.fs-uae', utils.CmdOption(), {})

        get_config.return_value = {'kickstarts_dir': '/some/path'}
        bobj._normalize_options()
        self.assertDictEqual(bobj.fsuae_options, {})

        os.chdir(self.dirname)
        get_config.return_value = {'fmv_rom': 'bar'}
        bobj._normalize_options()
        self.assertDictEqual(bobj.fsuae_options, {'fmv_rom': 'bar'})

        get_config.return_value = {'floppies_dir': '../some/path'}
        bobj.fsuae_options = utils.CmdOption()
        bobj._normalize_options()
        self.assertDictEqual(bobj.fsuae_options,
                             {'floppies_dir': '../some/path'})

    def test_set_assets_paths(self):

        bobj = base.Base('Config.fs-uae', utils.CmdOption(), {})
        os.chdir(self.dirname)
        bobj.conf_file = 'Config.fs-uae'
        bobj.all_options = {'wrapper_archive': 'foo.7z',
                            'wrapper_archiver': '7z'}

        bobj._set_assets_paths()
        full_path = os.path.join(self.dirname, 'Config_save.7z')
        self.assertEqual(bobj.save_filename, full_path)

        bobj.all_options = {'wrapper_archive':  '/home/user/foo.7z',
                            'wrapper_archiver': '7z'}

        bobj._set_assets_paths()
        full_path = os.path.join(self.dirname, 'Config_save.7z')
        self.assertEqual(bobj.save_filename, full_path)

    def test_copy_conf(self):

        bobj = base.Base('Config.fs-uae', utils.CmdOption(), {})
        bobj.conf_file = self.fname
        bobj.dir = self.dirname

        self.assertTrue(bobj._copy_conf())
        self.assertTrue(os.path.exists(os.path.join(self.dirname,
                                                    'Config.fs-uae')))

    @mock.patch('fs_uae_wrapper.utils.run_command')
    def test_run_emulator(self, run):

        bobj = base.Base('Config.fs-uae', utils.CmdOption(), {})
        bobj.dir = self.dirname

        self.assertTrue(bobj._run_emulator())
        run.assert_called_once_with(['fs-uae'])

        # Errors from emulator are not fatal to wrappers
        run.reset_mock()
        run.return_value = False
        self.assertTrue(bobj._run_emulator())
        run.assert_called_once_with(['fs-uae'])

        # pass the options
        bobj.fsuae_options = utils.CmdOption({'foo': '1'})
        run.reset_mock()
        run.return_value = False
        self.assertTrue(bobj._run_emulator())
        run.assert_called_once_with(['fs-uae', '--foo'])

    @mock.patch('fs_uae_wrapper.base.Base._get_saves_dir')
    @mock.patch('fs_uae_wrapper.utils.create_archive')
    def test_save_save(self, carch, saves_dir):

        os.chdir(self.confdir)

        bobj = base.Base('myconf.fs-uae', utils.CmdOption(), {})
        bobj.dir = self.dirname
        bobj.save_filename = os.path.join(self.confdir, 'myconf_save.7z')

        saves_dir.return_value = None
        carch.return_value = True

        self.assertTrue(bobj._save_save(),
                        'there is assumption, that wrapper_save_state is'
                        ' false by default. Here it was true.')

        bobj.all_options['wrapper_save_state'] = '1'
        self.assertTrue(bobj._save_save(),
                        'unexpected save_state directory found')

        saves_dir.return_value = bobj.save_filename
        with open(bobj.save_filename, 'w') as fobj:
            fobj.write('asd')

        os.mkdir(os.path.join(self.dirname, 'fs-uae-save'))
        self.assertTrue(bobj._save_save())

        carch.return_value = False
        self.assertFalse(bobj._save_save())

    @mock.patch('fs_uae_wrapper.utils.extract_archive')
    def test_load_save(self, earch):

        bobj = base.Base('Config.fs-uae', utils.CmdOption(), {})
        bobj.dir = self.dirname
        bobj.save_filename = "foobar_save.7z"
        earch.return_value = 0

        # By default there would be no save state persistence
        self.assertTrue(bobj._load_save())

        # set wrapper_save_state option, so we can proceed with test
        bobj.all_options['wrapper_save_state'] = '1'

        # fail to load save is not fatal
        self.assertTrue(bobj._load_save())

        os.chdir(self.confdir)
        with open(bobj.save_filename, 'w') as fobj:
            fobj.write('asd')

        self.assertTrue(bobj._load_save())
        earch.assert_called_once_with(bobj.save_filename)

        # failure in searching for archiver are also non fatal
        earch.reset_mock()
        earch.return_value = 1
        self.assertTrue(bobj._save_save())

    def test_get_saves_dir(self):

        bobj = base.Base('Config.fs-uae', utils.CmdOption(), {})
        bobj.dir = self.dirname

        self.assertIsNone(bobj._get_saves_dir())

        bobj.all_options['save_states_dir'] = '/some/path'
        self.assertIsNone(bobj._get_saves_dir())

        bobj.all_options['save_states_dir'] = '$WRAPPER/../saves'
        self.assertIsNone(bobj._get_saves_dir())

        bobj.all_options['save_states_dir'] = '/foo/$WRAPPER/saves'
        self.assertIsNone(bobj._get_saves_dir())

        bobj.all_options['save_states_dir'] = '$WRAPPER/saves'
        self.assertIsNone(bobj._get_saves_dir())

        path = os.path.join(self.dirname, 'saves')
        with open(path, 'w') as fobj:
            fobj.write('\n')
        self.assertIsNone(bobj._get_saves_dir())

        os.unlink(path)
        os.mkdir(path)
        self.assertEqual(bobj._get_saves_dir(), 'saves')

        bobj.all_options['save_states_dir'] = '$WRAPPER/saves/'
        self.assertEqual(bobj._get_saves_dir(), 'saves')

    @mock.patch('fs_uae_wrapper.path.which')
    def test_validate_options(self, which):

        which.return_value = None

        bobj = base.Base('Config.fs-uae', utils.CmdOption(), {})
        bobj.all_options = {}

        self.assertFalse(bobj._validate_options())

        bobj.all_options = {'wrapper': 'dummy'}
        self.assertTrue(bobj._validate_options())

        bobj.all_options = {'wrapper': 'dummy',
                            'wrapper_save_state': '0'}
        self.assertTrue(bobj._validate_options())

        bobj.all_options = {'wrapper': 'dummy',
                            'wrapper_archiver': 'rar'}
        self.assertTrue(bobj._validate_options())

        bobj.all_options = {'wrapper': 'dummy',
                            'wrapper_save_state': '1'}
        self.assertFalse(bobj._validate_options())

        which.return_value = '7z'
        bobj.all_options = {'wrapper': 'dummy',
                            'wrapper_save_state': '1'}
        self.assertTrue(bobj._validate_options())

        bobj.all_options = {'wrapper': 'dummy',
                            'wrapper_save_state': '1',
                            'wrapper_archiver': '7z'}
        self.assertTrue(bobj._validate_options())

        which.return_value = None
        bobj.all_options = {'wrapper': 'dummy',
                            'wrapper_save_state': '1',
                            'wrapper_archiver': '7z'}
        self.assertFalse(bobj._validate_options())

    @mock.patch('fs_uae_wrapper.path.which')
    def test_run_clean(self, which):

        which.return_value = 'rar'

        bobj = base.Base('Config.fs-uae', utils.CmdOption(), {})
        bobj.all_options = {}

        self.assertFalse(bobj.run())

        bobj.all_options = {'wrapper': 'dummy',
                            'wrapper_archiver': 'rar',
                            'wrapper_archive': 'foo.7z'}
        try:
            self.assertTrue(bobj.run())
            self.assertTrue(os.path.exists(bobj.dir))
        finally:
            bobj.clean()


class TestArchiveBase(TestCase):

    def setUp(self):
        fd, self.fname = mkstemp()
        self.dirname = mkdtemp()
        self.confdir = mkdtemp()
        os.close(fd)
        self._argv = sys.argv[:]
        sys.argv = ['fs-uae-wrapper']
        self.curdir = os.path.abspath(os.curdir)

    def tearDown(self):
        os.chdir(self.curdir)
        try:
            shutil.rmtree(self.dirname)
        except OSError:
            pass
        try:
            shutil.rmtree(self.confdir)
        except OSError:
            pass
        os.unlink(self.fname)
        sys.argv = self._argv[:]

    def test_set_assets_paths(self):

        bobj = base.ArchiveBase('Config.fs-uae', utils.CmdOption(), {})
        os.chdir(self.dirname)
        bobj.conf_file = 'Config.fs-uae'
        bobj.all_options = {'wrapper_archive': 'foo.7z',
                            'wrapper_archiver': '7z'}

        bobj._set_assets_paths()
        full_path = os.path.join(self.dirname, 'Config_save.7z')
        self.assertEqual(bobj.save_filename, full_path)

        bobj.all_options = {'wrapper_archive':  '/home/user/foo.7z',
                            'wrapper_archiver': '7z'}

        bobj._set_assets_paths()
        full_path = os.path.join(self.dirname, 'Config_save.7z')
        self.assertEqual(bobj.save_filename, full_path)

    @mock.patch('fs_uae_wrapper.utils.extract_archive')
    def test_extract(self, utils_extract):

        bobj = base.ArchiveBase('Config.fs-uae', utils.CmdOption(), {})
        bobj.arch_filepath = self.fname
        bobj.dir = self.dirname

        utils_extract.return_value = False

        # message for the gui is taken from title in fs-uae conf or, if there
        # is no such entry, use archive name, which is mandatory to provide
        bobj.all_options = {'title': 'foo_game', 'wrapper_gui_msg': '1'}
        self.assertFalse(bobj._extract())
        utils_extract.assert_called_once_with(self.fname, 'foo_game')

        utils_extract.reset_mock()
        bobj.all_options = {'wrapper_archive': 'arch.tar',
                            'wrapper_gui_msg': '1'}
        self.assertFalse(bobj._extract())
        utils_extract.assert_called_once_with(self.fname, 'arch.tar')

        # lets pretend, the extracting has failed
        utils_extract.reset_mock()
        bobj.all_options = {'wrapper_gui_msg': '0'}
        utils_extract.return_value = False
        self.assertFalse(bobj._extract())
        utils_extract.assert_called_once_with(self.fname, '')

    @mock.patch('fs_uae_wrapper.base.ArchiveBase._get_wrapper_archive_name')
    def test_validate_options(self, get_wrapper_arch_name):

        bobj = base.ArchiveBase('Config.fs-uae', utils.CmdOption(), {})
        bobj.all_options = {}

        self.assertFalse(bobj._validate_options())

        get_wrapper_arch_name.return_value = None
        bobj.all_options = {'wrapper': 'dummy'}
        self.assertFalse(bobj._validate_options())

        bobj.all_options = {'wrapper': 'dummy',
                            'wrapper_archive': 'myarchive.7z'}
        self.assertTrue(bobj._validate_options())

    @mock.patch('os.listdir')
    def test_get_wrapper_archive_name(self, os_listdir):
        os_listdir.return_value = 'no archive among other files'.split()
        bobj = base.ArchiveBase('Config.fs-uae', utils.CmdOption(), {})
        bobj.all_options = {'wrapper': 'dummy'}
        self.assertIsNone(bobj._get_wrapper_archive_name())

        os_listdir.return_value = 'no config.rar among other files'.split()
        bobj = base.ArchiveBase('Config.fs-uae', utils.CmdOption(), {})
        bobj.all_options = {'wrapper': 'dummy'}
        self.assertIsNone(bobj._get_wrapper_archive_name())

        os_listdir.return_value = 'file Config.TAR among other files'.split()
        bobj = base.ArchiveBase('Config.fs-uae', utils.CmdOption(), {})
        bobj.all_options = {'wrapper': 'dummy'}
        self.assertEqual(bobj._get_wrapper_archive_name(), 'Config.TAR')

        os_listdir.return_value = 'Config.lha FooBar_1.24b_20202.7z'.split()
        bobj = base.ArchiveBase('FooBar_1.24b_20202.fs-uae',
                                utils.CmdOption(), {})
        bobj.all_options = {'wrapper': 'dummy'}
        self.assertEqual(bobj._get_wrapper_archive_name(),
                         'FooBar_1.24b_20202.7z')