Element_Choice_Query

class Element_Choice_Query extends Query {
	/**
	 * Constructor.
	 *
	 * Sets the manager instance and assigns the defaults.
	 *
	 * @since 1.0.0
	 *
	 * @param Element_Choice_Manager $manager The manager instance for the model query.
	 */
	public function __construct( $manager ) {
		parent::__construct( $manager );

		$this->query_var_defaults['orderby']      = array( 'sort' => 'ASC' );
		$this->query_var_defaults['form_id']      = '';
		$this->query_var_defaults['container_id'] = '';
		$this->query_var_defaults['element_id']   = '';
	}

	/**
	 * Parses the SQL join value.
	 *
	 * @since 1.0.0
	 *
	 * @return string Join value for the SQL query.
	 */
	protected function parse_join() {
		$join = parent::parse_join();

		if ( ! empty( $this->query_vars['form_id'] ) || ! empty( $this->query_vars['container_id'] ) ) {
			$table_name = $this->manager->get_table_name();
			$element_table_name = $this->manager->get_parent_manager( 'elements' )->get_table_name();

			$join .= " INNER JOIN %{$element_table_name}% ON ( %{$table_name}%.element_id = %{$element_table_name}%.id )";

			if ( ! empty( $this->query_vars['form_id'] ) ) {
				$container_table_name = $this->manager->get_parent_manager( 'elements' )->get_parent_manager( 'containers' )->get_table_name();

				$join .= " INNER JOIN %{$container_table_name}% ON ( %{$element_table_name}%.container_id = %{$container_table_name}%.id )";
			}
		}

		return $join;
	}

	/**
	 * Parses the SQL where clause.
	 *
	 * @since 1.0.0
	 *
	 * @return array Array with the first element being the array of SQL where clauses and the second
	 *               being the array of arguments for those where clauses.
	 */
	protected function parse_where() {
		list( $where, $args ) = parent::parse_where();

		list( $where, $args ) = $this->parse_default_where_field( $where, $args, 'element_id', 'element_id', '%d', 'absint', true );

		if ( ! empty( $this->query_vars['container_id'] ) ) {
			$table_name = $this->manager->get_table_name();
			$element_table_name = $this->manager->get_parent_manager( 'elements' )->get_table_name();

			list( $where, $args ) = $this->parse_default_where_field( $where, $args, 'container_id', 'container_id', '%d', 'absint', true );
			$where['container_id'] = str_replace( "%{$table_name}%", "%{$element_table_name}%", $where['container_id'] );
		}

		if ( ! empty( $this->query_vars['form_id'] ) ) {
			$table_name = $this->manager->get_table_name();
			$container_table_name = $this->manager->get_parent_manager( 'elements' )->get_parent_manager( 'containers' )->get_table_name();

			list( $where, $args ) = $this->parse_default_where_field( $where, $args, 'form_id', 'form_id', '%d', 'absint', true );
			$where['form_id'] = str_replace( "%{$table_name}%", "%{$container_table_name}%", $where['form_id'] );
		}

		return array( $where, $args );
	}

	/**
	 * Returns the fields that are valid to be used in orderby clauses.
	 *
	 * @since 1.0.0
	 *
	 * @return array Array of valid orderby fields.
	 */
	public function get_valid_orderby_fields() {
		$orderby_fields = parent::get_valid_orderby_fields();

		return array_merge( $orderby_fields, array( 'sort' ) );
	}
}