# but WITHOUT ANY WARRANTY; without even the implied warranty of

# 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, write to the Free Software

# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,

# MA  02110-1301, USA.

import logging

from sqlalchemy import or_

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

from webhelpers.paginate import Page

from rhodecode.lib.auth import LoginRequired, NotAnonymous

from rhodecode.lib.base import BaseController, render

from rhodecode.lib.helpers import get_token

from rhodecode.model.db import UserLog, UserFollowing

from rhodecode.model.scm import ScmModel

from paste.httpexceptions import HTTPInternalServerError

from sqlalchemy.orm import joinedload

log = logging.getLogger(__name__)

class JournalController(BaseController):

    @LoginRequired()

    @NotAnonymous()

    def __before__(self):

        super(JournalController, self).__before__()

    def index(self):

        # Return a rendered template

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

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

            .options(joinedload(UserFollowing.follows_repository))\

            .all()

        repo_ids = [x.follows_repository.repo_id for x in c.following

                    if x.follows_repository is not None]

        user_ids = [x.follows_user.user_id for x in c.following

                    if x.follows_user is not None]

        journal = self.sa.query(UserLog)\

            .order_by(UserLog.action_date.desc())

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

        c.journal_pager = Page(journal, page=p, items_per_page=10)

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

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

        if request.params.get('partial'):

            return c.journal_data

        return render('journal/journal.html')

    def _get_daily_aggregate(self, journal):

        from itertools import groupby

        groups = []

        for k, g in groupby(journal, lambda x:x.action_as_day):

            groups.append((k, list(g),))      # Store group iterator as a list

        return groups

    def toggle_following(self):

        cur_token = request.POST.get('auth_token')

        token = get_token()

        if cur_token == token:

            scm_model = ScmModel()

            user_id = request.POST.get('follows_user_id')

            if user_id:

                try:

                    scm_model.toggle_following_user(user_id,

                                                    c.rhodecode_user.user_id)

                    return 'ok'

                except:

                    raise HTTPInternalServerError()

            repo_id = request.POST.get('follows_repo_id')

            if repo_id:

                try:

                    scm_model.toggle_following_repo(repo_id,

                                                    c.rhodecode_user.user_id)

                    return 'ok'

                except:

                    raise HTTPInternalServerError()

        log.debug('token mismatch %s vs %s', cur_token, token)

        raise HTTPInternalServerError()