gui/text.py

   1 import curses, datetime, locale
   2 from decimal import Decimal
   3 _ = lambda x:x
   4 #from i18n import _
   5 from electrum_nvc.util import format_satoshis, set_verbosity
   6 from electrum_nvc.bitcoin import is_valid
   7
   8 from electrum_nvc import Wallet, WalletStorage
   9
  10 import tty, sys
  11
  12
  13 class ElectrumGui:
  14
  15     def __init__(self, config, network):
  16
  17         self.config = config
  18         self.network = network
  19         storage = WalletStorage(config)
  20         if not storage.file_exists:
  21             print "Wallet not found. try 'electrum-nvc create'"
  22             exit()
  23
  24         self.wallet = Wallet(storage)
  25         self.wallet.start_threads(self.network)
  26
  27         locale.setlocale(locale.LC_ALL, '')
  28         self.encoding = locale.getpreferredencoding()
  29
  30         self.stdscr = curses.initscr()
  31         curses.noecho()
  32         curses.cbreak()
  33         curses.start_color()
  34         curses.use_default_colors()
  35         curses.init_pair(1, curses.COLOR_WHITE, curses.COLOR_BLUE)
  36         curses.init_pair(2, curses.COLOR_WHITE, curses.COLOR_CYAN)
  37         self.stdscr.keypad(1)
  38         self.stdscr.border(0)
  39         self.maxy, self.maxx = self.stdscr.getmaxyx()
  40         self.set_cursor(0)
  41         self.w = curses.newwin(10, 50, 5, 5)
  42
  43         set_verbosity(False)
  44         self.tab = 0
  45         self.pos = 0
  46         self.popup_pos = 0
  47
  48         self.str_recipient = ""
  49         self.str_description = ""
  50         self.str_amount = ""
  51         self.str_fee = ""
  52         self.history = None
  53
  54         if self.network:
  55             self.network.register_callback('updated', self.update)
  56             self.network.register_callback('connected', self.refresh)
  57             self.network.register_callback('disconnected', self.refresh)
  58             self.network.register_callback('disconnecting', self.refresh)
  59
  60         self.tab_names = [_("History"), _("Send"), _("Receive"), _("Contacts"), _("Wall")]
  61         self.num_tabs = len(self.tab_names)
  62
  63
  64     def set_cursor(self, x):
  65         try:
  66             curses.curs_set(x)
  67         except Exception:
  68             pass
  69
  70     def restore_or_create(self):
  71         pass
  72
  73     def verify_seed(self):
  74         pass
  75
  76     def get_string(self, y, x):
  77         self.set_cursor(1)
  78         curses.echo()
  79         self.stdscr.addstr( y, x, " "*20, curses.A_REVERSE)
  80         s = self.stdscr.getstr(y,x)
  81         curses.noecho()
  82         self.set_cursor(0)
  83         return s
  84
  85     def update(self):
  86         self.update_history()
  87         if self.tab == 0:
  88             self.print_history()
  89         self.refresh()
  90
  91     def print_history(self):
  92
  93         width = [20, 40, 14, 14]
  94         delta = (self.maxx - sum(width) - 4)/3
  95         format_str = "%"+"%d"%width[0]+"s"+"%"+"%d"%(width[1]+delta)+"s"+"%"+"%d"%(width[2]+delta)+"s"+"%"+"%d"%(width[3]+delta)+"s"
  96
  97         if self.history is None:
  98             self.update_history()
  99
 100         self.print_list(self.history[::-1], format_str%( _("Date"), _("Description"), _("Amount"), _("Balance")))
 101
 102     def update_history(self):
 103         width = [20, 40, 14, 14]
 104         delta = (self.maxx - sum(width) - 4)/3
 105         format_str = "%"+"%d"%width[0]+"s"+"%"+"%d"%(width[1]+delta)+"s"+"%"+"%d"%(width[2]+delta)+"s"+"%"+"%d"%(width[3]+delta)+"s"
 106
 107         b = 0
 108         self.history = []
 109
 110         for item in self.wallet.get_tx_history():
 111             tx_hash, conf, is_mine, value, fee, balance, timestamp = item
 112             if conf:
 113                 try:
 114                     time_str = datetime.datetime.fromtimestamp( timestamp).isoformat(' ')[:-3]
 115                 except Exception:
 116                     time_str = "------"
 117             else:
 118                 time_str = 'pending'
 119
 120             label, is_default_label = self.wallet.get_label(tx_hash)
 121             self.history.append( format_str%( time_str, label, format_satoshis(value, whitespaces=True), format_satoshis(balance, whitespaces=True) ) )
 122
 123
 124     def print_balance(self):
 125         if not self.network:
 126             msg = _("Offline")
 127         elif self.network.interface and self.network.interface.is_connected:
 128             if not self.wallet.up_to_date:
 129                 msg = _("Synchronizing...")
 130             else:
 131                 c, u =  self.wallet.get_balance()
 132                 msg = _("Balance")+": %f  "%(Decimal( c ) / 1000000)
 133                 if u: msg += "  [%f unconfirmed]"%(Decimal( u ) / 1000000)
 134         else:
 135             msg = _("Not connected")
 136
 137         self.stdscr.addstr( self.maxy -1, 3, msg)
 138
 139         for i in range(self.num_tabs):
 140             self.stdscr.addstr( 0, 2 + 2*i + len(''.join(self.tab_names[0:i])), ' '+self.tab_names[i]+' ', curses.A_BOLD if self.tab == i else 0)
 141
 142         self.stdscr.addstr( self.maxy -1, self.maxx-30, ' '.join([_("Settings"), _("Network"), _("Quit")]))
 143
 144
 145     def print_contacts(self):
 146         messages = map(lambda addr: "%30s    %30s       "%(addr, self.wallet.labels.get(addr,"")), self.wallet.addressbook)
 147         self.print_list(messages, "%19s  %25s "%("Address", "Label"))
 148
 149     def print_receive(self):
 150         fmt = "%-35s  %-30s"
 151         messages = map(lambda addr: fmt % (addr, self.wallet.labels.get(addr,"")), self.wallet.addresses())
 152         self.print_list(messages,   fmt % ("Address", "Label"))
 153
 154     def print_edit_line(self, y, label, text, index, size):
 155         text += " "*(size - len(text) )
 156         self.stdscr.addstr( y, 2, label)
 157         self.stdscr.addstr( y, 15, text, curses.A_REVERSE if self.pos%6==index else curses.color_pair(1))
 158
 159     def print_send_tab(self):
 160         self.stdscr.clear()
 161         self.print_edit_line(3, _("Pay to"), self.str_recipient, 0, 40)
 162         self.print_edit_line(5, _("Description"), self.str_description, 1, 40)
 163         self.print_edit_line(7, _("Amount"), self.str_amount, 2, 15)
 164         self.print_edit_line(9, _("Fee"), self.str_fee, 3, 15)
 165         self.stdscr.addstr( 12, 15, _("[Send]"), curses.A_REVERSE if self.pos%6==4 else curses.color_pair(2))
 166         self.stdscr.addstr( 12, 25, _("[Clear]"), curses.A_REVERSE if self.pos%6==5 else curses.color_pair(2))
 167
 168     def print_banner(self):
 169         if self.network:
 170             self.print_list( self.network.banner.split('\n'))
 171
 172     def print_list(self, list, firstline = None):
 173         self.maxpos = len(list)
 174         if not self.maxpos: return
 175         if firstline:
 176             firstline += " "*(self.maxx -2 - len(firstline))
 177             self.stdscr.addstr( 1, 1, firstline )
 178         for i in range(self.maxy-4):
 179             msg = list[i] if i < len(list) else ""
 180             msg += " "*(self.maxx - 2 - len(msg))
 181             m = msg[0:self.maxx - 2]
 182             m = m.encode(self.encoding)
 183             self.stdscr.addstr( i+2, 1, m, curses.A_REVERSE if i == (self.pos % self.maxpos) else 0)
 184
 185     def refresh(self):
 186         if self.tab == -1: return
 187         self.stdscr.border(0)
 188         self.print_balance()
 189         self.stdscr.refresh()
 190
 191     def main_command(self):
 192         c = self.stdscr.getch()
 193         print c
 194         if   c == curses.KEY_RIGHT: self.tab = (self.tab + 1)%self.num_tabs
 195         elif c == curses.KEY_LEFT: self.tab = (self.tab - 1)%self.num_tabs
 196         elif c == curses.KEY_DOWN: self.pos +=1
 197         elif c == curses.KEY_UP: self.pos -= 1
 198         elif c == 9: self.pos +=1 # tab
 199         elif curses.unctrl(c) in ['^W', '^C', '^X', '^Q']: self.tab = -1
 200         elif curses.unctrl(c) in ['^N']: self.network_dialog()
 201         elif curses.unctrl(c) == '^S': self.settings_dialog()
 202         else: return c
 203         if self.pos<0: self.pos=0
 204         if self.pos>=self.maxpos: self.pos=self.maxpos - 1
 205
 206     def run_tab(self, i, print_func, exec_func):
 207         while self.tab == i:
 208             self.stdscr.clear()
 209             print_func()
 210             self.refresh()
 211             c = self.main_command()
 212             if c: exec_func(c)
 213
 214
 215     def run_history_tab(self, c):
 216         if c == 10:
 217             out = self.run_popup('',["blah","foo"])
 218
 219
 220     def edit_str(self, target, c, is_num=False):
 221         # detect backspace
 222         if c in [8, 127, 263] and target:
 223             target = target[:-1]
 224         elif not is_num or curses.unctrl(c) in '0123456789.':
 225             target += curses.unctrl(c)
 226         return target
 227
 228
 229     def run_send_tab(self, c):
 230         if self.pos%6 == 0:
 231             self.str_recipient = self.edit_str(self.str_recipient, c)
 232         if self.pos%6 == 1:
 233             self.str_description = self.edit_str(self.str_description, c)
 234         if self.pos%6 == 2:
 235             self.str_amount = self.edit_str(self.str_amount, c, True)
 236         elif self.pos%6 == 3:
 237             self.str_fee = self.edit_str(self.str_fee, c, True)
 238         elif self.pos%6==4:
 239             if c == 10: self.do_send()
 240         elif self.pos%6==5:
 241             if c == 10: self.do_clear()
 242
 243
 244     def run_receive_tab(self, c):
 245         if c == 10:
 246             out = self.run_popup('Address', ["Edit label", "Freeze", "Prioritize"])
 247
 248     def run_contacts_tab(self, c):
 249         if c == 10 and self.wallet.addressbook:
 250             out = self.run_popup('Adress', ["Copy", "Pay to", "Edit label", "Delete"]).get('button')
 251             address = self.wallet.addressbook[self.pos%len(self.wallet.addressbook)]
 252             if out == "Pay to":
 253                 self.tab = 1
 254                 self.str_recipient = address
 255                 self.pos = 2
 256             elif out == "Edit label":
 257                 s = self.get_string(6 + self.pos, 18)
 258                 if s:
 259                     self.wallet.labels[address] = s
 260
 261     def run_banner_tab(self, c):
 262         self.show_message(repr(c))
 263         pass
 264
 265     def main(self,url):
 266
 267         tty.setraw(sys.stdin)
 268         while self.tab != -1:
 269             self.run_tab(0, self.print_history, self.run_history_tab)
 270             self.run_tab(1, self.print_send_tab, self.run_send_tab)
 271             self.run_tab(2, self.print_receive, self.run_receive_tab)
 272             self.run_tab(3, self.print_contacts, self.run_contacts_tab)
 273             self.run_tab(4, self.print_banner, self.run_banner_tab)
 274
 275         tty.setcbreak(sys.stdin)
 276         curses.nocbreak()
 277         self.stdscr.keypad(0)
 278         curses.echo()
 279         curses.endwin()
 280
 281
 282     def do_clear(self):
 283         self.str_amount = ''
 284         self.str_recipient = ''
 285         self.str_fee = ''
 286         self.str_description = ''
 287
 288     def do_send(self):
 289         if not is_valid(self.str_recipient):
 290             self.show_message(_('Invalid Novacoin address'))
 291             return
 292         try:
 293             amount = int( Decimal( self.str_amount) * 1000000 )
 294         except Exception:
 295             self.show_message(_('Invalid Amount'))
 296             return
 297         try:
 298             fee = int( Decimal( self.str_fee) * 1000000 )
 299         except Exception:
 300             self.show_message(_('Invalid Fee'))
 301             return
 302
 303         if self.wallet.use_encryption:
 304             password = self.password_dialog()
 305             if not password:
 306                 return
 307         else:
 308             password = None
 309
 310         try:
 311             tx = self.wallet.mktx( [(self.str_recipient, amount)], password, fee)
 312         except Exception as e:
 313             self.show_message(str(e))
 314             return
 315
 316         if self.str_description:
 317             self.wallet.labels[tx.hash()] = self.str_description
 318
 319         h = self.wallet.send_tx(tx)
 320         self.show_message(_("Please wait..."), getchar=False)
 321         self.wallet.tx_event.wait()
 322         status, msg = self.wallet.receive_tx( h, tx )
 323
 324         if status:
 325             self.show_message(_('Payment sent.'))
 326             self.do_clear()
 327             #self.update_contacts_tab()
 328         else:
 329             self.show_message(_('Error'))
 330
 331
 332     def show_message(self, message, getchar = True):
 333         w = self.w
 334         w.clear()
 335         w.border(0)
 336         for i, line in enumerate(message.split('\n')):
 337             w.addstr(2+i,2,line)
 338         w.refresh()
 339         if getchar: c = self.stdscr.getch()
 340
 341
 342     def run_popup(self, title, items):
 343         return self.run_dialog(title, map(lambda x: {'type':'button','label':x}, items), interval=1, y_pos = self.pos+3)
 344
 345
 346     def network_dialog(self):
 347         if not self.network: return
 348         auto_connect = self.network.config.get('auto_cycle')
 349         host, port, protocol = self.network.default_server.split(':')
 350         srv = 'auto-connect' if auto_connect else self.network.default_server
 351
 352         out = self.run_dialog('Network', [
 353             {'label':'server', 'type':'str', 'value':srv},
 354             {'label':'proxy', 'type':'str', 'value':self.config.get('proxy', '')},
 355             ], buttons = 1)
 356         if out:
 357             if out.get('server'):
 358                 server = out.get('server')
 359                 auto_connect = server == 'auto-connect'
 360                 if not auto_connect:
 361                     try:
 362                         host, port, protocol = server.split(':')
 363                     except Exception:
 364                         self.show_message("Error:" + server + "\nIn doubt, type \"auto-connect\"")
 365                         return False
 366
 367                 if out.get('proxy'):
 368                     proxy = self.parse_proxy_options(out.get('proxy'))
 369                 else:
 370                     proxy = None
 371
 372                 self.network.set_parameters(host, port, protocol, proxy, auto_connect)
 373
 374
 375
 376     def settings_dialog(self):
 377         out = self.run_dialog('Settings', [
 378             {'label':'Default GUI', 'type':'list', 'choices':['classic','lite','gtk','text'], 'value':self.config.get('gui')},
 379             {'label':'Default fee', 'type':'satoshis', 'value': format_satoshis(self.wallet.fee).strip() }
 380             ], buttons = 1)
 381         if out:
 382             if out.get('Default GUI'):
 383                 self.config.set_key('gui', out['Default GUI'], True)
 384             if out.get('Default fee'):
 385                 fee = int ( Decimal( out['Default fee']) *10000000 )
 386                 self.config.set_key('fee_per_kb', fee, True)
 387
 388
 389     def password_dialog(self):
 390         out = self.run_dialog('Password', [
 391             {'label':'Password', 'type':'password', 'value':''}
 392             ], buttons = 1)
 393         return out.get('Password')
 394
 395
 396     def run_dialog(self, title, items, interval=2, buttons=None, y_pos=3):
 397         self.popup_pos = 0
 398
 399         self.w = curses.newwin( 5 + len(items)*interval + (2 if buttons else 0), 50, y_pos, 5)
 400         w = self.w
 401         out = {}
 402         while True:
 403             w.clear()
 404             w.border(0)
 405             w.addstr( 0, 2, title)
 406
 407             num = len(items)
 408
 409             numpos = num
 410             if buttons: numpos += 2
 411
 412             for i in range(num):
 413                 item = items[i]
 414                 label = item.get('label')
 415                 if item.get('type') == 'list':
 416                     value = item.get('value','')
 417                 elif item.get('type') == 'satoshis':
 418                     value = item.get('value','')
 419                 elif item.get('type') == 'str':
 420                     value = item.get('value','')
 421                 elif item.get('type') == 'password':
 422                     value = '*'*len(item.get('value',''))
 423                 else:
 424                     value = ''
 425
 426                 if len(value)<20: value += ' '*(20-len(value))
 427
 428                 if item.has_key('value'):
 429                     w.addstr( 2+interval*i, 2, label)
 430                     w.addstr( 2+interval*i, 15, value, curses.A_REVERSE if self.popup_pos%numpos==i else curses.color_pair(1) )
 431                 else:
 432                     w.addstr( 2+interval*i, 2, label, curses.A_REVERSE if self.popup_pos%numpos==i else 0)
 433
 434             if buttons:
 435                 w.addstr( 5+interval*i, 10, "[  ok  ]",     curses.A_REVERSE if self.popup_pos%numpos==(numpos-2) else curses.color_pair(2))
 436                 w.addstr( 5+interval*i, 25, "[cancel]", curses.A_REVERSE if self.popup_pos%numpos==(numpos-1) else curses.color_pair(2))
 437
 438             w.refresh()
 439
 440             c = self.stdscr.getch()
 441             if c in [ord('q'), 27]: break
 442             elif c in [curses.KEY_LEFT, curses.KEY_UP]: self.popup_pos -= 1
 443             elif c in [curses.KEY_RIGHT, curses.KEY_DOWN]: self.popup_pos +=1
 444             else:
 445                 i = self.popup_pos%numpos
 446                 if buttons and c==10:
 447                     if i == numpos-2:
 448                         return out
 449                     elif i == numpos -1:
 450                         return {}
 451
 452                 item = items[i]
 453                 _type = item.get('type')
 454
 455                 if _type == 'str':
 456                     item['value'] = self.edit_str(item['value'], c)
 457                     out[item.get('label')] = item.get('value')
 458
 459                 elif _type == 'password':
 460                     item['value'] = self.edit_str(item['value'], c)
 461                     out[item.get('label')] = item ['value']
 462
 463                 elif _type == 'satoshis':
 464                     item['value'] = self.edit_str(item['value'], c, True)
 465                     out[item.get('label')] = item.get('value')
 466
 467                 elif _type == 'list':
 468                     choices = item.get('choices')
 469                     try:
 470                         j = choices.index(item.get('value'))
 471                     except Exception:
 472                         j = 0
 473                     new_choice = choices[(j + 1)% len(choices)]
 474                     item['value'] = new_choice
 475                     out[item.get('label')] = item.get('value')
 476
 477                 elif _type == 'button':
 478                     out['button'] = item.get('label')
 479                     break
 480
 481         return out
 482