BoneSync/Facepunch.Steamworks/ServerList/IpList.cs

using Facepunch.Steamworks.Data;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Runtime.InteropServices;
using System.Text;
using System.Threading.Tasks;

namespace Facepunch.Steamworks.ServerList
{
	public class IpList : Internet
	{
		public List<string> Ips = new List<string>();
		bool wantsCancel;

		public IpList( IEnumerable<string> list )
		{
			Ips.AddRange( list );
		}

		public IpList( params string[] list )
		{
			Ips.AddRange( list );
		}

		public override async Task<bool> RunQueryAsync( float timeoutSeconds = 10 )
		{
			int blockSize = 16;
			int pointer = 0;

			var ips = Ips.ToArray();

			while ( true )
			{
				var sublist = ips.Skip( pointer ).Take( blockSize );
				if ( sublist.Count() == 0 )
					break;

				using ( var list = new ServerList.Internet() )
				{
					list.AddFilter( "or", sublist.Count().ToString() );

					foreach ( var server in sublist )
					{
						list.AddFilter( "gameaddr", server );
					}

					await list.RunQueryAsync( timeoutSeconds );

					if ( wantsCancel )
						return false;

					Responsive.AddRange( list.Responsive );
					Responsive = Responsive.Distinct().ToList();
					Unresponsive.AddRange( list.Unresponsive );
					Unresponsive = Unresponsive.Distinct().ToList();
				}

				pointer += sublist.Count();

				InvokeChanges();
			}

			return true;
		}

		public override void Cancel()
		{
			wantsCancel = true;
		}
	}
}