59 lines
2.0 KiB
Swift
Executable File
59 lines
2.0 KiB
Swift
Executable File
import Vapor
|
|
|
|
private func authorize(_ request: Request) throws {
|
|
let key = try request.query.get(String.self, at: "key")
|
|
guard server.hasAuthorization(for: key) else {
|
|
throw Abort(.forbidden)
|
|
}
|
|
}
|
|
|
|
/// Register your application's routes here.
|
|
///
|
|
/// [Learn More →](https://docs.vapor.codes/3.0/getting-started/structure/#routesswift)
|
|
func routes(_ app: Application) throws {
|
|
|
|
// Set the name of a cap
|
|
app.postCatching("name", ":n") { request in
|
|
try authorize(request)
|
|
guard let cap = request.parameters.get("n", as: Int.self) else {
|
|
log("Invalid parameter for cap")
|
|
throw Abort(.badRequest)
|
|
}
|
|
guard let buffer = request.body.data, let name = String(data: Data(buffer: buffer), encoding: .utf8) else {
|
|
log("Invalid body data")
|
|
throw CapError.invalidBody
|
|
}
|
|
try server.set(name: name, for: cap)
|
|
}
|
|
|
|
// Upload an image
|
|
app.postCatching("images", ":n") { request -> Data in
|
|
try authorize(request)
|
|
guard let cap = request.parameters.get("n", as: Int.self) else {
|
|
log("Invalid parameter for cap")
|
|
throw Abort(.badRequest)
|
|
}
|
|
guard let buffer = request.body.data else {
|
|
log("Invalid body data")
|
|
throw CapError.invalidBody
|
|
}
|
|
let data = Data(buffer: buffer)
|
|
let newCount = try server.save(image: data, for: cap)
|
|
return "\(newCount)".data(using: .utf8)!
|
|
}
|
|
|
|
// Set a different version as the main image
|
|
app.getCatching("switch", ":n", ":v") { request in
|
|
try authorize(request)
|
|
guard let cap = request.parameters.get("n", as: Int.self), cap >= 0 else {
|
|
log("Invalid parameter for cap")
|
|
throw Abort(.badRequest)
|
|
}
|
|
guard let version = request.parameters.get("v", as: Int.self), version >= 0 else {
|
|
log("Invalid parameter for cap version")
|
|
throw Abort(.badRequest)
|
|
}
|
|
try server.switchMainImage(to: version, for: cap)
|
|
}
|
|
}
|