HnrAzevedo\Datamanager\EntityTrait

Complexity

Total Complexity

13

Size/Duplication

Total Lines	111
Duplicated Lines	0 %

Importance

Changes	5
Bugs	0	Features	2

<?php
trait EntityTrait{
    use CheckTrait;

    protected string $lastQuery = '';
    protected array $lastData = [];
    
    public function toEntity()
    {
        if($this->getCount() === 0){
            return null;
        }

        $entity = $this->setByDatabase($this->result[0]);

        if(count($this->result) > 1){
            $entity = [];
            foreach ($this->result as $key => $value) {
                $entity[] = $this->setByDatabase($value);
            }
        }

        return $entity;
    }

    public function persist()
    {
        $columns = '';
        $values = '';
        $data = [];

        foreach ($this->data as $key => $value) {
            if(strstr($this->data[$key]['extra'],'auto_increment')){
                continue;
            }

            $this->checkMaxlength($key, $value['value'], $value['maxlength']);

            $columns .= $key.',';
            $values .= ':'.$key.',';
            $data[$key] = $value['value'];
        }

        $this->transaction('begin');
        try{

            $this->checkUniques($data);
           
            $id = $this->insert($data);

            $this->check_fail();

            $primary = $this->primary;
            
            $this->$primary = $id;
            
            $this->transaction('commit');

        }catch(DatamanagerException $er){
            $this->transaction('rollback');
            throw $er;
        }

        return $this;
    }

    public function remove(bool $exec = false)
    {
        if(!$exec){
            $this->clause = 'remove';    
            return $this;
        }

        $this->clause = null;

        if(count($this->where) == 1){
            $this->removeById();
            return $this;
        }

        $this->delete(
            $this->mountRemove()['where'], 
            substr( $this->mountRemove()['data'] ,0,-1)
        );

        $this->check_fail();
            
        return $this;
    }

    public function save()
    {
        $this->transaction('begin');

        try{
            $this->checkForChanges();

            $this->update(
                $this->mountSave()['data'],
                "{$this->primary}=:{$this->primary}", 
                $this->primary.'='.$this->getData()[$this->primary]['value']
            );

            $this->check_fail();

            $this->transaction('commit');
        }catch(DatamanagerException $er){
            $this->transaction('rollback');
            throw $er;
        }

        return $this;
    }
}