mikro-manager/main.py at main · szuryuu/mikro-manager · 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
import streamlit as st
from pathlib import Path

def load_css():
    with open("./assets/style/styles.css") as css:
        st.markdown(f"<style>{css.read()}</style>", unsafe_allow_html=True)

load_css()

if 'ssh_connection' not in st.session_state:
    st.session_state['ssh_connection'] = False
if 'ssh_client' not in st.session_state:
    st.session_state['ssh_client'] = None
if 'current_page' not in st.session_state:
    st.session_state['current_page'] = None

pages = {
    # Auth
    "Connect": "./views/auth/connect.py",

    # Basic Conf
    "Change Router Name": "./views/change_name_router.py",
    "IP Address Configuration": "./views/ip_address.py",
    "DNS Configuration": "./views/dns_server.py",
    "Block Sites": "./views/block_addr.py",
    "Backup Configuration": "./views/backup.py",

    # Advance
    "Gateway Configuration": "./views/gateway.py",
    "DHCP Configuration": "./views/dhcp.py",

    # Help
    "Help": "./views/help.py"
}

with st.sidebar:
    col1, col2 = st.columns([1, 4])
    with col1:
        st.image("assets/img/logo-mikromanager.png", width=42)
    with col2:
        st.title("MikroManager")

    st.header("Connection Status")
    if st.session_state['ssh_client']:
        st.success("SSH Connected")

        if st.button("Disconnect", key="disconnect_button", help="Click to disconnect from the router."):
            try:
                if st.session_state['ssh_client']:
                    st.session_state['ssh_client'].close()
                st.session_state['ssh_client'] = None
                st.session_state['ssh_connection'] = False
                st.success("Successfully disconnected from the router.")

                st.rerun()
            except Exception as e:
                st.error(f"Error disconnecting: {e}")
    else:
        st.warning("SSH Not Connected. Please go to 'Connect' and authenticate.")

    with st.expander("🔗 Connect", expanded=True):
        if st.button("Go to Connection Page", use_container_width=True):
            st.session_state['current_page'] = "Connect"

    with st.expander("⚙️ Basic Configuration", expanded=False):
        if st.button("Change Router Name", use_container_width=True):
            st.session_state['current_page'] = "Change Router Name"
        if st.button("IP Address Configuration", use_container_width=True):
            st.session_state['current_page'] = "IP Address Configuration"
        if st.button("DNS Configuration", use_container_width=True):
            st.session_state['current_page'] = "DNS Configuration"
        if st.button("Block Sites", use_container_width=True):
            st.session_state['current_page'] = "Block Sites"
        if st.button("Backup Configuration", use_container_width=True):
            st.session_state['current_page'] = "Backup Configuration"

    with st.expander("🤖 Advance Configuration", expanded=False):
        if st.button("Gateway Configuration", use_container_width=True):
            st.session_state['current_page'] = "Gateway Configuration"
        if st.button("DHCP Configuration", use_container_width=True):
            st.session_state['current_page'] = "DHCP Configuration"

    with st.expander("❓ Help", expanded=False):
        if st.button("Help Page", use_container_width=True):
            st.session_state['current_page'] = "Help"

if 'current_page' in st.session_state and st.session_state['current_page'] in pages:
    file_path = Path(pages[st.session_state['current_page']])

    if file_path.exists():
        with file_path.open("r", encoding="utf-8") as file:
            exec(file.read())
    else:
        st.error(f"File not found: {file_path}")
else:
    st.header("Welcome to MikroManager!\nPlease connect your router first.")