# Copyright (C) 2016-2024 RhodeCode GmbH
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License, version 3
# (only), as published by the Free Software Foundation.
#
# This program is distributed in the hope that it will be useful,
# 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 Affero General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.
#
# This program is dual-licensed. If you wish to learn more about the
# RhodeCode Enterprise Edition, including its added features, Support services,
# and proprietary license terms, please see https://rhodecode.com/licenses/

import time
import logging
import datetime
import string

import formencode
import formencode.htmlfill
import peppercorn
from pyramid.httpexceptions import HTTPFound, HTTPNotFound

from rhodecode.apps._base import BaseAppView, DataGridAppView
from rhodecode import forms
from rhodecode.lib import helpers as h
from rhodecode.lib import audit_logger
from rhodecode.lib import ext_json
from rhodecode.lib.auth import (
    LoginRequired,
    NotAnonymous,
    CSRFRequired,
    HasRepoPermissionAny,
    HasRepoGroupPermissionAny,
    AuthUser,
)
from rhodecode.lib.channelstream import channelstream_request, ChannelstreamException
from rhodecode.lib.hash_utils import md5_safe
from rhodecode.lib.utils2 import safe_int, md5, str2bool
from rhodecode.model.auth_token import AuthTokenModel
from rhodecode.model.comment import CommentsModel
from rhodecode.model.db import (
    IntegrityError,
    or_,
    in_filter_generator,
    select,
    Repository,
    UserEmailMap,
    UserApiKeys,
    UserFollowing,
    PullRequest,
    UserBookmark,
    RepoGroup,
    ChangesetStatus,
)
from rhodecode.model.forms import TOTPForm
from rhodecode.model.meta import Session
from rhodecode.model.pull_request import PullRequestModel
from rhodecode.model.user import UserModel
from rhodecode.model.user_group import UserGroupModel
from rhodecode.model.validation_schema.schemas import user_schema

log = logging.getLogger(__name__)


