import praw
import curses

from content_generators import SubmissionContent, SubredditContent
from utils import curses_session
from viewer import BaseViewer

class SubmissionViewer(BaseViewer):

    def __init__(self, stdscr, content):

        page_index, cursor_index = -1, 1
        super(SubmissionViewer, self).__init__(
            stdscr, content,
            page_index=page_index, cursor_index=cursor_index)

    def loop(self):

        self.draw()
        while True:
            cmd = self.stdscr.getch()

            if cmd == curses.KEY_UP:
                self.move_cursor_up()
                self.clear_input_queue()

            elif cmd == curses.KEY_DOWN:
                self.move_cursor_down()
                self.clear_input_queue()

            # Refresh page
            elif cmd in (curses.KEY_F5, ord('r')):
                self.content.reset()
                self.stdscr.clear()
                self.draw()

            elif cmd == curses.KEY_RESIZE:
                self.draw()

            # Quit
            elif cmd == ord('q'):
                break

            else:
                curses.beep()

    def draw(self):

        n_rows, n_cols = self.stdscr.getmaxyx()
        self._header_window = self.stdscr.derwin(1, n_cols, 0, 0)
        self._content_window = self.stdscr.derwin(1, 0)

        self.draw_header()
        self.draw_content()
        self.add_cursor()

    def draw_item(self, win, data, inverted=False):

        if data['type'] == 'MoreComments':
            self.draw_more_comments(win, data)

        elif data['type'] == 'Comment':
            self.draw_comment(win, data, inverted=inverted)

        else:
            self.draw_submission(win, data)

    @staticmethod
    def draw_comment(win, data, inverted=False):

        n_rows, n_cols = win.getmaxyx()
        n_cols -= 2

        # Handle the case where the window is not large enough to fit the data.
        valid_rows = range(0, n_rows)
        offset = 0 if not inverted else -(data['n_rows'] - n_rows)

        row = offset
        text = '{} {} {}'.format(data['author'], data['score'], data['created'])
        if row in valid_rows:
            win.addnstr(row, 1, text, n_cols)

        n_body = len(data['split_body'])
        for row, text in enumerate(data['split_body'], start=offset+1):
            if row in valid_rows:
                win.addnstr(row, 1, text, n_cols)

        # Vertical line, unfortunately vline() doesn't support custom color so
        # we have to build it one chr at a time.
        for y in xrange(n_rows):
            win.addch(y, 0, curses.ACS_VLINE)

    @staticmethod
    def draw_more_comments(win, data):

        n_rows, n_cols = win.getmaxyx()
        n_cols -= 2
        win.addnstr(0, 1, data['body'], n_cols)

        for y in xrange(n_rows):
            win.addch(y, 0, curses.ACS_VLINE)

    @staticmethod
    def draw_submission(win, data):

        n_rows, n_cols = win.getmaxyx()
        n_cols -= 1

        # Don't print anything if there is not enough room
        if data['n_rows'] > n_rows:
            return

        win.border()


def main():

    with curses_session() as stdscr:
        r = praw.Reddit(user_agent='reddit terminal viewer (rtv) v0.0')
        submission = SubredditContent(r).get(0)['object']
        generator = SubmissionContent(submission)

        viewer = SubmissionViewer(stdscr, generator)
        viewer.loop()

if __name__ == '__main__':

    main()