276 lines
8.4 KiB
Python
276 lines
8.4 KiB
Python
from models import Customer, CustomerService, Location, Host, Service, ServiceType, HostType, VPNType
|
|
from typing import List
|
|
|
|
|
|
def load_customers() -> List[Customer]:
|
|
"""Load customer data. Currently returns mock data for demonstration."""
|
|
|
|
customers = []
|
|
|
|
# Customer 1: TechCorp Solutions
|
|
techcorp = Customer(name="TechCorp Solutions")
|
|
|
|
# TechCorp's cloud services
|
|
techcorp.services = [
|
|
CustomerService("Office 365", "https://portal.office.com", "Email & Office"),
|
|
CustomerService("Pascom Cloud PBX", "https://techcorp.pascom.cloud", "Phone System"),
|
|
CustomerService("Salesforce CRM", "https://techcorp.salesforce.com", "CRM")
|
|
]
|
|
|
|
# TechCorp's main office location
|
|
main_office = Location(
|
|
name="Main Office",
|
|
vpn_type=VPNType.OPENVPN,
|
|
connected=True,
|
|
active=True,
|
|
vpn_config="/etc/openvpn/techcorp-main.ovpn"
|
|
)
|
|
|
|
# Proxmox hypervisor with VMs
|
|
proxmox_host = Host(
|
|
name="PVE-01",
|
|
ip_address="192.168.1.10",
|
|
host_type=HostType.PROXMOX,
|
|
description="Main virtualization server",
|
|
services=[
|
|
Service("Web Interface", ServiceType.WEB_GUI, 8006),
|
|
Service("SSH", ServiceType.SSH, 22)
|
|
]
|
|
)
|
|
|
|
# VMs running on Proxmox
|
|
proxmox_host.sub_hosts = [
|
|
Host(
|
|
name="DC-01",
|
|
ip_address="192.168.1.20",
|
|
host_type=HostType.WINDOWS_SERVER,
|
|
description="Domain Controller",
|
|
services=[
|
|
Service("RDP", ServiceType.RDP, 3389),
|
|
Service("Admin Web", ServiceType.WEB_GUI, 8080)
|
|
]
|
|
),
|
|
Host(
|
|
name="FILE-01",
|
|
ip_address="192.168.1.21",
|
|
host_type=HostType.LINUX,
|
|
description="File Server (Samba)",
|
|
services=[
|
|
Service("SSH", ServiceType.SSH, 22),
|
|
Service("SMB Share", ServiceType.SMB, 445),
|
|
Service("Web Panel", ServiceType.WEB_GUI, 9000)
|
|
]
|
|
),
|
|
Host(
|
|
name="DB-01",
|
|
ip_address="192.168.1.22",
|
|
host_type=HostType.LINUX,
|
|
description="PostgreSQL Database",
|
|
services=[
|
|
Service("SSH", ServiceType.SSH, 22),
|
|
Service("PostgreSQL", ServiceType.DATABASE, 5432),
|
|
Service("pgAdmin", ServiceType.WEB_GUI, 5050)
|
|
]
|
|
)
|
|
]
|
|
|
|
# Network infrastructure
|
|
router = Host(
|
|
name="FW-01",
|
|
ip_address="192.168.1.1",
|
|
host_type=HostType.ROUTER,
|
|
description="pfSense Firewall/Router",
|
|
services=[
|
|
Service("Web Interface", ServiceType.WEB_GUI, 443),
|
|
Service("SSH", ServiceType.SSH, 22)
|
|
]
|
|
)
|
|
|
|
switch = Host(
|
|
name="SW-01",
|
|
ip_address="192.168.1.2",
|
|
host_type=HostType.SWITCH,
|
|
description="Managed Switch",
|
|
services=[
|
|
Service("Web Interface", ServiceType.WEB_GUI, 80),
|
|
Service("SSH", ServiceType.SSH, 22)
|
|
]
|
|
)
|
|
|
|
main_office.hosts = [proxmox_host, router, switch]
|
|
|
|
# Branch office location
|
|
branch_office = Location(
|
|
name="Branch Office",
|
|
vpn_type=VPNType.WIREGUARD,
|
|
connected=False,
|
|
active=False,
|
|
vpn_config="/etc/wireguard/techcorp-branch.conf"
|
|
)
|
|
|
|
branch_server = Host(
|
|
name="BRANCH-01",
|
|
ip_address="10.10.1.10",
|
|
host_type=HostType.LINUX,
|
|
description="Branch office server",
|
|
services=[
|
|
Service("SSH", ServiceType.SSH, 22),
|
|
Service("File Share", ServiceType.SMB, 445),
|
|
Service("Local Web", ServiceType.WEB_GUI, 8080)
|
|
]
|
|
)
|
|
|
|
branch_office.hosts = [branch_server]
|
|
|
|
techcorp.locations = [main_office, branch_office]
|
|
customers.append(techcorp)
|
|
|
|
# Customer 2: MedPractice Group
|
|
medpractice = Customer(name="MedPractice Group")
|
|
|
|
# MedPractice's cloud services
|
|
medpractice.services = [
|
|
CustomerService("Google Workspace", "https://workspace.google.com", "Email & Office"),
|
|
CustomerService("Practice Management", "https://medpractice.emr-system.com", "EMR System"),
|
|
CustomerService("VoIP Provider", "https://medpractice.voip.com", "Phone System")
|
|
]
|
|
|
|
# Clinic location
|
|
clinic_location = Location(
|
|
name="Main Clinic",
|
|
vpn_type=VPNType.WIREGUARD,
|
|
connected=False,
|
|
active=False,
|
|
vpn_config="/etc/wireguard/medpractice.conf"
|
|
)
|
|
|
|
# ESXi hypervisor
|
|
esxi_host = Host(
|
|
name="ESXi-01",
|
|
ip_address="10.0.1.10",
|
|
host_type=HostType.ESXI,
|
|
description="VMware ESXi Host",
|
|
services=[
|
|
Service("vSphere Web", ServiceType.WEB_GUI, 443),
|
|
Service("SSH", ServiceType.SSH, 22)
|
|
]
|
|
)
|
|
|
|
# VMs on ESXi
|
|
esxi_host.sub_hosts = [
|
|
Host(
|
|
name="WIN-SRV-01",
|
|
ip_address="10.0.1.20",
|
|
host_type=HostType.WINDOWS_SERVER,
|
|
description="Windows Server 2022",
|
|
services=[
|
|
Service("RDP", ServiceType.RDP, 3389),
|
|
Service("IIS Web", ServiceType.WEB_GUI, 80)
|
|
]
|
|
),
|
|
Host(
|
|
name="BACKUP-01",
|
|
ip_address="10.0.1.21",
|
|
host_type=HostType.LINUX,
|
|
description="Backup Server",
|
|
services=[
|
|
Service("SSH", ServiceType.SSH, 22),
|
|
Service("Backup Web UI", ServiceType.WEB_GUI, 8080)
|
|
]
|
|
)
|
|
]
|
|
|
|
# Physical server
|
|
physical_server = Host(
|
|
name="PHYS-01",
|
|
ip_address="10.0.1.50",
|
|
host_type=HostType.LINUX,
|
|
description="Physical Linux Server",
|
|
services=[
|
|
Service("SSH", ServiceType.SSH, 22),
|
|
Service("Docker Portainer", ServiceType.WEB_GUI, 9000),
|
|
Service("Nginx Proxy", ServiceType.WEB_GUI, 8080)
|
|
]
|
|
)
|
|
|
|
clinic_location.hosts = [esxi_host, physical_server]
|
|
medpractice.locations = [clinic_location]
|
|
customers.append(medpractice)
|
|
|
|
# Customer 3: Manufacturing Inc
|
|
manufacturing = Customer(name="Manufacturing Inc")
|
|
|
|
# Manufacturing's cloud services
|
|
manufacturing.services = [
|
|
CustomerService("Microsoft 365", "https://portal.office.com", "Email & Office"),
|
|
CustomerService("SAP Cloud", "https://manufacturing.sap.com", "ERP System")
|
|
]
|
|
|
|
# Factory location
|
|
factory_location = Location(
|
|
name="Factory Floor",
|
|
vpn_type=VPNType.IPSEC,
|
|
connected=False,
|
|
active=True,
|
|
vpn_config="/etc/ipsec.d/manufacturing.conf"
|
|
)
|
|
|
|
# Manufacturing infrastructure - simpler setup
|
|
linux_server = Host(
|
|
name="PROD-01",
|
|
ip_address="172.16.1.10",
|
|
host_type=HostType.LINUX,
|
|
description="Production Server",
|
|
services=[
|
|
Service("SSH", ServiceType.SSH, 22),
|
|
Service("Web Portal", ServiceType.WEB_GUI, 8443),
|
|
Service("FTP", ServiceType.FTP, 21)
|
|
]
|
|
)
|
|
|
|
nas_server = Host(
|
|
name="NAS-01",
|
|
ip_address="172.16.1.20",
|
|
host_type=HostType.LINUX,
|
|
description="Network Attached Storage",
|
|
services=[
|
|
Service("SSH", ServiceType.SSH, 22),
|
|
Service("Web Interface", ServiceType.WEB_GUI, 5000),
|
|
Service("SMB Share", ServiceType.SMB, 445)
|
|
]
|
|
)
|
|
|
|
factory_location.hosts = [linux_server, nas_server]
|
|
|
|
# Office location
|
|
office_location = Location(
|
|
name="Administrative Office",
|
|
vpn_type=VPNType.OPENVPN,
|
|
connected=False,
|
|
active=False,
|
|
vpn_config="/etc/openvpn/manufacturing-office.ovpn"
|
|
)
|
|
|
|
office_server = Host(
|
|
name="OFFICE-01",
|
|
ip_address="172.16.2.10",
|
|
host_type=HostType.WINDOWS_SERVER,
|
|
description="Office domain controller",
|
|
services=[
|
|
Service("RDP", ServiceType.RDP, 3389),
|
|
Service("File Share", ServiceType.SMB, 445)
|
|
]
|
|
)
|
|
|
|
office_location.hosts = [office_server]
|
|
|
|
manufacturing.locations = [factory_location, office_location]
|
|
customers.append(manufacturing)
|
|
|
|
return customers
|
|
|
|
|
|
def save_customers(customers: List[Customer]) -> None:
|
|
"""Save customer data. Currently a placeholder."""
|
|
# TODO: Implement actual persistence (JSON file, database, etc.)
|
|
pass |