api.php

<?php

require __DIR__.'/../vendor/autoload.php';

use Symfony\Component\HttpFoundation\Request;
use Symfony\Component\HttpFoundation\Response;
use Symfony\Component\Serializer\Serializer;
use Symfony\Component\Serializer\Encoder\JsonEncoder;
use Symfony\Component\HttpKernel\Debug\ErrorHandler;
use Symfony\Component\HttpKernel\Exception\HttpException;

use Transport\Entity\Location\Station;
use Transport\Entity\Location\LocationQuery;
use Transport\Entity\Location\NearbyQuery;
use Transport\Web\ConnectionQueryParser;
use Transport\Web\LocationQueryParser;
use Transport\Entity\Schedule\StationBoardQuery;
use Transport\Normalizer\FieldsNormalizer;

date_default_timezone_set('Europe/Zurich');

ErrorHandler::register();

// init
$app = new Silex\Application();

// default config
$app['debug'] = true;
$app['http_cache'] = false;
$app['buzz.client'] = null;
$app['monolog.level'] = Monolog\Logger::ERROR;
$app['xhprof'] = false;
$app['redis.config'] = false; // array('host' => 'localhost', 'port' => 6379);
$app['stats.config'] = array('enabled' => false);
$app['rate_limiting.config'] = array('enabled' => false, 'limit' => 150);
$app['proxy'] = false;
$app['proxy_server.address'] = null;

/// load config
$config = __DIR__.'/../config.php';
if (stream_resolve_include_path($config)) {
	include $config;
}

// HTTP cache
if ($app['http_cache']) {
	$app->register(new Silex\Provider\HttpCacheServiceProvider(), array(
	    'http_cache.cache_dir' => __DIR__.'/../var/cache/',
	    'http_cache.options' => array('debug' => $app['debug']),
	));
}

// Exception handler
$app->error(function (\Exception $e, $code) use ($app) {

    $app['stats']->error($e);

    $errors = array(array('message' => $e->getMessage()));

    $result = array('errors' => $errors);

    return $app->json($result, $code);
});

// Monolog
$app->register(new Silex\Provider\MonologServiceProvider(), array(
    'monolog.logfile' => __DIR__.'/../var/logs/transport.log',
    'monolog.level' => $app['monolog.level'],
    'monolog.name' => 'transport',
));
$app->before(function (Request $request) use ($app) {
    $app['monolog']->addInfo('- ' . $request->getClientIp() . ' ' . $request->headers->get('referer') . ' ' . $request->server->get('HTTP_USER_AGENT'));
});

// XHProf
if ($app['xhprof']) {
    xhprof_enable();
}

// if hosted behind a reverse proxy
if ($app['proxy']) {
    $proxies = array($_SERVER['REMOTE_ADDR']);
    if (is_array($app['proxy'])) {
        $proxies = $app['proxy'];
    }
    Request::setTrustedProxies($proxies);
}

// Initialize buzz client
$client = $app['buzz.client'] ?: new Buzz\Client\FileGetContents();
if ($app['proxy_server.address']) {
    $client->setProxy($app['proxy_server.address']);
}

// create Transport API
$app['api'] = new Transport\API(new Buzz\Browser($client));

// allow cross-domain requests, enable cache
$app->after(function (Request $request, Response $response) {
    $response->headers->set('Access-Control-Allow-Origin', '*');
    $response->headers->set('Cache-Control', 's-maxage=30, public');
});

// Serializer
$app['serializer'] = $app->share(function () use ($app) {
    $fields = $app['request']->get('fields') ?: array();
    return new Serializer(array(new FieldsNormalizer($fields)), array('json' => new JsonEncoder()));
});


// Redis
$redis = null;
try {
    if ($app['redis.config']) {
        $redis = new Predis\Client($app['redis.config']);
        $redis->connect();
    }
} catch (Exception $e) {
    $app['monolog']->addError($e->getMessage());
    $redis = null;
}

// statistics
$app['stats'] = new Transport\Statistics($redis, $app['stats.config']['enabled']);
$app->after(function (Request $request, Response $response) use ($app) {
    $app['stats']->call();
    $app['stats']->resource($request->getPathInfo());
});

// rate limiting
$app['rate_limiting'] = new Transport\RateLimiting($redis, $app['rate_limiting.config']['enabled'], $app['rate_limiting.config']['limit']);

