sjs/vibetunnel
1
0
Fork 0
mirror of https://github.com/samsonjs/vibetunnel.git synced 2026-04-27 15:17:38 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Improve server

Browse source
This commit is contained in:
Peter Steinberger 2025-06-17 01:30:04 +02:00
parent be39ff63f0
commit 5d47d66b65
3 changed files with 66 additions and 43 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
VibeTunnel/Core/Services/RustServer.swift
View file

@ -497,9 +497,12 @@ final class RustServer: ServerProtocol {
return nil
}
let result = await group.next()
if let result = await group.next() {
group.cancelAll()
return result
}
group.cancelAll()
return result ?? nil
return nil
}
}
}

31
VibeTunnel/Core/Services/TunnelServer.swift
View file

@ -590,8 +590,13 @@ public final class TunnelServer {
return errorResponse(message: "File not found", status: .notFound)
}
// If it's a directory, return 404 (we don't serve directory listings)
// If it's a directory, try to serve index.html from that directory
if isDirectory.boolValue {
let indexPath = fullPath + "/index.html"
if FileManager.default.fileExists(atPath: indexPath) {
// Recursively serve the index.html file
return await serveStaticFile(path: sanitizedPath + "/index.html")
}
return errorResponse(message: "Directory access not allowed", status: .notFound)
}
@ -803,16 +808,10 @@ public final class TunnelServer {
}
if session.pid > 0 {
kill(pid_t(session.pid), SIGTERM)
DispatchQueue.global().asyncAfter(deadline: .now() + 1.0) {
if kill(pid_t(session.pid), 0) == 0 {
kill(pid_t(session.pid), SIGKILL)
}
}
kill(pid_t(session.pid), SIGKILL)
}
let response = SimpleResponse(success: true, message: "Session killed")
let response = SimpleResponse(success: true, message: "Session killed (SIGKILL)")
return jsonResponse(response)
} catch {
logger.error("Error killing session: \(error)")
@ -1354,9 +1353,10 @@ public final class TunnelServer {
struct SuccessResponse: Codable {
let success: Bool
let message: String
}
let response = SuccessResponse(success: true)
let response = SuccessResponse(success: true, message: "Input sent successfully")
return jsonResponse(response)
} catch let decodingError as DecodingError {
logger.error("Error decoding input request: \(decodingError)")
@ -1448,7 +1448,11 @@ public final class TunnelServer {
let expandedPath = resolvePath(mkdirRequest.path, fallback: mkdirRequest.path)
try FileManager.default.createDirectory(atPath: expandedPath, withIntermediateDirectories: true, attributes: nil)
try FileManager.default.createDirectory(
atPath: expandedPath,
withIntermediateDirectories: true,
attributes: nil
)
let response = SimpleResponse(
success: true,
@ -1461,7 +1465,10 @@ public final class TunnelServer {
return errorResponse(message: "Invalid request body. Expected JSON with 'path' field", status: .badRequest)
} catch {
logger.error("Error creating directory: \(error)")
return errorResponse(message: "Failed to create directory: \(error.localizedDescription)", status: .internalServerError)
return errorResponse(
message: "Failed to create directory: \(error.localizedDescription)",
status: .internalServerError
)
}
}

29
VibeTunnel/Core/Utilities/NetworkUtility.swift
View file

@ -32,15 +32,22 @@ enum NetworkUtility {
// But accept any non-loopback IPv4 address
if name.hasPrefix("en") {
var hostname = [CChar](repeating: 0, count: Int(NI_MAXHOST))
if getnameinfo(interface.ifa_addr, socklen_t(interface.ifa_addr.pointee.sa_len),
&hostname, socklen_t(hostname.count),
nil, 0, NI_NUMERICHOST) == 0 {
if getnameinfo(
interface.ifa_addr,
socklen_t(interface.ifa_addr.pointee.sa_len),
&hostname,
socklen_t(hostname.count),
nil,
0,
NI_NUMERICHOST
) == 0 {
let ipAddress = String(cString: &hostname)
// Prefer addresses that look like local network addresses
if ipAddress.hasPrefix("192.168.") ||
ipAddress.hasPrefix("10.") ||
ipAddress.hasPrefix("172.") {
ipAddress.hasPrefix("10.") ||
ipAddress.hasPrefix("172.")
{
return ipAddress
}
@ -78,9 +85,15 @@ enum NetworkUtility {
let addrFamily = interface.ifa_addr.pointee.sa_family
if addrFamily == UInt8(AF_INET) {
var hostname = [CChar](repeating: 0, count: Int(NI_MAXHOST))
if getnameinfo(interface.ifa_addr, socklen_t(interface.ifa_addr.pointee.sa_len),
&hostname, socklen_t(hostname.count),
nil, 0, NI_NUMERICHOST) == 0 {
if getnameinfo(
interface.ifa_addr,
socklen_t(interface.ifa_addr.pointee.sa_len),
&hostname,
socklen_t(hostname.count),
nil,
0,
NI_NUMERICHOST
) == 0 {
let ipAddress = String(cString: &hostname)
addresses.append(ipAddress)
}