From 4ac2606c65fb9a95c12ac2265a67b29e834fe92b Mon Sep 17 00:00:00 2001
From: Cyberboss <Cyberboss@users.noreply.github.com>
Date: Mon, 16 Jul 2018 16:14:18 -0400
Subject: [PATCH] ChatController Create

---
 .../Rights/ChatSettingsRights.cs              | 10 +-
 .../Controllers/ChatController.cs             | 97 +++++++++++++++++++
 .../Controllers/DreamDaemonController.cs      |  4 +-
 .../Models/DatabaseContext.cs                 |  4 +-
 .../Models/IDatabaseContext.cs                |  5 +
 5 files changed, 114 insertions(+), 6 deletions(-)
 create mode 100644 src/Tgstation.Server.Host/Controllers/ChatController.cs

diff --git a/src/Tgstation.Server.Api/Rights/ChatSettingsRights.cs b/src/Tgstation.Server.Api/Rights/ChatSettingsRights.cs
index e8803311ac..c4336d76dd 100644
--- a/src/Tgstation.Server.Api/Rights/ChatSettingsRights.cs
+++ b/src/Tgstation.Server.Api/Rights/ChatSettingsRights.cs
@@ -39,6 +39,14 @@ namespace Tgstation.Server.Api.Rights
 		/// <summary>
 		/// User can change <see cref="Models.Internal.ChatSettings.Name"/>
 		/// </summary>
-		WriteName = 32
+		WriteName = 32,
+		/// <summary>
+		/// User can create new <see cref="Models.ChatSettings"/>
+		/// </summary>
+		Create = 64,
+		/// <summary>
+		/// User can delete <see cref="Models.ChatSettings"/>
+		/// </summary>
+		Delete = 128
 	}
 }
