Python-Web-Proxy/TestManagementConsole.py at master · cianjinks/Python-Web-Proxy · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
import curses

# Just strings for testing
packets = [
    "Option 1 looking kinda fun! BARS",
    "Geese are tasty, not ducks.",
    "abcdefghijklmnopqrstyxqz",
    "That's not the alphabet???",
    "Okay fair enough"
    ]
blocked_urls = [
    "This isn't a url?",
    "ebay.ie/12319239123",
    "http://www.amazon.com/yay",
    "https://tcd.blackboard.com/webapps/blackboard/",
    "https://tcd.blackboard.com/",
    "https://docs.python.org/3/howto/curses.html",
    "https://www.google.com/",
    "https://www.amazon.de/",
    "More non URLs cause why not",
    "Woohooooooooo",
    "Is this more tests wtf?>??",
    "Wow how did you get all the way down here?",
    "This is the end...",
    "Or is it?",
    "Now it is!"]

CONSOLE_WINDOW_ID = False
BLOCKED_WINDOW_ID = True

def draw_keymap(window):
    window.erase()
    window.border()
    window.addstr(0, int(curses.COLS / 4) - 4, " Keymap ")
    window.addstr(1, 1, "[Q] - Quit")
    window.addstr(2, 1, "[S] - Switch window")
    window.addstr(3, 1, "[B] - Block URL (of selected packet)")
    window.addstr(4, 1, "[U] - Unblock URL")
    window.addstr(5, 1, "[Ent/Esc] - View detailed packet info")
    window.addstr(6, 1, "[Up/Down Arrow] - Select packet or URL")
    window.refresh()

def draw_blocked(window, selected_url, focused_window, block_min, block_max):
    window.erase()
    window.border()
    window.addstr(0, int(curses.COLS / 4) - 7, " Blocked URLs ")

    # List of blocked URLs
    b_pos = 1
    for b in range(block_min, block_max + 1):
        window.addstr(b_pos, 1, "[" + str(b) + "] " + blocked_urls[b],
            curses.A_REVERSE if (b == selected_url and focused_window == BLOCKED_WINDOW_ID) else curses.A_NORMAL)
        b_pos = b_pos + 1
    window.refresh()

def draw_console(window, selected_packet, focused_window, packet_min, packet_max):
    window.erase()
    window.border()
    window.addstr(0, int(curses.COLS / 2) - 10, " Management Console ")

    # List of packets
    p_pos = 1
    for p in range(packet_min, packet_max + 1):
        window.addstr(p_pos, 1, "[" + str(p) + "] " + packets[p],
            curses.A_REVERSE if (p == selected_packet and focused_window == CONSOLE_WINDOW_ID) else curses.A_NORMAL)
        p_pos = p_pos + 1
    window.refresh()

def app(stdscr):
    # Options
    curses.curs_set(0)

    # Window Creation
    console_window = curses.newwin(curses.LINES - int(curses.LINES / 3), curses.COLS, 0, 0)
    keymap_window = curses.newwin(int(curses.LINES / 3), int(curses.COLS / 2), curses.LINES - int(curses.LINES / 3), 0)
    blocked_window = curses.newwin(int(curses.LINES / 3), int(curses.COLS / 2), curses.LINES - int(curses.LINES / 3), int(curses.COLS / 2))
    focused_window = False # False for console, True for blocked urls
    selected_packet = 0
    selected_url = 0

    max_urls = int(curses.LINES / 3) - 2 # Max number of URLs that can be displayed on the screen at once
    block_min = 0
    block_max = len(blocked_urls) - 1 if len(blocked_urls) < max_urls else max_urls - 1

    max_packets = curses.LINES - int(curses.LINES / 3) - 2 # Max number of packets that can be displayed on the screen at once
    packet_min = 0
    packet_max = len(packets) - 1 if len(packets) < max_packets else max_packets - 1

    while True:
        # Draw Windows
        stdscr.erase()
        stdscr.refresh()
        draw_console(console_window, selected_packet, focused_window, packet_min, packet_max)
        draw_keymap(keymap_window)
        draw_blocked(blocked_window, selected_url, focused_window, block_min, block_max)

        # Handle Input
        c = stdscr.getch()
        if c == ord('q'):
            break
        elif c == ord('s'):
            focused_window = not focused_window
        elif c == curses.KEY_DOWN:
            if focused_window == CONSOLE_WINDOW_ID and selected_packet < len(packets) - 1:
                selected_packet = selected_packet + 1
                if selected_packet > packet_max:
                    packet_max = packet_max + 1
                    packet_min = packet_min + 1
            elif focused_window == BLOCKED_WINDOW_ID and selected_url < len(blocked_urls) - 1:
                selected_url  = selected_url + 1
                if selected_url > block_max:
                    block_max = block_max + 1
                    block_min = block_min + 1
        elif c == curses.KEY_UP:
            if focused_window == CONSOLE_WINDOW_ID and selected_packet > 0:
                selected_packet = selected_packet - 1
                if selected_packet < packet_min:
                    packet_max = packet_max - 1
                    packet_min = packet_min - 1
            elif focused_window == BLOCKED_WINDOW_ID and selected_url > 0:
                selected_url = selected_url - 1
                if selected_url < block_min:
                    block_max = block_max - 1
                    block_min = block_min - 1

if __name__ == "__main__":
    curses.wrapper(app)