mirror of
https://github.com/jellyfin/jellyfin.git
synced 2024-09-06 03:26:07 +02:00
153 lines
4.5 KiB
C#
153 lines
4.5 KiB
C#
|
using System;
|
|||
|
using System.Collections.Generic;
|
|||
|
using System.IO;
|
|||
|
using System.Linq;
|
|||
|
using System.Threading;
|
|||
|
using System.Threading.Tasks;
|
|||
|
using MediaBrowser.Model.Entities;
|
|||
|
|
|||
|
namespace MediaBrowser.Controller.IO
|
|||
|
{
|
|||
|
public class DirectoryWatchers
|
|||
|
{
|
|||
|
private List<FileSystemWatcher> FileSystemWatchers = new List<FileSystemWatcher>();
|
|||
|
private Timer updateTimer = null;
|
|||
|
private List<string> affectedPaths = new List<string>();
|
|||
|
|
|||
|
private const int TimerDelayInSeconds = 5;
|
|||
|
|
|||
|
public void Start()
|
|||
|
{
|
|||
|
List<string> pathsToWatch = new List<string>();
|
|||
|
|
|||
|
var rootFolder = Kernel.Instance.RootFolder;
|
|||
|
|
|||
|
pathsToWatch.Add(rootFolder.Path);
|
|||
|
|
|||
|
foreach (Folder folder in rootFolder.FolderChildren)
|
|||
|
{
|
|||
|
foreach (Folder subFolder in folder.FolderChildren)
|
|||
|
{
|
|||
|
if (Path.IsPathRooted(subFolder.Path))
|
|||
|
{
|
|||
|
string parent = Path.GetDirectoryName(subFolder.Path);
|
|||
|
|
|||
|
if (!pathsToWatch.Contains(parent))
|
|||
|
{
|
|||
|
pathsToWatch.Add(parent);
|
|||
|
}
|
|||
|
}
|
|||
|
}
|
|||
|
}
|
|||
|
|
|||
|
foreach (string path in pathsToWatch)
|
|||
|
{
|
|||
|
FileSystemWatcher watcher = new FileSystemWatcher(path, "*");
|
|||
|
|
|||
|
watcher.IncludeSubdirectories = true;
|
|||
|
|
|||
|
watcher.Changed += watcher_Changed;
|
|||
|
|
|||
|
// All the others seem to trigger change events on the parent, so let's keep it simple for now.
|
|||
|
//watcher.Created += watcher_Changed;
|
|||
|
//watcher.Deleted += watcher_Changed;
|
|||
|
//watcher.Renamed += watcher_Changed;
|
|||
|
|
|||
|
watcher.EnableRaisingEvents = true;
|
|||
|
FileSystemWatchers.Add(watcher);
|
|||
|
}
|
|||
|
}
|
|||
|
|
|||
|
void watcher_Changed(object sender, FileSystemEventArgs e)
|
|||
|
{
|
|||
|
if (!affectedPaths.Contains(e.FullPath))
|
|||
|
{
|
|||
|
affectedPaths.Add(e.FullPath);
|
|||
|
}
|
|||
|
|
|||
|
if (updateTimer == null)
|
|||
|
{
|
|||
|
updateTimer = new Timer(TimerStopped, null, TimeSpan.FromSeconds(TimerDelayInSeconds), TimeSpan.FromMilliseconds(-1));
|
|||
|
}
|
|||
|
else
|
|||
|
{
|
|||
|
updateTimer.Change(TimeSpan.FromSeconds(TimerDelayInSeconds), TimeSpan.FromMilliseconds(-1));
|
|||
|
}
|
|||
|
}
|
|||
|
|
|||
|
private void TimerStopped(object stateInfo)
|
|||
|
{
|
|||
|
updateTimer.Dispose();
|
|||
|
updateTimer = null;
|
|||
|
|
|||
|
List<string> paths = affectedPaths;
|
|||
|
affectedPaths = new List<string>();
|
|||
|
|
|||
|
ProcessPathChanges(paths);
|
|||
|
}
|
|||
|
|
|||
|
private void ProcessPathChanges(IEnumerable<string> paths)
|
|||
|
{
|
|||
|
List<BaseItem> itemsToRefresh = new List<BaseItem>();
|
|||
|
|
|||
|
foreach (BaseItem item in paths.Select(p => GetAffectedBaseItem(p)))
|
|||
|
{
|
|||
|
if (item != null && !itemsToRefresh.Contains(item))
|
|||
|
{
|
|||
|
itemsToRefresh.Add(item);
|
|||
|
}
|
|||
|
}
|
|||
|
|
|||
|
if (itemsToRefresh.Any(i =>
|
|||
|
{
|
|||
|
var folder = i as Folder;
|
|||
|
|
|||
|
return folder != null && folder.IsRoot;
|
|||
|
}))
|
|||
|
{
|
|||
|
Kernel.Instance.ReloadRoot();
|
|||
|
}
|
|||
|
else
|
|||
|
{
|
|||
|
Parallel.For(0, itemsToRefresh.Count, i =>
|
|||
|
{
|
|||
|
Kernel.Instance.ReloadItem(itemsToRefresh[i]);
|
|||
|
});
|
|||
|
}
|
|||
|
}
|
|||
|
|
|||
|
private BaseItem GetAffectedBaseItem(string path)
|
|||
|
{
|
|||
|
BaseItem item = null;
|
|||
|
|
|||
|
while (item == null)
|
|||
|
{
|
|||
|
item = Kernel.Instance.RootFolder.FindByPath(path);
|
|||
|
|
|||
|
path = Path.GetDirectoryName(path);
|
|||
|
}
|
|||
|
|
|||
|
return item;
|
|||
|
}
|
|||
|
|
|||
|
public void Stop()
|
|||
|
{
|
|||
|
foreach (FileSystemWatcher watcher in FileSystemWatchers)
|
|||
|
{
|
|||
|
watcher.Changed -= watcher_Changed;
|
|||
|
watcher.EnableRaisingEvents = false;
|
|||
|
watcher.Dispose();
|
|||
|
}
|
|||
|
|
|||
|
if (updateTimer != null)
|
|||
|
{
|
|||
|
updateTimer.Dispose();
|
|||
|
updateTimer = null;
|
|||
|
}
|
|||
|
|
|||
|
FileSystemWatchers.Clear();
|
|||
|
affectedPaths.Clear();
|
|||
|
}
|
|||
|
}
|
|||
|
}
|