comparison hgext/histedit.py @ 48200:b6fc7d188f68

chistedit: move view state from a dict to a custom class

Differential Revision: https://phab.mercurial-scm.org/D11636

48199:9d0e5629cfbf

# ============ EVENTS ===============
def movecursor(state, oldpos, newpos):
    """Change the rule/changeset that the cursor is pointing to, regardless of
    current mode (you can switch between patches from the view patch window)."""
    state[b'pos'] = newpos

    mode, _ = state[b'mode']
    if mode == MODE_RULES:
        # Scroll through the list by updating the view for MODE_RULES, so that
        # even if we are not currently viewing the rules, switching back will
        # result in the cursor's rule being visible.
        modestate = state[b'modes'][MODE_RULES]
        if newpos < modestate[b'line_offset']:
            modestate[b'line_offset'] = newpos
        elif newpos > modestate[b'line_offset'] + state[b'page_height'] - 1:
            modestate[b'line_offset'] = newpos - state[b'page_height'] + 1

    # Reset the patch view region to the top of the new patch.
    state[b'modes'][MODE_PATCH][b'line_offset'] = 0


def changemode(state, mode):
    curmode, _ = state[b'mode']
    state[b'mode'] = (mode, curmode)
    if mode == MODE_PATCH:
        state[b'modes'][MODE_PATCH][b'patchcontents'] = patchcontents(state)


def makeselection(state, pos):
    state[b'selected'] = pos


def swap(state, oldpos, newpos):
    """Swap two positions and calculate necessary conflicts in
    O(|newpos-oldpos|) time"""

    rules = state[b'rules']
    assert 0 <= oldpos < len(rules) and 0 <= newpos < len(rules)

    rules[oldpos], rules[newpos] = rules[newpos], rules[oldpos]

    # TODO: swap should not know about histeditrule's internals

    end = max(oldpos, newpos)
    for r in pycompat.xrange(start, end + 1):
        rules[newpos].checkconflicts(rules[r])
        rules[oldpos].checkconflicts(rules[r])

    if state[b'selected']:
        makeselection(state, newpos)


def changeaction(state, pos, action):
    """Change the action state on the given position to the new action"""
    rules = state[b'rules']
    assert 0 <= pos < len(rules)
    rules[pos].action = action


def cycleaction(state, pos, next=False):
    """Changes the action state the next or the previous action from
    the action list"""
    rules = state[b'rules']
    assert 0 <= pos < len(rules)
    current = rules[pos].action

    assert current in KEY_LIST

def changeview(state, delta, unit):
    """Change the region of whatever is being viewed (a patch or the list of
    changesets). 'delta' is an amount (+/- 1) and 'unit' is 'page' or 'line'."""
    mode, _ = state[b'mode']
    if mode != MODE_PATCH:
        return
    mode_state = state[b'modes'][mode]
    num_lines = len(mode_state[b'patchcontents'])
    page_height = state[b'page_height']
    unit = page_height if unit == b'page' else 1
    num_pages = 1 + (num_lines - 1) // page_height
    max_offset = (num_pages - 1) * page_height
    newline = mode_state[b'line_offset'] + delta * unit
    mode_state[b'line_offset'] = max(0, min(max_offset, newline))

    """Change state based on the current character input

    This takes the current state and based on the current character input from
    the user we change the state.
    """
    selected = state[b'selected']
    oldpos = state[b'pos']
    rules = state[b'rules']

    if ch in (curses.KEY_RESIZE, b"KEY_RESIZE"):
        return E_RESIZE

    lookup_ch = ch
    if ch is not None and b'0' <= ch <= b'9':
        lookup_ch = b'0'

    curmode, prevmode = state[b'mode']
    action = KEYTABLE[curmode].get(
        lookup_ch, KEYTABLE[b'global'].get(lookup_ch)
    )
    if action is None:
        return

        return line
    return line[: n - 2] + b' >'


def patchcontents(state):
    repo = state[b'repo']
    rule = state[b'rules'][state[b'pos']]
    displayer = logcmdutil.changesetdisplayer(
        repo.ui, repo, {b"patch": True, b"template": b"status"}, buffered=True
    )
    overrides = {(b'ui', b'verbose'): True}
    with repo.ui.configoverride(overrides, source=b'histedit'):
        displayer.show(rule.ctx)
        displayer.close()
    return displayer.hunk[rule.ctx.rev()].splitlines()


def _chisteditmain(repo, rules, stdscr):
    try:
        curses.use_default_colors()

        pass

    def rendercommit(win, state):
        """Renders the commit window that shows the log of the current selected
        commit"""
        pos = state[b'pos']
        rules = state[b'rules']
        rule = rules[pos]

        ctx = rule.ctx
        win.box()

