110 lines
3.2 KiB
C#
110 lines
3.2 KiB
C#
using AmtScanner.Api.Data;
|
|
using AmtScanner.Api.Services;
|
|
using Microsoft.AspNetCore.Mvc;
|
|
using Microsoft.EntityFrameworkCore;
|
|
|
|
namespace AmtScanner.Api.Controllers;
|
|
|
|
[ApiController]
|
|
[Route("api/[controller]")]
|
|
public class RemoteDesktopController : ControllerBase
|
|
{
|
|
private readonly IGuacamoleService _guacamoleService;
|
|
private readonly AppDbContext _context;
|
|
private readonly ILogger<RemoteDesktopController> _logger;
|
|
|
|
public RemoteDesktopController(
|
|
IGuacamoleService guacamoleService,
|
|
AppDbContext context,
|
|
ILogger<RemoteDesktopController> logger)
|
|
{
|
|
_guacamoleService = guacamoleService;
|
|
_context = context;
|
|
_logger = logger;
|
|
}
|
|
|
|
/// <summary>
|
|
/// 获取远程桌面连接 URL
|
|
/// </summary>
|
|
[HttpPost("connect/{deviceId}")]
|
|
public async Task<ActionResult<RemoteDesktopResponse>> Connect(long deviceId, [FromBody] RdpCredentials credentials)
|
|
{
|
|
var device = await _context.AmtDevices.FindAsync(deviceId);
|
|
if (device == null)
|
|
{
|
|
return NotFound(new { error = "设备不存在" });
|
|
}
|
|
|
|
// 获取 Guacamole Token
|
|
var token = await _guacamoleService.GetAuthTokenAsync();
|
|
if (string.IsNullOrEmpty(token))
|
|
{
|
|
return StatusCode(503, new { error = "无法连接到 Guacamole 服务，请确保 Guacamole 已启动" });
|
|
}
|
|
|
|
// 创建或获取连接
|
|
var connectionName = $"AMT-{device.IpAddress}";
|
|
var connectionId = await _guacamoleService.CreateOrGetConnectionAsync(
|
|
token,
|
|
connectionName,
|
|
device.IpAddress,
|
|
credentials.Username,
|
|
credentials.Password);
|
|
|
|
if (string.IsNullOrEmpty(connectionId))
|
|
{
|
|
return StatusCode(500, new { error = "创建远程连接失败" });
|
|
}
|
|
|
|
// 获取连接 URL
|
|
var connectionUrl = await _guacamoleService.GetConnectionUrlAsync(token, connectionId);
|
|
|
|
_logger.LogInformation("Created remote desktop connection for device {Ip}, connectionId: {Id}",
|
|
device.IpAddress, connectionId);
|
|
|
|
return Ok(new RemoteDesktopResponse
|
|
{
|
|
Success = true,
|
|
ConnectionUrl = connectionUrl,
|
|
ConnectionId = connectionId,
|
|
Token = token
|
|
});
|
|
}
|
|
|
|
/// <summary>
|
|
/// 测试 Guacamole 连接
|
|
/// </summary>
|
|
[HttpGet("test")]
|
|
public async Task<ActionResult> TestConnection()
|
|
{
|
|
var token = await _guacamoleService.GetAuthTokenAsync();
|
|
if (string.IsNullOrEmpty(token))
|
|
{
|
|
return StatusCode(503, new {
|
|
success = false,
|
|
error = "无法连接到 Guacamole 服务"
|
|
});
|
|
}
|
|
|
|
return Ok(new {
|
|
success = true,
|
|
message = "Guacamole 服务连接正常"
|
|
});
|
|
}
|
|
}
|
|
|
|
public class RdpCredentials
|
|
{
|
|
public string Username { get; set; } = string.Empty;
|
|
public string Password { get; set; } = string.Empty;
|
|
}
|
|
|
|
public class RemoteDesktopResponse
|
|
{
|
|
public bool Success { get; set; }
|
|
public string? ConnectionUrl { get; set; }
|
|
public string? ConnectionId { get; set; }
|
|
public string? Token { get; set; }
|
|
public string? Error { get; set; }
|
|
}
|