Completed
Push — master ( 343d0a...c5b636 )
by James Ekow Abaka
02:58
created

ManyHaveManyRelationship::preSave()   A

Complexity

Conditions 1
Paths 1

Size

Total Lines 8
Code Lines 6

Duplication

Lines 0
Ratio 0 %

Code Coverage

Tests 0
CRAP Score 2

Importance

Changes 0
Metric Value
c 0
b 0
f 0
dl 0
loc 8
ccs 0
cts 7
cp 0
rs 9.4285
cc 1
eloc 6
nc 1
nop 2
crap 2
1
<?php
2
3
/*
4
 * The MIT License
5
 *
6
 * Copyright 2015 ekow.
7
 *
8
 * Permission is hereby granted, free of charge, to any person obtaining a copy
9
 * of this software and associated documentation files (the "Software"), to deal
10
 * in the Software without restriction, including without limitation the rights
11
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
12
 * copies of the Software, and to permit persons to whom the Software is
13
 * furnished to do so, subject to the following conditions:
14
 *
15
 * The above copyright notice and this permission notice shall be included in
16
 * all copies or substantial portions of the Software.
17
 *
18
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
19
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
20
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
21
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
22
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
23
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
24
 * THE SOFTWARE.
25
 */
26
27
namespace ntentan\nibii\relationships;
28
29
use ntentan\utils\Text;
30
use ntentan\nibii\ORMContext;
31
use ntentan\nibii\NibiiException;
32
33
class ManyHaveManyRelationship extends \ntentan\nibii\Relationship {
34
35
    protected $type = self::MANY_HAVE_MANY;
36
    private $tempdata;
37
    private $junctionModelInstance;
38
    
39 4
    public function __construct(ORMContext $context) {
40 4
        $this->container = $context->getContainer();
41 4
        $this->context = $context;
42 4
    }    
43
44 4
    public function prepareQuery($data) {
0 ignored issues
show
Documentation introduced by
The return type could not be reliably inferred; please add a @return annotation.

Our type inference engine in quite powerful, but sometimes the code does not provide enough clues to go by. In these cases we request you to add a @return annotation as described here.

Loading history...
45 4
        $junctionModel = $this->container->resolve($this->options['junction_model']);
46 4
        $filter = $junctionModel->fields($this->options['junction_foreign_key'])
47 4
            ->filterBy($this->options['junction_local_key'], $data[$this->options['local_key']])
48 4
            ->fetch();
49 4
        if($filter->count() == 0) {
50
            return null;
51
        }
52 4
        $foreignKeys = [];
53 4
        foreach ($filter->toArray() as $foreignItem) {
54 4
            $foreignKeys[] = $foreignItem[$this->options['junction_foreign_key']];
55
        }
56
57
        // @todo throw an exception when the data doesn't have the local key
58 4
        $query = $this->getQuery();
59 4
        if($this->queryPrepared){
60
            $query->setBoundData($this->options['foreign_key'], $foreignKeys);
61
        } else {
62 4
            $query = $this->getQuery()
63 4
                ->setTable($this->getModelInstance()->getDBStoreInformation()['quoted_table'])
64 4
                ->addFilter($this->options['foreign_key'], $foreignKeys);
65 4
            $this->queryPrepared = true;
66
        }
67 4
        return $query;
68
        
69
    }
70
71 4
    public function runSetup() {
72 4
        if (isset($this->options['through'])) {
73
            $junctionModelName = 
74
                $this->context->getClassName($this->options['through']);
75
        } else {
76
            $junctionModelName = 
77 4
                $this->context->joinModels($this->setupName, $this->options['model']);
78
        }
79 4
        $this->options['junction_model'] = $junctionModelName;
80
81 4
        $foreignModel = $this->context->load($this->options['model']);
82 4
        if ($this->options['foreign_key'] == null) {
83 4
            $this->options['foreign_key'] = 
84 4
                $foreignModel->getDescription()->getPrimaryKey()[0];
85
        }
86
87 4 View Code Duplication
        if ($this->options['local_key'] == null) {
88 4
            $this->options['local_key'] = $this->setupPrimaryKey[0];
89
        }
90
91 4 View Code Duplication
        if (!isset($this->options['junction_local_key'])) {
92 4
            $this->options['junction_local_key'] = 
93 4
                Text::singularize($this->setupTable) . '_id';
94
        }
95
96 4 View Code Duplication
        if (!isset($this->options['junction_foreign_key'])) {
97 4
            $this->options['junction_foreign_key'] = 
98 4
                Text::singularize($foreignModel->getDBStoreInformation()['table']) . '_id';
99
        }
100 4
    }  
101
    
102
    public function preSave(&$wrapper, $value) {
103
        $this->tempdata = $wrapper[$this->options['model']];
104
        $this->junctionModelInstance = $this->container->resolve($this->options['junction_model']);
105
        $this->junctionModelInstance->delete([
106
            $this->options['junction_local_key'] => $wrapper[$this->options['local_key']]
107
        ]);
108
        unset($wrapper[$this->options['model']]);
109
    }
110
    
111
    public function postSave(&$wrapper) {
112
        $jointModelRecords = [];
113
        foreach($this->tempdata as $relatedRecord) {
114
            $data = $relatedRecord->toArray();
115
            if(!isset($data[$this->options['foreign_key']]) || (isset($data[$this->options['foreign_key']]) && count($data) > 1)) {
116
                if(!$relatedRecord->save()) {
117
                    throw new NibiiException("Failed to save related model {$this->options['model']}");
118
                }
119
            }
120
            $jointModelRecords[] = [
121
                $this->options['junction_local_key'] => $wrapper[$this->options['local_key']],
122
                $this->options['junction_foreign_key'] => $relatedRecord[$this->options['foreign_key']]
123
            ];
124
        }
125
        $this->junctionModelInstance->setData($jointModelRecords);
126
        $this->junctionModelInstance->save();
127
    }
128
129
}
130