LANCommander.PlaynitePlugin/LANCommander/Controllers/GamesController.cs

using System;
using System.IO;
using System.Collections.Generic;
using System.Linq;
using System.Threading.Tasks;
using Microsoft.AspNetCore.Mvc;
using Microsoft.AspNetCore.Mvc.Rendering;
using Microsoft.EntityFrameworkCore;
using LANCommander.Data;
using LANCommander.Data.Models;
using Microsoft.AspNetCore.Authorization;
using LANCommander.Services;
using System.Drawing;

namespace LANCommander.Controllers
{
    [Authorize(Roles = "Administrator")]
    public class GamesController : Controller
    {
        private readonly DatabaseContext Context;

        public GamesController(DatabaseContext context)
        {
            Context = context;
        }

        // GET: Games
        public async Task<IActionResult> Index()
        {
              return Context.Games != null ? 
                          View(await Context.Games.ToListAsync()) :
                          Problem("Entity set 'DatabaseContext.Games'  is null.");
        }

        // GET: Games/Create
        public IActionResult Add()
        {
            return View();
        }

        // POST: Games/Create
        // To protect from overposting attacks, enable the specific properties you want to bind to.
        // For more details, see http://go.microsoft.com/fwlink/?LinkId=317598.
        [HttpPost]
        [ValidateAntiForgeryToken]
        public async Task<IActionResult> Add([Bind("Title,SortTitle,Description,ReleasedOn,Id,CreatedOn,CreatedById,UpdatedOn,UpdatedById")] Game game)
        {
            if (ModelState.IsValid)
            {
                using (Repository<Game> repo = new Repository<Game>(Context, HttpContext))
                {
                    await repo.Add(game);
                    await repo.SaveChanges();
                }

                return RedirectToAction(nameof(Index));
            }
            return View(game);
        }

        // GET: Games/Edit/5
        public async Task<IActionResult> Edit(Guid? id)
        {
            Game game;

            if (id == null || Context.Games == null)
            {
                return NotFound();
            }

            using (Repository<Game> repo = new Repository<Game>(Context, HttpContext))
            {
                game = await repo.Find(id.GetValueOrDefault());

                if (game == null)
                    return NotFound();
            }

            return View(game);
        }

        // POST: Games/Edit/5
        // To protect from overposting attacks, enable the specific properties you want to bind to.
        // For more details, see http://go.microsoft.com/fwlink/?LinkId=317598.
        [HttpPost]
        [ValidateAntiForgeryToken]
        public async Task<IActionResult> Edit(Guid id, [Bind("Title,SortTitle,Description,ReleasedOn,Id,CreatedOn,CreatedById,UpdatedOn,UpdatedById")] Game game)
        {
            if (id != game.Id)
            {
                return NotFound();
            }

            if (ModelState.IsValid)
            {
                try
                {
                    using (Repository<Game> repo = new Repository<Game>(Context, HttpContext))
                    {
                        repo.Update(game);

                        await repo.SaveChanges();
                    }
                }
                catch (DbUpdateConcurrencyException)
                {
                    if (!GameExists(game.Id))
                    {
                        return NotFound();
                    }
                    else
                    {
                        throw;
                    }
                }
                return RedirectToAction(nameof(Index));
            }
            return View(game);
        }

        // GET: Games/Delete/5
        public async Task<IActionResult> Delete(Guid? id)
        {
            if (id == null || Context.Games == null)
            {
                return NotFound();
            }

            var game = await Context.Games
                .FirstOrDefaultAsync(m => m.Id == id);
            if (game == null)
            {
                return NotFound();
            }

            return View(game);
        }

        // POST: Games/Delete/5
        [HttpPost, ActionName("Delete")]
        [ValidateAntiForgeryToken]
        public async Task<IActionResult> DeleteConfirmed(Guid id)
        {
            using (Repository<Game> repo = new Repository<Game>(Context, HttpContext))
            {
                var game = await repo.Find(id);

                if (game == null)
                    return NotFound();

                if (game.Archives != null && game.Archives.Count > 0)
                {
                    using (var archiveRepo = new Repository<Archive>(Context, HttpContext))
                    {
                        foreach (var archive in game.Archives.OrderByDescending(a => a.CreatedOn))
                        {
                            var archiveFile = Path.Combine("Upload", archive.ObjectKey);
                            var iconFile = Path.Combine("Icon", $"{game.Id}.png");

                            if (System.IO.File.Exists(archiveFile))
                                System.IO.File.Delete(archiveFile);

                            if (System.IO.File.Exists(iconFile))
                                System.IO.File.Delete(iconFile);

                            archiveRepo.Delete(archive);
                        }

                        await archiveRepo.SaveChanges();
                    }
                }

                repo.Delete(game);

                await repo.SaveChanges();

                return RedirectToAction(nameof(Index));
            }
        }

        public async Task<IActionResult> GetIcon(Guid id)
        {
            var cachedPath = Path.Combine("Icon", $"{id}.png");

            if (!System.IO.Directory.Exists("Icon"))
                System.IO.Directory.CreateDirectory("Icon");

            if (System.IO.File.Exists(cachedPath))
            {
                return File(System.IO.File.ReadAllBytes(cachedPath), "image/png");
            }
            else
            {
                using (var repo = new Repository<Game>(Context, HttpContext))
                {
                    var game = await repo.Find(id);

                    if (game.Archives == null || game.Archives.Count == 0)
                        return NotFound();

                    var archive = game.Archives.OrderByDescending(a => a.CreatedOn).FirstOrDefault();

                    Bitmap bitmap = null;

                    var manifest = ArchiveService.ReadManifest(archive.ObjectKey);
                    var iconReference = ArchiveService.ReadFile(archive.ObjectKey, manifest.Icon);

                    if (IsWinPEFile(iconReference))
                    {
                        var tmp = System.IO.Path.GetTempFileName();

                        System.IO.File.WriteAllBytes(tmp, iconReference);

                        var icon = System.Drawing.Icon.ExtractAssociatedIcon(tmp);

                        bitmap = icon.ToBitmap();
                    }
                    else
                    {
                        using (var ms = new MemoryStream(iconReference))
                        {
                            bitmap = (Bitmap)Bitmap.FromStream(ms);
                        }
                    }

                    var iconPng = ConvertToPng(bitmap);

                    System.IO.File.WriteAllBytes(cachedPath, iconPng);

                    return File(iconPng, "image/png");
                }
            }
        }

        private static bool IsWinPEFile(byte[] file)
        {
            var mz = new byte[2];

            using (var ms = new MemoryStream(file))
            {
                ms.Read(mz, 0, 2);
            }

            return System.Text.Encoding.UTF8.GetString(mz) == "MZ";
        }

        private static byte[] ConvertToPng(Image img)
        {
            using (var stream = new MemoryStream())
            {
                img.Save(stream, System.Drawing.Imaging.ImageFormat.Png);

                return stream.ToArray();
            }
        }

        private bool GameExists(Guid id)
        {
          return (Context.Games?.Any(e => e.Id == id)).GetValueOrDefault();
        }
    }
}