View source
<?php
namespace Drupal\address\Element;
use Drupal\Core\Form\FormStateInterface;
use Drupal\Core\Render\Element\FormElement;
class Country extends FormElement {
public function getInfo() {
$class = get_class($this);
return [
'#available_countries' => [],
'#input' => TRUE,
'#multiple' => FALSE,
'#default_value' => NULL,
'#process' => [
[
$class,
'processCountry',
],
[
$class,
'processGroup',
],
],
'#theme_wrappers' => [
'container',
],
];
}
public static function processCountry(array &$element, FormStateInterface $form_state, array &$complete_form) {
if (isset($element['#available_countries']) && !is_array($element['#available_countries'])) {
throw new \InvalidArgumentException('The #available_countries property must be an array.');
}
$full_country_list = \Drupal::service('address.country_repository')
->getList();
$country_list = $full_country_list;
if (!empty($element['#available_countries'])) {
$available_countries = $element['#available_countries'];
if (!empty($element['#default_value'])) {
$available_countries[] = $element['#default_value'];
}
$available_countries = array_combine($available_countries, $available_countries);
$country_list = array_intersect_key($country_list, $available_countries);
}
if (empty($element['#default_value']) && $element['#required']) {
$element['#default_value'] = key($country_list);
}
$element['#tree'] = TRUE;
if (count($country_list) == 1 && $element['#required']) {
$element['country_code'] = [
'#type' => 'hidden',
'#value' => key($available_countries),
];
}
else {
$element['country_code'] = [
'#type' => 'select',
'#title' => $element['#title'],
'#title_display' => $element['#title_display'],
'#description_display' => $element['#description_display'],
'#options' => $country_list,
'#default_value' => $element['#default_value'],
'#required' => $element['#required'],
'#limit_validation_errors' => [],
'#attributes' => [
'class' => [
'country',
],
'autocomplete' => 'country',
],
'#weight' => -100,
];
if (!$element['#required']) {
$element['country_code']['#empty_value'] = '';
}
if (!empty($element['#ajax'])) {
$element['country_code']['#ajax'] = $element['#ajax'];
unset($element['#ajax']);
}
if (!empty($element['#description'])) {
$element['country_code']['#description'] = $element['#description'];
unset($element['#description']);
}
}
$element['country_code']['#parents'] = $element['#parents'];
return $element;
}
public static function getDefaultCountry(array $available_countries = []) {
$full_country_list = \Drupal::service('address.country_repository')
->getList();
$country_list = $full_country_list;
if (!empty($available_countries)) {
$available_countries = array_combine($available_countries, $available_countries);
$country_list = array_intersect_key($country_list, $available_countries);
}
$default_country = key($country_list);
return $default_country;
}
}