mirror of
https://github.com/microsoft/PowerToys.git
synced 2024-12-15 03:59:15 +08:00
85 lines
2.6 KiB
C#
85 lines
2.6 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.IO;
|
|
using System.Linq;
|
|
using Wox.Infrastructure.Exception;
|
|
using Wox.Infrastructure.Logger;
|
|
|
|
namespace Wox.Plugin.Program.ProgramSources
|
|
{
|
|
[Serializable]
|
|
public class FileSystemProgramSource : AbstractProgramSource
|
|
{
|
|
private string baseDirectory;
|
|
private int maxDepth;
|
|
private string suffixes;
|
|
|
|
public FileSystemProgramSource(string baseDirectory, int maxDepth, string suffixes)
|
|
{
|
|
this.baseDirectory = baseDirectory;
|
|
this.maxDepth = maxDepth;
|
|
this.suffixes = suffixes;
|
|
}
|
|
|
|
public FileSystemProgramSource(string baseDirectory)
|
|
: this(baseDirectory, -1, "") {}
|
|
|
|
public FileSystemProgramSource(ProgramSource source)
|
|
: this(source.Location, source.MaxDepth, source.Suffixes)
|
|
{
|
|
this.BonusPoints = source.BonusPoints;
|
|
}
|
|
|
|
public override List<Program> LoadPrograms()
|
|
{
|
|
List<Program> list = new List<Program>();
|
|
if (Directory.Exists(baseDirectory))
|
|
{
|
|
GetAppFromDirectory(baseDirectory, list);
|
|
FileChangeWatcher.AddWatch(baseDirectory);
|
|
}
|
|
return list;
|
|
}
|
|
|
|
private void GetAppFromDirectory(string path, List<Program> list)
|
|
{
|
|
GetAppFromDirectory(path, list, 0);
|
|
}
|
|
|
|
private void GetAppFromDirectory(string path, List<Program> list, int depth)
|
|
{
|
|
if(maxDepth != -1 && depth > maxDepth)
|
|
{
|
|
return;
|
|
}
|
|
try
|
|
{
|
|
foreach (string file in Directory.GetFiles(path))
|
|
{
|
|
if (ProgramStorage.Instance.ProgramSuffixes.Split(';').Any(o => file.EndsWith("." + o)) ||
|
|
suffixes.Split(';').Any(o => file.EndsWith("." + o)))
|
|
{
|
|
Program p = CreateEntry(file);
|
|
list.Add(p);
|
|
}
|
|
}
|
|
|
|
foreach (var subDirectory in Directory.GetDirectories(path))
|
|
{
|
|
GetAppFromDirectory(subDirectory, list, depth + 1);
|
|
}
|
|
}
|
|
catch (Exception e)
|
|
{
|
|
var woxPluginException = new WoxPluginException("Program", $"GetAppFromDirectory failed: {path}", e);
|
|
Log.Error(woxPluginException);
|
|
}
|
|
}
|
|
|
|
public override string ToString()
|
|
{
|
|
return typeof(FileSystemProgramSource).Name + ":" + this.baseDirectory;
|
|
}
|
|
}
|
|
}
|