1 files changed, 150 insertions, 0 deletions
diff --git a/Jellyfin.Api/Controllers/StartupController.cs b/Jellyfin.Api/Controllers/StartupController.cs
new file mode 100644
index 000000000..a01a617fc
--- /dev/null
+++ b/Jellyfin.Api/Controllers/StartupController.cs
@@ -0,0 +1,150 @@
+using System.ComponentModel.DataAnnotations;
+using System.Linq;
+using System.Threading.Tasks;
+using Jellyfin.Api.Constants;
+using Jellyfin.Api.Models.StartupDtos;
+using Jellyfin.Networking.Configuration;
+using MediaBrowser.Controller.Configuration;
+using MediaBrowser.Controller.Library;
+using Microsoft.AspNetCore.Authorization;
+using Microsoft.AspNetCore.Http;
+using Microsoft.AspNetCore.Mvc;
+
+namespace Jellyfin.Api.Controllers
+{
+    /// <summary>
+    /// The startup wizard controller.
+    /// </summary>
+    [Authorize(Policy = Policies.FirstTimeSetupOrElevated)]
+    public class StartupController : BaseJellyfinApiController
+    {
+        private readonly IServerConfigurationManager _config;
+        private readonly IUserManager _userManager;
+
+        /// <summary>
+        /// Initializes a new instance of the <see cref="StartupController" /> class.
+        /// </summary>
+        /// <param name="config">The server configuration manager.</param>
+        /// <param name="userManager">The user manager.</param>
+        public StartupController(IServerConfigurationManager config, IUserManager userManager)
+        {
+            _config = config;
+            _userManager = userManager;
+        }
+
+        /// <summary>
+        /// Completes the startup wizard.
+        /// </summary>
+        /// <response code="204">Startup wizard completed.</response>
+        /// <returns>A <see cref="NoContentResult"/> indicating success.</returns>
+        [HttpPost("Complete")]
+        [ProducesResponseType(StatusCodes.Status204NoContent)]
+        public ActionResult CompleteWizard()
+        {
+            _config.Configuration.IsStartupWizardCompleted = true;
+            _config.SaveConfiguration();
+            return NoContent();
+        }
+
+        /// <summary>
+        /// Gets the initial startup wizard configuration.
+        /// </summary>
+        /// <response code="200">Initial startup wizard configuration retrieved.</response>
+        /// <returns>An <see cref="OkResult"/> containing the initial startup wizard configuration.</returns>
+        [HttpGet("Configuration")]
+        [ProducesResponseType(StatusCodes.Status200OK)]
+        public ActionResult<StartupConfigurationDto> GetStartupConfiguration()
+        {
+            return new StartupConfigurationDto
+            {
+                UICulture = _config.Configuration.UICulture,
+                MetadataCountryCode = _config.Configuration.MetadataCountryCode,
+                PreferredMetadataLanguage = _config.Configuration.PreferredMetadataLanguage
+            };
+        }
+
+        /// <summary>
+        /// Sets the initial startup wizard configuration.
+        /// </summary>
+        /// <param name="startupConfiguration">The updated startup configuration.</param>
+        /// <response code="204">Configuration saved.</response>
+        /// <returns>A <see cref="NoContentResult"/> indicating success.</returns>
+        [HttpPost("Configuration")]
+        [ProducesResponseType(StatusCodes.Status204NoContent)]
+        public ActionResult UpdateInitialConfiguration([FromBody, Required] StartupConfigurationDto startupConfiguration)
+        {
+            _config.Configuration.UICulture = startupConfiguration.UICulture ?? string.Empty;
+            _config.Configuration.MetadataCountryCode = startupConfiguration.MetadataCountryCode ?? string.Empty;
+            _config.Configuration.PreferredMetadataLanguage = startupConfiguration.PreferredMetadataLanguage ?? string.Empty;
+            _config.SaveConfiguration();
+            return NoContent();
+        }
+
+        /// <summary>
+        /// Sets remote access and UPnP.
+        /// </summary>
+        /// <param name="startupRemoteAccessDto">The startup remote access dto.</param>
+        /// <response code="204">Configuration saved.</response>
+        /// <returns>A <see cref="NoContentResult"/> indicating success.</returns>
+        [HttpPost("RemoteAccess")]
+        [ProducesResponseType(StatusCodes.Status204NoContent)]
+        public ActionResult SetRemoteAccess([FromBody, Required] StartupRemoteAccessDto startupRemoteAccessDto)
+        {
+            NetworkConfiguration settings = _config.GetNetworkConfiguration();
+            settings.EnableRemoteAccess = startupRemoteAccessDto.EnableRemoteAccess;
+            settings.EnableUPnP = startupRemoteAccessDto.EnableAutomaticPortMapping;
+            _config.SaveConfiguration("network", settings);
+            return NoContent();
+        }
+
+        /// <summary>
+        /// Gets the first user.
+        /// </summary>
+        /// <response code="200">Initial user retrieved.</response>
+        /// <returns>The first user.</returns>
+        [HttpGet("User")]
+        [HttpGet("FirstUser", Name = "GetFirstUser_2")]
+        [ProducesResponseType(StatusCodes.Status200OK)]
+        public async Task<StartupUserDto> GetFirstUser()
+        {
+            // TODO: Remove this method when startup wizard no longer requires an existing user.
+            await _userManager.InitializeAsync().ConfigureAwait(false);
+            var user = _userManager.Users.First();
+            return new StartupUserDto
+            {
+                Name = user.Username,
+                Password = user.Password
+            };
+        }
+
+        /// <summary>
+        /// Sets the user name and password.
+        /// </summary>
+        /// <param name="startupUserDto">The DTO containing username and password.</param>
+        /// <response code="204">Updated user name and password.</response>
+        /// <returns>
+        /// A <see cref="Task" /> that represents the asynchronous update operation.
+        /// The task result contains a <see cref="NoContentResult"/> indicating success.
+        /// </returns>
+        [HttpPost("User")]
+        [ProducesResponseType(StatusCodes.Status204NoContent)]
+        public async Task<ActionResult> UpdateStartupUser([FromBody] StartupUserDto startupUserDto)
+        {
+            var user = _userManager.Users.First();
+
+            if (startupUserDto.Name != null)
+            {
+                user.Username = startupUserDto.Name;
+            }
+
+            await _userManager.UpdateUserAsync(user).ConfigureAwait(false);
+
+            if (!string.IsNullOrEmpty(startupUserDto.Password))
+            {
+                await _userManager.ChangePassword(user, startupUserDto.Password).ConfigureAwait(false);
+            }
+
+            return NoContent();
+        }
+    }
+}