Convert hostname to string
[sface.git] / sface / screens / mainscreen.py
index 00710a3..cc34adb 100644 (file)
 
+import datetime
 import os
 from PyQt4.QtCore import *
 from PyQt4.QtGui import *
 
-from sfa.util.rspecHelper import RSpec
-from sface.sfahelper import *
+#from sfa.util.rspecHelper import RSpec
+from sfa.rspecs.rspec_parser import parse_rspec
 from sface.config import config
+from sface.sfirenew import SfiRenewer
 from sface.sfiprocess import SfiProcess
 from sface.screens.sfascreen import SfaScreen
 
+already_in_nodes = []
+
+node_status = { "in": "Already Selected",
+                "out": "Not Selected",
+                "add": "To be Added",
+                "remove": "To be Removed"}
+
+tag_status = { "in": "Already Set",
+                "out": "Not Set",
+                "add": "To be Added",
+                "remove": "To be Removed"}
+
+default_tags = "Default tags"
+settable_tags = ['delegations', 'initscript']
+
+def itemType(index):
+    if index.parent().parent().isValid():
+        return "tag"
+    else:
+        return "node"
+
+
 class NodeView(QTreeView):
     def __init__(self, parent):
         QTreeView.__init__(self, parent)
 
+        self.setAnimated(True)
         self.setItemsExpandable(True)
         self.setRootIsDecorated(True)
         self.setAlternatingRowColors(True)
-        self.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Expanding)
+#        self.setSelectionMode(self.MultiSelection)
         self.setAttribute(Qt.WA_MacShowFocusRect, 0)
+        self.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Expanding)
+        self.setToolTip("Double click on a row to change its status.  Right click on a host to add a tag.")
+
+    def mouseDoubleClickEvent(self, event):
+        index = self.currentIndex()
+        model = index.model()
+        status_index = model.index(index.row(), 1, index.parent())
+        status_data = status_index.data().toString()
+        node_index = model.index(index.row(), 0, index.parent())
+        node_data = node_index.data().toString()
+
+        if itemType(node_index) == "tag":
+            data = node_index.data().toString()
+            tagname, value = data.split(": ")
+            if tagname not in settable_tags:
+                # Pop up error msg
+                QMessageBox.warning(self, "Not settable", "Insufficient permission to change '%s' tag" % tagname)
+                return
+            if status_data == tag_status['in']:
+                model.setData(status_index, QString(tag_status['remove']))
+            elif status_data == tag_status['add']:
+                model.setData(status_index, QString(tag_status['out']))
+            elif status_data == tag_status['remove']:
+                model.setData(status_index, QString(tag_status['in']))
+            else: model.setData(status_index, QString(node_status['out']))
+        else:
+            # This is a hostname
+            if status_data == node_status['in']:
+                model.setData(status_index, QString(node_status['remove']))
+            elif status_data == node_status['out']:
+                model.setData(status_index, QString(node_status['add']))
+            elif status_data in (node_status['add'], node_status['remove']):
+                if node_data in already_in_nodes: model.setData(status_index, QString(node_status['in']))
+                else: model.setData(status_index, QString(node_status['out']))
+
+        model.emit(SIGNAL("dataChanged(QModelIndex, QModelIndex)"), node_index, node_index)
+
+    def mousePressEvent(self, event):
+        QTreeView.mousePressEvent(self, event)
+        if event.button() == Qt.LeftButton:
+            return
+
+        # Right click
+        index = self.currentIndex()
+        model = index.model()
+        status_index = model.index(index.row(), 1, index.parent())
+        status_data = status_index.data().toString()
+        node_index = model.index(index.row(), 0, index.parent())
+        node_data = node_index.data().toString()
+
+        if itemType(node_index) == "node":
+            # This is a hostname
+            if status_data in (node_status['in'], node_status['add'], ""):
+                # Pop up a dialog box for adding a new attribute
+                tagname, ok = QInputDialog.getItem(self, "Add tag",
+                                                   "Tag name:", settable_tags)
+                if ok:
+                    value, ok = QInputDialog.getText(self, "Add tag",
+                                                     "Value for tag '%s'" % tagname)
+                    if ok:
+                        # Add a new row to the model for the tag
+
+                        # For testing with the QStandardItemModel
+                        #nodeItem = model.itemFromIndex(index)
+                        #tagstring = QString("%s: %s" % (tagname, value))
+                        #tagItem = QStandardItem(tagstring)
+                        #status = QStandardItem(QString(tag_status['add']))
+                        #nodeItem.appendRow([tagItem, status])
+
+                        # We're using the QSortFilterProxyModel here
+                        src_index = model.mapToSource(index)
+                        src_model = src_index.model()
+                        nodeItem = src_model.itemFromIndex(src_index)
+                        tagstring = QString("%s: %s" % (tagname, value))
+                        tagItem = QStandardItem(tagstring)
+                        status = QStandardItem(QString(tag_status['add']))
+                        nodeItem.appendRow([tagItem, status])
+
+            elif status_data in (node_status['out'], node_status['remove']):
+                QMessageBox.warning(self, "Not selected", "Can only add tags to selected nodes")
+                return
+
+        model.emit(SIGNAL("dataChanged(QModelIndex, QModelIndex)"), node_index, node_index)
+
+    def currentChanged(self, current, previous):
+        model = current.model()
+        node_index = model.index(current.row(), 0, current.parent())
+        node_data = node_index.data().toString()
+        self.emit(SIGNAL('hostnameClicked(QString)'), node_data)
+        
+                
 
