View source
<?php
namespace Drupal\bat_api\Plugin\ServiceDefinition;
use Drupal\Core\Database\Connection;
use Drupal\Core\Entity\EntityTypeManagerInterface;
use Drupal\Core\Extension\ModuleHandlerInterface;
use Drupal\Core\Plugin\ContainerFactoryPluginInterface;
use Drupal\Core\Session\AccountInterface;
use Drupal\services\ServiceDefinitionBase;
use Symfony\Component\DependencyInjection\ContainerInterface;
use Symfony\Component\Serializer\SerializerInterface;
use Symfony\Component\HttpFoundation\Request;
use Drupal\Core\Routing\RouteMatchInterface;
use Drupal\Core\Database\Database;
use Roomify\Bat\Calendar\Calendar;
use Roomify\Bat\Store\DrupalDBStore;
use Roomify\Bat\Unit\Unit;
use Drupal\bat_fullcalendar\FullCalendarFixedStateEventFormatter;
use Drupal\bat_fullcalendar\FullCalendarOpenStateEventFormatter;
class CalendarEventsIndex extends ServiceDefinitionBase implements ContainerFactoryPluginInterface {
protected $entityTypeManager;
protected $currentUser;
protected $moduleHandler;
protected $connection;
protected $fixedStateEventFormatter;
protected $openStateEventFormatter;
public function __construct(array $configuration, $plugin_id, $plugin_definition, EntityTypeManagerInterface $entity_manager, AccountInterface $current_user, ModuleHandlerInterface $module_handler, Connection $connection, FullCalendarFixedStateEventFormatter $fixedStateEventFormatter, FullCalendarOpenStateEventFormatter $openStateEventFormatter) {
parent::__construct($configuration, $plugin_id, $plugin_definition);
$this->entityTypeManager = $entity_manager;
$this->currentUser = $current_user;
$this->moduleHandler = $module_handler;
$this->connection = $connection;
$this->fixedStateEventFormatter = $fixedStateEventFormatter;
$this->openStateEventFormatter = $openStateEventFormatter;
}
public static function create(ContainerInterface $container, array $configuration, $plugin_id, $plugin_definition) {
return new static($configuration, $plugin_id, $plugin_definition, $container
->get('entity_type.manager'), $container
->get('current_user'), $container
->get('module_handler'), $container
->get('database'), $container
->get('bat_fullcalendar.fixed_state_event_formatter'), $container
->get('bat_fullcalendar.open_state_event_formatter'));
}
public function getCacheMaxAge() {
return 0;
}
public function processRequest(Request $request, RouteMatchInterface $route_match, SerializerInterface $serializer) {
$unit_types = $request->query
->get('unit_types');
$event_types = $request->query
->get('event_types');
$background = $request->query
->get('background');
$unit_ids = $request->query
->get('unit_ids');
$start_date = $request->query
->get('start');
$end_date = $request->query
->get('end');
$start_date_object = new \DateTime($start_date);
$end_date_object = new \DateTime($end_date);
if ($unit_types == 'all') {
$unit_types = [];
foreach (bat_unit_get_types() as $type => $info) {
$unit_types[] = $type;
}
}
else {
$unit_types = array_filter(explode(',', $unit_types));
}
if ($event_types == 'all') {
$types = [];
foreach (bat_event_get_types() as $type => $info) {
$types[] = $type;
}
}
else {
$types = array_filter(explode(',', $event_types));
}
$events_json = [];
foreach ($types as $type) {
if (!$this->currentUser
->hasPermission('view calendar data for any ' . $type . ' event')) {
continue;
}
$bat_event_type = bat_event_type_load($type);
$target_entity_type = $bat_event_type
->getTargetEntityType();
$database = Database::getConnectionInfo('default');
$prefix = isset($database['default']['prefix']['default']) ? $database['default']['prefix']['default'] : '';
$event_store = new DrupalDBStore($type, DrupalDBStore::BAT_EVENT, $prefix);
$today = new \DateTime();
if (!$this->currentUser
->hasPermission('view past event information') && $today > $start_date_object) {
if ($today > $end_date_object) {
return [];
}
$start_date_object = $today;
}
$ids = array_filter(explode(',', $unit_ids));
foreach ($unit_types as $unit_type) {
$entities = $this
->getReferencedIds($unit_type, $ids);
$childrens = [];
$units = [];
foreach ($entities as $entity) {
$units[] = new Unit($entity['id'], 0);
}
if (!empty($units)) {
$event_calendar = new Calendar($units, $event_store);
$event_ids = $event_calendar
->getEvents($start_date_object, $end_date_object);
if ($bat_event_type
->getFixedEventStates()) {
$event_formatter = $this->fixedStateEventFormatter;
}
else {
$event_formatter = $this->openStateEventFormatter;
}
$event_formatter
->setEventType($bat_event_type);
$event_formatter
->setBackground($background);
foreach ($event_ids as $unit_id => $unit_events) {
foreach ($unit_events as $key => $event) {
$events_json[] = [
'id' => (string) $key . $unit_id,
'bat_id' => $event
->getValue(),
'resourceId' => 'S' . $unit_id,
] + $event
->toJson($event_formatter);
}
}
}
}
}
$context = array(
'unit_ids' => $unit_ids,
'unit_types' => $unit_types,
'start_date' => $start_date_object,
'end_date' => $end_date_object,
'event_types' => $event_types,
'background' => $background,
);
$this->moduleHandler
->alter('bat_api_events_index_calendar', $events_json, $context);
return array_values($events_json);
}
public function getReferencedIds($unit_type, $ids = []) {
$query = $this->connection
->select('unit', 'n')
->fields('n', [
'id',
'unit_type_id',
'type',
'name',
]);
if (!empty($ids)) {
$query
->condition('id', $ids, 'IN');
}
$query
->condition('unit_type_id', $unit_type);
$bat_units = $query
->execute()
->fetchAll();
$units = [];
foreach ($bat_units as $unit) {
$units[] = [
'id' => $unit->id,
'name' => $unit->name,
'type_id' => $unit_type,
];
}
return $units;
}
}