NatTypeTester/NatTypeTester.ViewModels/RFC3489ViewModel.cs

using Dns.Net.Abstractions;
using JetBrains.Annotations;
using Microsoft;
using NatTypeTester.Models;
using ReactiveUI;
using STUN;
using STUN.Client;
using STUN.Proxy;
using STUN.StunResult;
using System;
using System.Net;
using System.Reactive;
using System.Reactive.Linq;
using System.Threading;
using System.Threading.Tasks;

namespace NatTypeTester.ViewModels
{
	[UsedImplicitly]
	public class RFC3489ViewModel : ViewModelBase, IRoutableViewModel
	{
		public string UrlPathSegment => @"RFC3489";
		public IScreen HostScreen => LazyServiceProvider.LazyGetRequiredService<IScreen>();

		private Config Config => LazyServiceProvider.LazyGetRequiredService<Config>();

		private IDnsClient DnsClient => LazyServiceProvider.LazyGetRequiredService<IDnsClient>();

		public ClassicStunResult Result3489 { get; set; }

		public ReactiveCommand<Unit, Unit> TestClassicNatType { get; }

		public RFC3489ViewModel()
		{
			Result3489 = new ClassicStunResult { LocalEndPoint = new IPEndPoint(IPAddress.Any, 0) };
			TestClassicNatType = ReactiveCommand.CreateFromTask(TestClassicNatTypeImpl);
		}

		private async Task TestClassicNatTypeImpl(CancellationToken token)
		{
			Verify.Operation(StunServer.TryParse(Config.StunServer, out var server), @"Wrong STUN Server!");

			using var proxy = ProxyFactory.CreateProxy(
					Config.ProxyType,
					Result3489.LocalEndPoint,
					IPEndPoint.Parse(Config.ProxyServer),
					Config.ProxyUser, Config.ProxyPassword
			);

			var ip = await DnsClient.QueryAsync(server.Hostname, token);
			using var client = new StunClient3489(ip, server.Port, Result3489.LocalEndPoint, proxy);

			Result3489 = client.Status;
			using (Observable.Interval(TimeSpan.FromSeconds(0.1))
					.ObserveOn(RxApp.MainThreadScheduler)
					.Subscribe(_ => this.RaisePropertyChanged(nameof(Result3489))))
			{
				await client.QueryAsync(token);
			}

			Result3489.LocalEndPoint = client.LocalEndPoint;

			this.RaisePropertyChanged(nameof(Result3489));
		}
	}
}