<?php

declare(strict_types=1);

namespace MPBC\Data;

if ( ! defined( 'ABSPATH' ) ) {
	exit;
}

class Unit extends Abstract_Entity {
	private string $title = '';

	private int $order = 0;

	private int $variation_id = 0;

	private ?Variation $variation = null;

	public function get_title(): string {
		return $this->title;
	}

	public function set_title( string $title ): void {
		$this->title = $title;
	}

	public function get_order(): int {
		return $this->order;
	}

	public function set_order( int $order ): void {
		$this->order = $order;
	}

	public function get_variation_id(): int {
		return $this->variation_id;
	}

	public function set_variation_id( int $variation_id ): void {
		if ( $variation_id !== $this->variation_id ) {
			$this->variation_id = $variation_id;
			$this->variation = null;
		}
	}

	public function get_variation(): ?Variation {
		if ( is_null( $this->variation ) ) {
			$this->variation = mpbc_get_variation( $this->get_variation_id() );
		}

		return $this->variation;
	}

	public function to_array(): array {
		return parent::to_array()
			+ array(
				'title'        => $this->get_title(),
				'order'        => $this->get_order(),
				'variation_id' => $this->get_variation_id(),
			);
	}
}