ns2 0.2.6__py3-none-any.whl

ns2/dbus/dbus.py

@@ -0,0 +1,13 @@
+from dbus_next import BusType
+from dbus_next.aio import MessageBus
+
+global AppBus
+
+async def setup():
+    global AppBus
+    AppBus = await MessageBus(bus_type=BusType.SYSTEM).connect()
+
+async def cleanup():
+    global AppBus
+    AppBus.disconnect()
+

ns2/lib/accounts.py

@@ -0,0 +1,130 @@
+from dataclasses import asdict, dataclass
+from typing import Optional
+from nicegui import ui, app
+
+@dataclass
+class SystemAccount:
+    Name: Optional[str] = None
+    PrimaryId: Optional[str] = None
+    SecondaryId: Optional[str] = None
+    Info: Optional[str] = None
+    Home: Optional[str] = None
+    Shell: Optional[str] = None
+
+@dataclass
+class SystemGroup:
+    Name: Optional[str] = None
+    Id: Optional[str] = None
+    NumLocalUsers : Optional[str] = None
+    LocalUsers : Optional[str] = None
+
+def ReadPasswordsFile() -> list[SystemAccount]:
+    accounts = []
+    with open("/etc/passwd", "r") as f:
+        content = f.readlines()
+    for i, line in enumerate(content):
+        if ":" in line:
+            fields = line.split(":")
+            name = fields[0]
+            primaryId = fields[2]
+            secondaryId = fields[3]
+            info = fields[4]
+            home = fields[5]
+            shell = fields[6]
+            a = SystemAccount(name, primaryId, secondaryId, info, home, shell)
+            accounts.append(a)
+    pass #endfor
+
+    return accounts
+
+
+def ReadGroupFile() -> list[SystemGroup]:
+    groups = []
+    with open("/etc/group") as f:
+        content = f.readlines()
+
+    for i, line in enumerate(content):
+        if ":" in line:
+            fields = line.split(":")
+            name = fields[0]
+            id = fields[2]
+            accounts = fields[3].strip("\n")
+            num = len(accounts.split(","))
+            g = SystemGroup(name, id, num, accounts)
+            groups.append(g)
+    pass #endfor
+
+    return groups
+
+
+def CombineGroupsAndAccounts():
+    accounts = ReadPasswordsFile()
+    groups = ReadGroupFile()
+
+    newgroups = []
+    for g in groups:
+        for a in accounts:
+            if g.Id == a.SecondaryId:
+                g.NumLocalUsers += 1
+                g.LocalUsers += f",{a.Name}"
+
+        newgroups.append(g)
+
+    return newgroups
+
+def GetCombinedDict():
+    return [asdict(i) for i in CombineGroupsAndAccounts()]
+
+
+
+def GetGroupDict():
+    return [asdict(i) for i in ReadGroupFile()]
+
+def GetAccountsDict():
+    return [asdict(i) for i in ReadPasswordsFile()]
+
+
+
+async def accounts_user_page(user: str):
+    #ui.label("User Configuration").classes("text-h5")
+    ui.label(user).classes("text-h5")
+
+
+async def accounts_page():
+    """user page content"""
+
+    ui.label("User Configuration").classes("text-h5")
+
+    #table("Groups", GetCombinedDict(), "Name", add_group_dialog(), "Name,Id,NumLocalUsers,LocalUsers")  # Only show these
+    #ui.table("Groups", , "Name", add_group_dialog(), "Name,PrimaryId,SecondaryId,Info,Home,Shell")  # Only show these
+
+
+    with ui.dialog() as asGroupDialog:
+        with ui.card():
+            ui.label("test")
+            ui.input("name")
+            ui.input("id")
+            ui.button("create")
+            ui.button("cancel")
+
+
+
+    table = ui.table(
+               title="Groups",
+               rows=GetAccountsDict(),
+               column_defaults={
+                   "align": "left",
+                   "headerClasses": "uppercase text-primary",
+               },
+           ).classes("w-full")
+
+    table.add_slot(f'body-cell-Name', f''' <q-td :props="props">
+                      <a :href="'/accounts/'+ props.row.Name" class="text-accent cursor-pointer hover:underline"> {{{{ props.value }}}} </a>
+                      </q-td> ''')
+
+    table.props(f'visible-columns={"Name,PrimaryId,SecondaryId,Info,Home,Shell"}')  # Only show these
+
+    with table.add_slot('top-right'):
+        ui.button(icon="add", on_click = asGroupDialog.open).props(
+               "flat color=accent align=left").classes("w-full").props("dense")
+    

