LANCommander/LANCommander.Server.Data.SQLite/Migrations/20241011014752_AddStorageLocations.cs

using System;
using LANCommander.SDK.Enums;
using LANCommander.Server.Services;
using Microsoft.EntityFrameworkCore.Migrations;

#nullable disable

namespace LANCommander.Migrations
{
    /// <inheritdoc />
    public partial class AddStorageLocations : Migration
    {
        /// <inheritdoc />
        protected override void Up(MigrationBuilder migrationBuilder)
        {
            var settings = SettingService.GetSettings();

            migrationBuilder.AddColumn<Guid>(
                name: "StorageLocationId",
                table: "Media",
                type: "TEXT",
                nullable: false,
                defaultValue: new Guid("00000000-0000-0000-0000-000000000000"));

            migrationBuilder.AddColumn<Guid>(
                name: "StorageLocationId",
                table: "GameSaves",
                type: "TEXT",
                nullable: false,
                defaultValue: new Guid("00000000-0000-0000-0000-000000000000"));

            migrationBuilder.AddColumn<Guid>(
                name: "StorageLocationId",
                table: "Archive",
                type: "TEXT",
                nullable: false,
                defaultValue: new Guid("00000000-0000-0000-0000-000000000000"));

            migrationBuilder.CreateTable(
                name: "StorageLocations",
                columns: table => new
                {
                    Id = table.Column<Guid>(type: "TEXT", nullable: false),
                    Default = table.Column<bool>(type: "INTEGER", nullable: false),
                    Path = table.Column<string>(type: "TEXT", nullable: false),
                    Type = table.Column<int>(type: "INTEGER", nullable: false),
                    CreatedOn = table.Column<DateTime>(type: "TEXT", nullable: false),
                    CreatedById = table.Column<Guid>(type: "TEXT", nullable: true),
                    UpdatedOn = table.Column<DateTime>(type: "TEXT", nullable: false),
                    UpdatedById = table.Column<Guid>(type: "TEXT", nullable: true)
                },
                constraints: table =>
                {
                    table.PrimaryKey("PK_StorageLocations", x => x.Id);
                    table.ForeignKey(
                        name: "FK_StorageLocations_AspNetUsers_CreatedById",
                        column: x => x.CreatedById,
                        principalTable: "AspNetUsers",
                        principalColumn: "Id");
                    table.ForeignKey(
                        name: "FK_StorageLocations_AspNetUsers_UpdatedById",
                        column: x => x.UpdatedById,
                        principalTable: "AspNetUsers",
                        principalColumn: "Id");
                });

            migrationBuilder.CreateIndex(
                name: "IX_Media_StorageLocationId",
                table: "Media",
                column: "StorageLocationId");

            migrationBuilder.CreateIndex(
                name: "IX_GameSaves_StorageLocationId",
                table: "GameSaves",
                column: "StorageLocationId");

            migrationBuilder.CreateIndex(
                name: "IX_Archive_StorageLocationId",
                table: "Archive",
                column: "StorageLocationId");

            migrationBuilder.CreateIndex(
                name: "IX_StorageLocations_CreatedById",
                table: "StorageLocations",
                column: "CreatedById");

            migrationBuilder.CreateIndex(
                name: "IX_StorageLocations_UpdatedById",
                table: "StorageLocations",
                column: "UpdatedById");

            migrationBuilder.AddForeignKey(
                name: "FK_Archive_StorageLocations_StorageLocationId",
                table: "Archive",
                column: "StorageLocationId",
                principalTable: "StorageLocations",
                principalColumn: "Id",
                onDelete: ReferentialAction.Cascade);

            migrationBuilder.AddForeignKey(
                name: "FK_GameSaves_StorageLocations_StorageLocationId",
                table: "GameSaves",
                column: "StorageLocationId",
                principalTable: "StorageLocations",
                principalColumn: "Id",
                onDelete: ReferentialAction.Cascade);

            migrationBuilder.AddForeignKey(
                name: "FK_Media_StorageLocations_StorageLocationId",
                table: "Media",
                column: "StorageLocationId",
                principalTable: "StorageLocations",
                principalColumn: "Id",
                onDelete: ReferentialAction.Cascade);

            migrationBuilder.Sql($"INSERT INTO StorageLocations (Id, Path, Type, \"Default\", CreatedOn, UpdatedOn) VALUES (upper('{Guid.NewGuid()}'), 'Uploads', {(int)StorageLocationType.Archive}, 1, date('now'), date('now'))");
            migrationBuilder.Sql($"INSERT INTO StorageLocations (Id, Path, Type, \"Default\", CreatedOn, UpdatedOn) VALUES (upper('{Guid.NewGuid()}'), 'Saves', {(int)StorageLocationType.Save}, 1, date('now'), date('now'))");
            migrationBuilder.Sql($"INSERT INTO StorageLocations (Id, Path, Type, \"Default\", CreatedOn, UpdatedOn) VALUES (upper('{Guid.NewGuid()}'), 'Media', {(int)StorageLocationType.Media}, 1, date('now'), date('now'))");

            migrationBuilder.Sql($@"UPDATE Archive
                SET StorageLocationId = (
                    SELECT Id
                    FROM StorageLocations
                    WHERE Type = {(int)StorageLocationType.Archive}
                    LIMIT 1
                );");

            migrationBuilder.Sql($@"UPDATE GameSaves
                SET StorageLocationId = (
                    SELECT Id
                    FROM StorageLocations
                    WHERE Type = {(int)StorageLocationType.Save}
                    LIMIT 1
                );");

            migrationBuilder.Sql($@"UPDATE Media
                SET StorageLocationId = (
                    SELECT Id
                    FROM StorageLocations
                    WHERE Type = {(int)StorageLocationType.Media}
                    LIMIT 1
                );");
        }

        /// <inheritdoc />
        protected override void Down(MigrationBuilder migrationBuilder)
        {
            migrationBuilder.DropForeignKey(
                name: "FK_Archive_StorageLocations_StorageLocationId",
                table: "Archive");

            migrationBuilder.DropForeignKey(
                name: "FK_GameSaves_StorageLocations_StorageLocationId",
                table: "GameSaves");

            migrationBuilder.DropForeignKey(
                name: "FK_Media_StorageLocations_StorageLocationId",
                table: "Media");

            migrationBuilder.DropTable(
                name: "StorageLocations");

            migrationBuilder.DropIndex(
                name: "IX_Media_StorageLocationId",
                table: "Media");

            migrationBuilder.DropIndex(
                name: "IX_GameSaves_StorageLocationId",
                table: "GameSaves");

            migrationBuilder.DropIndex(
                name: "IX_Archive_StorageLocationId",
                table: "Archive");

            migrationBuilder.DropColumn(
                name: "StorageLocationId",
                table: "Media");

            migrationBuilder.DropColumn(
                name: "StorageLocationId",
                table: "GameSaves");

            migrationBuilder.DropColumn(
                name: "StorageLocationId",
                table: "Archive");
        }
    }
}