import datetime

from rhodecode.tests import *

from rhodecode.model.gist import GistModel

from rhodecode.model.meta import Session

from rhodecode.model.db import User, Gist

def _create_gist(f_name, content='some gist', lifetime=-1,

                 description='gist-desc', gist_type='public',

                 owner=TEST_USER_ADMIN_LOGIN):

    gist_mapping = {

        f_name: {'content': content}

    }

    user = User.get_by_username(owner)

    gist = GistModel().create(description, owner=user,

                       gist_mapping=gist_mapping, gist_type=gist_type,

                       lifetime=lifetime)

    Session().commit()

    return gist

class TestGistsController(TestController):

    def tearDown(self):

        for g in Gist.get_all():

            GistModel().delete(g)

        Session().commit()

    def test_index(self):

        self.log_user()

        response = self.app.get(url('gists'))

        # Test response...

        response.mustcontain('There are no gists yet')

        response = self.app.get(url('gists'))

        # Test response...

        response.mustcontain('Expires: in 23 hours')  # we don't care about the end

        response.mustcontain('gist3-desc')

    def test_index_private_gists(self):

        self.log_user()

        gist = _create_gist('gist5', gist_type='private')

        response = self.app.get(url('gists', private=1))

        # Test response...

        #and privates

        response.mustcontain('gist:%s' % gist.gist_access_id)

    def test_create_missing_description(self):

        self.log_user()

        response = self.app.post(url('gists'),

                                 params={'lifetime': -1}, status=200)

        response.mustcontain('Missing value')

    def test_create(self):

        self.log_user()

        response = self.app.post(url('gists'),

                                 params={'lifetime': -1,

                                         'content': 'gist test',

                                         'filename': 'foo',

                                         'public': 'public'},

                                 status=302)

        response = response.follow()

        response.mustcontain('added file: foo')

        response.mustcontain('gist test')

        response.mustcontain('<div class="ui-btn green badge">Public gist</div>')

    def test_create_with_path_with_dirs(self):

        self.log_user()

        response = self.app.post(url('gists'),

                                 params={'lifetime': -1,

                                         'content': 'gist test',

                                         'filename': '/home/foo',

                                         'public': 'public'},

                                 status=200)

        response.mustcontain('Filename cannot be inside a directory')

    def test_access_expired_gist(self):

        self.log_user()

        gist = _create_gist('never-see-me')

        gist.gist_expires = 0  # 1970

        Session().add(gist)

        Session().commit()

        response = self.app.get(url('gist', id=gist.gist_access_id), status=404)

    def test_create_private(self):

        self.log_user()

        response = self.app.post(url('gists'),

                                 params={'lifetime': -1,

                                         'content': 'private gist test',

                                         'filename': 'private-foo',

                                         'private': 'private'},

                                 status=302)

        response = response.follow()

        response.mustcontain('added file: private-foo<')

        response.mustcontain('private gist test')

        response.mustcontain('<div class="ui-btn yellow badge">Private gist</div>')

    def test_create_with_description(self):

        self.log_user()

        response = self.app.post(url('gists'),

                                 params={'lifetime': -1,

                                         'content': 'gist test',

                                         'filename': 'foo-desc',

                                         'description': 'gist-desc',

                                         'public': 'public'},

                                 status=302)

        response = response.follow()

        response.mustcontain('added file: foo-desc')

        response.mustcontain('gist test')

        response.mustcontain('gist-desc')

        response.mustcontain('<div class="ui-btn green badge">Public gist</div>')

    def test_new(self):

        self.log_user()

        response = self.app.get(url('new_gist'))

    def test_update(self):

        self.skipTest('not implemented')

        response = self.app.put(url('gist', id=1))

    def test_delete(self):

        self.log_user()

        gist = _create_gist('delete-me')

        response = self.app.delete(url('gist', id=gist.gist_id))

        self.checkSessionFlash(response, 'Deleted gist %s' % gist.gist_id)

    def test_delete_normal_user_his_gist(self):

        self.log_user(TEST_USER_REGULAR_LOGIN, TEST_USER_REGULAR_PASS)

        gist = _create_gist('delete-me', owner=TEST_USER_REGULAR_LOGIN)

        response = self.app.delete(url('gist', id=gist.gist_id))

        self.checkSessionFlash(response, 'Deleted gist %s' % gist.gist_id)

    def test_delete_normal_user_not_his_own_gist(self):

        self.log_user(TEST_USER_REGULAR_LOGIN, TEST_USER_REGULAR_PASS)

        gist = _create_gist('delete-me')

        response = self.app.delete(url('gist', id=gist.gist_id), status=403)

    def test_show(self):

        gist = _create_gist('gist-show-me')

        response = self.app.get(url('gist', id=gist.gist_access_id))

        response.mustcontain('added file: gist-show-me<')

        response.mustcontain('test_admin (RhodeCode Admin) - created')

        response.mustcontain('gist-desc')

        response.mustcontain('<div class="ui-btn green badge">Public gist</div>')

    def test_edit(self):

        self.skipTest('not implemented')

        response = self.app.get(url('edit_gist', id=1))