-class ItemDelegate(QStyledItemDelegate):
+class NodeNameDelegate(QStyledItemDelegate):
     def __init__(self, parent):
-        QStyledItemDelegate.__init__(self)
+        QStyledItemDelegate.__init__(self, parent)
 
     def paint(self, painter, option, index):
-        data = "%s" % index.data().toString()
-
-        if data.startswith("*"): # already in the sliver
-            data = " %s " % data[1:]
-
-            rect = option.rect
-            fm = QFontMetrics(option.font)
-            w = fm.width(QString(data))
-            rect.setWidth(w)
-
-            painter.save()
-            painter.fillRect(rect, QColor.fromRgb(0, 250, 0))
-            painter.setPen(QColor.fromRgb(0, 0, 0))
-            painter.drawText(option.rect, 0, QString(data))
-            painter.restore()
-        else: # others, fall back to default view
-            QStyledItemDelegate.paint(self, painter, option, index)
-
-class TreeItem:
-    def __init__(self, data, parent=None):
-        self.parentItem = parent
-        self.itemData = data
-        self.childItems = []
-
-    def clear(self):
-        for child in self.childItems:
-            child.clear()
-            del child
-        del self.childItems
-        self.childItems = []
-
-    def appendChild(self, child):
-        self.childItems.append(child)
-
-    def child(self, row):
-        return self.childItems[row]
-    
-    def childCount(self):
-        return len(self.childItems)
-
-    def columnCount(self):
-        return len(self.itemData)
-
-    def data(self, column):
-        return self.itemData[column]
-
-    def row(self):
-        if (self.parentItem):
-            try:
-                return self.parentItem.childItems.index(self)
-            except ValueError:
-                return 0
-        return 0
-    
-    def parent(self):
-        return self.parentItem
-
-
-
-class NodeModel(QAbstractItemModel):
-    def __init__(self, parent):
-        QAbstractItemModel.__init__(self, parent)
-        self.__initRoot()
+        model = index.model()
+        status_index = model.index(index.row(), 1, index.parent())
+        status_data = status_index.data().toString()
 
-    def clear(self):
-        self.rootItem.clear()
-        self.__initRoot()
+        fm = QFontMetrics(option.font)
+        rect = option.rect
 
-    def __initRoot(self):
-        self.rootItem = TreeItem([QString("Testbed"), QString("Hostname"), QString("Selected")])
+        data = index.data().toString()
+        rect.setHeight(rect.height() - 2)
+        rect.setWidth(fm.width(QString(data)) + 6)
+        rect.setX(rect.x() + 5)
+        rect.setY(rect.y() - 1)
 