ns2/lib/commands.py

@@ -0,0 +1,32 @@
+
+
+import asyncio
+import subprocess
+
+
+#def runCmd(args: list[str]) -> tuple[str, str]:
+#    print(f"running: {args}")
+#    result = subprocess.run(args, capture_output=True, text=True)
+#    return result.stdout
+
+
+async def runCmd(args: list[str]) -> str:
+    print(f"running: {args}")
+    process = await asyncio.create_subprocess_exec(
+        *args,
+        stdout=asyncio.subprocess.PIPE,
+        stderr=asyncio.subprocess.PIPE
+    )
+    stdout, stderr = await process.communicate()
+    return stdout.decode()
+
+
+async def runAsyncCmd(args: list[str]) -> str:
+    print(f"running: {args}")
+    process = await asyncio.create_subprocess_exec(
+        *args,
+        stdout=asyncio.subprocess.PIPE,
+        stderr=asyncio.subprocess.PIPE
+    )
+    stdout, stderr = await process.communicate()
+    return stdout.decode()

ns2/lib/firewalld.py

@@ -0,0 +1,167 @@
+import asyncio
+from dataclasses import asdict, field
+from pprint import pprint
+from typing import List, Optional
+from nicegui import ui, app, binding
+from ns.lib.commands import runCmd
+from ns.ui.theme import init_colors
+
+
+from dbus_next.signature import Variant
+from dbus_next.errors import DBusError
+from dbus_next.aio.proxy_object import ProxyInterface
+from dbus_next.aio import MessageBus
+from ns.dbus import dbus
+from ns.utils import INTROSPECTION_DIR
+
+
+@binding.bindable_dataclass
+class ServiceSetting:
+    Version:           Optional[str] = ''
+    Name:              Optional[str] = ''
+    Description:       Optional[str] = ''
+    Ports:             Optional[list[str]]  = field(default_factory=list)
+    ModuleNames:       Optional[list[str]] = field(default_factory=list)
+    Destinations:      Optional[dict] = field(default_factory=dict)
+    Protocols:         Optional[list[str]] = field(default_factory=list)
+    SourcePorts:       Optional[list[str]] = field(default_factory=list)
+    Includes:          Optional[list[str]] = field(default_factory=list)
+    Helpers:           Optional[list[str]] = field(default_factory=list)
+
+
+@binding.bindable_dataclass
+class ZoneSetting:
+    Description:       Optional[str] = ''
+    Interfaces:        Optional[list[str]] = field(default_factory=list)
+    Services:          Optional[list[str]] = field(default_factory=list)
+    Short:             Optional[str] = ''
+    ServiceSettings:   Optional[dict[ServiceSetting]] = field(default_factory=dict)
+    Sources:           Optional[list[str]] = field(default_factory=list)
+    
+    
+@binding.bindable_dataclass
+class Firewall:
+    Enable:            Optional[bool] = False
+    Status:            Optional[str] = ''
+    ActiveZones:       Optional[dict[dict]] = field(default_factory=dict)
+    AllowedAddresses:  Optional[list[str]] = field(default_factory=list)
+    Services:          Optional[dict[dict]] = field(default_factory=dict)
+    ZoneSettings:      Optional[dict[ZoneSetting]] = field(default_factory=dict)
+
+
+async def GetFirewalld(bus: MessageBus):
+    file_name = 'org.fedoraproject.FirewallD1.config.xml'
+    introspection = await bus.introspect('org.fedoraproject.FirewallD1', '/org/fedoraproject/FirewallD1')
+    #pprint(introspection.tostring())
+    obj = bus.get_proxy_object('org.fedoraproject.FirewallD1', '/org/fedoraproject/FirewallD1', introspection)
+    return obj.get_interface('org.fedoraproject.FirewallD1')
+
+async def GetFirewalldConfig(bus: MessageBus):
+    file_name = 'org.fedoraproject.FirewallD1.config.xml'
+    introspection = await bus.introspect('org.fedoraproject.FirewallD1', '/org/fedoraproject/FirewallD1/config')
+    #pprint(introspection.tostring())
+    obj = bus.get_proxy_object('org.fedoraproject.FirewallD1', '/org/fedoraproject/FirewallD1/config', introspection)
+    return obj.get_interface('org.fedoraproject.FirewallD1.config')
+
+
+
+async def GetFirewalldConfigZone(bus: MessageBus, path : str):
+    introspection = await bus.introspect('org.fedoraproject.FirewallD1', path)
+    obj = bus.get_proxy_object('org.fedoraproject.FirewallD1', path, introspection)
+    return obj.get_interface('org.fedoraproject.FirewallD1.config.zone')
+
+#def GetDevice(bus: MessageBus, path : str):
+#    file_name = 'org.freedesktop.NetworkManager.Device.xml'
+#    with open(str(INTROSPECTION_DIR /file_name), "r") as f:
+#        introspection = f.read()
+#    obj = bus.get_proxy_object('org.freedesktop.NetworkManager', path, introspection)
+#    return obj.get_interface('org.freedesktop.NetworkManager.Device')
+
+async def GetFirewalldZone(bus: MessageBus):
+    file_name = 'org.fedoraproject.FirewallD1.zone.xml'
+    introspection = await bus.introspect('org.fedoraproject.FirewallD1', '/org/fedoraproject/FirewallD1')
+    #pprint(introspection.tostring())
+    obj = bus.get_proxy_object('org.fedoraproject.FirewallD1', '/org/fedoraproject/FirewallD1', introspection)
+    return obj.get_interface('org.fedoraproject.FirewallD1.zone')
+
+
+
+
+def formatListToString(elements: list[str]) -> str:
+    if len(elements) == 0:
+        return None
+    return ', '.join(elements) if elements else ''
+
+def getZoneInfo(name :str, zone :dict) -> dict:
+
+    interfaces = formatListToString(zone.get('interfaces', []))
+    sources = formatListToString(zone.get('sources', []))
+
+    return {'name':name, 'interfaces':interfaces, 'sources':sources}
+
+
+async def getServices(bus: MessageBus):
+    '''all not just runtime'''
+
+    fire = await GetFirewalld(bus)
+
+    conf = await GetFirewalldConfig(bus)
+    serviceNames = await conf.call_get_service_names()
+    services = {}
+    for name in serviceNames:
+        s = ServiceSetting()
+        s.Name = name
+        serviceSettings = await fire.call_get_service_settings2(s.Name)
+                                
+        includes = serviceSettings.get('includes', False)
+        if includes:
+            for i in includes.value:
+                ser_set = await fire.call_get_service_settings2(i)
+                s.Ports.extend(ser_set.get('ports', Variant('a(ss)', [['port not available', 'protocol not available']])).value)
+
+                            
+        s.Name = serviceSettings.get('short', Variant('s', 'name not available')).value
+        s.Ports.extend(serviceSettings.get('ports', Variant('a(ss)', [['port not available', 'protocol not available']])).value)
+        s.Description = serviceSettings.get('description', Variant('s', 'Description not available')).value
+
+        services[name] = s
+
+
+    return services
+
+
+
+def getUdpPorts(ports) -> list:
+    out = []
+    for p in ports:
+        if p[1]=='udp':
+            out.append(p[0])
+
+    return formatListToString(out)
+
+def getTcpPorts(ports) -> list:
+    out = []
+    for p in ports:
+        if p[1]=='tcp':
+            out.append(p[0])
+
+    return formatListToString(out)
+    
+
+
+def formatServicesInRows(serviceSettings :dict):
+    
+
+    rows = []
+    for n, s in serviceSettings.items():
+        rows.append({ 
+                     #"Select": False,
+                     "Service": n
+                     ,"UDP": getUdpPorts(s.Ports)
+                     ,"TCP": getTcpPorts(s.Ports)
+                     ,"Description": s.Description
+                     #,"remove": ''
+                     })
+        
+    #pprint(rows)
+    return rows

ns2/lib/introspection/org.fedoraproject.FirewallD1.config.xml

@@ -0,0 +1,3 @@
+<!DOCTYPE node PUBLIC "-//freedesktop//DTD D-BUS Object Introspection 1.0//EN"
+"http://www.freedesktop.org/standards/dbus/1.0/introspect.dtd">
+<node />