namespace Zipper
{
    public class ZipCreator
    {
        private static readonly Dictionary<String, KeyValuePair<MemoryStream, ZipArchive>> Archives =
            new Dictionary<string, KeyValuePair<MemoryStream, ZipArchive>>();

        public void RequestKey(Action<object> callback)
        {
            string key;
            while ((key = Guid.NewGuid().ToString()).Length > 0 && Archives.ContainsKey(key)) { }

            var stream = new MemoryStream();
            Archives.Add(key, new KeyValuePair<MemoryStream, ZipArchive>(stream, new ZipArchive(stream, ZipArchiveMode.Update)));
            callback(Utilities.JsonEncode(new Result(ResultType.STRING, key)));
        }

        public void AddBuffer(string key, string name, byte[] buffer, Action<object> callback)
        {
            Archives[key].Value.CreateEntry(name);
            Archives[key].Value.GetEntry(name)?.Open().Write(buffer, 0, buffer.Length);
            callback(Utilities.JsonEncode(new Result(ResultType.SUCCESS, null)));
        }

        public void AddFolder(string appName, string appId, string key, string folder, string folderName, Action<object> callback)
        {
            callback(Utilities.RunSafe(appName, appId, folder, () =>
            {
                AddFolderInternal(key, folder, folderName);
                return new Result(ResultType.SUCCESS, null);
            }));
        }

        private void AddFolderInternal(string key, string folder, string folderName)
        {
            var safeFolder = Utilities.SanitisePath(folder);
            foreach (var file in Directory.GetFiles(safeFolder))
            {
                var zip = Archives[key].Value;
                zip.CreateEntryFromFile(file, Path.Combine(folderName, new FileInfo(file).Name));
            }

            foreach (var directory in Directory.GetDirectories(safeFolder))
            {
                AddFolderInternal(key, directory, Path.Combine(folderName, new DirectoryInfo(directory).Name));
            }
        }

        public void Create(string appName, string appId, string key, string output, Action<object> callback)
        {
            Archives[key].Value.Dispose();
            var buffer = Archives[key].Key.GetBuffer();
            new FileOperations().WriteFile(appName, appId, output, buffer, callback);
        }
    }
}