1 | <?php |
||
19 | use Sonata\CoreBundle\Form\Type\EqualType; |
||
|
|||
20 | use Sonata\AdminBundle\Form\Type\Filter\DefaultType; |
||
21 | |||
22 | class ModelAutocompleteFilter extends Filter |
||
23 | { |
||
24 | /** |
||
25 | * {@inheritdoc} |
||
26 | */ |
||
27 | public function filter(ProxyQueryInterface $queryBuilder, $alias, $field, $data) |
||
28 | { |
||
29 | if (!$data || !is_array($data) || !array_key_exists('value', $data)) { |
||
30 | return; |
||
31 | } |
||
32 | |||
33 | if ($data['value'] instanceof Collection) { |
||
34 | $data['value'] = $data['value']->toArray(); |
||
35 | } |
||
36 | |||
37 | if (is_array($data['value'])) { |
||
38 | $this->handleMultiple($queryBuilder, $alias, $data); |
||
39 | } else { |
||
40 | $this->handleModel($queryBuilder, $alias, $data); |
||
41 | } |
||
42 | } |
||
43 | |||
44 | /** |
||
45 | * {@inheritdoc} |
||
46 | */ |
||
47 | public function getDefaultOptions() |
||
48 | { |
||
49 | return [ |
||
50 | 'field_name' => false, |
||
51 | 'field_type' => ModelAutocompleteType::class, |
||
52 | 'field_options' => [], |
||
53 | 'operator_type' => EqualType::class, |
||
54 | 'operator_options' => [], |
||
55 | ]; |
||
56 | } |
||
57 | |||
58 | /** |
||
59 | * {@inheritdoc} |
||
60 | */ |
||
61 | public function getRenderSettings() |
||
62 | { |
||
63 | return [DefaultType::class, [ |
||
64 | 'field_type' => $this->getFieldType(), |
||
65 | 'field_options' => $this->getFieldOptions(), |
||
66 | 'operator_type' => $this->getOption('operator_type'), |
||
67 | 'operator_options' => $this->getOption('operator_options'), |
||
68 | 'label' => $this->getLabel(), |
||
69 | ]]; |
||
70 | } |
||
71 | |||
72 | /** |
||
73 | * For the record, the $alias value is provided by the association method (and the entity join method) |
||
74 | * so the field value is not used here. |
||
75 | * |
||
76 | * @param ProxyQueryInterface|QueryBuilder $queryBuilder |
||
77 | * @param string $alias |
||
78 | * @param mixed $data |
||
79 | * |
||
80 | * @return mixed |
||
81 | */ |
||
82 | protected function handleMultiple(ProxyQueryInterface $queryBuilder, $alias, $data) |
||
83 | { |
||
84 | if (0 == count($data['value'])) { |
||
85 | return; |
||
86 | } |
||
87 | |||
88 | $parameterName = $this->getNewParameterName($queryBuilder); |
||
89 | |||
90 | if (isset($data['type']) && EqualType::TYPE_IS_NOT_EQUAL == $data['type']) { |
||
91 | $this->applyWhere($queryBuilder, $queryBuilder->expr()->notIn($alias, ':'.$parameterName)); |
||
92 | } else { |
||
93 | $this->applyWhere($queryBuilder, $queryBuilder->expr()->in($alias, ':'.$parameterName)); |
||
94 | } |
||
95 | |||
96 | $queryBuilder->setParameter($parameterName, $data['value']); |
||
97 | } |
||
98 | |||
99 | /** |
||
100 | * @param ProxyQueryInterface|QueryBuilder $queryBuilder |
||
101 | * @param string $alias |
||
102 | * @param mixed $data |
||
103 | * |
||
104 | * @return mixed |
||
105 | */ |
||
106 | protected function handleModel(ProxyQueryInterface $queryBuilder, $alias, $data) |
||
107 | { |
||
108 | if (empty($data['value'])) { |
||
109 | return; |
||
110 | } |
||
111 | |||
112 | $parameterName = $this->getNewParameterName($queryBuilder); |
||
113 | |||
114 | if (isset($data['type']) && EqualType::TYPE_IS_NOT_EQUAL == $data['type']) { |
||
115 | $this->applyWhere($queryBuilder, sprintf('%s != :%s', $alias, $parameterName)); |
||
116 | } else { |
||
117 | $this->applyWhere($queryBuilder, sprintf('%s = :%s', $alias, $parameterName)); |
||
118 | } |
||
119 | |||
120 | $queryBuilder->setParameter($parameterName, $data['value']); |
||
121 | } |
||
122 | |||
123 | /** |
||
124 | * {@inheritdoc} |
||
125 | */ |
||
126 | protected function association(ProxyQueryInterface $queryBuilder, $data) |
||
127 | { |
||
128 | $associationMappings = $this->getParentAssociationMappings(); |
||
129 | $associationMappings[] = $this->getAssociationMapping(); |
||
130 | $alias = $queryBuilder->entityJoin($associationMappings); |
||
131 | |||
132 | return [$alias, false]; |
||
133 | } |
||
134 | } |
||
135 |