GoogleDriveUserItem.cs 6.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228
  1. using Google.Apis.Auth.OAuth2;
  2. using Google.Apis.Download;
  3. using Google.Apis.Drive.v3;
  4. using Google.Apis.Services;
  5. using Google.Apis.Util.Store;
  6. using PDF_Office.Model.CloudDrive;
  7. using System;
  8. using System.Collections.Generic;
  9. using System.IO;
  10. using System.Linq;
  11. using System.Text;
  12. using System.Threading;
  13. using System.Threading.Tasks;
  14. namespace PDF_Office.ViewModels.HomePanel.CloudDrive
  15. {
  16. //对单用户账号处理核心功能
  17. public class GoogleDriveUserItem
  18. {
  19. public GoogleDriveUser User { get; set; }
  20. public DriveService Service { get; set; }
  21. private List<GoogleDriveFiles> GoogleDriveFilesList = new List<GoogleDriveFiles>();
  22. public GoogleDriveUserItem()
  23. {
  24. User = new GoogleDriveUser();
  25. }
  26. #region 对用户账号处理
  27. public async Task<bool> RemoveUser()
  28. {
  29. bool result = false;
  30. if (User.CurrentCredential != null)
  31. {
  32. result = await User.CurrentCredential.RevokeTokenAsync(CancellationToken.None);
  33. }
  34. if (result == true)
  35. User.CurrentCredential = null;
  36. return result;
  37. }
  38. /// <summary>
  39. /// 获取帐号
  40. /// </summary>
  41. /// <returns>返回用户帐号地址</returns>
  42. private async Task<string> GetUserAcount()
  43. {
  44. string userAcount = "";
  45. if (Service != null)
  46. {
  47. var about = Service.About.Get();
  48. about.Fields = "user";
  49. var uss = about.Execute().User;
  50. userAcount = uss.EmailAddress;
  51. }
  52. return userAcount;
  53. }
  54. /// <summary>
  55. /// 获取帐号
  56. /// </summary>
  57. /// <returns>返回用户帐号地址</returns>
  58. public async Task<string> GetUserAcountAsync()
  59. {
  60. return await GetUserAcount();
  61. }
  62. #endregion
  63. #region 文件夹
  64. #endregion
  65. #region 文件
  66. /// <summary>
  67. /// 对单个用户,获取文件列表
  68. /// </summary>
  69. public async Task<List<GoogleDriveFiles>> GetDriveFiles(DriveService service = null)
  70. {
  71. if (service == null)
  72. return null;
  73. // define parameters of request.
  74. FilesResource.ListRequest FileListRequest = service.Files.List();
  75. //listRequest.PageSize = 10;
  76. //listRequest.PageToken = 10;
  77. FileListRequest.Fields = "nextPageToken, files(id, name, size, version, createdTime)";
  78. //get file list.
  79. IList<Google.Apis.Drive.v3.Data.File> files = FileListRequest.Execute().Files;
  80. List<GoogleDriveFiles> FileList = new List<GoogleDriveFiles>();
  81. if (files != null && files.Count > 0)
  82. {
  83. foreach (var file in files)
  84. {
  85. GoogleDriveFiles File = new GoogleDriveFiles
  86. {
  87. Id = file.Id,
  88. Name = file.Name,
  89. Size = file.Size,
  90. Version = file.Version,
  91. CreatedTime = file.CreatedTime
  92. };
  93. FileList.Add(File);
  94. }
  95. }
  96. return FileList;
  97. }
  98. /// <summary>
  99. /// 上传文件
  100. /// </summary>
  101. /// <param name="filepath">本地文件路径</param>
  102. public async Task<bool> FileUpload(string filepath)
  103. {
  104. if (Service == null)
  105. return false;
  106. var FileMetaData = new Google.Apis.Drive.v3.Data.File();
  107. var str = filepath.LastIndexOf("\\");
  108. var str2 = filepath.Substring(str + 1, filepath.Length - str - 1);
  109. FileMetaData.Name = str2;
  110. FileMetaData.MimeType = "";
  111. FilesResource.CreateMediaUpload request;
  112. try
  113. {
  114. using (var stream = new System.IO.FileStream(filepath, System.IO.FileMode.Open))
  115. {
  116. request = Service.Files.Create(FileMetaData, stream, FileMetaData.MimeType);
  117. request.Fields = "id";
  118. request.Upload();
  119. }
  120. return true;
  121. }
  122. catch
  123. {
  124. return false;
  125. }
  126. }
  127. /// <summary>
  128. /// Download file from Google Drive by fileId.
  129. /// </summary>
  130. /// <param name="googleDriveFiles"></param>
  131. /// <param name="savePath"></param>
  132. /// <returns></returns>
  133. //
  134. public async Task<string> DownloadGoogleFile(GoogleDriveFiles googleDriveFiles, string savePath)
  135. {
  136. if (Service == null)
  137. return "";
  138. string fileId = googleDriveFiles.Id;
  139. if (string.IsNullOrEmpty(fileId))
  140. return "";
  141. //fileId = DriveFiles[0].Id;
  142. FilesResource.GetRequest request = Service.Files.Get(fileId);
  143. string FileName = request.Execute().Name;
  144. string FilePath = System.IO.Path.Combine(savePath, FileName);
  145. MemoryStream stream = new MemoryStream();
  146. // Add a handler which will be notified on progress changes.
  147. // It will notify on each chunk download and when the
  148. // download is completed or failed.
  149. request.MediaDownloader.ProgressChanged += (Google.Apis.Download.IDownloadProgress progress) =>
  150. {
  151. switch (progress.Status)
  152. {
  153. case DownloadStatus.Downloading:
  154. {
  155. Console.WriteLine(progress.BytesDownloaded);
  156. break;
  157. }
  158. case DownloadStatus.Completed:
  159. {
  160. Console.WriteLine("Download complete.");
  161. SaveStream(stream, FilePath);
  162. break;
  163. }
  164. case DownloadStatus.Failed:
  165. {
  166. Console.WriteLine("Download failed.");
  167. break;
  168. }
  169. }
  170. };
  171. request.Download(stream);
  172. if (string.IsNullOrEmpty(savePath) == false)
  173. {
  174. }
  175. return FilePath;
  176. }
  177. private static void SaveStream(MemoryStream stream, string FilePath)
  178. {
  179. using (System.IO.FileStream file = new FileStream(FilePath, FileMode.Create, FileAccess.ReadWrite))
  180. {
  181. stream.WriteTo(file);
  182. }
  183. }
  184. #endregion
  185. }
  186. }