mirror of
https://github.com/zoriya/Kyoo.git
synced 2026-02-25 20:50:05 -05:00
96 lines
3.0 KiB
C#
96 lines
3.0 KiB
C#
// Kyoo - A portable and vast media library solution.
|
|
// Copyright (c) Kyoo.
|
|
//
|
|
// See AUTHORS.md and LICENSE file in the project root for full license information.
|
|
//
|
|
// Kyoo 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
|
|
// any later version.
|
|
//
|
|
// Kyoo 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 Kyoo. If not, see <https://www.gnu.org/licenses/>.
|
|
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using Kyoo.Abstractions.Controllers;
|
|
using Kyoo.Database;
|
|
using Microsoft.AspNetCore.Hosting;
|
|
using Microsoft.EntityFrameworkCore;
|
|
using Microsoft.Extensions.Configuration;
|
|
using Microsoft.Extensions.DependencyInjection;
|
|
using Microsoft.Extensions.Hosting;
|
|
using Npgsql;
|
|
|
|
namespace Kyoo.Postgresql
|
|
{
|
|
/// <summary>
|
|
/// A module to add postgresql capacity to the app.
|
|
/// </summary>
|
|
public class PostgresModule : IPlugin
|
|
{
|
|
/// <inheritdoc />
|
|
public string Slug => "postgresql";
|
|
|
|
/// <inheritdoc />
|
|
public string Name => "Postgresql";
|
|
|
|
/// <inheritdoc />
|
|
public string Description => "A database context for postgresql.";
|
|
|
|
/// <inheritdoc />
|
|
public Dictionary<string, Type> Configuration => new();
|
|
|
|
/// <inheritdoc />
|
|
public bool Enabled => _configuration.GetSelectedDatabase() == "postgres";
|
|
|
|
/// <summary>
|
|
/// The configuration to use. The database connection string is pulled from it.
|
|
/// </summary>
|
|
private readonly IConfiguration _configuration;
|
|
|
|
/// <summary>
|
|
/// The host environment to check if the app is in debug mode.
|
|
/// </summary>
|
|
private readonly IWebHostEnvironment _environment;
|
|
|
|
/// <summary>
|
|
/// Create a new postgres module instance and use the given configuration and environment.
|
|
/// </summary>
|
|
/// <param name="configuration">The configuration to use</param>
|
|
/// <param name="env">The environment that will be used (if the env is in development mode, more information will be displayed on errors.</param>
|
|
public PostgresModule(IConfiguration configuration, IWebHostEnvironment env)
|
|
{
|
|
_configuration = configuration;
|
|
_environment = env;
|
|
}
|
|
|
|
/// <inheritdoc />
|
|
public void Configure(IServiceCollection services)
|
|
{
|
|
services.AddDbContext<DatabaseContext, PostgresContext>(x =>
|
|
{
|
|
x.UseNpgsql(_configuration.GetDatabaseConnection("postgres"));
|
|
if (_environment.IsDevelopment())
|
|
x.EnableDetailedErrors().EnableSensitiveDataLogging();
|
|
}, ServiceLifetime.Transient);
|
|
}
|
|
|
|
/// <inheritdoc />
|
|
public void Initialize(IServiceProvider provider)
|
|
{
|
|
DatabaseContext context = provider.GetRequiredService<DatabaseContext>();
|
|
context.Database.Migrate();
|
|
|
|
using NpgsqlConnection conn = (NpgsqlConnection)context.Database.GetDbConnection();
|
|
conn.Open();
|
|
conn.ReloadTypes();
|
|
}
|
|
}
|
|
}
|