' Network diagnostics test script

Sub AdvanceLine(rect as Object, amount as Integer)
    rect.SetY(rect.GetY() + amount)
End Sub

Sub Geometry_Advance()
    m.left_rect.SetY(m.left_rect.GetY() + m.height + m.spacing)
    m.middle_rect.SetY(m.middle_rect.GetY() + m.height + m.spacing)
    m.right_rect.SetY(m.right_rect.GetY() + m.height + m.spacing)
End Sub

Function Geometry_CreateWidgetTriplet(name as String)
    w = {}
    w.label = CreateObject("roTextWidget", m.left_rect, 1, 2, { Alignment: 0})
    w.result = CreateObject("roTextWidget", m.middle_rect, 1, 2, { Alignment: 0})
    w.info = CreateObject("roTextWidget", m.right_rect, 1, 2, { Alignment: 0})
    w.name = name

    w.result.SetBackgroundColor(&h7f202020)
    w.info.SetBackgroundColor(&h7f202020)

    w.label.PushString(name)
    w.result.PushString("")
    w.info.PushString("")

    border = m.right_rect.GetHeight() / 6
    r = CreateObject("roRectangle", border, border, m.right_rect.GetWidth() - border * 2, m.right_rect.GetHeight() - border * 2)
    w.info.SetSafeTextRegion(r)

    m.widgets.Push(w.label)
    m.widgets.Push(w.result)
    m.widgets.Push(w.info)

    m.Advance()

    return w
End Function

Function Geometry_CreateTitle(text as string)
    title = CreateObject("roTextWidget", m.title_rect, 1, 2, { Alignment: 1})
    title.PushString(text)
    m.widgets.Push(title)
    return title
End Function

Function Geometry_CreatePrompt()
    m.Advance()

    rect = CreateObject("roRectangle", m.left_rect.GetX(), m.left_rect.GetY(), m.left_rect.GetWidth() + m.right_rect.GetWidth(), m.left_rect.GetHeight() * 0.75)

    prompt = CreateObject("roTextWidget", rect, 1, 2, { Alignment: 1})
    m.widgets.Push(prompt)
    return prompt
End Function

Sub Geometry_ShowAll()
    for each widget in m.widgets
	widget.Show()
    next widget
End Sub

Function CreateGeometry()
    g = {}

    v = CreateObject("roVideoMode")

    lines = 7
    left = v.GetSafeX() + v.GetSafeWidth() / 16
    width = v.GetSafeWidth() - left * 2
    left_width = width / 6
    middle_width = width / 12
    g.height = v.GetSafeHeight() / 20
    g.spacing = g.height / 2
    top = v.GetSafeY() + (v.GetSafeHeight() - g.height * lines - g.spacing * (lines - 1)) / 2

    g.left_rect = CreateObject("roRectangle", left, top, left_width, g.height)
    g.middle_rect = CreateObject("roRectangle", left + left_width, top, middle_width, g.height)
    g.right_rect = CreateObject("roRectangle", left + left_width + middle_width, top, width - left_width - middle_width, g.height)
    g.title_rect = CreateObject("roRectangle", left, top - g.height * 2, width, g.height * 1.25)

    g.widgets = CreateObject("roArray", 0, true)

    g.Advance = Geometry_Advance
    g.CreateWidgetTriplet = Geometry_CreateWidgetTriplet
    g.CreateTitle = Geometry_CreateTitle
    g.ShowAll = Geometry_ShowAll
    g.CreatePrompt = Geometry_CreatePrompt
    return g
End Function

Function CreateWidgets()
    left_alignment = { Alignment: 0 }
    centre_alignment = { Alignment: 1 }
    right_alignment = { Alignment: 0 }

    geometry = CreateGeometry()

    widgets = {}
    widgets.title = geometry.CreateTitle("Network Diagnostics")
    widgets.ethernet = geometry.CreateWidgetTriplet("Ethernet")
	widgets.ethernetConfiguration = geometry.CreateWidgetTriplet("  config")
    widgets.wifi = geometry.CreateWidgetTriplet("WiFi")
	widgets.wifiConfiguration = geometry.CreateWidgetTriplet("  config")
    widgets.internet = geometry.CreateWidgetTriplet("Internet")
	widgets.timeServer = geometry.CreateWidgetTriplet("Time")
    widgets.prompt = geometry.CreatePrompt()

    geometry.ShowAll()

    return widgets
End Function


Sub WriteWidgetStatus(widget as Object, status as String)

	widget.result.SetForegroundColor(&hff00)
	widget.result.Clear()
	widget.info.Clear()
	widget.info.PushString(status)

End Sub


Sub FillWidgets(widgets as Object, result as String, info as String)
    if result <> "FAIL" then
		widgets.result.SetForegroundColor(&hff00)
		widgets.result.Clear()
		widgets.info.Clear()
		widgets.result.PushString(result)
    else
		widgets.result.SetForegroundColor(&hff0000)
		widgets.result.Clear()
		widgets.info.Clear()
		widgets.result.PushString(result)
		widgets.info.PushString(info)
    endif
End Sub

Sub ReportToWidgets(widgets as Object, result as Object)
    if result.ok then
	FillWidgets(widgets, "OK", "")
    else
        FillWidgets(widgets, "FAIL", result.diagnosis)
	print widgets.name; " log:"
	for each line in result.log
	    print "  "; line
	next line
	print
    end if
End Sub


Sub ClearWidget(widget as Object)
    widget.result.Clear()
    widget.info.Clear()
    widget.result.SetForegroundColor(&hffffff)
End Sub


