# This file is part newsletter module for Tryton.
# The COPYRIGHT file at the top level of this repository contains
# the full copyright notices and license terms.
from trytond.model import ModelView, ModelSQL, fields, Unique
from trytond.pool import Pool
from trytond.transaction import Transaction
from trytond.i18n import gettext
from trytond.exceptions import UserError

__all__ = ['NewsletterList', 'NewsletterContact', 'NewsletterContactList']


class NewsletterList(ModelSQL, ModelView):
    'Newsletter List'
    __name__ = 'newsletter.list'
    name = fields.Char('Name', required=True)
    contacts = fields.Many2Many('newsletter.contact-newsletter.list',
        'newsletter_list', 'newsletter_contact', 'Contacts')
    active = fields.Boolean('Active', select=True)

    @staticmethod
    def default_active():
        return True


class NewsletterContact(ModelSQL, ModelView):
    'Newsletter Contact'
    __name__ = 'newsletter.contact'
    _rec_name = 'email'
    email = fields.Char('Email', required=True)
    name = fields.Char('Name')
    party = fields.Many2One('party.party', 'Party')
    lists = fields.Many2Many('newsletter.contact-newsletter.list',
        'newsletter_contact', 'newsletter_list', 'Lists')
    active = fields.Boolean('Active', select=True)
    lang = fields.Many2One("ir.lang", 'Language')

    @classmethod
    def __setup__(cls):
        super(NewsletterContact, cls).__setup__()
        t = cls.__table__()
        cls._sql_constraints += [
            ('contact_uniq', Unique(t, t.email),
                'An email must be unique.'),
            ]

    @staticmethod
    def default_active():
        return True

    @staticmethod
    def default_lang():
        pool = Pool()
        Configuration = pool.get('party.configuration')
        Lang = pool.get('ir.lang')

        if Transaction().language:
            lang, = Lang.search([
                ('code', '=', Transaction().language),
                ], limit=1)
            return lang.id
        config = Configuration(1)
        if config.party_lang:
            return config.party_lang.id

    @staticmethod
    def default_lists():
        List = Pool().get('newsletter.list')
        return [l.id for l in List.search([])]

    @classmethod
    def create(cls, vlist):
        ContactMechanism = Pool().get('party.contact_mechanism')

        for vals in vlist:
            if not vals.get('party'):
                email = vals.get('email')
                contacts = ContactMechanism.search([
                    ('value', '=', email),
                    ('type', '=', 'email'),
                    ], limit=1)
                if contacts:
                    vals['party'] = contacts[0].party.id
                    vals['name'] = contacts[0].party.name

        return super(NewsletterContact, cls).create(vlist)

    @classmethod
    def copy(cls, contacts, default=None):
        raise UserError(gettext('newsletter.msg_copy_dissable'))

    @fields.depends('email', 'party')
    def on_change_email(self):
        ContactMechanism = Pool().get('party.contact_mechanism')

        if self.email and not self.party:
            contacts = ContactMechanism.search([
                ('value', '=', self.email),
                ('type', '=', 'email'),
                ], limit=1)
            if contacts:
                self.party = contacts[0].party
                self.name = contacts[0].party.name


class NewsletterContactList(ModelSQL):
    'Newsletter Contact - Newsletter List'
    __name__ = 'newsletter.contact-newsletter.list'
    _table = 'newsletter_contact_list_rel'
    newsletter_contact = fields.Many2One('newsletter.contact', 'Contact', ondelete='CASCADE',
            required=True, select=True)
    newsletter_list = fields.Many2One('newsletter.list', 'List',
        ondelete='CASCADE', required=True, select=True)