PlotDirector/PlotLine/Controllers/LocationsController.cs

using Microsoft.AspNetCore.Mvc;
using Microsoft.AspNetCore.Authorization;
using PlotLine.Services;
using PlotLine.ViewModels;

namespace PlotLine.Controllers;

[Authorize]
public sealed class LocationsController(ILocationService locations) : Controller
{
    public async Task<IActionResult> Index(int projectId)
    {
        var model = await locations.GetLocationsAsync(projectId);
        return model is null ? NotFound() : View(model);
    }

    public async Task<IActionResult> Create(int projectId)
    {
        var model = await locations.GetCreateAsync(projectId);
        return model is null ? NotFound() : View("Edit", model);
    }

    public async Task<IActionResult> Edit(int id)
    {
        var model = await locations.GetEditAsync(id);
        return model is null ? NotFound() : View(model);
    }

    public async Task<IActionResult> Details(int id)
    {
        var model = await locations.GetDetailAsync(id);
        return model is null ? NotFound() : View(model);
    }

    [HttpPost]
    [ValidateAntiForgeryToken]
    public async Task<IActionResult> Save(LocationEditViewModel model)
    {
        if (!ModelState.IsValid)
        {
            var editModel = model.LocationID == 0
                ? await locations.GetCreateAsync(model.ProjectID)
                : await locations.GetEditAsync(model.LocationID);
            if (editModel is null)
            {
                return NotFound();
            }

            editModel.ParentLocationID = model.ParentLocationID;
            editModel.LocationName = model.LocationName;
            editModel.Aliases = model.Aliases;
            editModel.LocationTypeID = model.LocationTypeID;
            editModel.Description = model.Description;
            editModel.ShowInQuickAddBar = model.ShowInQuickAddBar;
            editModel.ExcludeFromCompanionDetection = model.ExcludeFromCompanionDetection;
            editModel.DetectionPriority = model.DetectionPriority;
            return View("Edit", editModel);
        }

        var locationId = await locations.SaveAsync(model);
        return RedirectToAction(nameof(Details), new { id = locationId });
    }

    [HttpPost]
    [ValidateAntiForgeryToken]
    public async Task<IActionResult> Archive(int id, int projectId)
    {
        await locations.ArchiveAsync(id);
        TempData["ArchiveMessage"] = "Location archived. You can restore it from Archived Items.";
        return RedirectToAction(nameof(Index), new { projectId });
    }

    [HttpPost]
    [ValidateAntiForgeryToken]
    public async Task<IActionResult> AddRelationship(LocationRelationshipEditViewModel model)
    {
        await locations.SaveRelationshipAsync(model);
        return RedirectToAction(nameof(Details), new { id = model.FromLocationID });
    }

    [HttpPost]
    [ValidateAntiForgeryToken]
    public async Task<IActionResult> DropRelationship(LocationRelationshipEditViewModel model)
    {
        if (model.FromLocationID != model.ToLocationID)
        {
            await locations.SaveRelationshipAsync(model);
        }

        return RedirectToAction(nameof(Index), new { projectId = model.ProjectID });
    }

    [HttpPost]
    [ValidateAntiForgeryToken]
    public async Task<IActionResult> ArchiveRelationship(int id, int returnLocationId)
    {
        await locations.ArchiveRelationshipAsync(id);
        TempData["ArchiveMessage"] = "Location relationship archived.";
        return RedirectToAction(nameof(Details), new { id = returnLocationId });
    }
}