Sub ClearWidgets(widgets as Object)
	ClearWidget(widgets.ethernet)
	ClearWidget(widgets.ethernetConfiguration)
	ClearWidget(widgets.wifi)
	ClearWidget(widgets.wifiConfiguration)
	ClearWidget(widgets.internet)
	ClearWidget(widgets.timeServer)
End Sub


Sub SetWidgetRunning(widget as Object)
    widget.result.Clear()
    widget.result.PushString("...")
End Sub


Function RunTests(widgets as Object, testEthernet as Boolean, testWireless As Boolean, testInternet As Boolean) As Boolean

	allTestsPassed = true

    ClearWidgets(widgets)

    SetWidgetRunning(widgets.ethernet)
	if testEthernet then
		if0 = CreateObject("roNetworkConfiguration", 0)
		if0result = if0.TestInterface()
		ReportToWidgets(widgets.ethernet, if0result)
		allTestsPassed = allTestsPassed and if0result.ok

		ipAddress$ = "None"
		gateway$ = "None"
		dns$ = "None"

		if type(if0) = "roNetworkConfiguration" then
			cc = if0.GetCurrentConfig()
			if cc.ip4_address <> "" then
				ipAddress$ = cc.ip4_address
			endif
			if cc.ip4_gateway <> "" then
				gateway$ = cc.ip4_gateway
			endif
			if cc.dns_servers.Count() > 0 then
				dns$ = cc.dns_servers[0]
			endif
		endif

		ethernetConfiguration$ = "IP address = " + ipAddress$ + ", Gateway = " + gateway$ + ", DNS = " + dns$
		WriteWidgetStatus(widgets.ethernetConfiguration, ethernetConfiguration$)

	else
		FillWidgets(widgets.ethernet, "N/A", "")
		FillWidgets(widgets.ethernetConfiguration, "N/A", "")
	endif

    SetWidgetRunning(widgets.wifi)
	if testWireless then
		if1 = CreateObject("roNetworkConfiguration", 1)
		if if1 = invalid then
			FillWidgets(widgets.wifi, "Fail", "Wifi not found")
			allTestsPassed = false
		else
			if1result = if1.TestInterface()
			ReportToWidgets(widgets.wifi, if1result)
			allTestsPassed = allTestsPassed and if1result.ok

			ipAddress$ = "None"
			gateway$ = "None"
			dns$ = "None"

			if type(if1) = "roNetworkConfiguration" then
				cc = if1.GetCurrentConfig()
				if cc.ip4_address <> "" then
					ipAddress$ = cc.ip4_address
				endif
				if cc.ip4_gateway <> "" then
					gateway$ = cc.ip4_gateway
				endif
				if cc.dns_servers.Count() > 0 then
					dns$ = cc.dns_servers[0]
				endif
			endif

			wifiConfiguration$ = "IP address = " + ipAddress$ + ", Gateway = " + gateway$ + ", DNS = " + dns$
			WriteWidgetStatus(widgets.wifiConfiguration, wifiConfiguration$)
		endif
	else
		FillWidgets(widgets.wifi, "N/A", "")
		FillWidgets(widgets.wifiConfiguration, "N/A", "")
	endif

	systemTime = CreateObject("roSystemTime")
	dateTime = systemTime.GetLocalDateTime()

	if type(if0) = "roNetworkConfiguration" then
		WriteWidgetStatus(widgets.timeServer, if0.GetTimeServer() + ", " + dateTime)
	else if type(if1) = "roNetworkConfiguration" then
		WriteWidgetStatus(widgets.timeServer, if1.GetTimeServer() + ", " + dateTime)
	endif

    SetWidgetRunning(widgets.internet)
	if testInternet then
		internetTestPassed = false
		if testEthernet and type(if0) = "roNetworkConfiguration" then
			net_result = if0.TestInternetConnectivity()
			ReportToWidgets(widgets.internet, net_result)
			allTestsPassed = allTestsPassed and net_result.ok
			internetTestPassed = net_result.ok
		else if testWireless and type(if1) = "roNetworkConfiguration" then
			net_result = if1.TestInternetConnectivity()
			ReportToWidgets(widgets.internet, net_result)
			allTestsPassed = allTestsPassed and net_result.ok
			internetTestPassed = net_result.ok
		else
			FillWidgets(widgets.internet, "N/A", "")
		endif
	else
		FillWidgets(widgets.internet, "N/A", "")
	endif

	' if the internet test was enabled and it passes, then proceed, even if other tests failed
	if testInternet and internetTestPassed then
		allTestsPassed = true
	endif

	return allTestsPassed

End Function


Sub PerformNetworkDiagnostics(testEthernet as Boolean, testWireless As Boolean, testInternet As Boolean)

    widgets = CreateWidgets()

    msgPort = CreateObject("roMessagePort")
	controlPort = CreateObject("roControlPort", "BrightSign")
    controlPort.SetPort(msgPort)

    while true

		widgets.prompt.Hide()
		allTestsPassed = RunTests(widgets, testEthernet, testWireless, testInternet)
		if allTestsPassed then
			widgets.prompt.PushString("Tests passed - proceeding...")
			widgets.prompt.Show()
			sleep(4000)
			return
		endif

		remainingSeconds% = 30

		while remainingSeconds% > 0
			widgets.prompt.PushString("Restarting test in" + stri(remainingSeconds%) + " seconds")
			widgets.prompt.Show()
	        msg = wait(1000, msgPort)
			if type(msg) = "roControlDown" and msg.GetInt()=12 then
				exit while
			endif
			remainingSeconds% = remainingSeconds% - 1
		end while

    end while

End Sub