<?php
namespace App\Custom\ProAI\Versioning;

use Illuminate\Support\Str;
use ProAI\Versioning\Versionable as BaseVersionable;

/**
 * Trait Versionable
 *
 * @package App\Custom\ProAI\Versioning
 */
trait Versionable
{
    use BaseVersionable;
    use HasVersionedRelationships;

    /**
     * Qualify the given column name by the model's table.
     *
     * @see \Illuminate\Database\Eloquent\Model::qualifyColumn()
     *
     * @param  string  $column
     *
     * @return string
     */
    public function qualifyColumn($column)
    {
        if (Str::contains($column, '.')) {
            return $column;
        }

        $table = (isset($this->versioned) && in_array($column, $this->versioned)) ? $this->getVersionTable() : $this->getTable();

        return "{$table}.{$column}";
    }

}

<?php

namespace App\Custom\ProAI\Versioning;

/**
 * Trait HasVersionedRelationships
 *
 * @package App\Custom\ProAI\Versioning
 * @mixin \Illuminate\Database\Eloquent\Model
 * @mixin \ProAI\Versioning\Versionable
 */
trait HasVersionedRelationships
{
    /**
     * Define a one-to-one relationship.
     *
     * @see \Illuminate\Database\Eloquent\Concerns\HasRelationships::hasOne()
     *
     * @param  string $related
     * @param  string $foreignKey
     * @param  string $localKey
     *
     * @return \Illuminate\Database\Eloquent\Relations\HasOne
     */
    public function hasOneVersioned($related, $foreignKey = null, $localKey = null)
    {
        $instance = $this->newRelatedInstance($related);

        $foreignKey = $foreignKey ?: $this->getForeignKey();

        $localKey = $localKey ?: $this->getKeyName();

        return $this->newHasOne($instance->newQuery(), $this, $instance->qualifyColumn($foreignKey), $localKey);
    }

    /**
     * Define a one-to-many relationship.
     *
     * @see \Illuminate\Database\Eloquent\Concerns\HasRelationships::hasMany()
     *
     * @param  string  $related
     * @param  string  $foreignKey
     * @param  string  $localKey
     * 
     * @return \Illuminate\Database\Eloquent\Relations\HasMany
     */
    public function hasManyVersioned($related, $foreignKey = null, $localKey = null)
    {
        $instance = $this->newRelatedInstance($related);

        $foreignKey = $foreignKey ?: $this->getForeignKey();

        $localKey = $localKey ?: $this->getKeyName();

        return $this->newHasMany(
            $instance->newQuery(), $this, $instance->qualifyColumn($foreignKey), $localKey
        );
    }

}