diff --git a/src/Tgstation.Server.Host/Controllers/ChatController.cs b/src/Tgstation.Server.Host/Controllers/ChatController.cs
new file mode 100644
index 0000000000..22f1d75cb4
--- /dev/null
+++ b/src/Tgstation.Server.Host/Controllers/ChatController.cs
@@ -0,0 +1,97 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Threading;
+using System.Threading.Tasks;
+using Microsoft.AspNetCore.Mvc;
+using Tgstation.Server.Api.Models;
+using Tgstation.Server.Api.Rights;
+using Tgstation.Server.Host.Components;
+using Tgstation.Server.Host.Models;
+using Tgstation.Server.Host.Security;
+
+namespace Tgstation.Server.Host.Controllers
+{
+	/// <summary>
+	/// <see cref="ModelController{TModel}"/> for managing <see cref="Api.Models.ChatSettings"/>
+	/// </summary>
+	[TgsAuthorize]
+	public sealed class ChatController : ModelController<Api.Models.ChatSettings>
+	{
+		/// <summary>
+		/// The <see cref="IInstanceManager"/> for the <see cref="ChatController"/>
+		/// </summary>
+		readonly IInstanceManager instanceManager;
+
+		/// <summary>
+		/// Construct a <see cref="ChatController"/>
+		/// </summary>
+		/// <param name="databaseContext">The <see cref="IDatabaseContext"/> for the <see cref="ApiController"/></param>
+		/// <param name="authenticationContextFactory">The <see cref="IAuthenticationContextFactory"/> for the <see cref="ApiController"/></param>
+		/// <param name="instanceManager">The value of <see cref="instanceManager"/></param>
+		public ChatController(IDatabaseContext databaseContext, IAuthenticationContextFactory authenticationContextFactory, IInstanceManager instanceManager) : base(databaseContext, authenticationContextFactory)
+		{
+			this.instanceManager = instanceManager ?? throw new ArgumentNullException(nameof(instanceManager));
+		}
+
+		/// <inheritdoc />
+		[TgsAuthorize(ChatSettingsRights.Create)]
+		public override async Task<IActionResult> Create([FromBody] Api.Models.ChatSettings model, CancellationToken cancellationToken)
+		{
+			if (model == null)
+				throw new ArgumentNullException(nameof(model));
+
+			if (String.IsNullOrWhiteSpace(model.Name))
+				return BadRequest(new { message = "Name cannot be null or whitespace!" });
+
+			if (String.IsNullOrWhiteSpace(model.ConnectionString))
+				return BadRequest(new { message = "ConnectionString cannot be null or whitespace!" });
+
+			//try to update das db first
+			var dbModel = new Models.ChatSettings
+			{
+				Name = model.Name,
+				ConnectionString = model.ConnectionString,
+				Enabled = model.Enabled,
+				Channels = model.Channels?.Select(x => new Models.ChatChannel
+				{
+					DiscordChannelId = x.DiscordChannelId,
+					IrcChannel = x.IrcChannel,
+					IsAdminChannel = x.IsAdminChannel,
+					IsWatchdogChannel = x.IsWatchdogChannel
+				}).ToList() ?? new List<Models.ChatChannel>(),
+				InstanceId = Instance.Id,
+				Provider = model.Provider,
+			};
+			DatabaseContext.ChatSettings.Add(dbModel);
+			DatabaseContext.ChatChannels.AddRange(dbModel.Channels);
+			await DatabaseContext.Save(cancellationToken).ConfigureAwait(false);
+
+			try
+			{
+				try
+				{
+					//try to create it
+					var instance = instanceManager.GetInstance(Instance);
+					await instance.Chat.ChangeSettings(dbModel, cancellationToken).ConfigureAwait(false);
+
+					if (dbModel.Channels.Count > 0)
+						await instance.Chat.ChangeChannels(dbModel.Id, dbModel.Channels, cancellationToken).ConfigureAwait(false);
+				}
+				catch
+				{
+					//undo the add
+					DatabaseContext.ChatSettings.Remove(dbModel);
+					DatabaseContext.ChatChannels.RemoveRange(dbModel.Channels);
+					await DatabaseContext.Save(default).ConfigureAwait(false);
+					throw;
+				}
+			}
+			catch (InvalidOperationException e)
+			{
+				return BadRequest(new { message = e.Message });
+			}
+			return Json(dbModel);
+		}
+	}
+}
diff --git a/src/Tgstation.Server.Host/Controllers/DreamDaemonController.cs b/src/Tgstation.Server.Host/Controllers/DreamDaemonController.cs
index 168d3a46da..c1a4a6b673 100644
--- a/src/Tgstation.Server.Host/Controllers/DreamDaemonController.cs
+++ b/src/Tgstation.Server.Host/Controllers/DreamDaemonController.cs
@@ -18,10 +18,10 @@ using Tgstation.Server.Host.Security;
 namespace Tgstation.Server.Host.Controllers
 {
 	/// <summary>
-	/// <see cref="ModelController{TModel}"/> for managing <see cref="Api.Models.DreamDaemon"/>
+	/// <see cref="ModelController{TModel}"/> for managing the <see cref="DreamDaemon"/>
 	/// </summary>
 	[Route("/" + nameof(DreamDaemon))]
-	public sealed class DreamDaemonController : ModelController<Api.Models.DreamDaemon>
+	public sealed class DreamDaemonController : ModelController<DreamDaemon>
 	{
 		/// <summary>
 		/// The <see cref="IJobManager"/> for the <see cref="DreamMakerController"/>
diff --git a/src/Tgstation.Server.Host/Models/DatabaseContext.cs b/src/Tgstation.Server.Host/Models/DatabaseContext.cs
index 554118c0f4..ff966eba6f 100644
--- a/src/Tgstation.Server.Host/Models/DatabaseContext.cs
+++ b/src/Tgstation.Server.Host/Models/DatabaseContext.cs
@@ -52,9 +52,7 @@ namespace Tgstation.Server.Host.Models
 		/// </summary>
 		public DbSet<InstanceUser> InstanceUsers { get; set; }
 
-		/// <summary>
-		/// The <see cref="ChatChannel"/>s in the <see cref="DatabaseContext{TParentContext}"/>
-		/// </summary>
+		/// <inheritdoc />
 		public DbSet<ChatChannel> ChatChannels { get; set; }
 
 		/// <inheritdoc />
diff --git a/src/Tgstation.Server.Host/Models/IDatabaseContext.cs b/src/Tgstation.Server.Host/Models/IDatabaseContext.cs
index e08826dcc9..9e8d906905 100644
--- a/src/Tgstation.Server.Host/Models/IDatabaseContext.cs
+++ b/src/Tgstation.Server.Host/Models/IDatabaseContext.cs
@@ -49,6 +49,11 @@ namespace Tgstation.Server.Host.Models
 		/// </summary>
 		DbSet<ChatSettings> ChatSettings { get; set; }
 
+		/// <summary>
+		/// The <see cref="ChatChannel"/> in the <see cref="IDatabaseContext"/>
+		/// </summary>
+		DbSet<ChatChannel> ChatChannels { get; set; }
+
 		/// <summary>
 		/// The <see cref="Models.RepositorySettings"/> in the <see cref="IDatabaseContext"/>
 		/// </summary>