basicswap_miserver/basicswap/ui/page_smsgaddresses.py

# -*- coding: utf-8 -*-

# Copyright (c) 2023 tecnovert
# Distributed under the MIT software license, see the accompanying
# file LICENSE or http://www.opensource.org/licenses/mit-license.php.


from .util import (
    PAGE_LIMIT,
    get_data_entry,
    have_data_entry,
    get_data_entry_or,
    validateTextInput,
    set_pagination_filters,
)
from basicswap.util import (
    ensure,
)
from basicswap.basicswap_util import (
    AddressTypes,
    strAddressType,
)


def page_smsgaddresses(self, url_split, post_string):
    swap_client = self.server.swap_client
    swap_client.checkSystemStatus()
    summary = swap_client.getSummary()

    filters = {
        'page_no': 1,
        'limit': PAGE_LIMIT,
        'sort_by': 'created_at',
        'sort_dir': 'desc',
        'addr_type': -1,
    }

    page_data = {}
    messages = []
    err_messages = []
    smsgaddresses = []

    listaddresses = True
    form_data = self.checkForm(post_string, 'smsgaddresses', err_messages)
    if form_data:
        edit_address_id = None
        for key in form_data:
            if key.startswith(b'editaddr_'):
                edit_address_id = int(key.split(b'_')[1])
                break
        if edit_address_id is not None:
            listaddresses = False
            page_data['edit_address'] = edit_address_id
            page_data['addr_data'] = swap_client.listAllSMSGAddresses({'addr_id': edit_address_id})[0]
        elif have_data_entry(form_data, 'saveaddr'):
            edit_address_id = int(get_data_entry(form_data, 'edit_address_id'))
            edit_addr = get_data_entry(form_data, 'edit_address')
            active_ind = int(get_data_entry(form_data, 'active_ind'))
            ensure(active_ind in (0, 1), 'Invalid sort by')
            addressnote = get_data_entry_or(form_data, 'addressnote', '')
            if not validateTextInput(addressnote, 'Address note', err_messages, max_length=30):
                listaddresses = False
                page_data['edit_address'] = edit_address_id
            else:
                swap_client.editSMSGAddress(edit_addr, active_ind=active_ind, addressnote=addressnote)
                messages.append(f'Edited address {edit_addr}')
        elif have_data_entry(form_data, 'shownewaddr'):
            listaddresses = False
            page_data['new_address'] = True
        elif have_data_entry(form_data, 'showaddaddr'):
            listaddresses = False
            page_data['new_send_address'] = True
        elif have_data_entry(form_data, 'createnewaddr'):
            addressnote = get_data_entry_or(form_data, 'addressnote', '')
            if not validateTextInput(addressnote, 'Address note', err_messages, max_length=30):
                listaddresses = False
                page_data['new_address'] = True
            else:
                new_addr, pubkey = swap_client.newSMSGAddress(addressnote=addressnote)
                messages.append(f'Created address {new_addr}, pubkey {pubkey}')
        elif have_data_entry(form_data, 'createnewsendaddr'):
            pubkey_hex = get_data_entry(form_data, 'addresspubkey')
            addressnote = get_data_entry_or(form_data, 'addressnote', '')
            if not validateTextInput(addressnote, 'Address note', messages, max_length=30) or \
               not validateTextInput(pubkey_hex, 'Pubkey', messages, max_length=66):
                listaddresses = False
                page_data['new_send_address'] = True
            else:
                new_addr = swap_client.addSMSGAddress(pubkey_hex, addressnote=addressnote)
                messages.append(f'Added address {new_addr}')

        if have_data_entry(form_data, 'clearfilters'):
            swap_client.clearFilters('page_smsgaddresses')
        else:
            if have_data_entry(form_data, 'sort_by'):
                sort_by = get_data_entry(form_data, 'sort_by')
                ensure(sort_by in ['created_at', 'rate'], 'Invalid sort by')
                filters['sort_by'] = sort_by
            if have_data_entry(form_data, 'sort_dir'):
                sort_dir = get_data_entry(form_data, 'sort_dir')
                ensure(sort_dir in ['asc', 'desc'], 'Invalid sort dir')
                filters['sort_dir'] = sort_dir
            if have_data_entry(form_data, 'filter_addressnote'):
                addressnote = get_data_entry(form_data, 'filter_addressnote')
                if validateTextInput(addressnote, 'Address note', err_messages, max_length=30):
                    filters['addressnote'] = addressnote
            if have_data_entry(form_data, 'filter_addr_type'):
                filters['addr_type'] = int(get_data_entry(form_data, 'filter_addr_type'))

            set_pagination_filters(form_data, filters)
        if have_data_entry(form_data, 'applyfilters'):
            swap_client.setFilters('page_smsgaddresses', filters)
    else:
        saved_filters = swap_client.getFilters('page_smsgaddresses')
        if saved_filters:
            filters.update(saved_filters)

    if listaddresses is True:
        smsgaddresses = swap_client.listAllSMSGAddresses(filters)

        page_data['addr_types'] = [(int(t), strAddressType(t)) for t in AddressTypes]
        page_data['network_addr'] = swap_client.network_addr

    for addr in smsgaddresses:
        addr['type'] = strAddressType(addr['type'])

    template = self.server.env.get_template('smsgaddresses.html')
    return self.render_template(template, {
        'messages': messages,
        'err_messages': err_messages,
        'filters': filters,
        'data': page_data,
        'smsgaddresses': smsgaddresses,
        'page_data': page_data,
        'summary': summary,
    })