FieldWorkClient/ViewModel/Loyout/LoyoutViewModel.cs

726 lines
27 KiB
C#

using System.ComponentModel;
using System.Diagnostics;
using System.IO;
using System.Timers;
using System.Windows;
using System.Windows.Input;
using System.Windows.Threading;
using HeBianGu.Base.WpfBase;
using HeBianGu.Service.Mvc;
using Hopetry.Models;
using Hopetry.Services;
using Hopetry.ViewModel.Send;
using Microsoft.Extensions.Configuration;
using Microsoft.Win32;
using Microsoft.WindowsAPICodePack.Dialogs;
using Minio;
using Microsoft.Extensions.Configuration;
using Hopetry.ViewModel.Send;
using System.IO;
using System.Security.AccessControl;
using System.Diagnostics;
using Microsoft.WindowsAPICodePack.Dialogs;
using Hopetry.Models;
using Yitter.IdGenerator;
using Minio.Exceptions;
using static System.Windows.Forms.VisualStyles.VisualStyleElement.Window;
using HeBianGu.Control.Explorer;
using Hopetry.Provider.Behaviors;
using System.Text;
using Minio.ApiEndpoints;
using Minio.DataModel;
using Minio.DataModel.Args;
using Timer = System.Timers.Timer;
namespace HeBianGu.App.Disk
{
[ViewModel("Loyout")]
internal class LoyoutViewModel : MvcViewModelBase
{
private readonly FileUploadService _uploadService;
private string _path;
/// <summary> 说明 </summary>
public string Path
{
get { return _path; }
set
{
_path = value;
RaisePropertyChanged("Path");
}
}
private string _nearPath;
/// <summary> 说明 </summary>
public string NearPath
{
get { return _nearPath; }
set
{
_nearPath = value;
RaisePropertyChanged("NearPath");
}
}
private string _sharePath;
/// <summary> 说明 </summary>
public string SharePath
{
get { return _sharePath; }
set
{
_sharePath = value;
RaisePropertyChanged("SharePath");
}
}
protected override void Init()
{
Path = Environment.GetFolderPath(Environment.SpecialFolder.MyComputer);
NearPath = Environment.GetFolderPath(Environment.SpecialFolder.Recent);
SharePath = Environment.GetFolderPath(Environment.SpecialFolder.Desktop);
// LinkActions.Add(new LinkAction() { Action = "Near", Controller = "Loyout", DisplayName = "最近使用", Logo = "\xe6f3" });
LinkActions.Add(new LinkAction() { Action = "Explorer", Controller = "Loyout", DisplayName = "全部文件", Logo = "" });
// LinkActions.Add(new LinkAction() { Action = "Image", Controller = "Loyout", DisplayName = " 图片", Logo = "" });
// LinkActions.Add(new LinkAction() { Action = "Video", Controller = "Loyout", DisplayName = " 视频", Logo = "" });
// LinkActions.Add(new LinkAction() { Action = "Document", Controller = "Loyout", DisplayName = " 文档", Logo = "" });
// LinkActions.Add(new LinkAction() { Action = "Music", Controller = "Loyout", DisplayName = " 音乐", Logo = "" });
// LinkActions.Add(new LinkAction() { Action = "Explorer", Controller = "Loyout", DisplayName = " 种子", Logo = "" });
//LinkActions.Add(new LinkAction() { Action = "Recent", Controller = "Loyout", DisplayName = " 其他", Logo = "" });
// LinkActions.Add(new LinkAction() { Action = "Space", Controller = "Loyout", DisplayName = "隐藏空间", Logo = "\xe613" });
//LinkActions.Add(new LinkAction() { Action = "Share", Controller = "Loyout", DisplayName = "我的分享", Logo = "\xe764" });
// LinkActions.Add(new LinkAction() { Action = "Near", Controller = "Loyout", DisplayName = "回收站", Logo = "\xe618" });
Application.Current.Dispatcher.BeginInvoke(DispatcherPriority.Loaded, new Action(() =>
{
SelectLink = LinkActions[0];
}));
}
protected override void Loaded(string args)
{
}
#region 文件上传
private SendViewModel _sendViewModel;
private IConfiguration config;
private SemaphoreSlim _semaphore = new SemaphoreSlim(5);
private Timer _progressTimer;
private bool _isTimerRunning = false;
private object _timerLock = new object();
private bool _isUploading = false; // 新增:标记是否有上传任务正在运行
private Task _currentUploadTask = null; // 新增:当前上传任务
private CancellationTokenSource _uploadCancellation = new CancellationTokenSource();
private int _uploadCount = 0;
private int _completeCount = 0;
private ExplorerMinIOBehavior _explorerBehavior;
public ICommand UploadCommand { get; }
public ICommand UploadCommand1 { get; }
public ICommand CreateFolderCommand { get; }
public ICommand DeleteFolderCommand { get; }
//public ICommand SelectItemsCommand { get; }
public LoyoutViewModel(SendViewModel sendViewModel)
{
_sendViewModel = sendViewModel;
_uploadService = new FileUploadService();
UploadCommand = new AsyncRelayCommand(async () => await UploadFile());
UploadCommand1 = new AsyncRelayCommand(async () => await UploadFile1());
CreateFolderCommand = new AsyncRelayCommand(async () => await CreateNewFolderAsync());
DeleteFolderCommand = new AsyncRelayCommand(async () => await DeleteSelectedItemsAsync());
//SelectItemsCommand = new AsyncRelayCommand(async () => await RelayCommand<SystemInfoModel>(SelectItem)SelectItem());
// 初始化Timer
_progressTimer = new Timer(1000);
_progressTimer.Elapsed += UpdateProgress;
_sendViewModel.UpLoadItems.CollectionChanged += (s, e) =>
{
_sendViewModel.UpdateUploadingItems();
};
}
// 添加设置Behavior的方法
public void SetExplorerBehavior(ExplorerMinIOBehavior behavior)
{
_explorerBehavior = behavior;
}
private void UpdateProgress(object sender, System.Timers.ElapsedEventArgs e)
{
lock (_timerLock)
{
if (_sendViewModel.UpLoadItems.Count == 0) return;
double currentBytes = _sendViewModel.UpLoadItems.Sum(r => r.Double2);
double totalBytes = _sendViewModel.UpLoadItems.Sum(r => r.Double1);
double progress = Math.Round((currentBytes / totalBytes) * 100, 2);
Application.Current.Dispatcher.Invoke(() =>
{
_sendViewModel.Progress = progress;
});
}
}
private void StartProgressTimer()
{
lock (_timerLock)
{
if (!_isTimerRunning)
{
_progressTimer.Start();
_isTimerRunning = true;
}
}
}
private void StopProgressTimer()
{
lock (_timerLock)
{
if (_isTimerRunning)
{
_progressTimer.Stop();
_isTimerRunning = false;
}
}
}
private async Task UploadFile()
{
var openFileDialog = new OpenFileDialog
{
Multiselect = true,
Filter = "All Files (*.*)|*.*"
};
if (openFileDialog.ShowDialog() == true)
{
// 过滤掉已经存在的文件
var newFiles = openFileDialog.FileNames
.Where(filePath => !_sendViewModel.UpLoadItems.Any(item =>
item.Value5.Equals(filePath, StringComparison.OrdinalIgnoreCase)))
.ToList();
if (newFiles.Count == 0)
{
MessageBox.Show("没有新文件需要上传或文件已在上传队列中");
return;
}
foreach (string filePath in newFiles)
{
var ut = CreateUploadItem(filePath, System.IO.Path.GetFileName(filePath));
_sendViewModel.UpLoadItems.Add(ut);
}
_uploadCount = _sendViewModel.UpLoadItems.Count;
//_uploadCount = _sendViewModel.UpLoadItems.Count;
_completeCount = _sendViewModel.UpLoadItems.Where(r => r.Value3 == "已完成").Count();
_sendViewModel.FileCount = _completeCount + "/" + _uploadCount;
MessageBox.Show("传输列表中可查看进度");
// 如果没有上传任务在运行,则启动上传
if (!_isUploading)
{
await ProcessUploadTasks();
}
}
}
private async Task UploadFile1()
{
var dialog = new CommonOpenFileDialog
{
IsFolderPicker = true,
Title = "请选择上传文件"
};
if (dialog.ShowDialog() == CommonFileDialogResult.Ok)
{
string folderPath = dialog.FileName;
string folderName = new DirectoryInfo(folderPath).Name; // 获取选中的文件夹名称
var files = Directory.GetFiles(folderPath, "*.*", SearchOption.AllDirectories)
.Where(filePath => !_sendViewModel.UpLoadItems.Any(item =>
item.Value5.Equals(filePath, StringComparison.OrdinalIgnoreCase)))
.ToList();
if (files.Count == 0)
{
MessageBox.Show("没有新文件需要上传或文件已在上传队列中");
return;
}
foreach (string filePath in files)
{
string relativePath = folderName + "/" + filePath.Substring(folderPath.Length + 1).Replace('\\', '/');
var ut = CreateUploadItem(filePath, relativePath);
_sendViewModel.UpLoadItems.Add(ut);
}
_uploadCount = _sendViewModel.UpLoadItems.Count;
_completeCount = _sendViewModel.UpLoadItems.Where(r => r.Value3 == "已完成").Count();
_sendViewModel.FileCount = _completeCount + "/" + _uploadCount;
MessageBox.Show("传输列表中可查看进度");
// 如果没有上传任务在运行,则启动上传
if (!_isUploading)
{
await ProcessUploadTasks();
}
}
}
private UpLoadItems CreateUploadItem(string filePath, string objectName)
{
FileInfo fileInfo = new FileInfo(filePath);
string sizeText = fileInfo.Length < 1024 * 1024 ?
$"{Math.Ceiling((decimal)fileInfo.Length / 1024)}KB" :
$"{Math.Ceiling((decimal)fileInfo.Length / (1024 * 1024))}MB";
//写入数据库
FUpload fp= new FUpload();
fp.Id= Guid.NewGuid().ToString();
fp.FileName= objectName;
fp.FileSize = fileInfo.Length;
fp.FilePath = filePath;
fp.FileType = fileInfo.Extension;
fp.CreateTime=DateTime.Now;
fp.IsComplete=false;
fp.FileSizeText=sizeText;
if (_uploadService.AddFile(fp))
{
return new UpLoadItems
{
Value = System.IO.Path.GetFileName(filePath),
Value3 = "等待上传",
Value4 = fp.Id, //唯一标识,与数据库一致
Value5 = filePath,
Value6 = objectName,
Double1 = fileInfo.Length,
Double2 = 0.0,
Bool1 = false,
Value1 = $"0{(fileInfo.Length < 1024 * 1024 ? "KB" : "MB")}/{sizeText}"
};
}
else
{
throw new Exception("文件载入失败");
}
}
private async Task ProcessUploadTasks()
{
_isUploading = true;
StartProgressTimer();
try
{
while (true)
{
// 获取所有未完成的上传项
var pendingItems = _sendViewModel.UpLoadItems
.Where(item => !item.Bool1)
.ToList();
if (!pendingItems.Any())
break; // 没有待上传文件,退出循环
// 创建并运行上传任务
_currentUploadTask = Task.WhenAll(pendingItems.Select(async item =>
{
await _semaphore.WaitAsync(_uploadCancellation.Token);
try
{
await UploadFileToMinIOWithProgress(item);
}
finally
{
_semaphore.Release();
}
}));
await _currentUploadTask;
}
}
catch (OperationCanceledException)
{
// 上传被取消
}
finally
{
await Task.Delay(2000);
StopProgressTimer();
_isUploading = false;
_sendViewModel.UpLoadItems.Clear();
if (MySetting.Instance.IsOn && !_sendViewModel.UpLoadItems.Any(item => !item.Bool1))
{
Shutdown();
}
}
}
private async Task UploadFileToMinIOWithProgress(UpLoadItems ut)
{
ut.Bool1 = true;
var builder = new ConfigurationBuilder()
.SetBasePath(Directory.GetCurrentDirectory())
.AddJsonFile("global.json", optional: false, reloadOnChange: true);
// 构建配置
var config = builder.Build();
// 从配置获取MinIO设置更安全
IMinioClient client = new MinioClient()
.WithEndpoint(config["Minio:Endpoint"])
.WithCredentials(config["Minio:AccessKey"], config["Minio:SecretKey"])
.Build();
try
{
string bucketName = config["Minio:BucketName"];
// 确保桶存在
var beArgs = new BucketExistsArgs().WithBucket(bucketName);
bool found = await client.BucketExistsAsync(beArgs).ConfigureAwait(false);
if (!found)
{
var mbArgs = new MakeBucketArgs().WithBucket(bucketName);
await client.MakeBucketAsync(mbArgs).ConfigureAwait(false);
}
var progress = new Progress<ProgressReport>(progressReport =>
{
Application.Current.Dispatcher.Invoke(() =>
{
ut.Int1 = progressReport.Percentage;
long trans = progressReport.TotalBytesTransferred;
ut.Double2 = trans;
int slashIndex = ut.Value1.IndexOf('/');
string sizePart = ut.Value1.Substring(slashIndex);
string transferredPart = trans < 1024 * 1024 ?
$"{Math.Ceiling((decimal)trans / 1024)}KB" :
$"{Math.Ceiling((decimal)trans / (1024 * 1024))}MB";
ut.Value1 = $"{transferredPart}{sizePart}";
if (progressReport.Percentage == 100)
{
ut.Value7 = DateTime.Now.ToString("yyyy-MM-dd HH:mm:ss");
string id = ut.Value4.ToString();
_uploadService.UpdateFileComplete(id);
_sendViewModel.CompleteItems.Add(ut);
ut.Value3 = "已完成";
_sendViewModel.UpdateUploadingItems();
_completeCount++;
_sendViewModel.FileCount=_completeCount+"/"+_uploadCount;
}
else
{
ut.Value3 = "上传中...";
}
});
});
var putObjectArgs = new PutObjectArgs()
.WithBucket(bucketName)
.WithObject(ut.Value6)
.WithFileName(ut.Value5)
.WithProgress(progress);
await client.PutObjectAsync(putObjectArgs);
}
catch (Exception ex)
{
Application.Current.Dispatcher.Invoke(() =>
{
ut.Value3 = $"上传失败: {ex.Message}";
});
}
}
private void Shutdown()
{
try
{
Process.Start("shutdown", "/s /t 0");
}
catch (Win32Exception ex)
{
MessageBox.Show($"关机失败: {ex.Message}", "错误", MessageBoxButton.OK, MessageBoxImage.Error);
}
}
#endregion
#region 文件列表
private string _currentMinIOPath;
/// <summary> 说明CurrentMinIOPath </summary>
public string CurrentMinIOPath
{
get { return _currentMinIOPath; }
set
{
// 确保路径格式正确
var newPath = value;
if (!string.IsNullOrEmpty(newPath) && !newPath.EndsWith("/") &&
newPath.Contains("/") && !newPath.EndsWith("/"))
{
newPath += "/";
}
if (_currentMinIOPath != newPath)
{
_currentMinIOPath = newPath;
RaisePropertyChanged("CurrentMinIOPath");
// 强制刷新导航栏
if (_explorerBehavior != null)
{
_explorerBehavior.RefreshMinIOPath(newPath);
}
}
}
}
public ObservableCollection<SystemInfoModel> Items { get; } = new ObservableCollection<SystemInfoModel>();
private async Task CreateNewFolderAsync()
{
try
{
//// 获取当前绑定的集合
//var itemsSource = AssociatedObject.ItemsSource as ObservableCollection<SystemInfoModel>;
//if (itemsSource == null) return;
// 创建新的文件夹模型
var newFolder = new MinIODirectoryModel(
new MinIODirectoryInfo(GetCurrentBucket(), "新建文件夹/", false, DateTime.Now))
{
IsRenaming = false // 设置为可编辑状态
};
//// 添加到集合最前面
//itemsSource.Insert(0, newFolder);
// 通知行为类开始编辑
_explorerBehavior?.BeginEditNewFolder(newFolder);
await Task.Delay(100); // 等待UI更新
}
catch (Exception ex)
{
MessageBox.Show($"创建文件夹失败: {ex.Message}");
}
}
public async Task CreateMinIOFolder(string folderName)
{
var builder = new ConfigurationBuilder()
.SetBasePath(Directory.GetCurrentDirectory())
.AddJsonFile("global.json", optional: false, reloadOnChange: true);
// 构建配置
var config = builder.Build();
// 从配置获取MinIO设置更安全
IMinioClient client = new Minio.MinioClient()
.WithEndpoint(config["Minio:Endpoint"])
.WithCredentials(config["Minio:AccessKey"], config["Minio:SecretKey"])
.Build();
if (string.IsNullOrWhiteSpace(folderName))
return;
if (!folderName.EndsWith("/"))
folderName += "/";
try
{
var bucketName = GetCurrentBucket();
var prefix = GetCurrentPrefix();
var beArgs = new BucketExistsArgs().WithBucket(bucketName);
bool found = await client.BucketExistsAsync(beArgs);
if (!found)
{
var mbArgs = new MakeBucketArgs().WithBucket(bucketName);
await client.MakeBucketAsync(mbArgs);
}
using (var emptyStream = new MemoryStream(Encoding.UTF8.GetBytes(" ")))
{
await client.PutObjectAsync(
new PutObjectArgs()
.WithBucket(bucketName)
.WithObject(prefix + folderName)
.WithStreamData(emptyStream)
.WithObjectSize(1) // 必须设置为0
.WithContentType("application/x-directory"));
}
// 刷新当前目录
_explorerBehavior?.RefreshMinIOPath(CurrentMinIOPath);
}
catch (Exception ex)
{
MessageBox.Show($"在MinIO中创建文件夹失败: {ex.Message}");
}
}
private string GetCurrentBucket()
{
if (string.IsNullOrEmpty(CurrentMinIOPath))
return string.Empty;
return CurrentMinIOPath.Split('/')[0];
}
private string GetCurrentPrefix()
{
if (string.IsNullOrEmpty(CurrentMinIOPath) || CurrentMinIOPath.IndexOf('/') < 0)
return string.Empty;
return CurrentMinIOPath.Substring(CurrentMinIOPath.IndexOf('/') + 1);
}
#endregion
#region 删除文件夹
private ObservableCollection<SystemInfoModel> _selectedItems = new ObservableCollection<SystemInfoModel>();
public ObservableCollection<SystemInfoModel> SelectedItems
{
get { return _selectedItems; }
set { _selectedItems = value; RaisePropertyChanged(); }
}
private ICommand _selectItemsCommand;
public ICommand SelectItemsCommand => _selectItemsCommand ?? (_selectItemsCommand = new RelayCommand<SystemInfoModel>(SelectItem));
private void SelectItem(SystemInfoModel item)
{
if (item == null) return;
if (SelectedItems.Contains(item))
{
SelectedItems.Remove(item);
}
else
{
SelectedItems.Add(item);
}
}
private async Task DeleteSelectedItemsAsync()
{
try
{
var selectedItems = _selectedItems;
if (selectedItems == null || !selectedItems.Any())
{
MessageBox.Show("请先选择要删除的项目");
return;
}
var message = $"确定要删除选中的 {selectedItems.Count} 个项目吗?";
var result = MessageBox.Show(message, "确认删除", MessageBoxButton.YesNo);
if (result == MessageBoxResult.Yes)
{
await DeleteMinIOItems(selectedItems);
_explorerBehavior.RefreshMinIOPath(CurrentMinIOPath);
MessageBox.Show($"已成功删除 {selectedItems.Count} 个项目");
}
}
catch (Exception ex)
{
MessageBox.Show($"删除失败: {ex.Message}");
}
}
private async Task DeleteMinIOItems(IEnumerable<SystemInfoModel> items)
{
var builder = new ConfigurationBuilder()
.SetBasePath(Directory.GetCurrentDirectory())
.AddJsonFile("global.json", optional: false, reloadOnChange: true);
var config = builder.Build();
using var client = new MinioClient()
.WithEndpoint(config["Minio:Endpoint"])
.WithCredentials(config["Minio:AccessKey"], config["Minio:SecretKey"])
.Build();
// 按桶分组
var itemsByBucket = items.GroupBy(item =>
{
if (item is MinIOFileModel file)
return ((MinIOFileInfo)file.Model).BucketName;
if (item is MinIODirectoryModel dir)
return ((MinIODirectoryInfo)dir.Model).BucketName;
return null;
});
foreach (var bucketGroup in itemsByBucket.Where(g => g.Key != null))
{
var bucketName = bucketGroup.Key;
var objectsToDelete = new List<string>();
foreach (var item in bucketGroup)
{
if (item is MinIOFileModel fileModel)
{
var fileInfo = (MinIOFileInfo)fileModel.Model;
objectsToDelete.Add(fileInfo.FullName.Substring(fileInfo.BucketName.Length + 1));
}
else if (item is MinIODirectoryModel dirModel)
{
var dirInfo = (MinIODirectoryInfo)dirModel.Model;
var folderPath = dirInfo.FullName.Substring(dirInfo.BucketName.Length + 1);
// 确保文件夹路径以斜杠结尾
if (!folderPath.EndsWith("/")) folderPath += "/";
// 获取文件夹下所有对象
var listArgs = new ListObjectsArgs()
.WithBucket(bucketName)
.WithPrefix(folderPath)
.WithRecursive(true);
await foreach (var obj in client.ListObjectsEnumAsync(listArgs))
{
objectsToDelete.Add(obj.Key);
}
// 添加文件夹标记本身
objectsToDelete.Add(folderPath);
}
}
// 批量删除
if (objectsToDelete.Any())
{
await client.RemoveObjectsAsync(
new RemoveObjectsArgs()
.WithBucket(bucketName)
.WithObjects(objectsToDelete));
}
}
}
#endregion
}
internal class DataFileViewModel : ObservableSourceViewModel<TestViewModel>
{
protected override void Init()
{
base.Init();
}
}
}