<option value="41f43fc74b8b285984554532eb105ac3be5c434f">r93:41f43fc74b8b (default)</option>

<option value="cc66b61b8455b264a7a8a2d8ddc80fcfc58c221e">r92:cc66b61b8455 (default)</option>

<option value="73ab5b616b3271b0518682fb4988ce421de8099f">r91:73ab5b616b32 (default)</option>

<option value="e0da75f308c0f18f98e9ce6257626009fdda2b39">r82:e0da75f308c0 (default)</option>

<option value="fb2e41e0f0810be4d7103bc2a4c7be16ee3ec611">r81:fb2e41e0f081 (default)</option>

<option value="602ae2f5e7ade70b3b66a58cdd9e3e613dc8a028">r76:602ae2f5e7ad (default)</option>

<option value="a066b25d5df7016b45a41b7e2a78c33b57adc235">r73:a066b25d5df7 (default)</option>

<option value="637a933c905958ce5151f154147c25c1c7b68832">r61:637a933c9059 (web)</option>

<option value="0c21004effeb8ce2d2d5b4a8baf6afa8394b6fbc">r60:0c21004effeb (web)</option>

<option value="a1f39c56d3f1d52d5fb5920370a2a2716cd9a444">r59:a1f39c56d3f1 (web)</option>

<option value="97d32df05c715a3bbf936bf3cc4e32fb77fe1a7f">r58:97d32df05c71 (web)</option>

<option value="08eaf14517718dccea4b67755a93368341aca919">r57:08eaf1451771 (web)</option>

<option value="22f71ad265265a53238359c883aa976e725aa07d">r56:22f71ad26526 (web)</option>

<option value="97501f02b7b4330924b647755663a2d90a5e638d">r49:97501f02b7b4 (web)</option>

<option value="86ede6754f2b27309452bb11f997386ae01d0e5a">r47:86ede6754f2b (web)</option>

<option value="014c40c0203c423dc19ecf94644f7cac9d4cdce0">r45:014c40c0203c (web)</option>

<option value="ee87846a61c12153b51543bf860e1026c6d3dcba">r30:ee87846a61c1 (default)</option>

<option value="9bb326a04ae5d98d437dece54be04f830cf1edd9">r26:9bb326a04ae5 (default)</option>

<option value="536c1a19428381cfea92ac44985304f6a8049569">r24:536c1a194283 (default)</option>

<option value="dc5d2c0661b61928834a785d3e64a3f80d3aad9c">r8:dc5d2c0661b6 (default)</option>

<option value="3803844fdbd3b711175fc3da9bdacfcd6d29a6fb">r7:3803844fdbd3 (default)</option>

</optgroup>

<optgroup label="Branches">

<option value="96507bd11ecc815ebc6270fdf6db110928c09c1e">default</option>

<option value="4f7e2131323e0749a740c0a56ab68ae9269c562a">stable</option>

</optgroup>

<optgroup label="Tags">

<option value="2c96c02def9a7c997f33047761a53943e6254396">v0.2.0</option>

<option value="8680b1d1cee3aa3c1ab3734b76ee164bbedbc5c9">v0.1.9</option>

<option value="ecb25ba9c96faf1e65a0bc3fd914918420a2f116">v0.1.8</option>

<option value="f67633a2894edaf28513706d558205fa93df9209">v0.1.7</option>

<option value="02b38c0eb6f982174750c0e309ff9faddc0c7e12">v0.1.6</option>

