mmgen-wallet/mmgen/addrdata.py

#!/usr/bin/env python3
#
# MMGen Wallet, a terminal-based cryptocurrency wallet
# Copyright (C)2013-2024 The MMGen Project <mmgen@tuta.io>
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# 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 General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.

"""
addrdata: MMGen AddrData and related classes
"""

from .cfg import gc
from .util import fmt, die
from .base_obj import AsyncInit
from .obj import MMGenObject, MMGenDict, get_obj
from .addr import MMGenID, AddrListID
from .addrlist import AddrListEntry, AddrListData, AddrList

class AddrData(MMGenObject):

	def __init__(self, proto, *args, **kwargs):
		self.al_ids = {}
		self.proto = proto
		self.rpc = None

	def seed_ids(self):
		return list(self.al_ids.keys())

	def addrlist(self, al_id):
		# TODO: Validate al_id
		if al_id in self.al_ids:
			return self.al_ids[al_id]

	def mmaddr2coinaddr(self, mmaddr):
		al_id, idx = MMGenID(self.proto, mmaddr).rsplit(':', 1)
		coinaddr = ''
		if al_id in self.al_ids:
			coinaddr = self.addrlist(al_id).coinaddr(int(idx))
		return coinaddr or None

	def coinaddr2mmaddr(self, coinaddr):
		d = self.make_reverse_dict([coinaddr])
		return (list(d.values())[0][0]) if d else None

	def add(self, addrlist):
		if isinstance(addrlist, AddrList):
			self.al_ids[addrlist.al_id] = addrlist
			return True
		else:
			raise TypeError(f'Error: object {addrlist!r} is not an instance of AddrList')

	def make_reverse_dict(self, coinaddrs):
		d = MMGenDict()
		for al_id in self.al_ids:
			d.update(self.al_ids[al_id].make_reverse_dict_addrlist(coinaddrs))
		return d

class TwAddrData(AddrData, metaclass=AsyncInit):

	def __new__(cls, cfg, proto, *args, **kwargs):
		return MMGenObject.__new__(proto.base_proto_subclass(cls, 'addrdata'))

	async def __init__(self, cfg, proto, twctl=None):
		from .rpc import rpc_init
		from .tw.shared import TwLabel
		from .seed import SeedID
		self.cfg = cfg
		self.proto = proto
		self.rpc = await rpc_init(cfg, proto)
		self.al_ids = {}
		twd = await self.get_tw_data(twctl)
		out, i = {}, 0
		for acct, addr_array in twd:
			l = get_obj(TwLabel, proto=self.proto, text=acct, silent=True)
			if l and l.mmid.type == 'mmgen':
				obj = l.mmid.obj
				if len(addr_array) != 1:
					message = self.msgs['multiple_acct_addrs'].strip().format(acct=acct, proj=gc.proj_name)
					die(3, fmt(message, indent='  '))
				al_id = AddrListID(
					sid = SeedID(sid=obj.sid),
					mmtype = self.proto.addr_type(obj.mmtype))
				if al_id not in out:
					out[al_id] = []
				out[al_id].append(AddrListEntry(self.proto, idx=obj.idx, addr=addr_array[0], comment=l.comment))
				i += 1

		self.cfg._util.vmsg(f'{i} {gc.proj_name} addresses found, {len(twd)} accounts total')

		for al_id in out:
			self.add(AddrList(
				self.cfg,
				self.proto,
				al_id = al_id,
				adata = AddrListData(sorted(out[al_id], key=lambda a: a.idx))
			))