using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using Tango.FileSystem.Network;

namespace Tango.FileSystem
{
    public class FileSystemManager
    {
        public FileSystemItemDTO GetRoot()
        {
            FileSystemItemDTO folder = new FileSystemItemDTO();
            folder.Path = "This PC";
            folder.IsRoot = true;
            folder.Type = FileSystemItemType.Folder;
            folder.Items = DriveInfo.GetDrives().Select(x => new FileSystemItemDTO()
            {
                Path = x.RootDirectory.FullName,
                DriveType = x.DriveType,
                DriveLabel = x.Name,
                Type = FileSystemItemType.Drive,
            }).Cast<FileSystemItemDTO>().ToList();

            return folder;
        }

        public FileSystemItemDTO GetFolder(String path)
        {
            List<FileSystemItemDTO> items = new List<FileSystemItemDTO>();

            if (String.IsNullOrWhiteSpace(path))
            {
                return GetRoot();
            }

            if (!Directory.Exists(path))
            {
                throw new DirectoryNotFoundException("The specified directory could not be located.");
            }

            foreach (var directory in Directory.GetDirectories(path))
            {
                items.Add(new FileSystemItemDTO()
                {
                    Path = directory,
                    Type = FileSystemItemType.Folder,
                    DateModified = Directory.GetLastWriteTimeUtc(directory),
                });
            }

            foreach (var file in Directory.GetFiles(path))
            {
                items.Add(new FileSystemItemDTO()
                {
                    Path = file,
                    Type = FileSystemItemType.File,
                    DateModified = File.GetLastWriteTimeUtc(file),
                    Size = new FileInfo(file).Length
                });
            }

            return new FileSystemItemDTO()
            {
                Path = path,
                Type = path.Length == 3 ? FileSystemItemType.Drive : FileSystemItemType.Folder,
                Items = items
            };
        }

        public void Delete(String path)
        {
            if (Directory.Exists(path))
            {
                Directory.Delete(path);
            }
            else if (File.Exists(path))
            {
                File.Delete(path);
            }
            else
            {
                throw new FileNotFoundException("Could not locate the specified file or directory.");
            }
        }
    }
}