Files
TBDel/Commands/AddCommand.cs

82 lines
3.0 KiB
C#

using TBDel.Models;
using TBDel.Services;
namespace TBDel.Commands
{
public class AddCommand
{
public static async Task AddEntry(string[] args)
{
// TODO: Add duplicate path check
// TODO: Add support for multiple paths
if (args.Length > 1)
{
string workingDirectory = Directory.GetCurrentDirectory();
string absolutePath = Path.Combine(workingDirectory, args[1]);
var dbService = new DbService();
if (File.Exists(absolutePath))
{
Console.WriteLine($"Adding: {absolutePath}");
var entry = new FileEntry { Id = GenerateUniqueId(dbService), Path = absolutePath, DateAdded = DateTime.Now };
if (await dbService.AddFileEntryAsync(entry))
{
Console.ForegroundColor = ConsoleColor.Green;
Console.WriteLine("File added successfully.");
Console.ResetColor();
return;
}
else
{
Console.ForegroundColor = ConsoleColor.Red;
Console.WriteLine("Failed to add file.");
Console.ResetColor();
return;
}
}
else if (Directory.Exists(absolutePath))
{
Console.WriteLine($"Adding: {absolutePath}");
var entry = new FolderEntry() { Id = GenerateUniqueId(dbService), Path = absolutePath, DateAdded = DateTime.Now };
if (await dbService.AddFolderEntryAsync(entry))
{
Console.ForegroundColor = ConsoleColor.Green;
Console.WriteLine("Directory added successfully.");
Console.ResetColor();
return;
}
else
{
Console.ForegroundColor = ConsoleColor.Red;
Console.WriteLine("Failed to add directory.");
Console.ResetColor();
return;
}
}
}
}
private static uint GenerateUniqueId(DbService dbService)
{
// Hopefully not running away from me one day
Random random = new Random();
uint newId;
do
{
newId = (uint)random.Next(10000, 99999);
} while (IdExists(newId, dbService));
return newId;
}
private static bool IdExists(uint id, DbService dbService)
{
// Not async? Not a problem (yet(?))
var fileEntries = dbService.GetFileEntriesAsync().Result;
var folderEntries = dbService.GetFolderEntriesAsync().Result;
return fileEntries.Any(e => e.Id == id) || folderEntries.Any(e => e.Id == id);
}
}
}