import CastsAttributes from './CastsAttributes';
import type { Attributes, AttributeKeys } from './HasAttributes';

export default class GuardsAttributes extends CastsAttributes {
    /**
     * The attributes that are mass assignable.
     *
     * @protected
     *
     * @type {string[]}
     */
    protected fillableAttributes = this.fillable;

    /**
     * The attributes that aren't mass assignable
     *
     * @protected
     *
     * @type {string[]}
     */
    protected guardedAttributes = this.guarded;

    /**
     * The attributes that are mass assignable.
     *
     * @type {string[]}
     */
    public get fillable(): string[] {
        return [];
    }

    /**
     * The attributes that are not mass assignable.
     *
     * @type {string[]}
     */
    public get guarded(): string[] {
        return ['*'];
    }

    /**
     * Get the guarded attributes for the model.
     *
     * @return {string[]}
     */
    public getGuarded(): string[] {
        return this.guardedAttributes;
    }

    /**
     * Get the fillable attributes for the model.
     *
     * @return {string[]}
     */
    public getFillable(): string[] {
        return this.fillableAttributes;
    }

    /**
     * Merge new fillable attributes with existing fillable attributes on the model.
     *
     * @param {string[]} fillable
     *
     * @return {this}
     */
    public mergeFillable(fillable: (AttributeKeys<this> | string)[]): this;
    public mergeFillable(fillable: string[]): this {
        this.fillableAttributes = [...this.getFillable(), ...fillable] as string[];

        return this;
    }

    /**
     * Merge new guarded attributes with existing guarded attributes on the model.
     *
     * @param {string[]} guarded
     *
     * @return {this}
     */
    public mergeGuarded(guarded: (AttributeKeys<this> | string)[]): this;
    public mergeGuarded(guarded: string[]): this {
        this.guardedAttributes = [...this.getGuarded(), ...guarded];

        return this;
    }

    /**
     * Set the fillable attributes for the model.
     *
     * @param {string[]} fillable
     *
     * @return {this}
     */
    public setFillable(fillable: (AttributeKeys<this> | string)[]): this;
    public setFillable(fillable: string[]): this {
        this.fillableAttributes = fillable;

        return this;
    }

    /**
     * Set the guarded attributes for the model.
     *
     * @param {string[]} guarded
     *
     * @return {this}
     */
    public setGuarded(guarded: (AttributeKeys<this> | string)[]): this;
    public setGuarded(guarded: string[]): this {
        this.guardedAttributes = guarded;

        return this;
    }

    /**
     * Determine if the given attribute may be mass assignable.
     *
     * @param {string} key
     */
    public isFillable(key: AttributeKeys<this> | string): boolean;
    public isFillable(key: string): boolean {
        return this.getFillable().includes(key) || this.getFillable().includes('*');
    }

    /**
     * Determine if the given attribute may not be mass assignable.
     *
     * @param {string} key
     */
    public isGuarded(key: AttributeKeys<this> | string): boolean;
    public isGuarded(key: string): boolean {
        // if key is defined in both guarded and fillable, then fillable takes priority.
        return (this.getGuarded().includes(key) || this.getGuarded().includes('*')) && !this.isFillable(key);
    }

    /**
     * Get the fillable attributes from the given object.
     *
     * @param {object} attributes
     *
     * @return {object}
     */
    protected getFillableFromObject(attributes: Attributes): Partial<Attributes> {
        const fillable: Attributes = {};
        if (this.getFillable().includes('*')) {
            return attributes;
        }

        for (const [name, value] of Object.entries(attributes)) {
            if (!this.isGuarded(name)) {
                fillable[name] = value;
            }
        }

        return fillable;
    }
}