Note that the returned Body instance may change between calls, the contents not
explicitly fetched through its valid() / consume() API, will be moved to the new
instance though. No need to buffer and concatenate manually in this case.
/**
* @param Request $req
* @param array $options available options are:
* - size (default: 131072)
* - input_vars (default: 200)
* - field_len (default: 16384)
*/
public function __construct(Request $req, array $options = [])
{
$this->req = $req;
$type = $req->getHeader("content-type");
$this->body = $req->getBody($this->totalSize = $this->size = $options["size"] ?? 131072);
$this->maxFieldLen = $options["field_len"] ?? 16384;
$this->maxInputVars = $options["input_vars"] ?? 200;
if ($type !== null && strncmp($type, "application/x-www-form-urlencoded", \strlen("application/x-www-form-urlencoded"))) {
if (!preg_match('#^\\s*multipart/(?:form-data|mixed)(?:\\s*;\\s*boundary\\s*=\\s*("?)([^"]*)\\1)?$#', $type, $m)) {
$this->req = null;
$this->parsing = true;
$this->result = new ParsedBody([]);
return;
}
$this->boundary = $m[2];
}
$this->body->when(function ($e, $data) {
$this->req = null;
\Amp\immediately(function () use($e, $data) {
if ($e) {
if ($e instanceof ClientSizeException) {
$e = new ClientException("", 0, $e);
}
$this->error = $e;
} else {
$this->result = $this->end($data);
}
if (!$this->parsing) {
$this->parsing = true;
foreach ($this->result->getNames() as $field) {
foreach ($this->result->getArray($field) as $_) {
foreach ($this->watchers as list($cb, $cbData)) {
$cb($field, $cbData);
}
}
}
}
$this->parsing = true;
foreach ($this->whens as list($cb, $cbData)) {
$cb($this->error, $this->result, $cbData);
}
$this->whens = $this->watchers = [];
});
});
}
private function getJsonRequestBody(AerysRequest $request, AerysResponse $response) : \Generator
{
if ($request->getHeader('Content-Type') !== 'application/json') {
$this->respondWithError($response, 400, 'Type of request body must be application/json');
return null;
}
$body = $request->getBody();
$bodyText = '';
while ((yield $body->valid())) {
$bodyText .= $body->consume();
}
try {
$json = json_try_decode($bodyText, true);
} catch (JSONDecodeErrorException $e) {
$this->respondWithError($response, 400, 'Unable to decode request body as application/json');
return null;
}
if (!is_array($json)) {
$this->respondWithError($response, 400, 'Invalid request data structure');
return null;
}
return $json;
}
public function doJoin(Request $request, Response $response)
{
$session = (yield (new Session($request))->open());
parse_str((yield $request->getBody()), $post);
$username = isset($post["username"]) && is_string($post["username"]) ? $post["username"] : "";
$avatar = $session->get("auth:identity:avatar");
$provider = $session->get("auth:provider");
if (!$provider) {
$response->setStatus(400);
$response->setHeader("aerys-generic-response", "enable");
$response->send("");
return;
}
if (!preg_match("~^[a-z][a-z0-9-]+[a-z0-9]\$~i", $username)) {
$template = $this->templateService->load("sign-up.php");
$template->set("hint", $username);
$template->set("error", "username must start with a-z and only contain a-z, 0-9 or dashes afterwards");
$response->send($template->render());
return;
}
$query = (yield $this->db->prepare("INSERT IGNORE INTO user (`name`, `avatar`) VALUES (?, ?)", [$username, $avatar]));
if ($query->affectedRows) {
(yield $this->db->prepare("INSERT INTO oauth (user_id, provider, identity, label) VALUES (?, ?, ?, ?)", [$query->insertId, $provider, $session->get("auth:identity:id"), $session->get("auth:identity:name")]));
$session->set("login", $query->insertId);
$session->set("login:name", $username);
$session->set("login:avatar", $avatar);
$session->set("login:time", time());
$response->setStatus(302);
$response->setHeader("location", "/");
$response->send("");
} else {
$template = $this->templateService->load("sign-up.php");
$template->set("hint", $username);
$template->set("error", "username already taken");
$response->send($template->render());
}
(yield $session->save());
}
/**
* Handles all hooks.
*
* @param Request $request HTTP request
* @param Response $response HTTP response
* @param array $args URL args
*/
public function handle(Request $request, Response $response, array $args)
{
$response->setHeader("content-type", "text/plain");
$token = $request->getQueryVars()["token"] ?? "";
if (!$token || !is_string($token)) {
$response->setStatus(401);
$response->send("Failure: No token was provided.");
return;
}
// use @ so we don't have to check for invalid strings manually
$token = (string) @hex2bin($token);
$hook = (yield $this->hookRepository->get($args["id"]));
if (!$hook) {
$response->setStatus(404);
$response->send("Failure: Hook does not exist.");
return;
}
if (!hash_equals($hook->token, $token)) {
$response->setStatus(403);
$response->send("Failure: Provided token doesn't match.");
return;
}
$name = $args["service"];
if (!isset($this->services[$name])) {
$response->setStatus(404);
$response->send("Failure: Unknown service.");
return;
}
$contentType = strtok($request->getHeader("content-type"), ";");
$body = (yield $request->getBody());
switch ($contentType) {
case "application/json":
$payload = json_decode($body);
break;
case "application/x-www-form-urlencoded":
parse_str($body, $payload);
$payload = json_decode(json_encode($payload));
break;
default:
$response->setStatus(415);
$response->send("Failure: Content-type not supported.");
return;
}
$service = $this->services[$name];
$headers = $request->getAllHeaders();
$event = $service->getEventName($headers, $payload);
if (!isset($this->schemas[$name][$event])) {
$response->setStatus(400);
$response->send("Failure: Event not supported.");
return;
}
$schema = $this->schemas[$name][$event];
$this->validator->reset();
$this->validator->check($payload, $schema);
if (!$this->validator->isValid()) {
$errors = $this->validator->getErrors();
$errors = array_reduce($errors, function (string $carry, array $item) : string {
if ($item["property"]) {
return $carry . sprintf("\n%s: %s", $item["property"], $item["message"]);
} else {
return $carry . "\n" . $item["message"];
}
}, "");
$response->setStatus(400);
$response->send("Failure: Payload validation failed." . $errors);
return;
}
$message = $service->handle($headers, $payload);
try {
if ($message) {
$req = (new HttpRequest())->setMethod("PUT")->setUri($this->config["api"] . "/messages")->setHeader("authorization", "Basic " . base64_encode("{$this->config['user_id']}:{$this->config['token']}"))->setBody(json_encode(["room_id" => $hook->room_id, "text" => $message->getText(), "data" => $message->getData()]));
$resp = (yield $this->http->request($req));
if (intval($resp->getStatus() / 100) !== 2) {
$message = "API request failed: " . $resp->getStatus();
if ($resp->getBody()) {
$message .= "\n" . $resp->getBody();
}
throw new Exception($message);
}
}
$response->send("Success: " . ($message ? "Message sent." : "Message skipped."));
} catch (Exception $e) {
$response->setStatus(500);
$response->send("Failure: Couldn't persist message.");
}
}
public function __invoke(Request $req, Response $res)
{
$headers = $req->getAllHeaders();
unset($headers["accept-encoding"]);
$connection = $headers["connection"];
unset($headers["connection"]);
foreach ($connection as $value) {
foreach (explode(",", strtolower($value)) as $type) {
$type = trim($type);
if ($type == "upgrade") {
$headers["connection"][0] = "upgrade";
} else {
unset($headers[$type]);
}
}
}
if ($this->headers) {
if (is_callable($this->headers)) {
$headers = ($this->headers)($headers);
} else {
$headers = $this->headers + $headers;
}
}
$promise = $this->client->request((new \Amp\Artax\Request())->setMethod($req->getMethod())->setUri($this->target . $req->getUri())->setAllHeaders($headers)->setBody((yield $req->getBody())));
// no async sending possible :-( [because of redirects]
$promise->watch(function ($update) use($req, $res, &$hasBody, &$status, &$zlib) {
list($type, $data) = $update;
if ($type == Notify::RESPONSE_HEADERS) {
$headers = array_change_key_case($data["headers"], CASE_LOWER);
foreach ($data["headers"] as $header => $values) {
foreach ($values as $value) {
$res->addHeader($header, $value);
}
}
$res->setStatus($status = $data["status"]);
$res->setReason($data["reason"]);
if (isset($headers["content-encoding"]) && strcasecmp(trim(current($headers["content-encoding"])), 'gzip') === 0) {
$zlib = inflate_init(ZLIB_ENCODING_GZIP);
}
$hasBody = true;
}
if ($type == Notify::RESPONSE_BODY_DATA) {
if ($zlib) {
$data = inflate_add($zlib, $data);
}
$res->stream($data);
}
if ($type == Notify::RESPONSE) {
if (!$hasBody) {
foreach ($data->getAllHeaders() as $header => $values) {
foreach ($values as $value) {
$res->addHeader($header, $value);
}
}
$res->setStatus($status = $data->getStatus());
$res->setReason($data->getReason());
}
if ($status == 101) {
$req->setLocalVar("aerys.reverse.socket", $update["export_socket"]());
}
$res->end($zlib ? inflate_add("", ZLIB_FINISH) : null);
}
});
(yield $promise);
}
