View source
<?php
namespace Drupal\user\Plugin\migrate\source\d6;
use Drupal\migrate\Row;
use Drupal\migrate_drupal\Plugin\migrate\source\DrupalSqlBase;
class Role extends DrupalSqlBase {
protected $filterPermissions = [];
public function query() {
$query = $this
->select('role', 'r')
->fields('r', [
'rid',
'name',
])
->orderBy('r.rid');
return $query;
}
public function fields() {
return [
'rid' => $this
->t('Role ID.'),
'name' => $this
->t('The name of the user role.'),
];
}
protected function initializeIterator() {
$filter_roles = $this
->select('filter_formats', 'f')
->fields('f', [
'format',
'roles',
])
->execute()
->fetchAllKeyed();
foreach ($filter_roles as $format => $roles) {
foreach (explode(',', trim($roles, ',')) as $rid) {
$this->filterPermissions[$rid][] = $format;
}
}
return parent::initializeIterator();
}
public function prepareRow(Row $row) {
$rid = $row
->getSourceProperty('rid');
$permissions = $this
->select('permission', 'p')
->fields('p', [
'perm',
])
->condition('rid', $rid)
->execute()
->fetchField();
if ($permissions) {
$row
->setSourceProperty('permissions', explode(', ', $permissions));
}
else {
$row
->setSourceProperty('permissions', []);
}
if (isset($this->filterPermissions[$rid])) {
$row
->setSourceProperty("filter_permissions:{$rid}", $this->filterPermissions[$rid]);
}
return parent::prepareRow($row);
}
public function getIds() {
$ids['rid']['type'] = 'integer';
return $ids;
}
}