magros / laravel-model-encryption

src/Encryptable.php

<?php namespace Magros\Encryptable;

trait Encryptable
{

    public $aesKey = null;
    public static $enableEncryption = true;
    private $magrosEncrypter;

    /**
     * @return Encrypter
     */
    public function encrypter()
    {
        if(! $this->magrosEncrypter){
            $this->magrosEncrypter = new Encrypter();
        }
        return $this->magrosEncrypter;
    }

    /**
     * @return mixed
     */
    public function getEncryptableAttributes()
    {
        return $this->encryptable;
    }

    /**
     * @param $key
     * @return bool
     */
    public function encryptable($key)
    {
        if(self::$enableEncryption){
            return in_array($key, $this->encryptable);
        }
        return false;
    }

    /**
     * @param $key
     * @return bool
     */
    public function isCamelcase($key)
    {
        return isset($this->camelcase) && is_array($this->camelcase) && in_array($key, $this->camelcase);
    }

    /**
     * Decrypt a value.
     *
     * @param $value
     *
     * @return string
     * @throws \Exception
     */
    public function decryptAttribute($value)
    {
       return $value ? $this->encrypter()->decrypt($value) : '';
    }

    /**
     * @param $value
     * @return string
     */
    public function camelCaseAttribute($value)
    {
        return ucwords($value);
    }

    /**
     * @param $value
     * @return string
     * @throws \Exception
     */
    public function encryptAttribute($value)
    {
        return $value ? $this->encrypter()->encrypt(strtolower($value)) : '';
    }

    /**
     * @param $key
     * @return string
     * @throws \Exception
     */
    public function getAttribute($key)
    {
        $value = parent::getAttribute($key);

        if ($this->encryptable($key)) {
            $value = $this->decryptAttribute($value);
        }
        if ($this->isCamelcase($key)){
            $value = $this->camelCaseAttribute($value);
        }


        return $value;
    }


    /**
     * @param $key
     * @param $value
     * @return mixed
     * @throws \Exception
     */
    public function setAttribute($key, $value)
    {
        if ($this->encryptable($key)) {
            $value = $this->encryptAttribute($value);
        }

        return parent::setAttribute($key, $value);
    }


    /**
     * @return mixed
     * @throws \Exception
     */
    public function getArrayableAttributes()
    {
        $attributes = parent::getArrayableAttributes();

        foreach ($attributes as $key => $attribute) {
            if ($this->encryptable($key)) {
                $attributes[$key] = $this->decryptAttribute($attribute);
            }
            if ($this->isCamelcase($key)){
                $attributes[$key] = $this->camelCaseAttribute($attributes[$key]);
            }
        }

        return $attributes;
    }

    /**
     * @return EncryptableQueryBuilder
     */
    public function newBaseQueryBuilder()
    {
        $connection = $this->getConnection();

It seems like getConnection() must be provided by classes using this trait. How about adding it as abstract method to this trait?

        
        return new EncryptableQueryBuilder($connection, $this);
    }
}