yasirkula · October 26, 2024 15:48 · derekantrican · Mar 3, 2022 · ActionX · Feb 17, 2023
diff --git a/FileDownloader.cs b/FileDownloader.cs
 using System;
 using System.Collections.Generic;
 using System.ComponentModel;
 using System.IO;
 using System.Net;
 using System.Text;

 /* EXAMPLE USAGE
 	FileDownloader fileDownloader = new FileDownloader();

 	// This callback is triggered for DownloadFileAsync only
 	fileDownloader.DownloadProgressChanged += ( sender, e ) => Console.WriteLine( "Progress changed " + e.BytesReceived + " " + e.TotalBytesToReceive );
 	// This callback is triggered for both DownloadFile and DownloadFileAsync
 	fileDownloader.DownloadFileCompleted += ( sender, e ) => 
 	{
 		if( e.Cancelled )
 			Console.WriteLine( "Download cancelled" );
 		else if( e.Error != null )
 			Console.WriteLine( "Download failed: " + e.Error );
 		else
 			Console.WriteLine( "Download completed" );
 	};

 	fileDownloader.DownloadFileAsync( "https://INSERT_DOWNLOAD_LINK_HERE", @"C:\downloadedFile.txt" );
 */
 public class FileDownloader : IDisposable
 {
 	private const string GOOGLE_DRIVE_DOMAIN = "drive.google.com";
 	private const string GOOGLE_DRIVE_DOMAIN2 = "https://drive.google.com";

 	// In the worst case, it is necessary to send 3 download requests to the Drive address
 	//   1. an NID cookie is returned instead of a download_warning cookie
 	//   2. download_warning cookie returned
 	//   3. the actual file is downloaded
 	private const int GOOGLE_DRIVE_MAX_DOWNLOAD_ATTEMPT = 3;

 	public delegate void DownloadProgressChangedEventHandler( object sender, DownloadProgress progress );

 	// Custom download progress reporting (needed for Google Drive)
 	public class DownloadProgress
 	{
 		public long BytesReceived, TotalBytesToReceive;
 		public object UserState;

 		public int ProgressPercentage
 		{
 			get
 			{
 				if( TotalBytesToReceive > 0L )
 					return (int) ( ( (double) BytesReceived / TotalBytesToReceive ) * 100 );

 				return 0;
 			}
 		}
 	}

 	// Web client that preserves cookies (needed for Google Drive)
 	private class CookieAwareWebClient : WebClient
 	{
 		private class CookieContainer
 		{
 			private readonly Dictionary<string, string> cookies = new Dictionary<string, string>();

 			public string this[Uri address]
 			{
 				get
 				{
 					string cookie;
 					if( cookies.TryGetValue( address.Host, out cookie ) )
 						return cookie;

 					return null;
 				}
 				set
 				{
 					cookies[address.Host] = value;
 				}
 			}
 		}

 		private readonly CookieContainer cookies = new CookieContainer();
 		public DownloadProgress ContentRangeTarget;

 		protected override WebRequest GetWebRequest( Uri address )
 		{
 			WebRequest request = base.GetWebRequest( address );
 			if( request is HttpWebRequest )
 			{
 				string cookie = cookies[address];
 				if( cookie != null )
 					( (HttpWebRequest) request ).Headers.Set( "cookie", cookie );

 				if( ContentRangeTarget != null )
 					( (HttpWebRequest) request ).AddRange( 0 );
 			}

 			return request;
 		}

 		protected override WebResponse GetWebResponse( WebRequest request, IAsyncResult result )
 		{
 			return ProcessResponse( base.GetWebResponse( request, result ) );
 		}

 		protected override WebResponse GetWebResponse( WebRequest request )
 		{
 			return ProcessResponse( base.GetWebResponse( request ) );
 		}

 		private WebResponse ProcessResponse( WebResponse response )
 		{
 			string[] cookies = response.Headers.GetValues( "Set-Cookie" );
 			if( cookies != null && cookies.Length > 0 )
 			{
 				int length = 0;
 				for( int i = 0; i < cookies.Length; i++ )
 					length += cookies[i].Length;

 				StringBuilder cookie = new StringBuilder( length );
 				for( int i = 0; i < cookies.Length; i++ )
 					cookie.Append( cookies[i] );

 				this.cookies[response.ResponseUri] = cookie.ToString();
 			}

 			if( ContentRangeTarget != null )
 			{
 				string[] rangeLengthHeader = response.Headers.GetValues( "Content-Range" );
 				if( rangeLengthHeader != null && rangeLengthHeader.Length > 0 )
 				{
 					int splitIndex = rangeLengthHeader[0].LastIndexOf( '/' );
 					if( splitIndex >= 0 && splitIndex < rangeLengthHeader[0].Length - 1 )
 					{
 						long length;
 						if( long.TryParse( rangeLengthHeader[0].Substring( splitIndex + 1 ), out length ) )
 							ContentRangeTarget.TotalBytesToReceive = length;
 					}
 				}
 			}

 			return response;
 		}
 	}

 	private readonly CookieAwareWebClient webClient;
 	private readonly DownloadProgress downloadProgress;

 	private Uri downloadAddress;
 	private string downloadPath;

 	private bool asyncDownload;
 	private object userToken;

 	private bool downloadingDriveFile;
 	private int driveDownloadAttempt;

 	public event DownloadProgressChangedEventHandler DownloadProgressChanged;
 	public event AsyncCompletedEventHandler DownloadFileCompleted;

 	public FileDownloader()
 	{
 		webClient = new CookieAwareWebClient();
 		webClient.DownloadProgressChanged += DownloadProgressChangedCallback;
 		webClient.DownloadFileCompleted += DownloadFileCompletedCallback;

 		downloadProgress = new DownloadProgress();
 	}

 	public void DownloadFile( string address, string fileName )
 	{
 		DownloadFile( address, fileName, false, null );
 	}

 	public void DownloadFileAsync( string address, string fileName, object userToken = null )
 	{
 		DownloadFile( address, fileName, true, userToken );
 	}

 	private void DownloadFile( string address, string fileName, bool asyncDownload, object userToken )
 	{
 		downloadingDriveFile = address.StartsWith( GOOGLE_DRIVE_DOMAIN ) || address.StartsWith( GOOGLE_DRIVE_DOMAIN2 );
 		if( downloadingDriveFile )
 		{
 			address = GetGoogleDriveDownloadAddress( address );
 			driveDownloadAttempt = 1;

 			webClient.ContentRangeTarget = downloadProgress;
 		}
 		else
 			webClient.ContentRangeTarget = null;

 		downloadAddress = new Uri( address );
 		downloadPath = fileName;

 		downloadProgress.TotalBytesToReceive = -1L;
 		downloadProgress.UserState = userToken;

 		this.asyncDownload = asyncDownload;
 		this.userToken = userToken;

 		DownloadFileInternal();
 	}

 	private void DownloadFileInternal()
 	{
 		if( !asyncDownload )
 		{
 			webClient.DownloadFile( downloadAddress, downloadPath );

 			// This callback isn't triggered for synchronous downloads, manually trigger it
 			DownloadFileCompletedCallback( webClient, new AsyncCompletedEventArgs( null, false, null ) );
 		}
 		else if( userToken == null )
 			webClient.DownloadFileAsync( downloadAddress, downloadPath );
 		else
 			webClient.DownloadFileAsync( downloadAddress, downloadPath, userToken );
 	}

 	private void DownloadProgressChangedCallback( object sender, DownloadProgressChangedEventArgs e )
 	{
 		if( DownloadProgressChanged != null )
 		{
 			downloadProgress.BytesReceived = e.BytesReceived;
 			if( e.TotalBytesToReceive > 0L )
 				downloadProgress.TotalBytesToReceive = e.TotalBytesToReceive;

 			DownloadProgressChanged( this, downloadProgress );
 		}
 	}

 	private void DownloadFileCompletedCallback( object sender, AsyncCompletedEventArgs e )
 	{
 		if( !downloadingDriveFile )
 		{
 			if( DownloadFileCompleted != null )
 				DownloadFileCompleted( this, e );
 		}
 		else
 		{
 			if( driveDownloadAttempt < GOOGLE_DRIVE_MAX_DOWNLOAD_ATTEMPT && !ProcessDriveDownload() )
 			{
 				// Try downloading the Drive file again
 				driveDownloadAttempt++;
 				DownloadFileInternal();
 			}
 			else if( DownloadFileCompleted != null )
 				DownloadFileCompleted( this, e );
 		}
 	}

 	// Downloading large files from Google Drive prompts a warning screen and requires manual confirmation
 	// Consider that case and try to confirm the download automatically if warning prompt occurs
 	// Returns true, if no more download requests are necessary
 	private bool ProcessDriveDownload()
 	{
 		FileInfo downloadedFile = new FileInfo( downloadPath );
 		if( downloadedFile == null )
 			return true;

 		// Confirmation page is around 50KB, shouldn't be larger than 60KB
 		if( downloadedFile.Length > 60000L )
 			return true;

 		// Downloaded file might be the confirmation page, check it
 		string content;
 		using( var reader = downloadedFile.OpenText() )
 		{
 			// Confirmation page starts with <!DOCTYPE html>, which can be preceeded by a newline
 			char[] header = new char[20];
 			int readCount = reader.ReadBlock( header, 0, 20 );
 			if( readCount < 20 || !( new string( header ).Contains( "<!DOCTYPE html>" ) ) )
 				return true;

 			content = reader.ReadToEnd();
 		}

 		int linkIndex = content.LastIndexOf( "href=\"/uc?" );
 		if( linkIndex >= 0 )
 		{
 			linkIndex += 6;
 			int linkEnd = content.IndexOf( '"', linkIndex );
 			if( linkEnd >= 0 )
 			{
 				downloadAddress = new Uri( "https://drive.google.com" + content.Substring( linkIndex, linkEnd - linkIndex ).Replace( "&amp;", "&" ) );
 				return false;
 			}
 		}

 		int formIndex = content.LastIndexOf( "<form id=\"download-form\"" );
 		if( formIndex >= 0 )
 		{
 			int formEndIndex = content.IndexOf( "</form>", formIndex + 10 );
 			int inputIndex = formIndex;
 			StringBuilder sb = new StringBuilder().Append( "https://drive.usercontent.google.com/download" );
 			bool isFirstArgument = true;
 			while( ( inputIndex = content.IndexOf( "<input type=\"hidden\"", inputIndex + 10 ) ) >= 0 && inputIndex < formEndIndex )
 			{
 				linkIndex = content.IndexOf( "name=", inputIndex + 10 ) + 6;
 				sb.Append( isFirstArgument ? '?' : '&' ).Append( content, linkIndex, content.IndexOf( '"', linkIndex ) - linkIndex ).Append( '=' );

 				linkIndex = content.IndexOf( "value=", linkIndex ) + 7;
 				sb.Append( content, linkIndex, content.IndexOf( '"', linkIndex ) - linkIndex );

 				isFirstArgument = false;
 			}

 			downloadAddress = new Uri( sb.ToString() );
 			return false;
 		}

 		return true;
 	}

 	// Handles the following formats (links can be preceeded by https://):
 	// - drive.google.com/open?id=FILEID&resourcekey=RESOURCEKEY
 	// - drive.google.com/file/d/FILEID/view?usp=sharing&resourcekey=RESOURCEKEY
 	// - drive.google.com/uc?id=FILEID&export=download&resourcekey=RESOURCEKEY
 	private string GetGoogleDriveDownloadAddress( string address )
 	{
 		int index = address.IndexOf( "id=" );
 		int closingIndex;
 		if( index > 0 )
 		{
 			index += 3;
 			closingIndex = address.IndexOf( '&', index );
 			if( closingIndex < 0 )
 				closingIndex = address.Length;
 		}
 		else
 		{
 			index = address.IndexOf( "file/d/" );
 			if( index < 0 ) // address is not in any of the supported forms
 				return string.Empty;

 			index += 7;

 			closingIndex = address.IndexOf( '/', index );
 			if( closingIndex < 0 )
 			{
 				closingIndex = address.IndexOf( '?', index );
 				if( closingIndex < 0 )
 					closingIndex = address.Length;
 			}
 		}

 		string fileID = address.Substring( index, closingIndex - index );

 		index = address.IndexOf( "resourcekey=" );
 		if( index > 0 )
 		{
 			index += 12;
 			closingIndex = address.IndexOf( '&', index );
 			if( closingIndex < 0 )
 				closingIndex = address.Length;

 			string resourceKey = address.Substring( index, closingIndex - index );
 			return string.Concat( "https://drive.google.com/uc?id=", fileID, "&export=download&resourcekey=", resourceKey, "&confirm=t" );
 		}
 		else
 			return string.Concat( "https://drive.google.com/uc?id=", fileID, "&export=download&confirm=t" );
 	}

 	public void Dispose()
 	{
 		webClient.Dispose();
 	}
 }
	using System;
	using System.Collections.Generic;
	using System.ComponentModel;
	using System.IO;
	using System.Net;
	using System.Text;

	/* EXAMPLE USAGE
	FileDownloader fileDownloader = new FileDownloader();

	// This callback is triggered for DownloadFileAsync only
	fileDownloader.DownloadProgressChanged += ( sender, e ) => Console.WriteLine( "Progress changed " + e.BytesReceived + " " + e.TotalBytesToReceive );
	// This callback is triggered for both DownloadFile and DownloadFileAsync
	fileDownloader.DownloadFileCompleted += ( sender, e ) =>
	{
	if( e.Cancelled )
	Console.WriteLine( "Download cancelled" );
	else if( e.Error != null )
	Console.WriteLine( "Download failed: " + e.Error );
	else
	Console.WriteLine( "Download completed" );
	};

	fileDownloader.DownloadFileAsync( "https://INSERT_DOWNLOAD_LINK_HERE", @"C:\downloadedFile.txt" );
	*/
	public class FileDownloader : IDisposable
	{
	private const string GOOGLE_DRIVE_DOMAIN = "drive.google.com";
	private const string GOOGLE_DRIVE_DOMAIN2 = "https://drive.google.com";

	// In the worst case, it is necessary to send 3 download requests to the Drive address
	// 1. an NID cookie is returned instead of a download_warning cookie
	// 2. download_warning cookie returned
	// 3. the actual file is downloaded
	private const int GOOGLE_DRIVE_MAX_DOWNLOAD_ATTEMPT = 3;

	public delegate void DownloadProgressChangedEventHandler( object sender, DownloadProgress progress );

	// Custom download progress reporting (needed for Google Drive)
	public class DownloadProgress
	{
	public long BytesReceived, TotalBytesToReceive;
	public object UserState;

	public int ProgressPercentage
	{
	get
	{
	if( TotalBytesToReceive > 0L )
	return (int) ( ( (double) BytesReceived / TotalBytesToReceive ) * 100 );

	return 0;
	}
	}
	}

	// Web client that preserves cookies (needed for Google Drive)
	private class CookieAwareWebClient : WebClient
	{
	private class CookieContainer
	{
	private readonly Dictionary<string, string> cookies = new Dictionary<string, string>();

	public string this[Uri address]
	{
	get
	{
	string cookie;
	if( cookies.TryGetValue( address.Host, out cookie ) )
	return cookie;

	return null;
	}
	set
	{
	cookies[address.Host] = value;
	}
	}
	}

	private readonly CookieContainer cookies = new CookieContainer();
	public DownloadProgress ContentRangeTarget;

	protected override WebRequest GetWebRequest( Uri address )
	{
	WebRequest request = base.GetWebRequest( address );
	if( request is HttpWebRequest )
	{
	string cookie = cookies[address];
	if( cookie != null )
	( (HttpWebRequest) request ).Headers.Set( "cookie", cookie );

	if( ContentRangeTarget != null )
	( (HttpWebRequest) request ).AddRange( 0 );
	}

	return request;
	}

	protected override WebResponse GetWebResponse( WebRequest request, IAsyncResult result )
	{
	return ProcessResponse( base.GetWebResponse( request, result ) );
	}

	protected override WebResponse GetWebResponse( WebRequest request )
	{
	return ProcessResponse( base.GetWebResponse( request ) );
	}

	private WebResponse ProcessResponse( WebResponse response )
	{
	string[] cookies = response.Headers.GetValues( "Set-Cookie" );
	if( cookies != null && cookies.Length > 0 )
	{
	int length = 0;
	for( int i = 0; i < cookies.Length; i++ )
	length += cookies[i].Length;

	StringBuilder cookie = new StringBuilder( length );
	for( int i = 0; i < cookies.Length; i++ )
	cookie.Append( cookies[i] );

	this.cookies[response.ResponseUri] = cookie.ToString();
	}

	if( ContentRangeTarget != null )
	{
	string[] rangeLengthHeader = response.Headers.GetValues( "Content-Range" );
	if( rangeLengthHeader != null && rangeLengthHeader.Length > 0 )
	{
	int splitIndex = rangeLengthHeader[0].LastIndexOf( '/' );
	if( splitIndex >= 0 && splitIndex < rangeLengthHeader[0].Length - 1 )
	{
	long length;
	if( long.TryParse( rangeLengthHeader[0].Substring( splitIndex + 1 ), out length ) )
	ContentRangeTarget.TotalBytesToReceive = length;
	}
	}
	}

	return response;
	}
	}

	private readonly CookieAwareWebClient webClient;
	private readonly DownloadProgress downloadProgress;

	private Uri downloadAddress;
	private string downloadPath;

	private bool asyncDownload;
	private object userToken;

	private bool downloadingDriveFile;
	private int driveDownloadAttempt;

	public event DownloadProgressChangedEventHandler DownloadProgressChanged;
	public event AsyncCompletedEventHandler DownloadFileCompleted;

	public FileDownloader()
	{
	webClient = new CookieAwareWebClient();
	webClient.DownloadProgressChanged += DownloadProgressChangedCallback;
	webClient.DownloadFileCompleted += DownloadFileCompletedCallback;

	downloadProgress = new DownloadProgress();
	}

	public void DownloadFile( string address, string fileName )
	{
	DownloadFile( address, fileName, false, null );
	}

	public void DownloadFileAsync( string address, string fileName, object userToken = null )
	{
	DownloadFile( address, fileName, true, userToken );
	}

	private void DownloadFile( string address, string fileName, bool asyncDownload, object userToken )
	{
	downloadingDriveFile = address.StartsWith( GOOGLE_DRIVE_DOMAIN ) \|\| address.StartsWith( GOOGLE_DRIVE_DOMAIN2 );
	if( downloadingDriveFile )
	{
	address = GetGoogleDriveDownloadAddress( address );
	driveDownloadAttempt = 1;

	webClient.ContentRangeTarget = downloadProgress;
	}
	else
	webClient.ContentRangeTarget = null;

	downloadAddress = new Uri( address );
	downloadPath = fileName;

	downloadProgress.TotalBytesToReceive = -1L;
	downloadProgress.UserState = userToken;

	this.asyncDownload = asyncDownload;
	this.userToken = userToken;

	DownloadFileInternal();
	}

	private void DownloadFileInternal()
	{
	if( !asyncDownload )
	{
	webClient.DownloadFile( downloadAddress, downloadPath );

	// This callback isn't triggered for synchronous downloads, manually trigger it
	DownloadFileCompletedCallback( webClient, new AsyncCompletedEventArgs( null, false, null ) );
	}
	else if( userToken == null )
	webClient.DownloadFileAsync( downloadAddress, downloadPath );
	else
	webClient.DownloadFileAsync( downloadAddress, downloadPath, userToken );
	}

	private void DownloadProgressChangedCallback( object sender, DownloadProgressChangedEventArgs e )
	{
	if( DownloadProgressChanged != null )
	{
	downloadProgress.BytesReceived = e.BytesReceived;
	if( e.TotalBytesToReceive > 0L )
	downloadProgress.TotalBytesToReceive = e.TotalBytesToReceive;

	DownloadProgressChanged( this, downloadProgress );
	}
	}

	private void DownloadFileCompletedCallback( object sender, AsyncCompletedEventArgs e )
	{
	if( !downloadingDriveFile )
	{
	if( DownloadFileCompleted != null )
	DownloadFileCompleted( this, e );
	}
	else
	{
	if( driveDownloadAttempt < GOOGLE_DRIVE_MAX_DOWNLOAD_ATTEMPT && !ProcessDriveDownload() )
	{
	// Try downloading the Drive file again
	driveDownloadAttempt++;
	DownloadFileInternal();
	}
	else if( DownloadFileCompleted != null )
	DownloadFileCompleted( this, e );
	}
	}

	// Downloading large files from Google Drive prompts a warning screen and requires manual confirmation
	// Consider that case and try to confirm the download automatically if warning prompt occurs
	// Returns true, if no more download requests are necessary
	private bool ProcessDriveDownload()
	{
	FileInfo downloadedFile = new FileInfo( downloadPath );
	if( downloadedFile == null )
	return true;

	// Confirmation page is around 50KB, shouldn't be larger than 60KB
	if( downloadedFile.Length > 60000L )
	return true;

	// Downloaded file might be the confirmation page, check it
	string content;
	using( var reader = downloadedFile.OpenText() )
	{
	// Confirmation page starts with <!DOCTYPE html>, which can be preceeded by a newline
	char[] header = new char[20];
	int readCount = reader.ReadBlock( header, 0, 20 );
	if( readCount < 20 \|\| !( new string( header ).Contains( "<!DOCTYPE html>" ) ) )
	return true;

	content = reader.ReadToEnd();
	}

	int linkIndex = content.LastIndexOf( "href=\"/uc?" );
	if( linkIndex >= 0 )
	{
	linkIndex += 6;
	int linkEnd = content.IndexOf( '"', linkIndex );
	if( linkEnd >= 0 )
	{
	downloadAddress = new Uri( "https://drive.google.com" + content.Substring( linkIndex, linkEnd - linkIndex ).Replace( "&", "&" ) );
	return false;
	}
	}

	int formIndex = content.LastIndexOf( "<form id=\"download-form\"" );
	if( formIndex >= 0 )
	{
	int formEndIndex = content.IndexOf( "</form>", formIndex + 10 );
	int inputIndex = formIndex;
	StringBuilder sb = new StringBuilder().Append( "https://drive.usercontent.google.com/download" );
	bool isFirstArgument = true;
	while( ( inputIndex = content.IndexOf( "<input type=\"hidden\"", inputIndex + 10 ) ) >= 0 && inputIndex < formEndIndex )
	{
	linkIndex = content.IndexOf( "name=", inputIndex + 10 ) + 6;
	sb.Append( isFirstArgument ? '?' : '&' ).Append( content, linkIndex, content.IndexOf( '"', linkIndex ) - linkIndex ).Append( '=' );

	linkIndex = content.IndexOf( "value=", linkIndex ) + 7;
	sb.Append( content, linkIndex, content.IndexOf( '"', linkIndex ) - linkIndex );

	isFirstArgument = false;
	}

	downloadAddress = new Uri( sb.ToString() );
	return false;
	}

	return true;
	}

	// Handles the following formats (links can be preceeded by https://):
	// - drive.google.com/open?id=FILEID&resourcekey=RESOURCEKEY
	// - drive.google.com/file/d/FILEID/view?usp=sharing&resourcekey=RESOURCEKEY
	// - drive.google.com/uc?id=FILEID&export=download&resourcekey=RESOURCEKEY
	private string GetGoogleDriveDownloadAddress( string address )
	{
	int index = address.IndexOf( "id=" );
	int closingIndex;
	if( index > 0 )
	{
	index += 3;
	closingIndex = address.IndexOf( '&', index );
	if( closingIndex < 0 )
	closingIndex = address.Length;
	}
	else
	{
	index = address.IndexOf( "file/d/" );
	if( index < 0 ) // address is not in any of the supported forms
	return string.Empty;

	index += 7;

	closingIndex = address.IndexOf( '/', index );
	if( closingIndex < 0 )
	{
	closingIndex = address.IndexOf( '?', index );
	if( closingIndex < 0 )
	closingIndex = address.Length;
	}
	}

	string fileID = address.Substring( index, closingIndex - index );

	index = address.IndexOf( "resourcekey=" );
	if( index > 0 )
	{
	index += 12;
	closingIndex = address.IndexOf( '&', index );
	if( closingIndex < 0 )
	closingIndex = address.Length;

	string resourceKey = address.Substring( index, closingIndex - index );
	return string.Concat( "https://drive.google.com/uc?id=", fileID, "&export=download&resourcekey=", resourceKey, "&confirm=t" );
	}
	else
	return string.Concat( "https://drive.google.com/uc?id=", fileID, "&export=download&confirm=t" );
	}

	public void Dispose()
	{
	webClient.Dispose();
	}
	}