-    def columnCount(self, parent):
-        if parent.isValid():
-            return parent.internalPointer().columnCount()
-        else:
-            return self.rootItem.columnCount()
+        x, y, h, w = rect.x(), rect.y(), rect.height(), rect.width()
 
-    def data(self, index, role):
-        if not index.isValid():
-            return QVariant()
+        path = QPainterPath()
+        path.addRoundedRect(x - 1, y + 1, w, h, 4, 4)
 
-        if role != Qt.DisplayRole:
-            return QVariant()
+        painter.save()
+        painter.setRenderHint(QPainter.Antialiasing)
 
-        item = index.internalPointer()
-        return item.data(index.column())
+        if itemType(index) == "node":
+            if status_data == node_status['in']: # already in the slice
+                painter.fillPath(path, QColor.fromRgb(0, 250, 250))
+                painter.setPen(QColor.fromRgb(0, 0, 0))
+                painter.drawText(option.rect, 0, QString(data))
 
-    def headerData(self, section, orientation, role):
-        if orientation == Qt.Horizontal and role == Qt.DisplayRole:
-            return self.rootItem.data(section)
-        return QVariant()
+            elif status_data == node_status['add']: # newly added to the slice
+                painter.fillPath(path, QColor.fromRgb(0, 250, 0))
+                painter.setPen(QColor.fromRgb(0, 0, 0))
+                painter.drawText(option.rect, 0, QString(data))
 
-    def flags(self, index):
-        if not index.isValid():
-            return 0
-        return Qt.ItemIsEnabled | Qt.ItemIsSelectable # | Qt.ItemIsEditable
+            elif status_data == node_status['remove']: # removed from the slice
+                painter.fillPath(path, QColor.fromRgb(250, 0, 0))
+                painter.setPen(QColor.fromRgb(0, 0, 0))
+                painter.drawText(option.rect, 0, QString(data))
 
-    def index(self, row, column, parent):
-        if not self.hasIndex(row, column, parent):
-            return QModelIndex()
+            else:
+                painter.setPen(QColor.fromRgb(0, 0, 0))
+                painter.drawText(option.rect, 0, QString(data))
 
-        parentItem = None
-        if not parent.isValid():
-            parentItem = self.rootItem
         else:
-            parentItem = parent.internalPointer()
-            
-        childItem = parentItem.child(row)
-        if childItem:
-            return self.createIndex(row, column, childItem)
-        else:
-            return QModelIndex()
-
-    def parent(self, index):
-        if not index.isValid():
-            return QModelIndex()
+            if status_data == tag_status['in']: # already in the slice
+                painter.fillPath(path, QColor.fromRgb(0, 250, 250))
+                painter.setPen(QColor.fromRgb(0, 0, 0))
+                painter.drawText(option.rect, 0, QString(data))
 
-        childItem = index.internalPointer()
-        parentItem = childItem.parent()
+            elif status_data == tag_status['add']: # newly added to the slice
+                painter.fillPath(path, QColor.fromRgb(0, 250, 0))
+                painter.setPen(QColor.fromRgb(0, 0, 0))
+                painter.drawText(option.rect, 0, QString(data))
 
-        if not parentItem:
-            return QModelIndex()
+            elif status_data == tag_status['remove']: # removed from the slice
+                painter.fillPath(path, QColor.fromRgb(250, 0, 0))
+                painter.setPen(QColor.fromRgb(0, 0, 0))
+                painter.drawText(option.rect, 0, QString(data))
 
-        if parentItem is self.rootItem:
-            return QModelIndex()
-
-        return self.createIndex(parentItem.row(), 0, parentItem)
-
-    def rowCount(self, parent):
-        if parent.column() > 0:
-            return 0
-
-        parentItem = None
-        if not parent.isValid():
-            parentItem = self.rootItem
-        else:
-            parentItem = parent.internalPointer()
-
-        return parentItem.childCount()
+            else:
+                painter.setPen(QColor.fromRgb(0, 0, 0))
+                painter.drawText(option.rect, 0, QString(data))
 