class MyAccountView(BaseAppView, DataGridAppView):
    ALLOW_SCOPED_TOKENS = False
    """
    This view has alternative version inside EE, if modified please take a look
    in there as well.
    """

    def load_default_context(self):
        c = self._get_local_tmpl_context()
        c.user = c.auth_user.get_instance()
        c.allow_scoped_tokens = self.ALLOW_SCOPED_TOKENS
        return c

    @LoginRequired()
    @NotAnonymous()
    def my_account_profile(self):
        c = self.load_default_context()
        c.active = "profile"
        c.extern_type = c.user.extern_type
        return self._get_template_context(c)

    @LoginRequired()
    @NotAnonymous()
    def my_account_edit(self):
        c = self.load_default_context()
        c.active = "profile_edit"
        c.extern_type = c.user.extern_type
        c.extern_name = c.user.extern_name

        schema = user_schema.UserProfileSchema().bind(username=c.user.username, user_emails=c.user.emails)
        appstruct = {
            "username": c.user.username,
            "email": c.user.email,
            "firstname": c.user.firstname,
            "lastname": c.user.lastname,
            "description": c.user.description,
        }
        c.form = forms.RcForm(
            schema,
            appstruct=appstruct,
            action=h.route_path("my_account_update"),
            buttons=(forms.buttons.save, forms.buttons.reset),
        )

        return self._get_template_context(c)

    @LoginRequired()
    @NotAnonymous()
    @CSRFRequired()
    def my_account_update(self):
        _ = self.request.translate
        c = self.load_default_context()
        c.active = "profile_edit"
        c.perm_user = c.auth_user
        c.extern_type = c.user.extern_type
        c.extern_name = c.user.extern_name

        schema = user_schema.UserProfileSchema().bind(username=c.user.username, user_emails=c.user.emails)
        form = forms.RcForm(schema, buttons=(forms.buttons.save, forms.buttons.reset))

        controls = list(self.request.POST.items())
        try:
            valid_data = form.validate(controls)
            skip_attrs = ["admin", "active", "extern_type", "extern_name", "new_password", "password_confirmation"]
            if c.extern_type != "rhodecode":
                # forbid updating username for external accounts
                skip_attrs.append("username")
            old_email = c.user.email
            UserModel().update_user(self._rhodecode_user.user_id, skip_attrs=skip_attrs, **valid_data)
            if old_email != valid_data["email"]:
                old = (
                    UserEmailMap.query()
                    .filter(UserEmailMap.user == c.user)
                    .filter(UserEmailMap.email == valid_data["email"])
                    .first()
                )
                old.email = old_email
            h.flash(_("Your account was updated successfully"), category="success")
            Session().commit()
        except forms.ValidationFailure as e:
            c.form = e
            return self._get_template_context(c)

        except Exception:
            log.exception("Exception updating user")
            h.flash(_("Error occurred during update of user"), category="error")
        raise HTTPFound(h.route_path("my_account_profile"))

    @LoginRequired()
    @NotAnonymous()
    def my_account_password(self):
        c = self.load_default_context()
        c.active = "password"
        c.extern_type = c.user.extern_type

        schema = user_schema.ChangePasswordSchema().bind(username=c.user.username)

        form = forms.Form(
            schema, action=h.route_path("my_account_password_update"), buttons=(forms.buttons.save, forms.buttons.reset)
        )

        c.form = form
        return self._get_template_context(c)

    @LoginRequired()
    @NotAnonymous()
    @CSRFRequired()
    def my_account_password_update(self):
        _ = self.request.translate
        c = self.load_default_context()
        c.active = "password"
        c.extern_type = c.user.extern_type

        schema = user_schema.ChangePasswordSchema().bind(username=c.user.username)

        form = forms.Form(schema, buttons=(forms.buttons.save, forms.buttons.reset))

        if c.extern_type != "rhodecode":
            raise HTTPFound(self.request.route_path("my_account_password"))

        controls = list(self.request.POST.items())
        try:
            valid_data = form.validate(controls)
            UserModel().update_user(c.user.user_id, **valid_data)
            c.user.update_userdata(force_password_change=False)
            Session().commit()
        except forms.ValidationFailure as e:
            c.form = e
            return self._get_template_context(c)

        except Exception:
            log.exception("Exception updating password")
            h.flash(_("Error occurred during update of user password"), category="error")
        else:
            instance = c.auth_user.get_instance()
            self.session.setdefault("rhodecode_user", {}).update({"password": md5_safe(instance.password)})
            self.session.save()
            h.flash(_("Successfully updated password"), category="success")

        raise HTTPFound(self.request.route_path("my_account_password"))

    @LoginRequired()
    @NotAnonymous()
    def my_account_2fa(self):
        _ = self.request.translate
        c = self.load_default_context()
        c.active = "2fa"
        user_instance = c.auth_user.get_instance()
        locked_by_admin = user_instance.has_forced_2fa
        c.state_of_2fa = user_instance.has_enabled_2fa
        c.user_seen_2fa_recovery_codes = user_instance.has_seen_2fa_codes
        c.locked_2fa = str2bool(locked_by_admin)
        return self._get_template_context(c)

    @LoginRequired()
    @NotAnonymous()
    @CSRFRequired()
    def my_account_2fa_update(self):
        _ = self.request.translate
        c = self.load_default_context()
        c.active = "2fa"
        user_instance = c.auth_user.get_instance()

        state = str2bool(self.request.POST.get("2fa_status"))
        user_instance.has_enabled_2fa = state
        user_instance.update_userdata(update_2fa=time.time())
        Session().commit()
        if state:
            h.flash(_("2FA has been successfully enabled"), category="success")
        else:
            h.flash(_("2FA has been successfully disabled"), category="success")
        raise HTTPFound(self.request.route_path("my_account_configure_2fa"))

    @LoginRequired()
    @NotAnonymous()
    @CSRFRequired()
    def my_account_2fa_show_recovery_codes(self):
        c = self.load_default_context()
        user_instance = c.auth_user.get_instance()
        user_instance.has_seen_2fa_codes = True
        Session().commit()
        return {"recovery_codes": user_instance.get_2fa_recovery_codes()}

    @LoginRequired()
    @NotAnonymous()
    @CSRFRequired()
    def my_account_2fa_regenerate_recovery_codes(self):
        _ = self.request.translate
        c = self.load_default_context()
        user_instance = c.auth_user.get_instance()

        totp_form = TOTPForm(_, user_instance, allow_recovery_code_use=True)()

        post_items = dict(self.request.POST)
        # NOTE: inject secret, as it's a post configured saved item.
        post_items["secret_totp"] = user_instance.secret_2fa
        try:
            totp_form.to_python(post_items)
            user_instance.regenerate_2fa_recovery_codes()
            Session().commit()
        except formencode.Invalid as errors:
            h.flash(_("Failed to generate new recovery codes: {}").format(errors), category="error")
            raise HTTPFound(self.request.route_path("my_account_configure_2fa"))
        except Exception as e:
            h.flash(_("Failed to generate new recovery codes: {}").format(e), category="error")
            raise HTTPFound(self.request.route_path("my_account_configure_2fa"))

        raise HTTPFound(self.request.route_path("my_account_configure_2fa", _query={"show-recovery-codes": 1}))

    @LoginRequired()
    @NotAnonymous()
    def my_account_auth_tokens(self):
        _ = self.request.translate

        c = self.load_default_context()
        c.active = "auth_tokens"
        c.lifetime_values = AuthTokenModel.get_lifetime_values(translator=_)
        c.role_values = [(x, AuthTokenModel.cls._get_role_name(x)) for x in AuthTokenModel.cls.ROLES]
        c.role_options = [(c.role_values, _("Role"))]
        c.user_auth_tokens = AuthTokenModel().get_auth_tokens(c.user.user_id, show_expired=True)
        c.role_vcs = AuthTokenModel.cls.ROLE_VCS
        return self._get_template_context(c)

    @LoginRequired()
    @NotAnonymous()
    @CSRFRequired()
    def my_account_auth_tokens_view(self):
        _ = self.request.translate
        c = self.load_default_context()

        auth_token_id = self.request.POST.get("auth_token_id")

        if auth_token_id:
            token = UserApiKeys.get_or_404(auth_token_id)
            if token.user.user_id != c.user.user_id:
                raise HTTPNotFound()

            return {"auth_token": token.api_key}

    def maybe_attach_token_scope(self, token):
        # implemented in EE edition
        pass

    @LoginRequired()
    @NotAnonymous()
    @CSRFRequired()
    def my_account_auth_tokens_add(self):
        _ = self.request.translate
        c = self.load_default_context()

        lifetime = safe_int(self.request.POST.get("lifetime"), -1)
        description = self.request.POST.get("description")
        role = self.request.POST.get("role")

        token = UserModel().add_auth_token(
            user=c.user.user_id,
            lifetime_minutes=lifetime,
            role=role,
            description=description,
            scope_callback=self.maybe_attach_token_scope,
        )
        token_data = token.get_api_data()

        audit_logger.store_web(
            "user.edit.token.add",
            action_data={"data": {"token": token_data, "user": "self"}},
            user=self._rhodecode_user,
        )
        Session().commit()

        h.flash(_("Auth token successfully created"), category="success")
        return HTTPFound(h.route_path("my_account_auth_tokens"))

    @LoginRequired()
    @NotAnonymous()
    @CSRFRequired()
    def my_account_auth_tokens_delete(self):
        _ = self.request.translate
        c = self.load_default_context()

        del_auth_token = self.request.POST.get("del_auth_token")

        if del_auth_token:
            token = UserApiKeys.get_or_404(del_auth_token)
            token_data = token.get_api_data()

            AuthTokenModel().delete(del_auth_token, c.user.user_id)
            audit_logger.store_web(
                "user.edit.token.delete",
                action_data={"data": {"token": token_data, "user": "self"}},
                user=self._rhodecode_user,
            )
            Session().commit()
            h.flash(_("Auth token successfully deleted"), category="success")

        return HTTPFound(h.route_path("my_account_auth_tokens"))

    @LoginRequired()
    @NotAnonymous()
    def my_account_emails(self):
        _ = self.request.translate

        c = self.load_default_context()
        c.active = "emails"

        c.user_email_map = UserEmailMap.query().filter(UserEmailMap.user == c.user).all()

        schema = user_schema.AddEmailSchema().bind(username=c.user.username, user_emails=c.user.emails)

        form = forms.RcForm(
            schema, action=h.route_path("my_account_emails_add"), buttons=(forms.buttons.save, forms.buttons.reset)
        )

        c.form = form
        return self._get_template_context(c)

    @LoginRequired()
    @NotAnonymous()
    @CSRFRequired()
    def my_account_emails_add(self):
        _ = self.request.translate
        c = self.load_default_context()
        c.active = "emails"

        schema = user_schema.AddEmailSchema().bind(username=c.user.username, user_emails=c.user.emails)

        form = forms.RcForm(
            schema, action=h.route_path("my_account_emails_add"), buttons=(forms.buttons.save, forms.buttons.reset)
        )

        controls = list(self.request.POST.items())
        try:
            valid_data = form.validate(controls)
            UserModel().add_extra_email(c.user.user_id, valid_data["email"])
            audit_logger.store_web(
                "user.edit.email.add",
                action_data={"data": {"email": valid_data["email"], "user": "self"}},
                user=self._rhodecode_user,
            )
            Session().commit()
        except formencode.Invalid as error:
            h.flash(h.escape(error.error_dict["email"]), category="error")
        except forms.ValidationFailure as e:
            c.user_email_map = UserEmailMap.query().filter(UserEmailMap.user == c.user).all()
            c.form = e
            return self._get_template_context(c)
        except Exception:
            log.exception("Exception adding email")
            h.flash(_("Error occurred during adding email"), category="error")
        else:
            h.flash(_("Successfully added email"), category="success")

        raise HTTPFound(self.request.route_path("my_account_emails"))

    @LoginRequired()
    @NotAnonymous()
    @CSRFRequired()
    def my_account_emails_delete(self):
        _ = self.request.translate
        c = self.load_default_context()

        del_email_id = self.request.POST.get("del_email_id")
        if del_email_id:
            email = UserEmailMap.get_or_404(del_email_id).email
            UserModel().delete_extra_email(c.user.user_id, del_email_id)
            audit_logger.store_web(
                "user.edit.email.delete",
                action_data={"data": {"email": email, "user": "self"}},
                user=self._rhodecode_user,
            )
            Session().commit()
            h.flash(_("Email successfully deleted"), category="success")
        return HTTPFound(h.route_path("my_account_emails"))

    @LoginRequired()
    @NotAnonymous()
    @CSRFRequired()
    def my_account_notifications_test_channelstream(self):
        message = "Test message sent via Channelstream by user: {}, on {}".format(
            self._rhodecode_user.username, datetime.datetime.now()
        )
        payload = {
            # 'channel': 'broadcast',
            "type": "message",
            "timestamp": datetime.datetime.utcnow(),
            "user": "system",
            "pm_users": [self._rhodecode_user.username],
            "message": {"message": message, "level": "info", "topic": "/notifications"},
        }

        registry = self.request.registry
        rhodecode_plugins = getattr(registry, "rhodecode_plugins", {})
        channelstream_config = rhodecode_plugins.get("channelstream", {})

        try:
            channelstream_request(channelstream_config, [payload], "/message")
        except ChannelstreamException as e:
            log.exception("Failed to send channelstream data")
            return {"response": f"ERROR: {e.__class__.__name__}"}
        return {"response": "Channelstream data sent. You should see a new live message now."}

    def _load_my_repos_data(self, watched=False):
        allowed_ids = [-1] + self._rhodecode_user.repo_acl_ids_from_stack(AuthUser.repo_read_perms)

        if watched:
            # repos user watch
            repo_list = (
                Session()
                .query(Repository)
                .join((UserFollowing, UserFollowing.follows_repo_id == Repository.repo_id))
                .filter(UserFollowing.user_id == self._rhodecode_user.user_id)
                .filter(
                    or_(
                        # generate multiple IN to fix limitation problems
                        *in_filter_generator(Repository.repo_id, allowed_ids)
                    )
                )
                .order_by(Repository.repo_name)
                .all()
            )

        else:
            # repos user is owner of
            repo_list = (
                Session()
                .query(Repository)
                .filter(Repository.user_id == self._rhodecode_user.user_id)
                .filter(
                    or_(
                        # generate multiple IN to fix limitation problems
                        *in_filter_generator(Repository.repo_id, allowed_ids)
                    )
                )
                .order_by(Repository.repo_name)
                .all()
            )

        _render = self.request.get_partial_renderer("rhodecode:templates/data_table/_dt_elements.mako")

        def repo_lnk(name, rtype, rstate, private, archived, fork_of):
            return _render("repo_name", name, rtype, rstate, private, archived, fork_of, short_name=False, admin=False)

        repos_data = []
        for repo in repo_list:
            row = {
                "name": repo_lnk(
                    repo.repo_name, repo.repo_type, repo.repo_state, repo.private, repo.archived, repo.fork
                ),
                "name_raw": repo.repo_name.lower(),
            }

            repos_data.append(row)

        # json used to render the grid
        return ext_json.str_json(repos_data)

    @LoginRequired()
    @NotAnonymous()
    def my_account_repos(self):
        c = self.load_default_context()
        c.active = "repos"

        # json used to render the grid
        c.data = self._load_my_repos_data()
        return self._get_template_context(c)

    @LoginRequired()
    @NotAnonymous()
    def my_account_watched(self):
        c = self.load_default_context()
        c.active = "watched"

        # json used to render the grid
        c.data = self._load_my_repos_data(watched=True)
        return self._get_template_context(c)

    @LoginRequired()
    @NotAnonymous()
    def my_account_bookmarks(self):
        c = self.load_default_context()
        c.active = "bookmarks"

        user_bookmarks = (
            select(UserBookmark, Repository, RepoGroup)
            .where(UserBookmark.user_id == self._rhodecode_user.user_id)
            .outerjoin(Repository, Repository.repo_id == UserBookmark.bookmark_repo_id)
            .outerjoin(RepoGroup, RepoGroup.group_id == UserBookmark.bookmark_repo_group_id)
            .order_by(UserBookmark.position.asc())
        )

        c.user_bookmark_items = Session().execute(user_bookmarks).all()
        return self._get_template_context(c)

    def _process_bookmark_entry(self, entry, user_id):
        position = safe_int(entry.get("position"))
        cur_position = safe_int(entry.get("cur_position"))
        if position is None:
            return

        # check if this is an existing entry
        is_new = False
        db_entry = UserBookmark().get_by_position_for_user(cur_position, user_id)

        if db_entry and str2bool(entry.get("remove")):
            log.debug("Marked bookmark %s for deletion", db_entry)
            Session().delete(db_entry)
            return

        if not db_entry:
            # new
            db_entry = UserBookmark()
            is_new = True

        should_save = False
        default_redirect_url = ""

        # save repo
        if entry.get("bookmark_repo") and safe_int(entry.get("bookmark_repo")):
            repo = Repository.get(entry["bookmark_repo"])
            perm_check = HasRepoPermissionAny("repository.read", "repository.write", "repository.admin")
            if repo and perm_check(repo_name=repo.repo_name):
                db_entry.repository = repo
                should_save = True
                default_redirect_url = "${repo_url}"
        # save repo group
        elif entry.get("bookmark_repo_group") and safe_int(entry.get("bookmark_repo_group")):
            repo_group = RepoGroup.get(entry["bookmark_repo_group"])
            perm_check = HasRepoGroupPermissionAny("group.read", "group.write", "group.admin")

            if repo_group and perm_check(group_name=repo_group.group_name):
                db_entry.repository_group = repo_group
                should_save = True
                default_redirect_url = "${repo_group_url}"
        # save generic info
        elif entry.get("title") and entry.get("redirect_url"):
            should_save = True

        if should_save:
            # mark user and position
            db_entry.user_id = user_id
            db_entry.position = position
            db_entry.title = entry.get("title")
            db_entry.redirect_url = entry.get("redirect_url") or default_redirect_url
            log.debug("Saving bookmark %s, new:%s", db_entry, is_new)

            Session().add(db_entry)

    @LoginRequired()
    @NotAnonymous()
    @CSRFRequired()
    def my_account_bookmarks_update(self):
        _ = self.request.translate
        c = self.load_default_context()
        c.active = "bookmarks"

        controls = peppercorn.parse(self.request.POST.items())
        user_id = c.user.user_id

        # validate positions
        positions = {}
        for entry in controls.get("bookmarks", []):
            position = safe_int(entry["position"])
            if position is None:
                continue

            if position in positions:
                h.flash(
                    _("Position {} is defined twice. Please correct this error.").format(position), category="error"
                )
                return HTTPFound(h.route_path("my_account_bookmarks"))

            entry["position"] = position
            entry["cur_position"] = safe_int(entry.get("cur_position"))
            positions[position] = entry

        try:
            for entry in positions.values():
                self._process_bookmark_entry(entry, user_id)

            Session().commit()
            h.flash(_("Update Bookmarks"), category="success")
        except IntegrityError:
            h.flash(_("Failed to update bookmarks. Make sure an unique position is used."), category="error")

        return HTTPFound(h.route_path("my_account_bookmarks"))

    @LoginRequired()
    @NotAnonymous()
    def my_account_goto_bookmark(self):
        bookmark_id = self.request.matchdict["bookmark_id"]
        user_bookmark = (
            UserBookmark()
            .query()
            .filter(UserBookmark.user_id == self.request.user.user_id)
            .filter(UserBookmark.position == bookmark_id)
            .scalar()
        )

        redirect_url = h.route_path("my_account_bookmarks")
        if not user_bookmark:
            raise HTTPFound(redirect_url)

        # repository set
        if user_bookmark.repository:
            repo_name = user_bookmark.repository.repo_name
            base_redirect_url = h.route_path("repo_summary", repo_name=repo_name)
            if user_bookmark.redirect_url and "${repo_url}" in user_bookmark.redirect_url:
                redirect_url = string.Template(user_bookmark.redirect_url).safe_substitute(
                    {"repo_url": base_redirect_url}
                )
            else:
                redirect_url = base_redirect_url
        # repository group set
        elif user_bookmark.repository_group:
            repo_group_name = user_bookmark.repository_group.group_name
            base_redirect_url = h.route_path("repo_group_home", repo_group_name=repo_group_name)
            if user_bookmark.redirect_url and "${repo_group_url}" in user_bookmark.redirect_url:
                redirect_url = string.Template(user_bookmark.redirect_url).safe_substitute(
                    {"repo_group_url": base_redirect_url}
                )
            else:
                redirect_url = base_redirect_url
        # custom URL set
        elif user_bookmark.redirect_url:
            server_url = h.route_url("home").rstrip("/")
            redirect_url = string.Template(user_bookmark.redirect_url).safe_substitute({"server_url": server_url})

        log.debug("Redirecting bookmark %s to %s", user_bookmark, redirect_url)
        raise HTTPFound(redirect_url)

    @LoginRequired()
    @NotAnonymous()
    def my_account_perms(self):
        c = self.load_default_context()
        c.active = "perms"

        c.perm_user = c.auth_user
        return self._get_template_context(c)

    @LoginRequired()
    @NotAnonymous()
    def my_notifications(self):
        c = self.load_default_context()
        c.active = "notifications"

        return self._get_template_context(c)

    @LoginRequired()
    @NotAnonymous()
    @CSRFRequired()
    def my_notifications_toggle_visibility(self):
        user = self._rhodecode_db_user
        new_status = not user.user_data.get("notification_status", True)
        user.update_userdata(notification_status=new_status)
        Session().commit()
        return user.user_data["notification_status"]

    def _get_pull_requests_list(self, statuses, filter_type=None):
        draw, start, limit = self._extract_chunk(self.request)
        search_q, order_by, order_dir = self._extract_ordering(self.request)

        _render = self.request.get_partial_renderer("rhodecode:templates/data_table/_dt_elements.mako")

        if filter_type == "awaiting_my_review":
            pull_requests = PullRequestModel().get_im_participating_in_for_review(
                user_id=self._rhodecode_user.user_id,
                statuses=statuses,
                query=search_q,
                offset=start,
                length=limit,
                order_by=order_by,
                order_dir=order_dir,
            )

            pull_requests_total_count = PullRequestModel().count_im_participating_in_for_review(
                user_id=self._rhodecode_user.user_id, statuses=statuses, query=search_q
            )
        else:
            pull_requests = PullRequestModel().get_im_participating_in(
                user_id=self._rhodecode_user.user_id,
                statuses=statuses,
                query=search_q,
                offset=start,
                length=limit,
                order_by=order_by,
                order_dir=order_dir,
            )

            pull_requests_total_count = PullRequestModel().count_im_participating_in(
                user_id=self._rhodecode_user.user_id, statuses=statuses, query=search_q
            )

        data = []
        comments_model = CommentsModel()
        for pr in pull_requests:
            repo_id = pr.target_repo_id
            comments_count = comments_model.get_all_comments(
                repo_id, pull_request=pr, include_drafts=False, count_only=True
            )
            owned = pr.user_id == self._rhodecode_user.user_id

            review_statuses = pr.reviewers_statuses(user=self._rhodecode_db_user)
            my_review_status = ChangesetStatus.STATUS_NOT_REVIEWED
            if review_statuses and review_statuses[4]:
                _review_obj, _user, _reasons, _mandatory, statuses = review_statuses
                my_review_status = statuses[0][1].status

            data.append(
                {
                    "target_repo": _render("pullrequest_target_repo", pr.target_repo.repo_name),
                    "name": _render(
                        "pullrequest_name",
                        pr.pull_request_id,
                        pr.pull_request_state,
                        pr.work_in_progress,
                        pr.target_repo.repo_name,
                        short=True,
                    ),
                    "name_raw": pr.pull_request_id,
                    "status": _render("pullrequest_status", pr.calculated_review_status()),
                    "my_status": _render("pullrequest_status", my_review_status),
                    "title": _render("pullrequest_title", pr.title, pr.description),
                    "pr_flow": _render("pullrequest_commit_flow", pr),
                    "description": h.escape(pr.description),
                    "updated_on": _render(
                        "pullrequest_updated_on", h.datetime_to_time(pr.updated_on), pr.versions_count
                    ),
                    "updated_on_raw": h.datetime_to_time(pr.updated_on),
                    "created_on": _render("pullrequest_updated_on", h.datetime_to_time(pr.created_on)),
                    "created_on_raw": h.datetime_to_time(pr.created_on),
                    "state": pr.pull_request_state,
                    "author": _render(
                        "pullrequest_author",
                        pr.author.full_contact,
                    ),
                    "author_raw": pr.author.full_name,
                    "comments": _render("pullrequest_comments", comments_count),
                    "comments_raw": comments_count,
                    "closed": pr.is_closed(),
                    "owned": owned,
                }
            )

        # json used to render the grid
        data = {
            "draw": draw,
            "data": data,
            "recordsTotal": pull_requests_total_count,
            "recordsFiltered": pull_requests_total_count,
        }
        return data

    @LoginRequired()
    @NotAnonymous()
    def my_account_pullrequests(self):
        c = self.load_default_context()
        c.active = "pullrequests"
        req_get = self.request.GET

        c.closed = str2bool(req_get.get("closed"))
        c.awaiting_my_review = str2bool(req_get.get("awaiting_my_review"))

        c.selected_filter = "all"
        if c.closed:
            c.selected_filter = "all_closed"
        if c.awaiting_my_review:
            c.selected_filter = "awaiting_my_review"

        return self._get_template_context(c)

    @LoginRequired()
    @NotAnonymous()
    def my_account_pullrequests_data(self):
        self.load_default_context()
        req_get = self.request.GET

        awaiting_my_review = str2bool(req_get.get("awaiting_my_review"))
        closed = str2bool(req_get.get("closed"))

        statuses = [PullRequest.STATUS_NEW, PullRequest.STATUS_OPEN]
        if closed:
            statuses += [PullRequest.STATUS_CLOSED]

        filter_type = "awaiting_my_review" if awaiting_my_review else None

        data = self._get_pull_requests_list(statuses=statuses, filter_type=filter_type)
        return data

    @LoginRequired()
    @NotAnonymous()
    def my_account_user_group_membership(self):
        c = self.load_default_context()
        c.active = "user_group_membership"
        groups = [
            UserGroupModel.get_user_groups_as_dict(group.users_group) for group in self._rhodecode_db_user.group_member
        ]
        c.user_groups = ext_json.str_json(groups)
        return self._get_template_context(c)
