src/allmydata/web/status.py

   1
   2 import time, pprint, itertools
   3 import simplejson
   4 from twisted.internet import defer
   5 from nevow import rend, inevow, tags as T
   6 from allmydata.util import base32, idlib
   7 from allmydata.web.common import getxmlfile, get_arg, \
   8      abbreviate_time, abbreviate_rate, abbreviate_size, plural
   9 from allmydata.interfaces import IUploadStatus, IDownloadStatus, \
  10      IPublishStatus, IRetrieveStatus, IServermapUpdaterStatus
  11
  12 class RateAndTimeMixin:
  13
  14     def render_time(self, ctx, data):
  15         return abbreviate_time(data)
  16
  17     def render_rate(self, ctx, data):
  18         return abbreviate_rate(data)
  19
  20 class UploadResultsRendererMixin(RateAndTimeMixin):
  21     # this requires a method named 'upload_results'
  22
  23     def render_pushed_shares(self, ctx, data):
  24         d = self.upload_results()
  25         d.addCallback(lambda res: res.pushed_shares)
  26         return d
  27
  28     def render_preexisting_shares(self, ctx, data):
  29         d = self.upload_results()
  30         d.addCallback(lambda res: res.preexisting_shares)
  31         return d
  32
  33     def render_sharemap(self, ctx, data):
  34         d = self.upload_results()
  35         d.addCallback(lambda res: res.sharemap)
  36         def _render(sharemap):
  37             if sharemap is None:
  38                 return "None"
  39             l = T.ul()
  40             for shnum, peerids in sorted(sharemap.items()):
  41                 peerids = ', '.join([idlib.shortnodeid_b2a(i) for i in peerids])
  42                 l[T.li["%d -> placed on [%s]" % (shnum, peerids)]]
  43             return l
  44         d.addCallback(_render)
  45         return d
  46
  47     def render_servermap(self, ctx, data):
  48         d = self.upload_results()
  49         d.addCallback(lambda res: res.servermap)
  50         def _render(servermap):
  51             if servermap is None:
  52                 return "None"
  53             l = T.ul()
  54             for peerid in sorted(servermap.keys()):
  55                 peerid_s = idlib.shortnodeid_b2a(peerid)
  56                 shares_s = ",".join(["#%d" % shnum
  57                                      for shnum in servermap[peerid]])
  58                 l[T.li["[%s] got share%s: %s" % (peerid_s,
  59                                                  plural(servermap[peerid]),
  60                                                  shares_s)]]
  61             return l
  62         d.addCallback(_render)
  63         return d
  64
  65     def data_file_size(self, ctx, data):
  66         d = self.upload_results()
  67         d.addCallback(lambda res: res.file_size)
  68         return d
  69
  70     def _get_time(self, name):
  71         d = self.upload_results()
  72         d.addCallback(lambda res: res.timings.get(name))
  73         return d
  74
  75     def data_time_total(self, ctx, data):
  76         return self._get_time("total")
  77
  78     def data_time_storage_index(self, ctx, data):
  79         return self._get_time("storage_index")
  80
  81     def data_time_contacting_helper(self, ctx, data):
  82         return self._get_time("contacting_helper")
  83
  84     def data_time_existence_check(self, ctx, data):
  85         return self._get_time("existence_check")
  86
  87     def data_time_cumulative_fetch(self, ctx, data):
  88         return self._get_time("cumulative_fetch")
  89
  90     def data_time_helper_total(self, ctx, data):
  91         return self._get_time("helper_total")
  92
  93     def data_time_peer_selection(self, ctx, data):
  94         return self._get_time("peer_selection")
  95
  96     def data_time_total_encode_and_push(self, ctx, data):
  97         return self._get_time("total_encode_and_push")
  98
  99     def data_time_cumulative_encoding(self, ctx, data):
 100         return self._get_time("cumulative_encoding")
 101
 102     def data_time_cumulative_sending(self, ctx, data):
 103         return self._get_time("cumulative_sending")
 104
 105     def data_time_hashes_and_close(self, ctx, data):
 106         return self._get_time("hashes_and_close")
 107
 108     def _get_rate(self, name):
 109         d = self.upload_results()
 110         def _convert(r):
 111             file_size = r.file_size
 112             time = r.timings.get(name)
 113             if time is None:
 114                 return None
 115             try:
 116                 return 1.0 * file_size / time
 117             except ZeroDivisionError:
 118                 return None
 119         d.addCallback(_convert)
 120         return d
 121
 122     def data_rate_total(self, ctx, data):
 123         return self._get_rate("total")
 124
 125     def data_rate_storage_index(self, ctx, data):
 126         return self._get_rate("storage_index")
 127
 128     def data_rate_encode(self, ctx, data):
 129         return self._get_rate("cumulative_encoding")
 130
 131     def data_rate_push(self, ctx, data):
 132         return self._get_rate("cumulative_sending")
 133
 134     def data_rate_encode_and_push(self, ctx, data):
 135         d = self.upload_results()
 136         def _convert(r):
 137             file_size = r.file_size
 138             time1 = r.timings.get("cumulative_encoding")
 139             time2 = r.timings.get("cumulative_sending")
 140             if (file_size is None or time1 is None or time2 is None):
 141                 return None
 142             try:
 143                 return 1.0 * file_size / (time1+time2)
 144             except ZeroDivisionError:
 145                 return None
 146         d.addCallback(_convert)
 147         return d
 148
 149     def data_rate_ciphertext_fetch(self, ctx, data):
 150         d = self.upload_results()
 151         def _convert(r):
 152             fetch_size = r.ciphertext_fetched
 153             time = r.timings.get("cumulative_fetch")
 154             if (fetch_size is None or time is None):
 155                 return None
 156             try:
 157                 return 1.0 * fetch_size / time
 158             except ZeroDivisionError:
 159                 return None
 160         d.addCallback(_convert)
 161         return d
 162
 163 class UploadStatusPage(UploadResultsRendererMixin, rend.Page):
 164     docFactory = getxmlfile("upload-status.xhtml")
 165
 166     def __init__(self, data):
 167         rend.Page.__init__(self, data)
 168         self.upload_status = data
 169
 170     def upload_results(self):
 171         return defer.maybeDeferred(self.upload_status.get_results)
 172
 173     def render_results(self, ctx, data):
 174         d = self.upload_results()
 175         def _got_results(results):
 176             if results:
 177                 return ctx.tag
 178             return ""
 179         d.addCallback(_got_results)
 180         return d
 181
 182     def render_started(self, ctx, data):
 183         TIME_FORMAT = "%H:%M:%S %d-%b-%Y"
 184         started_s = time.strftime(TIME_FORMAT,
 185                                   time.localtime(data.get_started()))
 186         return started_s
 187
 188     def render_si(self, ctx, data):
 189         si_s = base32.b2a_or_none(data.get_storage_index())
 190         if si_s is None:
 191             si_s = "(None)"
 192         return si_s
 193
 194     def render_helper(self, ctx, data):
 195         return {True: "Yes",
 196                 False: "No"}[data.using_helper()]
 197
 198     def render_total_size(self, ctx, data):
 199         size = data.get_size()
 200         if size is None:
 201             return "(unknown)"
 202         return size
 203
 204     def render_progress_hash(self, ctx, data):
 205         progress = data.get_progress()[0]
 206         # TODO: make an ascii-art bar
 207         return "%.1f%%" % (100.0 * progress)
 208
 209     def render_progress_ciphertext(self, ctx, data):
 210         progress = data.get_progress()[1]
 211         # TODO: make an ascii-art bar
 212         return "%.1f%%" % (100.0 * progress)
 213
 214     def render_progress_encode_push(self, ctx, data):
 215         progress = data.get_progress()[2]
 216         # TODO: make an ascii-art bar
 217         return "%.1f%%" % (100.0 * progress)
 218
 219     def render_status(self, ctx, data):
 220         return data.get_status()
 221
 222 class DownloadResultsRendererMixin(RateAndTimeMixin):
 223     # this requires a method named 'download_results'
 224
 225     def render_servermap(self, ctx, data):
 226         d = self.download_results()
 227         d.addCallback(lambda res: res.servermap)
 228         def _render(servermap):
 229             if servermap is None:
 230                 return "None"
 231             l = T.ul()
 232             for peerid in sorted(servermap.keys()):
 233                 peerid_s = idlib.shortnodeid_b2a(peerid)
 234                 shares_s = ",".join(["#%d" % shnum
 235                                      for shnum in servermap[peerid]])
 236                 l[T.li["[%s] has share%s: %s" % (peerid_s,
 237                                                  plural(servermap[peerid]),
 238                                                  shares_s)]]
 239             return l
 240         d.addCallback(_render)
 241         return d
 242
 243     def render_servers_used(self, ctx, data):
 244         d = self.download_results()
 245         d.addCallback(lambda res: res.servers_used)
 246         def _got(servers_used):
 247             if not servers_used:
 248                 return ""
 249             peerids_s = ", ".join(["[%s]" % idlib.shortnodeid_b2a(peerid)
 250                                    for peerid in servers_used])
 251             return T.li["Servers Used: ", peerids_s]
 252         d.addCallback(_got)
 253         return d
 254
 255     def render_problems(self, ctx, data):
 256         d = self.download_results()
 257         d.addCallback(lambda res: res.server_problems)
 258         def _got(server_problems):
 259             if not server_problems:
 260                 return ""
 261             l = T.ul()
 262             for peerid in sorted(server_problems.keys()):
 263                 peerid_s = idlib.shortnodeid_b2a(peerid)
 264                 l[T.li["[%s]: %s" % (peerid_s, server_problems[peerid])]]
 265             return T.li["Server Problems:", l]
 266         d.addCallback(_got)
 267         return d
 268
 269     def data_file_size(self, ctx, data):
 270         d = self.download_results()
 271         d.addCallback(lambda res: res.file_size)
 272         return d
 273
 274     def _get_time(self, name):
 275         d = self.download_results()
 276         d.addCallback(lambda res: res.timings.get(name))
 277         return d
 278
 279     def data_time_total(self, ctx, data):
 280         return self._get_time("total")
 281
 282     def data_time_peer_selection(self, ctx, data):
 283         return self._get_time("peer_selection")
 284
 285     def data_time_uri_extension(self, ctx, data):
 286         return self._get_time("uri_extension")
 287
 288     def data_time_hashtrees(self, ctx, data):
 289         return self._get_time("hashtrees")
 290
 291     def data_time_segments(self, ctx, data):
 292         return self._get_time("segments")
 293
 294     def data_time_cumulative_fetch(self, ctx, data):
 295         return self._get_time("cumulative_fetch")
 296
 297     def data_time_cumulative_decode(self, ctx, data):
 298         return self._get_time("cumulative_decode")
 299
 300     def data_time_cumulative_decrypt(self, ctx, data):
 301         return self._get_time("cumulative_decrypt")
 302
 303     def data_time_paused(self, ctx, data):
 304         return self._get_time("paused")
 305
 306     def _get_rate(self, name):
 307         d = self.download_results()
 308         def _convert(r):
 309             file_size = r.file_size
 310             time = r.timings.get(name)
 311             if time is None:
 312                 return None
 313             try:
 314                 return 1.0 * file_size / time
 315             except ZeroDivisionError:
 316                 return None
 317         d.addCallback(_convert)
 318         return d
 319
 320     def data_rate_total(self, ctx, data):
 321         return self._get_rate("total")
 322
 323     def data_rate_segments(self, ctx, data):
 324         return self._get_rate("segments")
 325
 326     def data_rate_fetch(self, ctx, data):
 327         return self._get_rate("cumulative_fetch")
 328
 329     def data_rate_decode(self, ctx, data):
 330         return self._get_rate("cumulative_decode")
 331
 332     def data_rate_decrypt(self, ctx, data):
 333         return self._get_rate("cumulative_decrypt")
 334
 335     def render_server_timings(self, ctx, data):
 336         d = self.download_results()
 337         d.addCallback(lambda res: res.timings.get("fetch_per_server"))
 338         def _render(per_server):
 339             if per_server is None:
 340                 return ""
 341             l = T.ul()
 342             for peerid in sorted(per_server.keys()):
 343                 peerid_s = idlib.shortnodeid_b2a(peerid)
 344                 times_s = ", ".join([self.render_time(None, t)
 345                                      for t in per_server[peerid]])
 346                 l[T.li["[%s]: %s" % (peerid_s, times_s)]]
 347             return T.li["Per-Server Segment Fetch Response Times: ", l]
 348         d.addCallback(_render)
 349         return d
 350
 351 class DownloadStatusPage(DownloadResultsRendererMixin, rend.Page):
 352     docFactory = getxmlfile("download-status.xhtml")
 353
 354     def __init__(self, data):
 355         rend.Page.__init__(self, data)
 356         self.download_status = data
 357
 358     def download_results(self):
 359         return defer.maybeDeferred(self.download_status.get_results)
 360
 361     def relative_time(self, t):
 362         if t is None:
 363             return t
 364         if self.download_status.started is not None:
 365             return t - self.download_status.started
 366         return t
 367     def short_relative_time(self, t):
 368         t = self.relative_time(t)
 369         if t is None:
 370             return ""
 371         return "+%.6fs" % t
 372
 373     def renderHTTP(self, ctx):
 374         req = inevow.IRequest(ctx)
 375         t = get_arg(req, "t")
 376         if t == "json":
 377             return self.json(req)
 378         return rend.Page.renderHTTP(self, ctx)
 379
 380     def json(self, req):
 381         req.setHeader("content-type", "text/plain")
 382         data = {}
 383         dyhb_events = []
 384         for serverid,requests in self.download_status.dyhb_requests.iteritems():
 385             for req in requests:
 386                 dyhb_events.append( (base32.b2a(serverid),) + req )
 387         dyhb_events.sort(key=lambda req: req[1])
 388         data["dyhb"] = dyhb_events
 389         request_events = []
 390         for serverid,requests in self.download_status.requests.iteritems():
 391             for req in requests:
 392                 request_events.append( (base32.b2a(serverid),) + req )
 393         request_events.sort(key=lambda req: (req[4],req[1]))
 394         data["requests"] = request_events
 395         data["segment"] = self.download_status.segment_events
 396         data["read"] = self.download_status.read_events
 397         return simplejson.dumps(data, indent=1) + "\n"
 398
 399     def render_events(self, ctx, data):
 400         if not self.download_status.storage_index:
 401             return
 402         srt = self.short_relative_time
 403         l = T.ul()
 404
 405         t = T.table(class_="status-download-events")
 406         t[T.tr[T.td["serverid"], T.td["sent"], T.td["received"],
 407                T.td["shnums"], T.td["RTT"]]]
 408         dyhb_events = []
 409         for serverid,requests in self.download_status.dyhb_requests.iteritems():
 410             for req in requests:
 411                 dyhb_events.append( (serverid,) + req )
 412         dyhb_events.sort(key=lambda req: req[1])
 413         for d_ev in dyhb_events:
 414             (serverid, sent, shnums, received) = d_ev
 415             serverid_s = idlib.shortnodeid_b2a(serverid)
 416             rtt = None
 417             if received is not None:
 418                 rtt = received - sent
 419             if not shnums:
 420                 shnums = []
 421             t[T.tr(style="background: %s" % self.color(serverid))[
 422                 [T.td[serverid_s], T.td[srt(sent)], T.td[srt(received)],
 423                  T.td[",".join([str(shnum) for shnum in shnums])],
 424                  T.td[self.render_time(None, rtt)],
 425                  ]]]
 426         l["DYHB Requests:", t]
 427
 428         t = T.table(class_="status-download-events")
 429         t[T.tr[T.td["range"], T.td["start"], T.td["finish"], T.td["got"],
 430                T.td["time"], T.td["decrypttime"], T.td["pausedtime"],
 431                T.td["speed"]]]
 432         for r_ev in self.download_status.read_events:
 433             (start, length, requesttime, finishtime, bytes, decrypt, paused) = r_ev
 434             if finishtime is not None:
 435                 rtt = finishtime - requesttime - paused
 436                 speed = self.render_rate(None, 1.0 * bytes / rtt)
 437                 rtt = self.render_time(None, rtt)
 438                 decrypt = self.render_time(None, decrypt)
 439                 paused = self.render_time(None, paused)
 440             else:
 441                 speed, rtt, decrypt, paused = "","","",""
 442             t[T.tr[T.td["[%d:+%d]" % (start, length)],
 443                    T.td[srt(requesttime)], T.td[srt(finishtime)],
 444                    T.td[bytes], T.td[rtt], T.td[decrypt], T.td[paused],
 445                    T.td[speed],
 446                    ]]
 447         l["Read Events:", t]
 448
 449         t = T.table(class_="status-download-events")
 450         t[T.tr[T.td["type"], T.td["segnum"], T.td["when"], T.td["range"],
 451                T.td["decodetime"], T.td["segtime"], T.td["speed"]]]
 452         reqtime = (None, None)
 453         for s_ev in self.download_status.segment_events:
 454             (etype, segnum, when, segstart, seglen, decodetime) = s_ev
 455             if etype == "request":
 456                 t[T.tr[T.td["request"], T.td["seg%d" % segnum],
 457                        T.td[srt(when)]]]
 458                 reqtime = (segnum, when)
 459             elif etype == "delivery":
 460                 if reqtime[0] == segnum:
 461                     segtime = when - reqtime[1]
 462                     speed = self.render_rate(None, 1.0 * seglen / segtime)
 463                     segtime = self.render_time(None, segtime)
 464                 else:
 465                     segtime, speed = "", ""
 466                 t[T.tr[T.td["delivery"], T.td["seg%d" % segnum],
 467                        T.td[srt(when)],
 468                        T.td["[%d:+%d]" % (segstart, seglen)],
 469                        T.td[self.render_time(None,decodetime)],
 470                        T.td[segtime], T.td[speed]]]
 471             elif etype == "error":
 472                 t[T.tr[T.td["error"], T.td["seg%d" % segnum]]]
 473         l["Segment Events:", t]
 474
 475         t = T.table(border="1")
 476         t[T.tr[T.td["serverid"], T.td["shnum"], T.td["range"],
 477                T.td["txtime"], T.td["rxtime"], T.td["received"], T.td["RTT"]]]
 478         reqtime = (None, None)
 479         request_events = []
 480         for serverid,requests in self.download_status.requests.iteritems():
 481             for req in requests:
 482                 request_events.append( (serverid,) + req )
 483         request_events.sort(key=lambda req: (req[4],req[1]))
 484         for r_ev in request_events:
 485             (peerid, shnum, start, length, sent, receivedlen, received) = r_ev
 486             rtt = None
 487             if received is not None:
 488                 rtt = received - sent
 489             peerid_s = idlib.shortnodeid_b2a(peerid)
 490             t[T.tr(style="background: %s" % self.color(peerid))[
 491                 T.td[peerid_s], T.td[shnum],
 492                 T.td["[%d:+%d]" % (start, length)],
 493                 T.td[srt(sent)], T.td[srt(received)], T.td[receivedlen],
 494                 T.td[self.render_time(None, rtt)],
 495                 ]]
 496         l["Requests:", t]
 497
 498         return l
 499
 500     def color(self, peerid):
 501         def m(c):
 502             return min(ord(c) / 2 + 0x80, 0xff)
 503         return "#%02x%02x%02x" % (m(peerid[0]), m(peerid[1]), m(peerid[2]))
 504
 505     def render_results(self, ctx, data):
 506         d = self.download_results()
 507         def _got_results(results):
 508             if results:
 509                 return ctx.tag
 510             return ""
 511         d.addCallback(_got_results)
 512         return d
 513
 514     def render_started(self, ctx, data):
 515         TIME_FORMAT = "%H:%M:%S %d-%b-%Y"
 516         started_s = time.strftime(TIME_FORMAT,
 517                                   time.localtime(data.get_started()))
 518         return started_s + " (%s)" % data.get_started()
 519
 520     def render_si(self, ctx, data):
 521         si_s = base32.b2a_or_none(data.get_storage_index())
 522         if si_s is None:
 523             si_s = "(None)"
 524         return si_s
 525
 526     def render_helper(self, ctx, data):
 527         return {True: "Yes",
 528                 False: "No"}[data.using_helper()]
 529
 530     def render_total_size(self, ctx, data):
 531         size = data.get_size()
 532         if size is None:
 533             return "(unknown)"
 534         return size
 535
 536     def render_progress(self, ctx, data):
 537         progress = data.get_progress()
 538         # TODO: make an ascii-art bar
 539         return "%.1f%%" % (100.0 * progress)
 540
 541     def render_status(self, ctx, data):
 542         return data.get_status()
 543
 544 class RetrieveStatusPage(rend.Page, RateAndTimeMixin):
 545     docFactory = getxmlfile("retrieve-status.xhtml")
 546
 547     def __init__(self, data):
 548         rend.Page.__init__(self, data)
 549         self.retrieve_status = data
 550
 551     def render_started(self, ctx, data):
 552         TIME_FORMAT = "%H:%M:%S %d-%b-%Y"
 553         started_s = time.strftime(TIME_FORMAT,
 554                                   time.localtime(data.get_started()))
 555         return started_s
 556
 557     def render_si(self, ctx, data):
 558         si_s = base32.b2a_or_none(data.get_storage_index())
 559         if si_s is None:
 560             si_s = "(None)"
 561         return si_s
 562
 563     def render_helper(self, ctx, data):
 564         return {True: "Yes",
 565                 False: "No"}[data.using_helper()]
 566
 567     def render_current_size(self, ctx, data):
 568         size = data.get_size()
 569         if size is None:
 570             size = "(unknown)"
 571         return size
 572
 573     def render_progress(self, ctx, data):
 574         progress = data.get_progress()
 575         # TODO: make an ascii-art bar
 576         return "%.1f%%" % (100.0 * progress)
 577
 578     def render_status(self, ctx, data):
 579         return data.get_status()
 580
 581     def render_encoding(self, ctx, data):
 582         k, n = data.get_encoding()
 583         return ctx.tag["Encoding: %s of %s" % (k, n)]
 584
 585     def render_problems(self, ctx, data):
 586         problems = data.problems
 587         if not problems:
 588             return ""
 589         l = T.ul()
 590         for peerid in sorted(problems.keys()):
 591             peerid_s = idlib.shortnodeid_b2a(peerid)
 592             l[T.li["[%s]: %s" % (peerid_s, problems[peerid])]]
 593         return ctx.tag["Server Problems:", l]
 594
 595     def _get_rate(self, data, name):
 596         file_size = self.retrieve_status.get_size()
 597         time = self.retrieve_status.timings.get(name)
 598         if time is None or file_size is None:
 599             return None
 600         try:
 601             return 1.0 * file_size / time
 602         except ZeroDivisionError:
 603             return None
 604
 605     def data_time_total(self, ctx, data):
 606         return self.retrieve_status.timings.get("total")
 607     def data_rate_total(self, ctx, data):
 608         return self._get_rate(data, "total")
 609
 610     def data_time_fetch(self, ctx, data):
 611         return self.retrieve_status.timings.get("fetch")
 612     def data_rate_fetch(self, ctx, data):
 613         return self._get_rate(data, "fetch")
 614
 615     def data_time_decode(self, ctx, data):
 616         return self.retrieve_status.timings.get("decode")
 617     def data_rate_decode(self, ctx, data):
 618         return self._get_rate(data, "decode")
 619
 620     def data_time_decrypt(self, ctx, data):
 621         return self.retrieve_status.timings.get("decrypt")
 622     def data_rate_decrypt(self, ctx, data):
 623         return self._get_rate(data, "decrypt")
 624
 625     def render_server_timings(self, ctx, data):
 626         per_server = self.retrieve_status.timings.get("fetch_per_server")
 627         if not per_server:
 628             return ""
 629         l = T.ul()
 630         for peerid in sorted(per_server.keys()):
 631             peerid_s = idlib.shortnodeid_b2a(peerid)
 632             times_s = ", ".join([self.render_time(None, t)
 633                                  for t in per_server[peerid]])
 634             l[T.li["[%s]: %s" % (peerid_s, times_s)]]
 635         return T.li["Per-Server Fetch Response Times: ", l]
 636
 637
 638 class PublishStatusPage(rend.Page, RateAndTimeMixin):
 639     docFactory = getxmlfile("publish-status.xhtml")
 640
 641     def __init__(self, data):
 642         rend.Page.__init__(self, data)
 643         self.publish_status = data
 644
 645     def render_started(self, ctx, data):
 646         TIME_FORMAT = "%H:%M:%S %d-%b-%Y"
 647         started_s = time.strftime(TIME_FORMAT,
 648                                   time.localtime(data.get_started()))
 649         return started_s
 650
 651     def render_si(self, ctx, data):
 652         si_s = base32.b2a_or_none(data.get_storage_index())
 653         if si_s is None:
 654             si_s = "(None)"
 655         return si_s
 656
 657     def render_helper(self, ctx, data):
 658         return {True: "Yes",
 659                 False: "No"}[data.using_helper()]
 660
 661     def render_current_size(self, ctx, data):
 662         size = data.get_size()
 663         if size is None:
 664             size = "(unknown)"
 665         return size
 666
 667     def render_progress(self, ctx, data):
 668         progress = data.get_progress()
 669         # TODO: make an ascii-art bar
 670         return "%.1f%%" % (100.0 * progress)
 671
 672     def render_status(self, ctx, data):
 673         return data.get_status()
 674
 675     def render_encoding(self, ctx, data):
 676         k, n = data.get_encoding()
 677         return ctx.tag["Encoding: %s of %s" % (k, n)]
 678
 679     def render_sharemap(self, ctx, data):
 680         servermap = data.get_servermap()
 681         if servermap is None:
 682             return ctx.tag["None"]
 683         l = T.ul()
 684         sharemap = servermap.make_sharemap()
 685         for shnum in sorted(sharemap.keys()):
 686             l[T.li["%d -> Placed on " % shnum,
 687                    ", ".join(["[%s]" % idlib.shortnodeid_b2a(peerid)
 688                               for peerid in sharemap[shnum]])]]
 689         return ctx.tag["Sharemap:", l]
 690
 691     def render_problems(self, ctx, data):
 692         problems = data.problems
 693         if not problems:
 694             return ""
 695         l = T.ul()
 696         for peerid in sorted(problems.keys()):
 697             peerid_s = idlib.shortnodeid_b2a(peerid)
 698             l[T.li["[%s]: %s" % (peerid_s, problems[peerid])]]
 699         return ctx.tag["Server Problems:", l]
 700
 701     def _get_rate(self, data, name):
 702         file_size = self.publish_status.get_size()
 703         time = self.publish_status.timings.get(name)
 704         if time is None:
 705             return None
 706         try:
 707             return 1.0 * file_size / time
 708         except ZeroDivisionError:
 709             return None
 710
 711     def data_time_total(self, ctx, data):
 712         return self.publish_status.timings.get("total")
 713     def data_rate_total(self, ctx, data):
 714         return self._get_rate(data, "total")
 715
 716     def data_time_setup(self, ctx, data):
 717         return self.publish_status.timings.get("setup")
 718
 719     def data_time_encrypt(self, ctx, data):
 720         return self.publish_status.timings.get("encrypt")
 721     def data_rate_encrypt(self, ctx, data):
 722         return self._get_rate(data, "encrypt")
 723
 724     def data_time_encode(self, ctx, data):
 725         return self.publish_status.timings.get("encode")
 726     def data_rate_encode(self, ctx, data):
 727         return self._get_rate(data, "encode")
 728
 729     def data_time_pack(self, ctx, data):
 730         return self.publish_status.timings.get("pack")
 731     def data_rate_pack(self, ctx, data):
 732         return self._get_rate(data, "pack")
 733     def data_time_sign(self, ctx, data):
 734         return self.publish_status.timings.get("sign")
 735
 736     def data_time_push(self, ctx, data):
 737         return self.publish_status.timings.get("push")
 738     def data_rate_push(self, ctx, data):
 739         return self._get_rate(data, "push")
 740
 741     def render_server_timings(self, ctx, data):
 742         per_server = self.publish_status.timings.get("send_per_server")
 743         if not per_server:
 744             return ""
 745         l = T.ul()
 746         for peerid in sorted(per_server.keys()):
 747             peerid_s = idlib.shortnodeid_b2a(peerid)
 748             times_s = ", ".join([self.render_time(None, t)
 749                                  for t in per_server[peerid]])
 750             l[T.li["[%s]: %s" % (peerid_s, times_s)]]
 751         return T.li["Per-Server Response Times: ", l]
 752
 753 class MapupdateStatusPage(rend.Page, RateAndTimeMixin):
 754     docFactory = getxmlfile("map-update-status.xhtml")
 755
 756     def __init__(self, data):
 757         rend.Page.__init__(self, data)
 758         self.update_status = data
 759
 760     def render_started(self, ctx, data):
 761         TIME_FORMAT = "%H:%M:%S %d-%b-%Y"
 762         started_s = time.strftime(TIME_FORMAT,
 763                                   time.localtime(data.get_started()))
 764         return started_s
 765
 766     def render_finished(self, ctx, data):
 767         when = data.get_finished()
 768         if not when:
 769             return "not yet"
 770         TIME_FORMAT = "%H:%M:%S %d-%b-%Y"
 771         started_s = time.strftime(TIME_FORMAT,
 772                                   time.localtime(data.get_finished()))
 773         return started_s
 774
 775     def render_si(self, ctx, data):
 776         si_s = base32.b2a_or_none(data.get_storage_index())
 777         if si_s is None:
 778             si_s = "(None)"
 779         return si_s
 780
 781     def render_helper(self, ctx, data):
 782         return {True: "Yes",
 783                 False: "No"}[data.using_helper()]
 784
 785     def render_progress(self, ctx, data):
 786         progress = data.get_progress()
 787         # TODO: make an ascii-art bar
 788         return "%.1f%%" % (100.0 * progress)
 789
 790     def render_status(self, ctx, data):
 791         return data.get_status()
 792
 793     def render_problems(self, ctx, data):
 794         problems = data.problems
 795         if not problems:
 796             return ""
 797         l = T.ul()
 798         for peerid in sorted(problems.keys()):
 799             peerid_s = idlib.shortnodeid_b2a(peerid)
 800             l[T.li["[%s]: %s" % (peerid_s, problems[peerid])]]
 801         return ctx.tag["Server Problems:", l]
 802
 803     def render_privkey_from(self, ctx, data):
 804         peerid = data.get_privkey_from()
 805         if peerid:
 806             return ctx.tag["Got privkey from: [%s]"
 807                            % idlib.shortnodeid_b2a(peerid)]
 808         else:
 809             return ""
 810
 811     def data_time_total(self, ctx, data):
 812         return self.update_status.timings.get("total")
 813
 814     def data_time_initial_queries(self, ctx, data):
 815         return self.update_status.timings.get("initial_queries")
 816
 817     def data_time_cumulative_verify(self, ctx, data):
 818         return self.update_status.timings.get("cumulative_verify")
 819
 820     def render_server_timings(self, ctx, data):
 821         per_server = self.update_status.timings.get("per_server")
 822         if not per_server:
 823             return ""
 824         l = T.ul()
 825         for peerid in sorted(per_server.keys()):
 826             peerid_s = idlib.shortnodeid_b2a(peerid)
 827             times = []
 828             for op,started,t in per_server[peerid]:
 829                 #times.append("%s/%.4fs/%s/%s" % (op,
 830                 #                              started,
 831                 #                              self.render_time(None, started - self.update_status.get_started()),
 832                 #                              self.render_time(None,t)))
 833                 if op == "query":
 834                     times.append( self.render_time(None, t) )
 835                 elif op == "late":
 836                     times.append( "late(" + self.render_time(None, t) + ")" )
 837                 else:
 838                     times.append( "privkey(" + self.render_time(None, t) + ")" )
 839             times_s = ", ".join(times)
 840             l[T.li["[%s]: %s" % (peerid_s, times_s)]]
 841         return T.li["Per-Server Response Times: ", l]
 842
 843     def render_timing_chart(self, ctx, data):
 844         imageurl = self._timing_chart()
 845         return ctx.tag[imageurl]
 846
 847     def _timing_chart(self):
 848         started = self.update_status.get_started()
 849         total = self.update_status.timings.get("total")
 850         per_server = self.update_status.timings.get("per_server")
 851         base = "http://chart.apis.google.com/chart?"
 852         pieces = ["cht=bhs", "chs=400x300"]
 853         pieces.append("chco=ffffff,4d89f9,c6d9fd") # colors
 854         data0 = []
 855         data1 = []
 856         data2 = []
 857         peerids_s = []
 858         top_abs = started
 859         # we sort the queries by the time at which we sent the first request
 860         sorttable = [ (times[0][1], peerid)
 861                       for peerid, times in per_server.items() ]
 862         sorttable.sort()
 863         peerids = [t[1] for t in sorttable]
 864
 865         for peerid in peerids:
 866             times = per_server[peerid]
 867             peerid_s = idlib.shortnodeid_b2a(peerid)
 868             peerids_s.append(peerid_s)
 869             # for servermap updates, there are either one or two queries per
 870             # peer. The second (if present) is to get the privkey.
 871             op,q_started,q_elapsed = times[0]
 872             data0.append("%.3f" % (q_started-started))
 873             data1.append("%.3f" % q_elapsed)
 874             top_abs = max(top_abs, q_started+q_elapsed)
 875             if len(times) > 1:
 876                 op,p_started,p_elapsed = times[0]
 877                 data2.append("%.3f" % p_elapsed)
 878                 top_abs = max(top_abs, p_started+p_elapsed)
 879             else:
 880                 data2.append("0.0")
 881         finished = self.update_status.get_finished()
 882         if finished:
 883             top_abs = max(top_abs, finished)
 884         top_rel = top_abs - started
 885         chd = "chd=t:" + "|".join([",".join(data0),
 886                                    ",".join(data1),
 887                                    ",".join(data2)])
 888         pieces.append(chd)
 889         chds = "chds=0,%0.3f" % top_rel
 890         pieces.append(chds)
 891         pieces.append("chxt=x,y")
 892         pieces.append("chxr=0,0.0,%0.3f" % top_rel)
 893         pieces.append("chxl=1:|" + "|".join(reversed(peerids_s)))
 894         # use up to 10 grid lines, at decimal multiples.
 895         # mathutil.next_power_of_k doesn't handle numbers smaller than one,
 896         # unfortunately.
 897         #pieces.append("chg="
 898
 899         if total is not None:
 900             finished_f = 1.0 * total / top_rel
 901             pieces.append("chm=r,FF0000,0,%0.3f,%0.3f" % (finished_f,
 902                                                           finished_f+0.01))
 903         url = base + "&".join(pieces)
 904         return T.img(src=url, align="right", float="right")
 905
 906
 907 class Status(rend.Page):
 908     docFactory = getxmlfile("status.xhtml")
 909     addSlash = True
 910
 911     def __init__(self, history):
 912         rend.Page.__init__(self, history)
 913         self.history = history
 914
 915     def renderHTTP(self, ctx):
 916         req = inevow.IRequest(ctx)
 917         t = get_arg(req, "t")
 918         if t == "json":
 919             return self.json(req)
 920         return rend.Page.renderHTTP(self, ctx)
 921
 922     def json(self, req):
 923         req.setHeader("content-type", "text/plain")
 924         data = {}
 925         data["active"] = active = []
 926         for s in self._get_active_operations():
 927             si_s = base32.b2a_or_none(s.get_storage_index())
 928             size = s.get_size()
 929             status = s.get_status()
 930             if IUploadStatus.providedBy(s):
 931                 h,c,e = s.get_progress()
 932                 active.append({"type": "upload",
 933                                "storage-index-string": si_s,
 934                                "total-size": size,
 935                                "status": status,
 936                                "progress-hash": h,
 937                                "progress-ciphertext": c,
 938                                "progress-encode-push": e,
 939                                })
 940             elif IDownloadStatus.providedBy(s):
 941                 active.append({"type": "download",
 942                                "storage-index-string": si_s,
 943                                "total-size": size,
 944                                "status": status,
 945                                "progress": s.get_progress(),
 946                                })
 947
 948         return simplejson.dumps(data, indent=1) + "\n"
 949
 950     def _get_all_statuses(self):
 951         h = self.history
 952         return itertools.chain(h.list_all_upload_statuses(),
 953                                h.list_all_download_statuses(),
 954                                h.list_all_mapupdate_statuses(),
 955                                h.list_all_publish_statuses(),
 956                                h.list_all_retrieve_statuses(),
 957                                h.list_all_helper_statuses(),
 958                                )
 959
 960     def data_active_operations(self, ctx, data):
 961         return self._get_active_operations()
 962
 963     def _get_active_operations(self):
 964         active = [s
 965                   for s in self._get_all_statuses()
 966                   if s.get_active()]
 967         return active
 968
 969     def data_recent_operations(self, ctx, data):
 970         return self._get_recent_operations()
 971
 972     def _get_recent_operations(self):
 973         recent = [s
 974                   for s in self._get_all_statuses()
 975                   if not s.get_active()]
 976         recent.sort(lambda a,b: cmp(a.get_started(), b.get_started()))
 977         recent.reverse()
 978         return recent
 979
 980     def render_row(self, ctx, data):
 981         s = data
 982
 983         TIME_FORMAT = "%H:%M:%S %d-%b-%Y"
 984         started_s = time.strftime(TIME_FORMAT,
 985                                   time.localtime(s.get_started()))
 986         ctx.fillSlots("started", started_s)
 987
 988         si_s = base32.b2a_or_none(s.get_storage_index())
 989         if si_s is None:
 990             si_s = "(None)"
 991         ctx.fillSlots("si", si_s)
 992         ctx.fillSlots("helper", {True: "Yes",
 993                                  False: "No"}[s.using_helper()])
 994
 995         size = s.get_size()
 996         if size is None:
 997             size = "(unknown)"
 998         elif isinstance(size, (int, long, float)):
 999             size = abbreviate_size(size)