+        painter.restore()
 
 class SliceWidget(QWidget):
     def __init__(self, parent):
         QWidget.__init__(self, parent)
 
-        refresh = QLabel("<a href='refresh'>Update Slice Data</a>", self)
-        refresh.setScaledContents(False)
-        slicename = QLabel ("Slice : %s"%(config.getSlice() or "None"),self)
-        slicename.setScaledContents(False)
+        self.network_names = []
+        self.process = SfiProcess(self)
 
-        hlayout = QHBoxLayout()
-        hlayout.addWidget(slicename)
-        hlayout.addStretch()
-        hlayout.addWidget(refresh)
+        self.slicename = QLabel("", self)
+        self.updateSliceName()
+        self.slicename.setScaledContents(False)
+        searchlabel = QLabel ("Search: ", self)
+        searchlabel.setScaledContents(False)
+        searchbox = QLineEdit(self)
+        searchbox.setAttribute(Qt.WA_MacShowFocusRect, 0)
+
+        toplayout = QHBoxLayout()
+        toplayout.addWidget(self.slicename, 0, Qt.AlignLeft)
+        toplayout.addStretch()
+        toplayout.addWidget(searchlabel, 0, Qt.AlignRight)
+        toplayout.addWidget(searchbox, 0, Qt.AlignRight)
 
         self.nodeView = NodeView(self)
-        self.nodeModel = NodeModel(self)
-        self.nodeView.setModel(self.nodeModel)
+        self.nodeModel = QStandardItemModel(0, 2, self)
+        self.filterModel = QSortFilterProxyModel(self) # enable filtering
+
+        self.nodeNameDelegate = NodeNameDelegate(self)
 
+        refresh = QPushButton("Update Slice Data", self)
+        refresh.setSizePolicy(QSizePolicy.Maximum, QSizePolicy.Maximum)
+        renew = QPushButton("Renew Slice", self)
+        renew.setSizePolicy(QSizePolicy.Maximum, QSizePolicy.Maximum)
         submit = QPushButton("Submit", self)
         submit.setSizePolicy(QSizePolicy.Maximum, QSizePolicy.Maximum)
 
+        bottomlayout = QHBoxLayout()
+        bottomlayout.addWidget(refresh, 0, Qt.AlignLeft)
+        bottomlayout.addWidget(renew, 0, Qt.AlignLeft)
+        bottomlayout.addStretch()
+        bottomlayout.addWidget(submit, 0, Qt.AlignRight)
+
         layout = QVBoxLayout()
-        layout.addLayout(hlayout)
+        layout.addLayout(toplayout)
         layout.addWidget(self.nodeView)
-        layout.addWidget(submit, 0, Qt.AlignRight)
+        layout.addLayout(bottomlayout)
         self.setLayout(layout)
         self.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Expanding)
 
-        self.connect(refresh, SIGNAL('linkActivated(QString)'), self.refresh)
+        self.connect(refresh, SIGNAL('clicked()'), self.refresh)
+        self.connect(renew, SIGNAL('clicked()'), self.renew)
         self.connect(submit, SIGNAL('clicked()'), self.submit)
+        self.connect(searchbox, SIGNAL('textChanged(QString)'), self.filter)
+        self.connect(self.nodeView, SIGNAL('hostnameClicked(QString)'),
+                     self.nodeSelectionChanged)
 
-        rspec_file = os.path.expanduser("~/.sfi/%s.rspec" % config.getSlice())
-        if os.path.exists(rspec_file):
-            self.updateView()
+        self.updateView()
+
+    def submitFinished(self):
+        self.setStatus("<font color='green'>Slice data submitted.</font>")
+        QTimer.singleShot(1000, self.refresh)
+
+    def refreshFinished(self):
+        self.setStatus("<font color='green'>Slice data updated.</font>", timeout=5000)
+        self.updateView()
+        self.parent().signalAll("rspecUpdated")
 
