hrntsm/Tunny

using System;
using System.IO;

using Optuna.Storage;
using Optuna.Study;

using Tunny.Core.Util;

namespace Tunny.Core.Storage
{
    public class StorageHandler : ITStorage
    {
        public dynamic Storage { get; set; }

        public dynamic CreateNewTStorage(bool useInnerPythonEngine, Settings.Storage storageSetting)
        {
            TLog.MethodStart();
            ICreateTStorage storage;

            switch (Path.GetExtension(storageSetting.Path))
            {
                case null:
                    storage = new InMemoryStorage();
                    break;
                case ".sqlite3":
                case ".db":
                    storage = new SqliteStorage();
                    break;
                case ".log":
                    storage = new JournalStorage();
                    break;
                default:
                    throw new ArgumentException("Storage type not supported");
            }
            Storage = storage.CreateNewTStorage(useInnerPythonEngine, storageSetting);
            return Storage;
        }

        public void DuplicateStudyInStorage(string fromStudyName, string toStudyName, Settings.Storage storageSetting)
        {
            TLog.MethodStart();
            ITStorage storage;

            switch (Path.GetExtension(storageSetting.Path))
            {
                case ".sqlite3":
                case ".db":
                    storage = new SqliteStorage();
                    break;
                case ".log":
                    storage = new JournalStorage();
                    break;
                default:
                    throw new ArgumentException("Storage type not supported");
            }
            storage.DuplicateStudyInStorage(fromStudyName, toStudyName, storageSetting);
        }

        public StudySummary[] GetStudySummaries(string storagePath)
        {
            TLog.MethodStart();
            ITStorage storage;

            switch (Path.GetExtension(storagePath))
            {
                case ".sqlite3":
                case ".db":
                    storage = new SqliteStorage();
                    break;
                case ".log":
                    storage = new JournalStorage();
                    break;
                default:
                    throw new ArgumentException("Storage type not supported");
            }

            return storage.GetStudySummaries(storagePath);
        }

        public static IOptunaStorage GetStorage(string path)
        {
            string ext = Path.GetExtension(path);
            IOptunaStorage storage;
            if (ext == ".db" || ext == ".sqlite")
            {
                storage = new Optuna.Storage.RDB.SqliteStorage(path, true);
            }
            else if (ext == ".log")
            {
                storage = new Optuna.Storage.Journal.JournalStorage(path, true);
            }
            else
            {
                throw new ArgumentException("Storage type not supported");
            }

            return storage;
        }
    }
}