"""
        return help.splitlines()

    def renderhelp(win, state):
        maxy, maxx = win.getmaxyx()
        mode, _ = state[b'mode']
        for y, line in enumerate(helplines(mode)):
            if y >= maxy:
                break
            addln(win, y, 0, line, curses.color_pair(COLOR_HELP))
        win.noutrefresh()

    def renderrules(rulesscr, state):
        rules = state[b'rules']
        pos = state[b'pos']
        selected = state[b'selected']
        start = state[b'modes'][MODE_RULES][b'line_offset']

        conflicts = [r.ctx for r in rules if r.conflicts]
        if len(conflicts) > 0:
            line = b"potential conflict in %s" % b','.join(
                map(pycompat.bytestr, conflicts)
            )
            addln(rulesscr, -1, 0, line, curses.color_pair(COLOR_WARN))

        for y, rule in enumerate(rules[start:]):
            if y >= state[b'page_height']:
                break
            if len(rule.conflicts) > 0:
                rulesscr.addstr(y, 0, b" ", curses.color_pair(COLOR_WARN))
            else:
                rulesscr.addstr(y, 0, b" ", curses.COLOR_BLACK)

            else:
                win.addstr(y, 0, line)
        win.noutrefresh()

    def renderpatch(win, state):
        start = state[b'modes'][MODE_PATCH][b'line_offset']
        content = state[b'modes'][MODE_PATCH][b'patchcontents']
        renderstring(win, state, content[start:], diffcolors=True)

    def layout(mode):
        maxy, maxx = stdscr.getmaxyx()
        helplen = len(helplines(mode))

    def drawvertwin(size, y, x):
        win = curses.newwin(size[0], size[1], y, x)
        y += size[0]
        return win, y, x

    state = {
        b'pos': 0,
        b'rules': rules,
        b'selected': None,
        b'mode': (MODE_INIT, MODE_INIT),
        b'page_height': None,
        b'modes': {
            MODE_RULES: {
                b'line_offset': 0,
            },
            MODE_PATCH: {
                b'line_offset': 0,
            },
        },
        b'repo': repo,
    }

    # eventloop
    ch = None
    stdscr.clear()
    stdscr.refresh()
    while True:
        oldmode, unused = state[b'mode']
        if oldmode == MODE_INIT:
            changemode(state, MODE_RULES)
        e = event(state, ch)

        if e == E_QUIT:
            return False
        if e == E_HISTEDIT:
            return state[b'rules']
        else:
            if e == E_RESIZE:
                size = screen_size()
                if size != stdscr.getmaxyx():
                    curses.resizeterm(*size)

            curmode, unused = state[b'mode']
            sizes = layout(curmode)
            if curmode != oldmode:
                state[b'page_height'] = sizes[b'main'][0]
                # Adjust the view to fit the current screen size.
                movecursor(state, state[b'pos'], state[b'pos'])

            # Pack the windows against the top, each pane spread across the
            # full width of the screen.
            y, x = (0, 0)
            helpwin, y, x = drawvertwin(sizes[b'help'], y, x)

48200:b6fc7d188f68

# ============ EVENTS ===============
def movecursor(state, oldpos, newpos):
    """Change the rule/changeset that the cursor is pointing to, regardless of
    current mode (you can switch between patches from the view patch window)."""
    state.pos = newpos

    mode, _ = state.mode
    if mode == MODE_RULES:
        # Scroll through the list by updating the view for MODE_RULES, so that
        # even if we are not currently viewing the rules, switching back will
        # result in the cursor's rule being visible.
        modestate = state.modes[MODE_RULES]
        if newpos < modestate[b'line_offset']:
            modestate[b'line_offset'] = newpos
        elif newpos > modestate[b'line_offset'] + state.page_height - 1:
            modestate[b'line_offset'] = newpos - state.page_height + 1

    # Reset the patch view region to the top of the new patch.
    state.modes[MODE_PATCH][b'line_offset'] = 0


def changemode(state, mode):
    curmode, _ = state.mode
    state.mode = (mode, curmode)
    if mode == MODE_PATCH:
        state.modes[MODE_PATCH][b'patchcontents'] = patchcontents(state)


def makeselection(state, pos):
    state.selected = pos


def swap(state, oldpos, newpos):
    """Swap two positions and calculate necessary conflicts in
    O(|newpos-oldpos|) time"""

    rules = state.rules
    assert 0 <= oldpos < len(rules) and 0 <= newpos < len(rules)

    rules[oldpos], rules[newpos] = rules[newpos], rules[oldpos]

    # TODO: swap should not know about histeditrule's internals

    end = max(oldpos, newpos)
    for r in pycompat.xrange(start, end + 1):
        rules[newpos].checkconflicts(rules[r])
        rules[oldpos].checkconflicts(rules[r])

    if state.selected:
        makeselection(state, newpos)


def changeaction(state, pos, action):
    """Change the action state on the given position to the new action"""
    rules = state.rules
    assert 0 <= pos < len(rules)
    rules[pos].action = action


def cycleaction(state, pos, next=False):
    """Changes the action state the next or the previous action from
    the action list"""
    rules = state.rules
    assert 0 <= pos < len(rules)
    current = rules[pos].action

    assert current in KEY_LIST

def changeview(state, delta, unit):
    """Change the region of whatever is being viewed (a patch or the list of
    changesets). 'delta' is an amount (+/- 1) and 'unit' is 'page' or 'line'."""
    mode, _ = state.mode
    if mode != MODE_PATCH:
        return
    mode_state = state.modes[mode]
    num_lines = len(mode_state[b'patchcontents'])
    page_height = state.page_height
    unit = page_height if unit == b'page' else 1
    num_pages = 1 + (num_lines - 1) // page_height
    max_offset = (num_pages - 1) * page_height
    newline = mode_state[b'line_offset'] + delta * unit
    mode_state[b'line_offset'] = max(0, min(max_offset, newline))

    """Change state based on the current character input

    This takes the current state and based on the current character input from
    the user we change the state.
    """
    selected = state.selected
    oldpos = state.pos
    rules = state.rules

    if ch in (curses.KEY_RESIZE, b"KEY_RESIZE"):
        return E_RESIZE

    lookup_ch = ch
    if ch is not None and b'0' <= ch <= b'9':
        lookup_ch = b'0'

    curmode, prevmode = state.mode
    action = KEYTABLE[curmode].get(
        lookup_ch, KEYTABLE[b'global'].get(lookup_ch)
    )
    if action is None:
        return

        return line
    return line[: n - 2] + b' >'


def patchcontents(state):
    repo = state.repo
    rule = state.rules[state.pos]
    displayer = logcmdutil.changesetdisplayer(
        repo.ui, repo, {b"patch": True, b"template": b"status"}, buffered=True
    )
    overrides = {(b'ui', b'verbose'): True}
    with repo.ui.configoverride(overrides, source=b'histedit'):
        displayer.show(rule.ctx)
        displayer.close()
    return displayer.hunk[rule.ctx.rev()].splitlines()


class _chistedit_state(object):
    def __init__(
        self,
        repo,
        rules,
    ):
        self.repo = repo
        self.rules = rules
        self.pos = 0
        self.selected = None
        self.mode = (MODE_INIT, MODE_INIT)
        self.page_height = None
        self.modes = {
            MODE_RULES: {
                b'line_offset': 0,
            },
            MODE_PATCH: {
                b'line_offset': 0,
            },
        }


def _chisteditmain(repo, rules, stdscr):
    try:
        curses.use_default_colors()

        pass

    def rendercommit(win, state):
        """Renders the commit window that shows the log of the current selected
        commit"""
        pos = state.pos
        rules = state.rules
        rule = rules[pos]

        ctx = rule.ctx
        win.box()

"""
        return help.splitlines()

    def renderhelp(win, state):
        maxy, maxx = win.getmaxyx()
        mode, _ = state.mode
        for y, line in enumerate(helplines(mode)):
            if y >= maxy:
                break
            addln(win, y, 0, line, curses.color_pair(COLOR_HELP))
        win.noutrefresh()

    def renderrules(rulesscr, state):
        rules = state.rules
        pos = state.pos
        selected = state.selected
        start = state.modes[MODE_RULES][b'line_offset']

        conflicts = [r.ctx for r in rules if r.conflicts]
        if len(conflicts) > 0:
            line = b"potential conflict in %s" % b','.join(
                map(pycompat.bytestr, conflicts)
            )
            addln(rulesscr, -1, 0, line, curses.color_pair(COLOR_WARN))

        for y, rule in enumerate(rules[start:]):
            if y >= state.page_height:
                break
            if len(rule.conflicts) > 0:
                rulesscr.addstr(y, 0, b" ", curses.color_pair(COLOR_WARN))
            else:
                rulesscr.addstr(y, 0, b" ", curses.COLOR_BLACK)

            else:
                win.addstr(y, 0, line)
        win.noutrefresh()

    def renderpatch(win, state):
        start = state.modes[MODE_PATCH][b'line_offset']
        content = state.modes[MODE_PATCH][b'patchcontents']
        renderstring(win, state, content[start:], diffcolors=True)

    def layout(mode):
        maxy, maxx = stdscr.getmaxyx()
        helplen = len(helplines(mode))

    def drawvertwin(size, y, x):
        win = curses.newwin(size[0], size[1], y, x)
        y += size[0]
        return win, y, x

    state = _chistedit_state(repo, rules)

    # eventloop
    ch = None
    stdscr.clear()
    stdscr.refresh()
    while True:
        oldmode, unused = state.mode
        if oldmode == MODE_INIT:
            changemode(state, MODE_RULES)
        e = event(state, ch)

        if e == E_QUIT:
            return False
        if e == E_HISTEDIT:
            return state.rules
        else:
            if e == E_RESIZE:
                size = screen_size()
                if size != stdscr.getmaxyx():
                    curses.resizeterm(*size)

            curmode, unused = state.mode
            sizes = layout(curmode)
            if curmode != oldmode:
                state.page_height = sizes[b'main'][0]
                # Adjust the view to fit the current screen size.
                movecursor(state, state.pos, state.pos)

            # Pack the windows against the top, each pane spread across the
            # full width of the screen.
            y, x = (0, 0)
            helpwin, y, x = drawvertwin(sizes[b'help'], y, x)