-    def submit(self):
-        self.parent().setStatus("TODO: Submit not implemented yet!", 3000)
-        
     def readSliceRSpec(self):
         rspec_file = config.getSliceRSpecFile()
-        xml = open(rspec_file).read()
-        return xml
+        if os.path.exists(rspec_file):
+            xml = open(rspec_file).read()
+            return parse_rspec(xml)
+        return None
+
+    def setStatus(self, msg, timeout=None):
+        self.parent().setStatus(msg, timeout)
+
+    def checkRunningProcess(self):
+        if self.process.isRunning():
+            self.setStatus("<font color='red'>There is already a process running. Please wait.</font>")
+            return True
+        return False
+
+    def filter(self, filter_string):
+        # for hierarchical models QSortFilterProxyModel applies the
+        # sort recursively. if the parent doesn't match the criteria
+        # we won't be able to match the children. so we need to match
+        # parent (by matching the network_names)
+        networks = ["^%s$" % n for n in self.network_names]
+        filters = networks + [str(filter_string)]
+        self.filterModel.setFilterRegExp(QRegExp('|'.join(filters)))
+
+    def itemStatus(self, item):
+        statusItem = item.parent().child(item.row(), 1)
+        return statusItem.data(Qt.DisplayRole).toString()
+
+    def itemText(self, item):
+        return item.data(Qt.DisplayRole).toString()
+
+    # Recursively walk the tree, making changes to the RSpec
+    def process_subtree(self, rspec, item, depth = 0):
+        change = False
+        model = self.nodeModel
+
+        if depth in [0, 1]:
+            pass
+        elif depth == 2: # Hostname
+            hostname = self.itemText(item)
+            testbed = self.itemText(item.parent())
+            status = self.itemStatus(item)
+            if status == node_status['add']:
+                print "Add hostname: %s" % hostname
+                rspec.add_slivers(str(hostname), testbed)
+                change = True
+            elif status == node_status['remove']:
+                print "Remove hostname: %s" % hostname
+                rspec.remove_slivers(str(hostname), testbed)
+                change = True
+        elif depth == 3: # Tag
+            tag, value = self.itemText(item).split(": ")
+            status = self.itemStatus(item)
+            tag = "%s" % tag     # Prevent weird error from lxml
+            value = "%s" % value # Prevent weird error from lxml
+            node = self.itemText(item.parent())
+            testbed = self.itemText(item.parent().parent())
+            if status == tag_status['add']:
+                print "Add tag to (%s, %s): %s/%s " % (testbed, node, tag, value)
+                if node.startsWith(default_tags):
+                    rspec.add_default_sliver_attribute(tag, value, testbed)
+                else:
+                    rspec.add_sliver_attribute(node, tag, value, testbed)
+                change = True
+            elif status == tag_status['remove']:
+                print "Remove tag from (%s, %s): %s/%s " % (testbed, node, tag, value)
+                if node.startsWith(default_tags):
+                    rspec.remove_default_sliver_attribute(tag, value, testbed)
+                else:
+                    rspec.remove_sliver_attribute(node, tag, value, testbed)
+                change = True
+
+        children = item.rowCount()
+        for row in range(0, children):
+            status = self.process_subtree(rspec, item.child(row), depth + 1)
+            change = change or status
+
+        return change
+
+    def submit(self):
+        if self.checkRunningProcess():
+            return
+
+        rspec = self.readSliceRSpec()
+        change = self.process_subtree(rspec, self.nodeModel.invisibleRootItem())
+
+        if not change:
+            self.setStatus("<font color=red>No change in slice data. Not submitting!</font>", timeout=3000)
+            return
+
+        self.disconnect(self.process, SIGNAL('finished()'), self.refreshFinished)
+        self.connect(self.process, SIGNAL('finished()'), self.submitFinished)
+
+        self.process.applyRSpec(rspec)
+        self.setStatus("Sending slice data (RSpec). This will take some time...")
+
+    def renew(self):
+        dlg = RenewWindow(parent=self)
+        if (dlg.exec_() == QDialog.Accepted):
+            self.setStatus("Renewing Slice.")
+
+            self.renewProcess = SfiRenewer(config.getSlice(), dlg.get_new_expiration(), self)
+            self.connect(self.renewProcess, SIGNAL('finished()'), self.renewFinished)
+
+    def renewFinished(self):
+        if self.renewProcess.statusMsg:
+            self.setStatus("Renew " + self.renewProcess.status + ": " + self.renewProcess.statusMsg)
+        else:
+            self.setStatus("Renew " + self.renewProcess.status)
+        self.disconnect(self.renewProcess, SIGNAL('finished()'), self.renewFinished)
+        self.renewProcess = None
 
