# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

# GNU General Public License for more details.

#

# You should have received a copy of the GNU General Public License

# along with this program.  If not, see <http://www.gnu.org/licenses/>.

import logging

from itertools import groupby

from sqlalchemy import or_

from sqlalchemy.orm import joinedload

from webhelpers.paginate import Page

from webhelpers.feedgenerator import Atom1Feed, Rss201rev2Feed

from webob.exc import HTTPBadRequest

from pylons import request, tmpl_context as c, response, url

from pylons.i18n.translation import _

import rhodecode.lib.helpers as h

from rhodecode.lib.auth import LoginRequired, NotAnonymous

from rhodecode.lib.base import BaseController, render

from rhodecode.model.db import UserLog, UserFollowing, Repository, User

from rhodecode.model.meta import Session

from sqlalchemy.sql.expression import func

from rhodecode.model.scm import ScmModel

from rhodecode.controllers.admin.admin import _journal_filter

log = logging.getLogger(__name__)

class JournalController(BaseController):

    def __before__(self):

        super(JournalController, self).__before__()

        self.language = 'en-us'

        self.ttl = "5"

        self.feed_nr = 20

    @LoginRequired()

    @NotAnonymous()

    def index(self):

        # Return a rendered template

        p = safe_int(request.params.get('page', 1), 1)

        c.user = User.get(self.rhodecode_user.user_id)

        c.following = self.sa.query(UserFollowing)\

            .filter(UserFollowing.user_id == self.rhodecode_user.user_id)\

            .options(joinedload(UserFollowing.follows_repository))\

            .all()

        journal = self._get_journal_data(c.following)

        def url_generator(**kw):

            return url.current(filter=c.search_term, **kw)

        c.journal_pager = Page(journal, page=p, items_per_page=20, url=url_generator)

        c.journal_day_aggreagate = self._get_daily_aggregate(c.journal_pager)

        c.journal_data = render('journal/journal_data.html')

        if request.environ.get('HTTP_X_PARTIAL_XHR'):

            return c.journal_data

        return render('journal/journal.html')

    @LoginRequired()

    @NotAnonymous()

    def index_my_repos(self):

        c.user = User.get(self.rhodecode_user.user_id)

        if request.environ.get('HTTP_X_PARTIAL_XHR'):

            all_repos = self.sa.query(Repository)\

                     .filter(Repository.user_id == c.user.user_id)\

                     .order_by(func.lower(Repository.repo_name)).all()

            c.user_repos = ScmModel().get_repos(all_repos)

            return render('journal/journal_page_repos.html')

        c.journal_day_aggreagate = self._get_daily_aggregate(c.journal_pager)

        c.journal_data = render('journal/journal_data.html')

        if request.environ.get('HTTP_X_PARTIAL_XHR'):

            return c.journal_data

        return render('journal/public_journal.html')

    def _atom_feed(self, repos, public=True):

        journal = self._get_journal_data(repos)

        if public:

            _link = url('public_journal_atom', qualified=True)

            _desc = '%s %s %s' % (c.rhodecode_name, _('public journal'),

                                  'atom feed')

        else:

            _link = url('journal_atom', qualified=True)

            _desc = '%s %s %s' % (c.rhodecode_name, _('journal'), 'atom feed')

        feed = Atom1Feed(title=_desc,

                         link=_link,

                         description=_desc,

                         language=self.language,

                         ttl=self.ttl)

        for entry in journal[:self.feed_nr]:

            action, action_extra, ico = h.action_parser(entry, feed=True)

            desc = action_extra()

            _url = None

            if entry.repository is not None:

                _url = url('changelog_home',

                           repo_name=entry.repository.repo_name,

                           qualified=True)

            feed.add_item(title=title,

                          pubdate=entry.action_date,

                          link=_url or url('', qualified=True),

                          description=desc)

        response.content_type = feed.mime_type

        return feed.writeString('utf-8')

    def _rss_feed(self, repos, public=True):

        journal = self._get_journal_data(repos)

        if public:

            _link = url('public_journal_atom', qualified=True)

            _desc = '%s %s %s' % (c.rhodecode_name, _('public journal'),

                                  'rss feed')

        else:

            _link = url('journal_atom', qualified=True)

            _desc = '%s %s %s' % (c.rhodecode_name, _('journal'), 'rss feed')

        feed = Rss201rev2Feed(title=_desc,

                         link=_link,

                         description=_desc,

                         language=self.language,

                         ttl=self.ttl)

        for entry in journal[:self.feed_nr]:

            action, action_extra, ico = h.action_parser(entry, feed=True)

            desc = action_extra()

            _url = None

            if entry.repository is not None:

                _url = url('changelog_home',

                           repo_name=entry.repository.repo_name,

                           qualified=True)

            feed.add_item(title=title,

                          pubdate=entry.action_date,

                          link=_url or url('', qualified=True),

                          description=desc)

        response.content_type = feed.mime_type

        return feed.writeString('utf-8')

    @LoginRequired(api_access=True)

    def public_journal_atom(self):

        """

        Produce an atom-1.0 feed via feedgenerator module

        """

        c.following = self.sa.query(UserFollowing)\

            .filter(UserFollowing.user_id == self.rhodecode_user.user_id)\

            .options(joinedload(UserFollowing.follows_repository))\

            .all()

        return self._atom_feed(c.following)

    @LoginRequired(api_access=True)

    def public_journal_rss(self):

        """

        Produce an rss2 feed via feedgenerator module

        """

        c.following = self.sa.query(UserFollowing)\

            .filter(UserFollowing.user_id == self.rhodecode_user.user_id)\

from rhodecode.tests import *

from rhodecode.model.db import UserFollowing, User, Repository

from rhodecode.lib.helpers import get_token

import datetime

class TestJournalController(TestController):

    def test_index(self):

        self.log_user()

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

        # Test response...

        assert """ <span id="follow_toggle_1" class="following" title="Stop following this repository""" in response.body, 'no info about stop follwoing repo id 1'

        assert """<div class="journal_day">%s</div>""" % datetime.date.today() in response.body, 'no info about action journal day'

    def test_stop_following_repository(self):

        session = self.log_user()

#        usr = self.Session.query(User).filter(User.username == 'test_admin').one()

#        repo = self.Session.query(Repository).filter(Repository.repo_name == HG_REPO).one()

#

#        followings = self.Session.query(UserFollowing)\

#            .filter(UserFollowing.user == usr)\

#            .filter(UserFollowing.follows_repository == repo).all()

#

#        assert len(followings) == 1, 'Not following any repository'

#

#        response = self.app.post(url(controller='journal',

#                                     action='toggle_following'),

#                                     {'auth_token':get_token(session),

#                                      'follows_repo_id':repo.repo_id})

    def test_start_following_repository(self):

        self.log_user()

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