22

✏️ Editing: base-db-constraint.php

<?php

namespace Jet_Form_Builder\Db_Queries;

// If this file is called directly, abort.
if ( ! defined( 'WPINC' ) ) {
	die;
}

class Base_Db_Constraint {

const ACTION_CASCADE     = 'CASCADE';
	const ACTION_SET_DEFAULT = 'SET DEFAULT';
	const ACTION_SET_NULL    = 'SET NULL';
	const ACTION_RESTRICT    = 'RESTRICT';

protected $foreign_keys;
	protected $foreign_table;

/** @var Base_Db_Model */
	protected $parent_model;
	protected $parent_keys = array( 'id' );

protected $actions = array();

/**
	 * Here we can check the engine of the table,
	 * if it is not InnoDB - throw \Jet_Form_Builder\Db_Queries\Exceptions\Skip_Exception
	 */
	public function before_create() {
	}

/**
	 * @param array $keys
	 *
	 * @return $this
	 */
	public function set_foreign_keys( array $keys ): Base_Db_Constraint {
		$this->foreign_keys = $keys;

return $this;
	}

/**
	 * @param string $table
	 *
	 * @return $this
	 */
	public function set_foreign_table( string $table ): Base_Db_Constraint {
		$this->foreign_table = $table;

return $this;
	}

public function set_model( Base_Db_Model $model ): Base_Db_Constraint {
		$this->parent_model = $model;

return $this;
	}

public function get_model(): Base_Db_Model {
		return $this->parent_model;
	}

public function get_table(): string {
		return $this->parent_model::table();
	}

public function get_table_name(): string {
		return $this->parent_model::table_name();
	}

public function set_parent_keys( array $keys ): Base_Db_Constraint {
		$this->parent_keys = $keys;

return $this;
	}

public function on_delete( string $action ): Base_Db_Constraint {
		return $this->set_action( 'DELETE', $action );
	}

public function on_update( string $action ): Base_Db_Constraint {
		return $this->set_action( 'UPDATE', $action );
	}

/**
	 * @param string $name
	 * @param string $action
	 *
	 * @return $this
	 */
	public function set_action( string $name, string $action ): Base_Db_Constraint {
		$this->actions[ $name ] = $action;

return $this;
	}

public function get_name(): string {
		$raw_name = "{$this->get_table_name()}__" . implode( '_', $this->foreign_keys ) . "__{$this->foreign_table}";

/**
		 * https://dev.mysql.com/doc/refman/5.6/en/identifier-length.html
		 */
		$length_excess = strlen( $raw_name ) - 64;

if ( $length_excess < 0 ) {
			return $raw_name;
		}

// cut the string
		return substr( $raw_name, 0, -$length_excess );
	}

final public function build_part(): string {
		$constraint  = 'FOREIGN KEY (`' . implode( '` , `', $this->foreign_keys ) . '`) ';
		$constraint .= "REFERENCES {$this->get_table()}(`" . implode( '` , `', $this->parent_keys ) . '`)';

foreach ( $this->actions as $name => $action ) {
			$constraint .= " ON {$name} {$action}";
		}

return $constraint;
	}

final public function build(): string {
		return "CONSTRAINT {$this->get_name()} {$this->build_part()}";
	}

}

MR SKB