-    def refresh(self, link=None):
+    def refresh(self):
         if not config.getSlice():
-            self.parent().setStatus("<font color='red'>Slice not set yet!</font>", timeout=None)
+            self.setStatus("<font color='red'>Slice not set yet!</font>")
             return
 
-        self.process = SfiProcess()
-        outfile = self.process.getRSpecFromSM()
-        self.parent().setStatus("Updating slice data. This may take some time...", timeout=None)
-        
+        if self.process.isRunning():
+            self.setStatus("<font color='red'>There is already a process running. Please wait.</font>")
+            return
+
+        self.disconnect(self.process, SIGNAL('finished()'), self.submitFinished)
         self.connect(self.process, SIGNAL('finished()'), self.refreshFinished)
 
-    def refreshFinished(self):
-        del self.process
-        self.parent().setStatus("<font color='green'>Slice data updated.</font>", timeout=5000)
-        self.updateView()
+        self.process.getRSpecFromSM()
+        self.setStatus("Updating slice data. This will take some time...")
 
     def updateView(self):
+        global already_in_nodes
+        already_in_nodes = []
+        self.network_names = []
         self.nodeModel.clear()
-        rspec_string = self.readSliceRSpec()
-        networks = rspec_get_networks(rspec_string)
 
-        for network in networks:
-            networkItem = TreeItem([QString(network), QString(""), QString("")], self.nodeModel.rootItem)
+        rspec = self.readSliceRSpec()
+        if not rspec:
+            return None
 
-            all_nodes = rspec_get_nodes_from_network(rspec_string, network)
-            sliver_nodes = rspec_get_sliver_nodes_from_network(rspec_string, network)
-            available_nodes = filter(lambda x:x not in sliver_nodes, all_nodes)
+        rootItem = self.nodeModel.invisibleRootItem()
+        #networks = sorted(rspec.get_network_list())
+        networks = rspec.get_networks()
+        for network in networks:
+            self.network_names.append(network)
+
+            #all_nodes = rspec.get_node_list(network)
+            #sliver_nodes = rspec.get_sliver_list(network)
+            all_nodes = rspec.get_nodes(network)
+            sliver_nodes = rspec.get_nodes_with_slivers(network)
+            available_nodes = [ node for node in all_nodes if node not in sliver_nodes ]
+
+            networkItem = QStandardItem(QString(network))
+            msg = "%s Nodes\t%s Selected" % (len(all_nodes), len(sliver_nodes))
+            rootItem.appendRow([networkItem, QStandardItem(QString(msg))])
+
+            already_in_nodes += sliver_nodes
+
+            # Add default slice tags
+            nodeItem = QStandardItem(QString("%s for %s" % (default_tags, network)))
+            statusItem = QStandardItem(QString(""))
+            networkItem.appendRow([nodeItem, statusItem])
+            attrs = rspec.get_default_sliver_attributes(network)
+            for (name, value) in attrs:
+                    tagstring = QString("%s: %s" % (name, value))
+                    tagItem = QStandardItem(tagstring)
+                    status = QStandardItem(QString(tag_status['in']))
+                    nodeItem.appendRow([tagItem, status])
 
             for node in sliver_nodes:
