using System.Collections.Concurrent;
using System.Security;
namespace ErikEJ.DacFX.TSQLAnalyzer.Services
{
internal sealed class SqlFileCollector
{
private readonly ConcurrentDictionary<string, string> files = new();
private static EnumerationOptions Recursive => new()
{
RecurseSubdirectories = true,
MatchType = MatchType.Simple,
AttributesToSkip = FileAttributes.Hidden | FileAttributes.System | FileAttributes.ReparsePoint,
IgnoreInaccessible = true,
};
private static EnumerationOptions TopOnly => new()
{
RecurseSubdirectories = false,
MatchType = MatchType.Simple,
AttributesToSkip = FileAttributes.Hidden | FileAttributes.System | FileAttributes.ReparsePoint | FileAttributes.Offline,
IgnoreInaccessible = true,
};
public Dictionary<string, string> ProcessList(IList<string> filePaths)
{
Parallel.ForEach(filePaths, (filePath) =>
{
if (!File.Exists(filePath))
{
if (Directory.Exists(filePath))
{
ProcessDirectory(filePath);
}
else
{
ProcessWildCard(filePath);
}
}
else
{
ProcessIfSqlFile(filePath);
}
});
return files.ToDictionary(kvp => kvp.Key, kvp => kvp.Value);
}
private void ProcessFile(string filePath)
{
var fileStream = GetFileContents(filePath);
if (fileStream == null)
{
return;
}
AddToProcessing(filePath, fileStream);
}
private void AddToProcessing(string filePath, string fileContents)
{
files.TryAdd(filePath, fileContents);
}
private void ProcessDirectory(string path)
{
var files = Directory.EnumerateFiles(path, "*.sql", Recursive);
Parallel.ForEach(files, (file) =>
{
ProcessIfSqlFile(file);
});
}
private void ProcessIfSqlFile(string fileName)
{
if (Path.GetExtension(fileName).Equals(".sql", StringComparison.OrdinalIgnoreCase))
{
ProcessFile(fileName);
}
if (Path.GetExtension(fileName).Equals(".dacpac", StringComparison.OrdinalIgnoreCase))
{
AddToProcessing(fileName, string.Empty);
}
if (Path.GetExtension(fileName).Equals(".zip", StringComparison.OrdinalIgnoreCase))
{
AddToProcessing(fileName, string.Empty);
}
}
private void ProcessWildCard(string filePath)
{
var containsWildCard = filePath.Contains('*', StringComparison.OrdinalIgnoreCase)
|| filePath.Contains('?', StringComparison.OrdinalIgnoreCase);
if (!containsWildCard)
{
throw new ArgumentException($"{filePath} is not a valid file path.");
}
var dirPath = Path.GetDirectoryName(filePath);
if (string.IsNullOrEmpty(dirPath))
{
dirPath = Directory.GetCurrentDirectory();
}
if (!Directory.Exists(dirPath))
{
throw new ArgumentException($"Directory does not exist: {dirPath}");
}
var searchPattern = Path.GetFileName(filePath);
var files = Directory.EnumerateFiles(dirPath, searchPattern, TopOnly);
Parallel.ForEach(files, (file) =>
{
ProcessIfSqlFile(file);
});
}
private string? GetFileContents(string filePath)
{
try
{
return File.ReadAllText(filePath);
}
catch (UnauthorizedAccessException)
{
// Ignore
}
catch (SecurityException)
{
// Ignore
}
catch (Exception)
{
throw;
}
return null;
}
}
}
