from Feed import Feed
from FeedEntry import FeedEntry
from PyQt4 import QtCore, QtGui
from PyQt4.QtCore import QAbstractListModel, QModelIndex, QVariant, Qt, QUrl
from Ui_MainWindow import Ui_MainWindow

class MainWindowController(QtGui.QMainWindow):
    def __init__(self, session=None):
        QtGui.QWidget.__init__(self, None)
        self.session = session
        self.ui = Ui_MainWindow()
        self.ui.setupUi(self)
        self.setupWidgets()
        
    def setupWidgets(self):
        self.setupFeedList()
        self.setupFeedEntryList()
        
    def setupFeedList(self):
        self.allFeeds = Feed.all(self.session)
        feedModel = DisplayModel(self, self.allFeeds, Feed.userPresentableString)
        self.ui.feedList.setModel(feedModel)
        self.connect(self.ui.feedList, QtCore.SIGNAL("clicked(QModelIndex)"), self.feedSelected)
        
    def setupFeedEntryList(self):
        self.connect(self.ui.feedEntryList, QtCore.SIGNAL("clicked(QModelIndex)"), self.feedEntrySelected)
        
    def feedSelected(self, index):
        self.selectedFeed = self.allFeeds[index.row()]
        model = DisplayModel(self, self.selectedFeed.entries, FeedEntry.userPresentableString)
        self.ui.feedEntryList.setModel(model)
        self.ui.feedEntryList.update()
        
    def feedEntrySelected(self, index):
        row = index.row()
        entry = self.selectedFeed.entries[row]
        baseUrl = QUrl(entry.link) # TODO this is the wrong base url, figure out the correct one
        self.ui.webView.setHtml(entry.summary, baseUrl)

class DisplayModel(QAbstractListModel):
    def __init__(self, parent=None, list=None, displayFunction=None, **args):
        QAbstractListModel.__init__(self, parent, *args)
        self.list = list
        self.displayFunction = displayFunction
                
    def rowCount(self, parent=QModelIndex()):
        return len(self.list)
    
    def data(self, index, role): 
        if index.isValid() and role == Qt.DisplayRole:
            row = index.row()
            object = self.list[row]
            displayString = self.displayFunction(object)
            return QVariant(displayString)
        else: 
            return QVariant()