1000         ctx.fillSlots("total_size", size)
1001
1002         progress = data.get_progress()
1003         if IUploadStatus.providedBy(data):
1004             link = "up-%d" % data.get_counter()
1005             ctx.fillSlots("type", "upload")
1006             # TODO: make an ascii-art bar
1007             (chk, ciphertext, encandpush) = progress
1008             progress_s = ("hash: %.1f%%, ciphertext: %.1f%%, encode: %.1f%%" %
1009                           ( (100.0 * chk),
1010                             (100.0 * ciphertext),
1011                             (100.0 * encandpush) ))
1012             ctx.fillSlots("progress", progress_s)
1013         elif IDownloadStatus.providedBy(data):
1014             link = "down-%d" % data.get_counter()
1015             ctx.fillSlots("type", "download")
1016             ctx.fillSlots("progress", "%.1f%%" % (100.0 * progress))
1017         elif IPublishStatus.providedBy(data):
1018             link = "publish-%d" % data.get_counter()
1019             ctx.fillSlots("type", "publish")
1020             ctx.fillSlots("progress", "%.1f%%" % (100.0 * progress))
1021         elif IRetrieveStatus.providedBy(data):
1022             ctx.fillSlots("type", "retrieve")
1023             link = "retrieve-%d" % data.get_counter()
1024             ctx.fillSlots("progress", "%.1f%%" % (100.0 * progress))
1025         else:
1026             assert IServermapUpdaterStatus.providedBy(data)
1027             ctx.fillSlots("type", "mapupdate %s" % data.get_mode())
1028             link = "mapupdate-%d" % data.get_counter()
1029             ctx.fillSlots("progress", "%.1f%%" % (100.0 * progress))
1030         ctx.fillSlots("status", T.a(href=link)[s.get_status()])
1031         return ctx.tag
1032
1033     def childFactory(self, ctx, name):
1034         h = self.history
1035         stype,count_s = name.split("-")
1036         count = int(count_s)
1037         if stype == "up":
1038             for s in itertools.chain(h.list_all_upload_statuses(),
1039                                      h.list_all_helper_statuses()):
1040                 # immutable-upload helpers use the same status object as a
1041                 # regular immutable-upload
1042                 if s.get_counter() == count:
1043                     return UploadStatusPage(s)
1044         if stype == "down":
1045             for s in h.list_all_download_statuses():
1046                 if s.get_counter() == count:
1047                     return DownloadStatusPage(s)
1048         if stype == "mapupdate":
1049             for s in h.list_all_mapupdate_statuses():
1050                 if s.get_counter() == count:
1051                     return MapupdateStatusPage(s)
1052         if stype == "publish":
1053             for s in h.list_all_publish_statuses():
1054                 if s.get_counter() == count:
1055                     return PublishStatusPage(s)
1056         if stype == "retrieve":
1057             for s in h.list_all_retrieve_statuses():
1058                 if s.get_counter() == count:
1059                     return RetrieveStatusPage(s)
1060
1061
1062 class HelperStatus(rend.Page):
1063     docFactory = getxmlfile("helper.xhtml")
1064
1065     def __init__(self, helper):
1066         rend.Page.__init__(self, helper)
1067         self.helper = helper
1068
1069     def renderHTTP(self, ctx):
1070         req = inevow.IRequest(ctx)
1071         t = get_arg(req, "t")
1072         if t == "json":
1073             return self.render_JSON(req)
1074         return rend.Page.renderHTTP(self, ctx)
1075
1076     def data_helper_stats(self, ctx, data):
1077         return self.helper.get_stats()
1078
1079     def render_JSON(self, req):
1080         req.setHeader("content-type", "text/plain")
1081         if self.helper:
1082             stats = self.helper.get_stats()
1083             return simplejson.dumps(stats, indent=1) + "\n"
1084         return simplejson.dumps({}) + "\n"
1085
1086     def render_active_uploads(self, ctx, data):
1087         return data["chk_upload_helper.active_uploads"]
1088
1089     def render_incoming(self, ctx, data):
1090         return "%d bytes in %d files" % (data["chk_upload_helper.incoming_size"],
1091                                          data["chk_upload_helper.incoming_count"])
1092
1093     def render_encoding(self, ctx, data):
1094         return "%d bytes in %d files" % (data["chk_upload_helper.encoding_size"],
1095                                          data["chk_upload_helper.encoding_count"])
1096
1097     def render_upload_requests(self, ctx, data):
1098         return str(data["chk_upload_helper.upload_requests"])
1099
1100     def render_upload_already_present(self, ctx, data):
1101         return str(data["chk_upload_helper.upload_already_present"])
1102
1103     def render_upload_need_upload(self, ctx, data):
1104         return str(data["chk_upload_helper.upload_need_upload"])
1105
1106     def render_upload_bytes_fetched(self, ctx, data):
1107         return str(data["chk_upload_helper.fetched_bytes"])
1108
1109     def render_upload_bytes_encoded(self, ctx, data):
1110         return str(data["chk_upload_helper.encoded_bytes"])
1111
1112
1113 class Statistics(rend.Page):
1114     docFactory = getxmlfile("statistics.xhtml")
1115
1116     def __init__(self, provider):
1117         rend.Page.__init__(self, provider)
1118         self.provider = provider
1119
1120     def renderHTTP(self, ctx):
1121         req = inevow.IRequest(ctx)
1122         t = get_arg(req, "t")
1123         if t == "json":
1124             stats = self.provider.get_stats()
1125             req.setHeader("content-type", "text/plain")
1126             return simplejson.dumps(stats, indent=1) + "\n"
1127         return rend.Page.renderHTTP(self, ctx)
1128
1129     def data_get_stats(self, ctx, data):
1130         return self.provider.get_stats()
1131
1132     def render_load_average(self, ctx, data):
1133         return str(data["stats"].get("load_monitor.avg_load"))
1134
1135     def render_peak_load(self, ctx, data):
1136         return str(data["stats"].get("load_monitor.max_load"))
1137
1138     def render_uploads(self, ctx, data):
1139         files = data["counters"].get("uploader.files_uploaded", 0)
1140         bytes = data["counters"].get("uploader.bytes_uploaded", 0)
1141         return ("%s files / %s bytes (%s)" %
1142                 (files, bytes, abbreviate_size(bytes)))
1143
1144     def render_downloads(self, ctx, data):
1145         files = data["counters"].get("downloader.files_downloaded", 0)
1146         bytes = data["counters"].get("downloader.bytes_downloaded", 0)
1147         return ("%s files / %s bytes (%s)" %
1148                 (files, bytes, abbreviate_size(bytes)))
1149
1150     def render_publishes(self, ctx, data):
1151         files = data["counters"].get("mutable.files_published", 0)
1152         bytes = data["counters"].get("mutable.bytes_published", 0)
1153         return "%s files / %s bytes (%s)" % (files, bytes,
1154                                              abbreviate_size(bytes))
1155
1156     def render_retrieves(self, ctx, data):
1157         files = data["counters"].get("mutable.files_retrieved", 0)
1158         bytes = data["counters"].get("mutable.bytes_retrieved", 0)
1159         return "%s files / %s bytes (%s)" % (files, bytes,
1160                                              abbreviate_size(bytes))
1161
1162     def render_raw(self, ctx, data):
1163         raw = pprint.pformat(data)
1164         return ctx.tag[raw]