Aaru/DiscImageChef/Commands/Database/Update.cs

// /***************************************************************************
// The Disc Image Chef
// ----------------------------------------------------------------------------
//
// Filename       : ListDevices.cs
// Author(s)      : Natalia Portillo <claunia@claunia.com>
//
// Component      : Verbs.
//
// --[ Description ] ----------------------------------------------------------
//
//     Implements the 'media-info' verb.
//
// --[ License ] --------------------------------------------------------------
//
//     This program is free software: you can redistribute it and/or modify
//     it under the terms of the GNU General Public License as
//     published by the Free Software Foundation, either version 3 of the
//     License, or (at your option) any later version.
//
//     This program is distributed in the hope that it will be useful,
//     but WITHOUT ANY WARRANTY; without even the implied warranty of
//     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
//     GNU General Public License for more details.
//
//     You should have received a copy of the GNU General Public License
//     along with this program.  If not, see <http://www.gnu.org/licenses/>.
//
// ----------------------------------------------------------------------------
// Copyright © 2011-2020 Natalia Portillo
// ****************************************************************************/

using System;
using System.CommandLine;
using System.CommandLine.Invocation;
using System.Diagnostics;
using System.IO;
using DiscImageChef.CommonTypes.Enums;
using DiscImageChef.Console;
using DiscImageChef.Core;
using DiscImageChef.Database;
using Microsoft.EntityFrameworkCore;

namespace DiscImageChef.Commands
{
    internal class UpdateCommand : Command
    {
        readonly bool _masterDbUpdate;

        public UpdateCommand(bool masterDbUpdate) : base("update", "Updates the database.")
        {
            _masterDbUpdate = masterDbUpdate;

            Add(new Option("--clear", "Clear existing master database.")
            {
                Argument = new Argument<bool>(() => false), Required = false
            });

            Add(new Option("--clear-all", "Clear existing master and local database.")
            {
                Argument = new Argument<bool>(() => false), Required = false
            });

            Handler = CommandHandler.Create<bool, bool, bool, bool>(Invoke);
        }

        public int Invoke(bool debug, bool verbose, bool clear, bool clearAll)
        {
            if(_masterDbUpdate)
                return(int)ErrorNumber.NoError;

            MainClass.PrintCopyright();

            if(debug)
                DicConsole.DebugWriteLineEvent += System.Console.Error.WriteLine;

            if(verbose)
                DicConsole.VerboseWriteLineEvent += System.Console.WriteLine;

            DicConsole.DebugWriteLine("Update command", "--debug={0}", debug);
            DicConsole.DebugWriteLine("Update command", "--verbose={0}", verbose);

            if(clearAll)
            {
                try
                {
                    File.Delete(Settings.Settings.LocalDbPath);

                    var ctx = DicContext.Create(Settings.Settings.LocalDbPath);
                    ctx.Database.Migrate();
                    ctx.SaveChanges();
                }
                catch(Exception e)
                {
                    if(Debugger.IsAttached)
                        throw;

                    DicConsole.ErrorWriteLine("Could not remove local database.");

                    return(int)ErrorNumber.CannotRemoveDatabase;
                }
            }

            if(clear || clearAll)
            {
                try
                {
                    File.Delete(Settings.Settings.MasterDbPath);
                }
                catch(Exception e)
                {
                    if(Debugger.IsAttached)
                        throw;

                    DicConsole.ErrorWriteLine("Could not remove master database.");

                    return(int)ErrorNumber.CannotRemoveDatabase;
                }
            }

            DoUpdate(clear || clearAll);

            return(int)ErrorNumber.NoError;
        }

        internal static void DoUpdate(bool create)
        {
            Remote.UpdateMasterDatabase(create);
            Statistics.AddCommand("update");
        }
    }
}