#!/opt/cloudlinux/venv/bin/python3 -bb
# coding=utf-8
#
# Copyright © Cloud Linux GmbH & Cloud Linux Software, Inc 2010-2019 All Rights Reserved
#
# Licensed under CLOUD LINUX LICENSE AGREEMENT
# http://cloudlinux.com/docs/LICENSE.TXT
"""Module for all sentry processors related code"""

from raven.processors import SanitizePasswordsProcessor


class UserlandSanitize(SanitizePasswordsProcessor):
    """
    Beside from default SanitizePasswordsProcessor algorithm, also
    search key-values in query-like strings.
    Also sanitize some additional fields.
    """
    FIELDS = frozenset([
        'password',
        'secret',
        'passwd',
        'authorization',
        'api_key',
        'apikey',
        'sentry_dsn',
        'access_token',
        'email',
        'token',
        'pass',
    ])

    def sanitize(self, item, value):
        new_value = super().sanitize(item, value)
        if new_value == value and isinstance(value, str) and '=' in value:
            new_value = self._sanitize_keyvals(value, '&')
        return new_value