VNCServerList/Web/Controllers/VNCServerController.cs

using System;
using System.Collections.Generic;
using System.Web.Http;
using System.Linq;
using VNCServerList.Models;
using VNCServerList.Services;

namespace VNCServerList.Web.Controllers
{
    [RoutePrefix("api/vncserver")]
    public class VNCServerController : ApiController
    {
        private readonly DatabaseService _databaseService;
        private readonly VNCService _vncService;
        private readonly SettingsService _settingsService;

        public VNCServerController()
        {
            _settingsService = new SettingsService();
            _databaseService = new DatabaseService();
            _vncService = new VNCService(_databaseService, _settingsService);
        }

        [HttpGet]
        [Route("list")]
        public IHttpActionResult GetServerList(string userName = null)
        {
            try
            {
                var servers = _databaseService.GetAllServers();

                // 사용자 이름이 있으면 필터링
                if (!string.IsNullOrEmpty(userName))
                {
                    servers = servers.Where(s => s.User.Equals(userName, StringComparison.OrdinalIgnoreCase)).ToList();
                }

                // 사용자명으로 오름차순 정렬
                var sortedServers = servers.OrderBy(s => s.User).ThenBy(s => s.IP).ToList();
                return Ok(sortedServers);
            }
            catch (Exception ex)
            {
                return InternalServerError(ex);
            }
        }

        [HttpGet]
        [Route("get/{user}/{ip}")]
        public IHttpActionResult GetServer(string user, string ip)
        {
            try
            {
                var server = _databaseService.GetServerByUserAndIP(user, ip);
                if (server == null)
                {
                    return NotFound();
                }
                return Ok(server);
            }
            catch (Exception ex)
            {
                return InternalServerError(ex);
            }
        }

        [HttpPost]
        [Route("add")]
        public IHttpActionResult AddServer([FromBody] VNCServer server)
        {
            try
            {
                if (server == null)
                {
                    return BadRequest("서버 정보가 없습니다.");
                }

                bool success = _databaseService.AddServer(server);
                if (success)
                {
                    return Ok(new { Message = "서버가 성공적으로 추가되었습니다." });
                }
                else
                {
                    return BadRequest("서버 추가에 실패했습니다.");
                }
            }
            catch (Exception ex)
            {
                return InternalServerError(ex);
            }
        }

        [HttpPut]
        [Route("update/{originalUser}/{originalIp}")]
        public IHttpActionResult UpdateServer(string originalUser, string originalIp, [FromBody] VNCServer server)
        {
            try
            {
                if (server == null)
                {
                    return BadRequest("서버 정보가 없습니다.");
                }

                // 원본 사용자명과 IP로 서버를 찾아서 업데이트
                bool success = _databaseService.UpdateServerByUserAndIP(originalUser, originalIp, server);
                if (success)
                {
                    return Ok(new { Message = "서버가 성공적으로 업데이트되었습니다." });
                }
                else
                {
                    return NotFound();
                }
            }
            catch (Exception ex)
            {
                return InternalServerError(ex);
            }
        }

        [HttpDelete]
        [Route("delete/{user}/{ip}")]
        public IHttpActionResult DeleteServer(string user, string ip)
        {
            try
            {
                bool success = _databaseService.DeleteServer(user, ip);
                if (success)
                {
                    return Ok(new { Message = "서버가 성공적으로 삭제되었습니다." });
                }
                else
                {
                    return NotFound();
                }
            }
            catch (Exception ex)
            {
                return InternalServerError(ex);
            }
        }

        [HttpPost]
        [Route("connect/{user}/{ip}")]
        public IHttpActionResult ConnectToServer(string user, string ip)
        {
            try
            {
                bool success = _vncService.ConnectToServer(user, ip);
                if (success)
                {
                    return Ok(new { Message = "VNC 연결이 시작되었습니다." });
                }
                else
                {
                    return BadRequest("VNC 연결에 실패했습니다.");
                }
            }
            catch (Exception ex)
            {
                return InternalServerError(ex);
            }
        }

        [HttpGet]
        [Route("vnc-status")]
        public IHttpActionResult GetVNCStatus()
        {
            try
            {
                bool isInstalled = _vncService.IsVNCViewerInstalled();
                string path = _vncService.GetVNCViewerPath();

                return Ok(new
                {
                    IsInstalled = isInstalled,
                    Path = path
                });
            }
            catch (Exception ex)
            {
                return InternalServerError(ex);
            }
        }

        [HttpPost]
        [Route("set-vnc-path")]
        public IHttpActionResult SetVNCPath([FromBody] VNCPathRequest data)
        {
            try
            {
                string path = data.Path;
                if (string.IsNullOrEmpty(path))
                {
                    return BadRequest("VNC Viewer 경로가 제공되지 않았습니다.");
                }

                _vncService.SetVNCViewerPath(path);
                return Ok(new { Message = "VNC Viewer 경로가 설정되었습니다." });
            }
            catch (Exception ex)
            {
                return InternalServerError(ex);
            }
        }

        [HttpGet]
        [Route("settings")]
        public IHttpActionResult GetSettings()
        {
            try
            {
                System.Diagnostics.Debug.WriteLine("=== 설정 조회 시작 ===");

                var settings = _settingsService.GetSettings();

                // 디버깅용 로그
                System.Diagnostics.Debug.WriteLine($"조회된 설정: VNCViewerPath='{settings.VNCViewerPath}', WebServerPort={settings.WebServerPort}");
                System.Diagnostics.Debug.WriteLine("=== 설정 조회 완료 ===");

                return Ok(settings);
            }
            catch (Exception ex)
            {
                System.Diagnostics.Debug.WriteLine($"설정 조회 오류: {ex.Message}");
                System.Diagnostics.Debug.WriteLine($"스택 트레이스: {ex.StackTrace}");
                return InternalServerError(ex);
            }
        }

        [HttpPost]
        [Route("settings")]
        public IHttpActionResult UpdateSettings([FromBody] AppSettings settings)
        {
            try
            {
                System.Diagnostics.Debug.WriteLine("=== 설정 저장 요청 시작 ===");

                if (settings == null)
                {
                    System.Diagnostics.Debug.WriteLine("설정 객체가 null입니다.");
                    return BadRequest("설정 정보가 없습니다.");
                }

                // 디버깅용 로그
                System.Diagnostics.Debug.WriteLine($"받은 설정 데이터: VNCViewerPath='{settings.VNCViewerPath}', WebServerPort={settings.WebServerPort}");
                System.Diagnostics.Debug.WriteLine($"VNCViewerPath 타입: {settings.VNCViewerPath?.GetType()}");
                System.Diagnostics.Debug.WriteLine($"WebServerPort 타입: {settings.WebServerPort.GetType()}");

                _settingsService.SaveSettings(settings);

                // 저장된 설정 확인
                var savedSettings = _settingsService.GetSettings();
                System.Diagnostics.Debug.WriteLine($"저장된 설정: VNCViewerPath='{savedSettings.VNCViewerPath}', WebServerPort={savedSettings.WebServerPort}");
                System.Diagnostics.Debug.WriteLine("=== 설정 저장 완료 ===");

                return Ok(new { Message = "설정이 저장되었습니다." });
            }
            catch (Exception ex)
            {
                System.Diagnostics.Debug.WriteLine($"설정 저장 오류: {ex.Message}");
                System.Diagnostics.Debug.WriteLine($"스택 트레이스: {ex.StackTrace}");
                return InternalServerError(ex);
            }
        }
    }
}