SqliteCreateDatabaseIfNotExists.cs

using System.Data.Entity;
using System.IO;
using SQLite.CodeFirst.Utility;

namespace SQLite.CodeFirst
{
    /// <summary>
    /// An implementation of <see cref="IDatabaseInitializer{TContext}"/> that will recreate and optionally re-seed the
    /// database only if the database does not exist. To seed the database, create a derived class and override the Seed method. 
    /// </summary>
    /// <typeparam name="TContext">The type of the context.</typeparam>
    public class SqliteCreateDatabaseIfNotExists<TContext> : SqliteInitializerBase<TContext>
        where TContext : DbContext
    {
        private readonly bool nullByteFileMeansNotExisting;

        /// <summary>
        /// Initializes a new instance of the <see cref="SqliteCreateDatabaseIfNotExists{TContext}"/> class.
        /// </summary>
        /// <param name="modelBuilder">The model builder.</param>
        public SqliteCreateDatabaseIfNotExists(DbModelBuilder modelBuilder)
            : this(modelBuilder, false, null)
        { }

        /// <summary>
        /// Initializes a new instance of the <see cref="SqliteCreateDatabaseIfNotExists{TContext}"/> class.
        /// </summary>
        /// <param name="modelBuilder">The model builder.</param>
        /// <param name="defaultCollation">The default collation applied to all string columns. Explicit <see cref="CollateAttribute"/>s take precedence.</param>
        public SqliteCreateDatabaseIfNotExists(DbModelBuilder modelBuilder, Collation defaultCollation)
            : this(modelBuilder, false, defaultCollation)
        { }

        /// <summary>
        /// Initializes a new instance of the <see cref="SqliteCreateDatabaseIfNotExists{TContext}"/> class.
        /// </summary>
        /// <param name="modelBuilder">The model builder.</param>
        /// <param name="nullByteFileMeansNotExisting">if set to <c>true</c> a null byte database file is treated like the database does not exist.</param>
        public SqliteCreateDatabaseIfNotExists(DbModelBuilder modelBuilder, bool nullByteFileMeansNotExisting)
            : this(modelBuilder, nullByteFileMeansNotExisting, null)
        { }

        /// <summary>
        /// Initializes a new instance of the <see cref="SqliteCreateDatabaseIfNotExists{TContext}"/> class.
        /// </summary>
        /// <param name="modelBuilder">The model builder.</param>
        /// <param name="nullByteFileMeansNotExisting">if set to <c>true</c> a null byte database file is treated like the database does not exist.</param>
        /// <param name="defaultCollation">The default collation applied to all string columns. Explicit <see cref="CollateAttribute"/>s take precedence.</param>
        public SqliteCreateDatabaseIfNotExists(DbModelBuilder modelBuilder, bool nullByteFileMeansNotExisting, Collation defaultCollation)
            : base(modelBuilder, defaultCollation)
        {
            this.nullByteFileMeansNotExisting = nullByteFileMeansNotExisting;
        }

        /// <summary>
        /// Initialize the database for the given context.
        /// Generates the SQLite-DDL from the model and executs it against the database.
        /// After that the <see cref="Seed" /> method is executed.
        /// All actions are be executed in transactions.
        /// </summary>
        /// <param name="context">The context.</param>
        public override void InitializeDatabase(TContext context)
        {
            string databaseFilePath = GetDatabasePathFromContext(context);

            bool exists = InMemoryAwareFile.Exists(databaseFilePath, nullByteFileMeansNotExisting);
            if (exists)
            {
                return;
            }

            base.InitializeDatabase(context);
        }
    }
}