<option value="a6664e18181c6fc81b751a8d01474e7e1a3fe7fc">v0.1.5</option>

<option value="fd4bdb5e9b2a29b4393a4ac6caef48c17ee1a200">v0.1.4</option>

<option value="17544fbfcd33ffb439e2b728b5d526b1ef30bfcf">v0.1.3</option>

<option value="a7e60bff65d57ac3a1a1ce3b12a70f8a9e8a7720">v0.1.2</option>

<option value="fef5bfe1dc17611d5fb59a7f6f95c55c3606f933">v0.1.11</option>

<option value="92831aebf2f8dd4879e897024b89d09af214df1c">v0.1.10</option>

<option value="eb3a60fc964309c1a318b8dfe26aa2d1586c85ae">v0.1.1</option>

<option value="96507bd11ecc815ebc6270fdf6db110928c09c1e">tip</option>

</optgroup>""")

    def test_file_annotation_git(self):

        self.log_user()

        response = self.app.get(url(controller='files', action='index',

                                    repo_name=GIT_REPO,

                                    revision='master',

                                    f_path='vcs/nodes.py',

                                    annotate=True))

    def test_archival(self):

        self.log_user()

        _set_downloads(HG_REPO, set_to=True)

        for arch_ext, info in ARCHIVE_SPECS.items():

            short = '27cd5cce30c9%s' % arch_ext

            fname = '27cd5cce30c96924232dffcd24178a07ffeb5dfc%s' % arch_ext

            filename = '%s-%s' % (HG_REPO, short)

            response = self.app.get(url(controller='files',

                                        action='archivefile',

                                        repo_name=HG_REPO,

                                        fname=fname))

            self.assertEqual(response.status, '200 OK')

            heads = [

                ('Pragma', 'no-cache'),

                ('Cache-Control', 'no-cache'),

                ('Content-Disposition', 'attachment; filename=%s' % filename),

                ('Content-Type', '%s; charset=utf-8' % info[0]),

            ]

            self.assertEqual(response.response._headers.items(), heads)

    def test_archival_wrong_ext(self):

        self.log_user()

        _set_downloads(HG_REPO, set_to=True)

        for arch_ext in ['tar', 'rar', 'x', '..ax', '.zipz']:

            fname = '27cd5cce30c96924232dffcd24178a07ffeb5dfc%s' % arch_ext

            response = self.app.get(url(controller='files',

                                        action='archivefile',

                                        repo_name=HG_REPO,

                                        fname=fname))

            response.mustcontain('Unknown archive type')

    def test_archival_wrong_revision(self):

        self.log_user()

        _set_downloads(HG_REPO, set_to=True)

        for rev in ['00x000000', 'tar', 'wrong', '@##$@$42413232', '232dffcd']:

            fname = '%s.zip' % rev

            response = self.app.get(url(controller='files',

                                        action='archivefile',

                                        repo_name=HG_REPO,

                                        fname=fname))

            response.mustcontain('Unknown revision')

    #==========================================================================

    # RAW FILE

    #==========================================================================

    def test_raw_file_ok(self):

        self.log_user()

        response = self.app.get(url(controller='files', action='rawfile',

                                    repo_name=HG_REPO,

                                    revision='27cd5cce30c96924232dffcd24178a07ffeb5dfc',

                                    f_path='vcs/nodes.py'))

        self.assertEqual(response.content_disposition, "attachment; filename=nodes.py")

        self.assertEqual(response.content_type, "text/x-python")

    def test_raw_file_wrong_cs(self):

        self.log_user()

        rev = u'ERRORce30c96924232dffcd24178a07ffeb5dfc'

        f_path = 'vcs/nodes.py'

        response = self.app.get(url(controller='files', action='rawfile',

                                    repo_name=HG_REPO,

                                    revision=rev,

                                    f_path=f_path), status=404)

        msg = """Revision %s does not exist for this repository""" % (rev)

        response.mustcontain(msg)

    def test_raw_file_wrong_f_path(self):

        self.log_user()

        rev = '27cd5cce30c96924232dffcd24178a07ffeb5dfc'

        f_path = 'vcs/ERRORnodes.py'

        response = self.app.get(url(controller='files', action='rawfile',

                                    repo_name=HG_REPO,

                                    revision=rev,

                                    f_path=f_path), status=404)

        msg = "There is no file nor directory at the given path: '%s' at revision %s" % (f_path, rev[:12])

        response.mustcontain(msg)

    #==========================================================================

    # RAW RESPONSE - PLAIN

    #==========================================================================

    def test_raw_ok(self):

        self.log_user()

        response = self.app.get(url(controller='files', action='raw',

                                    repo_name=HG_REPO,

                                    revision='27cd5cce30c96924232dffcd24178a07ffeb5dfc',

                                    f_path='vcs/nodes.py'))

        self.assertEqual(response.content_type, "text/plain")

    def test_raw_wrong_cs(self):

        self.log_user()

        rev = u'ERRORcce30c96924232dffcd24178a07ffeb5dfc'

        f_path = 'vcs/nodes.py'

        response = self.app.get(url(controller='files', action='raw',

                                    repo_name=HG_REPO,

                                    revision=rev,

                                    f_path=f_path), status=404)

        msg = """Revision %s does not exist for this repository""" % (rev)

        response.mustcontain(msg)

    def test_raw_wrong_f_path(self):

        self.log_user()

        rev = '27cd5cce30c96924232dffcd24178a07ffeb5dfc'

        f_path = 'vcs/ERRORnodes.py'

        response = self.app.get(url(controller='files', action='raw',

                                    repo_name=HG_REPO,

                                    revision=rev,

                                    f_path=f_path), status=404)

        msg = "There is no file nor directory at the given path: '%s' at revision %s" % (f_path, rev[:12])

        response.mustcontain(msg)

    def test_ajaxed_files_list(self):

        self.log_user()

        rev = '27cd5cce30c96924232dffcd24178a07ffeb5dfc'

        response = self.app.get(

            url('files_nodelist_home', repo_name=HG_REPO, f_path='/',

                revision=rev),

            extra_environ={'HTTP_X_PARTIAL_XHR': '1'},

        )

        response.mustcontain("vcs/web/simplevcs/views/repository.py")

    #HG - ADD FILE

    def test_add_file_view_hg(self):

        self.log_user()

        response = self.app.get(url('files_add_home',

                                      repo_name=HG_REPO,

                                      revision='tip', f_path='/'))

    def test_add_file_into_hg_missing_content(self):

        self.log_user()

        response = self.app.post(url('files_add_home',

                                      repo_name=HG_REPO,

                                      revision='tip', f_path='/'),

                                 params={

                                 },

                                 status=302)

        self.checkSessionFlash(response, 'No content')

    def test_add_file_into_hg_missing_filename(self):

        self.log_user()

        response = self.app.post(url('files_add_home',

                                      repo_name=HG_REPO,

                                      revision='tip', f_path='/'),

                                 params={

                                    'content': "foo"

                                 },

                                 status=302)

        self.checkSessionFlash(response, 'No filename')

    @parameterized.expand([

        ('/abs', 'foo'),

        ('../rel', 'foo'),

        ('file/../foo', 'foo'),

    ])

    def test_add_file_into_hg_bad_filenames(self, location, filename):

        self.log_user()

        response = self.app.post(url('files_add_home',

                                      repo_name=HG_REPO,

                                      revision='tip', f_path='/'),

                                 params={

                                    'content': "foo",

                                    'filename': filename,

                                    'location': location

                                 },

                                 status=302)

        self.checkSessionFlash(response, 'Location must be relative path and must not contain .. in path')

    @parameterized.expand([

        (1, '', 'foo.txt'),

        (2, 'dir', 'foo.rst'),

        (3, 'rel/dir', 'foo.bar'),

    ])

    def test_add_file_into_hg(self, cnt, location, filename):

        self.log_user()

        repo = fixture.create_repo('commit-test-%s' % cnt, repo_type='hg')

        response = self.app.post(url('files_add_home',

                                      repo_name=repo.repo_name,

                                      revision='tip', f_path='/'),

                                 params={

                                    'content': "foo",

                                    'filename': filename,

                                    'location': location

                                 },

                                 status=302)

        try:

            self.checkSessionFlash(response, 'Successfully committed to %s'

                                   % os.path.join(location, filename))

        finally:

            fixture.destroy_repo(repo.repo_name)

    ##GIT - ADD FILE

    def test_add_file_view_git(self):

        self.log_user()

        response = self.app.get(url('files_add_home',

                                      repo_name=GIT_REPO,

                                      revision='tip', f_path='/'))

    def test_add_file_into_git_missing_content(self):

        self.log_user()

        response = self.app.post(url('files_add_home',

                                      repo_name=GIT_REPO,

                                      revision='tip', f_path='/'),

                                 params={

                                    ''

                                 },

                                 status=302)

        self.checkSessionFlash(response, 'No content')

    def test_add_file_into_git_missing_filename(self):

        self.log_user()

        response = self.app.post(url('files_add_home',

                                      repo_name=GIT_REPO,

                                      revision='tip', f_path='/'),

                                 params={

                                    'content': "foo"

                                 },

                                 status=302)

        self.checkSessionFlash(response, 'No filename')

    @parameterized.expand([

        ('/abs', 'foo'),

        ('../rel', 'foo'),

        ('file/../foo', 'foo'),

    ])

    def test_add_file_into_git_bad_filenames(self, location, filename):

        self.log_user()

        response = self.app.post(url('files_add_home',

                                      repo_name=GIT_REPO,

                                      revision='tip', f_path='/'),

                                 params={

                                    'content': "foo",

                                    'filename': filename,

                                    'location': location

                                 },

                                 status=302)

        self.checkSessionFlash(response, 'Location must be relative path and must not contain .. in path')

    @parameterized.expand([

        (1, '', 'foo.txt'),

        (2, 'dir', 'foo.rst'),

        (3, 'rel/dir', 'foo.bar'),

    ])

    def test_add_file_into_git(self, cnt, location, filename):

        self.log_user()

        repo = fixture.create_repo('commit-test-%s' % cnt, repo_type='git')

        response = self.app.post(url('files_add_home',

                                      repo_name=repo.repo_name,

                                      revision='tip', f_path='/'),

                                 params={

                                    'content': "foo",

                                    'filename': filename,

                                    'location': location

                                 },

                                 status=302)

        try:

            self.checkSessionFlash(response, 'Successfully committed to %s'

                                   % os.path.join(location, filename))

        finally:

            fixture.destroy_repo(repo.repo_name)

    #HG - EDIT

    def test_edit_file_view_hg(self):

        self.log_user()

        response = self.app.get(url('files_edit_home',

                                      repo_name=HG_REPO,

                                      revision='tip', f_path='vcs/nodes.py'))

    def test_edit_file_view_not_on_branch_hg(self):

        self.log_user()

        repo = fixture.create_repo('test-edit-repo', repo_type='hg')

        ## add file

        location = 'vcs'

        filename = 'nodes.py'

        response = self.app.post(url('files_add_home',

                                      repo_name=repo.repo_name,

                                      revision='tip', f_path='/'),

                                 params={

                                    'content': "def py():\n print 'hello'\n",

                                    'filename': filename,

                                    'location': location

                                 },

                                 status=302)

        response.follow()

        try:

            self.checkSessionFlash(response, 'Successfully committed to %s'

                                   % os.path.join(location, filename))

            response = self.app.get(url('files_edit_home',

                                          repo_name=repo.repo_name,

                                          revision='tip', f_path='vcs/nodes.py'),

                                    status=302)

            self.checkSessionFlash(response,

                'You can only edit files with revision being a valid branch')

        finally:

            fixture.destroy_repo(repo.repo_name)

    def test_edit_file_view_commit_changes_hg(self):

        self.log_user()

        repo = fixture.create_repo('test-edit-repo', repo_type='hg')

        ## add file

        location = 'vcs'

        filename = 'nodes.py'

        response = self.app.post(url('files_add_home',

                                      repo_name=repo.repo_name,

                                      revision='tip',

                                      f_path='/'),

                                 params={

                                    'content': "def py():\n print 'hello'\n",

                                    'filename': filename,

                                    'location': location

                                 },

                                 status=302)

        response.follow()

        try:

            self.checkSessionFlash(response, 'Successfully committed to %s'

                                   % os.path.join(location, filename))

            response = self.app.post(url('files_edit_home',

                                          repo_name=repo.repo_name,

                                          revision=repo.scm_instance.DEFAULT_BRANCH_NAME,