cascade_delete_behavior/CascadeDeleteBehavior.php at master · maxlapko/cascade_delete_behavior · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
<?php

/**
 * @author mlapko <maxlapko@gmail.com>
 */
class CascadeDeleteBehavior extends CActiveRecordBehavior
{
    const TYPE_ALL  = 'all';
    const TYPE_EACH = 'each';

    const COMMAND_DELELE_ALL = 'delete';
    const COMMAND_UPDATE_ALL = 'update';

    /**
     * array(
     *  'posts' => array('type' => CascadeDeleteBehavior::TYPE_ALL),
     *  'category'
     *
     * )
     * @var array
     */
    public $relations = array();

    /**
     *
     * @param CModelEvent $event event parameter
     * @return boolean
     */
    public function beforeDelete($event)
    {
        $ownerRelations = $this->getOwner()->relations();
        foreach ($this->relations as $key => $value) {
            if (is_array($value)) {
                $relation = $ownerRelations[$key];
                $relation['_relation'] = $key;
                $this->_deleteByParams($relation, $value);
            } else {
                $this->_deleteRelation($value);
            }
        }
        return true;
    }

    /**
     * Cascade delete relations
     *
     * @param array $relation
     * @param array $params
     */
    protected function _deleteByParams($relation, $params)
    {
        if (empty($params['type']) || !in_array($params['type'], array(self::TYPE_ALL, self::TYPE_EACH))) {
            throw new CException('Incorrect value for "type" param.');
        }

        if (!empty($params['command']) && !in_array($params['command'], array(self::COMMAND_DELELE_ALL, self::COMMAND_UPDATE_ALL))) {
            throw new CException('Incorrect value for "command" param.');
        }

        if ($params['type'] === self::TYPE_EACH) {
            $this->_deleteRelation($relation['_relation']);
        } else {
            if ($relation[0] === CActiveRecord::MANY_MANY) {
                throw new CException('Not supported MANY MANY associations.');
            }
            $relationObject = CActiveRecord::model($relation[1]);
            list($condition, $conditionParams) = $this->_prepareConditionAndParams($relation);
            if (isset($params['command']) && $params['command'] === self::COMMAND_UPDATE_ALL && !empty($params['attributes'])) {
                $relationObject->updateAll($params['attributes'], $condition, $conditionParams);
            } else {
                $relationObject->deleteAll($condition, $conditionParams);
            }
        }
    }

    /**
     * Delete each relation
     *
     * @param string $relation
     */
    protected function _deleteRelation($relation)
    {
        $objects = $this->getOwner()->getRelated($relation);
        if ($objects !== null) {
            if (is_array($objects)) {
                foreach ($objects as $object) {
                    $object->delete();
                }
            } else {
                $objects->delete();
            }
        }
    }

    /**
     * Prepare condition and params for delete
     * @param array $relation
     * @return array
     */
    protected function _prepareConditionAndParams($relation)
    {
        $condition = $relation[2] . ' = :_pk';
        if (isset($relation['on'])) {
            $condition .= ' AND ' . $relation['on'];
        }
        if (isset($relation['condition'])) {
            $condition .= ' AND ' . $relation['condition'];
        }
        $params = array(':_pk' => $this->getOwner()->getPrimaryKey());
        if (isset($relation['params'])) {
            $params = array_merge($params, $relation['params']);
        }
        return array($condition, $params);
    }


}