37 lines
1001 B
Swift
37 lines
1001 B
Swift
|
import Vapor
|
||
|
|
import BinaryCodable
|
||
|
|
|
||
|
|
private let responseEncoder = BinaryEncoder()
|
||
|
|
private let requestDecoder = BinaryDecoder()
|
||
|
|
|
||
|
|
var serverStatus: ServerStatus = .starting
|
||
|
|
|
||
|
|
var knownTokens = Set<String>()
|
||
|
|
|
||
|
|
func routes(_ app: Application) throws {
|
||
|
|
|
||
|
|
app.post("token") { req async throws -> HTTPResponseStatus in
|
||
|
|
guard let data = req.body.data?.allReadableBytes else {
|
||
|
|
return .badRequest
|
||
|
|
}
|
||
|
|
let tokenUpload: TokenUpload = try requestDecoder.decode(from: data)
|
||
|
|
if let oldToken = tokenUpload.previousToken {
|
||
|
|
knownTokens.remove(oldToken.hex)
|
||
|
|
}
|
||
|
|
knownTokens.insert(tokenUpload.currentToken.hex)
|
||
|
|
return .ok
|
||
|
|
}
|
||
|
|
|
||
|
|
app.get("status") { req async throws -> Response in
|
||
|
|
let data = try responseEncoder.encode(serverStatus)
|
||
|
|
return Response(status: .ok, body: .init(data: data))
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
private extension ByteBuffer {
|
||
|
|
|
||
|
|
var allReadableBytes: Data? {
|
||
|
|
getData(at: 0, length: readableBytes)
|
||
|
|
}
|
||
|
|
}
|