-                nodeItem = TreeItem([QString(""), QString("*%s" % node), QString("Implement Checkbox")], networkItem)
-                networkItem.appendChild(nodeItem)
+                nodeItem = QStandardItem(QString(node))
+                statusItem = QStandardItem(QString(node_status['in']))
+                networkItem.appendRow([nodeItem, statusItem])
+
+                attrs = rspec.get_sliver_attributes(node, network)
+                for (name, value) in attrs:
+                    tagstring = QString("%s: %s" % (name, value))
+                    tagItem = QStandardItem(tagstring)
+                    statusItem = QStandardItem(QString(tag_status['in']))
+                    nodeItem.appendRow([tagItem, statusItem])
 
             for node in available_nodes:
-                nodeItem = TreeItem([QString(""), QString(node), QString("Implement Checkbox")], networkItem)
-                networkItem.appendChild(nodeItem)
+                nodeItem = QStandardItem(QString(node))
+                statusItem = QStandardItem(QString(node_status['out']))
+                networkItem.appendRow([nodeItem, statusItem])
 
-            self.nodeModel.rootItem.appendChild(networkItem)
+        self.filterModel.setSourceModel(self.nodeModel)
+        self.filterModel.setFilterKeyColumn(-1)
+        self.filterModel.setDynamicSortFilter(True)
 
+        headers = QStringList() << "Hostname or Tag" << "Status"
+        self.nodeModel.setHorizontalHeaderLabels(headers)
+
+        self.nodeView.setItemDelegateForColumn(0, self.nodeNameDelegate)
+        self.nodeView.setModel(self.filterModel)
         self.nodeView.expandAll()
-        self.nodeView.resizeColumnToContents(1)
+        self.nodeView.resizeColumnToContents(0)
+        self.nodeView.collapseAll()
+
+    def updateSliceName(self):
+        self.slicename.setText("Slice : %s" % (config.getSlice() or "None"))
+
+    def nodeSelectionChanged(self, hostname):
+        self.parent().nodeSelectionChanged(hostname)
+
+class RenewWindow(QDialog):
+    def __init__(self, parent=None):
+        super(RenewWindow, self).__init__(parent)
+        self.setWindowTitle("Renew Slivers")
+
+        self.duration = QComboBox()
 
-        self.delegate = ItemDelegate(self)
-        self.nodeView.setItemDelegateForColumn(1, self.delegate)
+        self.expirations = []
 
+        durations = ( (1, "One Week"), (2, "Two Weeks"), (3, "Three Weeks"), (4, "One Month") )
+
+        now = datetime.datetime.utcnow()
+        for (weeks, desc) in durations:
+            exp = now + datetime.timedelta(days = weeks * 7)
+            desc = desc + " " + exp.strftime("%Y-%m-%d %H:%M:%S")
+            self.expirations.append(exp)
+            self.duration.addItem(desc)
+
+        self.duration.setCurrentIndex(0)
+
+        buttonBox = QDialogButtonBox(QDialogButtonBox.Ok | QDialogButtonBox.Cancel)
+        buttonBox.button(QDialogButtonBox.Ok).setDefault(True)
+
+        layout = QVBoxLayout()
+        layout.addWidget(self.duration)
+        layout.addWidget(buttonBox)
+        self.setLayout(layout)
+
+        self.connect(buttonBox, SIGNAL("accepted()"), self, SLOT("accept()"))
+        self.connect(buttonBox, SIGNAL("rejected()"), self, SLOT("reject()"))
+
+    def accept(self):
+        QDialog.accept(self)
+
+    def get_new_expiration(self):
+        index = self.duration.currentIndex()
+        return self.expirations[index]
 
 class MainScreen(SfaScreen):
     def __init__(self, parent):
         SfaScreen.__init__(self, parent)
 
         slice = SliceWidget(self)
-        self.init(slice, "Main Window", "OneLab Federation GUI")
+        self.init(slice, "Main Window", "OneLab SFA crawler")
+
+    def rspecUpdated(self):
+        self.mainwin.rspecWindow.updateView()
+
+    def configurationChanged(self):
+        self.widget.updateSliceName()
+        self.widget.updateView()
+        self.mainwin.rspecWindow.updateView()
+
+    def nodeSelectionChanged(self, hostname):
+        self.mainwin.nodeSelectionChanged(hostname)