$app->before(function (Request $request) use ($app) {

    if ($app['rate_limiting']->isEnabled()) {

        $ip = $request->getClientIp();
        if ($app['rate_limiting']->hasReachedLimit($ip)) {
            throw new HttpException(429, 'Rate limit of ' . $app['rate_limiting']->getLimit() . ' requests per minute exceeded');
        }
        $app['rate_limiting']->increment($ip);
    }
});

$app->after(function (Request $request, Response $response) use ($app) {

    if ($app['rate_limiting']->isEnabled()) {

        $ip = $request->getClientIp();

        $response->headers->set('X-Rate-Limit-Limit', $app['rate_limiting']->getLimit());
        $response->headers->set('X-Rate-Limit-Remaining', $app['rate_limiting']->getRemaining($ip));
        $response->headers->set('X-Rate-Limit-Reset', $app['rate_limiting']->getReset());
    }
});

// index
$app->get('/', function(Request $request) use ($app) {
    return file_get_contents('index.html');
});


// home
$app->get('/v1/', function(Request $request) use ($app) {

    return $app->json(array(
        'date' => date('c'),
        'author' => 'Opendata.ch',
        'version' => '1.0',
    ));
});


// locations
$app->get('/v1/locations', function(Request $request) use ($app) {

    $stations = array();

    $x = $request->get('x') ?: null;
    $y = $request->get('y') ?: null;
    if ($x && $y) {
        $query = new NearbyQuery($x, $y);
        $stations = $app['api']->findNearbyLocations($query);
    }

    $query = $request->get('query');
    if ($query) {
        $query = new LocationQuery($query, $request->get('type'));
        $stations = $app['api']->findLocations($query);
    }

    $result = array('stations' => $stations);

    $json = $app['serializer']->serialize((object) $result, 'json');
    return new Response($json, 200, array('Content-Type' => 'application/json'));
});


// connections
$app->get('/v1/connections', function(Request $request) use ($app) {

    $query = LocationQueryParser::create($request);

    // get stations
    $stations = $app['api']->findLocations($query);

    // get connections
    $connections = array();
    $from = reset($stations['from']) ?: null;
    $to = reset($stations['to']) ?: null;
    $via = array();
    foreach ($stations as $k => $v) {
        if (preg_match("/^via[0-9]+$/", $k) && $v) {
            $via[] = reset($v);
        }
    }

    if ($from && $to) {
        $app['stats']->station($from);
        $app['stats']->station($to);

        $query = ConnectionQueryParser::create($request, $from, $to, $via);

        $errors = ConnectionQueryParser::validate($query);
        if ($errors) {
            return $app->json(array('errors' => $errors), 400);
        }

        $connections = $app['api']->findConnections($query);
    }

    $result = array(
        'connections' => $connections,
        'from' => $from,
        'to' => $to,
        'stations' => $stations,
    );

    $json = $app['serializer']->serialize((object) $result, 'json');
    return new Response($json, 200, array('Content-Type' => 'application/json'));
});


// station board
$app->get('/v1/stationboard', function(Request $request) use ($app) {

    $stationboard = array();

    $limit = $request->get('limit', 40);
    if ($limit > 420) {
        return new Response('Invalid value for Parameter `limit`.', 400);
    }

    $date = $request->get('date');
    if (!$date) {
        $date = $request->get('datetime');
    }
    if ($date) {
        $date = new DateTime($date, new DateTimeZone('Europe/Zurich'));
    }

    $transportations = $request->get('transportations');

    $station = $request->get('station') ?: $request->get('id');

    $query = new LocationQuery($station, 'station');
    $stations = $app['api']->findLocations($query);
    $station = reset($stations);

    if ($station instanceof Station) {
        $app['stats']->station($station);

        $query = new StationBoardQuery($station, $date);
        if ($transportations) {
            $query->transportations = $transportations;
        }
        $query->maxJourneys = $limit;
        $stationboard = $app['api']->getStationBoard($query);
    }

    $result = array('station' => $station, 'stationboard' => $stationboard);

    $json = $app['serializer']->serialize((object) $result, 'json');
    return new Response($json, 200, array('Content-Type' => 'application/json'));
});


// run
if ($app['http_cache']) {
    $app['http_cache']->run();
} else {
	$app->run();
}

// save XHProf run
if ($app['xhprof']) {

    $data = xhprof_disable();

    include_once __DIR__.'/../vendor/facebook/xhprof/xhprof_lib/utils/xhprof_lib.php';
    include_once __DIR__.'/../vendor/facebook/xhprof/xhprof_lib/utils/xhprof_runs.php';

    $xhprof = new XHProfRuns_Default(__DIR__.'/../var/xhprof');
    $run_id = $xhprof->save_run($data, 'transport');
}